Creating notifications on new issue

models/notification.go

@@ -52,3 +52,83 @@ func (n *Notification) BeforeUpdate() {
 	n.Updated = now
 	n.UpdatedUnix = nowUnix
 }
+
+func CreateOrUpdateIssueNotifications(issue *Issue) error {
+	watches, err := getWatchers(x, issue.RepoID)
+	if err != nil {
+		return err
+	}
+
+	sess := x.NewSession()
+	if err := sess.Begin(); err != nil {
+		return err
+	}
+
+	defer sess.Close()
+
+	for _, watch := range watches {
+		exists, err := issueNotificationExists(sess, watch.UserID, watch.RepoID)
+		if err != nil {
+			return err
+		}
+
+		if exists {
+			err = updateIssueNotification(sess, watch.UserID, issue.ID)
+		} else {
+			err = createIssueNotification(sess, watch.UserID, issue)
+		}
+
+		if err != nil {
+			return err
+		}
+	}
+
+	return sess.Commit()
+}
+
+func issueNotificationExists(e Engine, userID, issueID int64) (bool, error) {
+	count, err := e.
+		Where("user_id = ?", userID).
+		And("issue_id = ?", issueID).
+		Count(Notification{})
+	return count > 0, err
+}
+
+func createIssueNotification(e Engine, userID int64, issue *Issue) error {
+	notification := &Notification{
+		UserID:  userID,
+		RepoID:  issue.RepoID,
+		Status:  NotificationStatusUnread,
+		IssueID: issue.ID,
+	}
+
+	if issue.IsPull {
+		notification.Source = NotificationSourcePullRequest
+	} else {
+		notification.Source = NotificationSourceIssue
+	}
+
+	_, err := e.Insert(notification)
+	return err
+}
+
+func updateIssueNotification(e Engine, userID, issueID int64) error {
+	notification, err := getIssueNotification(e, userID, issueID)
+	if err != nil {
+		return err
+	}
+
+	notification.Status = NotificationStatusUnread
+
+	_, err = e.Id(notification.ID).Update(notification)
+	return err
+}
+
+func getIssueNotification(e Engine, userID, issueID int64) (*Notification, error) {
+	notification := new(Notification)
+	_, err := e.
+		Where("user_id = ?").
+		And("issue_id = ?", issueID).
+		Get(notification)
+	return notification, err
+}