mirror of
				https://github.com/go-gitea/gitea.git
				synced 2025-10-31 21:28:11 +09:00 
			
		
		
		
	partially fix #19345 This PR add some `Link` methods for different objects. The `Link` methods are not different from `HTMLURL`, they are lack of the absolute URL. And most of UI `HTMLURL` have been replaced to `Link` so that users can visit them from a different domain or IP. This PR also introduces a new javascript configuration `window.config.reqAppUrl` which is different from `appUrl` which is still an absolute url but the domain has been replaced to the current requested domain.
		
			
				
	
	
		
			823 lines
		
	
	
		
			23 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			823 lines
		
	
	
		
			23 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| // Copyright 2016 The Gitea Authors. All rights reserved.
 | |
| // SPDX-License-Identifier: MIT
 | |
| 
 | |
| package activities
 | |
| 
 | |
| import (
 | |
| 	"context"
 | |
| 	"fmt"
 | |
| 	"net/url"
 | |
| 	"strconv"
 | |
| 
 | |
| 	"code.gitea.io/gitea/models/db"
 | |
| 	issues_model "code.gitea.io/gitea/models/issues"
 | |
| 	"code.gitea.io/gitea/models/organization"
 | |
| 	access_model "code.gitea.io/gitea/models/perm/access"
 | |
| 	repo_model "code.gitea.io/gitea/models/repo"
 | |
| 	"code.gitea.io/gitea/models/unit"
 | |
| 	user_model "code.gitea.io/gitea/models/user"
 | |
| 	"code.gitea.io/gitea/modules/container"
 | |
| 	"code.gitea.io/gitea/modules/log"
 | |
| 	"code.gitea.io/gitea/modules/setting"
 | |
| 	"code.gitea.io/gitea/modules/timeutil"
 | |
| 
 | |
| 	"xorm.io/builder"
 | |
| 	"xorm.io/xorm"
 | |
| )
 | |
| 
 | |
| type (
 | |
| 	// NotificationStatus is the status of the notification (read or unread)
 | |
| 	NotificationStatus uint8
 | |
| 	// NotificationSource is the source of the notification (issue, PR, commit, etc)
 | |
| 	NotificationSource uint8
 | |
| )
 | |
| 
 | |
| const (
 | |
| 	// NotificationStatusUnread represents an unread notification
 | |
| 	NotificationStatusUnread NotificationStatus = iota + 1
 | |
| 	// NotificationStatusRead represents a read notification
 | |
| 	NotificationStatusRead
 | |
| 	// NotificationStatusPinned represents a pinned notification
 | |
| 	NotificationStatusPinned
 | |
| )
 | |
| 
 | |
| const (
 | |
| 	// NotificationSourceIssue is a notification of an issue
 | |
| 	NotificationSourceIssue NotificationSource = iota + 1
 | |
| 	// NotificationSourcePullRequest is a notification of a pull request
 | |
| 	NotificationSourcePullRequest
 | |
| 	// NotificationSourceCommit is a notification of a commit
 | |
| 	NotificationSourceCommit
 | |
| 	// NotificationSourceRepository is a notification for a repository
 | |
| 	NotificationSourceRepository
 | |
| )
 | |
| 
 | |
| // Notification represents a notification
 | |
| type Notification struct {
 | |
| 	ID     int64 `xorm:"pk autoincr"`
 | |
| 	UserID int64 `xorm:"INDEX NOT NULL"`
 | |
| 	RepoID int64 `xorm:"INDEX NOT NULL"`
 | |
| 
 | |
| 	Status NotificationStatus `xorm:"SMALLINT INDEX NOT NULL"`
 | |
| 	Source NotificationSource `xorm:"SMALLINT INDEX NOT NULL"`
 | |
| 
 | |
| 	IssueID   int64  `xorm:"INDEX NOT NULL"`
 | |
| 	CommitID  string `xorm:"INDEX"`
 | |
| 	CommentID int64
 | |
| 
 | |
| 	UpdatedBy int64 `xorm:"INDEX NOT NULL"`
 | |
| 
 | |
| 	Issue      *issues_model.Issue    `xorm:"-"`
 | |
| 	Repository *repo_model.Repository `xorm:"-"`
 | |
| 	Comment    *issues_model.Comment  `xorm:"-"`
 | |
| 	User       *user_model.User       `xorm:"-"`
 | |
| 
 | |
| 	CreatedUnix timeutil.TimeStamp `xorm:"created INDEX NOT NULL"`
 | |
| 	UpdatedUnix timeutil.TimeStamp `xorm:"updated INDEX NOT NULL"`
 | |
| }
 | |
| 
 | |
| func init() {
 | |
| 	db.RegisterModel(new(Notification))
 | |
| }
 | |
| 
 | |
| // FindNotificationOptions represent the filters for notifications. If an ID is 0 it will be ignored.
 | |
| type FindNotificationOptions struct {
 | |
| 	db.ListOptions
 | |
| 	UserID            int64
 | |
| 	RepoID            int64
 | |
| 	IssueID           int64
 | |
| 	Status            []NotificationStatus
 | |
| 	Source            []NotificationSource
 | |
| 	UpdatedAfterUnix  int64
 | |
| 	UpdatedBeforeUnix int64
 | |
| }
 | |
| 
 | |
| // ToCond will convert each condition into a xorm-Cond
 | |
| func (opts *FindNotificationOptions) ToCond() builder.Cond {
 | |
| 	cond := builder.NewCond()
 | |
| 	if opts.UserID != 0 {
 | |
| 		cond = cond.And(builder.Eq{"notification.user_id": opts.UserID})
 | |
| 	}
 | |
| 	if opts.RepoID != 0 {
 | |
| 		cond = cond.And(builder.Eq{"notification.repo_id": opts.RepoID})
 | |
| 	}
 | |
| 	if opts.IssueID != 0 {
 | |
| 		cond = cond.And(builder.Eq{"notification.issue_id": opts.IssueID})
 | |
| 	}
 | |
| 	if len(opts.Status) > 0 {
 | |
| 		cond = cond.And(builder.In("notification.status", opts.Status))
 | |
| 	}
 | |
| 	if len(opts.Source) > 0 {
 | |
| 		cond = cond.And(builder.In("notification.source", opts.Source))
 | |
| 	}
 | |
| 	if opts.UpdatedAfterUnix != 0 {
 | |
| 		cond = cond.And(builder.Gte{"notification.updated_unix": opts.UpdatedAfterUnix})
 | |
| 	}
 | |
| 	if opts.UpdatedBeforeUnix != 0 {
 | |
| 		cond = cond.And(builder.Lte{"notification.updated_unix": opts.UpdatedBeforeUnix})
 | |
| 	}
 | |
| 	return cond
 | |
| }
 | |
| 
 | |
| // ToSession will convert the given options to a xorm Session by using the conditions from ToCond and joining with issue table if required
 | |
| func (opts *FindNotificationOptions) ToSession(ctx context.Context) *xorm.Session {
 | |
| 	sess := db.GetEngine(ctx).Where(opts.ToCond())
 | |
| 	if opts.Page != 0 {
 | |
| 		sess = db.SetSessionPagination(sess, opts)
 | |
| 	}
 | |
| 	return sess
 | |
| }
 | |
| 
 | |
| // GetNotifications returns all notifications that fit to the given options.
 | |
| func GetNotifications(ctx context.Context, options *FindNotificationOptions) (nl NotificationList, err error) {
 | |
| 	err = options.ToSession(ctx).OrderBy("notification.updated_unix DESC").Find(&nl)
 | |
| 	return nl, err
 | |
| }
 | |
| 
 | |
| // CountNotifications count all notifications that fit to the given options and ignore pagination.
 | |
| func CountNotifications(ctx context.Context, opts *FindNotificationOptions) (int64, error) {
 | |
| 	return db.GetEngine(ctx).Where(opts.ToCond()).Count(&Notification{})
 | |
| }
 | |
| 
 | |
| // CreateRepoTransferNotification creates  notification for the user a repository was transferred to
 | |
| func CreateRepoTransferNotification(ctx context.Context, doer, newOwner *user_model.User, repo *repo_model.Repository) error {
 | |
| 	return db.WithTx(ctx, func(ctx context.Context) error {
 | |
| 		var notify []*Notification
 | |
| 
 | |
| 		if newOwner.IsOrganization() {
 | |
| 			users, err := organization.GetUsersWhoCanCreateOrgRepo(ctx, newOwner.ID)
 | |
| 			if err != nil || len(users) == 0 {
 | |
| 				return err
 | |
| 			}
 | |
| 			for i := range users {
 | |
| 				notify = append(notify, &Notification{
 | |
| 					UserID:    i,
 | |
| 					RepoID:    repo.ID,
 | |
| 					Status:    NotificationStatusUnread,
 | |
| 					UpdatedBy: doer.ID,
 | |
| 					Source:    NotificationSourceRepository,
 | |
| 				})
 | |
| 			}
 | |
| 		} else {
 | |
| 			notify = []*Notification{{
 | |
| 				UserID:    newOwner.ID,
 | |
| 				RepoID:    repo.ID,
 | |
| 				Status:    NotificationStatusUnread,
 | |
| 				UpdatedBy: doer.ID,
 | |
| 				Source:    NotificationSourceRepository,
 | |
| 			}}
 | |
| 		}
 | |
| 
 | |
| 		return db.Insert(ctx, notify)
 | |
| 	})
 | |
| }
 | |
| 
 | |
| // CreateOrUpdateIssueNotifications creates an issue notification
 | |
| // for each watcher, or updates it if already exists
 | |
| // receiverID > 0 just send to receiver, else send to all watcher
 | |
| func CreateOrUpdateIssueNotifications(issueID, commentID, notificationAuthorID, receiverID int64) error {
 | |
| 	ctx, committer, err := db.TxContext(db.DefaultContext)
 | |
| 	if err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 	defer committer.Close()
 | |
| 
 | |
| 	if err := createOrUpdateIssueNotifications(ctx, issueID, commentID, notificationAuthorID, receiverID); err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 
 | |
| 	return committer.Commit()
 | |
| }
 | |
| 
 | |
| func createOrUpdateIssueNotifications(ctx context.Context, issueID, commentID, notificationAuthorID, receiverID int64) error {
 | |
| 	// init
 | |
| 	var toNotify container.Set[int64]
 | |
| 	notifications, err := getNotificationsByIssueID(ctx, issueID)
 | |
| 	if err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 
 | |
| 	issue, err := issues_model.GetIssueByID(ctx, issueID)
 | |
| 	if err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 
 | |
| 	if receiverID > 0 {
 | |
| 		toNotify = make(container.Set[int64], 1)
 | |
| 		toNotify.Add(receiverID)
 | |
| 	} else {
 | |
| 		toNotify = make(container.Set[int64], 32)
 | |
| 		issueWatches, err := issues_model.GetIssueWatchersIDs(ctx, issueID, true)
 | |
| 		if err != nil {
 | |
| 			return err
 | |
| 		}
 | |
| 		toNotify.AddMultiple(issueWatches...)
 | |
| 		if !(issue.IsPull && issues_model.HasWorkInProgressPrefix(issue.Title)) {
 | |
| 			repoWatches, err := repo_model.GetRepoWatchersIDs(ctx, issue.RepoID)
 | |
| 			if err != nil {
 | |
| 				return err
 | |
| 			}
 | |
| 			toNotify.AddMultiple(repoWatches...)
 | |
| 		}
 | |
| 		issueParticipants, err := issue.GetParticipantIDsByIssue(ctx)
 | |
| 		if err != nil {
 | |
| 			return err
 | |
| 		}
 | |
| 		toNotify.AddMultiple(issueParticipants...)
 | |
| 
 | |
| 		// dont notify user who cause notification
 | |
| 		delete(toNotify, notificationAuthorID)
 | |
| 		// explicit unwatch on issue
 | |
| 		issueUnWatches, err := issues_model.GetIssueWatchersIDs(ctx, issueID, false)
 | |
| 		if err != nil {
 | |
| 			return err
 | |
| 		}
 | |
| 		for _, id := range issueUnWatches {
 | |
| 			toNotify.Remove(id)
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	err = issue.LoadRepo(ctx)
 | |
| 	if err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 
 | |
| 	// notify
 | |
| 	for userID := range toNotify {
 | |
| 		issue.Repo.Units = nil
 | |
| 		user, err := user_model.GetUserByID(ctx, userID)
 | |
| 		if err != nil {
 | |
| 			if user_model.IsErrUserNotExist(err) {
 | |
| 				continue
 | |
| 			}
 | |
| 
 | |
| 			return err
 | |
| 		}
 | |
| 		if issue.IsPull && !access_model.CheckRepoUnitUser(ctx, issue.Repo, user, unit.TypePullRequests) {
 | |
| 			continue
 | |
| 		}
 | |
| 		if !issue.IsPull && !access_model.CheckRepoUnitUser(ctx, issue.Repo, user, unit.TypeIssues) {
 | |
| 			continue
 | |
| 		}
 | |
| 
 | |
| 		if notificationExists(notifications, issue.ID, userID) {
 | |
| 			if err = updateIssueNotification(ctx, userID, issue.ID, commentID, notificationAuthorID); err != nil {
 | |
| 				return err
 | |
| 			}
 | |
| 			continue
 | |
| 		}
 | |
| 		if err = createIssueNotification(ctx, userID, issue, commentID, notificationAuthorID); err != nil {
 | |
| 			return err
 | |
| 		}
 | |
| 	}
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| func getNotificationsByIssueID(ctx context.Context, issueID int64) (notifications []*Notification, err error) {
 | |
| 	err = db.GetEngine(ctx).
 | |
| 		Where("issue_id = ?", issueID).
 | |
| 		Find(¬ifications)
 | |
| 	return notifications, err
 | |
| }
 | |
| 
 | |
| func notificationExists(notifications []*Notification, issueID, userID int64) bool {
 | |
| 	for _, notification := range notifications {
 | |
| 		if notification.IssueID == issueID && notification.UserID == userID {
 | |
| 			return true
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	return false
 | |
| }
 | |
| 
 | |
| func createIssueNotification(ctx context.Context, userID int64, issue *issues_model.Issue, commentID, updatedByID int64) error {
 | |
| 	notification := &Notification{
 | |
| 		UserID:    userID,
 | |
| 		RepoID:    issue.RepoID,
 | |
| 		Status:    NotificationStatusUnread,
 | |
| 		IssueID:   issue.ID,
 | |
| 		CommentID: commentID,
 | |
| 		UpdatedBy: updatedByID,
 | |
| 	}
 | |
| 
 | |
| 	if issue.IsPull {
 | |
| 		notification.Source = NotificationSourcePullRequest
 | |
| 	} else {
 | |
| 		notification.Source = NotificationSourceIssue
 | |
| 	}
 | |
| 
 | |
| 	return db.Insert(ctx, notification)
 | |
| }
 | |
| 
 | |
| func updateIssueNotification(ctx context.Context, userID, issueID, commentID, updatedByID int64) error {
 | |
| 	notification, err := getIssueNotification(ctx, userID, issueID)
 | |
| 	if err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 
 | |
| 	// NOTICE: Only update comment id when the before notification on this issue is read, otherwise you may miss some old comments.
 | |
| 	// But we need update update_by so that the notification will be reorder
 | |
| 	var cols []string
 | |
| 	if notification.Status == NotificationStatusRead {
 | |
| 		notification.Status = NotificationStatusUnread
 | |
| 		notification.CommentID = commentID
 | |
| 		cols = []string{"status", "update_by", "comment_id"}
 | |
| 	} else {
 | |
| 		notification.UpdatedBy = updatedByID
 | |
| 		cols = []string{"update_by"}
 | |
| 	}
 | |
| 
 | |
| 	_, err = db.GetEngine(ctx).ID(notification.ID).Cols(cols...).Update(notification)
 | |
| 	return err
 | |
| }
 | |
| 
 | |
| func getIssueNotification(ctx context.Context, userID, issueID int64) (*Notification, error) {
 | |
| 	notification := new(Notification)
 | |
| 	_, err := db.GetEngine(ctx).
 | |
| 		Where("user_id = ?", userID).
 | |
| 		And("issue_id = ?", issueID).
 | |
| 		Get(notification)
 | |
| 	return notification, err
 | |
| }
 | |
| 
 | |
| // NotificationsForUser returns notifications for a given user and status
 | |
| func NotificationsForUser(ctx context.Context, user *user_model.User, statuses []NotificationStatus, page, perPage int) (notifications NotificationList, err error) {
 | |
| 	if len(statuses) == 0 {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	sess := db.GetEngine(ctx).
 | |
| 		Where("user_id = ?", user.ID).
 | |
| 		In("status", statuses).
 | |
| 		OrderBy("updated_unix DESC")
 | |
| 
 | |
| 	if page > 0 && perPage > 0 {
 | |
| 		sess.Limit(perPage, (page-1)*perPage)
 | |
| 	}
 | |
| 
 | |
| 	err = sess.Find(¬ifications)
 | |
| 	return notifications, err
 | |
| }
 | |
| 
 | |
| // CountUnread count unread notifications for a user
 | |
| func CountUnread(ctx context.Context, userID int64) int64 {
 | |
| 	exist, err := db.GetEngine(ctx).Where("user_id = ?", userID).And("status = ?", NotificationStatusUnread).Count(new(Notification))
 | |
| 	if err != nil {
 | |
| 		log.Error("countUnread", err)
 | |
| 		return 0
 | |
| 	}
 | |
| 	return exist
 | |
| }
 | |
| 
 | |
| // LoadAttributes load Repo Issue User and Comment if not loaded
 | |
| func (n *Notification) LoadAttributes(ctx context.Context) (err error) {
 | |
| 	if err = n.loadRepo(ctx); err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 	if err = n.loadIssue(ctx); err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 	if err = n.loadUser(ctx); err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 	if err = n.loadComment(ctx); err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 	return err
 | |
| }
 | |
| 
 | |
| func (n *Notification) loadRepo(ctx context.Context) (err error) {
 | |
| 	if n.Repository == nil {
 | |
| 		n.Repository, err = repo_model.GetRepositoryByID(ctx, n.RepoID)
 | |
| 		if err != nil {
 | |
| 			return fmt.Errorf("getRepositoryByID [%d]: %w", n.RepoID, err)
 | |
| 		}
 | |
| 	}
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| func (n *Notification) loadIssue(ctx context.Context) (err error) {
 | |
| 	if n.Issue == nil && n.IssueID != 0 {
 | |
| 		n.Issue, err = issues_model.GetIssueByID(ctx, n.IssueID)
 | |
| 		if err != nil {
 | |
| 			return fmt.Errorf("getIssueByID [%d]: %w", n.IssueID, err)
 | |
| 		}
 | |
| 		return n.Issue.LoadAttributes(ctx)
 | |
| 	}
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| func (n *Notification) loadComment(ctx context.Context) (err error) {
 | |
| 	if n.Comment == nil && n.CommentID != 0 {
 | |
| 		n.Comment, err = issues_model.GetCommentByID(ctx, n.CommentID)
 | |
| 		if err != nil {
 | |
| 			if issues_model.IsErrCommentNotExist(err) {
 | |
| 				return issues_model.ErrCommentNotExist{
 | |
| 					ID:      n.CommentID,
 | |
| 					IssueID: n.IssueID,
 | |
| 				}
 | |
| 			}
 | |
| 			return err
 | |
| 		}
 | |
| 	}
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| func (n *Notification) loadUser(ctx context.Context) (err error) {
 | |
| 	if n.User == nil {
 | |
| 		n.User, err = user_model.GetUserByID(ctx, n.UserID)
 | |
| 		if err != nil {
 | |
| 			return fmt.Errorf("getUserByID [%d]: %w", n.UserID, err)
 | |
| 		}
 | |
| 	}
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| // GetRepo returns the repo of the notification
 | |
| func (n *Notification) GetRepo() (*repo_model.Repository, error) {
 | |
| 	return n.Repository, n.loadRepo(db.DefaultContext)
 | |
| }
 | |
| 
 | |
| // GetIssue returns the issue of the notification
 | |
| func (n *Notification) GetIssue() (*issues_model.Issue, error) {
 | |
| 	return n.Issue, n.loadIssue(db.DefaultContext)
 | |
| }
 | |
| 
 | |
| // HTMLURL formats a URL-string to the notification
 | |
| func (n *Notification) HTMLURL() string {
 | |
| 	switch n.Source {
 | |
| 	case NotificationSourceIssue, NotificationSourcePullRequest:
 | |
| 		if n.Comment != nil {
 | |
| 			return n.Comment.HTMLURL()
 | |
| 		}
 | |
| 		return n.Issue.HTMLURL()
 | |
| 	case NotificationSourceCommit:
 | |
| 		return n.Repository.HTMLURL() + "/commit/" + url.PathEscape(n.CommitID)
 | |
| 	case NotificationSourceRepository:
 | |
| 		return n.Repository.HTMLURL()
 | |
| 	}
 | |
| 	return ""
 | |
| }
 | |
| 
 | |
| // Link formats a relative URL-string to the notification
 | |
| func (n *Notification) Link() string {
 | |
| 	switch n.Source {
 | |
| 	case NotificationSourceIssue, NotificationSourcePullRequest:
 | |
| 		if n.Comment != nil {
 | |
| 			return n.Comment.Link()
 | |
| 		}
 | |
| 		return n.Issue.Link()
 | |
| 	case NotificationSourceCommit:
 | |
| 		return n.Repository.Link() + "/commit/" + url.PathEscape(n.CommitID)
 | |
| 	case NotificationSourceRepository:
 | |
| 		return n.Repository.Link()
 | |
| 	}
 | |
| 	return ""
 | |
| }
 | |
| 
 | |
| // APIURL formats a URL-string to the notification
 | |
| func (n *Notification) APIURL() string {
 | |
| 	return setting.AppURL + "api/v1/notifications/threads/" + strconv.FormatInt(n.ID, 10)
 | |
| }
 | |
| 
 | |
| // NotificationList contains a list of notifications
 | |
| type NotificationList []*Notification
 | |
| 
 | |
| // LoadAttributes load Repo Issue User and Comment if not loaded
 | |
| func (nl NotificationList) LoadAttributes(ctx context.Context) error {
 | |
| 	var err error
 | |
| 	for i := 0; i < len(nl); i++ {
 | |
| 		err = nl[i].LoadAttributes(ctx)
 | |
| 		if err != nil && !issues_model.IsErrCommentNotExist(err) {
 | |
| 			return err
 | |
| 		}
 | |
| 	}
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| func (nl NotificationList) getPendingRepoIDs() []int64 {
 | |
| 	ids := make(container.Set[int64], len(nl))
 | |
| 	for _, notification := range nl {
 | |
| 		if notification.Repository != nil {
 | |
| 			continue
 | |
| 		}
 | |
| 		ids.Add(notification.RepoID)
 | |
| 	}
 | |
| 	return ids.Values()
 | |
| }
 | |
| 
 | |
| // LoadRepos loads repositories from database
 | |
| func (nl NotificationList) LoadRepos(ctx context.Context) (repo_model.RepositoryList, []int, error) {
 | |
| 	if len(nl) == 0 {
 | |
| 		return repo_model.RepositoryList{}, []int{}, nil
 | |
| 	}
 | |
| 
 | |
| 	repoIDs := nl.getPendingRepoIDs()
 | |
| 	repos := make(map[int64]*repo_model.Repository, len(repoIDs))
 | |
| 	left := len(repoIDs)
 | |
| 	for left > 0 {
 | |
| 		limit := db.DefaultMaxInSize
 | |
| 		if left < limit {
 | |
| 			limit = left
 | |
| 		}
 | |
| 		rows, err := db.GetEngine(ctx).
 | |
| 			In("id", repoIDs[:limit]).
 | |
| 			Rows(new(repo_model.Repository))
 | |
| 		if err != nil {
 | |
| 			return nil, nil, err
 | |
| 		}
 | |
| 
 | |
| 		for rows.Next() {
 | |
| 			var repo repo_model.Repository
 | |
| 			err = rows.Scan(&repo)
 | |
| 			if err != nil {
 | |
| 				rows.Close()
 | |
| 				return nil, nil, err
 | |
| 			}
 | |
| 
 | |
| 			repos[repo.ID] = &repo
 | |
| 		}
 | |
| 		_ = rows.Close()
 | |
| 
 | |
| 		left -= limit
 | |
| 		repoIDs = repoIDs[limit:]
 | |
| 	}
 | |
| 
 | |
| 	failed := []int{}
 | |
| 
 | |
| 	reposList := make(repo_model.RepositoryList, 0, len(repoIDs))
 | |
| 	for i, notification := range nl {
 | |
| 		if notification.Repository == nil {
 | |
| 			notification.Repository = repos[notification.RepoID]
 | |
| 		}
 | |
| 		if notification.Repository == nil {
 | |
| 			log.Error("Notification[%d]: RepoID: %d not found", notification.ID, notification.RepoID)
 | |
| 			failed = append(failed, i)
 | |
| 			continue
 | |
| 		}
 | |
| 		var found bool
 | |
| 		for _, r := range reposList {
 | |
| 			if r.ID == notification.RepoID {
 | |
| 				found = true
 | |
| 				break
 | |
| 			}
 | |
| 		}
 | |
| 		if !found {
 | |
| 			reposList = append(reposList, notification.Repository)
 | |
| 		}
 | |
| 	}
 | |
| 	return reposList, failed, nil
 | |
| }
 | |
| 
 | |
| func (nl NotificationList) getPendingIssueIDs() []int64 {
 | |
| 	ids := make(container.Set[int64], len(nl))
 | |
| 	for _, notification := range nl {
 | |
| 		if notification.Issue != nil {
 | |
| 			continue
 | |
| 		}
 | |
| 		ids.Add(notification.IssueID)
 | |
| 	}
 | |
| 	return ids.Values()
 | |
| }
 | |
| 
 | |
| // LoadIssues loads issues from database
 | |
| func (nl NotificationList) LoadIssues(ctx context.Context) ([]int, error) {
 | |
| 	if len(nl) == 0 {
 | |
| 		return []int{}, nil
 | |
| 	}
 | |
| 
 | |
| 	issueIDs := nl.getPendingIssueIDs()
 | |
| 	issues := make(map[int64]*issues_model.Issue, len(issueIDs))
 | |
| 	left := len(issueIDs)
 | |
| 	for left > 0 {
 | |
| 		limit := db.DefaultMaxInSize
 | |
| 		if left < limit {
 | |
| 			limit = left
 | |
| 		}
 | |
| 		rows, err := db.GetEngine(ctx).
 | |
| 			In("id", issueIDs[:limit]).
 | |
| 			Rows(new(issues_model.Issue))
 | |
| 		if err != nil {
 | |
| 			return nil, err
 | |
| 		}
 | |
| 
 | |
| 		for rows.Next() {
 | |
| 			var issue issues_model.Issue
 | |
| 			err = rows.Scan(&issue)
 | |
| 			if err != nil {
 | |
| 				rows.Close()
 | |
| 				return nil, err
 | |
| 			}
 | |
| 
 | |
| 			issues[issue.ID] = &issue
 | |
| 		}
 | |
| 		_ = rows.Close()
 | |
| 
 | |
| 		left -= limit
 | |
| 		issueIDs = issueIDs[limit:]
 | |
| 	}
 | |
| 
 | |
| 	failures := []int{}
 | |
| 
 | |
| 	for i, notification := range nl {
 | |
| 		if notification.Issue == nil {
 | |
| 			notification.Issue = issues[notification.IssueID]
 | |
| 			if notification.Issue == nil {
 | |
| 				if notification.IssueID != 0 {
 | |
| 					log.Error("Notification[%d]: IssueID: %d Not Found", notification.ID, notification.IssueID)
 | |
| 					failures = append(failures, i)
 | |
| 				}
 | |
| 				continue
 | |
| 			}
 | |
| 			notification.Issue.Repo = notification.Repository
 | |
| 		}
 | |
| 	}
 | |
| 	return failures, nil
 | |
| }
 | |
| 
 | |
| // Without returns the notification list without the failures
 | |
| func (nl NotificationList) Without(failures []int) NotificationList {
 | |
| 	if len(failures) == 0 {
 | |
| 		return nl
 | |
| 	}
 | |
| 	remaining := make([]*Notification, 0, len(nl))
 | |
| 	last := -1
 | |
| 	var i int
 | |
| 	for _, i = range failures {
 | |
| 		remaining = append(remaining, nl[last+1:i]...)
 | |
| 		last = i
 | |
| 	}
 | |
| 	if len(nl) > i {
 | |
| 		remaining = append(remaining, nl[i+1:]...)
 | |
| 	}
 | |
| 	return remaining
 | |
| }
 | |
| 
 | |
| func (nl NotificationList) getPendingCommentIDs() []int64 {
 | |
| 	ids := make(container.Set[int64], len(nl))
 | |
| 	for _, notification := range nl {
 | |
| 		if notification.CommentID == 0 || notification.Comment != nil {
 | |
| 			continue
 | |
| 		}
 | |
| 		ids.Add(notification.CommentID)
 | |
| 	}
 | |
| 	return ids.Values()
 | |
| }
 | |
| 
 | |
| // LoadComments loads comments from database
 | |
| func (nl NotificationList) LoadComments(ctx context.Context) ([]int, error) {
 | |
| 	if len(nl) == 0 {
 | |
| 		return []int{}, nil
 | |
| 	}
 | |
| 
 | |
| 	commentIDs := nl.getPendingCommentIDs()
 | |
| 	comments := make(map[int64]*issues_model.Comment, len(commentIDs))
 | |
| 	left := len(commentIDs)
 | |
| 	for left > 0 {
 | |
| 		limit := db.DefaultMaxInSize
 | |
| 		if left < limit {
 | |
| 			limit = left
 | |
| 		}
 | |
| 		rows, err := db.GetEngine(ctx).
 | |
| 			In("id", commentIDs[:limit]).
 | |
| 			Rows(new(issues_model.Comment))
 | |
| 		if err != nil {
 | |
| 			return nil, err
 | |
| 		}
 | |
| 
 | |
| 		for rows.Next() {
 | |
| 			var comment issues_model.Comment
 | |
| 			err = rows.Scan(&comment)
 | |
| 			if err != nil {
 | |
| 				rows.Close()
 | |
| 				return nil, err
 | |
| 			}
 | |
| 
 | |
| 			comments[comment.ID] = &comment
 | |
| 		}
 | |
| 		_ = rows.Close()
 | |
| 
 | |
| 		left -= limit
 | |
| 		commentIDs = commentIDs[limit:]
 | |
| 	}
 | |
| 
 | |
| 	failures := []int{}
 | |
| 	for i, notification := range nl {
 | |
| 		if notification.CommentID > 0 && notification.Comment == nil && comments[notification.CommentID] != nil {
 | |
| 			notification.Comment = comments[notification.CommentID]
 | |
| 			if notification.Comment == nil {
 | |
| 				log.Error("Notification[%d]: CommentID[%d] failed to load", notification.ID, notification.CommentID)
 | |
| 				failures = append(failures, i)
 | |
| 				continue
 | |
| 			}
 | |
| 			notification.Comment.Issue = notification.Issue
 | |
| 		}
 | |
| 	}
 | |
| 	return failures, nil
 | |
| }
 | |
| 
 | |
| // GetNotificationCount returns the notification count for user
 | |
| func GetNotificationCount(ctx context.Context, user *user_model.User, status NotificationStatus) (count int64, err error) {
 | |
| 	count, err = db.GetEngine(ctx).
 | |
| 		Where("user_id = ?", user.ID).
 | |
| 		And("status = ?", status).
 | |
| 		Count(&Notification{})
 | |
| 	return count, err
 | |
| }
 | |
| 
 | |
| // UserIDCount is a simple coalition of UserID and Count
 | |
| type UserIDCount struct {
 | |
| 	UserID int64
 | |
| 	Count  int64
 | |
| }
 | |
| 
 | |
| // GetUIDsAndNotificationCounts between the two provided times
 | |
| func GetUIDsAndNotificationCounts(since, until timeutil.TimeStamp) ([]UserIDCount, error) {
 | |
| 	sql := `SELECT user_id, count(*) AS count FROM notification ` +
 | |
| 		`WHERE user_id IN (SELECT user_id FROM notification WHERE updated_unix >= ? AND ` +
 | |
| 		`updated_unix < ?) AND status = ? GROUP BY user_id`
 | |
| 	var res []UserIDCount
 | |
| 	return res, db.GetEngine(db.DefaultContext).SQL(sql, since, until, NotificationStatusUnread).Find(&res)
 | |
| }
 | |
| 
 | |
| // SetIssueReadBy sets issue to be read by given user.
 | |
| func SetIssueReadBy(ctx context.Context, issueID, userID int64) error {
 | |
| 	if err := issues_model.UpdateIssueUserByRead(userID, issueID); err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 
 | |
| 	return setIssueNotificationStatusReadIfUnread(ctx, userID, issueID)
 | |
| }
 | |
| 
 | |
| func setIssueNotificationStatusReadIfUnread(ctx context.Context, userID, issueID int64) error {
 | |
| 	notification, err := getIssueNotification(ctx, userID, issueID)
 | |
| 	// ignore if not exists
 | |
| 	if err != nil {
 | |
| 		return nil
 | |
| 	}
 | |
| 
 | |
| 	if notification.Status != NotificationStatusUnread {
 | |
| 		return nil
 | |
| 	}
 | |
| 
 | |
| 	notification.Status = NotificationStatusRead
 | |
| 
 | |
| 	_, err = db.GetEngine(ctx).ID(notification.ID).Update(notification)
 | |
| 	return err
 | |
| }
 | |
| 
 | |
| // SetRepoReadBy sets repo to be visited by given user.
 | |
| func SetRepoReadBy(ctx context.Context, userID, repoID int64) error {
 | |
| 	_, err := db.GetEngine(ctx).Where(builder.Eq{
 | |
| 		"user_id": userID,
 | |
| 		"status":  NotificationStatusUnread,
 | |
| 		"source":  NotificationSourceRepository,
 | |
| 		"repo_id": repoID,
 | |
| 	}).Cols("status").Update(&Notification{Status: NotificationStatusRead})
 | |
| 	return err
 | |
| }
 | |
| 
 | |
| // SetNotificationStatus change the notification status
 | |
| func SetNotificationStatus(ctx context.Context, notificationID int64, user *user_model.User, status NotificationStatus) (*Notification, error) {
 | |
| 	notification, err := GetNotificationByID(ctx, notificationID)
 | |
| 	if err != nil {
 | |
| 		return notification, err
 | |
| 	}
 | |
| 
 | |
| 	if notification.UserID != user.ID {
 | |
| 		return nil, fmt.Errorf("Can't change notification of another user: %d, %d", notification.UserID, user.ID)
 | |
| 	}
 | |
| 
 | |
| 	notification.Status = status
 | |
| 
 | |
| 	_, err = db.GetEngine(ctx).ID(notificationID).Update(notification)
 | |
| 	return notification, err
 | |
| }
 | |
| 
 | |
| // GetNotificationByID return notification by ID
 | |
| func GetNotificationByID(ctx context.Context, notificationID int64) (*Notification, error) {
 | |
| 	notification := new(Notification)
 | |
| 	ok, err := db.GetEngine(ctx).
 | |
| 		Where("id = ?", notificationID).
 | |
| 		Get(notification)
 | |
| 	if err != nil {
 | |
| 		return nil, err
 | |
| 	}
 | |
| 
 | |
| 	if !ok {
 | |
| 		return nil, db.ErrNotExist{Resource: "notification", ID: notificationID}
 | |
| 	}
 | |
| 
 | |
| 	return notification, nil
 | |
| }
 | |
| 
 | |
| // UpdateNotificationStatuses updates the statuses of all of a user's notifications that are of the currentStatus type to the desiredStatus
 | |
| func UpdateNotificationStatuses(ctx context.Context, user *user_model.User, currentStatus, desiredStatus NotificationStatus) error {
 | |
| 	n := &Notification{Status: desiredStatus, UpdatedBy: user.ID}
 | |
| 	_, err := db.GetEngine(ctx).
 | |
| 		Where("user_id = ? AND status = ?", user.ID, currentStatus).
 | |
| 		Cols("status", "updated_by", "updated_unix").
 | |
| 		Update(n)
 | |
| 	return err
 | |
| }
 |