From 1ee4969d007bfe0ded9f52d471c168c666e6e80e Mon Sep 17 00:00:00 2001
From: Hatter Jiang <jht5945@gmail.com>
Date: Sun, 30 Jun 2024 20:14:34 +0800
Subject: [PATCH] feat: pinentry-cli more funcs

---
 .../hatter/tools/pinentry/PinEntryTool.java   | 23 +++++++++++++++----
 1 file changed, 19 insertions(+), 4 deletions(-)

diff --git a/src/main/java/me/hatter/tools/pinentry/PinEntryTool.java b/src/main/java/me/hatter/tools/pinentry/PinEntryTool.java
index f133ad9..1d0d7c4 100644
--- a/src/main/java/me/hatter/tools/pinentry/PinEntryTool.java
+++ b/src/main/java/me/hatter/tools/pinentry/PinEntryTool.java
@@ -16,10 +16,6 @@ public class PinEntryTool {
     private String pinEntry = null;
     private boolean disableFallbackCli = false;
 
-    public static void main(String[] args) throws PinEntryException {
-        System.out.println(PinEntryTool.instance().getPin());
-    }
-
     public static PinEntryTool instance() {
         return new PinEntryTool();
     }
@@ -43,6 +39,25 @@ public class PinEntryTool {
         return this;
     }
 
+    public String getDefaultPinOrInputWithoutThrows(String defaultPin) {
+        try {
+            return getDefaultPinOrInput(defaultPin);
+        } catch (PinEntryException e) {
+            throw new RuntimeException("Get default PIN or input field: " + e.getMessage(), e);
+        }
+    }
+
+    public String getDefaultPinOrInput(String defaultPin) throws PinEntryException {
+        return getDefaultPinOrInput(defaultPin, null, null);
+    }
+
+    public String getDefaultPinOrInput(String defaultPin, String description, String prompt) throws PinEntryException {
+        if (StringUtil.isNotEmpty(defaultPin)) {
+            return defaultPin;
+        }
+        return getPin(description, prompt);
+    }
+
     public String getPin() throws PinEntryException {
         return getPin(null, null);
     }