chore: adjust api

Author: linux-china

dotenvx-spring-boot/src/main/java/org/mvnsearch/dotenvx/spring/configuration/DotenvxEncryptorBuilder.java

@@ -4,6 +4,8 @@
 import org.mvnsearch.dotenvx.spring.encryptor.DotenvxEncryptorImpl;
 import org.springframework.lang.Nullable;
 
+import java.util.HashMap;
+
 /**
  * <p>DotenvxEncryptorBuilder class.</p>
  *
@@ -12,8 +14,11 @@
  */
 public class DotenvxEncryptorBuilder {
 
+    @Nullable
     private String publicKeyHex;
+    @Nullable
     private String privateKeyHex;
+    private HashMap<String, String> profileKeyPairs = new HashMap<>();
 
     /**
      * <p>Constructor for DotenvxEncryptorBuilder.</p>
@@ -26,14 +31,18 @@ public DotenvxEncryptorBuilder(@Nullable String publicKeyHex, @Nullable String p
         this.privateKeyHex = privateKeyHex;
     }
 
+    public DotenvxEncryptorBuilder() {
+    }
+
     /**
      * set public key hex
      *
      * @param publicKeyHex public key hex
      * @return this builder
      */
-    public DotenvxEncryptorBuilder setPublicKeyHex(@Nullable String publicKeyHex) {
+    public DotenvxEncryptorBuilder withPrimaryKeyPair(@Nullable String publicKeyHex, @Nullable String privateKeyHex) {
         this.publicKeyHex = publicKeyHex;
+        this.privateKeyHex = privateKeyHex;
         return this;
     }
 
@@ -43,8 +52,10 @@ public DotenvxEncryptorBuilder setPublicKeyHex(@Nullable String publicKeyHex) {
      * @param privateKeyHex private key hex
      * @return this builder
      */
-    public DotenvxEncryptorBuilder setPrivateKeyHex(@Nullable String privateKeyHex) {
-        this.privateKeyHex = privateKeyHex;
+    public DotenvxEncryptorBuilder withProfileKeyPair(@Nullable String publicKeyHex, @Nullable String privateKeyHex) {
+        if (publicKeyHex != null && privateKeyHex != null) {
+            this.profileKeyPairs.put(publicKeyHex, privateKeyHex);
+        }
         return this;
     }
 
@@ -54,6 +65,6 @@ public DotenvxEncryptorBuilder setPrivateKeyHex(@Nullable String privateKeyHex)
      * @return a {@link DotenvxEncryptor} object
      */
     public DotenvxEncryptor build() {
-        return new DotenvxEncryptorImpl(publicKeyHex, privateKeyHex);
+        return new DotenvxEncryptorImpl(publicKeyHex, privateKeyHex, profileKeyPairs);
     }
 }

dotenvx-spring-boot/src/main/java/org/mvnsearch/dotenvx/spring/encryptor/DotenvxEncryptorImpl.java

@@ -5,6 +5,9 @@
 import org.mvnsearch.dotenvx.spring.exception.EncryptionException;
 import org.springframework.lang.Nullable;
 
+import java.util.HashMap;
+import java.util.Map;
+
 /**
  * dotenvx encryptor implementation
  *
@@ -13,10 +16,12 @@
 public class DotenvxEncryptorImpl implements DotenvxEncryptor {
     private final String publicKeyHex;
     private final String privateKeyHex;
+    private HashMap<String, String> profileKeyPairs = new HashMap<>();
 
-    public DotenvxEncryptorImpl(@Nullable String publicKeyHex, @Nullable String privateKeyHex) {
+    public DotenvxEncryptorImpl(@Nullable String publicKeyHex, @Nullable String privateKeyHex, HashMap<String, String> profileKeyPairs) {
         this.publicKeyHex = publicKeyHex;
         this.privateKeyHex = privateKeyHex;
+        this.profileKeyPairs = profileKeyPairs;
     }
 
     /**
@@ -49,8 +54,18 @@ public String decrypt(String base64EncodedText) throws DecryptionException {
                 base64EncodedText = base64EncodedText.substring("encrypted:".length());
             }
             return Ecies.decrypt(privateKeyHex, base64EncodedText);
-        } catch (Exception e) {
-            throw new DecryptionException("Failed to decrypt text: " + base64EncodedText, e);
+        } catch (Exception ignore) {
+        }
+        for (Map.Entry<String, String> keyPair : profileKeyPairs.entrySet()) {
+            try {
+                String privateKey = keyPair.getValue();
+                if (base64EncodedText.startsWith("encrypted:")) {
+                    base64EncodedText = base64EncodedText.substring("encrypted:".length());
+                }
+                return Ecies.decrypt(privateKey, base64EncodedText);
+            } catch (Exception ignore) {
+            }
         }
+        throw new DecryptionException("Failed to decrypt text: " + base64EncodedText);
     }
 }