111 lines
2.5 KiB
Go
111 lines
2.5 KiB
Go
package builtin
|
|
|
|
import (
|
|
"database/sql"
|
|
|
|
"github.com/drone/drone/pkg/types"
|
|
)
|
|
|
|
type Userstore struct {
|
|
*sql.DB
|
|
}
|
|
|
|
func NewUserstore(db *sql.DB) *Userstore {
|
|
return &Userstore{db}
|
|
}
|
|
|
|
// User returns a user by user ID.
|
|
func (db *Userstore) User(id int64) (*types.User, error) {
|
|
return getUser(db, rebind(stmtUserSelect), id)
|
|
}
|
|
|
|
// UserLogin returns a user by user login.
|
|
func (db *Userstore) UserLogin(login string) (*types.User, error) {
|
|
return getUser(db, rebind(stmtUserSelectUserLogin), login)
|
|
}
|
|
|
|
// UserList returns a list of all registered users.
|
|
func (db *Userstore) UserList() ([]*types.User, error) {
|
|
return getUsers(db, rebind(stmtUserSelectList))
|
|
}
|
|
|
|
// UserFeed retrieves a digest of recent builds
|
|
// from the datastore accessible to the specified user.
|
|
func (db *Userstore) UserFeed(user *types.User, limit, offset int) ([]*types.RepoCommit, error) {
|
|
rows, err := db.Query(rebind(userFeedQuery), user.ID, limit, offset)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
return scanRepoCommits(rows)
|
|
}
|
|
|
|
// UserCount returns a count of all registered users.
|
|
func (db *Userstore) UserCount() (int, error) {
|
|
var count int
|
|
err := db.QueryRow(stmtUserSelectCount).Scan(&count)
|
|
return count, err
|
|
}
|
|
|
|
// AddUser inserts a new user into the datastore.
|
|
// If the user login already exists an error is returned.
|
|
func (db *Userstore) AddUser(user *types.User) error {
|
|
return createUser(db, rebind(stmtUserInsert), user)
|
|
}
|
|
|
|
// SetUser updates an existing user.
|
|
func (db *Userstore) SetUser(user *types.User) error {
|
|
return updateUser(db, rebind(stmtUserUpdate), user)
|
|
}
|
|
|
|
// DelUser removes the user from the datastore.
|
|
func (db *Userstore) DelUser(user *types.User) error {
|
|
var _, err = db.Exec(rebind(stmtUserDelete), user.ID)
|
|
return err
|
|
}
|
|
|
|
// SQL query to retrieve a build feed for the given
|
|
// user account.
|
|
const userFeedQuery = `
|
|
SELECT
|
|
r.repo_id
|
|
,r.repo_owner
|
|
,r.repo_name
|
|
,r.repo_full_name
|
|
,b.build_number
|
|
,b.build_status
|
|
,b.build_started
|
|
,b.build_finished
|
|
FROM
|
|
builds b
|
|
,repos r
|
|
,stars s
|
|
WHERE b.build_repo_id = r.repo_id
|
|
AND r.repo_id = s.star_repo_id
|
|
AND s.star_user_id = ?
|
|
ORDER BY b.build_number DESC
|
|
LIMIT ? OFFSET ?
|
|
`
|
|
|
|
func scanRepoCommits(rows *sql.Rows) ([]*types.RepoCommit, error) {
|
|
var err error
|
|
var vv []*types.RepoCommit
|
|
for rows.Next() {
|
|
v := &types.RepoCommit{}
|
|
err = rows.Scan(
|
|
&v.ID,
|
|
&v.Owner,
|
|
&v.Name,
|
|
&v.FullName,
|
|
&v.Number,
|
|
&v.Status,
|
|
&v.Started,
|
|
&v.Finished,
|
|
)
|
|
if err != nil {
|
|
return vv, err
|
|
}
|
|
vv = append(vv, v)
|
|
}
|
|
return vv, rows.Err()
|
|
}
|