bench-forgejo/cmd/update.go

84 lines
2.1 KiB
Go
Raw Normal View History

2014-04-10 23:50:58 +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-05-02 06:51:46 +05:30
package cmd
2014-04-10 23:50:58 +05:30
import (
"os"
"strconv"
"strings"
2014-04-10 23:50:58 +05:30
2016-08-30 17:27:58 +05:30
"github.com/urfave/cli"
2014-07-26 09:54:27 +05:30
"code.gitea.io/git"
"code.gitea.io/gitea/models"
"code.gitea.io/gitea/modules/log"
"code.gitea.io/gitea/modules/setting"
2014-04-10 23:50:58 +05:30
)
2016-11-04 17:12:18 +05:30
// CmdUpdate represents the available update sub-command.
2014-04-10 23:50:58 +05:30
var CmdUpdate = cli.Command{
2014-05-05 10:25:17 +05:30
Name: "update",
2016-02-22 08:25:59 +05:30
Usage: "This command should only be called by Git hook",
2014-05-05 10:25:17 +05:30
Description: `Update get pushed info and insert into database`,
Action: runUpdate,
Flags: []cli.Flag{
cli.StringFlag{
Name: "config, c",
Value: "custom/conf/app.ini",
Usage: "Custom configuration file path",
},
},
2014-04-10 23:50:58 +05:30
}
func runUpdate(c *cli.Context) error {
if c.IsSet("config") {
setting.CustomConf = c.String("config")
}
2016-02-16 09:41:22 +05:30
2016-02-18 01:47:52 +05:30
setup("update.log")
2016-02-16 09:41:22 +05:30
if len(os.Getenv("SSH_ORIGINAL_COMMAND")) == 0 {
log.GitLogger.Trace("SSH_ORIGINAL_COMMAND is empty")
return nil
2014-04-11 07:57:13 +05:30
}
2014-04-10 23:50:58 +05:30
args := c.Args()
if len(args) != 3 {
2016-02-16 09:41:22 +05:30
log.GitLogger.Fatal(2, "Arguments received are not equal to three")
} else if len(args[0]) == 0 {
log.GitLogger.Fatal(2, "First argument 'refName' is empty, shouldn't use")
2014-04-10 23:50:58 +05:30
}
// protected branch check
branchName := strings.TrimPrefix(args[0], git.BranchPrefix)
repoID, _ := strconv.ParseInt(os.Getenv(models.ProtectedBranchRepoID), 10, 64)
log.GitLogger.Trace("pushing to %d %v", repoID, branchName)
accessMode := models.ParseAccessMode(os.Getenv(models.ProtectedBranchAccessMode))
// skip admin or owner AccessMode
if accessMode == models.AccessModeWrite {
protectBranch, err := models.GetProtectedBranchBy(repoID, branchName)
if err != nil {
log.GitLogger.Fatal(2, "retrieve protected branches information failed")
}
if protectBranch != nil {
log.GitLogger.Fatal(2, "protected branches can not be pushed to")
}
}
2014-06-28 21:26:41 +05:30
task := models.UpdateTask{
UUID: os.Getenv("GITEA_UUID"),
2014-06-28 21:26:41 +05:30
RefName: args[0],
OldCommitID: args[1],
NewCommitID: args[2],
2014-06-28 21:26:41 +05:30
}
if err := models.AddUpdateTask(&task); err != nil {
log.GitLogger.Fatal(2, "AddUpdateTask: %v", err)
2014-06-24 01:52:34 +05:30
}
return nil
2014-04-10 23:50:58 +05:30
}