package rpc

import (
	"fmt"
	"github.com/kaspanet/kaspad/rpc/model"
	"github.com/kaspanet/kaspad/util/daghash"
)

const (
	// maxBlocksInGetChainFromBlockResult is the max amount of blocks that
	// are allowed in a GetChainFromBlockResult.
	maxBlocksInGetChainFromBlockResult = 1000
)

// handleGetChainFromBlock implements the getChainFromBlock command.
func handleGetChainFromBlock(s *Server, cmd interface{}, closeChan <-chan struct{}) (interface{}, error) {
	if s.acceptanceIndex == nil {
		return nil, &model.RPCError{
			Code: model.ErrRPCNoAcceptanceIndex,
			Message: "The acceptance index must be " +
				"enabled to get the selected parent chain " +
				"(specify --acceptanceindex)",
		}
	}

	c := cmd.(*model.GetChainFromBlockCmd)
	var startHash *daghash.Hash
	if c.StartHash != nil {
		startHash = &daghash.Hash{}
		err := daghash.Decode(startHash, *c.StartHash)
		if err != nil {
			return nil, rpcDecodeHexError(*c.StartHash)
		}
	}

	s.dag.RLock()
	defer s.dag.RUnlock()

	// If startHash is not in the selected parent chain, there's nothing
	// to do; return an error.
	if startHash != nil && !s.dag.IsInDAG(startHash) {
		return nil, &model.RPCError{
			Code:    model.ErrRPCBlockNotFound,
			Message: "Block not found in the DAG",
		}
	}

	// Retrieve the selected parent chain.
	removedChainHashes, addedChainHashes, err := s.dag.SelectedParentChain(startHash)
	if err != nil {
		return nil, err
	}

	// Limit the amount of blocks in the response
	if len(addedChainHashes) > maxBlocksInGetChainFromBlockResult {
		addedChainHashes = addedChainHashes[:maxBlocksInGetChainFromBlockResult]
	}

	// Collect addedChainBlocks.
	addedChainBlocks, err := collectChainBlocks(s, addedChainHashes)
	if err != nil {
		return nil, &model.RPCError{
			Code:    model.ErrRPCInternal.Code,
			Message: fmt.Sprintf("could not collect chain blocks: %s", err),
		}
	}

	// Collect removedHashes.
	removedHashes := make([]string, len(removedChainHashes))
	for i, hash := range removedChainHashes {
		removedHashes[i] = hash.String()
	}

	result := &model.GetChainFromBlockResult{
		RemovedChainBlockHashes: removedHashes,
		AddedChainBlocks:        addedChainBlocks,
		Blocks:                  nil,
	}

	// If the user specified to include the blocks, collect them as well.
	if c.IncludeBlocks {
		getBlockVerboseResults, err := hashesToGetBlockVerboseResults(s, addedChainHashes)
		if err != nil {
			return nil, err
		}
		result.Blocks = getBlockVerboseResults
	}

	return result, nil
}