update.go 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  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. 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. task := models.UpdateTask{
  37. UUID: os.Getenv("uuid"),
  38. RefName: args[0],
  39. OldCommitID: args[1],
  40. NewCommitID: args[2],
  41. }
  42. if err := models.AddUpdateTask(&task); err != nil {
  43. log.GitLogger.Fatal(2, "AddUpdateTask: %v", err)
  44. }
  45. }