Ensure json extension in wallet (neo-project#608)

* Update MainService.cs * Update MainService.cs
ProDog · Jul 9, 2020 · 92c09b8 · 92c09b8
1 parent b4b0e53
commit 92c09b8
Showing 1 changed file with 14 additions and 8 deletions.
diff --git a/neo-cli/CLI/MainService.cs b/neo-cli/CLI/MainService.cs
@@ -320,15 +320,21 @@ public void OpenWallet(string path, string password)
                 throw new FileNotFoundException();
             }
 
-            if (Path.GetExtension(path) == ".db3")
+            switch (Path.GetExtension(path).ToLowerInvariant())
             {
-                CurrentWallet = UserWallet.Open(path, password);
-            }
-            else
-            {
-                NEP6Wallet nep6wallet = new NEP6Wallet(path);
-                nep6wallet.Unlock(password);
-                CurrentWallet = nep6wallet;
+                case ".db3":
+                    {
+                        CurrentWallet = UserWallet.Open(path, password);
+                        break;
+                    }
+                case ".json":
+                    {
+                        NEP6Wallet nep6wallet = new NEP6Wallet(path);
+                        nep6wallet.Unlock(password);
+                        CurrentWallet = nep6wallet;
+                        break;
+                    }
+                default: throw new NotSupportedException();
             }
         }