43 lines
1.1 KiB
Go
43 lines
1.1 KiB
Go
package builtin
|
|
|
|
import (
|
|
"database/sql"
|
|
|
|
"github.com/drone/drone/pkg/types"
|
|
)
|
|
|
|
type Tokenstore struct {
|
|
*sql.DB
|
|
}
|
|
|
|
func NewTokenstore(db *sql.DB) *Tokenstore {
|
|
return &Tokenstore{db}
|
|
}
|
|
|
|
// Token returns a token by ID.
|
|
func (db *Tokenstore) Token(id int64) (*types.Token, error) {
|
|
return getToken(db, rebind(stmtTokenSelect), id)
|
|
}
|
|
|
|
// TokenLabel returns a token by label
|
|
func (db *Tokenstore) TokenLabel(user *types.User, label string) (*types.Token, error) {
|
|
return getToken(db, rebind(stmtTokenSelectTokenUserLabel), user.ID, label)
|
|
}
|
|
|
|
// TokenList returns a list of all user tokens.
|
|
func (db *Tokenstore) TokenList(user *types.User) ([]*types.Token, error) {
|
|
return getTokens(db, rebind(stmtTokenSelectTokenUserId), user.ID)
|
|
}
|
|
|
|
// AddToken inserts a new token into the datastore.
|
|
// If the token label already exists for the user
|
|
// an error is returned.
|
|
func (db *Tokenstore) AddToken(token *types.Token) error {
|
|
return createToken(db, rebind(stmtTokenInsert), token)
|
|
}
|
|
|
|
// DelToken removes the DelToken from the datastore.
|
|
func (db *Tokenstore) DelToken(token *types.Token) error {
|
|
var _, err = db.Exec(rebind(stmtTokenDelete), token.ID)
|
|
return err
|
|
}
|