138 lines
3.6 KiB
Go
138 lines
3.6 KiB
Go
package builtin
|
|
|
|
import (
|
|
"database/sql"
|
|
"time"
|
|
|
|
common "github.com/drone/drone/pkg/types"
|
|
"github.com/russross/meddler"
|
|
)
|
|
|
|
type Repostore struct {
|
|
*sql.DB
|
|
}
|
|
|
|
func NewRepostore(db *sql.DB) *Repostore {
|
|
return &Repostore{db}
|
|
}
|
|
|
|
// Repo retrieves a specific repo from the
|
|
// datastore for the given ID.
|
|
func (db *Repostore) Repo(id int64) (*common.Repo, error) {
|
|
var repo = new(common.Repo)
|
|
var err = meddler.Load(db, repoTable, repo, id)
|
|
return repo, err
|
|
}
|
|
|
|
// RepoName retrieves a repo from the datastore
|
|
// for the specified name.
|
|
func (db *Repostore) RepoName(owner, name string) (*common.Repo, error) {
|
|
var repo = new(common.Repo)
|
|
var err = meddler.QueryRow(db, repo, rebind(repoNameQuery), owner, name)
|
|
return repo, err
|
|
}
|
|
|
|
// RepoList retrieves a list of all repos from
|
|
// the datastore accessible by the given user ID.
|
|
func (db *Repostore) RepoList(user *common.User) ([]*common.Repo, error) {
|
|
var repos []*common.Repo
|
|
var err = meddler.QueryAll(db, &repos, rebind(repoListQuery), user.ID)
|
|
return repos, err
|
|
}
|
|
|
|
// // RepoKeys retrieves a set of repository keys from
|
|
// // the datastore for the specified name.
|
|
// func (db *Repostore) RepoKeypair(repo *common.Repo) (*common.Keypair, error) {
|
|
// var keypair = new(common.Keypair)
|
|
// var err = meddler.QueryRow(db, keypair, rebind(repoKeysQuery), repo.ID)
|
|
// return keypair, err
|
|
// }
|
|
|
|
// // RepoParams retrieves a set of repository params from
|
|
// // the datastore for the specified name.
|
|
// func (db *Repostore) RepoParams(repo *common.Repo) (*common.Params, error) {
|
|
// var params = new(common.Params)
|
|
// var err = meddler.QueryRow(db, params, rebind(repoParamsQuery), repo.ID)
|
|
// return params, err
|
|
// }
|
|
|
|
// AddRepo inserts a repo in the datastore.
|
|
func (db *Repostore) AddRepo(repo *common.Repo) error {
|
|
repo.Created = time.Now().UTC().Unix()
|
|
repo.Updated = time.Now().UTC().Unix()
|
|
return meddler.Insert(db, repoTable, repo)
|
|
}
|
|
|
|
// SetRepo updates a repo in the datastore.
|
|
func (db *Repostore) SetRepo(repo *common.Repo) error {
|
|
repo.Updated = time.Now().UTC().Unix()
|
|
return meddler.Update(db, repoTable, repo)
|
|
}
|
|
|
|
// // SetRepoKeypair upserts a keypair in the datastore.
|
|
// func (db *Repostore) SetRepoKeypair(keys *common.Keypair) error {
|
|
// return meddler.Save(db, repoKeyTable, keys)
|
|
// }
|
|
|
|
// // SetRepoKeypair upserts a param set in the datastore.
|
|
// func (db *Repostore) SetRepoParams(params *common.Params) error {
|
|
// return meddler.Save(db, repoParamTable, params)
|
|
// }
|
|
|
|
// DelRepo removes the repo from the datastore.
|
|
func (db *Repostore) DelRepo(repo *common.Repo) error {
|
|
var _, err = db.Exec(rebind(repoDeleteStmt), repo.ID)
|
|
return err
|
|
}
|
|
|
|
// Repo table names in database.
|
|
const (
|
|
repoTable = "repos"
|
|
repoKeyTable = "repo_keys"
|
|
repoParamTable = "repo_params"
|
|
)
|
|
|
|
// SQL statement to retrieve a Repo by name.
|
|
const repoNameQuery = `
|
|
SELECT *
|
|
FROM repos
|
|
WHERE repo_owner = ?
|
|
AND repo_name = ?
|
|
LIMIT 1;
|
|
`
|
|
|
|
// SQL statement to retrieve a list of Repos
|
|
// with permissions for the given User ID.
|
|
const repoListQuery = `
|
|
SELECT r.*
|
|
FROM
|
|
repos r
|
|
,stars s
|
|
WHERE r.repo_id = s.repo_id
|
|
AND s.user_id = ?
|
|
`
|
|
|
|
// SQL statement to retrieve a keypair for
|
|
// a Repository.
|
|
const repoKeysQuery = `
|
|
SELECT *
|
|
FROM repo_keys
|
|
WHERE repo_id = ?
|
|
LIMIT 1;
|
|
`
|
|
|
|
// SQL statement to retrieve a keypair for
|
|
// a Repository.
|
|
const repoParamsQuery = `
|
|
SELECT *
|
|
FROM repo_params
|
|
WHERE repo_id = ?
|
|
LIMIT 1;
|
|
`
|
|
|
|
// SQL statement to delete a User by ID.
|
|
const (
|
|
repoDeleteStmt = `DELETE FROM repos WHERE repo_id = ?`
|
|
repoKeypairDeleteStmt = `DELETE FROM repo_params WHERE repo_id = ?`
|
|
repoParamsDeleteStmt = `DELETE FROM repo_keys WHERE repo_id = ?`
|
|
)
|