forgejo/update.go

147 lines
3.2 KiB
Go
Raw Normal View History

2014-03-16 07:53:41 +01:00
// Copyright 2014 The Gogs Authors. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
2014-03-16 05:18:34 +01:00
package main
2014-03-25 17:00:36 +01:00
import (
"os"
2014-03-26 06:21:09 +01:00
"os/exec"
"strings"
"strconv"
"container/list"
2014-03-25 17:00:36 +01:00
"github.com/codegangsta/cli"
2014-03-25 17:05:09 +01:00
//"github.com/gogits/gogs/modules/log"
2014-03-26 06:21:09 +01:00
"github.com/gogits/gogs/models"
"github.com/gogits/gogs/modules/base"
2014-03-25 17:05:09 +01:00
"github.com/qiniu/log"
2014-03-26 06:21:09 +01:00
"github.com/gogits/git"
2014-03-25 17:00:36 +01:00
)
2014-03-16 05:18:34 +01:00
var CmdUpdate = cli.Command{
Name: "update",
Usage: "This command just should be called by ssh shell",
Description: `
gogs serv provide access auth for repositories`,
Action: runUpdate,
Flags: []cli.Flag{},
}
2014-03-18 14:58:58 +01:00
// for command: ./gogs update
2014-03-25 17:00:36 +01:00
func runUpdate(c *cli.Context) {
2014-03-25 17:05:09 +01:00
//level := "0"
//os.MkdirAll("log", os.ModePerm)
//log.NewLogger(10000, "file", fmt.Sprintf(`{"level":%s,"filename":"%s"}`, level, "log/serv.log"))
//log.Info("start update logging...")
2014-03-23 09:31:13 +01:00
2014-03-25 17:05:09 +01:00
w, _ := os.Create("update.log")
defer w.Close()
log.SetOutput(w)
2014-03-26 06:21:09 +01:00
args := c.Args()
if len(args) != 3 {
log.Error("received less 3 parameters")
return
2014-03-25 17:05:09 +01:00
}
2014-03-26 06:21:09 +01:00
refName := args[0]
oldCommitId := args[1]
newCommitId := args[2]
2014-03-25 17:16:13 +01:00
userName := os.Getenv("userName")
2014-03-26 06:21:09 +01:00
userId := os.Getenv("userId")
2014-03-25 17:16:13 +01:00
//repoId := os.Getenv("repoId")
2014-03-26 06:21:09 +01:00
repoName := os.Getenv("repoName")
2014-03-16 05:18:34 +01:00
2014-03-25 17:16:13 +01:00
log.Info("username", userName)
2014-03-26 06:31:27 +01:00
log.Info("repoName", repoName)
2014-03-26 06:21:09 +01:00
f := models.RepoPath(userName, repoName)
2014-03-26 06:31:27 +01:00
log.Debug("f", f)
2014-03-26 06:21:09 +01:00
gitUpdate := exec.Command("git", "update-server-info")
gitUpdate.Dir = f
gitUpdate.Run()
2014-03-16 05:18:34 +01:00
repo, err := git.OpenRepository(f)
if err != nil {
2014-03-23 09:31:13 +01:00
log.Error("runUpdate.Open repoId: %v", err)
2014-03-16 05:18:34 +01:00
return
}
2014-03-26 06:21:09 +01:00
ref, err := repo.LookupReference(refName)
2014-03-16 05:18:34 +01:00
if err != nil {
2014-03-23 09:31:13 +01:00
log.Error("runUpdate.Ref repoId: %v", err)
2014-03-16 05:18:34 +01:00
return
}
2014-03-26 06:21:09 +01:00
oldOid, err := git.NewOidFromString(oldCommitId)
if err != nil {
log.Error("runUpdate.Ref repoId: %v", err)
return
}
oldCommit, err := repo.LookupCommit(oldOid)
if err != nil {
log.Error("runUpdate.Ref repoId: %v", err)
return
}
newOid, err := git.NewOidFromString(newCommitId)
if err != nil {
log.Error("runUpdate.Ref repoId: %v", err)
return
}
newCommit, err := repo.LookupCommit(newOid)
if err != nil {
log.Error("runUpdate.Ref repoId: %v", err)
return
}
var l *list.List
// if a new branch
if strings.HasPrefix(oldCommitId, "0000000") {
l, err = ref.AllCommits()
} else {
l = ref.CommitsBetween(newCommit, oldCommit)
}
2014-03-16 05:18:34 +01:00
if err != nil {
2014-03-23 09:31:13 +01:00
log.Error("runUpdate.Commit repoId: %v", err)
2014-03-16 05:18:34 +01:00
return
}
2014-03-16 16:02:59 +01:00
2014-03-16 07:53:41 +01:00
sUserId, err := strconv.Atoi(userId)
2014-03-16 05:18:34 +01:00
if err != nil {
2014-03-16 07:53:41 +01:00
log.Error("runUpdate.Parse userId: %v", err)
return
}
2014-03-26 06:21:09 +01:00
repos, err := models.GetRepositoryByName(int64(sUserId), repoName)
2014-03-16 07:53:41 +01:00
if err != nil {
2014-03-26 06:21:09 +01:00
log.Error("runUpdate.GetRepositoryByName userId: %v", err)
2014-03-16 07:53:41 +01:00
return
}
2014-03-26 06:21:09 +01:00
2014-03-16 16:02:59 +01:00
commits := make([][]string, 0)
2014-03-26 06:21:09 +01:00
var maxCommits = 3
for e := l.Front(); e != nil; e = e.Next() {
commit := e.Value.(*git.Commit)
commits = append(commits, []string{commit.Id().String(), commit.Message()})
if len(commits) >= maxCommits {
break
}
}
//commits = append(commits, []string{lastCommit.Id().String(), lastCommit.Message()})
2014-03-16 07:53:41 +01:00
if err = models.CommitRepoAction(int64(sUserId), userName,
2014-03-26 06:21:09 +01:00
repos.Id, repoName, refName, &base.PushCommits{l.Len(), commits}); err != nil {
2014-03-16 07:53:41 +01:00
log.Error("runUpdate.models.CommitRepoAction: %v", err)
2014-03-26 06:21:09 +01:00
}
2014-03-16 05:18:34 +01:00
}