package main

import (
	"github.com/guggero/chantools/btc"
	"github.com/guggero/chantools/lnd"
	"os"
	"testing"

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

const (
	testPath        = "m/123'/45'/67'/8/9"
	keyContent      = "bcrt1qnl5qfvpfcmj7y56nugpermluu46x79sfz0ku70"
	keyContentBIP39 = "bcrt1q3pae32m7jdqm5ulf80yc3n59xy4s4xm5a28ekr"
)

func TestDeriveKey(t *testing.T) {
	h := newHarness(t)

	// Derive a specific key from the serialized root key.
	derive := &deriveKeyCommand{
		Path:    testPath,
		rootKey: &rootKey{RootKey: rootKeyAezeed},
	}

	err := derive.Execute(nil, nil)
	require.NoError(t, err)

	h.assertLogContains(keyContent)
}

func TestDeriveKeyAezeedNoPassphrase(t *testing.T) {
	h := newHarness(t)

	// Derive a specific key from the serialized root key.
	derive := &deriveKeyCommand{
		Path:    testPath,
		rootKey: &rootKey{},
	}

	err := os.Setenv(lnd.MnemonicEnvName, seedAezeedNoPassphrase)
	require.NoError(t, err)
	err = os.Setenv(lnd.PassphraseEnvName, "-")
	require.NoError(t, err)

	err = derive.Execute(nil, nil)
	require.NoError(t, err)

	h.assertLogContains(keyContent)
}

func TestDeriveKeyAezeedWithPassphrase(t *testing.T) {
	h := newHarness(t)

	// Derive a specific key from the serialized root key.
	derive := &deriveKeyCommand{
		Path:    testPath,
		rootKey: &rootKey{},
	}

	err := os.Setenv(lnd.MnemonicEnvName, seedAezeedWithPassphrase)
	require.NoError(t, err)
	err = os.Setenv(lnd.PassphraseEnvName, testPassPhrase)
	require.NoError(t, err)

	err = derive.Execute(nil, nil)
	require.NoError(t, err)

	h.assertLogContains(keyContent)
}

func TestDeriveKeySeedBip39(t *testing.T) {
	h := newHarness(t)

	// Derive a specific key from the serialized root key.
	derive := &deriveKeyCommand{
		Path:    testPath,
		rootKey: &rootKey{BIP39: true},
	}

	err := os.Setenv(btc.BIP39MnemonicEnvName, seedBip39)
	require.NoError(t, err)
	err = os.Setenv(btc.BIP39PassphraseEnvName, "-")
	require.NoError(t, err)

	err = derive.Execute(nil, nil)
	require.NoError(t, err)

	h.assertLogContains(keyContentBIP39)
}