mirror of https://github.com/go-gitea/gitea
Move commit repo action from models to repofiles package (#7645)
* move commit repo action from models to repofiles package * fix unit testspull/7674/head
Lunny Xiao
5 years ago
committed by
GitHub
parent
4d643a59db
commit
e7d4895732
@ -0,0 +1,211 @@ |
|||||||
|
// Copyright 2019 The Gitea Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by a MIT-style
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package repofiles |
||||||
|
|
||||||
|
import ( |
||||||
|
"encoding/json" |
||||||
|
"fmt" |
||||||
|
"strings" |
||||||
|
|
||||||
|
"code.gitea.io/gitea/models" |
||||||
|
"code.gitea.io/gitea/modules/git" |
||||||
|
"code.gitea.io/gitea/modules/log" |
||||||
|
"code.gitea.io/gitea/modules/setting" |
||||||
|
api "code.gitea.io/gitea/modules/structs" |
||||||
|
) |
||||||
|
|
||||||
|
// CommitRepoActionOptions represent options of a new commit action.
|
||||||
|
type CommitRepoActionOptions struct { |
||||||
|
PusherName string |
||||||
|
RepoOwnerID int64 |
||||||
|
RepoName string |
||||||
|
RefFullName string |
||||||
|
OldCommitID string |
||||||
|
NewCommitID string |
||||||
|
Commits *models.PushCommits |
||||||
|
} |
||||||
|
|
||||||
|
// CommitRepoAction adds new commit action to the repository, and prepare
|
||||||
|
// corresponding webhooks.
|
||||||
|
func CommitRepoAction(opts CommitRepoActionOptions) error { |
||||||
|
pusher, err := models.GetUserByName(opts.PusherName) |
||||||
|
if err != nil { |
||||||
|
return fmt.Errorf("GetUserByName [%s]: %v", opts.PusherName, err) |
||||||
|
} |
||||||
|
|
||||||
|
repo, err := models.GetRepositoryByName(opts.RepoOwnerID, opts.RepoName) |
||||||
|
if err != nil { |
||||||
|
return fmt.Errorf("GetRepositoryByName [owner_id: %d, name: %s]: %v", opts.RepoOwnerID, opts.RepoName, err) |
||||||
|
} |
||||||
|
|
||||||
|
refName := git.RefEndName(opts.RefFullName) |
||||||
|
|
||||||
|
// Change default branch and empty status only if pushed ref is non-empty branch.
|
||||||
|
if repo.IsEmpty && opts.NewCommitID != git.EmptySHA && strings.HasPrefix(opts.RefFullName, git.BranchPrefix) { |
||||||
|
repo.DefaultBranch = refName |
||||||
|
repo.IsEmpty = false |
||||||
|
if refName != "master" { |
||||||
|
gitRepo, err := git.OpenRepository(repo.RepoPath()) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
if err := gitRepo.SetDefaultBranch(repo.DefaultBranch); err != nil { |
||||||
|
if !git.IsErrUnsupportedVersion(err) { |
||||||
|
return err |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Change repository empty status and update last updated time.
|
||||||
|
if err = models.UpdateRepository(repo, false); err != nil { |
||||||
|
return fmt.Errorf("UpdateRepository: %v", err) |
||||||
|
} |
||||||
|
|
||||||
|
isNewBranch := false |
||||||
|
opType := models.ActionCommitRepo |
||||||
|
// Check it's tag push or branch.
|
||||||
|
if strings.HasPrefix(opts.RefFullName, git.TagPrefix) { |
||||||
|
opType = models.ActionPushTag |
||||||
|
if opts.NewCommitID == git.EmptySHA { |
||||||
|
opType = models.ActionDeleteTag |
||||||
|
} |
||||||
|
opts.Commits = &models.PushCommits{} |
||||||
|
} else if opts.NewCommitID == git.EmptySHA { |
||||||
|
opType = models.ActionDeleteBranch |
||||||
|
opts.Commits = &models.PushCommits{} |
||||||
|
} else { |
||||||
|
// if not the first commit, set the compare URL.
|
||||||
|
if opts.OldCommitID == git.EmptySHA { |
||||||
|
isNewBranch = true |
||||||
|
} else { |
||||||
|
opts.Commits.CompareURL = repo.ComposeCompareURL(opts.OldCommitID, opts.NewCommitID) |
||||||
|
} |
||||||
|
|
||||||
|
if err = models.UpdateIssuesCommit(pusher, repo, opts.Commits.Commits, refName); err != nil { |
||||||
|
log.Error("updateIssuesCommit: %v", err) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
if len(opts.Commits.Commits) > setting.UI.FeedMaxCommitNum { |
||||||
|
opts.Commits.Commits = opts.Commits.Commits[:setting.UI.FeedMaxCommitNum] |
||||||
|
} |
||||||
|
|
||||||
|
data, err := json.Marshal(opts.Commits) |
||||||
|
if err != nil { |
||||||
|
return fmt.Errorf("Marshal: %v", err) |
||||||
|
} |
||||||
|
|
||||||
|
if err = models.NotifyWatchers(&models.Action{ |
||||||
|
ActUserID: pusher.ID, |
||||||
|
ActUser: pusher, |
||||||
|
OpType: opType, |
||||||
|
Content: string(data), |
||||||
|
RepoID: repo.ID, |
||||||
|
Repo: repo, |
||||||
|
RefName: refName, |
||||||
|
IsPrivate: repo.IsPrivate, |
||||||
|
}); err != nil { |
||||||
|
return fmt.Errorf("NotifyWatchers: %v", err) |
||||||
|
} |
||||||
|
|
||||||
|
defer func() { |
||||||
|
go models.HookQueue.Add(repo.ID) |
||||||
|
}() |
||||||
|
|
||||||
|
apiPusher := pusher.APIFormat() |
||||||
|
apiRepo := repo.APIFormat(models.AccessModeNone) |
||||||
|
|
||||||
|
var shaSum string |
||||||
|
var isHookEventPush = false |
||||||
|
switch opType { |
||||||
|
case models.ActionCommitRepo: // Push
|
||||||
|
isHookEventPush = true |
||||||
|
|
||||||
|
if isNewBranch { |
||||||
|
gitRepo, err := git.OpenRepository(repo.RepoPath()) |
||||||
|
if err != nil { |
||||||
|
log.Error("OpenRepository[%s]: %v", repo.RepoPath(), err) |
||||||
|
} |
||||||
|
|
||||||
|
shaSum, err = gitRepo.GetBranchCommitID(refName) |
||||||
|
if err != nil { |
||||||
|
log.Error("GetBranchCommitID[%s]: %v", opts.RefFullName, err) |
||||||
|
} |
||||||
|
if err = models.PrepareWebhooks(repo, models.HookEventCreate, &api.CreatePayload{ |
||||||
|
Ref: refName, |
||||||
|
Sha: shaSum, |
||||||
|
RefType: "branch", |
||||||
|
Repo: apiRepo, |
||||||
|
Sender: apiPusher, |
||||||
|
}); err != nil { |
||||||
|
return fmt.Errorf("PrepareWebhooks: %v", err) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
case models.ActionDeleteBranch: // Delete Branch
|
||||||
|
isHookEventPush = true |
||||||
|
|
||||||
|
if err = models.PrepareWebhooks(repo, models.HookEventDelete, &api.DeletePayload{ |
||||||
|
Ref: refName, |
||||||
|
RefType: "branch", |
||||||
|
PusherType: api.PusherTypeUser, |
||||||
|
Repo: apiRepo, |
||||||
|
Sender: apiPusher, |
||||||
|
}); err != nil { |
||||||
|
return fmt.Errorf("PrepareWebhooks.(delete branch): %v", err) |
||||||
|
} |
||||||
|
|
||||||
|
case models.ActionPushTag: // Create
|
||||||
|
isHookEventPush = true |
||||||
|
|
||||||
|
gitRepo, err := git.OpenRepository(repo.RepoPath()) |
||||||
|
if err != nil { |
||||||
|
log.Error("OpenRepository[%s]: %v", repo.RepoPath(), err) |
||||||
|
} |
||||||
|
shaSum, err = gitRepo.GetTagCommitID(refName) |
||||||
|
if err != nil { |
||||||
|
log.Error("GetTagCommitID[%s]: %v", opts.RefFullName, err) |
||||||
|
} |
||||||
|
if err = models.PrepareWebhooks(repo, models.HookEventCreate, &api.CreatePayload{ |
||||||
|
Ref: refName, |
||||||
|
Sha: shaSum, |
||||||
|
RefType: "tag", |
||||||
|
Repo: apiRepo, |
||||||
|
Sender: apiPusher, |
||||||
|
}); err != nil { |
||||||
|
return fmt.Errorf("PrepareWebhooks: %v", err) |
||||||
|
} |
||||||
|
case models.ActionDeleteTag: // Delete Tag
|
||||||
|
isHookEventPush = true |
||||||
|
|
||||||
|
if err = models.PrepareWebhooks(repo, models.HookEventDelete, &api.DeletePayload{ |
||||||
|
Ref: refName, |
||||||
|
RefType: "tag", |
||||||
|
PusherType: api.PusherTypeUser, |
||||||
|
Repo: apiRepo, |
||||||
|
Sender: apiPusher, |
||||||
|
}); err != nil { |
||||||
|
return fmt.Errorf("PrepareWebhooks.(delete tag): %v", err) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
if isHookEventPush { |
||||||
|
if err = models.PrepareWebhooks(repo, models.HookEventPush, &api.PushPayload{ |
||||||
|
Ref: opts.RefFullName, |
||||||
|
Before: opts.OldCommitID, |
||||||
|
After: opts.NewCommitID, |
||||||
|
CompareURL: setting.AppURL + opts.Commits.CompareURL, |
||||||
|
Commits: opts.Commits.ToAPIPayloadCommits(repo.HTMLURL()), |
||||||
|
Repo: apiRepo, |
||||||
|
Pusher: apiPusher, |
||||||
|
Sender: apiPusher, |
||||||
|
}); err != nil { |
||||||
|
return fmt.Errorf("PrepareWebhooks: %v", err) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return nil |
||||||
|
} |
||||||
@ -0,0 +1,126 @@ |
|||||||
|
// Copyright 2019 The Gitea Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by a MIT-style
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package repofiles |
||||||
|
|
||||||
|
import ( |
||||||
|
"testing" |
||||||
|
|
||||||
|
"code.gitea.io/gitea/models" |
||||||
|
"code.gitea.io/gitea/modules/git" |
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert" |
||||||
|
) |
||||||
|
|
||||||
|
func testCorrectRepoAction(t *testing.T, opts CommitRepoActionOptions, actionBean *models.Action) { |
||||||
|
models.AssertNotExistsBean(t, actionBean) |
||||||
|
assert.NoError(t, CommitRepoAction(opts)) |
||||||
|
models.AssertExistsAndLoadBean(t, actionBean) |
||||||
|
models.CheckConsistencyFor(t, &models.Action{}) |
||||||
|
} |
||||||
|
|
||||||
|
func TestCommitRepoAction(t *testing.T) { |
||||||
|
samples := []struct { |
||||||
|
userID int64 |
||||||
|
repositoryID int64 |
||||||
|
commitRepoActionOptions CommitRepoActionOptions |
||||||
|
action models.Action |
||||||
|
}{ |
||||||
|
{ |
||||||
|
userID: 2, |
||||||
|
repositoryID: 2, |
||||||
|
commitRepoActionOptions: CommitRepoActionOptions{ |
||||||
|
RefFullName: "refName", |
||||||
|
OldCommitID: "oldCommitID", |
||||||
|
NewCommitID: "newCommitID", |
||||||
|
Commits: &models.PushCommits{ |
||||||
|
Commits: []*models.PushCommit{ |
||||||
|
{ |
||||||
|
Sha1: "abcdef1", |
||||||
|
CommitterEmail: "user2@example.com", |
||||||
|
CommitterName: "User Two", |
||||||
|
AuthorEmail: "user4@example.com", |
||||||
|
AuthorName: "User Four", |
||||||
|
Message: "message1", |
||||||
|
}, |
||||||
|
{ |
||||||
|
Sha1: "abcdef2", |
||||||
|
CommitterEmail: "user2@example.com", |
||||||
|
CommitterName: "User Two", |
||||||
|
AuthorEmail: "user2@example.com", |
||||||
|
AuthorName: "User Two", |
||||||
|
Message: "message2", |
||||||
|
}, |
||||||
|
}, |
||||||
|
Len: 2, |
||||||
|
}, |
||||||
|
}, |
||||||
|
action: models.Action{ |
||||||
|
OpType: models.ActionCommitRepo, |
||||||
|
RefName: "refName", |
||||||
|
}, |
||||||
|
}, |
||||||
|
{ |
||||||
|
userID: 2, |
||||||
|
repositoryID: 1, |
||||||
|
commitRepoActionOptions: CommitRepoActionOptions{ |
||||||
|
RefFullName: git.TagPrefix + "v1.1", |
||||||
|
OldCommitID: git.EmptySHA, |
||||||
|
NewCommitID: "newCommitID", |
||||||
|
Commits: &models.PushCommits{}, |
||||||
|
}, |
||||||
|
action: models.Action{ |
||||||
|
OpType: models.ActionPushTag, |
||||||
|
RefName: "v1.1", |
||||||
|
}, |
||||||
|
}, |
||||||
|
{ |
||||||
|
userID: 2, |
||||||
|
repositoryID: 1, |
||||||
|
commitRepoActionOptions: CommitRepoActionOptions{ |
||||||
|
RefFullName: git.TagPrefix + "v1.1", |
||||||
|
OldCommitID: "oldCommitID", |
||||||
|
NewCommitID: git.EmptySHA, |
||||||
|
Commits: &models.PushCommits{}, |
||||||
|
}, |
||||||
|
action: models.Action{ |
||||||
|
OpType: models.ActionDeleteTag, |
||||||
|
RefName: "v1.1", |
||||||
|
}, |
||||||
|
}, |
||||||
|
{ |
||||||
|
userID: 2, |
||||||
|
repositoryID: 1, |
||||||
|
commitRepoActionOptions: CommitRepoActionOptions{ |
||||||
|
RefFullName: git.BranchPrefix + "feature/1", |
||||||
|
OldCommitID: "oldCommitID", |
||||||
|
NewCommitID: git.EmptySHA, |
||||||
|
Commits: &models.PushCommits{}, |
||||||
|
}, |
||||||
|
action: models.Action{ |
||||||
|
OpType: models.ActionDeleteBranch, |
||||||
|
RefName: "feature/1", |
||||||
|
}, |
||||||
|
}, |
||||||
|
} |
||||||
|
|
||||||
|
for _, s := range samples { |
||||||
|
models.PrepareTestEnv(t) |
||||||
|
|
||||||
|
user := models.AssertExistsAndLoadBean(t, &models.User{ID: s.userID}).(*models.User) |
||||||
|
repo := models.AssertExistsAndLoadBean(t, &models.Repository{ID: s.repositoryID, OwnerID: user.ID}).(*models.Repository) |
||||||
|
repo.Owner = user |
||||||
|
|
||||||
|
s.commitRepoActionOptions.PusherName = user.Name |
||||||
|
s.commitRepoActionOptions.RepoOwnerID = user.ID |
||||||
|
s.commitRepoActionOptions.RepoName = repo.Name |
||||||
|
|
||||||
|
s.action.ActUserID = user.ID |
||||||
|
s.action.RepoID = repo.ID |
||||||
|
s.action.Repo = repo |
||||||
|
s.action.IsPrivate = repo.IsPrivate |
||||||
|
|
||||||
|
testCorrectRepoAction(t, s.commitRepoActionOptions, &s.action) |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue