feat: extended master key support multi chainnet.

Author: Zhangguiguang

core/wallet/mnemonic.go

@@ -1,6 +1,8 @@
 package wallet
 
 import (
+	"errors"
+
 	"github.com/btcsuite/btcd/btcutil/hdkeychain"
 	"github.com/btcsuite/btcd/chaincfg"
 	"github.com/tyler-smith/go-bip39"
@@ -20,12 +22,33 @@ func IsValidMnemonic(mnemonic string) bool {
 	return err == nil
 }
 
-func ExtendMasterKey(mnemonic string) (string, error) {
+// ExtendMasterKey derives a master key from the given mnemonic and chain network identifier.
+//
+// Parameters:
+//   - mnemonic: A string representing the mnemonic phrase used to generate the seed.
+//   - chainnet: The blockchain network, which must be either
+//     "mainnet", "testnet", "signet", "simnet" or "regtest".
+func ExtendMasterKey(mnemonic string, chainnet string) (string, error) {
+	var net chaincfg.Params
+	switch chainnet {
+	case "mainnet", "bitcoin":
+		net = chaincfg.MainNetParams
+	case "testnet", "testnet3":
+		net = chaincfg.TestNet3Params
+	case "signet":
+		net = chaincfg.SigNetParams
+	case "simnet":
+		net = chaincfg.SimNetParams
+	case "regtest":
+		net = chaincfg.RegressionNetParams
+	default:
+		return "", errors.New("invalid chainnet")
+	}
 	seed, err := bip39.NewSeedWithErrorChecking(mnemonic, "")
 	if err != nil {
 		return "", err
 	}
-	masterKey, err := hdkeychain.NewMaster(seed, &chaincfg.MainNetParams)
+	masterKey, err := hdkeychain.NewMaster(seed, &net)
 	if err != nil {
 		return "", err
 	}

core/wallet/mnemonic_test.go

@@ -4,7 +4,6 @@ import (
 	"testing"
 
 	"github.com/ChainSafe/go-schnorrkel"
-	"github.com/stretchr/testify/require"
 	"github.com/tyler-smith/go-bip39"
 )
 
@@ -42,8 +41,42 @@ func TestSeed(t *testing.T) {
 
 func TestExtendMasterKey(t *testing.T) {
 	mn := "unaware oxygen allow method allow property predict various slice travel please priority"
-	extendedMasterKey, err := ExtendMasterKey(mn)
-	require.NoError(t, err)
-	want := "xprv9s21ZrQH143K2gM869SUAeSEGDM81UnyWG2tM7ZygYwN3PkveZmY799G6q7zkQYSQCMn7M7AMpP8Z6etKUoZo8x4hH1WVcKBCMZnVTNM8AD"
-	require.Equal(t, want, extendedMasterKey)
+	tests := []struct {
+		chainnet string
+		want     string
+		wantErr  bool
+	}{
+		{
+			chainnet: "mainnet",
+			want:     "xprv9s21ZrQH143K2gM869SUAeSEGDM81UnyWG2tM7ZygYwN3PkveZmY799G6q7zkQYSQCMn7M7AMpP8Z6etKUoZo8x4hH1WVcKBCMZnVTNM8AD",
+		},
+		{
+			chainnet: "testnet",
+			want:     "tprv8ZgxMBicQKsPdVaekiHyLJ4DaLmLEzpyqox1DXzSAXRqpzW1dw7HctWi21HekmvkmdtZ7SivXAxw1xCdSh9WcCDfDvDp9y3E7TKCw7HgLYY",
+		},
+		{
+			chainnet: "signet",
+			want:     "tprv8ZgxMBicQKsPdVaekiHyLJ4DaLmLEzpyqox1DXzSAXRqpzW1dw7HctWi21HekmvkmdtZ7SivXAxw1xCdSh9WcCDfDvDp9y3E7TKCw7HgLYY",
+		},
+		{
+			chainnet: "simnet",
+			want:     "sprv8Erh3X3hFeKunCPXvMWM8CxiQNctJNqUvhRnS96YnX3xmtgnPGwizprZzoL4ksGqMzmkMy8N4WcP8fb4izjVoxY4MaXPa4DjqjFZYXFdKru",
+		},
+		{
+			chainnet: "regtest",
+			want:     "tprv8ZgxMBicQKsPdVaekiHyLJ4DaLmLEzpyqox1DXzSAXRqpzW1dw7HctWi21HekmvkmdtZ7SivXAxw1xCdSh9WcCDfDvDp9y3E7TKCw7HgLYY",
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.chainnet, func(t *testing.T) {
+			got, err := ExtendMasterKey(mn, tt.chainnet)
+			if (err != nil) != tt.wantErr {
+				t.Errorf("ExtendMasterKey() error = %v, wantErr %v", err, tt.wantErr)
+				return
+			}
+			if got != tt.want {
+				t.Errorf("ExtendMasterKey() = %v, want %v", got, tt.want)
+			}
+		})
+	}
 }