Implement _nt._supports_virtual_terminal

Author: msimacek

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/modules/PosixModuleBuiltins.java

@@ -611,6 +611,15 @@ static long getPid(@Bind PythonContext context) {
         }
     }
 
+    @Builtin(name = "_supports_virtual_terminal", minNumOfPositionalArgs = 0, os = PythonOS.PLATFORM_WIN32)
+    @GenerateNodeFactory
+    abstract static class SupportsVirtualTerminalNode extends PythonBuiltinNode {
+        @Specialization
+        static boolean supportsVirtualTerminal(@Bind PythonContext context) {
+            return context.getPosixSupport().supportsVirtualTerminal();
+        }
+    }
+
     @Builtin(name = "_is_inputhook_installed", minNumOfPositionalArgs = 0)
     @GenerateNodeFactory
     abstract static class IsInputHookInstalledNode extends PythonBuiltinNode {

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/runtime/EmulatedPosixSupport.java

@@ -611,6 +611,12 @@ public int msvcrtKbhit() throws UnsupportedPosixFeatureException {
         throw createUnsupportedFeature("msvcrt.kbhit");
     }
 
+    @Override
+    @TruffleBoundary
+    public boolean supportsVirtualTerminal() {
+        return false;
+    }
+
     @Override
     @TruffleBoundary
     public int[] pipe() throws PosixException {

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/runtime/LoggingPosixSupport.java

@@ -311,6 +311,13 @@ public final int msvcrtKbhit() throws PosixException {
         }
     }
 
+    @Override
+    @TruffleBoundary
+    public final boolean supportsVirtualTerminal() {
+        logEnter("supportsVirtualTerminal", "");
+        return logExit("supportsVirtualTerminal", "%s", delegate.supportsVirtualTerminal());
+    }
+
     @Override
     @TruffleBoundary
     public final int[] pipe() throws PosixException {

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/runtime/NativePosixSupport.java

@@ -236,6 +236,9 @@ abstract static class PosixNativeFunctionInvoker {
         @DowncallSignature(returnType = SINT32, argumentTypes = {})
         abstract int call_msvcrt_kbhit();
 
+        @DowncallSignature(returnType = SINT32)
+        abstract int supports_virtual_terminal();
+
         @DowncallSignature(returnType = SINT32, argumentTypes = {POINTER})
         abstract int call_pipe2(long pipefd);
 
@@ -902,6 +905,12 @@ public int msvcrtKbhit() throws PosixException {
         return result;
     }
 
+    @Override
+    @TruffleBoundary
+    public boolean supportsVirtualTerminal() {
+        return posixNativeFunctionInvoker.supports_virtual_terminal() != 0;
+    }
+
     @Override
     @TruffleBoundary
     public int[] pipe() throws PosixException {

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/runtime/PosixSupport.java

@@ -986,6 +986,9 @@ public abstract int forkExec(Object[] executables, Object[] args, Object cwd, Ob
     @TruffleBoundary
     public abstract int msvcrtKbhit() throws PosixException;
 
+    @TruffleBoundary
+    public abstract boolean supportsVirtualTerminal();
+
     @TruffleBoundary
     public abstract int[] pipe() throws PosixException;
 

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/runtime/PreInitPosixSupport.java

@@ -279,6 +279,13 @@ public final int msvcrtKbhit() throws PosixException {
         return nativePosixSupport.msvcrtKbhit();
     }
 
+    @Override
+    @TruffleBoundary
+    public final boolean supportsVirtualTerminal() {
+        checkNotInPreInitialization();
+        return nativePosixSupport.supportsVirtualTerminal();
+    }
+
     @Override
     @TruffleBoundary
     public final int[] pipe() throws PosixException {

graalpython/python-libposix/src/posix.c

@@ -1371,6 +1371,12 @@ GP_EXPORT int32_t call_msvcrt_kbhit(void) {
     return _kbhit();
 }
 
+GP_EXPORT int32_t supports_virtual_terminal(void) {
+    DWORD mode = 0;
+    HANDLE handle = GetStdHandle(STD_ERROR_HANDLE);
+    return GetConsoleMode(handle, &mode) && (mode & ENABLE_VIRTUAL_TERMINAL_PROCESSING) != 0;
+}
+
 GP_EXPORT int32_t call_pipe2(int32_t *pipefd) {
     int result = _pipe(pipefd, 8192, _O_BINARY | _O_NOINHERIT);
     if (result < 0) {
@@ -2632,6 +2638,10 @@ int32_t call_msvcrt_locking(int32_t fd, int32_t mode, int64_t nbytes) {
     return -1;
 }
 
+int32_t supports_virtual_terminal(void) {
+    return 0;
+}
+
 int32_t call_msvcrt_kbhit(void) {
     errno = ENOSYS;
     capture_errno();