harness-drone/server/database/user.go

package database

import (
	"database/sql"
	"time"

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

type UserManager interface {
	// Find finds the User by ID.
	Find(id int64) (*model.User, error)

	// FindLogin finds the User by remote login.
	FindLogin(remote, login string) (*model.User, error)

	// FindToken finds the User by token.
	FindToken(token string) (*model.User, error)

	// List finds all registered users of the system.
	List() ([]*model.User, error)

	// Insert persists the User to the datastore.
	Insert(user *model.User) error

	// Update persists changes to the User to the datastore.
	Update(user *model.User) error

	// Delete removes the User from the datastore.
	Delete(user *model.User) error

	// Exist returns true if Users exist in the system.
	Exist() bool
}

// userManager manages a list of users in a SQL database.
type userManager struct {
	*sql.DB
}

// SQL query to retrieve a User by remote login.
const findUserLoginQuery = `
SELECT *
FROM users
WHERE user_remote=?
AND   user_login=?
LIMIT 1
`

// SQL query to retrieve a User by remote login.
const findUserTokenQuery = `
SELECT *
FROM users
WHERE user_token=?
LIMIT 1
`

// SQL query to retrieve a list of all users.
const listUserQuery = `
SELECT *
FROM users
ORDER BY user_name ASC
`

// SQL statement to delete a User by ID.
const deleteUserStmt = `
DELETE FROM users WHERE user_id=?
`

// SQL statement to check if users exist.
const confirmUserStmt = `
select 0 from users limit 1
`

// NewUserManager initiales a new UserManager intended to
// manage and persist commits.
func NewUserManager(db *sql.DB) UserManager {
	return &userManager{db}
}

func (db *userManager) Find(id int64) (*model.User, error) {
	dst := model.User{}
	err := meddler.Load(db, "users", &dst, id)
	return &dst, err
}

func (db *userManager) FindLogin(remote, login string) (*model.User, error) {
	dst := model.User{}
	err := meddler.QueryRow(db, &dst, findUserLoginQuery, remote, login)
	return &dst, err
}

func (db *userManager) FindToken(token string) (*model.User, error) {
	dst := model.User{}
	err := meddler.QueryRow(db, &dst, findUserTokenQuery, token)
	return &dst, err
}

func (db *userManager) List() ([]*model.User, error) {
	var dst []*model.User
	err := meddler.QueryAll(db, &dst, listUserQuery)
	return dst, err
}

func (db *userManager) Insert(user *model.User) error {
	user.Created = time.Now().Unix()
	user.Updated = time.Now().Unix()
	return meddler.Insert(db, "users", user)
}

func (db *userManager) Update(user *model.User) error {
	user.Updated = time.Now().Unix()
	return meddler.Update(db, "users", user)
}

func (db *userManager) Delete(user *model.User) error {
	_, err := db.Exec(deleteUserStmt, user.ID)
	return err
}

func (db *userManager) Exist() bool {
	row := db.QueryRow(confirmUserStmt)
	var result int
	row.Scan(&result)
	return result == 1
}