diff options
author | Lunny Xiao <xiaolunwen@gmail.com> | 2022-03-29 22:57:33 +0800 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-03-29 16:57:33 +0200 |
commit | 74731c3a5aea71c81e4ca75bde96154f3adf3cfa (patch) | |
tree | 64b01935ab9912de71318eac447721e41b200135 /models | |
parent | bd97736b9c7a16023bc9abf17be6157284f655b1 (diff) | |
download | gitea-74731c3a5aea71c81e4ca75bde96154f3adf3cfa.tar.gz gitea-74731c3a5aea71c81e4ca75bde96154f3adf3cfa.zip |
Move some issue methods as functions (#19255)
* Move some issue methods as functions
* Fix bug
Diffstat (limited to 'models')
-rw-r--r-- | models/issue.go | 85 | ||||
-rw-r--r-- | models/issue_assignees.go | 8 | ||||
-rw-r--r-- | models/issue_assignees_test.go | 6 | ||||
-rw-r--r-- | models/issue_comment.go | 2 | ||||
-rw-r--r-- | models/issue_dependency_test.go | 2 | ||||
-rw-r--r-- | models/issue_test.go | 6 | ||||
-rw-r--r-- | models/issue_xref_test.go | 4 | ||||
-rw-r--r-- | models/pull.go | 2 |
8 files changed, 47 insertions, 68 deletions
diff --git a/models/issue.go b/models/issue.go index d2a2b6a329..99688f2ec9 100644 --- a/models/issue.go +++ b/models/issue.go @@ -457,23 +457,6 @@ func (issue *Issue) IsPoster(uid int64) bool { return issue.OriginalAuthorID == 0 && issue.PosterID == uid } -func (issue *Issue) hasLabel(e db.Engine, labelID int64) bool { - return hasIssueLabel(e, issue.ID, labelID) -} - -// HasLabel returns true if issue has been labeled by given ID. -func (issue *Issue) HasLabel(labelID int64) bool { - return issue.hasLabel(db.GetEngine(db.DefaultContext), labelID) -} - -func (issue *Issue) addLabel(ctx context.Context, label *Label, doer *user_model.User) error { - return newIssueLabel(ctx, issue, label, doer) -} - -func (issue *Issue) addLabels(ctx context.Context, labels []*Label, doer *user_model.User) error { - return newIssueLabels(ctx, issue, labels, doer) -} - func (issue *Issue) getLabels(e db.Engine) (err error) { if len(issue.Labels) > 0 { return nil @@ -486,17 +469,13 @@ func (issue *Issue) getLabels(e db.Engine) (err error) { return nil } -func (issue *Issue) removeLabel(ctx context.Context, doer *user_model.User, label *Label) error { - return deleteIssueLabel(ctx, issue, label, doer) -} - -func (issue *Issue) clearLabels(ctx context.Context, doer *user_model.User) (err error) { +func clearIssueLabels(ctx context.Context, issue *Issue, doer *user_model.User) (err error) { if err = issue.getLabels(db.GetEngine(ctx)); err != nil { return fmt.Errorf("getLabels: %v", err) } for i := range issue.Labels { - if err = issue.removeLabel(ctx, doer, issue.Labels[i]); err != nil { + if err = deleteIssueLabel(ctx, issue, issue.Labels[i], doer); err != nil { return fmt.Errorf("removeLabel: %v", err) } } @@ -504,9 +483,9 @@ func (issue *Issue) clearLabels(ctx context.Context, doer *user_model.User) (err return nil } -// ClearLabels removes all issue labels as the given user. +// ClearIssueLabels removes all issue labels as the given user. // Triggers appropriate WebHooks, if any. -func (issue *Issue) ClearLabels(doer *user_model.User) (err error) { +func ClearIssueLabels(issue *Issue, doer *user_model.User) (err error) { ctx, committer, err := db.TxContext() if err != nil { return err @@ -527,7 +506,7 @@ func (issue *Issue) ClearLabels(doer *user_model.User) (err error) { return ErrRepoLabelNotExist{} } - if err = issue.clearLabels(ctx, doer); err != nil { + if err = clearIssueLabels(ctx, issue, doer); err != nil { return err } @@ -552,9 +531,9 @@ func (ts labelSorter) Swap(i, j int) { []*Label(ts)[i], []*Label(ts)[j] = []*Label(ts)[j], []*Label(ts)[i] } -// ReplaceLabels removes all current labels and add new labels to the issue. +// ReplaceIssueLabels removes all current labels and add new labels to the issue. // Triggers appropriate WebHooks, if any. -func (issue *Issue) ReplaceLabels(labels []*Label, doer *user_model.User) (err error) { +func ReplaceIssueLabels(issue *Issue, labels []*Label, doer *user_model.User) (err error) { ctx, committer, err := db.TxContext() if err != nil { return err @@ -601,13 +580,13 @@ func (issue *Issue) ReplaceLabels(labels []*Label, doer *user_model.User) (err e toRemove = append(toRemove, issue.Labels[removeIndex:]...) if len(toAdd) > 0 { - if err = issue.addLabels(ctx, toAdd, doer); err != nil { + if err = newIssueLabels(ctx, issue, toAdd, doer); err != nil { return fmt.Errorf("addLabels: %v", err) } } for _, l := range toRemove { - if err = issue.removeLabel(ctx, doer, l); err != nil { + if err = deleteIssueLabel(ctx, issue, l, doer); err != nil { return fmt.Errorf("removeLabel: %v", err) } } @@ -636,7 +615,7 @@ func updateIssueCols(ctx context.Context, issue *Issue, cols ...string) error { return nil } -func (issue *Issue) changeStatus(ctx context.Context, doer *user_model.User, isClosed, isMergePull bool) (*Comment, error) { +func changeIssueStatus(ctx context.Context, issue *Issue, doer *user_model.User, isClosed, isMergePull bool) (*Comment, error) { // Reload the issue currentIssue, err := getIssueByID(db.GetEngine(ctx), issue.ID) if err != nil { @@ -656,10 +635,10 @@ func (issue *Issue) changeStatus(ctx context.Context, doer *user_model.User, isC } issue.IsClosed = isClosed - return issue.doChangeStatus(ctx, doer, isMergePull) + return doChangeIssueStatus(ctx, issue, doer, isMergePull) } -func (issue *Issue) doChangeStatus(ctx context.Context, doer *user_model.User, isMergePull bool) (*Comment, error) { +func doChangeIssueStatus(ctx context.Context, issue *Issue, doer *user_model.User, isMergePull bool) (*Comment, error) { e := db.GetEngine(ctx) // Check for open dependencies if issue.IsClosed && issue.Repo.IsDependenciesEnabledCtx(ctx) { @@ -701,7 +680,7 @@ func (issue *Issue) doChangeStatus(ctx context.Context, doer *user_model.User, i } } - if err := issue.updateClosedNum(ctx); err != nil { + if err := updateIssueClosedNum(ctx, issue); err != nil { return nil, err } @@ -721,8 +700,8 @@ func (issue *Issue) doChangeStatus(ctx context.Context, doer *user_model.User, i }) } -// ChangeStatus changes issue status to open or closed. -func (issue *Issue) ChangeStatus(doer *user_model.User, isClosed bool) (*Comment, error) { +// ChangeIssueStatus changes issue status to open or closed. +func ChangeIssueStatus(issue *Issue, doer *user_model.User, isClosed bool) (*Comment, error) { ctx, committer, err := db.TxContext() if err != nil { return nil, err @@ -736,7 +715,7 @@ func (issue *Issue) ChangeStatus(doer *user_model.User, isClosed bool) (*Comment return nil, err } - comment, err := issue.changeStatus(ctx, doer, isClosed, false) + comment, err := changeIssueStatus(ctx, issue, doer, isClosed, false) if err != nil { return nil, err } @@ -748,8 +727,8 @@ func (issue *Issue) ChangeStatus(doer *user_model.User, isClosed bool) (*Comment return comment, nil } -// ChangeTitle changes the title of this issue, as the given user. -func (issue *Issue) ChangeTitle(doer *user_model.User, oldTitle string) (err error) { +// ChangeIssueTitle changes the title of this issue, as the given user. +func ChangeIssueTitle(issue *Issue, doer *user_model.User, oldTitle string) (err error) { ctx, committer, err := db.TxContext() if err != nil { return err @@ -782,8 +761,8 @@ func (issue *Issue) ChangeTitle(doer *user_model.User, oldTitle string) (err err return committer.Commit() } -// ChangeRef changes the branch of this issue, as the given user. -func (issue *Issue) ChangeRef(doer *user_model.User, oldRef string) (err error) { +// ChangeIssueRef changes the branch of this issue, as the given user. +func ChangeIssueRef(issue *Issue, doer *user_model.User, oldRef string) (err error) { ctx, committer, err := db.TxContext() if err != nil { return err @@ -840,8 +819,8 @@ func AddDeletePRBranchComment(doer *user_model.User, repo *repo_model.Repository return committer.Commit() } -// UpdateAttachments update attachments by UUIDs for the issue -func (issue *Issue) UpdateAttachments(uuids []string) (err error) { +// UpdateIssueAttachments update attachments by UUIDs for the issue +func UpdateIssueAttachments(issueID int64, uuids []string) (err error) { ctx, committer, err := db.TxContext() if err != nil { return err @@ -852,7 +831,7 @@ func (issue *Issue) UpdateAttachments(uuids []string) (err error) { return fmt.Errorf("getAttachmentsByUUIDs [uuids: %v]: %v", uuids, err) } for i := 0; i < len(attachments); i++ { - attachments[i].IssueID = issue.ID + attachments[i].IssueID = issueID if err := repo_model.UpdateAttachmentCtx(ctx, attachments[i]); err != nil { return fmt.Errorf("update attachment [id: %d]: %v", attachments[i].ID, err) } @@ -860,8 +839,8 @@ func (issue *Issue) UpdateAttachments(uuids []string) (err error) { return committer.Commit() } -// ChangeContent changes issue content, as the given user. -func (issue *Issue) ChangeContent(doer *user_model.User, content string) (err error) { +// ChangeIssueContent changes issue content, as the given user. +func ChangeIssueContent(issue *Issue, doer *user_model.User, content string) (err error) { ctx, committer, err := db.TxContext() if err != nil { return err @@ -1034,7 +1013,7 @@ func newIssue(ctx context.Context, doer *user_model.User, opts NewIssueOptions) continue } - if err = opts.Issue.addLabel(ctx, label, opts.Issue.Poster); err != nil { + if err = newIssueLabel(ctx, opts.Issue, label, opts.Issue.Poster); err != nil { return fmt.Errorf("addLabel [id: %d]: %v", label.ID, err) } } @@ -2011,7 +1990,7 @@ func UpdateIssueByAPI(issue *Issue, doer *user_model.User) (statusChangeComment } if currentIssue.IsClosed != issue.IsClosed { - statusChangeComment, err = issue.doChangeStatus(ctx, doer, false) + statusChangeComment, err = doChangeIssueStatus(ctx, issue, doer, false) if err != nil { return nil, false, err } @@ -2235,7 +2214,7 @@ func (issue *Issue) BlockingDependencies() ([]*DependencyInfo, error) { return issue.getBlockingDependencies(db.GetEngine(db.DefaultContext)) } -func (issue *Issue) updateClosedNum(ctx context.Context) (err error) { +func updateIssueClosedNum(ctx context.Context, issue *Issue) (err error) { if issue.IsPull { err = repoStatsCorrectNumClosed(ctx, issue.RepoID, true, "num_closed_pulls") } else { @@ -2245,9 +2224,9 @@ func (issue *Issue) updateClosedNum(ctx context.Context) (err error) { } // FindAndUpdateIssueMentions finds users mentioned in the given content string, and saves them in the database. -func (issue *Issue) FindAndUpdateIssueMentions(ctx context.Context, doer *user_model.User, content string) (mentions []*user_model.User, err error) { +func FindAndUpdateIssueMentions(ctx context.Context, issue *Issue, doer *user_model.User, content string) (mentions []*user_model.User, err error) { rawMentions := references.FindAllMentionsMarkdown(content) - mentions, err = issue.ResolveMentionsByVisibility(ctx, doer, rawMentions) + mentions, err = ResolveIssueMentionsByVisibility(ctx, issue, doer, rawMentions) if err != nil { return nil, fmt.Errorf("UpdateIssueMentions [%d]: %v", issue.ID, err) } @@ -2257,9 +2236,9 @@ func (issue *Issue) FindAndUpdateIssueMentions(ctx context.Context, doer *user_m return } -// ResolveMentionsByVisibility returns the users mentioned in an issue, removing those that +// ResolveIssueMentionsByVisibility returns the users mentioned in an issue, removing those that // don't have access to reading it. Teams are expanded into their users, but organizations are ignored. -func (issue *Issue) ResolveMentionsByVisibility(ctx context.Context, doer *user_model.User, mentions []string) (users []*user_model.User, err error) { +func ResolveIssueMentionsByVisibility(ctx context.Context, issue *Issue, doer *user_model.User, mentions []string) (users []*user_model.User, err error) { if len(mentions) == 0 { return } diff --git a/models/issue_assignees.go b/models/issue_assignees.go index b3511f8b59..a91793f2f1 100644 --- a/models/issue_assignees.go +++ b/models/issue_assignees.go @@ -92,15 +92,15 @@ func clearAssigneeByUserID(sess db.Engine, userID int64) (err error) { return } -// ToggleAssignee changes a user between assigned and not assigned for this issue, and make issue comment for it. -func (issue *Issue) ToggleAssignee(doer *user_model.User, assigneeID int64) (removed bool, comment *Comment, err error) { +// ToggleIssueAssignee changes a user between assigned and not assigned for this issue, and make issue comment for it. +func ToggleIssueAssignee(issue *Issue, doer *user_model.User, assigneeID int64) (removed bool, comment *Comment, err error) { ctx, committer, err := db.TxContext() if err != nil { return false, nil, err } defer committer.Close() - removed, comment, err = issue.toggleAssignee(ctx, doer, assigneeID, false) + removed, comment, err = toggleIssueAssignee(ctx, issue, doer, assigneeID, false) if err != nil { return false, nil, err } @@ -112,7 +112,7 @@ func (issue *Issue) ToggleAssignee(doer *user_model.User, assigneeID int64) (rem return removed, comment, nil } -func (issue *Issue) toggleAssignee(ctx context.Context, doer *user_model.User, assigneeID int64, isCreate bool) (removed bool, comment *Comment, err error) { +func toggleIssueAssignee(ctx context.Context, issue *Issue, doer *user_model.User, assigneeID int64, isCreate bool) (removed bool, comment *Comment, err error) { sess := db.GetEngine(ctx) removed, err = toggleUserAssignee(sess, issue, assigneeID) if err != nil { diff --git a/models/issue_assignees_test.go b/models/issue_assignees_test.go index dd9a42b572..41a3ad86e3 100644 --- a/models/issue_assignees_test.go +++ b/models/issue_assignees_test.go @@ -23,17 +23,17 @@ func TestUpdateAssignee(t *testing.T) { // Assign multiple users user2, err := user_model.GetUserByID(2) assert.NoError(t, err) - _, _, err = issue.ToggleAssignee(&user_model.User{ID: 1}, user2.ID) + _, _, err = ToggleIssueAssignee(issue, &user_model.User{ID: 1}, user2.ID) assert.NoError(t, err) user3, err := user_model.GetUserByID(3) assert.NoError(t, err) - _, _, err = issue.ToggleAssignee(&user_model.User{ID: 1}, user3.ID) + _, _, err = ToggleIssueAssignee(issue, &user_model.User{ID: 1}, user3.ID) assert.NoError(t, err) user1, err := user_model.GetUserByID(1) // This user is already assigned (see the definition in fixtures), so running UpdateAssignee should unassign him assert.NoError(t, err) - _, _, err = issue.ToggleAssignee(&user_model.User{ID: 1}, user1.ID) + _, _, err = ToggleIssueAssignee(issue, &user_model.User{ID: 1}, user1.ID) assert.NoError(t, err) // Check if he got removed diff --git a/models/issue_comment.go b/models/issue_comment.go index 500ed6d038..7fb013ae92 100644 --- a/models/issue_comment.go +++ b/models/issue_comment.go @@ -858,7 +858,7 @@ func updateCommentInfos(ctx context.Context, opts *CreateCommentOptions, comment } } case CommentTypeReopen, CommentTypeClose: - if err = opts.Issue.updateClosedNum(ctx); err != nil { + if err = updateIssueClosedNum(ctx, opts.Issue); err != nil { return err } } diff --git a/models/issue_dependency_test.go b/models/issue_dependency_test.go index 61215dedef..1789de5dc0 100644 --- a/models/issue_dependency_test.go +++ b/models/issue_dependency_test.go @@ -48,7 +48,7 @@ func TestCreateIssueDependency(t *testing.T) { assert.False(t, left) // Close #2 and check again - _, err = issue2.ChangeStatus(user1, true) + _, err = ChangeIssueStatus(issue2, user1, true) assert.NoError(t, err) left, err = IssueNoDependenciesLeft(issue1) diff --git a/models/issue_test.go b/models/issue_test.go index 9afbce67c1..71e0e8a2f7 100644 --- a/models/issue_test.go +++ b/models/issue_test.go @@ -34,7 +34,7 @@ func TestIssue_ReplaceLabels(t *testing.T) { for i, labelID := range labelIDs { labels[i] = unittest.AssertExistsAndLoadBean(t, &Label{ID: labelID, RepoID: repo.ID}).(*Label) } - assert.NoError(t, issue.ReplaceLabels(labels, doer)) + assert.NoError(t, ReplaceIssueLabels(issue, labels, doer)) unittest.AssertCount(t, &IssueLabel{IssueID: issueID}, len(labelIDs)) for _, labelID := range labelIDs { unittest.AssertExistsAndLoadBean(t, &IssueLabel{IssueID: issueID, LabelID: labelID}) @@ -116,7 +116,7 @@ func TestIssue_ClearLabels(t *testing.T) { assert.NoError(t, unittest.PrepareTestDatabase()) issue := unittest.AssertExistsAndLoadBean(t, &Issue{ID: test.issueID}).(*Issue) doer := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: test.doerID}).(*user_model.User) - assert.NoError(t, issue.ClearLabels(doer)) + assert.NoError(t, ClearIssueLabels(issue, doer)) unittest.AssertNotExistsBean(t, &IssueLabel{IssueID: test.issueID}) } } @@ -459,7 +459,7 @@ func TestIssue_ResolveMentions(t *testing.T) { r := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{OwnerID: o.ID, LowerName: repo}).(*repo_model.Repository) issue := &Issue{RepoID: r.ID} d := unittest.AssertExistsAndLoadBean(t, &user_model.User{LowerName: doer}).(*user_model.User) - resolved, err := issue.ResolveMentionsByVisibility(db.DefaultContext, d, mentions) + resolved, err := ResolveIssueMentionsByVisibility(db.DefaultContext, issue, d, mentions) assert.NoError(t, err) ids := make([]int64, len(resolved)) for i, user := range resolved { diff --git a/models/issue_xref_test.go b/models/issue_xref_test.go index 1deeb44ad5..1ae5fb360f 100644 --- a/models/issue_xref_test.go +++ b/models/issue_xref_test.go @@ -83,7 +83,7 @@ func TestXRef_NeuterCrossReferences(t *testing.T) { d := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2}).(*user_model.User) i.Title = "title2, no mentions" - assert.NoError(t, i.ChangeTitle(d, title)) + assert.NoError(t, ChangeIssueTitle(i, d, title)) ref = unittest.AssertExistsAndLoadBean(t, &Comment{IssueID: itarget.ID, RefIssueID: i.ID, RefCommentID: 0}).(*Comment) assert.Equal(t, CommentTypeIssueRef, ref.Type) @@ -98,7 +98,7 @@ func TestXRef_ResolveCrossReferences(t *testing.T) { i1 := testCreateIssue(t, 1, 2, "title1", "content1", false) i2 := testCreateIssue(t, 1, 2, "title2", "content2", false) i3 := testCreateIssue(t, 1, 2, "title3", "content3", false) - _, err := i3.ChangeStatus(d, true) + _, err := ChangeIssueStatus(i3, d, true) assert.NoError(t, err) pr := testCreatePR(t, 1, 2, "titlepr", fmt.Sprintf("closes #%d", i1.Index)) diff --git a/models/pull.go b/models/pull.go index d6bfbbf09f..863c8d708f 100644 --- a/models/pull.go +++ b/models/pull.go @@ -420,7 +420,7 @@ func (pr *PullRequest) SetMerged() (bool, error) { return false, err } - if _, err := pr.Issue.changeStatus(ctx, pr.Merger, true, true); err != nil { + if _, err := changeIssueStatus(ctx, pr.Issue, pr.Merger, true, true); err != nil { return false, fmt.Errorf("Issue.changeStatus: %v", err) } |