Merge pull request #4 from alexarchambault/develop

Add windows-ansi-ps module

Author: alexarchambault

build.sbt

@@ -25,7 +25,7 @@ lazy val shared = Def.settings(
 )
 
 
-lazy val core = project
+lazy val jni = project
   .settings(
     shared,
     name := "windows-ansi",
@@ -35,10 +35,10 @@ lazy val core = project
     )
   )
 
-lazy val `test-cli` = project
+lazy val ps = project
   .settings(
     shared,
-    skip.in(publish) := true
+    name := "windows-ansi-ps"
   )
 
 // root project

core/src/main/java/io/github/alexarchambault/windowsansi/NativeImageFeature.java renamed to jni/src/main/java/io/github/alexarchambault/windowsansi/NativeImageFeature.java

@@ -1,2 +1 @@
 addSbtPlugin("com.geirsson" % "sbt-ci-release" % "1.5.0")
-addSbtPlugin("org.xerial.sbt" % "sbt-pack" % "0.12")

core/src/main/java/io/github/alexarchambault/windowsansi/WindowsAnsi.java renamed to jni/src/main/java/io/github/alexarchambault/windowsansi/WindowsAnsi.java

@@ -0,0 +1,30 @@
+package io.github.alexarchambault.windowsansi;
+
+import java.io.IOException;
+import java.nio.charset.StandardCharsets;
+import java.util.Base64;
+
+public final class PowershellRunner {
+
+    public static void runScript(String script) throws InterruptedException, IOException {
+
+        String fullScript = "& {\n" +
+                script +
+                "\n}";
+
+        Base64.Encoder base64 = Base64.getEncoder();
+        String encodedScript = base64.encodeToString(fullScript.getBytes(StandardCharsets.UTF_16LE));
+
+        ProcessBuilder builder = new ProcessBuilder(
+                "powershell.exe",
+                "-NoProfile",
+                "-NonInteractive",
+                "-EncodedCommand", encodedScript);
+        builder.inheritIO();
+
+        int retCode = builder.start().waitFor();
+        if (retCode != 0)
+            throw new IOException("Error running powershell script (return code: " + retCode + ")");
+    }
+
+}

core/src/main/java/io/github/alexarchambault/windowsansi/WindowsAnsiSubstitutions.java renamed to jni/src/main/java/io/github/alexarchambault/windowsansi/WindowsAnsiSubstitutions.java

@@ -0,0 +1,53 @@
+package io.github.alexarchambault.windowsansi;
+
+import java.io.IOException;
+
+public final class WindowsAnsiPs {
+
+    static boolean isWindows;
+
+    static {
+        isWindows = System.getProperty("os.name")
+                .toLowerCase(java.util.Locale.ROOT)
+                .contains("windows");
+    }
+
+    // adapted from https://github.com/rprichard/winpty/blob/7e59fe2d09adf0fa2aa606492e7ca98efbc5184e/misc/ConinMode.ps1
+    static String script = "$signature = @'\n" +
+            "[DllImport(\"kernel32.dll\", SetLastError = true)]\n" +
+            "public static extern IntPtr GetStdHandle(int nStdHandle);\n" +
+            "[DllImport(\"kernel32.dll\", SetLastError = true)]\n" +
+            "public static extern uint GetConsoleMode(\n" +
+            "    IntPtr hConsoleHandle,\n" +
+            "    out uint lpMode);\n" +
+            "[DllImport(\"kernel32.dll\", SetLastError = true)]\n" +
+            "public static extern uint SetConsoleMode(\n" +
+            "    IntPtr hConsoleHandle,\n" +
+            "    uint dwMode);\n" +
+            "public const int STD_OUTPUT_HANDLE = -11;\n" +
+            "public const int ENABLE_VIRTUAL_TERMINAL_PROCESSING = 0x0004;\n" +
+            "'@\n" +
+            "\n" +
+            "$WinAPI = Add-Type -MemberDefinition $signature `\n" +
+            "    -Name WinAPI -Namespace ConinModeScript `\n" +
+            "    -PassThru\n" +
+            "\n" +
+            "$handle = $WinAPI::GetStdHandle($WinAPI::STD_OUTPUT_HANDLE)\n" +
+            "$mode = 0\n" +
+            "$ret = $WinAPI::GetConsoleMode($handle, [ref]$mode)\n" +
+            "if ($ret -eq 0) {\n" +
+            "    throw \"GetConsoleMode failed (is stdin a console?)\"\n" +
+            "}\n" +
+            "$ret = $WinAPI::SetConsoleMode($handle, $mode -bor $WinAPI::ENABLE_VIRTUAL_TERMINAL_PROCESSING)\n" +
+            "if ($ret -eq 0) {\n" +
+            "    throw \"SetConsoleMode failed (is stdin a console?)\"\n" +
+            "}\n";
+
+    // not sure what happens on Windows versions that don't support ANSI mode
+    public static void setup() throws InterruptedException, IOException {
+        if (isWindows) {
+            PowershellRunner.runScript(script);
+        }
+    }
+
+}