package inmemory

import (
	"github.com/vulcanize/vulcanizedb/pkg/core"
	"github.com/vulcanize/vulcanizedb/pkg/datastore"
)

type BlockRepository struct {
	*InMemory
}

func (blockRepository *BlockRepository) CreateOrUpdateBlock(block core.Block) (int64, error) {
	blockRepository.CreateOrUpdateBlockCallCount++
	blockRepository.blocks[block.Number] = block
	for _, transaction := range block.Transactions {
		blockRepository.receipts[transaction.Hash] = transaction.Receipt
		blockRepository.logs[transaction.TxHash] = transaction.Logs
	}
	return 0, nil
}

func (blockRepository *BlockRepository) GetBlock(blockNumber int64) (core.Block, error) {
	if block, ok := blockRepository.blocks[blockNumber]; ok {
		return block, nil
	}
	return core.Block{}, datastore.ErrBlockDoesNotExist(blockNumber)
}

func (blockRepository *BlockRepository) MissingBlockNumbers(startingBlockNumber int64, endingBlockNumber int64, nodeId string) []int64 {
	missingNumbers := []int64{}
	for blockNumber := int64(startingBlockNumber); blockNumber <= endingBlockNumber; blockNumber++ {
		if _, ok := blockRepository.blocks[blockNumber]; !ok {
			missingNumbers = append(missingNumbers, blockNumber)
		}
	}
	return missingNumbers
}

func (blockRepository *BlockRepository) SetBlocksStatus(chainHead int64) {
	for key, block := range blockRepository.blocks {
		if key < (chainHead - blocksFromHeadBeforeFinal) {
			tmp := block
			tmp.IsFinal = true
			blockRepository.blocks[key] = tmp
		}
	}
}

func (blockRepository *BlockRepository) BlockCount() int {
	return len(blockRepository.blocks)
}