update.go 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. // Copyright 2014 The Gogs Authors. All rights reserved.
  2. // Use of this source code is governed by a MIT-style
  3. // license that can be found in the LICENSE file.
  4. package cmd
  5. import (
  6. "os"
  7. "github.com/codegangsta/cli"
  8. "github.com/gogits/gogs/models"
  9. "github.com/gogits/gogs/modules/log"
  10. "github.com/gogits/gogs/modules/setting"
  11. )
  12. var CmdUpdate = cli.Command{
  13. Name: "update",
  14. Usage: "This command should only be called by SSH shell",
  15. Description: `Update get pushed info and insert into database`,
  16. Action: runUpdate,
  17. Flags: []cli.Flag{
  18. cli.StringFlag{"config, c", "custom/conf/app.ini", "Custom configuration file path", ""},
  19. },
  20. }
  21. func runUpdate(c *cli.Context) {
  22. if c.IsSet("config") {
  23. setting.CustomConf = c.String("config")
  24. }
  25. cmd := os.Getenv("SSH_ORIGINAL_COMMAND")
  26. if cmd == "" {
  27. return
  28. }
  29. setup("update.log")
  30. args := c.Args()
  31. if len(args) != 3 {
  32. log.GitLogger.Fatal(2, "received less 3 parameters")
  33. } else if args[0] == "" {
  34. log.GitLogger.Fatal(2, "refName is empty, shouldn't use")
  35. }
  36. uuid := os.Getenv("uuid")
  37. task := models.UpdateTask{
  38. Uuid: uuid,
  39. RefName: args[0],
  40. OldCommitId: args[1],
  41. NewCommitId: args[2],
  42. }
  43. if err := models.AddUpdateTask(&task); err != nil {
  44. log.GitLogger.Fatal(2, err.Error())
  45. }
  46. }