package types

import (
	"fmt"
	"io"

	"github.com/vapor/encoding/blockchain"
	"github.com/vapor/errors"
	"github.com/vapor/protocol/bc"
)

// serflag variables for input types.
const (
	CrossChainInputType uint8 = iota
	SpendInputType
	CoinbaseInputType
	VetoInputType
)

type (
	// TxInput is the top level struct of tx input.
	TxInput struct {
		AssetVersion uint64
		TypedInput
		CommitmentSuffix []byte
		WitnessSuffix    []byte
	}

	// TypedInput return the txinput type.
	TypedInput interface {
		InputType() uint8
	}
)

// AssetAmount return the asset id and amount of the txinput.
func (t *TxInput) AssetAmount() bc.AssetAmount {
	switch inp := t.TypedInput.(type) {
	case *SpendInput:
		return inp.AssetAmount

	case *CrossChainInput:
		return inp.AssetAmount

	case *VetoInput:
		return inp.AssetAmount
	}

	return bc.AssetAmount{}
}

// AssetID return the assetID of the txinput
func (t *TxInput) AssetID() bc.AssetID {
	switch inp := t.TypedInput.(type) {
	case *SpendInput:
		return *inp.AssetId

	case *CrossChainInput:
		return *inp.AssetAmount.AssetId

	case *VetoInput:
		return *inp.AssetId

	}
	return bc.AssetID{}
}

// Amount return the asset amount of the txinput
func (t *TxInput) Amount() uint64 {
	switch inp := t.TypedInput.(type) {
	case *SpendInput:
		return inp.Amount

	case *CrossChainInput:
		return inp.AssetAmount.Amount

	case *VetoInput:
		return inp.Amount

	}
	return 0
}

// ControlProgram return the control program of the spend input
func (t *TxInput) ControlProgram() []byte {
	switch inp := t.TypedInput.(type) {
	case *SpendInput:
		return inp.ControlProgram

	case *CrossChainInput:
		return inp.ControlProgram

	case *VetoInput:
		return inp.ControlProgram

	}

	return nil
}

// Arguments get the args for the input
func (t *TxInput) Arguments() [][]byte {
	switch inp := t.TypedInput.(type) {
	case *SpendInput:
		return inp.Arguments

	case *CrossChainInput:
		return inp.Arguments

	case *VetoInput:
		return inp.Arguments
	}
	return nil
}

// SetArguments set the args for the input
func (t *TxInput) SetArguments(args [][]byte) {
	switch inp := t.TypedInput.(type) {
	case *SpendInput:
		inp.Arguments = args

	case *CrossChainInput:
		inp.Arguments = args

	case *VetoInput:
		inp.Arguments = args
	}
}

// SpentOutputID calculate the hash of spended output
func (t *TxInput) SpentOutputID() (o bc.Hash, err error) {
	switch inp := t.TypedInput.(type) {
	case *SpendInput:
		o, err = ComputeOutputID(&inp.SpendCommitment, SpendInputType, nil)

	case *VetoInput:
		o, err = ComputeOutputID(&inp.SpendCommitment, VetoInputType, inp.Vote)

	case *CrossChainInput:
		o, err = ComputeOutputID(&inp.SpendCommitment, SpendInputType, nil)
	}

	return o, err
}

func (t *TxInput) readFrom(r *blockchain.Reader) (err error) {
	if t.AssetVersion, err = blockchain.ReadVarint63(r); err != nil {
		return err
	}

	t.CommitmentSuffix, err = blockchain.ReadExtensibleString(r, func(r *blockchain.Reader) error {
		if t.AssetVersion != 1 {
			return nil
		}
		var icType [1]byte
		if _, err = io.ReadFull(r, icType[:]); err != nil {
			return errors.Wrap(err, "reading input commitment type")
		}

		switch icType[0] {
		case SpendInputType:
			si := new(SpendInput)
			t.TypedInput = si
			if si.SpendCommitmentSuffix, err = si.SpendCommitment.readFrom(r, 1); err != nil {
				return err
			}

		case CoinbaseInputType:
			ci := new(CoinbaseInput)
			t.TypedInput = ci
			if ci.Arbitrary, err = blockchain.ReadVarstr31(r); err != nil {
				return err
			}

		case CrossChainInputType:
			ci := new(CrossChainInput)
			t.TypedInput = ci
			if ci.SpendCommitmentSuffix, err = ci.SpendCommitment.readFrom(r, 1); err != nil {
				return err
			}

			if ci.IssuanceVMVersion, err = blockchain.ReadVarint63(r); err != nil {
				return err
			}

			if ci.AssetDefinition, err = blockchain.ReadVarstr31(r); err != nil {
				return err
			}

			if ci.IssuanceProgram, err = blockchain.ReadVarstr31(r); err != nil {
				return err
			}

		case VetoInputType:
			ui := new(VetoInput)
			t.TypedInput = ui
			if ui.VetoCommitmentSuffix, err = ui.SpendCommitment.readFrom(r, 1); err != nil {
				return err
			}

			if ui.Vote, err = blockchain.ReadVarstr31(r); err != nil {
				return err
			}

		default:
			return fmt.Errorf("unsupported input type %d", icType[0])
		}
		return nil
	})
	if err != nil {
		return err
	}

	t.WitnessSuffix, err = blockchain.ReadExtensibleString(r, func(r *blockchain.Reader) error {
		if t.AssetVersion != 1 {
			return nil
		}

		var err error
		switch inp := t.TypedInput.(type) {
		case *SpendInput:
			inp.Arguments, err = blockchain.ReadVarstrList(r)

		case *CrossChainInput:
			inp.Arguments, err = blockchain.ReadVarstrList(r)

		case *VetoInput:
			inp.Arguments, err = blockchain.ReadVarstrList(r)
		}

		return err
	})
	return err
}

func (t *TxInput) writeTo(w io.Writer) error {
	if _, err := blockchain.WriteVarint63(w, t.AssetVersion); err != nil {
		return errors.Wrap(err, "writing asset version")
	}

	if _, err := blockchain.WriteExtensibleString(w, t.CommitmentSuffix, t.writeInputCommitment); err != nil {
		return errors.Wrap(err, "writing input commitment")
	}

	if _, err := blockchain.WriteExtensibleString(w, t.WitnessSuffix, t.writeInputWitness); err != nil {
		return errors.Wrap(err, "writing input witness")
	}

	return nil
}

func (t *TxInput) writeInputCommitment(w io.Writer) (err error) {
	if t.AssetVersion != 1 {
		return nil
	}

	switch inp := t.TypedInput.(type) {
	case *SpendInput:
		if _, err = w.Write([]byte{SpendInputType}); err != nil {
			return err
		}

		return inp.SpendCommitment.writeExtensibleString(w, inp.SpendCommitmentSuffix, t.AssetVersion)

	case *CrossChainInput:
		if _, err = w.Write([]byte{CrossChainInputType}); err != nil {
			return err
		}

		if err := inp.SpendCommitment.writeExtensibleString(w, inp.SpendCommitmentSuffix, t.AssetVersion); err != nil {
			return err
		}

		if _, err := blockchain.WriteVarint63(w, inp.IssuanceVMVersion); err != nil {
			return err
		}

		if _, err := blockchain.WriteVarstr31(w, inp.AssetDefinition); err != nil {
			return err
		}

		if _, err := blockchain.WriteVarstr31(w, inp.IssuanceProgram); err != nil {
			return err
		}

	case *CoinbaseInput:
		if _, err := w.Write([]byte{CoinbaseInputType}); err != nil {
			return err
		}

		if _, err := blockchain.WriteVarstr31(w, inp.Arbitrary); err != nil {
			return errors.Wrap(err, "writing coinbase arbitrary")
		}

	case *VetoInput:
		if _, err = w.Write([]byte{VetoInputType}); err != nil {
			return err
		}

		if err := inp.SpendCommitment.writeExtensibleString(w, inp.VetoCommitmentSuffix, t.AssetVersion); err != nil {
			return err
		}

		_, err := blockchain.WriteVarstr31(w, inp.Vote)
		return err
	}
	return nil
}

func (t *TxInput) writeInputWitness(w io.Writer) error {
	if t.AssetVersion != 1 {
		return nil
	}

	var err error
	switch inp := t.TypedInput.(type) {
	case *SpendInput:
		_, err = blockchain.WriteVarstrList(w, inp.Arguments)

	case *CrossChainInput:
		_, err = blockchain.WriteVarstrList(w, inp.Arguments)

	case *VetoInput:
		_, err = blockchain.WriteVarstrList(w, inp.Arguments)
	}
	return err
}