bench-forgejo/update.go

165 lines
3.7 KiB
Go
Raw Normal View History

2014-03-16 12:23:41 +05:30
// 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 09:48:34 +05:30
package main
2014-03-25 21:30:36 +05:30
import (
2014-03-26 18:52:08 +05:30
"container/list"
"os"
"os/exec"
2014-03-31 18:56:15 +05:30
"path"
2014-03-26 18:52:08 +05:30
"strconv"
"strings"
"github.com/codegangsta/cli"
2014-04-06 22:30:20 +05:30
qlog "github.com/qiniu/log"
2014-03-26 18:52:08 +05:30
"github.com/gogits/git"
"github.com/gogits/gogs/models"
"github.com/gogits/gogs/modules/base"
2014-03-25 21:30:36 +05:30
)
2014-03-16 09:48:34 +05:30
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-31 18:56:15 +05:30
func newUpdateLogger(execDir string) {
logPath := execDir + "/log/update.log"
os.MkdirAll(path.Dir(logPath), os.ModePerm)
2014-04-06 23:11:58 +05:30
f, err := os.OpenFile(logPath, os.O_WRONLY|os.O_APPEND|os.O_CREATE, os.ModePerm)
2014-04-06 22:30:20 +05:30
if err != nil {
qlog.Fatal(err)
}
qlog.SetOutput(f)
qlog.Info("Start logging update...")
2014-03-31 18:56:15 +05:30
}
2014-04-10 20:32:08 +05:30
func update(refName, oldCommitId, newCommitId string) {
2014-03-28 08:18:36 +05:30
isNew := strings.HasPrefix(oldCommitId, "0000000")
if isNew &&
strings.HasPrefix(newCommitId, "0000000") {
2014-04-06 22:30:20 +05:30
qlog.Fatal("old rev and new rev both 000000")
2014-03-28 08:18:36 +05:30
}
2014-03-25 21:46:13 +05:30
userName := os.Getenv("userName")
2014-03-26 10:51:09 +05:30
userId := os.Getenv("userId")
2014-03-25 21:46:13 +05:30
//repoId := os.Getenv("repoId")
2014-03-26 10:51:09 +05:30
repoName := os.Getenv("repoName")
2014-03-16 09:48:34 +05:30
2014-03-26 10:51:09 +05:30
f := models.RepoPath(userName, repoName)
gitUpdate := exec.Command("git", "update-server-info")
gitUpdate.Dir = f
gitUpdate.Run()
2014-03-16 09:48:34 +05:30
repo, err := git.OpenRepository(f)
if err != nil {
2014-04-06 22:30:20 +05:30
qlog.Fatalf("runUpdate.Open repoId: %v", err)
2014-03-16 09:48:34 +05:30
}
2014-03-26 10:51:09 +05:30
newOid, err := git.NewOidFromString(newCommitId)
if err != nil {
2014-04-06 22:30:20 +05:30
qlog.Fatalf("runUpdate.Ref repoId: %v", err)
2014-03-26 10:51:09 +05:30
}
newCommit, err := repo.LookupCommit(newOid)
if err != nil {
2014-04-06 22:30:20 +05:30
qlog.Fatalf("runUpdate.Ref repoId: %v", err)
2014-03-26 10:51:09 +05:30
}
var l *list.List
// if a new branch
2014-03-28 08:18:36 +05:30
if isNew {
l, err = repo.CommitsBefore(newCommit.Id())
if err != nil {
2014-04-06 22:30:20 +05:30
qlog.Fatalf("Find CommitsBefore erro:", err)
2014-03-28 08:18:36 +05:30
}
2014-03-26 10:51:09 +05:30
} else {
2014-03-28 08:18:36 +05:30
oldOid, err := git.NewOidFromString(oldCommitId)
if err != nil {
2014-04-06 22:30:20 +05:30
qlog.Fatalf("runUpdate.Ref repoId: %v", err)
2014-03-28 08:18:36 +05:30
}
oldCommit, err := repo.LookupCommit(oldOid)
if err != nil {
2014-04-06 22:30:20 +05:30
qlog.Fatalf("runUpdate.Ref repoId: %v", err)
2014-03-28 08:18:36 +05:30
}
l = repo.CommitsBetween(newCommit, oldCommit)
2014-03-26 10:51:09 +05:30
}
2014-03-16 09:48:34 +05:30
if err != nil {
2014-04-06 22:30:20 +05:30
qlog.Fatalf("runUpdate.Commit repoId: %v", err)
2014-03-16 09:48:34 +05:30
}
2014-03-16 20:32:59 +05:30
2014-03-16 12:23:41 +05:30
sUserId, err := strconv.Atoi(userId)
2014-03-16 09:48:34 +05:30
if err != nil {
2014-04-06 22:30:20 +05:30
qlog.Fatalf("runUpdate.Parse userId: %v", err)
2014-03-16 12:23:41 +05:30
}
2014-03-26 10:51:09 +05:30
repos, err := models.GetRepositoryByName(int64(sUserId), repoName)
2014-03-16 12:23:41 +05:30
if err != nil {
2014-04-06 22:30:20 +05:30
qlog.Fatalf("runUpdate.GetRepositoryByName userId: %v", err)
2014-03-16 12:23:41 +05:30
}
2014-03-26 10:51:09 +05:30
2014-03-29 16:59:52 +05:30
commits := make([]*base.PushCommit, 0)
2014-03-26 10:51:09 +05:30
var maxCommits = 3
2014-03-29 17:29:02 +05:30
var actEmail string
2014-03-26 10:51:09 +05:30
for e := l.Front(); e != nil; e = e.Next() {
commit := e.Value.(*git.Commit)
2014-03-29 17:29:02 +05:30
if actEmail == "" {
actEmail = commit.Committer.Email
}
2014-03-29 16:59:52 +05:30
commits = append(commits,
&base.PushCommit{commit.Id().String(),
commit.Message(),
commit.Author.Email,
commit.Author.Name})
2014-03-26 10:51:09 +05:30
if len(commits) >= maxCommits {
break
}
}
//commits = append(commits, []string{lastCommit.Id().String(), lastCommit.Message()})
2014-03-29 17:29:02 +05:30
if err = models.CommitRepoAction(int64(sUserId), userName, actEmail,
2014-03-26 11:15:55 +05:30
repos.Id, repoName, git.BranchName(refName), &base.PushCommits{l.Len(), commits}); err != nil {
2014-04-06 22:30:20 +05:30
qlog.Fatalf("runUpdate.models.CommitRepoAction: %v", err)
2014-03-26 10:51:09 +05:30
}
2014-03-16 09:48:34 +05:30
}
2014-04-10 20:32:08 +05:30
// for command: ./gogs update
func runUpdate(c *cli.Context) {
execDir, _ := base.ExecDir()
newUpdateLogger(execDir)
base.NewConfigContext()
models.LoadModelsConfig()
if models.UseSQLite3 {
os.Chdir(execDir)
}
models.SetEngine()
args := c.Args()
if len(args) != 3 {
qlog.Fatal("received less 3 parameters")
}
refName := args[0]
if refName == "" {
qlog.Fatal("refName is empty, shouldn't use")
}
oldCommitId := args[1]
newCommitId := args[2]
update(refName, oldCommitId, newCommitId)
}