package index

import (
	"git.gammaspectra.live/P2Pool/p2pool-observer/monero/crypto"
	"git.gammaspectra.live/P2Pool/p2pool-observer/p2pool/sidechain"
	"git.gammaspectra.live/P2Pool/p2pool-observer/types"
	"git.gammaspectra.live/P2Pool/p2pool-observer/utils"
)

const MainBlockSelectFields = "id, height, timestamp, reward, coinbase_id, difficulty, metadata, side_template_id, coinbase_private_key"

type MainBlock struct {
	Id         types.Hash `json:"id"`
	Height     uint64     `json:"height"`
	Timestamp  uint64     `json:"timestamp"`
	Reward     uint64     `json:"reward"`
	CoinbaseId types.Hash `json:"coinbase_id"`
	Difficulty uint64     `json:"difficulty"`

	// Metadata should be jsonb blob, can be NULL. metadata such as pool ownership, links to other p2pool networks, and other interesting data
	Metadata map[string]any `json:"metadata,omitempty"`

	// sidechain data for blocks we own
	// SideTemplateId can be NULL
	SideTemplateId types.Hash `json:"side_template_id,omitempty"`
	// CoinbasePrivateKey private key for coinbase outputs we own (all 0x00 = not known, but should have one)
	CoinbasePrivateKey crypto.PrivateKeyBytes `json:"coinbase_private_key,omitempty"`
}

func (b *MainBlock) GetMetadata(key string) any {
	return b.Metadata[key]
}

func (b *MainBlock) SetMetadata(key string, v any) {
	b.Metadata[key] = v
}

func (b *MainBlock) ScanFromRow(_ *sidechain.Consensus, row RowScanInterface) error {
	var metadataBuf []byte
	b.Metadata = make(map[string]any)
	if err := row.Scan(&b.Id, &b.Height, &b.Timestamp, &b.Reward, &b.CoinbaseId, &b.Difficulty, &metadataBuf, &b.SideTemplateId, &b.CoinbasePrivateKey); err != nil {
		return err
	} else if err = utils.UnmarshalJSON(metadataBuf, &b.Metadata); err != nil {
		return err
	}
	return nil
}