AccountKind helpers

aspectron · Nov 2, 2023 · ca0e580 · ca0e580
1 parent 8c65f48
commit ca0e580
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 1 deletion.
diff --git a/cli/src/wizards/import.rs b/cli/src/wizards/import.rs
@@ -41,7 +41,7 @@ pub async fn ask(term: &Arc<Terminal>) -> Result<Vec<String>> {
 pub(crate) async fn import_with_mnemonic(ctx: &Arc<KaspaCli>, account_kind: AccountKind, additional_xpubs: &[String]) -> Result<()> {
     let wallet = ctx.wallet();
 
-    let is_legacy = matches!(account_kind, AccountKind::Legacy);
+    let is_legacy = account_kind.is_legacy();
 
     if !wallet.is_open() {
         return Err(Error::WalletIsNotOpen);

diff --git a/wallet/core/src/runtime/account/kind.rs b/wallet/core/src/runtime/account/kind.rs
@@ -21,6 +21,32 @@ u8_try_from! {
     }
 }
 
+impl AccountKind {
+    pub fn is_legacy(&self) -> bool {
+        matches!(self, AccountKind::Legacy)
+    }
+
+    pub fn is_bip32(&self) -> bool {
+        matches!(self, AccountKind::Bip32)
+    }
+
+    pub fn is_multisig(&self) -> bool {
+        matches!(self, AccountKind::MultiSig)
+    }
+
+    pub fn is_keypair(&self) -> bool {
+        matches!(self, AccountKind::Keypair)
+    }
+
+    pub fn is_hardware(&self) -> bool {
+        matches!(self, AccountKind::Hardware)
+    }
+
+    pub fn is_resident(&self) -> bool {
+        matches!(self, AccountKind::Resident)
+    }
+}
+
 impl std::fmt::Display for AccountKind {
     fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
         match self {