mirror of
				https://gitee.com/gitea/gitea
				synced 2025-11-04 08:30:25 +08:00 
			
		
		
		
	* Check if project has the same repository id with issue when assign project to issue * Check if issue's repository id match project's repository id * Add more permission checking * Remove invalid argument * Fix errors * Add generic check * Remove duplicated check * Return error + add check for new issues * Apply suggestions from code review Co-authored-by: KN4CK3R <admin@oldschoolhack.me> Co-authored-by: Gusted <williamzijl7@hotmail.com> Co-authored-by: KN4CK3R <admin@oldschoolhack.me> Co-authored-by: 6543 <6543@obermui.de>
		
			
				
	
	
		
			86 lines
		
	
	
		
			2.2 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			86 lines
		
	
	
		
			2.2 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
// Copyright 2019 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 issue
 | 
						|
 | 
						|
import (
 | 
						|
	"context"
 | 
						|
	"fmt"
 | 
						|
 | 
						|
	"code.gitea.io/gitea/models/db"
 | 
						|
	issues_model "code.gitea.io/gitea/models/issues"
 | 
						|
	user_model "code.gitea.io/gitea/models/user"
 | 
						|
	"code.gitea.io/gitea/modules/notification"
 | 
						|
)
 | 
						|
 | 
						|
func changeMilestoneAssign(ctx context.Context, doer *user_model.User, issue *issues_model.Issue, oldMilestoneID int64) error {
 | 
						|
	// Only check if milestone exists if we don't remove it.
 | 
						|
	if issue.MilestoneID > 0 {
 | 
						|
		has, err := issues_model.HasMilestoneByRepoID(ctx, issue.RepoID, issue.MilestoneID)
 | 
						|
		if err != nil {
 | 
						|
			return fmt.Errorf("HasMilestoneByRepoID: %v", err)
 | 
						|
		}
 | 
						|
		if !has {
 | 
						|
			return fmt.Errorf("HasMilestoneByRepoID: issue doesn't exist")
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	if err := issues_model.UpdateIssueCols(ctx, issue, "milestone_id"); err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
 | 
						|
	if oldMilestoneID > 0 {
 | 
						|
		if err := issues_model.UpdateMilestoneCounters(ctx, oldMilestoneID); err != nil {
 | 
						|
			return err
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	if issue.MilestoneID > 0 {
 | 
						|
		if err := issues_model.UpdateMilestoneCounters(ctx, issue.MilestoneID); err != nil {
 | 
						|
			return err
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	if oldMilestoneID > 0 || issue.MilestoneID > 0 {
 | 
						|
		if err := issue.LoadRepo(ctx); err != nil {
 | 
						|
			return err
 | 
						|
		}
 | 
						|
 | 
						|
		opts := &issues_model.CreateCommentOptions{
 | 
						|
			Type:           issues_model.CommentTypeMilestone,
 | 
						|
			Doer:           doer,
 | 
						|
			Repo:           issue.Repo,
 | 
						|
			Issue:          issue,
 | 
						|
			OldMilestoneID: oldMilestoneID,
 | 
						|
			MilestoneID:    issue.MilestoneID,
 | 
						|
		}
 | 
						|
		if _, err := issues_model.CreateCommentCtx(ctx, opts); err != nil {
 | 
						|
			return err
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	return nil
 | 
						|
}
 | 
						|
 | 
						|
// ChangeMilestoneAssign changes assignment of milestone for issue.
 | 
						|
func ChangeMilestoneAssign(issue *issues_model.Issue, doer *user_model.User, oldMilestoneID int64) (err error) {
 | 
						|
	ctx, committer, err := db.TxContext()
 | 
						|
	if err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
	defer committer.Close()
 | 
						|
 | 
						|
	if err = changeMilestoneAssign(ctx, doer, issue, oldMilestoneID); err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
 | 
						|
	if err = committer.Commit(); err != nil {
 | 
						|
		return fmt.Errorf("Commit: %v", err)
 | 
						|
	}
 | 
						|
 | 
						|
	notification.NotifyIssueChangeMilestone(doer, issue, oldMilestoneID)
 | 
						|
 | 
						|
	return nil
 | 
						|
}
 |