package vm

import (
	"fmt"
	"math/big"
)

var (
	GasQuickStep   = big.NewInt(2)
	GasFastestStep = big.NewInt(3)
	GasFastStep    = big.NewInt(5)
	GasMidStep     = big.NewInt(8)
	GasSlowStep    = big.NewInt(10)
	GasExtStep     = big.NewInt(20)

	GasStorageGet        = big.NewInt(50)
	GasStorageAdd        = big.NewInt(20000)
	GasStorageMod        = big.NewInt(5000)
	GasLogBase           = big.NewInt(375)
	GasLogTopic          = big.NewInt(375)
	GasLogByte           = big.NewInt(8)
	GasCreate            = big.NewInt(32000)
	GasCreateByte        = big.NewInt(200)
	GasCall              = big.NewInt(40)
	GasCallValueTransfer = big.NewInt(9000)
	GasStipend           = big.NewInt(2300)
	GasCallNewAccount    = big.NewInt(25000)
	GasReturn            = big.NewInt(0)
	GasStop              = big.NewInt(0)
	GasJumpDest          = big.NewInt(1)

	RefundStorage = big.NewInt(15000)
	RefundSuicide = big.NewInt(24000)

	GasMemWord           = big.NewInt(3)
	GasQuadCoeffDenom    = big.NewInt(512)
	GasContractByte      = big.NewInt(200)
	GasTransaction       = big.NewInt(21000)
	GasTxDataNonzeroByte = big.NewInt(68)
	GasTxDataZeroByte    = big.NewInt(4)
	GasTx                = big.NewInt(21000)
	GasExp               = big.NewInt(10)
	GasExpByte           = big.NewInt(10)

	GasSha3Base     = big.NewInt(30)
	GasSha3Word     = big.NewInt(6)
	GasSha256Base   = big.NewInt(60)
	GasSha256Word   = big.NewInt(12)
	GasRipemdBase   = big.NewInt(600)
	GasRipemdWord   = big.NewInt(12)
	GasEcrecover    = big.NewInt(3000)
	GasIdentityBase = big.NewInt(15)
	GasIdentityWord = big.NewInt(3)
	GasCopyWord     = big.NewInt(3)
)

func baseCheck(op OpCode, stack *stack, gas *big.Int) error {
	// PUSH and DUP are a bit special. They all cost the same but we do want to have checking on stack push limit
	// PUSH is also allowed to calculate the same price for all PUSHes
	// DUP requirements are handled elsewhere (except for the stack limit check)
	if op >= PUSH1 && op <= PUSH32 {
		op = PUSH1
	}
	if op >= SWAP1 && op <= SWAP16 {
		op = SWAP1
	}

	if r, ok := _baseCheck[op]; ok {
		err := stack.require(r.stackPop)
		if err != nil {
			return err
		}

		if r.stackPush && len(stack.data)-r.stackPop == 1024 {
			return fmt.Errorf("stack limit reached (%d)", maxStack)
		}

		gas.Add(gas, r.gas)
	}
	return nil
}

func toWordSize(size *big.Int) *big.Int {
	tmp := new(big.Int)
	tmp.Add(size, u256(31))
	tmp.Div(tmp, u256(32))
	return tmp
}

type req struct {
	stackPop  int
	gas       *big.Int
	stackPush bool
}

var _baseCheck = map[OpCode]req{
	// opcode  |  stack pop | gas price | stack push
	ADD:          {2, GasFastestStep, true},
	LT:           {2, GasFastestStep, true},
	GT:           {2, GasFastestStep, true},
	SLT:          {2, GasFastestStep, true},
	SGT:          {2, GasFastestStep, true},
	EQ:           {2, GasFastestStep, true},
	ISZERO:       {1, GasFastestStep, true},
	SUB:          {2, GasFastestStep, true},
	AND:          {2, GasFastestStep, true},
	OR:           {2, GasFastestStep, true},
	XOR:          {2, GasFastestStep, true},
	NOT:          {1, GasFastestStep, true},
	BYTE:         {2, GasFastestStep, true},
	CALLDATALOAD: {1, GasFastestStep, true},
	CALLDATACOPY: {3, GasFastestStep, true},
	MLOAD:        {1, GasFastestStep, true},
	MSTORE:       {2, GasFastestStep, false},
	MSTORE8:      {2, GasFastestStep, false},
	CODECOPY:     {3, GasFastestStep, false},
	MUL:          {2, GasFastStep, true},
	DIV:          {2, GasFastStep, true},
	SDIV:         {2, GasFastStep, true},
	MOD:          {2, GasFastStep, true},
	SMOD:         {2, GasFastStep, true},
	SIGNEXTEND:   {2, GasFastStep, true},
	ADDMOD:       {3, GasMidStep, true},
	MULMOD:       {3, GasMidStep, true},
	JUMP:         {1, GasMidStep, false},
	JUMPI:        {2, GasSlowStep, false},
	EXP:          {2, GasSlowStep, true},
	ADDRESS:      {0, GasQuickStep, true},
	ORIGIN:       {0, GasQuickStep, true},
	CALLER:       {0, GasQuickStep, true},
	CALLVALUE:    {0, GasQuickStep, true},
	CODESIZE:     {0, GasQuickStep, true},
	GASPRICE:     {0, GasQuickStep, true},
	COINBASE:     {0, GasQuickStep, true},
	TIMESTAMP:    {0, GasQuickStep, true},
	NUMBER:       {0, GasQuickStep, true},
	CALLDATASIZE: {0, GasQuickStep, true},
	DIFFICULTY:   {0, GasQuickStep, true},
	GASLIMIT:     {0, GasQuickStep, true},
	POP:          {1, GasQuickStep, false},
	PC:           {0, GasQuickStep, true},
	MSIZE:        {0, GasQuickStep, true},
	GAS:          {0, GasQuickStep, true},
	BLOCKHASH:    {1, GasExtStep, true},
	BALANCE:      {0, GasExtStep, true},
	EXTCODESIZE:  {1, GasExtStep, true},
	EXTCODECOPY:  {4, GasExtStep, false},
	SLOAD:        {1, GasStorageGet, true},
	SSTORE:       {2, Zero, false},
	SHA3:         {1, GasSha3Base, true},
	CREATE:       {3, GasCreate, true},
	CALL:         {7, GasCall, true},
	CALLCODE:     {7, GasCall, true},
	JUMPDEST:     {0, GasJumpDest, false},
	SUICIDE:      {1, Zero, false},
	RETURN:       {2, Zero, false},
	PUSH1:        {0, GasFastStep, true},
	DUP1:         {0, Zero, true},
}