package store

// persistent key-value store
// multiple key-value store can be managed by a single db connection
// each store has a name, before usage the name of the store must be
// ensured to exist
//
// TODO: this is a toy implementation

import (
	"os"
	"path/filepath"
	"io/ioutil"
)


type Conn struct {
	path string
}

func Open(root string) (c *Conn, err error) {
	c = new(Conn)
	c.path, err = filepath.Abs(root)
	if err != nil {
		return
	}
	err = os.MkdirAll(c.path, 0755)
	if err != nil {
		return
	}
	return
}

func (c *Conn) Get(name, k string) (v []byte, err error) {
	return ioutil.ReadFile(filepath.Join(c.path, name, k))
}

func (c *Conn) Ensure(name string) (err error) {
	return os.MkdirAll(filepath.Join(c.path, name), 0755)
}

func (c *Conn) Set(name, k string, v []byte) (err error) {
	fn := filepath.Join(c.path, name, k)
	f, err := os.OpenFile(fn+".tmp", os.O_CREATE|os.O_TRUNC|os.O_WRONLY|os.O_SYNC, 0666)
	if err != nil {
		return
	}
	defer f.Close()
	_, err = f.Write(v)
	if err != nil {
		return
	}
	err = os.Rename(fn+".tmp", fn)
	return
}

func (c *Conn) Append(name, k string, v []byte) (err error) {
	fn := filepath.Join(c.path, name, k)
	f, err := os.OpenFile(fn, os.O_CREATE|os.O_APPEND|os.O_WRONLY|os.O_SYNC, 0666)
	if err != nil {
		return
	}
	defer f.Close()
	_, err = f.Write(v)
	return
}

func (c *Conn) Close() (err error) {
	return
}