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

Author: saygox

commitizen/commands/commit.py

@@ -1,8 +1,12 @@
 import contextlib
 import os
+import selectors
 import sys
 import tempfile
 
+from asyncio import set_event_loop_policy, get_event_loop_policy, DefaultEventLoopPolicy
+from io import IOBase
+
 import questionary
 
 from commitizen import factory, git, out
@@ -19,14 +23,30 @@
 )
 
 
-class WrapStdin:
-    def __init__(self):
-        fd = os.open("/dev/tty", os.O_RDWR | os.O_NOCTTY)
-        tty = open(fd, "wb+", buffering=0)
+class CZEventLoopPolicy(DefaultEventLoopPolicy):
+    def get_event_loop(self):
+        self.set_event_loop(self._loop_factory(selectors.SelectSelector()))
+        return self._local._loop
+
+class WrapStdx:
+    def __init__(self, stdx:IOBase):
+        self._fileno = stdx.fileno()
+        if sys.platform == 'linux':
+            if self._fileno == 0:
+                fd = os.open("/dev/tty", os.O_RDWR | os.O_NOCTTY)
+                tty = open(fd, "wb+", buffering=0)
+            else:
+                tty = open("/dev/tty", "w")
+        else:
+            fd = os.open("/dev/tty", os.O_RDWR | os.O_NOCTTY)
+            if self._fileno == 0:
+                tty = open(fd, "wb+", buffering=0)
+            else:
+                tty = open(fd, "rb+", buffering=0)
         self.tty = tty
 
     def __getattr__(self, key):
-        if key == "encoding":
+        if key == "encoding" and (sys.platform != 'linux' or self._fileno == 0) :
             return "UTF-8"
         return getattr(self.tty, key)
 
@@ -84,9 +104,11 @@ def __call__(self):
             old_stdin = sys.stdin
             old_stdout = sys.stdout
             old_stderr = sys.stderr
-            sys.stdin = WrapStdin()
-            sys.stdout = open("/dev/tty", "w")
-            sys.stderr = open("/dev/tty", "w")
+            old_event_loop_policy=get_event_loop_policy()
+            set_event_loop_policy(CZEventLoopPolicy())
+            sys.stdin = WrapStdx(sys.stdin)
+            sys.stdout = WrapStdx(sys.stdout)
+            sys.stderr = WrapStdx(sys.stderr)
 
         if git.is_staging_clean() and not dry_run:
             raise NothingToCommitError("No files added to staging!")
@@ -98,18 +120,21 @@ def __call__(self):
         else:
             m = self.prompt_commit_questions()
 
-        out.info(f"\n{m}\n")
-
-        if dry_run:
-            raise DryRunExit()
-
         if commit_msg_file:
             sys.stdin.close()
             sys.stdout.close()
             sys.stderr.close()
+            set_event_loop_policy(old_event_loop_policy)
             sys.stdin = old_stdin
             sys.stdout = old_stdout
             sys.stderr = old_stderr
+
+        out.info(f"\n{m}\n")
+
+        if dry_run:
+            raise DryRunExit()
+
+        if commit_msg_file:
             defaultmesaage = ""
             with open(commit_msg_file) as f:
                 defaultmesaage = f.read()
@@ -126,6 +151,7 @@ def __call__(self):
         else:
             c = git.commit(m)
 
+
         if c.return_code != 0:
             out.error(c.err)
 

tests/commands/test_commit_command.py

@@ -197,7 +197,8 @@ def test_commit_from_pre_commit_msg_hook(config, mocker, capsys):
     }
 
     commit_mock = mocker.patch("commitizen.git.commit")
-    mocker.patch("commitizen.commands.commit.WrapStdin")
+    commit_mock.return_value = cmd.Command("success", "", "", "", 0)
+    mocker.patch("commitizen.commands.commit.WrapStdx")
     mocker.patch("os.open")
     reader_mock = mocker.mock_open(read_data="\n\n#test\n")
     mocker.patch("builtins.open", reader_mock, create=True)
@@ -208,13 +209,38 @@ def test_commit_from_pre_commit_msg_hook(config, mocker, capsys):
     assert "Commit message is successful!" in out
     commit_mock.assert_not_called()
 
-
-def test_WrapStdin(mocker):
+def test_WrapStdx(mocker):
     mocker.patch("os.open")
     reader_mock = mocker.mock_open(read_data="data")
     mocker.patch("builtins.open", reader_mock, create=True)
 
-    wrap_stdin = commands.commit.WrapStdin()
+    stdin_mock_fileno=mocker.patch.object(sys.stdin, 'fileno')
+    stdin_mock_fileno.return_value = 0
+    wrap_stdin = commands.commit.WrapStdx(sys.stdin)
 
     assert wrap_stdin.encoding == "UTF-8"
     assert wrap_stdin.read() == "data"
+
+
+    writer_mock = mocker.mock_open(read_data="data")
+    mocker.patch("builtins.open", writer_mock, create=True)
+    stdout_mock_fileno=mocker.patch.object(sys.stdout, 'fileno')
+    stdout_mock_fileno.return_value = 1
+    wrap_stout = commands.commit.WrapStdx(sys.stdout)
+    wrap_stout.write("data")
+
+    writer_mock.assert_called_once_with("/dev/tty", 'w')
+    writer_mock().write.assert_called_once_with("data")
+
+
+    writer_mock = mocker.mock_open(read_data="data")
+    mocker.patch("builtins.open", writer_mock, create=True)
+    stderr_mock_fileno=mocker.patch.object(sys.stdout, 'fileno')
+    stderr_mock_fileno.return_value = 2
+    wrap_sterr = commands.commit.WrapStdx(sys.stderr)
+
+
+    wrap_sterr.write("data")
+
+    writer_mock.assert_called_once_with("/dev/tty", 'w')
+    writer_mock().write.assert_called_once_with("data")