package keycard

import (
	"github.com/status-im/keycard-go/apdu"
	"github.com/status-im/keycard-go/globalplatform"
	"github.com/status-im/keycard-go/identifiers"
	"github.com/status-im/keycard-go/types"
)

type CashCommandSet struct {
	c                   types.Channel
	CashApplicationInfo *types.CashApplicationInfo
}

func NewCashCommandSet(c types.Channel) *CashCommandSet {
	return &CashCommandSet{
		c:                   c,
		CashApplicationInfo: &types.CashApplicationInfo{},
	}
}

func (cs *CashCommandSet) Select() error {
	cmd := globalplatform.NewCommandSelect(identifiers.CashInstanceAID)
	cmd.SetLe(0)
	resp, err := cs.c.Send(cmd)
	if err = cs.checkOK(resp, err); err != nil {
		return err
	}

	appInfo, err := types.ParseCashApplicationInfo(resp.Data)
	if err != nil {
		return err
	}

	cs.CashApplicationInfo = appInfo

	return nil
}

func (cs *CashCommandSet) Sign(data []byte) (*types.Signature, error) {
	cmd, err := NewCommandSign(data, 0x00, "")
	if err != nil {
		return nil, err
	}

	resp, err := cs.c.Send(cmd)
	if err = cs.checkOK(resp, err); err != nil {
		return nil, err
	}

	return types.ParseSignature(data, resp.Data)
}

func (cs *CashCommandSet) checkOK(resp *apdu.Response, err error, allowedResponses ...uint16) error {
	if err != nil {
		return err
	}

	if len(allowedResponses) == 0 {
		allowedResponses = []uint16{apdu.SwOK}
	}

	for _, code := range allowedResponses {
		if code == resp.Sw {
			return nil
		}
	}

	return apdu.NewErrBadResponse(resp.Sw, "unexpected response")
}