package splitstore

import (
	"path/filepath"

	"golang.org/x/xerrors"

	"github.com/filecoin-project/go-state-types/abi"
	cid "github.com/ipfs/go-cid"
)

type TrackingStore interface {
	Put(cid.Cid, abi.ChainEpoch) error
	PutBatch([]cid.Cid, abi.ChainEpoch) error
	Get(cid.Cid) (abi.ChainEpoch, error)
	Delete(cid.Cid) error
	DeleteBatch(map[cid.Cid]struct{}) error
	ForEach(func(cid.Cid, abi.ChainEpoch) error) error
	Sync() error
	Close() error
}

func NewTrackingStore(path string, trackingStoreType string) (TrackingStore, error) {
	switch trackingStoreType {
	case "", "bolt":
		return NewBoltTrackingStore(filepath.Join(path, "snoop.bolt"))
	case "lmdb":
		return NewLMDBTrackingStore(filepath.Join(path, "snoop.lmdb"))
	default:
		return nil, xerrors.Errorf("unknown tracking store type %s", trackingStoreType)
	}
}