woodpecker/datastore/bolt/token.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)
	})
}