package database import ( "time" . "github.com/drone/drone/pkg/model" "github.com/russross/meddler" ) // Name of the Repos table in the database const repoTable = "repos" // SQL Queries to retrieve a list of all repos belonging to a User. const repoStmt = ` SELECT id, slug, host, owner, name, private, disabled, disabled_pr, scm, url, username, password, public_key, private_key, params, timeout, priveleged, created, updated, user_id, team_id FROM repos WHERE user_id = ? AND team_id = 0 ORDER BY slug ASC ` // SQL Queries to retrieve a list of all repos belonging to a Team. const repoTeamStmt = ` SELECT id, slug, host, owner, name, private, disabled, disabled_pr, scm, url, username, password, public_key, private_key, params, timeout, priveleged, created, updated, user_id, team_id FROM repos WHERE team_id = ? ORDER BY slug ASC ` // SQL Queries to retrieve a repo by id. const repoFindStmt = ` SELECT id, slug, host, owner, name, private, disabled, disabled_pr, scm, url, username, password, public_key, private_key, params, timeout, priveleged, created, updated, user_id, team_id FROM repos WHERE id = ? ` // SQL Queries to retrieve a repo by name. const repoFindSlugStmt = ` SELECT id, slug, host, owner, name, private, disabled, disabled_pr, scm, url, username, password, public_key, private_key, params, timeout, priveleged, created, updated, user_id, team_id FROM repos WHERE slug = ? ` // Returns the Repo with the given ID. func GetRepo(id int64) (*Repo, error) { repo := Repo{} err := meddler.QueryRow(db, &repo, repoFindStmt, id) return &repo, err } // Returns the Repo with the given slug. func GetRepoSlug(slug string) (*Repo, error) { repo := Repo{} err := meddler.QueryRow(db, &repo, repoFindSlugStmt, slug) return &repo, err } // Creates a new Repository. func SaveRepo(repo *Repo) error { if repo.ID == 0 { repo.Created = time.Now().UTC() } repo.Updated = time.Now().UTC() return meddler.Save(db, repoTable, repo) } // Deletes an existing Repository. // TODO need to delete builds too. func DeleteRepo(id int64) error { _, err := db.Exec("DELETE FROM repos WHERE id = ?", id) db.Exec("DELETE FROM commits WHERE repo_id = ?", id) return err } // Returns a list of all Repos associated // with the specified User ID. func ListRepos(id int64) ([]*Repo, error) { var repos []*Repo err := meddler.QueryAll(db, &repos, repoStmt, id) return repos, err } // Returns a list of all Repos associated // with the specified Team ID. func ListReposTeam(id int64) ([]*Repo, error) { var repos []*Repo err := meddler.QueryAll(db, &repos, repoTeamStmt, id) return repos, err }