package descriptor

import (
	"encoding/hex"
	"errors"

	"github.com/btcsuite/btcd/btcec/v2/schnorr"
	"github.com/decred/dcrd/dcrec/secp256k1/v4"
)

// tr(unspendable, { and(pk(user), pk(asp)), and(older(timeout), pk(user)) })
const DefaultVtxoDescriptorTemplate = "tr(%s,{ and(pk(%s), pk(%s)), and(older(%d), pk(%s)) })"

// tr(unspendable, { { and(pk(sender), pk(asp)), and(older(timeout), pk(sender)) }, and(pk(receiver), pk(asp)) })
const ReversibleVtxoScriptTemplate = "tr(%s,{ { and(pk(%s), pk(%s)), and(older(%d), pk(%s)) }, and(pk(%s), pk(%s)) })"

func ParseReversibleVtxoDescriptor(
	desc TaprootDescriptor,
) (user, sender, asp *secp256k1.PublicKey, timeout uint, err error) {
	if len(desc.ScriptTree) != 3 {
		return nil, nil, nil, 0, errors.New("not a reversible vtxo script descriptor")
	}

	for _, leaf := range desc.ScriptTree {
		if andLeaf, ok := leaf.(*And); ok {
			if first, ok := andLeaf.First.(*PK); ok {
				if second, ok := andLeaf.Second.(*PK); ok {
					keyBytes, err := hex.DecodeString(first.Key.Hex)
					if err != nil {
						return nil, nil, nil, 0, err
					}
					if sender == nil {
						sender, err = schnorr.ParsePubKey(keyBytes)
						if err != nil {
							return nil, nil, nil, 0, err
						}
					} else {
						user, err = schnorr.ParsePubKey(keyBytes)
						if err != nil {
							return nil, nil, nil, 0, err
						}
					}

					if asp == nil {
						keyBytes, err = hex.DecodeString(second.Key.Hex)
						if err != nil {
							return nil, nil, nil, 0, err
						}

						asp, err = schnorr.ParsePubKey(keyBytes)
						if err != nil {
							return nil, nil, nil, 0, err
						}
					}
				}
			}

			if first, ok := andLeaf.First.(*Older); ok {
				if second, ok := andLeaf.Second.(*PK); ok {
					timeout = first.Timeout
					keyBytes, err := hex.DecodeString(second.Key.Hex)
					if err != nil {
						return nil, nil, nil, 0, err
					}

					sender, err = schnorr.ParsePubKey(keyBytes)
					if err != nil {
						return nil, nil, nil, 0, err
					}
				}
			}
		}
	}

	if user == nil {
		return nil, nil, nil, 0, errors.New("descriptor is invalid")
	}

	if asp == nil {
		return nil, nil, nil, 0, errors.New("descriptor is invalid")
	}

	if timeout == 0 {
		return nil, nil, nil, 0, errors.New("descriptor is invalid")
	}

	if sender == nil {
		return nil, nil, nil, 0, errors.New("descriptor is invalid")
	}

	return
}

func ParseDefaultVtxoDescriptor(
	desc TaprootDescriptor,
) (user, asp *secp256k1.PublicKey, timeout uint, err error) {
	if len(desc.ScriptTree) != 2 {
		return nil, nil, 0, errors.New("not a default vtxo script descriptor")
	}

	for _, leaf := range desc.ScriptTree {
		if andLeaf, ok := leaf.(*And); ok {
			if first, ok := andLeaf.First.(*PK); ok {
				if second, ok := andLeaf.Second.(*PK); ok {
					keyBytes, err := hex.DecodeString(first.Key.Hex)
					if err != nil {
						return nil, nil, 0, err
					}

					user, err = schnorr.ParsePubKey(keyBytes)
					if err != nil {
						return nil, nil, 0, err
					}

					keyBytes, err = hex.DecodeString(second.Key.Hex)
					if err != nil {
						return nil, nil, 0, err
					}

					asp, err = schnorr.ParsePubKey(keyBytes)
					if err != nil {
						return nil, nil, 0, err
					}
				}
			}

			if first, ok := andLeaf.First.(*Older); ok {
				if second, ok := andLeaf.Second.(*PK); ok {
					timeout = first.Timeout
					keyBytes, err := hex.DecodeString(second.Key.Hex)
					if err != nil {
						return nil, nil, 0, err
					}

					user, err = schnorr.ParsePubKey(keyBytes)
					if err != nil {
						return nil, nil, 0, err
					}
				}
			}
		}
	}

	if user == nil {
		return nil, nil, 0, errors.New("boarding descriptor is invalid")
	}

	if asp == nil {
		return nil, nil, 0, errors.New("boarding descriptor is invalid")
	}

	if timeout == 0 {
		return nil, nil, 0, errors.New("boarding descriptor is invalid")
	}

	return
}