mirror of
				https://codeberg.org/forgejo/forgejo.git
				synced 2025-10-30 22:11:07 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			254 lines
		
	
	
	
		
			8.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			254 lines
		
	
	
	
		
			8.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| // Copyright 2017 Gitea. 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 (
 | |
| 	"fmt"
 | |
| 	"strings"
 | |
| 	"time"
 | |
| 
 | |
| 	"code.gitea.io/git"
 | |
| 	"code.gitea.io/gitea/modules/log"
 | |
| 	"code.gitea.io/gitea/modules/setting"
 | |
| 	api "code.gitea.io/sdk/gitea"
 | |
| 
 | |
| 	"github.com/go-xorm/xorm"
 | |
| )
 | |
| 
 | |
| // CommitStatusState holds the state of a Status
 | |
| // It can be "pending", "success", "error", "failure", and "warning"
 | |
| type CommitStatusState string
 | |
| 
 | |
| // IsWorseThan returns true if this State is worse than the given State
 | |
| func (css CommitStatusState) IsWorseThan(css2 CommitStatusState) bool {
 | |
| 	switch css {
 | |
| 	case CommitStatusError:
 | |
| 		return true
 | |
| 	case CommitStatusFailure:
 | |
| 		return css2 != CommitStatusError
 | |
| 	case CommitStatusWarning:
 | |
| 		return css2 != CommitStatusError && css2 != CommitStatusFailure
 | |
| 	case CommitStatusSuccess:
 | |
| 		return css2 != CommitStatusError && css2 != CommitStatusFailure && css2 != CommitStatusWarning
 | |
| 	default:
 | |
| 		return css2 != CommitStatusError && css2 != CommitStatusFailure && css2 != CommitStatusWarning && css2 != CommitStatusSuccess
 | |
| 	}
 | |
| }
 | |
| 
 | |
| const (
 | |
| 	// CommitStatusPending is for when the Status is Pending
 | |
| 	CommitStatusPending CommitStatusState = "pending"
 | |
| 	// CommitStatusSuccess is for when the Status is Success
 | |
| 	CommitStatusSuccess CommitStatusState = "success"
 | |
| 	// CommitStatusError is for when the Status is Error
 | |
| 	CommitStatusError CommitStatusState = "error"
 | |
| 	// CommitStatusFailure is for when the Status is Failure
 | |
| 	CommitStatusFailure CommitStatusState = "failure"
 | |
| 	// CommitStatusWarning is for when the Status is Warning
 | |
| 	CommitStatusWarning CommitStatusState = "warning"
 | |
| )
 | |
| 
 | |
| // CommitStatus holds a single Status of a single Commit
 | |
| type CommitStatus struct {
 | |
| 	ID          int64             `xorm:"pk autoincr"`
 | |
| 	Index       int64             `xorm:"INDEX UNIQUE(repo_sha_index)"`
 | |
| 	RepoID      int64             `xorm:"INDEX UNIQUE(repo_sha_index)"`
 | |
| 	Repo        *Repository       `xorm:"-"`
 | |
| 	State       CommitStatusState `xorm:"VARCHAR(7) NOT NULL"`
 | |
| 	SHA         string            `xorm:"VARCHAR(64) NOT NULL INDEX UNIQUE(repo_sha_index)"`
 | |
| 	TargetURL   string            `xorm:"TEXT"`
 | |
| 	Description string            `xorm:"TEXT"`
 | |
| 	Context     string            `xorm:"TEXT"`
 | |
| 	Creator     *User             `xorm:"-"`
 | |
| 	CreatorID   int64
 | |
| 
 | |
| 	Created     time.Time `xorm:"-"`
 | |
| 	CreatedUnix int64     `xorm:"INDEX"`
 | |
| 	Updated     time.Time `xorm:"-"`
 | |
| 	UpdatedUnix int64     `xorm:"INDEX"`
 | |
| }
 | |
| 
 | |
| // BeforeInsert is invoked from XORM before inserting an object of this type.
 | |
| func (status *CommitStatus) BeforeInsert() {
 | |
| 	status.CreatedUnix = time.Now().Unix()
 | |
| 	status.UpdatedUnix = status.CreatedUnix
 | |
| }
 | |
| 
 | |
| // BeforeUpdate is invoked from XORM before updating this object.
 | |
| func (status *CommitStatus) BeforeUpdate() {
 | |
| 	status.UpdatedUnix = time.Now().Unix()
 | |
| }
 | |
| 
 | |
| // AfterSet is invoked from XORM after setting the value of a field of
 | |
| // this object.
 | |
| func (status *CommitStatus) AfterSet(colName string, _ xorm.Cell) {
 | |
| 	switch colName {
 | |
| 	case "created_unix":
 | |
| 		status.Created = time.Unix(status.CreatedUnix, 0).Local()
 | |
| 	case "updated_unix":
 | |
| 		status.Updated = time.Unix(status.UpdatedUnix, 0).Local()
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (status *CommitStatus) loadRepo(e Engine) (err error) {
 | |
| 	if status.Repo == nil {
 | |
| 		status.Repo, err = getRepositoryByID(e, status.RepoID)
 | |
| 		if err != nil {
 | |
| 			return fmt.Errorf("getRepositoryByID [%d]: %v", status.RepoID, err)
 | |
| 		}
 | |
| 	}
 | |
| 	if status.Creator == nil && status.CreatorID > 0 {
 | |
| 		status.Creator, err = getUserByID(e, status.CreatorID)
 | |
| 		if err != nil {
 | |
| 			return fmt.Errorf("getUserByID [%d]: %v", status.CreatorID, err)
 | |
| 		}
 | |
| 	}
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| // APIURL returns the absolute APIURL to this commit-status.
 | |
| func (status *CommitStatus) APIURL() string {
 | |
| 	status.loadRepo(x)
 | |
| 	return fmt.Sprintf("%sapi/v1/%s/statuses/%s",
 | |
| 		setting.AppURL, status.Repo.FullName(), status.SHA)
 | |
| }
 | |
| 
 | |
| // APIFormat assumes some fields assigned with values:
 | |
| // Required - Repo, Creator
 | |
| func (status *CommitStatus) APIFormat() *api.Status {
 | |
| 	status.loadRepo(x)
 | |
| 	apiStatus := &api.Status{
 | |
| 		Created:     status.Created,
 | |
| 		Updated:     status.Created,
 | |
| 		State:       api.StatusState(status.State),
 | |
| 		TargetURL:   status.TargetURL,
 | |
| 		Description: status.Description,
 | |
| 		ID:          status.Index,
 | |
| 		URL:         status.APIURL(),
 | |
| 		Context:     status.Context,
 | |
| 	}
 | |
| 	if status.Creator != nil {
 | |
| 		apiStatus.Creator = status.Creator.APIFormat()
 | |
| 	}
 | |
| 
 | |
| 	return apiStatus
 | |
| }
 | |
| 
 | |
| // GetCommitStatuses returns all statuses for a given commit.
 | |
| func GetCommitStatuses(repo *Repository, sha string, page int) ([]*CommitStatus, error) {
 | |
| 	statuses := make([]*CommitStatus, 0, 10)
 | |
| 	return statuses, x.Limit(10, page*10).Where("repo_id = ?", repo.ID).And("sha = ?", sha).Find(&statuses)
 | |
| }
 | |
| 
 | |
| // GetLatestCommitStatus returns all statuses with a unique context for a given commit.
 | |
| func GetLatestCommitStatus(repo *Repository, sha string, page int) ([]*CommitStatus, error) {
 | |
| 	statuses := make([]*CommitStatus, 0, 10)
 | |
| 	return statuses, x.Limit(10, page*10).
 | |
| 		Where("repo_id = ?", repo.ID).And("sha = ?", sha).Select("*").
 | |
| 		GroupBy("context").Desc("created_unix").Find(&statuses)
 | |
| }
 | |
| 
 | |
| // GetCommitStatus populates a given status for a given commit.
 | |
| // NOTE: If ID or Index isn't given, and only Context, TargetURL and/or Description
 | |
| //       is given, the CommitStatus created _last_ will be returned.
 | |
| func GetCommitStatus(repo *Repository, sha string, status *CommitStatus) (*CommitStatus, error) {
 | |
| 	conds := &CommitStatus{
 | |
| 		Context:     status.Context,
 | |
| 		State:       status.State,
 | |
| 		TargetURL:   status.TargetURL,
 | |
| 		Description: status.Description,
 | |
| 	}
 | |
| 	has, err := x.Where("repo_id = ?", repo.ID).And("sha = ?", sha).Desc("created_unix").Get(conds)
 | |
| 	if err != nil {
 | |
| 		return nil, fmt.Errorf("GetCommitStatus[%s, %s]: %v", repo.RepoPath(), sha, err)
 | |
| 	}
 | |
| 	if !has {
 | |
| 		return nil, fmt.Errorf("GetCommitStatus[%s, %s]: not found", repo.RepoPath(), sha)
 | |
| 	}
 | |
| 
 | |
| 	return conds, nil
 | |
| }
 | |
| 
 | |
| // NewCommitStatusOptions holds options for creating a CommitStatus
 | |
| type NewCommitStatusOptions struct {
 | |
| 	Repo         *Repository
 | |
| 	Creator      *User
 | |
| 	SHA          string
 | |
| 	CommitStatus *CommitStatus
 | |
| }
 | |
| 
 | |
| func newCommitStatus(sess *xorm.Session, opts NewCommitStatusOptions) error {
 | |
| 	opts.CommitStatus.Description = strings.TrimSpace(opts.CommitStatus.Description)
 | |
| 	opts.CommitStatus.Context = strings.TrimSpace(opts.CommitStatus.Context)
 | |
| 	opts.CommitStatus.TargetURL = strings.TrimSpace(opts.CommitStatus.TargetURL)
 | |
| 	opts.CommitStatus.SHA = opts.SHA
 | |
| 	opts.CommitStatus.CreatorID = opts.Creator.ID
 | |
| 
 | |
| 	if opts.Repo == nil {
 | |
| 		return fmt.Errorf("newCommitStatus[nil, %s]: no repository specified", opts.SHA)
 | |
| 	}
 | |
| 	opts.CommitStatus.RepoID = opts.Repo.ID
 | |
| 
 | |
| 	if opts.Creator == nil {
 | |
| 		return fmt.Errorf("newCommitStatus[%s, %s]: no user specified", opts.Repo.RepoPath(), opts.SHA)
 | |
| 	}
 | |
| 
 | |
| 	gitRepo, err := git.OpenRepository(opts.Repo.RepoPath())
 | |
| 	if err != nil {
 | |
| 		return fmt.Errorf("OpenRepository[%s]: %v", opts.Repo.RepoPath(), err)
 | |
| 	}
 | |
| 	if _, err := gitRepo.GetCommit(opts.SHA); err != nil {
 | |
| 		return fmt.Errorf("GetCommit[%s]: %v", opts.SHA, err)
 | |
| 	}
 | |
| 
 | |
| 	// Get the next Status Index
 | |
| 	var nextIndex int64
 | |
| 	lastCommitStatus := &CommitStatus{
 | |
| 		SHA:    opts.SHA,
 | |
| 		RepoID: opts.Repo.ID,
 | |
| 	}
 | |
| 	has, err := sess.Desc("index").Limit(1).Get(lastCommitStatus)
 | |
| 	if err != nil {
 | |
| 		sess.Rollback()
 | |
| 		return fmt.Errorf("newCommitStatus[%s, %s]: %v", opts.Repo.RepoPath(), opts.SHA, err)
 | |
| 	}
 | |
| 	if has {
 | |
| 		log.Debug("newCommitStatus[%s, %s]: found", opts.Repo.RepoPath(), opts.SHA)
 | |
| 		nextIndex = lastCommitStatus.Index
 | |
| 	}
 | |
| 	opts.CommitStatus.Index = nextIndex + 1
 | |
| 	log.Debug("newCommitStatus[%s, %s]: %d", opts.Repo.RepoPath(), opts.SHA, opts.CommitStatus.Index)
 | |
| 
 | |
| 	// Insert new CommitStatus
 | |
| 	if _, err = sess.Insert(opts.CommitStatus); err != nil {
 | |
| 		sess.Rollback()
 | |
| 		return fmt.Errorf("newCommitStatus[%s, %s]: %v", opts.Repo.RepoPath(), opts.SHA, err)
 | |
| 	}
 | |
| 
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| // NewCommitStatus creates a new CommitStatus given a bunch of parameters
 | |
| // NOTE: All text-values will be trimmed from whitespaces.
 | |
| // Requires: Repo, Creator, SHA
 | |
| func NewCommitStatus(repo *Repository, creator *User, sha string, status *CommitStatus) error {
 | |
| 	sess := x.NewSession()
 | |
| 	defer sess.Close()
 | |
| 
 | |
| 	if err := sess.Begin(); err != nil {
 | |
| 		return fmt.Errorf("NewCommitStatus[repo_id: %d, user_id: %d, sha: %s]: %v", repo.ID, creator.ID, sha, err)
 | |
| 	}
 | |
| 
 | |
| 	if err := newCommitStatus(sess, NewCommitStatusOptions{
 | |
| 		Repo:         repo,
 | |
| 		Creator:      creator,
 | |
| 		SHA:          sha,
 | |
| 		CommitStatus: status,
 | |
| 	}); err != nil {
 | |
| 		return fmt.Errorf("NewCommitStatus[repo_id: %d, user_id: %d, sha: %s]: %v", repo.ID, creator.ID, sha, err)
 | |
| 	}
 | |
| 
 | |
| 	return sess.Commit()
 | |
| }
 |