gokeybase/v1/kvstore.go

package keybase

import (
	"encoding/json"
	"errors"
)

// kvAPIOut sends a JSON request to the kvstore API and returns its response.
func kvAPIOut(k *Keybase, kv KVAPI) (KVAPI, error) {
	jsonBytes, _ := json.Marshal(kv)

	cmdOut, err := k.Exec("kvstore", "api", "-m", string(jsonBytes))
	if err != nil {
		return KVAPI{}, err
	}

	var r KVAPI
	if err := json.Unmarshal(cmdOut, &r); err != nil {
		return KVAPI{}, err
	}

	if r.Error != nil {
		return KVAPI{}, errors.New(r.Error.Message)
	}

	return r, nil
}

// Namespaces returns all namespaces for a team
func (kv KV) Namespaces() (KVAPI, error) {
	m := KVAPI{
		Params: &kvParams{},
	}
	m.Params.Options = kvOptions{
		Team: kv.Team,
	}

	m.Method = "list"

	r, err := kvAPIOut(kv.keybase, m)
	if err != nil {
		return r, err
	}
	return r, nil
}

// Keys returns all non-deleted keys for a namespace
func (kv KV) Keys(namespace string) (KVAPI, error) {
	m := KVAPI{
		Params: &kvParams{},
	}
	m.Params.Options = kvOptions{
		Team:      kv.Team,
		Namespace: namespace,
	}

	m.Method = "list"

	r, err := kvAPIOut(kv.keybase, m)
	if err != nil {
		return r, err
	}
	return r, nil
}

// Get returns an entry
func (kv KV) Get(namespace string, key string, revision ...uint) (KVAPI, error) {
	m := KVAPI{
		Params: &kvParams{},
	}
	m.Params.Options = kvOptions{
		Team:      kv.Team,
		Namespace: namespace,
		EntryKey:  key,
	}

	if len(revision) > 0 {
		m.Params.Options.Revision = revision[0]
	}

	m.Method = "get"

	r, err := kvAPIOut(kv.keybase, m)
	if err != nil {
		return r, err
	}
	return r, nil
}

// Put adds an entry
func (kv KV) Put(namespace string, key string, value string, revision ...uint) (KVAPI, error) {
	m := KVAPI{
		Params: &kvParams{},
	}
	m.Params.Options = kvOptions{
		Team:       kv.Team,
		Namespace:  namespace,
		EntryKey:   key,
		EntryValue: value,
	}

	if len(revision) > 0 {
		m.Params.Options.Revision = revision[0]
	}

	m.Method = "put"

	r, err := kvAPIOut(kv.keybase, m)
	if err != nil {
		return r, err
	}
	return r, nil
}

// Delete removes an entry
func (kv KV) Delete(namespace string, key string, revision ...uint) (KVAPI, error) {
	m := KVAPI{
		Params: &kvParams{},
	}
	m.Params.Options = kvOptions{
		Team:      kv.Team,
		Namespace: namespace,
		EntryKey:  key,
	}

	if len(revision) > 0 {
		m.Params.Options.Revision = revision[0]
	}

	m.Method = "del"

	r, err := kvAPIOut(kv.keybase, m)
	if err != nil {
		return r, err
	}
	return r, nil
}