package manager

import (
	"github.com/meshplus/bitxhub-core/governance"
	"github.com/meshplus/bitxhub-kit/storage"
	"github.com/sirupsen/logrus"

	"gitlab.33.cn/link33/sidecar/pkg/json"
)

var _ governance.Persister = (*Persister)(nil)

type Persister struct {
	addr    string
	storage storage.Storage
	logger  logrus.FieldLogger
}

func NewPersister(add string, store storage.Storage, logger logrus.FieldLogger) Persister {
	return Persister{
		addr:    add,
		storage: store,
		logger:  logger,
	}
}

func (m Persister) Caller() string {
	return m.addr
}

func (m Persister) Logger() logrus.FieldLogger {
	return m.logger
}

func (m Persister) Has(key string) bool {
	return m.storage.Has([]byte(key))
}

func (m Persister) Get(key string) (bool, []byte) {
	return true, m.storage.Get([]byte(key))
}

func (m Persister) GetObject(key string, ret interface{}) bool {
	ok, data := m.Get(key)
	if !ok {
		return false
	}
	err := json.Unmarshal(data, ret)
	return err == nil
}

func (m Persister) Set(key string, value []byte) {
	m.storage.Put([]byte(key), value)
}

func (m Persister) SetObject(key string, value interface{}) {
	data, err := json.Marshal(value)
	if err != nil {
		panic(err.Error())
	}
	m.Set(key, data)
}

func (m Persister) Delete(key string) {
	m.storage.Delete([]byte(key))
}

func (m Persister) Query(prefix string) (bool, [][]byte) {
	var ret [][]byte
	it := m.storage.Prefix([]byte(prefix))
	for it.Next() {
		val := make([]byte, len(it.Value()))
		copy(val, it.Value())
		ret = append(ret, val)
	}
	return len(ret) != 0, ret
}

func (m Persister) GetAccount(_ string) (bool, interface{}) {
	return false, nil
}