package multiformat

import (
	"encoding/hex"
	"fmt"
	"testing"

	"github.com/stretchr/testify/require"
)

var (
	// secp256k1 sample public key
	secPkB = []byte{
		0x04,
		0x26, 0x1c, 0x55, 0x67, 0x5e, 0x55, 0xff, 0x25,
		0xed, 0xb5, 0x0b, 0x34, 0x5c, 0xfb, 0x3a, 0x3f,
		0x35, 0xf6, 0x07, 0x12, 0xd2, 0x51, 0xcb, 0xaa,
		0xab, 0x97, 0xbd, 0x50, 0x05, 0x4c, 0x6e, 0xbc,
		0x3c, 0xd4, 0xe2, 0x22, 0x00, 0xc6, 0x8d, 0xaf,
		0x74, 0x93, 0xe1, 0xf8, 0xda, 0x6a, 0x19, 0x0a,
		0x68, 0xa6, 0x71, 0xe2, 0xd3, 0x97, 0x78, 0x09,
		0x61, 0x24, 0x24, 0xc7, 0xc3, 0x88, 0x8b, 0xc6,
	}
	secPk   = "f" + hex.EncodeToString(secPkB)
	secPkBt = append([]byte{0xe7, 0x01}, secPkB...)
	secPkt  = "f" + hex.EncodeToString(secPkBt)

	// bls12-381 G1 sample public key
	bls12G1PkB = []byte{
		0x17, 0xf1, 0xd3, 0xa7, 0x31, 0x97, 0xd7, 0x94,
		0x26, 0x95, 0x63, 0x8c, 0x4f, 0xa9, 0xac, 0x0f,
		0xc3, 0x68, 0x8c, 0x4f, 0x97, 0x74, 0xb9, 0x05,
		0xa1, 0x4e, 0x3a, 0x3f, 0x17, 0x1b, 0xac, 0x58,
		0x6c, 0x55, 0xe8, 0x3f, 0xf9, 0x7a, 0x1a, 0xef,
		0xfb, 0x3a, 0xf0, 0x0a, 0xdb, 0x22, 0xc6, 0xbb,
		0x08, 0xb3, 0xf4, 0x81, 0xe3, 0xaa, 0xa0, 0xf1,
		0xa0, 0x9e, 0x30, 0xed, 0x74, 0x1d, 0x8a, 0xe4,
		0xfc, 0xf5, 0xe0, 0x95, 0xd5, 0xd0, 0x0a, 0xf6,
		0x00, 0xdb, 0x18, 0xcb, 0x2c, 0x04, 0xb3, 0xed,
		0xd0, 0x3c, 0xc7, 0x44, 0xa2, 0x88, 0x8a, 0xe4,
		0x0c, 0xaa, 0x23, 0x29, 0x46, 0xc5, 0xe7, 0xe1,
	}
	bls12G1Pk   = "f" + hex.EncodeToString(bls12G1PkB)
	bls12G1PkBt = append([]byte{0xea, 0x01}, bls12G1PkB...)
	bls12G1Pkt  = "f" + hex.EncodeToString(bls12G1PkBt)

	// bls12 381 G2 sample public key
	bls12G2PkB = []byte{
		0x13, 0xe0, 0x2b, 0x60, 0x52, 0x71, 0x9f, 0x60,
		0x7d, 0xac, 0xd3, 0xa0, 0x88, 0x27, 0x4f, 0x65,
		0x59, 0x6b, 0xd0, 0xd0, 0x99, 0x20, 0xb6, 0x1a,
		0xb5, 0xda, 0x61, 0xbb, 0xdc, 0x7f, 0x50, 0x49,
		0x33, 0x4c, 0xf1, 0x12, 0x13, 0x94, 0x5d, 0x57,
		0xe5, 0xac, 0x7d, 0x05, 0x5d, 0x04, 0x2b, 0x7e,
		0x02, 0x4a, 0xa2, 0xb2, 0xf0, 0x8f, 0x0a, 0x91,
		0x26, 0x08, 0x05, 0x27, 0x2d, 0xc5, 0x10, 0x51,
		0xc6, 0xe4, 0x7a, 0xd4, 0xfa, 0x40, 0x3b, 0x02,
		0xb4, 0x51, 0x0b, 0x64, 0x7a, 0xe3, 0xd1, 0x77,
		0x0b, 0xac, 0x03, 0x26, 0xa8, 0x05, 0xbb, 0xef,
		0xd4, 0x80, 0x56, 0xc8, 0xc1, 0x21, 0xbd, 0xb8,
		0x06, 0x06, 0xc4, 0xa0, 0x2e, 0xa7, 0x34, 0xcc,
		0x32, 0xac, 0xd2, 0xb0, 0x2b, 0xc2, 0x8b, 0x99,
		0xcb, 0x3e, 0x28, 0x7e, 0x85, 0xa7, 0x63, 0xaf,
		0x26, 0x74, 0x92, 0xab, 0x57, 0x2e, 0x99, 0xab,
		0x3f, 0x37, 0x0d, 0x27, 0x5c, 0xec, 0x1d, 0xa1,
		0xaa, 0xa9, 0x07, 0x5f, 0xf0, 0x5f, 0x79, 0xbe,
		0x0c, 0xe5, 0xd5, 0x27, 0x72, 0x7d, 0x6e, 0x11,
		0x8c, 0xc9, 0xcd, 0xc6, 0xda, 0x2e, 0x35, 0x1a,
		0xad, 0xfd, 0x9b, 0xaa, 0x8c, 0xbd, 0xd3, 0xa7,
		0x6d, 0x42, 0x9a, 0x69, 0x51, 0x60, 0xd1, 0x2c,
		0x92, 0x3a, 0xc9, 0xcc, 0x3b, 0xac, 0xa2, 0x89,
		0xe1, 0x93, 0x54, 0x86, 0x08, 0xb8, 0x28, 0x01,
	}
	bls12G2Pk   = "f" + hex.EncodeToString(bls12G2PkB)
	bls12G2PkBt = append([]byte{0xeb, 0x01}, bls12G2PkB...)
	bls12G2Pkt  = "f" + hex.EncodeToString(bls12G2PkBt)
)

func TestSerialisePublicKey(t *testing.T) {
	cs := []struct {
		Description string
		OutBase     string
		Key         string
		Expected    string
		Error       error
	}{
		{
			"invalid key, with valid key type",
			"z",
			"0xe701ff42ea",
			"",
			fmt.Errorf("invalid public key format, '[11111111 1000010 11101010]'"),
		},
		{
			"invalid key type, with invalid key",
			"z",
			"0xeeff42ea",
			"",
			fmt.Errorf("unsupported public key type '10BFEE'"),
		},
		{
			"invalid encoding type, with valid key",
			"p",
			secPkt,
			"",
			fmt.Errorf("selected encoding not supported"),
		},
		{
			"valid key, no key type defined",
			"z",
			secPk,
			"",
			fmt.Errorf("unsupported public key type '4'"),
		},
		{
			"valid key, with base58 bitcoin encoding",
			"z",
			secPkt,
			"zQ3shPyZJnxZK4Bwyx9QsaksNKDYTPmpwPvGSjMYVHoXHeEgB",
			nil,
		},
		{
			"valid key, with traditional hex encoding",
			"0x",
			secPkt,
			"fe70102261c55675e55ff25edb50b345cfb3a3f35f60712d251cbaaab97bd50054c6ebc",
			nil,
		},
		{
			"valid secp256k1 key, with multiencoding hex encoding",
			"f",
			secPkt,
			"fe70102261c55675e55ff25edb50b345cfb3a3f35f60712d251cbaaab97bd50054c6ebc",
			nil,
		},
		{
			"valid bls12-381 g1 key, with multiencoding hex encoding",
			"f",
			bls12G1Pkt,
			"fea0197f1d3a73197d7942695638c4fa9ac0fc3688c4f9774b905a14e3a3f171bac586c55e83ff97a1aeffb3af00adb22c6bb",
			nil,
		},
		{
			"valid bls12-381 g1 key, with base58 bitcoin encoding",
			"z",
			bls12G1Pkt,
			"z3tEFUdV4D3tCMG6Fr1deVvt32DCS1Y4SxDGoELedXaMUdTdr5FfZvBnbK9bWMhAGj3RHk",
			nil,
		},
		{
			"valid bls12-381 g1 key, with no key type",
			"f",
			bls12G1Pk,
			"",
			fmt.Errorf("unsupported public key type '17'"),
		},
		{
			"valid bls12-381 g2 key, with multiencoding hex encoding",
			"f",
			bls12G2Pkt,
			"feb0193e02b6052719f607dacd3a088274f65596bd0d09920b61ab5da61bbdc7f5049334cf11213945d57e5ac7d055d042b7e024aa2b2f08f0a91260805272dc51051c6e47ad4fa403b02b4510b647ae3d1770bac0326a805bbefd48056c8c121bdb8",
			nil,
		},
		{
			"valid bls12-381 g2 key, with base58 bitcoin encoding",
			"z",
			bls12G2Pkt,
			"zUC77n3BqSWuoGMY7ut91NDoWzpithCd4GwPLAnv9fc7drWY4wBTvMX1y9eGSAuiBpktqGAocND2KXdu1HqNgrd6i3vCZKCLqZ3nQFaEA2FpTs7ZEChRpWReLvYyXNYUHvQjyKd",
			nil,
		},
		{
			"valid bls12-381 g2 key, with no key type",
			"f",
			bls12G2Pk,
			"",
			fmt.Errorf("unsupported public key type '13'"),
		},
	}

	for _, c := range cs {
		cpk, err := SerializePublicKey(c.Key, c.OutBase)

		require.Equal(t, c.Expected, cpk, c.Description)

		if c.Error != nil {
			require.EqualError(t, err, c.Error.Error(), c.Description)
			continue
		}

		require.NoError(t, err, c.Description)
	}
}

func TestDeserialisePublicKey(t *testing.T) {
	cs := []struct {
		Description string
		Key         string
		OutBase     string
		Expected    string
		Error       error
	}{
		{
			"valid key with valid encoding type '0x'",
			"0xe70102261c55675e55ff25edb50b345cfb3a3f35f60712d251cbaaab97bd50054c6ebc",
			"f",
			secPkt,
			nil,
		},
		{
			"valid key with valid encoding type 'f'",
			"fe70102261c55675e55ff25edb50b345cfb3a3f35f60712d251cbaaab97bd50054c6ebc",
			"f",
			secPkt,
			nil,
		},
		{
			"valid key with valid encoding type 'z'",
			"zQ3shPyZJnxZK4Bwyx9QsaksNKDYTPmpwPvGSjMYVHoXHeEgB",
			"f",
			secPkt,
			nil,
		},
		{
			"valid key with mismatched encoding type 'f' instead of 'z'",
			"fQ3shPyZJnxZK4Bwyx9QsaksNKDYTPmpwPvGSjMYVHoXHeEgB",
			"f",
			"",
			fmt.Errorf("encoding/hex: invalid byte: U+0051 'Q'"),
		},
		{
			"valid key with no encoding type, in base58 encoding",
			"Q3shPyZJnxZK4Bwyx9QsaksNKDYTPmpwPvGSjMYVHoXHeEgB",
			"f",
			"",
			fmt.Errorf("selected encoding not supported"),
		},
		{
			"valid bls12-381 g1 key encoding type 'z'",
			"z3tEFUdV4D3tCMG6Fr1deVvt32DCS1Y4SxDGoELedXaMUdTdr5FfZvBnbK9bWMhAGj3RHk",
			"f",
			bls12G1Pkt,
			nil,
		},
		{
			"valid bls12-381 g1 key encoding type 'f'",
			"fea0197f1d3a73197d7942695638c4fa9ac0fc3688c4f9774b905a14e3a3f171bac586c55e83ff97a1aeffb3af00adb22c6bb",
			"f",
			bls12G1Pkt,
			nil,
		},
		{
			"valid bls12-381 g1 key encoding type '0x'",
			"0xea0197f1d3a73197d7942695638c4fa9ac0fc3688c4f9774b905a14e3a3f171bac586c55e83ff97a1aeffb3af00adb22c6bb",
			"f",
			bls12G1Pkt,
			nil,
		},
		{
			"valid bls12-381 g2 key encoding type 'z'",
			"zUC77n3BqSWuoGMY7ut91NDoWzpithCd4GwPLAnv9fc7drWY4wBTvMX1y9eGSAuiBpktqGAocND2KXdu1HqNgrd6i3vCZKCLqZ3nQFaEA2FpTs7ZEChRpWReLvYyXNYUHvQjyKd",
			"f",
			bls12G2Pkt,
			nil,
		},
		{
			"valid bls12-381 g2 key encoding type 'f'",
			"feb0193e02b6052719f607dacd3a088274f65596bd0d09920b61ab5da61bbdc7f5049334cf11213945d57e5ac7d055d042b7e024aa2b2f08f0a91260805272dc51051c6e47ad4fa403b02b4510b647ae3d1770bac0326a805bbefd48056c8c121bdb8",
			"f",
			bls12G2Pkt,
			nil,
		},
		{
			"valid bls12-381 g2 key encoding type '0x'",
			"0xeb0193e02b6052719f607dacd3a088274f65596bd0d09920b61ab5da61bbdc7f5049334cf11213945d57e5ac7d055d042b7e024aa2b2f08f0a91260805272dc51051c6e47ad4fa403b02b4510b647ae3d1770bac0326a805bbefd48056c8c121bdb8",
			"f",
			bls12G2Pkt,
			nil,
		},
	}

	for _, c := range cs {
		key, err := DeserializePublicKey(c.Key, c.OutBase)

		require.Exactly(t, c.Expected, key, c.Description)

		if c.Error != nil {
			require.EqualError(t, err, c.Error.Error(), c.Description)
			continue
		}

		require.NoError(t, err, c.Description)
	}
}

func TestSerializeLegacyKey(t *testing.T) {

	key := "0x04deaafa03e3a646e54a36ec3f6968c1d3686847d88420f00c0ab6ee517ee1893398fca28aacd2af74f2654738c21d10bad3d88dc64201ebe0de5cf1e313970d3d"
	expected := "zQ3shudJrBctPznsRLvbsCtvZFTdi3b34uzYDuqE9Wq9m9T1C"

	res, err := SerializeLegacyKey(key)

	require.NoError(t, err)
	require.Equal(t, expected, res)

	wrongKey := key[3:]

	_, err = SerializeLegacyKey(wrongKey)

	require.Error(t, err)
}