feat(commands/commit): apply prepare-commit-msg hook for Win

Author: saygox

commitizen/wrap_stdio_unix.py

@@ -11,7 +11,7 @@ def get_event_loop(self):
         return self._local._loop
 
 
-class WrapStdioLinux:
+class WrapStdioUnix:
     def __init__(self, stdx: IOBase):
         self._fileno = stdx.fileno()
         fd = os.open("/dev/tty", os.O_RDWR | os.O_NOCTTY)
@@ -43,15 +43,15 @@ def _wrap_stdio():
 
     global backup_stdin
     backup_stdin = sys.stdin
-    sys.stdin = WrapStdioLinux(sys.stdin)
+    sys.stdin = WrapStdioUnix(sys.stdin)
 
     global backup_stdout
     backup_stdout = sys.stdout
-    sys.stdout = WrapStdioLinux(sys.stdout)
+    sys.stdout = WrapStdioUnix(sys.stdout)
 
     global backup_stderr
     backup_stdout = sys.stderr
-    sys.stderr = WrapStdioLinux(sys.stderr)
+    sys.stderr = WrapStdioUnix(sys.stderr)
 
 
 def _unwrap_stdio():

commitizen/wrap_stdio_windows.py

@@ -1,6 +1,58 @@
-def _wrap_stdio():
-    pass
+import sys
 
+if sys.platform == "win32":  # pragma: no cover
+    import msvcrt
+    import os
+    from ctypes import c_ulong, windll  # noqa
+    from ctypes.wintypes import HANDLE
+    from io import IOBase
 
-def _unwrap_stdio():
-    pass
+    STD_INPUT_HANDLE = c_ulong(-10)
+    STD_OUTPUT_HANDLE = c_ulong(-11)
+
+    class WrapStdioWindows:
+        def __init__(self, stdx: IOBase):
+            self._fileno = stdx.fileno()
+            if self._fileno == 0:
+                fd = os.open("CONIN$", os.O_RDWR | os.O_BINARY)
+                tty = open(fd, "r")
+                handle = HANDLE(msvcrt.get_osfhandle(fd))  # noqa
+                windll.kernel32.SetStdHandle(STD_INPUT_HANDLE, handle)
+            elif self._fileno == 1:
+                fd = os.open("CONOUT$", os.O_RDWR | os.O_BINARY)
+                tty = open(fd, "w")
+                handle = HANDLE(msvcrt.get_osfhandle(fd))  # noqa
+                windll.kernel32.SetStdHandle(STD_OUTPUT_HANDLE, handle)
+            else:
+                raise Exception("not defined type")
+            self._tty = tty
+
+        def __getattr__(self, key):
+            if key == "encoding" and self._fileno == 0:
+                return "UTF-8"
+            return getattr(self._tty, key)
+
+        def __del__(self):
+            if "_tty" in self.__dict__:
+                self._tty.close()
+
+    backup_stdin = None
+    backup_stdout = None
+
+    def _wrap_stdio():
+        global backup_stdin
+        backup_stdin = sys.stdin
+        sys.stdin = WrapStdioWindows(sys.stdin)
+
+        global backup_stdout
+        backup_stdout = sys.stdout
+        sys.stdout = WrapStdioWindows(sys.stdout)
+
+    def _unwrap_stdio():
+        global backup_stdin
+        sys.stdin.close()
+        sys.stdin = backup_stdin
+
+        global backup_stdout
+        sys.stdout.close()
+        sys.stdout = backup_stdout