update.go 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  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 Git hook",
  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) error {
  22. if c.IsSet("config") {
  23. setting.CustomConf = c.String("config")
  24. }
  25. setup("update.log")
  26. if len(os.Getenv("SSH_ORIGINAL_COMMAND")) == 0 {
  27. log.GitLogger.Trace("SSH_ORIGINAL_COMMAND is empty")
  28. return nil
  29. }
  30. args := c.Args()
  31. if len(args) != 3 {
  32. log.GitLogger.Fatal(2, "Arguments received are not equal to three")
  33. } else if len(args[0]) == 0 {
  34. log.GitLogger.Fatal(2, "First argument '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. return nil
  46. }