Skip to content

Commit 2945473

Browse files
authored
move some milestone functions to a standalone package (#8213)
1 parent 99f3ee3 commit 2945473

File tree

5 files changed

+65
-44
lines changed

5 files changed

+65
-44
lines changed

Diff for: models/issue_milestone.go

-41
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,6 @@ package models
77
import (
88
"fmt"
99

10-
"code.gitea.io/gitea/modules/log"
1110
"code.gitea.io/gitea/modules/setting"
1211
api "code.gitea.io/gitea/modules/structs"
1312
"code.gitea.io/gitea/modules/timeutil"
@@ -394,46 +393,6 @@ func ChangeMilestoneAssign(issue *Issue, doer *User, oldMilestoneID int64) (err
394393
if err = sess.Commit(); err != nil {
395394
return fmt.Errorf("Commit: %v", err)
396395
}
397-
398-
var hookAction api.HookIssueAction
399-
if issue.MilestoneID > 0 {
400-
hookAction = api.HookIssueMilestoned
401-
} else {
402-
hookAction = api.HookIssueDemilestoned
403-
}
404-
405-
if err = issue.LoadAttributes(); err != nil {
406-
return err
407-
}
408-
409-
mode, _ := AccessLevel(doer, issue.Repo)
410-
if issue.IsPull {
411-
err = issue.PullRequest.LoadIssue()
412-
if err != nil {
413-
log.Error("LoadIssue: %v", err)
414-
return
415-
}
416-
err = PrepareWebhooks(issue.Repo, HookEventPullRequest, &api.PullRequestPayload{
417-
Action: hookAction,
418-
Index: issue.Index,
419-
PullRequest: issue.PullRequest.APIFormat(),
420-
Repository: issue.Repo.APIFormat(mode),
421-
Sender: doer.APIFormat(),
422-
})
423-
} else {
424-
err = PrepareWebhooks(issue.Repo, HookEventIssues, &api.IssuePayload{
425-
Action: hookAction,
426-
Index: issue.Index,
427-
Issue: issue.APIFormat(),
428-
Repository: issue.Repo.APIFormat(mode),
429-
Sender: doer.APIFormat(),
430-
})
431-
}
432-
if err != nil {
433-
log.Error("PrepareWebhooks [is_pull: %v]: %v", issue.IsPull, err)
434-
} else {
435-
go HookQueue.Add(issue.RepoID)
436-
}
437396
return nil
438397
}
439398

Diff for: routers/api/v1/repo/issue.go

+2-1
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@ import (
1919
api "code.gitea.io/gitea/modules/structs"
2020
"code.gitea.io/gitea/modules/timeutil"
2121
"code.gitea.io/gitea/modules/util"
22+
milestone_service "code.gitea.io/gitea/services/milestone"
2223
)
2324

2425
// ListIssues list the issues of a repository
@@ -345,7 +346,7 @@ func EditIssue(ctx *context.APIContext, form api.EditIssueOption) {
345346
issue.MilestoneID != *form.Milestone {
346347
oldMilestoneID := issue.MilestoneID
347348
issue.MilestoneID = *form.Milestone
348-
if err = models.ChangeMilestoneAssign(issue, ctx.User, oldMilestoneID); err != nil {
349+
if err = milestone_service.ChangeMilestoneAssign(issue, ctx.User, oldMilestoneID); err != nil {
349350
ctx.Error(500, "ChangeMilestoneAssign", err)
350351
return
351352
}

Diff for: routers/api/v1/repo/pull.go

+2-1
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ import (
1818
"code.gitea.io/gitea/modules/pull"
1919
api "code.gitea.io/gitea/modules/structs"
2020
"code.gitea.io/gitea/modules/timeutil"
21+
milestone_service "code.gitea.io/gitea/services/milestone"
2122
)
2223

2324
// ListPullRequests returns a list of all PRs
@@ -402,7 +403,7 @@ func EditPullRequest(ctx *context.APIContext, form api.EditPullRequestOption) {
402403
issue.MilestoneID != form.Milestone {
403404
oldMilestoneID := issue.MilestoneID
404405
issue.MilestoneID = form.Milestone
405-
if err = models.ChangeMilestoneAssign(issue, ctx.User, oldMilestoneID); err != nil {
406+
if err = milestone_service.ChangeMilestoneAssign(issue, ctx.User, oldMilestoneID); err != nil {
406407
ctx.Error(500, "ChangeMilestoneAssign", err)
407408
return
408409
}

Diff for: routers/repo/issue.go

+2-1
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ import (
2626
"code.gitea.io/gitea/modules/setting"
2727
api "code.gitea.io/gitea/modules/structs"
2828
"code.gitea.io/gitea/modules/util"
29+
milestone_service "code.gitea.io/gitea/services/milestone"
2930

3031
"github.com/unknwon/com"
3132
)
@@ -1086,7 +1087,7 @@ func UpdateIssueMilestone(ctx *context.Context) {
10861087
continue
10871088
}
10881089
issue.MilestoneID = milestoneID
1089-
if err := models.ChangeMilestoneAssign(issue, ctx.User, oldMilestoneID); err != nil {
1090+
if err := milestone_service.ChangeMilestoneAssign(issue, ctx.User, oldMilestoneID); err != nil {
10901091
ctx.ServerError("ChangeMilestoneAssign", err)
10911092
return
10921093
}

Diff for: services/milestone/milestone.go

+59
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,59 @@
1+
// Copyright 2019 The Gitea 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+
5+
package models
6+
7+
import (
8+
"code.gitea.io/gitea/models"
9+
"code.gitea.io/gitea/modules/log"
10+
api "code.gitea.io/gitea/modules/structs"
11+
)
12+
13+
// ChangeMilestoneAssign changes assignment of milestone for issue.
14+
func ChangeMilestoneAssign(issue *models.Issue, doer *models.User, oldMilestoneID int64) (err error) {
15+
if err = models.ChangeMilestoneAssign(issue, doer, oldMilestoneID); err != nil {
16+
return
17+
}
18+
19+
var hookAction api.HookIssueAction
20+
if issue.MilestoneID > 0 {
21+
hookAction = api.HookIssueMilestoned
22+
} else {
23+
hookAction = api.HookIssueDemilestoned
24+
}
25+
26+
if err = issue.LoadAttributes(); err != nil {
27+
return err
28+
}
29+
30+
mode, _ := models.AccessLevel(doer, issue.Repo)
31+
if issue.IsPull {
32+
err = issue.PullRequest.LoadIssue()
33+
if err != nil {
34+
log.Error("LoadIssue: %v", err)
35+
return
36+
}
37+
err = models.PrepareWebhooks(issue.Repo, models.HookEventPullRequest, &api.PullRequestPayload{
38+
Action: hookAction,
39+
Index: issue.Index,
40+
PullRequest: issue.PullRequest.APIFormat(),
41+
Repository: issue.Repo.APIFormat(mode),
42+
Sender: doer.APIFormat(),
43+
})
44+
} else {
45+
err = models.PrepareWebhooks(issue.Repo, models.HookEventIssues, &api.IssuePayload{
46+
Action: hookAction,
47+
Index: issue.Index,
48+
Issue: issue.APIFormat(),
49+
Repository: issue.Repo.APIFormat(mode),
50+
Sender: doer.APIFormat(),
51+
})
52+
}
53+
if err != nil {
54+
log.Error("PrepareWebhooks [is_pull: %v]: %v", issue.IsPull, err)
55+
} else {
56+
go models.HookQueue.Add(issue.RepoID)
57+
}
58+
return nil
59+
}

0 commit comments

Comments
 (0)