[SimpleWallet] Use first 20 characters of wallet base filename for pr…

…ompt.
mtl1979 · Oct 22, 2019 · 7ac1680 · 7ac1680
1 parent 5e5c20f
commit 7ac1680
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/SimpleWallet/SimpleWallet.cpp b/src/SimpleWallet/SimpleWallet.cpp
@@ -1624,7 +1624,16 @@ ColouredMsg getPrompt(std::shared_ptr<WalletInfo> &walletInfo)
         walletName = walletInfo->walletFileName.substr(0, extPos);
     }
 
-    std::string shortName = walletName.substr(0, promptLength);
+    std::string shortName = walletName;
+    size_t pos;
+
+    if ((pos = shortName.rfind("\\")) != std::string::npos) {
+      shortName = shortName.substr(pos + 1, promptLength);
+    } else if ((pos = shortName.rfind("/")) != std::string::npos) {
+      shortName = shortName.substr(pos + 1, promptLength);
+    } else if (shortName.length() > promptLength) {
+      shortName = shortName.substr(0, promptLength);
+    }
 
     return InformationMsg("[BTOR " + shortName + "]: ");
 }