67 lines
1.6 KiB
Go
67 lines
1.6 KiB
Go
package bolt
|
|
|
|
import (
|
|
"bytes"
|
|
|
|
"github.com/boltdb/bolt"
|
|
"github.com/drone/drone/common"
|
|
)
|
|
|
|
// GetToken gets a token by sha value.
|
|
func (db *DB) GetToken(user, label string) (*common.Token, error) {
|
|
token := &common.Token{}
|
|
key := []byte(user + "/" + label)
|
|
|
|
err := db.View(func(t *bolt.Tx) error {
|
|
return get(t, bucketTokens, key, token)
|
|
})
|
|
|
|
return token, err
|
|
}
|
|
|
|
// InsertToken inserts a new user token in the datastore.
|
|
// If the token already exists and error is returned.
|
|
func (db *DB) InsertToken(token *common.Token) error {
|
|
key := []byte(token.Login + "/" + token.Label)
|
|
return db.Update(func(t *bolt.Tx) error {
|
|
// gets an index
|
|
var idx [][]byte
|
|
err := get(t, bucketUserTokens, []byte(token.Login), &idx)
|
|
if err != nil && err != ErrKeyNotFound {
|
|
return err
|
|
}
|
|
// pushes to the index
|
|
idx = append(idx, key)
|
|
err = update(t, bucketUserTokens, []byte(token.Login), &idx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return insert(t, bucketTokens, key, token)
|
|
})
|
|
}
|
|
|
|
// DeleteUser deletes the token.
|
|
func (db *DB) DeleteToken(token *common.Token) error {
|
|
key := []byte(token.Login + "/" + token.Label)
|
|
return db.Update(func(t *bolt.Tx) error {
|
|
// gets an index
|
|
var idx [][]byte
|
|
err := get(t, bucketUserTokens, []byte(token.Login), &idx)
|
|
if err != nil && err != ErrKeyNotFound {
|
|
return err
|
|
}
|
|
// pushes to the index
|
|
for i, val := range idx {
|
|
if bytes.Equal(val, key) {
|
|
idx = idx[:i+copy(idx[i:], idx[i+1:])]
|
|
break
|
|
}
|
|
}
|
|
err = update(t, bucketUserTokens, []byte(token.Login), &idx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return delete(t, bucketUser, key)
|
|
})
|
|
}
|