domain

type ForfeitTx struct {
	Txid string
	Tx   string
}

type MarketHour struct {
	StartTime     time.Time
	EndTime       time.Time
	Period        time.Duration
	RoundInterval time.Duration
	UpdatedAt     time.Time
}

type MarketHourRepo interface {
	Get(ctx context.Context) (*MarketHour, error)
	Upsert(ctx context.Context, marketHour MarketHour) error
	Close()
}

type NoteRepository interface {
	Contains(ctx context.Context, id uint64) (bool, error)
	Add(ctx context.Context, id uint64) error
	Close()
}

type Outpoint VtxoKey

type Receiver struct {
	Amount         uint64
	OnchainAddress string // onchain
	PubKey         string // offchain
}

type Round struct {
	Id                string
	StartingTimestamp int64
	EndingTimestamp   int64
	Stage             Stage
	TxRequests        map[string]TxRequest
	Txid              string
	UnsignedTx        string
	ForfeitTxs        []ForfeitTx
	VtxoTree          tree.TxTree
	Connectors        tree.TxTree
	ConnectorAddress  string
	DustAmount        uint64
	Version           uint
	Swept             bool // true if all the vtxos are vtxo.Swept or vtxo.Redeemed
	// contains filtered or unexported fields
}

type RoundEvent interface {
	IsEvent()
}

type RoundEventRepository interface {
	Save(ctx context.Context, id string, events ...RoundEvent) (*Round, error)
	Load(ctx context.Context, id string) (*Round, error)
	RegisterEventsHandler(func(*Round))
	Close()
}

type RoundFailed struct {
	Id        string
	Err       string
	Timestamp int64
}

type RoundFinalizationStarted struct {
	Id               string
	VtxoTree         tree.TxTree
	Connectors       tree.TxTree
	ConnectorAddress string
	RoundTx          string
	MinRelayFeeRate  int64
	ConnectorsIndex  map[string]Outpoint
}

type RoundFinalized struct {
	Id         string
	Txid       string
	ForfeitTxs []ForfeitTx
	Timestamp  int64
}

type RoundRepository interface {
	AddOrUpdateRound(ctx context.Context, round Round) error
	GetRoundWithId(ctx context.Context, id string) (*Round, error)
	GetRoundWithTxid(ctx context.Context, txid string) (*Round, error)
	GetRoundStats(ctx context.Context, roundTxid string) (*RoundStats, error)
	GetRoundForfeitTxs(ctx context.Context, roundTxid string) ([]ForfeitTx, error)
	GetRoundConnectorTree(ctx context.Context, roundTxid string) (tree.TxTree, error)
	GetVtxoTreeWithTxid(ctx context.Context, txid string) (tree.TxTree, error)
	GetExpiredRoundsTxid(ctx context.Context) ([]string, error)
	GetRoundsIds(ctx context.Context, startedAfter int64, startedBefore int64) ([]string, error)
	GetSweptRoundsConnectorAddress(ctx context.Context) ([]string, error)
	GetTxsWithTxids(ctx context.Context, txids []string) ([]string, error)
	GetExistingRounds(ctx context.Context, txids []string) (map[string]any, error)
	Close()
}

type RoundStage int

type RoundStarted struct {
	Id        string
	Timestamp int64
}

type RoundStats struct {
	Swept              bool
	TotalForfeitAmount uint64
	TotalInputVtxos    int32
	TotalBatchAmount   uint64
	TotalOutputVtxos   int32
	ExpiresAt          int64
	Started            int64
	Ended              int64
}

type SigningType uint8

type Stage struct {
	Code   RoundStage
	Ended  bool
	Failed bool
}

type TxRequest struct {
	Id        string
	Inputs    []Vtxo
	Receivers []Receiver
}

type TxRequestsRegistered struct {
	Id         string
	TxRequests []TxRequest
}

type Vtxo struct {
	VtxoKey
	Amount    uint64
	PubKey    string
	RoundTxid string
	SpentBy   string // round txid or redeem txid
	Spent     bool
	Redeemed  bool
	Swept     bool
	ExpireAt  int64
	RedeemTx  string // empty if in-round vtxo
	CreatedAt int64
}

type VtxoKey struct {
	Txid string
	VOut uint32
}

type VtxoRepository interface {
	AddVtxos(ctx context.Context, vtxos []Vtxo) error
	SpendVtxos(ctx context.Context, vtxos []VtxoKey, txid string) error
	RedeemVtxos(ctx context.Context, vtxos []VtxoKey) error
	GetVtxos(ctx context.Context, vtxos []VtxoKey) ([]Vtxo, error)
	GetVtxosForRound(ctx context.Context, txid string) ([]Vtxo, error)
	SweepVtxos(ctx context.Context, vtxos []VtxoKey) error
	GetAllNonRedeemedVtxos(ctx context.Context, pubkey string) ([]Vtxo, []Vtxo, error)
	GetAllSweepableVtxos(ctx context.Context) ([]Vtxo, error)
	GetSpendableVtxosWithPubKey(ctx context.Context, pubkey string) ([]Vtxo, error)
	GetAll(ctx context.Context) ([]Vtxo, error)
	GetAllVtxosWithPubKey(ctx context.Context, pubkey string) ([]Vtxo, []Vtxo, error)
	GetAllVtxosWithPubKeys(ctx context.Context, pubkeys []string, spendableOnly, spentOnly bool) ([]Vtxo, error)
	UpdateExpireAt(ctx context.Context, vtxos []VtxoKey, expireAt int64) error
	GetLeafVtxosForRound(ctx context.Context, txid string) ([]Vtxo, error)
	Close()
}