From 0a9c03d1e1f850a8e50ebe83ed4fafdbf6f58e89 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Th=C3=A9ophile=20Bastian?= <contact@tobast.fr>
Date: Thu, 7 May 2020 13:16:44 +0200
Subject: [PATCH] Use tmux and vim configs. Fixes #1. Fixes #2.

---
 .gitignore            |  1 +
 patch2vimedit/main.py | 60 +++++++++++++++++++++++++++++++------------
 patch2vimedit/tmux.py | 23 +++++++++++++----
 3 files changed, 63 insertions(+), 21 deletions(-)

diff --git a/.gitignore b/.gitignore
index bce835a..d984604 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,3 +1,4 @@
 virtualenv
 venv
 *.pyc
+test
diff --git a/patch2vimedit/main.py b/patch2vimedit/main.py
index b3e63a5..d38e6ad 100644
--- a/patch2vimedit/main.py
+++ b/patch2vimedit/main.py
@@ -3,6 +3,7 @@ import pathlib
 import os
 import sys
 import time
+import tempfile
 
 import patch
 from tmux import TmuxSession
@@ -35,25 +36,23 @@ def get_patch(patch_path):
         return patch.PatchSet(stream)
 
 
-def main():
-    """ Entry-point function """
-    args = parse_args()
+def configure_home(home):
+    """ Configures the temporary home """
+    tmux_conf_path = home / ".tmux.conf"
+    vim_conf_path = home / ".vimrc"
 
-    if args.directory:
-        chdir_to = pathlib.Path(args.directory)
-        if not chdir_to.is_dir():
-            print(
-                (
-                    "The required working directory '{}' does not exist, or is not a "
-                    "directory."
-                ).format(str(chdir_to)),
-                file=sys.stderr,
-            )
-            sys.exit(1)
-        os.chdir(chdir_to)
+    print("Configuring tmux in {}".format(tmux_conf_path))
+    with tmux_conf_path.open("w") as tmux_conf:
+        tmux_conf.write("set -g status off\n")
+    with vim_conf_path.open("w") as vim_conf:
+        vim_conf.write(
+            "syntax on\nset bg=dark\nset number\nset ts=4\nset sw=4\nset et\n"
+        )
 
-    patchset = get_patch(args.patch)
+    return tmux_conf_path, vim_conf_path
 
+
+def apply_patchset(patchset):
     tmux_session = TmuxSession.create_detached()
     tmux_fg_process = tmux_session.process_attach(read_only=True)
     time.sleep(0.2)
@@ -74,5 +73,34 @@ def main():
         tmux_session.session.kill_session()
 
 
+def main():
+    """ Entry-point function """
+    args = parse_args()
+
+    patchset = get_patch(args.patch)
+
+    if args.directory:
+        chdir_to = pathlib.Path(args.directory)
+        if not chdir_to.is_dir():
+            print(
+                (
+                    "The required working directory '{}' does not exist, or is not a "
+                    "directory."
+                ).format(str(chdir_to)),
+                file=sys.stderr,
+            )
+            sys.exit(1)
+        os.chdir(chdir_to)
+
+    with tempfile.TemporaryDirectory() as tmp_home_path:
+        os.environ["HOME"] = tmp_home_path
+        tmp_home = pathlib.Path(tmp_home_path)
+        tmux_conf, _ = configure_home(tmp_home)
+        TmuxSession.initialize_server(
+            socket_name="patch2vimedit", config_file=tmux_conf.as_posix()
+        )
+        apply_patchset(patchset)
+
+
 if __name__ == "__main__":
     main()
diff --git a/patch2vimedit/tmux.py b/patch2vimedit/tmux.py
index 0424b54..a6ba356 100644
--- a/patch2vimedit/tmux.py
+++ b/patch2vimedit/tmux.py
@@ -9,16 +9,29 @@ import time
 class TmuxSession:
     """ A tmux session """
 
-    tmux_server = libtmux.Server()
+    tmux_server = None
+    tmux_socket = None
 
     def __init__(self, session):
+        if self.tmux_server is None:
+            raise Exception("Server not initialized")
         self.session = session
         self.session_id = session.id
-        self.keyboard_speed = 0.01
+        self.keyboard_speed = 0.00001
+
+    @classmethod
+    def initialize_server(cls, socket_name=None, config_file=None):
+        """ Initialize the tmux server """
+        cls.tmux_socket = socket_name
+        cls.tmux_server = libtmux.Server(
+            socket_name=socket_name, config_file=config_file
+        )
 
     @classmethod
     def create_detached(cls, name=None):
         """ Create a new detached tmux session, returning the TmuxSession """
+        if cls.tmux_server is None:
+            raise Exception("Server not initialized")
         if not name:
             name = "patch2vimsession-{:04x}".format(random.randint(0, 1 << 16 - 1))
         session = cls.tmux_server.new_session(session_name=name, attach=False)
@@ -32,8 +45,7 @@ class TmuxSession:
     def attach_ro(self):
         """ Same as `attach`, but attaches a tmux session in a read-only fashion (see
         `tmux attach -r`). """
-        # libtmux is not expressive enough to avoid `subprocess` here...
-        subprocess.run(["tmux", "attach", "-r", "-t", self.session.id], check=True)
+        self.tmux_server.cmd("attach", "-r", "-t", self.session.id)
 
     def session_exists(self):
         """ Checks that the session exists """
@@ -61,7 +73,8 @@ class TmuxSession:
 
     def type_delay(self):
         """ Introduce a small delay, to emulate a human typing """
-        time.sleep(self.keyboard_speed + random.gauss(0, self.keyboard_speed * 0.1))
+        delay = self.keyboard_speed + random.gauss(0, self.keyboard_speed * 0.1)
+        time.sleep(delay)
 
     def type_keys(self, *args):
         """ Same as `send_keys`, but emulates a human typing with pauses. This is