109 lines
		
	
	
		
			3.1 KiB
		
	
	
	
		
			Go
		
	
	
	
			
		
		
	
	
			109 lines
		
	
	
		
			3.1 KiB
		
	
	
	
		
			Go
		
	
	
	
| // Copyright 2020 The Gitea Authors. All rights reserved.
 | |
| // SPDX-License-Identifier: MIT
 | |
| 
 | |
| package project
 | |
| 
 | |
| import (
 | |
| 	"context"
 | |
| 	"fmt"
 | |
| 
 | |
| 	"code.gitea.io/gitea/models/db"
 | |
| 	"code.gitea.io/gitea/modules/log"
 | |
| )
 | |
| 
 | |
| // ProjectIssue saves relation from issue to a project
 | |
| type ProjectIssue struct { //revive:disable-line:exported
 | |
| 	ID        int64 `xorm:"pk autoincr"`
 | |
| 	IssueID   int64 `xorm:"INDEX"`
 | |
| 	ProjectID int64 `xorm:"INDEX"`
 | |
| 
 | |
| 	// If 0, then it has not been added to a specific board in the project
 | |
| 	ProjectBoardID int64 `xorm:"INDEX"`
 | |
| 
 | |
| 	// the sorting order on the board
 | |
| 	Sorting int64 `xorm:"NOT NULL DEFAULT 0"`
 | |
| }
 | |
| 
 | |
| func init() {
 | |
| 	db.RegisterModel(new(ProjectIssue))
 | |
| }
 | |
| 
 | |
| func deleteProjectIssuesByProjectID(ctx context.Context, projectID int64) error {
 | |
| 	_, err := db.GetEngine(ctx).Where("project_id=?", projectID).Delete(&ProjectIssue{})
 | |
| 	return err
 | |
| }
 | |
| 
 | |
| // NumIssues return counter of all issues assigned to a project
 | |
| func (p *Project) NumIssues() int {
 | |
| 	c, err := db.GetEngine(db.DefaultContext).Table("project_issue").
 | |
| 		Where("project_id=?", p.ID).
 | |
| 		GroupBy("issue_id").
 | |
| 		Cols("issue_id").
 | |
| 		Count()
 | |
| 	if err != nil {
 | |
| 		log.Error("NumIssues: %v", err)
 | |
| 		return 0
 | |
| 	}
 | |
| 	return int(c)
 | |
| }
 | |
| 
 | |
| // NumClosedIssues return counter of closed issues assigned to a project
 | |
| func (p *Project) NumClosedIssues() int {
 | |
| 	c, err := db.GetEngine(db.DefaultContext).Table("project_issue").
 | |
| 		Join("INNER", "issue", "project_issue.issue_id=issue.id").
 | |
| 		Where("project_issue.project_id=? AND issue.is_closed=?", p.ID, true).
 | |
| 		Cols("issue_id").
 | |
| 		Count()
 | |
| 	if err != nil {
 | |
| 		log.Error("NumClosedIssues: %v", err)
 | |
| 		return 0
 | |
| 	}
 | |
| 	return int(c)
 | |
| }
 | |
| 
 | |
| // NumOpenIssues return counter of open issues assigned to a project
 | |
| func (p *Project) NumOpenIssues() int {
 | |
| 	c, err := db.GetEngine(db.DefaultContext).Table("project_issue").
 | |
| 		Join("INNER", "issue", "project_issue.issue_id=issue.id").
 | |
| 		Where("project_issue.project_id=? AND issue.is_closed=?", p.ID, false).
 | |
| 		Cols("issue_id").
 | |
| 		Count()
 | |
| 	if err != nil {
 | |
| 		log.Error("NumOpenIssues: %v", err)
 | |
| 		return 0
 | |
| 	}
 | |
| 	return int(c)
 | |
| }
 | |
| 
 | |
| // MoveIssuesOnProjectBoard moves or keeps issues in a column and sorts them inside that column
 | |
| func MoveIssuesOnProjectBoard(board *Board, sortedIssueIDs map[int64]int64) error {
 | |
| 	return db.WithTx(db.DefaultContext, func(ctx context.Context) error {
 | |
| 		sess := db.GetEngine(ctx)
 | |
| 
 | |
| 		issueIDs := make([]int64, 0, len(sortedIssueIDs))
 | |
| 		for _, issueID := range sortedIssueIDs {
 | |
| 			issueIDs = append(issueIDs, issueID)
 | |
| 		}
 | |
| 		count, err := sess.Table(new(ProjectIssue)).Where("project_id=?", board.ProjectID).In("issue_id", issueIDs).Count()
 | |
| 		if err != nil {
 | |
| 			return err
 | |
| 		}
 | |
| 		if int(count) != len(sortedIssueIDs) {
 | |
| 			return fmt.Errorf("all issues have to be added to a project first")
 | |
| 		}
 | |
| 
 | |
| 		for sorting, issueID := range sortedIssueIDs {
 | |
| 			_, err = sess.Exec("UPDATE `project_issue` SET project_board_id=?, sorting=? WHERE issue_id=?", board.ID, sorting, issueID)
 | |
| 			if err != nil {
 | |
| 				return err
 | |
| 			}
 | |
| 		}
 | |
| 		return nil
 | |
| 	})
 | |
| }
 | |
| 
 | |
| func (b *Board) removeIssues(ctx context.Context) error {
 | |
| 	_, err := db.GetEngine(ctx).Exec("UPDATE `project_issue` SET project_board_id = 0 WHERE project_board_id = ? ", b.ID)
 | |
| 	return err
 | |
| }
 |