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() }