go-gitea/models/issue_dependency.go
Konrad Langenberg fe8181af73 Simplified checking for open dependencies
Signed-off-by: Konrad <konrad@kola-entertainments.de>
2017-09-24 16:29:25 +02:00

155 lines
3.7 KiB
Go

// Copyright 2017 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 models
import (
"time"
)
// IssueDependency is connection request for receiving issue notification.
type IssueDependency struct {
ID int64 `xorm:"pk autoincr"`
UserID int64 `xorm:"UNIQUE(watch) NOT NULL"`
IssueID int64 `xorm:"UNIQUE(watch) NOT NULL"`
DependencyID int64 `xorm:"UNIQUE(watch) NOT NULL"`
Created time.Time `xorm:"-"`
CreatedUnix int64 `xorm:"INDEX created"`
Updated time.Time `xorm:"-"`
UpdatedUnix int64 `xorm:"updated"`
}
// Define Dependency Type Constants
const (
DependencyTypeBlockedBy int64 = 1
DependencyTypeBlocking int64 = 2
)
// CreateIssueDependency creates a new dependency for an issue
func CreateIssueDependency(user *User, issue, dep *Issue) (exists bool, err error) {
sess := x.NewSession()
// Check if it aleready exists
exists, err = issueDepExists(x, issue.ID, dep.ID)
if err != nil {
return exists, err
}
// If it not exists, create it, otherwise show an error message
if !exists {
newIssueDependency := &IssueDependency{
UserID: user.ID,
IssueID: issue.ID,
DependencyID: dep.ID,
}
if _, err := x.Insert(newIssueDependency); err != nil {
return exists, err
}
// Add comment referencing the new dependency
_, err = createIssueDependencyComment(sess, user, issue, dep, true)
if err != nil {
return exists, err
}
// Create a new comment for the dependent issue
_, err = createIssueDependencyComment(sess, user, dep, issue, true)
if err != nil {
return exists, err
}
}
return exists, nil
}
// RemoveIssueDependency removes a dependency from an issue
func RemoveIssueDependency(user *User, issue *Issue, dep *Issue, depType int64) (err error) {
sess := x.NewSession()
// Check if it exists
exists, err := issueDepExists(x, issue.ID, dep.ID)
if err != nil {
return err
}
// If it exists, remove it, otherwise show an error message
if exists {
if depType == DependencyTypeBlockedBy {
_, err := x.Delete(&IssueDependency{IssueID: issue.ID, DependencyID: dep.ID})
if err != nil {
return err
}
}
if depType == DependencyTypeBlocking {
_, err := x.Delete(&IssueDependency{IssueID: dep.ID, DependencyID: issue.ID})
if err != nil {
return err
}
}
// Add comment referencing the removed dependency
_, err = createIssueDependencyComment(sess, user, issue, dep, false)
if err != nil {
return err
}
// Create a new comment for the dependent issue
_, err = createIssueDependencyComment(sess, user, dep, issue, false)
if err != nil {
return err
}
}
return nil
}
// Check if the dependency already exists
func issueDepExists(e Engine, issueID int64, depID int64) (exists bool, err error) {
deps := new(IssueDependency)
exists, err = e.Where("(issue_id = ? AND dependency_id = ?) OR (issue_id = ? AND dependency_id = ?)", issueID, depID, depID, issueID).Get(deps)
if err != nil {
return exists, err
}
return exists, nil
}
// IssueDependencyIssue custom type for mysql join
type IssueDependencyIssue struct {
IssueDependency `xorm:"extends"`
Issue `xorm:"extends"`
}
// TableName returns table name for mysql join
func (IssueDependencyIssue) TableName() string {
return "issue_dependency"
}
// IssueNoDependenciesLeft checks if issue can be closed
func IssueNoDependenciesLeft(issue *Issue) bool {
var issueDeps []IssueDependencyIssue
err := x.Join("INNER", "issue", "issue.id = issue_dependency.issue_id").Find(&issueDeps)
if err != nil {
return false
}
for _, issueDep := range issueDeps {
if !issueDep.IsClosed {
return false
}
}
return true
}