harness-drone/server/database/repo.go

package database

import (
	"database/sql"
	"time"

	"github.com/drone/drone/shared/model"
	"github.com/russross/meddler"
)

type RepoManager interface {
	// Find retrieves the Repo by ID.
	Find(id int64) (*model.Repo, error)

	// FindName retrieves the Repo by the remote, owner and name.
	FindName(remote, owner, name string) (*model.Repo, error)

	// Insert persists a new Repo to the datastore.
	Insert(repo *model.Repo) error

	// Insert persists a modified Repo to the datastore.
	Update(repo *model.Repo) error

	// Delete removes a Repo from the datastore.
	Delete(repo *model.Repo) error

	// List retrieves all repositories from the datastore.
	List(user int64) ([]*model.Repo, error)

	// List retrieves all public repositories from the datastore.
	//ListPublic(user int64) ([]*Repo, error)
}

func NewRepoManager(db *sql.DB) RepoManager {
	return &repoManager{db}
}

type repoManager struct {
	*sql.DB
}

func (db *repoManager) Find(id int64) (*model.Repo, error) {
	const query = "select * from repos where repo_id = ?"
	var repo = model.Repo{}
	var err = meddler.QueryRow(db, &repo, query, id)
	return &repo, err
}

func (db *repoManager) FindName(remote, owner, name string) (*model.Repo, error) {
	const query = "select * from repos where repo_host = ? and repo_owner = ? and repo_name = ?"
	var repo = model.Repo{}
	var err = meddler.QueryRow(db, &repo, query, remote, owner, name)
	return &repo, err
}

func (db *repoManager) List(user int64) ([]*model.Repo, error) {
	const query = "select * from repos where repo_id IN (select repo_id from perms where user_id = ?)"
	var repos []*model.Repo
	err := meddler.QueryAll(db, &repos, query, user)
	return repos, err
}

func (db *repoManager) Insert(repo *model.Repo) error {
	repo.Created = time.Now().Unix()
	repo.Updated = time.Now().Unix()
	return meddler.Insert(db, "repos", repo)
}

func (db *repoManager) Update(repo *model.Repo) error {
	repo.Updated = time.Now().Unix()
	return meddler.Update(db, "repos", repo)
}

func (db *repoManager) Delete(repo *model.Repo) error {
	const stmt = "delete from repos where repo_id = ?"
	_, err := db.Exec(stmt, repo.ID)
	return err
}