vault_index.go

package main

import (
	"encoding/json"
	"fmt"
)

type VaultIndex struct {
	parent *Vault
	path   string
	Data   map[string]interface{}
}

func (v *Vault) GetIndex(path string) (*VaultIndex, error) {
	Data := map[string]interface{}{}
	exists, err := v.Get(path, &Data)
	if err != nil {
		return nil, err
	}
	if !exists {
		Data = make(map[string]interface{})
	}
	return &VaultIndex{
		parent: v,
		path:   path,
		Data:   Data,
	}, nil
}

func (vi *VaultIndex) Lookup(key string) (interface{}, error) {
	v, ok := vi.Data[key]
	if !ok {
		return nil, fmt.Errorf("key '%s' not found in index", key)
	}
	return v, nil
}

func (vi *VaultIndex) Save() error {
	if len(vi.Data) == 0 {
		return vi.parent.Delete(vi.path)
	}
	return vi.parent.Put(vi.path, vi.Data)
}

func (vi *VaultIndex) JSON() string {
	b, err := json.Marshal(vi.Data)
	if err != nil {
		return `{"error":"json failed"}`
	}

	return string(b)
}