From 1c46d68abaf08890e10386b3af674233b4dda2d3 Mon Sep 17 00:00:00 2001 From: Lunny Xiao Date: Sat, 28 Jun 2014 23:56:41 +0800 Subject: [PATCH] bug fixed for message tag --- cmd/serve.go | 25 +++++++++++- cmd/update.go | 27 +++++++++---- models/models.go | 3 +- models/update.go | 76 ++++++++++++++++++++++++----------- modules/middleware/context.go | 2 + modules/middleware/repo.go | 7 ++-- 6 files changed, 105 insertions(+), 35 deletions(-) diff --git a/cmd/serve.go b/cmd/serve.go index 2a76da793..351553b91 100644 --- a/cmd/serve.go +++ b/cmd/serve.go @@ -13,6 +13,7 @@ import ( "strings" "github.com/codegangsta/cli" + "github.com/satori/go.uuid" "github.com/gogits/gogs/models" "github.com/gogits/gogs/modules/log" @@ -165,7 +166,9 @@ func runServ(k *cli.Context) { return } - models.SetRepoEnvs(user.Id, user.Name, repoName, repoUserName) + //models.SetRepoEnvs(user.Id, user.Name, repoName, repoUserName) + uuid := uuid.NewV4().String() + os.Setenv("uuid", uuid) gitcmd := exec.Command(verb, repoPath) gitcmd.Dir = setting.RepoRootPath @@ -177,4 +180,24 @@ func runServ(k *cli.Context) { println("Gogs: internal error:", err) log.GitLogger.Fatal("Fail to execute git command: %v", err) } + + if isWrite { + tasks, err := models.GetUpdateTasksByUuid(uuid) + if err != nil { + log.GitLogger.Fatal("Fail to get update task: %v", err) + } + + for _, task := range tasks { + err = models.Update(task.RefName, task.OldCommitId, task.NewCommitId, + user.Name, repoUserName, repoName, user.Id) + if err != nil { + log.GitLogger.Fatal("Fail to update: %v", err) + } + } + + err = models.DelUpdateTasksByUuid(uuid) + if err != nil { + log.GitLogger.Fatal("Fail to del update task: %v", err) + } + } } diff --git a/cmd/update.go b/cmd/update.go index c030b6cfb..551fd09ef 100644 --- a/cmd/update.go +++ b/cmd/update.go @@ -6,10 +6,8 @@ package cmd import ( "os" - "strconv" "github.com/codegangsta/cli" - "github.com/gogits/gogs/models" "github.com/gogits/gogs/modules/log" ) @@ -37,12 +35,27 @@ func runUpdate(c *cli.Context) { log.GitLogger.Fatal("refName is empty, shouldn't use") } - userName := os.Getenv("userName") - userId, _ := strconv.ParseInt(os.Getenv("userId"), 10, 64) - repoUserName := os.Getenv("repoUserName") - repoName := os.Getenv("repoName") + //userName := os.Getenv("userName") + //userId, _ := strconv.ParseInt(os.Getenv("userId"), 10, 64) + //repoUserName := os.Getenv("repoUserName") + //repoName := os.Getenv("repoName") + uuid := os.Getenv("uuid") + + task := models.UpdateTask{ + Uuid: uuid, + RefName: args[0], + OldCommitId: args[1], + NewCommitId: args[2], + } + + log.GitLogger.Error("%v", task) - if err := models.Update(args[0], args[1], args[2], userName, repoUserName, repoName, userId); err != nil { + if err := models.AddUpdateTask(&task); err != nil { log.GitLogger.Fatal(err.Error()) } + + /*if err := models.Update(args[0], args[1], args[2], userName, repoUserName, repoName, userId); err != nil { + log.GitLogger.Fatal(err.Error()) + }*/ + //setEnvs(args[0], args[1], args[2], userName, repoUserName, repoName, userId) } diff --git a/models/models.go b/models/models.go index 4e65c00bc..070784f13 100644 --- a/models/models.go +++ b/models/models.go @@ -35,7 +35,8 @@ func init() { tables = append(tables, new(User), new(PublicKey), new(Repository), new(Watch), new(Action), new(Access), new(Issue), new(Comment), new(Oauth2), new(Follow), new(Mirror), new(Release), new(LoginSource), new(Webhook), new(IssueUser), - new(Milestone), new(Label), new(HookTask), new(Team), new(OrgUser), new(TeamUser)) + new(Milestone), new(Label), new(HookTask), new(Team), new(OrgUser), new(TeamUser), + new(UpdateTask)) } func LoadModelsConfig() { diff --git a/models/update.go b/models/update.go index 6702ad3b4..cf7f5d2a7 100644 --- a/models/update.go +++ b/models/update.go @@ -16,6 +16,36 @@ import ( "github.com/gogits/gogs/modules/log" ) +type UpdateTask struct { + Id int64 + Uuid string `xorm:"index"` + RefName string + OldCommitId string + NewCommitId string +} + +func AddUpdateTask(task *UpdateTask) error { + _, err := x.Insert(task) + return err +} + +func GetUpdateTasksByUuid(uuid string) ([]*UpdateTask, error) { + task := &UpdateTask{ + Uuid: uuid, + } + tasks := make([]*UpdateTask, 0) + err := x.Find(&tasks, task) + if err != nil { + return nil, err + } + return tasks, nil +} + +func DelUpdateTasksByUuid(uuid string) error { + _, err := x.Delete(&UpdateTask{Uuid: uuid}) + return err +} + func Update(refName, oldCommitId, newCommitId, userName, repoUserName, repoName string, userId int64) error { //fmt.Println(refName, oldCommitId, newCommitId) //fmt.Println(userName, repoUserName, repoName) @@ -42,29 +72,6 @@ func Update(refName, oldCommitId, newCommitId, userName, repoUserName, repoName return fmt.Errorf("runUpdate.Open repoId: %v", err) } - newCommit, err := repo.GetCommit(newCommitId) - if err != nil { - return fmt.Errorf("runUpdate GetCommit of newCommitId: %v", err) - } - - var l *list.List - // if a new branch - if isNew { - l, err = newCommit.CommitsBefore() - if err != nil { - return fmt.Errorf("Find CommitsBefore erro: %v", err) - } - } else { - l, err = newCommit.CommitsBeforeUntil(oldCommitId) - if err != nil { - return fmt.Errorf("Find CommitsBeforeUntil erro: %v", err) - } - } - - if err != nil { - return fmt.Errorf("runUpdate.Commit repoId: %v", err) - } - ru, err := GetUserByName(repoUserName) if err != nil { return fmt.Errorf("runUpdate.GetUserByName: %v", err) @@ -103,6 +110,29 @@ func Update(refName, oldCommitId, newCommitId, userName, repoUserName, repoName return err } + newCommit, err := repo.GetCommit(newCommitId) + if err != nil { + return fmt.Errorf("runUpdate GetCommit of newCommitId: %v", err) + } + + var l *list.List + // if a new branch + if isNew { + l, err = newCommit.CommitsBefore() + if err != nil { + return fmt.Errorf("Find CommitsBefore erro: %v", err) + } + } else { + l, err = newCommit.CommitsBeforeUntil(oldCommitId) + if err != nil { + return fmt.Errorf("Find CommitsBeforeUntil erro: %v", err) + } + } + + if err != nil { + return fmt.Errorf("runUpdate.Commit repoId: %v", err) + } + // if commits push commits := make([]*base.PushCommit, 0) var maxCommits = 3 diff --git a/modules/middleware/context.go b/modules/middleware/context.go index 45f0140a2..8e7ac4209 100644 --- a/modules/middleware/context.go +++ b/modules/middleware/context.go @@ -56,8 +56,10 @@ type Context struct { Repository *models.Repository Owner *models.User Commit *git.Commit + Tag *git.Tag GitRepo *git.Repository BranchName string + TagName string CommitId string RepoLink string CloneLink struct { diff --git a/modules/middleware/repo.go b/modules/middleware/repo.go index 0c6402755..7ba211c71 100644 --- a/modules/middleware/repo.go +++ b/modules/middleware/repo.go @@ -185,16 +185,16 @@ func RepoAssignment(redirect bool, args ...bool) martini.Handler { ctx.Repo.CommitId = ctx.Repo.Commit.Id.String() } else if gitRepo.IsTagExist(refName) { - ctx.Repo.IsBranch = true + ctx.Repo.IsTag = true ctx.Repo.BranchName = refName - ctx.Repo.Commit, err = gitRepo.GetCommitOfTag(refName) + ctx.Repo.Tag, err = gitRepo.GetTag(refName) if err != nil { ctx.Handle(404, "RepoAssignment invalid tag", nil) return } + ctx.Repo.Commit, _ = ctx.Repo.Tag.Commit() ctx.Repo.CommitId = ctx.Repo.Commit.Id.String() - } else if len(refName) == 40 { ctx.Repo.IsCommit = true ctx.Repo.CommitId = refName @@ -244,6 +244,7 @@ func RepoAssignment(redirect bool, args ...bool) martini.Handler { } ctx.Data["BranchName"] = ctx.Repo.BranchName + ctx.Data["TagName"] = ctx.Repo.TagName brs, err := ctx.Repo.GitRepo.GetBranches() if err != nil { log.Error("RepoAssignment(GetBranches): %v", err)