package wallet_test

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

	"github.com/stretchr/testify/assert"
	"github.com/textileio/go-textile/keypair"
	. "github.com/textileio/go-textile/wallet"
)

func ExampleDeriveFromPath() {
	seed, _ := hex.DecodeString("000102030405060708090a0b0c0d0e0f")
	key, err := DeriveForPath(TextilePrimaryAccountPath, seed)
	if err != nil {
		panic(err)
	}

	kp, err := keypair.FromRawSeed(key.RawSeed())
	if err != nil {
		panic(err)
	}

	fmt.Println(kp.Seed())
	fmt.Println(kp.Address())

	// Output:
	// SX2XbMf8FHtYq6xZ4vUuZhmKTezrn67XKYRn1Fc7ZLcN2VXE
	// P7bFoVXV3dALSLwYAnyzzVvHkfTDwbpaLwA9PL9659hBqWAX
}

func ExampleDeriveMultipleKeys() {
	seed, _ := hex.DecodeString("000102030405060708090a0b0c0d0e0f")

	for i := 0; i < 10; i++ {
		path := fmt.Sprintf(TextileAccountPathFormat, i)
		key, err := DeriveForPath(path, seed)
		if err != nil {
			panic(err)
		}

		kp, err := keypair.FromRawSeed(key.RawSeed())
		if err != nil {
			panic(err)
		}

		fmt.Println(path, kp.Seed(), kp.Address())
	}

	// Output:
	// m/44'/406'/0' SX2XbMf8FHtYq6xZ4vUuZhmKTezrn67XKYRn1Fc7ZLcN2VXE P7bFoVXV3dALSLwYAnyzzVvHkfTDwbpaLwA9PL9659hBqWAX
	// m/44'/406'/1' SWBR6LyJUDz2Kjx6vdsQci8N8CMiUCm1NB6HRkxfQ6qSMef3 P9UXU6xtgnEYJ51J2x5q1R1JEBMoYr6BuFy9vM8wkmvLV3JL
	// m/44'/406'/2' SUs72keJyyqYgSuYAZWM5RVDwSNmFEDFPq9X2yHNc88LTsPF P8wnHf2UCbMRPzNSTY28UyC8WjJEtPC67WuDUcsntCM8ubCx
	// m/44'/406'/3' SWW631fBve4zRvCd2Uv1UQk1EYF9muFAAT4xmyApiWzXNCqL P5kA2UdGNf6GJ7LJBiM3VKUyD59kF9t46j742Go6SykxTnQg
	// m/44'/406'/4' SXC1gk5X7WNyNhm6yeft1n6P5gWarBibkJFm8RR9uzYsR6iT P89CyiEf234U16yrjpwEVzW36PrEaeTjkyJnoqSujdPCoLDA
	// m/44'/406'/5' SVTepyrL9oad1wBFEGWsPHtFbN3JB6NSs2rfbbcaoYH1vCqS P65pF4U9LNSJ1pL8AyV4JmFbmYQsdJ2CoDKNKeefxQnGnKW6
	// m/44'/406'/6' SW9JWH8MzRhS8hqvimc3ES5QdHQUThSav8NprZ9i19S9tXLt P6nmNBVUPRjLCtWqUVwbbnaGX2yFSPZ6sdPWyvJgXunomwk9
	// m/44'/406'/7' SX3wzcqcZmYF8fvXakvCiUYsGA6g8ZnkpJrD6jsxTaKh9m3M P43y96cDLLmbjip4dpBkhkLRSz7SxC3GBBz1eEbAA2m1ydu9
	// m/44'/406'/8' SSxrsRHUbw8f1WrMzUk44os1Bv9DLFHQGsk9pEeosJt6u1kc P4WYDTUi4oosjxwYEfLnwDF52J7XCzw6fLD4Dx2WxE67px3V
	// m/44'/406'/9' SXm1Ttvq7tHgfaNAC2waYQ8H9KASqbnwzeNbuC2zctv4DsM8 P6nFLE44WaCa4QPEb5ygMkdWQxLHuf1rkSo7LFHTY1Nq4uQA
}

func ExampleDeriveMultipleKeysFaster() {
	seed, _ := hex.DecodeString("000102030405060708090a0b0c0d0e0f")
	mainKey, err := DeriveForPath(TextileAccountPrefix, seed)
	if err != nil {
		panic(err)
	}

	for i := uint32(0); i < 10; i++ {
		key, err := mainKey.Derive(FirstHardenedIndex + i)
		if err != nil {
			panic(err)
		}

		kp, err := keypair.FromRawSeed(key.RawSeed())
		if err != nil {
			panic(err)
		}

		fmt.Println(fmt.Sprintf(TextileAccountPathFormat, i), kp.Seed(), kp.Address())
	}

	// Output:
	// m/44'/406'/0' SX2XbMf8FHtYq6xZ4vUuZhmKTezrn67XKYRn1Fc7ZLcN2VXE P7bFoVXV3dALSLwYAnyzzVvHkfTDwbpaLwA9PL9659hBqWAX
	// m/44'/406'/1' SWBR6LyJUDz2Kjx6vdsQci8N8CMiUCm1NB6HRkxfQ6qSMef3 P9UXU6xtgnEYJ51J2x5q1R1JEBMoYr6BuFy9vM8wkmvLV3JL
	// m/44'/406'/2' SUs72keJyyqYgSuYAZWM5RVDwSNmFEDFPq9X2yHNc88LTsPF P8wnHf2UCbMRPzNSTY28UyC8WjJEtPC67WuDUcsntCM8ubCx
	// m/44'/406'/3' SWW631fBve4zRvCd2Uv1UQk1EYF9muFAAT4xmyApiWzXNCqL P5kA2UdGNf6GJ7LJBiM3VKUyD59kF9t46j742Go6SykxTnQg
	// m/44'/406'/4' SXC1gk5X7WNyNhm6yeft1n6P5gWarBibkJFm8RR9uzYsR6iT P89CyiEf234U16yrjpwEVzW36PrEaeTjkyJnoqSujdPCoLDA
	// m/44'/406'/5' SVTepyrL9oad1wBFEGWsPHtFbN3JB6NSs2rfbbcaoYH1vCqS P65pF4U9LNSJ1pL8AyV4JmFbmYQsdJ2CoDKNKeefxQnGnKW6
	// m/44'/406'/6' SW9JWH8MzRhS8hqvimc3ES5QdHQUThSav8NprZ9i19S9tXLt P6nmNBVUPRjLCtWqUVwbbnaGX2yFSPZ6sdPWyvJgXunomwk9
	// m/44'/406'/7' SX3wzcqcZmYF8fvXakvCiUYsGA6g8ZnkpJrD6jsxTaKh9m3M P43y96cDLLmbjip4dpBkhkLRSz7SxC3GBBz1eEbAA2m1ydu9
	// m/44'/406'/8' SSxrsRHUbw8f1WrMzUk44os1Bv9DLFHQGsk9pEeosJt6u1kc P4WYDTUi4oosjxwYEfLnwDF52J7XCzw6fLD4Dx2WxE67px3V
	// m/44'/406'/9' SXm1Ttvq7tHgfaNAC2waYQ8H9KASqbnwzeNbuC2zctv4DsM8 P6nFLE44WaCa4QPEb5ygMkdWQxLHuf1rkSo7LFHTY1Nq4uQA
}

func BenchmarkDerive(b *testing.B) {
	seed, _ := hex.DecodeString("000102030405060708090a0b0c0d0e0f")

	for i := 0; i < b.N; i++ {
		_, err := DeriveForPath(TextilePrimaryAccountPath, seed)
		if err != nil {
			panic(err)
		}
	}
}

func BenchmarkDeriveFast(b *testing.B) {
	seed, _ := hex.DecodeString("000102030405060708090a0b0c0d0e0f")
	mainKey, err := DeriveForPath(TextileAccountPrefix, seed)
	if err != nil {
		panic(err)
	}

	for i := 0; i < b.N; i++ {
		_, err := mainKey.Derive(FirstHardenedIndex)
		if err != nil {
			panic(err)
		}
	}
}

func TestIsValidPath(t *testing.T) {
	assert.True(t, IsValidPath("m/0'"))
	assert.True(t, IsValidPath("m/0'/100'"))
	assert.True(t, IsValidPath("m/0'/100'/200'"))
	assert.True(t, IsValidPath("m/0'/100'/200'/300'"))

	assert.False(t, IsValidPath("foobar"))
	assert.False(t, IsValidPath("m"))                           // Master key only
	assert.False(t, IsValidPath("m/0"))                         // Missing '
	assert.False(t, IsValidPath("m/0'/"))                       // Trailing slash
	assert.False(t, IsValidPath("m/0'/893478327492379497823'")) // Overflow
}

// https://github.com/satoshilabs/slips/blob/master/slip-0010.md#test-vector-1-for-ed25519
func TestDeriveVector1(t *testing.T) {
	seed, err := hex.DecodeString("000102030405060708090a0b0c0d0e0f")
	assert.NoError(t, err)

	key, err := NewMasterKey(seed)
	assert.NoError(t, err)
	assert.Equal(t, "2b4be7f19ee27bbf30c667b642d5f4aa69fd169872f8fc3059c08ebae2eb19e7", hex.EncodeToString(key.Key))
	assert.Equal(t, "90046a93de5380a72b5e45010748567d5ea02bbf6522f979e05c0d8d8ca9fffb", hex.EncodeToString(key.ChainCode))
	publicKey, err := key.PublicKey()
	assert.NoError(t, err)
	assert.Equal(t, "00a4b2856bfec510abab89753fac1ac0e1112364e7d250545963f135f2a33188ed", hex.EncodeToString(append([]byte{0x0}, publicKey...)))

	tests := []struct {
		Path       string
		ChainCode  string
		PrivateKey string
		PublicKey  string
	}{
		{
			Path:       "m/0'",
			ChainCode:  "8b59aa11380b624e81507a27fedda59fea6d0b779a778918a2fd3590e16e9c69",
			PrivateKey: "68e0fe46dfb67e368c75379acec591dad19df3cde26e63b93a8e704f1dade7a3",
			PublicKey:  "008c8a13df77a28f3445213a0f432fde644acaa215fc72dcdf300d5efaa85d350c",
		},
		{
			Path:       "m/0'/1'",
			ChainCode:  "a320425f77d1b5c2505a6b1b27382b37368ee640e3557c315416801243552f14",
			PrivateKey: "b1d0bad404bf35da785a64ca1ac54b2617211d2777696fbffaf208f746ae84f2",
			PublicKey:  "001932a5270f335bed617d5b935c80aedb1a35bd9fc1e31acafd5372c30f5c1187",
		},
		{
			Path:       "m/0'/1'/2'",
			ChainCode:  "2e69929e00b5ab250f49c3fb1c12f252de4fed2c1db88387094a0f8c4c9ccd6c",
			PrivateKey: "92a5b23c0b8a99e37d07df3fb9966917f5d06e02ddbd909c7e184371463e9fc9",
			PublicKey:  "00ae98736566d30ed0e9d2f4486a64bc95740d89c7db33f52121f8ea8f76ff0fc1",
		},
		{
			Path:       "m/0'/1'/2'/2'",
			ChainCode:  "8f6d87f93d750e0efccda017d662a1b31a266e4a6f5993b15f5c1f07f74dd5cc",
			PrivateKey: "30d1dc7e5fc04c31219ab25a27ae00b50f6fd66622f6e9c913253d6511d1e662",
			PublicKey:  "008abae2d66361c879b900d204ad2cc4984fa2aa344dd7ddc46007329ac76c429c",
		},
		{
			Path:       "m/0'/1'/2'/2'/1000000000'",
			ChainCode:  "68789923a0cac2cd5a29172a475fe9e0fb14cd6adb5ad98a3fa70333e7afa230",
			PrivateKey: "8f94d394a8e8fd6b1bc2f3f49f5c47e385281d5c17e65324b0f62483e37e8793",
			PublicKey:  "003c24da049451555d51a7014a37337aa4e12d41e485abccfa46b47dfb2af54b7a",
		},
	}

	for _, test := range tests {
		key, err = DeriveForPath(test.Path, seed)
		assert.NoError(t, err)
		assert.Equal(t, test.PrivateKey, hex.EncodeToString(key.Key))
		assert.Equal(t, test.ChainCode, hex.EncodeToString(key.ChainCode))
		publicKey, err := key.PublicKey()
		assert.NoError(t, err)
		assert.Equal(t, test.PublicKey, hex.EncodeToString(append([]byte{0x0}, publicKey...)))
	}
}

// https://github.com/satoshilabs/slips/blob/master/slip-0010.md#test-vector-2-for-ed25519
func TestDeriveVector2(t *testing.T) {
	seed, err := hex.DecodeString("fffcf9f6f3f0edeae7e4e1dedbd8d5d2cfccc9c6c3c0bdbab7b4b1aeaba8a5a29f9c999693908d8a8784817e7b7875726f6c696663605d5a5754514e4b484542")
	assert.NoError(t, err)

	key, err := NewMasterKey(seed)
	assert.NoError(t, err)
	assert.Equal(t, "171cb88b1b3c1db25add599712e36245d75bc65a1a5c9e18d76f9f2b1eab4012", hex.EncodeToString(key.Key))
	assert.Equal(t, "ef70a74db9c3a5af931b5fe73ed8e1a53464133654fd55e7a66f8570b8e33c3b", hex.EncodeToString(key.ChainCode))
	publicKey, err := key.PublicKey()
	assert.NoError(t, err)
	assert.Equal(t, "008fe9693f8fa62a4305a140b9764c5ee01e455963744fe18204b4fb948249308a", hex.EncodeToString(append([]byte{0x0}, publicKey...)))

	tests := []struct {
		Path       string
		ChainCode  string
		PrivateKey string
		PublicKey  string
	}{
		{
			Path:       "m/0'",
			ChainCode:  "0b78a3226f915c082bf118f83618a618ab6dec793752624cbeb622acb562862d",
			PrivateKey: "1559eb2bbec5790b0c65d8693e4d0875b1747f4970ae8b650486ed7470845635",
			PublicKey:  "0086fab68dcb57aa196c77c5f264f215a112c22a912c10d123b0d03c3c28ef1037",
		},
		{
			Path:       "m/0'/2147483647'",
			ChainCode:  "138f0b2551bcafeca6ff2aa88ba8ed0ed8de070841f0c4ef0165df8181eaad7f",
			PrivateKey: "ea4f5bfe8694d8bb74b7b59404632fd5968b774ed545e810de9c32a4fb4192f4",
			PublicKey:  "005ba3b9ac6e90e83effcd25ac4e58a1365a9e35a3d3ae5eb07b9e4d90bcf7506d",
		},
		{
			Path:       "m/0'/2147483647'/1'",
			ChainCode:  "73bd9fff1cfbde33a1b846c27085f711c0fe2d66fd32e139d3ebc28e5a4a6b90",
			PrivateKey: "3757c7577170179c7868353ada796c839135b3d30554bbb74a4b1e4a5a58505c",
			PublicKey:  "002e66aa57069c86cc18249aecf5cb5a9cebbfd6fadeab056254763874a9352b45",
		},
		{
			Path:       "m/0'/2147483647'/1'/2147483646'",
			ChainCode:  "0902fe8a29f9140480a00ef244bd183e8a13288e4412d8389d140aac1794825a",
			PrivateKey: "5837736c89570de861ebc173b1086da4f505d4adb387c6a1b1342d5e4ac9ec72",
			PublicKey:  "00e33c0f7d81d843c572275f287498e8d408654fdf0d1e065b84e2e6f157aab09b",
		},
		{
			Path:       "m/0'/2147483647'/1'/2147483646'/2'",
			ChainCode:  "5d70af781f3a37b829f0d060924d5e960bdc02e85423494afc0b1a41bbe196d4",
			PrivateKey: "551d333177df541ad876a60ea71f00447931c0a9da16f227c11ea080d7391b8d",
			PublicKey:  "0047150c75db263559a70d5778bf36abbab30fb061ad69f69ece61a72b0cfa4fc0",
		},
	}

	for _, test := range tests {
		key, err = DeriveForPath(test.Path, seed)
		assert.NoError(t, err)
		assert.Equal(t, test.PrivateKey, hex.EncodeToString(key.Key))
		assert.Equal(t, test.ChainCode, hex.EncodeToString(key.ChainCode))
		publicKey, err := key.PublicKey()
		assert.NoError(t, err)
		assert.Equal(t, test.PublicKey, hex.EncodeToString(append([]byte{0x0}, publicKey...)))
	}
}