woodpecker/model/repo.go

package model

import (
	"github.com/drone/drone/shared/database"
	"github.com/russross/meddler"
	"strings"
)

type RepoLite struct {
	Owner    string `json:"owner"`
	Name     string `json:"name"`
	FullName string `json:"full_name"`
	Avatar   string `json:"avatar_url"`
}

type Repo struct {
	ID          int64  `json:"id"                meddler:"repo_id,pk"`
	UserID      int64  `json:"-"                 meddler:"repo_user_id"`
	Owner       string `json:"owner"             meddler:"repo_owner"`
	Name        string `json:"name"              meddler:"repo_name"`
	FullName    string `json:"full_name"         meddler:"repo_full_name"`
	Avatar      string `json:"avatar_url"        meddler:"repo_avatar"`
	Link        string `json:"link_url"          meddler:"repo_link"`
	Clone       string `json:"clone_url"         meddler:"repo_clone"`
	Branch      string `json:"default_branch"    meddler:"repo_branch"`
	Timeout     int64  `json:"timeout"           meddler:"repo_timeout"`
	IsPrivate   bool   `json:"private"           meddler:"repo_private"`
	IsTrusted   bool   `json:"trusted"           meddler:"repo_trusted"`
	IsStarred   bool   `json:"starred,omitempty" meddler:"-"`
	AllowPull   bool   `json:"allow_pr"          meddler:"repo_allow_pr"`
	AllowPush   bool   `json:"allow_push"        meddler:"repo_allow_push"`
	AllowDeploy bool   `json:"allow_deploys"     meddler:"repo_allow_deploys"`
	AllowTag    bool   `json:"allow_tags"        meddler:"repo_allow_tags"`
	Hash        string `json:"-"                 meddler:"repo_hash"`
}

func GetRepo(db meddler.DB, id int64) (*Repo, error) {
	var repo = new(Repo)
	var err = meddler.Load(db, repoTable, repo, id)
	return repo, err
}

func GetRepoName(db meddler.DB, owner, name string) (*Repo, error) {
	return GetRepoFullName(db, owner+"/"+name)
}

func GetRepoFullName(db meddler.DB, name string) (*Repo, error) {
	var repo = new(Repo)
	var err = meddler.QueryRow(db, repo, database.Rebind(repoNameQuery), name)
	return repo, err
}

func GetRepoList(db meddler.DB, user *User) ([]*Repo, error) {
	// we don't have real-time access to the intersection
	// of github repos and drone repos. So we cheat by simply
	// getting the distinct list of repos that the user
	// has created builds for.
	var repos = []*Repo{}
	var err = meddler.QueryAll(db, &repos, database.Rebind(repoListQuery), user.Login)
	return repos, err
}

func GetRepoListOf(db meddler.DB, listof []string) ([]*Repo, error) {
	var repos = []*Repo{}
	var qs = make([]string, len(listof), len(listof))
	var in = make([]interface{}, len(listof), len(listof))
	for i, repo := range listof {
		qs[i] = "?"
		in[i] = repo
	}
	var stmt = "SELECT * FROM repos WHERE repo_id IN (" + strings.Join(qs, ",") + ")"
	var err = meddler.QueryAll(db, &repos, database.Rebind(stmt), in...)
	return repos, err
}

func CreateRepo(db meddler.DB, repo *Repo) error {
	return meddler.Insert(db, repoTable, repo)
}

func UpdateRepo(db meddler.DB, repo *Repo) error {
	return meddler.Update(db, repoTable, repo)
}

func DeleteRepo(db meddler.DB, repo *Repo) error {
	var _, err = db.Exec(database.Rebind(repoDeleteStmt), repo.ID)
	return err
}

const repoTable = "repos"

const repoNameQuery = `
SELECT *
FROM repos
WHERE repo_full_name = ?
LIMIT 1;
`

const repoStarsQuery = `
SELECT r.*
FROM
 repos r
,stars s
WHERE r.repo_id = s.star_repo_id
  AND s.star_user_id = ?
`

const repoListQuery = `
SELECT *
FROM repos
WHERE repo_id IN (
	SELECT DISTINCT build_repo_id
	FROM builds
	WHERE build_author = ?
)
ORDER BY repo_full_name
`

const repoDeleteStmt = `
DELETE FROM repos
WHERE repo_id = ?
`