mirror of https://github.com/go-gitea/gitea
Multiple assignees (#3705)
kolaente
7 years ago
committed by
Lauris BH
parent
238a997ec0
commit
95f2e2b57b
@ -0,0 +1,8 @@ |
|||||||
|
- |
||||||
|
id: 1 |
||||||
|
assignee_id: 1 |
||||||
|
issue_id: 1 |
||||||
|
- |
||||||
|
id: 2 |
||||||
|
assignee_id: 1 |
||||||
|
issue_id: 6 |
||||||
@ -0,0 +1,263 @@ |
|||||||
|
// Copyright 2018 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 models |
||||||
|
|
||||||
|
import ( |
||||||
|
"fmt" |
||||||
|
|
||||||
|
"code.gitea.io/gitea/modules/log" |
||||||
|
|
||||||
|
api "code.gitea.io/sdk/gitea" |
||||||
|
"github.com/go-xorm/xorm" |
||||||
|
) |
||||||
|
|
||||||
|
// IssueAssignees saves all issue assignees
|
||||||
|
type IssueAssignees struct { |
||||||
|
ID int64 `xorm:"pk autoincr"` |
||||||
|
AssigneeID int64 `xorm:"INDEX"` |
||||||
|
IssueID int64 `xorm:"INDEX"` |
||||||
|
} |
||||||
|
|
||||||
|
// This loads all assignees of an issue
|
||||||
|
func (issue *Issue) loadAssignees(e Engine) (err error) { |
||||||
|
// Reset maybe preexisting assignees
|
||||||
|
issue.Assignees = []*User{} |
||||||
|
|
||||||
|
err = e.Table("`user`"). |
||||||
|
Join("INNER", "issue_assignees", "assignee_id = `user`.id"). |
||||||
|
Where("issue_assignees.issue_id = ?", issue.ID). |
||||||
|
Find(&issue.Assignees) |
||||||
|
|
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
// Check if we have at least one assignee and if yes put it in as `Assignee`
|
||||||
|
if len(issue.Assignees) > 0 { |
||||||
|
issue.Assignee = issue.Assignees[0] |
||||||
|
} |
||||||
|
|
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
// GetAssigneesByIssue returns everyone assigned to that issue
|
||||||
|
func GetAssigneesByIssue(issue *Issue) (assignees []*User, err error) { |
||||||
|
err = issue.loadAssignees(x) |
||||||
|
if err != nil { |
||||||
|
return assignees, err |
||||||
|
} |
||||||
|
|
||||||
|
return issue.Assignees, nil |
||||||
|
} |
||||||
|
|
||||||
|
// IsUserAssignedToIssue returns true when the user is assigned to the issue
|
||||||
|
func IsUserAssignedToIssue(issue *Issue, user *User) (isAssigned bool, err error) { |
||||||
|
isAssigned, err = x.Exist(&IssueAssignees{IssueID: issue.ID, AssigneeID: user.ID}) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
// DeleteNotPassedAssignee deletes all assignees who aren't passed via the "assignees" array
|
||||||
|
func DeleteNotPassedAssignee(issue *Issue, doer *User, assignees []*User) (err error) { |
||||||
|
var found bool |
||||||
|
|
||||||
|
for _, assignee := range issue.Assignees { |
||||||
|
|
||||||
|
found = false |
||||||
|
for _, alreadyAssignee := range assignees { |
||||||
|
if assignee.ID == alreadyAssignee.ID { |
||||||
|
found = true |
||||||
|
break |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
if !found { |
||||||
|
// This function also does comments and hooks, which is why we call it seperatly instead of directly removing the assignees here
|
||||||
|
if err := UpdateAssignee(issue, doer, assignee.ID); err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
// MakeAssigneeList concats a string with all names of the assignees. Useful for logs.
|
||||||
|
func MakeAssigneeList(issue *Issue) (assigneeList string, err error) { |
||||||
|
err = issue.loadAssignees(x) |
||||||
|
if err != nil { |
||||||
|
return "", err |
||||||
|
} |
||||||
|
|
||||||
|
for in, assignee := range issue.Assignees { |
||||||
|
assigneeList += assignee.Name |
||||||
|
|
||||||
|
if len(issue.Assignees) > (in + 1) { |
||||||
|
assigneeList += ", " |
||||||
|
} |
||||||
|
} |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
// ClearAssigneeByUserID deletes all assignments of an user
|
||||||
|
func clearAssigneeByUserID(sess *xorm.Session, userID int64) (err error) { |
||||||
|
_, err = sess.Delete(&IssueAssignees{AssigneeID: userID}) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
// AddAssigneeIfNotAssigned adds an assignee only if he isn't aleady assigned to the issue
|
||||||
|
func AddAssigneeIfNotAssigned(issue *Issue, doer *User, assigneeID int64) (err error) { |
||||||
|
// Check if the user is already assigned
|
||||||
|
isAssigned, err := IsUserAssignedToIssue(issue, &User{ID: assigneeID}) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
if !isAssigned { |
||||||
|
return issue.ChangeAssignee(doer, assigneeID) |
||||||
|
} |
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
// UpdateAssignee deletes or adds an assignee to an issue
|
||||||
|
func UpdateAssignee(issue *Issue, doer *User, assigneeID int64) (err error) { |
||||||
|
return issue.ChangeAssignee(doer, assigneeID) |
||||||
|
} |
||||||
|
|
||||||
|
// ChangeAssignee changes the Assignee of this issue.
|
||||||
|
func (issue *Issue) ChangeAssignee(doer *User, assigneeID int64) (err error) { |
||||||
|
sess := x.NewSession() |
||||||
|
defer sess.Close() |
||||||
|
|
||||||
|
if err := sess.Begin(); err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
if err := issue.changeAssignee(sess, doer, assigneeID); err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
return sess.Commit() |
||||||
|
} |
||||||
|
|
||||||
|
func (issue *Issue) changeAssignee(sess *xorm.Session, doer *User, assigneeID int64) (err error) { |
||||||
|
|
||||||
|
// Update the assignee
|
||||||
|
removed, err := updateIssueAssignee(sess, issue, assigneeID) |
||||||
|
if err != nil { |
||||||
|
return fmt.Errorf("UpdateIssueUserByAssignee: %v", err) |
||||||
|
} |
||||||
|
|
||||||
|
// Repo infos
|
||||||
|
if err = issue.loadRepo(sess); err != nil { |
||||||
|
return fmt.Errorf("loadRepo: %v", err) |
||||||
|
} |
||||||
|
|
||||||
|
// Comment
|
||||||
|
if _, err = createAssigneeComment(sess, doer, issue.Repo, issue, assigneeID, removed); err != nil { |
||||||
|
return fmt.Errorf("createAssigneeComment: %v", err) |
||||||
|
} |
||||||
|
|
||||||
|
if issue.IsPull { |
||||||
|
issue.PullRequest = &PullRequest{Issue: issue} |
||||||
|
apiPullRequest := &api.PullRequestPayload{ |
||||||
|
Index: issue.Index, |
||||||
|
PullRequest: issue.PullRequest.APIFormat(), |
||||||
|
Repository: issue.Repo.APIFormat(AccessModeNone), |
||||||
|
Sender: doer.APIFormat(), |
||||||
|
} |
||||||
|
if removed { |
||||||
|
apiPullRequest.Action = api.HookIssueUnassigned |
||||||
|
} else { |
||||||
|
apiPullRequest.Action = api.HookIssueAssigned |
||||||
|
} |
||||||
|
if err := PrepareWebhooks(issue.Repo, HookEventPullRequest, apiPullRequest); err != nil { |
||||||
|
log.Error(4, "PrepareWebhooks [is_pull: %v, remove_assignee: %v]: %v", issue.IsPull, removed, err) |
||||||
|
return nil |
||||||
|
} |
||||||
|
} |
||||||
|
go HookQueue.Add(issue.RepoID) |
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
// UpdateAPIAssignee is a helper function to add or delete one or multiple issue assignee(s)
|
||||||
|
// Deleting is done the Github way (quote from their api documentation):
|
||||||
|
// https://developer.github.com/v3/issues/#edit-an-issue
|
||||||
|
// "assignees" (array): Logins for Users to assign to this issue.
|
||||||
|
// Pass one or more user logins to replace the set of assignees on this Issue.
|
||||||
|
// Send an empty array ([]) to clear all assignees from the Issue.
|
||||||
|
func UpdateAPIAssignee(issue *Issue, oneAssignee string, multipleAssignees []string, doer *User) (err error) { |
||||||
|
var allNewAssignees []*User |
||||||
|
|
||||||
|
// Keep the old assignee thingy for compatibility reasons
|
||||||
|
if oneAssignee != "" { |
||||||
|
// Prevent double adding assignees
|
||||||
|
var isDouble bool |
||||||
|
for _, assignee := range multipleAssignees { |
||||||
|
if assignee == oneAssignee { |
||||||
|
isDouble = true |
||||||
|
break |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
if !isDouble { |
||||||
|
multipleAssignees = append(multipleAssignees, oneAssignee) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Loop through all assignees to add them
|
||||||
|
for _, assigneeName := range multipleAssignees { |
||||||
|
assignee, err := GetUserByName(assigneeName) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
allNewAssignees = append(allNewAssignees, assignee) |
||||||
|
} |
||||||
|
|
||||||
|
// Delete all old assignees not passed
|
||||||
|
if err = DeleteNotPassedAssignee(issue, doer, allNewAssignees); err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
// Add all new assignees
|
||||||
|
// Update the assignee. The function will check if the user exists, is already
|
||||||
|
// assigned (which he shouldn't as we deleted all assignees before) and
|
||||||
|
// has access to the repo.
|
||||||
|
for _, assignee := range allNewAssignees { |
||||||
|
// Extra method to prevent double adding (which would result in removing)
|
||||||
|
err = AddAssigneeIfNotAssigned(issue, doer, assignee.ID) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
// MakeIDsFromAPIAssigneesToAdd returns an array with all assignee IDs
|
||||||
|
func MakeIDsFromAPIAssigneesToAdd(oneAssignee string, multipleAssignees []string) (assigneeIDs []int64, err error) { |
||||||
|
|
||||||
|
// Keeping the old assigning method for compatibility reasons
|
||||||
|
if oneAssignee != "" { |
||||||
|
|
||||||
|
// Prevent double adding assignees
|
||||||
|
var isDouble bool |
||||||
|
for _, assignee := range multipleAssignees { |
||||||
|
if assignee == oneAssignee { |
||||||
|
isDouble = true |
||||||
|
break |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
if !isDouble { |
||||||
|
multipleAssignees = append(multipleAssignees, oneAssignee) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Get the IDs of all assignees
|
||||||
|
assigneeIDs = GetUserIDsByNames(multipleAssignees) |
||||||
|
|
||||||
|
return |
||||||
|
} |
||||||
@ -0,0 +1,71 @@ |
|||||||
|
// Copyright 2018 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 models |
||||||
|
|
||||||
|
import ( |
||||||
|
"testing" |
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert" |
||||||
|
) |
||||||
|
|
||||||
|
func TestUpdateAssignee(t *testing.T) { |
||||||
|
assert.NoError(t, PrepareTestDatabase()) |
||||||
|
|
||||||
|
// Fake issue with assignees
|
||||||
|
issue, err := GetIssueByID(1) |
||||||
|
assert.NoError(t, err) |
||||||
|
|
||||||
|
// Assign multiple users
|
||||||
|
user2, err := GetUserByID(2) |
||||||
|
assert.NoError(t, err) |
||||||
|
err = UpdateAssignee(issue, &User{ID: 1}, user2.ID) |
||||||
|
assert.NoError(t, err) |
||||||
|
|
||||||
|
user3, err := GetUserByID(3) |
||||||
|
assert.NoError(t, err) |
||||||
|
err = UpdateAssignee(issue, &User{ID: 1}, user3.ID) |
||||||
|
assert.NoError(t, err) |
||||||
|
|
||||||
|
user1, err := GetUserByID(1) // This user is already assigned (see the definition in fixtures), so running UpdateAssignee should unassign him
|
||||||
|
assert.NoError(t, err) |
||||||
|
err = UpdateAssignee(issue, &User{ID: 1}, user1.ID) |
||||||
|
assert.NoError(t, err) |
||||||
|
|
||||||
|
// Check if he got removed
|
||||||
|
isAssigned, err := IsUserAssignedToIssue(issue, user1) |
||||||
|
assert.NoError(t, err) |
||||||
|
assert.False(t, isAssigned) |
||||||
|
|
||||||
|
// Check if they're all there
|
||||||
|
assignees, err := GetAssigneesByIssue(issue) |
||||||
|
assert.NoError(t, err) |
||||||
|
|
||||||
|
var expectedAssignees []*User |
||||||
|
expectedAssignees = append(expectedAssignees, user2) |
||||||
|
expectedAssignees = append(expectedAssignees, user3) |
||||||
|
|
||||||
|
for in, assignee := range assignees { |
||||||
|
assert.Equal(t, assignee.ID, expectedAssignees[in].ID) |
||||||
|
} |
||||||
|
|
||||||
|
// Check if the user is assigned
|
||||||
|
isAssigned, err = IsUserAssignedToIssue(issue, user2) |
||||||
|
assert.NoError(t, err) |
||||||
|
assert.True(t, isAssigned) |
||||||
|
|
||||||
|
// This user should not be assigned
|
||||||
|
isAssigned, err = IsUserAssignedToIssue(issue, &User{ID: 4}) |
||||||
|
assert.NoError(t, err) |
||||||
|
assert.False(t, isAssigned) |
||||||
|
|
||||||
|
// Clean everyone
|
||||||
|
err = DeleteNotPassedAssignee(issue, user1, []*User{}) |
||||||
|
assert.NoError(t, err) |
||||||
|
|
||||||
|
// Check they're gone
|
||||||
|
assignees, err = GetAssigneesByIssue(issue) |
||||||
|
assert.NoError(t, err) |
||||||
|
assert.Equal(t, 0, len(assignees)) |
||||||
|
} |
||||||
@ -0,0 +1,129 @@ |
|||||||
|
// Copyright 2018 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 migrations |
||||||
|
|
||||||
|
import ( |
||||||
|
"code.gitea.io/gitea/modules/util" |
||||||
|
|
||||||
|
"github.com/go-xorm/xorm" |
||||||
|
) |
||||||
|
|
||||||
|
func addMultipleAssignees(x *xorm.Engine) error { |
||||||
|
|
||||||
|
// Redeclare issue struct
|
||||||
|
type Issue struct { |
||||||
|
ID int64 `xorm:"pk autoincr"` |
||||||
|
RepoID int64 `xorm:"INDEX UNIQUE(repo_index)"` |
||||||
|
Index int64 `xorm:"UNIQUE(repo_index)"` // Index in one repository.
|
||||||
|
PosterID int64 `xorm:"INDEX"` |
||||||
|
Title string `xorm:"name"` |
||||||
|
Content string `xorm:"TEXT"` |
||||||
|
MilestoneID int64 `xorm:"INDEX"` |
||||||
|
Priority int |
||||||
|
AssigneeID int64 `xorm:"INDEX"` |
||||||
|
IsClosed bool `xorm:"INDEX"` |
||||||
|
IsPull bool `xorm:"INDEX"` // Indicates whether is a pull request or not.
|
||||||
|
NumComments int |
||||||
|
Ref string |
||||||
|
|
||||||
|
DeadlineUnix util.TimeStamp `xorm:"INDEX"` |
||||||
|
CreatedUnix util.TimeStamp `xorm:"INDEX created"` |
||||||
|
UpdatedUnix util.TimeStamp `xorm:"INDEX updated"` |
||||||
|
ClosedUnix util.TimeStamp `xorm:"INDEX"` |
||||||
|
} |
||||||
|
|
||||||
|
allIssues := []Issue{} |
||||||
|
err := x.Find(&allIssues) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
// Create the table
|
||||||
|
type IssueAssignees struct { |
||||||
|
ID int64 `xorm:"pk autoincr"` |
||||||
|
AssigneeID int64 `xorm:"INDEX"` |
||||||
|
IssueID int64 `xorm:"INDEX"` |
||||||
|
} |
||||||
|
err = x.Sync2(IssueAssignees{}) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
// Range over all issues and insert a new entry for each issue/assignee
|
||||||
|
sess := x.NewSession() |
||||||
|
defer sess.Close() |
||||||
|
|
||||||
|
err = sess.Begin() |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
for _, issue := range allIssues { |
||||||
|
if issue.AssigneeID != 0 { |
||||||
|
_, err := sess.Insert(IssueAssignees{IssueID: issue.ID, AssigneeID: issue.AssigneeID}) |
||||||
|
if err != nil { |
||||||
|
sess.Rollback() |
||||||
|
return err |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Updated the comment table
|
||||||
|
type Comment struct { |
||||||
|
ID int64 `xorm:"pk autoincr"` |
||||||
|
Type int |
||||||
|
PosterID int64 `xorm:"INDEX"` |
||||||
|
IssueID int64 `xorm:"INDEX"` |
||||||
|
LabelID int64 |
||||||
|
OldMilestoneID int64 |
||||||
|
MilestoneID int64 |
||||||
|
OldAssigneeID int64 |
||||||
|
AssigneeID int64 |
||||||
|
RemovedAssignee bool |
||||||
|
OldTitle string |
||||||
|
NewTitle string |
||||||
|
|
||||||
|
CommitID int64 |
||||||
|
Line int64 |
||||||
|
Content string `xorm:"TEXT"` |
||||||
|
RenderedContent string `xorm:"-"` |
||||||
|
|
||||||
|
CreatedUnix util.TimeStamp `xorm:"INDEX created"` |
||||||
|
UpdatedUnix util.TimeStamp `xorm:"INDEX updated"` |
||||||
|
|
||||||
|
// Reference issue in commit message
|
||||||
|
CommitSHA string `xorm:"VARCHAR(40)"` |
||||||
|
} |
||||||
|
if err := x.Sync2(Comment{}); err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
// Migrate comments
|
||||||
|
// First update everything to not have nulls in db
|
||||||
|
if _, err := sess.Where("type = ?", 9).Cols("removed_assignee").Update(Comment{RemovedAssignee: false}); err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
allAssignementComments := []Comment{} |
||||||
|
if err := sess.Where("type = ?", 9).Find(&allAssignementComments); err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
for _, comment := range allAssignementComments { |
||||||
|
// Everytime where OldAssigneeID is > 0, the assignement was removed.
|
||||||
|
if comment.OldAssigneeID > 0 { |
||||||
|
_, err = sess.ID(comment.ID).Update(Comment{RemovedAssignee: true}) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
if err := dropTableColumns(sess, "issue", "assignee_id"); err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
if err := dropTableColumns(sess, "issue_user", "is_assigned"); err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
return sess.Commit() |
||||||
|
} |
||||||
File diff suppressed because one or more lines are too long
Loading…
Reference in new issue