package chain

import (
	"context"
	"math/big"

	"github.com/ethereum/go-ethereum"
	"github.com/ethereum/go-ethereum/common"
	"github.com/ethereum/go-ethereum/common/hexutil"
	"github.com/ethereum/go-ethereum/core/types"
	"github.com/ethereum/go-ethereum/ethclient"
	"github.com/status-im/status-go/rpc"
	"github.com/status-im/status-go/services/rpcstats"
)

type Client struct {
	eth       *ethclient.Client
	ChainID   uint64
	rpcClient *rpc.Client
}

type FeeHistory struct {
	BaseFeePerGas []string `json:"baseFeePerGas"`
}

func NewClient(rpc *rpc.Client, chainID uint64) (*Client, error) {
	ethClient, err := rpc.EthClient(chainID)
	if err != nil {
		return nil, err
	}
	return &Client{ethClient, chainID, rpc}, nil
}

func NewLegacyClient(rpc *rpc.Client) (*Client, error) {
	return NewClient(rpc, rpc.UpstreamChainID)
}

func NewClients(rpc *rpc.Client, chainIDs []uint64) (res []*Client, err error) {
	for _, chainID := range chainIDs {
		client, err := NewClient(rpc, chainID)
		if err != nil {
			return nil, err
		}
		res = append(res, client)
	}
	return res, nil
}

func (cc *Client) ToBigInt() *big.Int {
	return big.NewInt(int64(cc.ChainID))
}

func (cc *Client) HeaderByHash(ctx context.Context, hash common.Hash) (*types.Header, error) {
	rpcstats.CountCall("eth_getBlockByHash")
	return cc.eth.HeaderByHash(ctx, hash)
}

func (cc *Client) HeaderByNumber(ctx context.Context, number *big.Int) (*types.Header, error) {
	rpcstats.CountCall("eth_getBlockByNumber")
	return cc.eth.HeaderByNumber(ctx, number)
}

func (cc *Client) BlockByHash(ctx context.Context, hash common.Hash) (*types.Block, error) {
	rpcstats.CountCall("eth_getBlockByHash")
	return cc.eth.BlockByHash(ctx, hash)
}

func (cc *Client) BlockByNumber(ctx context.Context, number *big.Int) (*types.Block, error) {
	rpcstats.CountCall("eth_getBlockByNumber")
	return cc.eth.BlockByNumber(ctx, number)
}

func (cc *Client) BalanceAt(ctx context.Context, account common.Address, blockNumber *big.Int) (*big.Int, error) {
	rpcstats.CountCall("eth_getBalance")
	return cc.eth.BalanceAt(ctx, account, blockNumber)
}

func (cc *Client) NonceAt(ctx context.Context, account common.Address, blockNumber *big.Int) (uint64, error) {
	rpcstats.CountCall("eth_getTransactionCount")
	return cc.eth.NonceAt(ctx, account, blockNumber)
}

func (cc *Client) TransactionReceipt(ctx context.Context, txHash common.Hash) (*types.Receipt, error) {
	rpcstats.CountCall("eth_getTransactionReceipt")
	return cc.eth.TransactionReceipt(ctx, txHash)
}

func (cc *Client) TransactionByHash(ctx context.Context, hash common.Hash) (tx *types.Transaction, isPending bool, err error) {
	rpcstats.CountCall("eth_getTransactionByHash")
	return cc.eth.TransactionByHash(ctx, hash)
}

func (cc *Client) FilterLogs(ctx context.Context, q ethereum.FilterQuery) ([]types.Log, error) {
	rpcstats.CountCall("eth_getLogs")
	return cc.eth.FilterLogs(ctx, q)
}

func (cc *Client) CodeAt(ctx context.Context, contract common.Address, blockNumber *big.Int) ([]byte, error) {
	rpcstats.CountCall("eth_getCode")
	return cc.eth.CodeAt(ctx, contract, blockNumber)
}

func (cc *Client) CallContract(ctx context.Context, call ethereum.CallMsg, blockNumber *big.Int) ([]byte, error) {
	rpcstats.CountCall("eth_call")
	return cc.eth.CallContract(ctx, call, blockNumber)
}

func (cc *Client) GetBaseFeeFromBlock(blockNumber *big.Int) (string, error) {
	var feeHistory FeeHistory
	err := cc.rpcClient.Call(&feeHistory, cc.ChainID, "eth_feeHistory", "0x1", (*hexutil.Big)(blockNumber), nil)
	if err != nil {
		return "", err
	}

	var baseGasFee string = ""
	if len(feeHistory.BaseFeePerGas) > 0 {
		baseGasFee = feeHistory.BaseFeePerGas[0]
	}

	return baseGasFee, err
}