package arangodb

import (
	"context"
	"fmt"

	"github.com/arangodb/go-driver"
	"github.com/chrislusf/seaweedfs/weed/filer"
	"github.com/chrislusf/seaweedfs/weed/glog"
)

func (store *ArangodbStore) KvPut(ctx context.Context, key []byte, value []byte) (err error) {
	model := &Model{
		Key:       hashString(".kvstore." + string(key)),
		Directory: ".kvstore." + string(key),
		Meta:      bytesToArray(value),
	}

	exists, err := store.kvCollection.DocumentExists(ctx, model.Key)
	if err != nil {
		return fmt.Errorf("kv put: %v", err)
	}
	if exists {
		_, err = store.kvCollection.UpdateDocument(ctx, model.Key, model)
	} else {
		_, err = store.kvCollection.CreateDocument(ctx, model)
	}
	if err != nil {
		return fmt.Errorf("kv put: %v", err)
	}

	return nil
}
func (store *ArangodbStore) KvGet(ctx context.Context, key []byte) (value []byte, err error) {
	var model Model
	_, err = store.kvCollection.ReadDocument(ctx, hashString(".kvstore."+string(key)), &model)
	if driver.IsNotFound(err) {
		return nil, filer.ErrKvNotFound
	}
	if err != nil {
		glog.Errorf("kv get: %s %v", string(key), err)
		return nil, filer.ErrKvNotFound
	}
	return arrayToBytes(model.Meta), nil
}

func (store *ArangodbStore) KvDelete(ctx context.Context, key []byte) (err error) {
	_, err = store.kvCollection.RemoveDocument(ctx, hashString(".kvstore."+string(key)))
	if err != nil {
		glog.Errorf("kv del: %v", err)
		return filer.ErrKvNotFound
	}
	return nil
}
