From 28121c5e62f15339033b5f8f921c6715234813ed Mon Sep 17 00:00:00 2001
From: Neal Xu <imxuneal@gmail.com>
Date: Fri, 23 Feb 2024 18:17:05 +0800
Subject: [PATCH] feat: retur all k1 chain_types (#67)

---
 token-core/tcx/src/handler.rs       | 23 ++++++++++++++++++++++-
 token-core/tcx/tests/import_test.rs | 13 ++++++++++++-
 2 files changed, 34 insertions(+), 2 deletions(-)

diff --git a/token-core/tcx/src/handler.rs b/token-core/tcx/src/handler.rs
index 39acfa14..fab8b342 100644
--- a/token-core/tcx/src/handler.rs
+++ b/token-core/tcx/src/handler.rs
@@ -244,6 +244,16 @@ fn decode_private_key(private_key: &str) -> Result<DecodedPrivateKey> {
         if let Ok(decoded_data) = decoded {
             if decoded_data.len() == 32 {
                 private_key_bytes = decoded_data;
+                chain_types = vec![
+                    "BITCOIN".to_string(),
+                    "ETHEREUM".to_string(),
+                    "BITCOINCASH".to_string(),
+                    "LITECOIN".to_string(),
+                    "EOS".to_string(),
+                    "TRON".to_string(),
+                    "FILECOIN".to_string(),
+                ];
+                curve = CurveType::SECP256k1;
             } else if decoded_data.len() == 64 {
                 let sr25519_key = Sr25519PrivateKey::from_slice(&decoded_data)?;
                 private_key_bytes = sr25519_key.to_bytes();
@@ -1285,7 +1295,18 @@ mod tests {
 
         let private_key = "0x43fe394358d14f2e096f4efe80894b4e51a3fdcb73c06b77e937b80deb8c746b";
         let decoded = decode_private_key(&private_key).unwrap();
-        assert_eq!(decoded.chain_types, Vec::<String>::new());
+        assert_eq!(
+            decoded.chain_types,
+            vec![
+                "BITCOIN".to_string(),
+                "ETHEREUM".to_string(),
+                "BITCOINCASH".to_string(),
+                "LITECOIN".to_string(),
+                "EOS".to_string(),
+                "TRON".to_string(),
+                "FILECOIN".to_string(),
+            ]
+        );
         assert_eq!(decoded.curve, CurveType::SECP256k1);
         assert_eq!(decoded.network, "".to_string());
         assert_eq!(decoded.source, Source::Private);
diff --git a/token-core/tcx/tests/import_test.rs b/token-core/tcx/tests/import_test.rs
index 80262183..d5c88910 100644
--- a/token-core/tcx/tests/import_test.rs
+++ b/token-core/tcx/tests/import_test.rs
@@ -1096,7 +1096,18 @@ pub fn test_import_hex_private_key() {
         let ret = call_api("import_private_key", param).unwrap();
         let import_result: ImportPrivateKeyResult =
             ImportPrivateKeyResult::decode(ret.as_slice()).unwrap();
-        assert_eq!(Vec::<String>::new(), import_result.identified_chain_types);
+        assert_eq!(
+            vec![
+                "BITCOIN".to_string(),
+                "ETHEREUM".to_string(),
+                "BITCOINCASH".to_string(),
+                "LITECOIN".to_string(),
+                "EOS".to_string(),
+                "TRON".to_string(),
+                "FILECOIN".to_string(),
+            ],
+            import_result.identified_chain_types
+        );
         assert_eq!("secp256k1", import_result.identified_curve);
         assert_eq!("", import_result.identified_network);
         assert_eq!("PRIVATE", import_result.source);