forgejo/cmd/update.go

86 lines
1.8 KiB
Go
Raw Normal View History

2014-04-10 18:20:58 +00: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-05-02 01:21:46 +00:00
package cmd
2014-04-10 18:20:58 +00:00
import (
"os"
"path"
"strconv"
"github.com/codegangsta/cli"
2014-04-21 10:37:57 +00:00
qlog "github.com/qiniu/log"
2014-04-10 18:20:58 +00:00
"github.com/gogits/gogs/models"
"github.com/gogits/gogs/modules/base"
)
var CmdUpdate = cli.Command{
Name: "update",
2014-05-02 01:21:46 +00:00
Usage: "This command should only be called by SSH shell",
2014-04-10 18:20:58 +00:00
Description: `
2014-05-02 01:21:46 +00:00
Update get pushed info and insert into database`,
2014-04-10 18:20:58 +00:00
Action: runUpdate,
Flags: []cli.Flag{},
}
func newUpdateLogger(execDir string) {
logPath := execDir + "/log/update.log"
os.MkdirAll(path.Dir(logPath), os.ModePerm)
f, err := os.OpenFile(logPath, os.O_WRONLY|os.O_APPEND|os.O_CREATE, os.ModePerm)
if err != nil {
qlog.Fatal(err)
}
qlog.SetOutput(f)
qlog.Info("Start logging update...")
}
2014-04-11 02:27:13 +00:00
func updateEnv(refName, oldCommitId, newCommitId string) {
os.Setenv("refName", refName)
os.Setenv("oldCommitId", oldCommitId)
os.Setenv("newCommitId", newCommitId)
qlog.Error("set envs:", refName, oldCommitId, newCommitId)
2014-04-10 18:20:58 +00:00
}
// for command: ./gogs update
func runUpdate(c *cli.Context) {
2014-04-11 02:27:13 +00:00
cmd := os.Getenv("SSH_ORIGINAL_COMMAND")
if cmd == "" {
return
}
2014-04-10 18:20:58 +00:00
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")
}
2014-04-11 02:27:13 +00:00
if args[0] == "" {
2014-04-10 18:20:58 +00:00
qlog.Fatal("refName is empty, shouldn't use")
}
2014-04-11 02:27:13 +00:00
//updateEnv(args[0], args[1], args[2])
userName := os.Getenv("userName")
userId := os.Getenv("userId")
iUserId, _ := strconv.ParseInt(userId, 10, 64)
//repoId := os.Getenv("repoId")
repoName := os.Getenv("repoName")
models.Update(args[0], args[1], args[2], userName, repoName, iUserId)
2014-04-10 18:20:58 +00:00
}