woodpecker/pkg/store/builtin/star.go
2015-06-13 17:49:21 -07:00

47 lines
1.1 KiB
Go

package builtin
import (
"database/sql"
"github.com/drone/drone/pkg/types"
)
type Starstore struct {
*sql.DB
}
func NewStarstore(db *sql.DB) *Starstore {
return &Starstore{db}
}
// Starred returns true if the user starred
// the given repository.
func (db *Starstore) Starred(user *types.User, repo *types.Repo) (bool, error) {
_, err := getStar(db, rebind(stmtStarSelectStarUserRepo), user.ID, repo.ID)
return (err == nil), err
}
// AddStar inserts a starred repo / user in the datastore.
func (db *Starstore) AddStar(user *types.User, repo *types.Repo) error {
var star = &Star{UserID: user.ID, RepoID: repo.ID}
return createStar(db, rebind(stmtStarInsert), star)
}
// DelStar removes starred repo / user from the datastore.
func (db *Starstore) DelStar(user *types.User, repo *types.Repo) error {
var _, err = db.Exec(rebind(stmtStartDeleteUserRepo), user.ID, repo.ID)
return err
}
type Star struct {
ID int64
UserID int64 `sql:"unique:ux_star_user_repo"`
RepoID int64 `sql:"unique:ux_star_user_repo"`
}
// SQL statement to delete a star by ID.
const stmtStartDeleteUserRepo = `
DELETE FROM stars
WHERE star_user_id=?
AND star_repo_id=?
`