package consensusstatestore

import (
	"github.com/kaspanet/kaspad/domain/consensus/model"
	"github.com/kaspanet/kaspad/domain/consensus/model/externalapi"
)

// consensusStateStore represents a store for the current consensus state
type consensusStateStore struct {
	stagedTips               []*externalapi.DomainHash
	stagedVirtualDiffParents []*externalapi.DomainHash
	stagedVirtualUTXODiff    *model.UTXODiff
}

// New instantiates a new ConsensusStateStore
func New() model.ConsensusStateStore {
	return &consensusStateStore{}
}

func (c consensusStateStore) Discard() {
	c.stagedTips = nil
	c.stagedVirtualUTXODiff = nil
	c.stagedVirtualDiffParents = nil
}

func (c consensusStateStore) Commit(dbTx model.DBTransaction) error {
	err := c.commitTips(dbTx)
	if err != nil {
		return err
	}
	err = c.commitVirtualDiffParents(dbTx)
	if err != nil {
		return err
	}
	err = c.commitVirtualUTXODiff(dbTx)
	if err != nil {
		return err
	}

	return nil
}

func (c consensusStateStore) IsStaged() bool {
	return c.stagedTips != nil ||
		c.stagedVirtualDiffParents != nil ||
		c.stagedVirtualUTXODiff != nil
}

func (c consensusStateStore) VirtualUTXOSetIterator(dbContext model.DBReader) (model.ReadOnlyUTXOSetIterator, error) {
	panic("implement me")
}

func (c consensusStateStore) StageVirtualUTXOSet(virtualUTXOSetIterator model.ReadOnlyUTXOSetIterator) {
	panic("implement me")
}