Handle hidden chars on Win in CLI config ITs

quarkus-qe · Aug 25, 2024 · 9556b40 · 9556b40
1 parent b39e0ac
commit 9556b40
Show file tree

Hide file tree

Showing 3 changed files with 54 additions and 2 deletions.
diff --git a/...cli/src/main/java/io/quarkus/test/bootstrap/config/QuarkusEncryptConfigCommandResult.java b/...cli/src/main/java/io/quarkus/test/bootstrap/config/QuarkusEncryptConfigCommandResult.java
@@ -5,6 +5,8 @@
 import java.util.Objects;
 import java.util.function.Consumer;
 
+import io.quarkus.test.util.QuarkusCLIUtils;
+
 public class QuarkusEncryptConfigCommandResult extends QuarkusConfigCommandResult {
 
     private static final String SECRET_ENCRYPTED_TO = "was encrypted to";
@@ -19,7 +21,11 @@ public class QuarkusEncryptConfigCommandResult extends QuarkusConfigCommandResul
 
     public String getGeneratedEncryptionKey() {
         if (output.contains(WITH_GENERATED_KEY)) {
-            return output.transform(o -> o.substring(o.lastIndexOf(" "))).trim();
+            return output
+                    .transform(o -> o.substring(o.lastIndexOf(" ")))
+                    .transform(QuarkusCLIUtils::toUtf8)
+                    .transform(QuarkusCLIUtils::removeAnsiAndHiddenChars)
+                    .trim();
         }
         return null;
     }
@@ -29,6 +35,8 @@ public String getEncryptedSecret() {
             encryptedSecret = output
                     .transform(o -> o.split(SECRET_ENCRYPTED_TO)[1])
                     .transform(remaining -> remaining.split(WITH_GENERATED_KEY)[0])
+                    .transform(QuarkusCLIUtils::toUtf8)
+                    .transform(QuarkusCLIUtils::removeAnsiAndHiddenChars)
                     .trim();
         }
         return encryptedSecret;

diff --git a/...st-cli/src/main/java/io/quarkus/test/bootstrap/config/QuarkusSetConfigCommandBuilder.java b/...st-cli/src/main/java/io/quarkus/test/bootstrap/config/QuarkusSetConfigCommandBuilder.java
@@ -2,6 +2,9 @@
 
 import java.util.ArrayList;
 
+import io.quarkus.test.util.QuarkusCLIUtils;
+import io.smallrye.common.os.OS;
+
 public class QuarkusSetConfigCommandBuilder {
 
     private final boolean updateScenario;
@@ -23,7 +26,11 @@ public QuarkusSetConfigCommandBuilder name(String name) {
     }
 
     public QuarkusSetConfigCommandBuilder value(String value) {
-        this.propertyValue = value;
+        if (OS.WINDOWS.isCurrent()) {
+            this.propertyValue = QuarkusCLIUtils.escapeSecretCharsForWindows(value);
+        } else {
+            this.propertyValue = value;
+        }
         return this;
     }
 

diff --git a/quarkus-test-cli/src/main/java/io/quarkus/test/util/QuarkusCLIUtils.java b/quarkus-test-cli/src/main/java/io/quarkus/test/util/QuarkusCLIUtils.java
@@ -11,6 +11,8 @@
 import java.io.FileWriter;
 import java.io.IOException;
 import java.io.OutputStream;
+import java.nio.charset.Charset;
+import java.nio.charset.StandardCharsets;
 import java.nio.file.Files;
 import java.nio.file.Paths;
 import java.util.List;
@@ -27,12 +29,15 @@
 import org.codehaus.plexus.util.xml.pull.XmlPullParserException;
 
 import io.quarkus.test.bootstrap.QuarkusCliRestService;
+import io.smallrye.common.os.OS;
 
 public abstract class QuarkusCLIUtils {
     public static final String RESOURCES_DIR = Paths.get("src", "main", "resources").toString();
     public static final String PROPERTIES_FILE = "application.properties";
     public static final String PROPERTIES_YAML_FILE = "application.yml";
     public static final String POM_FILE = "pom.xml";
+    private static final String ANSI_BOLD_TEXT_ESCAPE_SEQ = "[1m";
+    private static final char ESCAPE_CHARACTER = 27;
 
     /**
      * This constant stands for number of fields in groupId:artifactId:version string, when separated via ":".
@@ -384,4 +389,36 @@ public static String escapeSecretCharsForWindows(String secret) {
                 .replace("\"", "\\\"")
                 + "\"";
     }
+
+    /**
+     * When Quarkus CLI prints out text, especially by {@code quarkus config encrypt} command,
+     * important parts (like encoded secrets) can be highlighted or there can be hidden chars.
+     * We recognize hidden chars etc. This method handles both situation. It's definitely imperfect,
+     * but we only deal with scenarios (issues) we run on.
+     */
+    public static String removeAnsiAndHiddenChars(String text) {
+        if (OS.current() == OS.WINDOWS) {
+            var result = text
+                    .trim()
+                    .transform(t -> {
+                        if (t.contains(ANSI_BOLD_TEXT_ESCAPE_SEQ)) {
+                            return t.substring(ANSI_BOLD_TEXT_ESCAPE_SEQ.length());
+                        }
+                        return t;
+                    })
+                    .transform(t -> {
+                        int idx = t.indexOf(ESCAPE_CHARACTER);
+                        if (idx >= 0) {
+                            return t.substring(0, idx);
+                        }
+                        return t;
+                    });
+            return result;
+        }
+        return text;
+    }
+
+    public static String toUtf8(String t) {
+        return new String(t.getBytes(Charset.defaultCharset()), StandardCharsets.UTF_8);
+    }
 }