package rpc

import (
	"context"
	"encoding/json"
	"errors"

	"github.com/ethereum/go-ethereum/rpc"
	"github.com/harmony-one/harmony/hmy"
)

type PublicParityTracerService struct {
	*PublicTracerService
}

// trace_block RPC
func (s *PublicParityTracerService) Block(ctx context.Context, number rpc.BlockNumber) (interface{}, error) {
	block := s.hmy.BlockChain.GetBlockByNumber(uint64(number))
	if block == nil {
		return block, errors.New("block not exist")
	}
	traceJs := "blockTracer"
	results, err := s.hmy.TraceBlock(ctx, block, &hmy.TraceConfig{Tracer: &traceJs})
	if err != nil {
		return results, err
	}
	var resultArray = make([]interface{}, 0)
	for _, result := range results {
		raw, ok := result.Result.(json.RawMessage)
		if !ok {
			return results, errors.New("expected json.RawMessage")
		}
		var subArray []interface{}
		json.Unmarshal(raw, &subArray)
		resultArray = append(resultArray, subArray...)
	}
	return resultArray, nil
}