package tests

import (
	"encoding/json"
	"math/big"

	"github.com/holiman/uint256"

	"github.com/ledgerwatch/turbo-geth/common/hexutil"
	"github.com/ledgerwatch/turbo-geth/common/math"
	"github.com/ledgerwatch/turbo-geth/core/types"
)

var _ = (*stTransactionMarshaling)(nil)

// MarshalJSON marshals as JSON.
func (s stTransaction) MarshalJSON() ([]byte, error) {
	type stTransaction struct {
		GasPrice    *math.HexOrDecimal256 `json:"gasPrice"`
		Nonce       math.HexOrDecimal64   `json:"nonce"`
		To          string                `json:"to"`
		Data        []string              `json:"data"`
		AccessLists []*types.AccessList   `json:"accessLists,omitempty"`
		GasLimit    []math.HexOrDecimal64 `json:"gasLimit"`
		Value       []string              `json:"value"`
		PrivateKey  hexutil.Bytes         `json:"secretKey"`
	}
	var enc stTransaction
	enc.GasPrice = (*math.HexOrDecimal256)(s.GasPrice.ToBig())
	enc.Nonce = math.HexOrDecimal64(s.Nonce)
	enc.To = s.To
	enc.Data = s.Data
	enc.AccessLists = s.AccessLists
	if s.GasLimit != nil {
		enc.GasLimit = make([]math.HexOrDecimal64, len(s.GasLimit))
		for k, v := range s.GasLimit {
			enc.GasLimit[k] = math.HexOrDecimal64(v)
		}
	}
	enc.Value = s.Value
	enc.PrivateKey = s.PrivateKey
	return json.Marshal(&enc)
}

// UnmarshalJSON unmarshals from JSON.
func (s *stTransaction) UnmarshalJSON(input []byte) error {
	type stTransaction struct {
		GasPrice    *math.HexOrDecimal256 `json:"gasPrice"`
		Nonce       *math.HexOrDecimal64  `json:"nonce"`
		To          *string               `json:"to"`
		Data        []string              `json:"data"`
		AccessLists []*types.AccessList   `json:"accessLists,omitempty"`
		GasLimit    []math.HexOrDecimal64 `json:"gasLimit"`
		Value       []string              `json:"value"`
		PrivateKey  *hexutil.Bytes        `json:"secretKey"`
	}
	var dec stTransaction
	if err := json.Unmarshal(input, &dec); err != nil {
		return err
	}
	if dec.GasPrice != nil {
		s.GasPrice, _ = uint256.FromBig((*big.Int)(dec.GasPrice))
	}
	if dec.Nonce != nil {
		s.Nonce = uint64(*dec.Nonce)
	}
	if dec.To != nil {
		s.To = *dec.To
	}
	if dec.Data != nil {
		s.Data = dec.Data
	}
	if dec.AccessLists != nil {
		s.AccessLists = dec.AccessLists
	}
	if dec.GasLimit != nil {
		s.GasLimit = make([]uint64, len(dec.GasLimit))
		for k, v := range dec.GasLimit {
			s.GasLimit[k] = uint64(v)
		}
	}
	if dec.Value != nil {
		s.Value = dec.Value
	}
	if dec.PrivateKey != nil {
		s.PrivateKey = *dec.PrivateKey
	}
	return nil
}