Track assignee for issue (#808)

* track assignee for issue

* fix lint

* use getUserByID instead Get

1 files changed, 21 insertions, 0 deletions

diff --git a/models/issue.go b/models/issue.go
index 4b80c41325..7d7591837e 100644
--- a/models/issue.go
+++ b/models/issue.go
@@ -703,11 +703,23 @@ func (issue *Issue) ChangeContent(doer *User, content string) (err error) {
 
 // ChangeAssignee changes the Asssignee field of this issue.
 func (issue *Issue) ChangeAssignee(doer *User, assigneeID int64) (err error) {
+	var oldAssigneeID = issue.AssigneeID
 	issue.AssigneeID = assigneeID
 	if err = UpdateIssueUserByAssignee(issue); err != nil {
 		return fmt.Errorf("UpdateIssueUserByAssignee: %v", err)
 	}
 
+	sess := x.NewSession()
+	defer sess.Close()
+
+	if err = issue.loadRepo(sess); err != nil {
+		return fmt.Errorf("loadRepo: %v", err)
+	}
+
+	if _, err = createAssigneeComment(sess, doer, issue.Repo, issue, oldAssigneeID, assigneeID); err != nil {
+		return fmt.Errorf("createAssigneeComment: %v", err)
+	}
+
 	issue.Assignee, err = GetUserByID(issue.AssigneeID)
 	if err != nil && !IsErrUserNotExist(err) {
 		log.Error(4, "GetUserByID [assignee_id: %v]: %v", issue.AssigneeID, err)
@@ -798,6 +810,15 @@ func newIssue(e *xorm.Session, doer *User, opts NewIssueOptions) (err error) {
 		}
 	}
 
+	if opts.Issue.AssigneeID > 0 {
+		if err = opts.Issue.loadRepo(e); err != nil {
+			return err
+		}
+		if _, err = createAssigneeComment(e, doer, opts.Issue.Repo, opts.Issue, -1, opts.Issue.AssigneeID); err != nil {
+			return err
+		}
+	}
+
 	if opts.IsPull {
 		_, err = e.Exec("UPDATE `repository` SET num_pulls = num_pulls + 1 WHERE id = ?", opts.Issue.RepoID)
 	} else {