package vm

import (
	"code.google.com/p/go.crypto/ripemd160"
	"crypto/sha256"
	"github.com/tendermint/tendermint/vm/secp256k1"
	"github.com/tendermint/tendermint/vm/sha3"

	. "github.com/tendermint/tendermint/common"
)

var nativeContracts = make(map[Word]NativeContract)

func init() {
	nativeContracts[Uint64ToWord(1)] = ecrecoverFunc
	nativeContracts[Uint64ToWord(2)] = sha256Func
	nativeContracts[Uint64ToWord(3)] = ripemd160Func
	nativeContracts[Uint64ToWord(4)] = identityFunc
}

//-----------------------------------------------------------------------------

type NativeContract func(input []byte, gas *uint64) (output []byte, err error)

func ecrecoverFunc(input []byte, gas *uint64) (output []byte, err error) {
	// Deduct gas
	gasRequired := GasEcRecover
	if *gas < gasRequired {
		return nil, ErrInsufficientGas
	} else {
		*gas -= gasRequired
	}
	// Recover
	hash := input[:32]
	v := byte(input[32] - 27) // ignore input[33:64], v is small.
	sig := append(input[64:], v)

	recovered, err := secp256k1.RecoverPubkey(hash, sig)
	if err != nil {
		return nil, err
	}
	hashed := sha3.Sha3(recovered[1:])
	return RightPadBytes(hashed, 32), nil
}

func sha256Func(input []byte, gas *uint64) (output []byte, err error) {
	// Deduct gas
	gasRequired := uint64((len(input)+31)/32)*GasSha256Word + GasSha256Base
	if *gas < gasRequired {
		return nil, ErrInsufficientGas
	} else {
		*gas -= gasRequired
	}
	// Hash
	hasher := sha256.New()
	_, err = hasher.Write(input)
	if err != nil {
		panic(err)
	}
	return hasher.Sum(nil), nil
}

func ripemd160Func(input []byte, gas *uint64) (output []byte, err error) {
	// Deduct gas
	gasRequired := uint64((len(input)+31)/32)*GasRipemd160Word + GasRipemd160Base
	if *gas < gasRequired {
		return nil, ErrInsufficientGas
	} else {
		*gas -= gasRequired
	}
	// Hash
	hasher := ripemd160.New()
	_, err = hasher.Write(input)
	if err != nil {
		panic(err)
	}
	return RightPadBytes(hasher.Sum(nil), 32), nil
}

func identityFunc(input []byte, gas *uint64) (output []byte, err error) {
	// Deduct gas
	gasRequired := uint64((len(input)+31)/32)*GasIdentityWord + GasIdentityBase
	if *gas < gasRequired {
		return nil, ErrInsufficientGas
	} else {
		*gas -= gasRequired
	}
	// Return identity
	return input, nil
}