From 4143874df0b31db319cd2f304adcc2aa1d5346ee Mon Sep 17 00:00:00 2001
From: mollkeith <mollkeith.elastos@gmail.com>
Date: Mon, 17 Feb 2025 13:50:58 +0800
Subject: [PATCH] support parse btc keystore

---
 app/keystore-generator/main.go | 21 ++++++++++++++++-----
 1 file changed, 16 insertions(+), 5 deletions(-)

diff --git a/app/keystore-generator/main.go b/app/keystore-generator/main.go
index e1bc9ef..7bf15c3 100644
--- a/app/keystore-generator/main.go
+++ b/app/keystore-generator/main.go
@@ -27,12 +27,23 @@ func main() {
 	flag.Parse()
 
 	if *fileToParse != "" {
-		privateKey, err := eth.ParseKeystore(*fileToParse, *password)
-		if err != nil {
-			fmt.Println("Error:", err)
-			os.Exit(1)
+		switch *chain {
+		case "eth":
+			privateKey, err := eth.ParseKeystore(*fileToParse, *password)
+			if err != nil {
+				fmt.Println("Error:", err)
+				os.Exit(1)
+			}
+			fmt.Println("Private key:", privateKey)
+		case "btc":
+			privateKey, err := btc.ParseKeystore(*fileToParse, *password)
+			if err != nil {
+				fmt.Println("Error:", err)
+				os.Exit(1)
+			}
+			fmt.Println("Private key:", privateKey)
 		}
-		fmt.Println("Private key:", privateKey)
+
 	} else {
 		switch *chain {
 		case "eth":