Status-API (#1332)

8 files changed, 529 insertions, 0 deletions

diff --git a/models/fixtures/commit_status.yml b/models/fixtures/commit_status.yml
new file mode 100644
index 0000000000..20d57975ef
--- /dev/null
+++ b/models/fixtures/commit_status.yml
@@ -0,0 +1,54 @@
+-
+  id: 1
+  index: 1
+  repo_id: 1
+  state: "pending"
+  sha: "1234123412341234123412341234123412341234"
+  target_url: https://example.com/builds/
+  description: My awesome CI-service
+  context: ci/awesomeness
+  creator_id: 2
+
+-
+  id: 2
+  index: 2
+  repo_id: 1
+  state: "warning"
+  sha: "1234123412341234123412341234123412341234"
+  target_url: https://example.com/converage/
+  description: My awesome Coverage service
+  context: cov/awesomeness
+  creator_id: 2
+
+-
+  id: 3
+  index: 3
+  repo_id: 1
+  state: "success"
+  sha: "1234123412341234123412341234123412341234"
+  target_url: https://example.com/converage/
+  description: My awesome Coverage service
+  context: cov/awesomeness
+  creator_id: 2
+
+-
+  id: 4
+  index: 4
+  repo_id: 1
+  state: "failure"
+  sha: "1234123412341234123412341234123412341234"
+  target_url: https://example.com/builds/
+  description: My awesome CI-service
+  context: ci/awesomeness
+  creator_id: 2
+
+-
+  id: 5
+  index: 5
+  repo_id: 1
+  state: "error"
+  sha: "1234123412341234123412341234123412341234"
+  target_url: https://example.com/builds/
+  description: My awesome deploy service
+  context: deploy/awesomeness
+  creator_id: 2
diff --git a/models/migrations/migrations.go b/models/migrations/migrations.go
index fb7b66cd00..92dd4c8ccd 100644
--- a/models/migrations/migrations.go
+++ b/models/migrations/migrations.go
@@ -106,6 +106,8 @@ var migrations = []Migration{
 	NewMigration("change mirror interval from hours to time.Duration", convertIntervalToDuration),
 	// v28 -> v29
 	NewMigration("add field for repo size", addRepoSize),
+	// v29 -> v30
+	NewMigration("add commit status table", addCommitStatus),
 }
 
 // Migrate database to current version
diff --git a/models/migrations/v29.go b/models/migrations/v29.go
new file mode 100644
index 0000000000..eadb0f3d87
--- /dev/null
+++ b/models/migrations/v29.go
@@ -0,0 +1,34 @@
+// Copyright 2017 The Gogs 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 migrations
+
+import (
+	"fmt"
+
+	"github.com/go-xorm/xorm"
+)
+
+// CommitStatus see models/status.go
+type CommitStatus struct {
+	ID          int64  `xorm:"pk autoincr"`
+	Index       int64  `xorm:"INDEX UNIQUE(repo_sha_index)"`
+	RepoID      int64  `xorm:"INDEX UNIQUE(repo_sha_index)"`
+	State       string `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"`
+	CreatorID   int64  `xorm:"INDEX"`
+
+	CreatedUnix int64 `xorm:"INDEX"`
+	UpdatedUnix int64 `xorm:"INDEX"`
+}
+
+func addCommitStatus(x *xorm.Engine) error {
+	if err := x.Sync2(new(CommitStatus)); err != nil {
+		return fmt.Errorf("Sync2: %v", err)
+	}
+	return nil
+}
diff --git a/models/models.go b/models/models.go
index a1332ac23e..02499b6eb6 100644
--- a/models/models.go
+++ b/models/models.go
@@ -118,6 +118,7 @@ func init() {
 		new(ProtectedBranch),
 		new(UserOpenID),
 		new(IssueWatch),
+		new(CommitStatus),
 	)
 
 	gonicNames := []string{"SSL", "UID"}
diff --git a/models/status.go b/models/status.go
new file mode 100644
index 0000000000..c06dad822e
--- /dev/null
+++ b/models/status.go
@@ -0,0 +1,265 @@
+// 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)
+	sess := x.NewSession()
+	defer sess.Close()
+	return statuses, sess.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)
+	sess := x.NewSession()
+	defer sess.Close()
+
+	return statuses, sess.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(e Engine, 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)
+	}
+
+	sess := x.NewSession()
+	defer sess.Close()
+	if err = sess.Begin(); err != nil {
+		return fmt.Errorf("newCommitStatus[%s, %s]: %v", opts.Repo.RepoPath(), 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 sess.Commit()
+}
+
+// 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 nil
+}
diff --git a/models/status_test.go b/models/status_test.go
new file mode 100644
index 0000000000..5585fb1dcc
--- /dev/null
+++ b/models/status_test.go
@@ -0,0 +1,39 @@
+// 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 (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestGetCommitStatuses(t *testing.T) {
+	assert.NoError(t, PrepareTestDatabase())
+
+	repo1 := AssertExistsAndLoadBean(t, &Repository{ID: 1}).(*Repository)
+
+	sha1 := "1234123412341234123412341234123412341234"
+
+	statuses, err := GetCommitStatuses(repo1, sha1, 0)
+	assert.NoError(t, err)
+	if assert.Equal(t, 5, len(statuses), "Expected to get 5 statuses") {
+
+		assert.Equal(t, statuses[0].Context, "ci/awesomeness")
+		assert.Equal(t, statuses[0].State, CommitStatusPending)
+
+		assert.Equal(t, statuses[1].Context, "cov/awesomeness")
+		assert.Equal(t, statuses[1].State, CommitStatusWarning)
+
+		assert.Equal(t, statuses[2].Context, "cov/awesomeness")
+		assert.Equal(t, statuses[2].State, CommitStatusSuccess)
+
+		assert.Equal(t, statuses[3].Context, "ci/awesomeness")
+		assert.Equal(t, statuses[3].State, CommitStatusFailure)
+
+		assert.Equal(t, statuses[4].Context, "deploy/awesomeness")
+		assert.Equal(t, statuses[4].State, CommitStatusError)
+	}
+}
diff --git a/routers/api/v1/api.go b/routers/api/v1/api.go
index 74aa995c97..e33dc12d28 100644
--- a/routers/api/v1/api.go
+++ b/routers/api/v1/api.go
@@ -412,6 +412,13 @@ func RegisterRoutes(m *macaron.Macaron) {
 					})
 
 				}, mustAllowPulls, context.ReferencesGitRepo())
+				m.Group("/statuses", func() {
+					m.Combo("/:sha").Get(repo.GetCommitStatuses).Post(reqRepoWriter(), bind(api.CreateStatusOption{}), repo.NewCommitStatus)
+				})
+				m.Group("/commits/:ref", func() {
+					m.Get("/status", repo.GetCombinedCommitStatus)
+					m.Get("/statuses", repo.GetCommitStatuses)
+				})
 			}, repoAssignment())
 		}, reqToken())
 
diff --git a/routers/api/v1/repo/status.go b/routers/api/v1/repo/status.go
new file mode 100644
index 0000000000..d9b101df05
--- /dev/null
+++ b/routers/api/v1/repo/status.go
@@ -0,0 +1,127 @@
+// 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 repo
+
+import (
+	"fmt"
+
+	"code.gitea.io/gitea/models"
+	"code.gitea.io/gitea/modules/context"
+	api "code.gitea.io/sdk/gitea"
+)
+
+// NewCommitStatus creates a new CommitStatus
+func NewCommitStatus(ctx *context.APIContext, form api.CreateStatusOption) {
+	sha := ctx.Params("sha")
+	if len(sha) == 0 {
+		sha = ctx.Params("ref")
+	}
+	if len(sha) == 0 {
+		ctx.Error(400, "ref/sha not given", nil)
+		return
+	}
+	status := &models.CommitStatus{
+		State:       models.CommitStatusState(form.State),
+		TargetURL:   form.TargetURL,
+		Description: form.Description,
+		Context:     form.Context,
+	}
+	if err := models.NewCommitStatus(ctx.Repo.Repository, ctx.User, sha, status); err != nil {
+		ctx.Error(500, "NewCommitStatus", err)
+		return
+	}
+
+	newStatus, err := models.GetCommitStatus(ctx.Repo.Repository, sha, status)
+	if err != nil {
+		ctx.Error(500, "GetCommitStatus", err)
+		return
+	}
+	ctx.JSON(201, newStatus.APIFormat())
+}
+
+// GetCommitStatuses returns all statuses for any given commit hash
+func GetCommitStatuses(ctx *context.APIContext) {
+	sha := ctx.Params("sha")
+	if len(sha) == 0 {
+		sha = ctx.Params("ref")
+	}
+	if len(sha) == 0 {
+		ctx.Error(400, "ref/sha not given", nil)
+		return
+	}
+	repo := ctx.Repo.Repository
+
+	page := ctx.ParamsInt("page")
+
+	statuses, err := models.GetCommitStatuses(repo, sha, page)
+	if err != nil {
+		ctx.Error(500, "GetCommitStatuses", fmt.Errorf("GetCommitStatuses[%s, %s, %d]: %v", repo.FullName(), sha, page, err))
+	}
+
+	apiStatuses := make([]*api.Status, 0, len(statuses))
+	for _, status := range statuses {
+		apiStatuses = append(apiStatuses, status.APIFormat())
+	}
+
+	ctx.JSON(200, apiStatuses)
+}
+
+type combinedCommitStatus struct {
+	State      models.CommitStatusState `json:"state"`
+	SHA        string                   `json:"sha"`
+	TotalCount int                      `json:"total_count"`
+	Statuses   []*api.Status            `json:"statuses"`
+	Repo       *api.Repository          `json:"repository"`
+	CommitURL  string                   `json:"commit_url"`
+	URL        string                   `json:"url"`
+}
+
+// GetCombinedCommitStatus returns the combined status for any given commit hash
+func GetCombinedCommitStatus(ctx *context.APIContext) {
+	sha := ctx.Params("sha")
+	if len(sha) == 0 {
+		sha = ctx.Params("ref")
+	}
+	if len(sha) == 0 {
+		ctx.Error(400, "ref/sha not given", nil)
+		return
+	}
+	repo := ctx.Repo.Repository
+
+	page := ctx.ParamsInt("page")
+
+	statuses, err := models.GetLatestCommitStatus(repo, sha, page)
+	if err != nil {
+		ctx.Error(500, "GetLatestCommitStatus", fmt.Errorf("GetLatestCommitStatus[%s, %s, %d]: %v", repo.FullName(), sha, page, err))
+		return
+	}
+
+	if len(statuses) == 0 {
+		ctx.Status(200)
+		return
+	}
+
+	acl, err := models.AccessLevel(ctx.User.ID, repo)
+	if err != nil {
+		ctx.Error(500, "AccessLevel", fmt.Errorf("AccessLevel[%d, %s]: %v", ctx.User.ID, repo.FullName(), err))
+		return
+	}
+	retStatus := &combinedCommitStatus{
+		SHA:        sha,
+		TotalCount: len(statuses),
+		Repo:       repo.APIFormat(acl),
+		URL:        "",
+	}
+
+	retStatus.Statuses = make([]*api.Status, 0, len(statuses))
+	for _, status := range statuses {
+		retStatus.Statuses = append(retStatus.Statuses, status.APIFormat())
+		if status.State.IsWorseThan(retStatus.State) {
+			retStatus.State = status.State
+		}
+	}
+
+	ctx.JSON(200, retStatus)
+}