Merge pull request #4 from abstractlyZach/window-management

Add basic window management

Author: abstractlyZach

.flake8

@@ -3,6 +3,6 @@ select = ANN,B,B9,BLK,C,E,F,I,S,W
 ignore = E203,E501,W503,ANN101
 max-complexity = 10
 max-line-length = 90
-application-import-names = "magic_tiler,tests"
+application-import-names = magic_tiler,tests
 import-order-style = google
 per-file-ignores = tests/*:S101,ANN

mypy.ini

@@ -0,0 +1,4 @@
+[mypy]
+
+[mypy-i3ipc]
+ignore_missing_imports = True

pyproject.toml

@@ -33,8 +33,16 @@ source = ["src", "*/site-packages"]
 [tool.coverage.run]
 branch = true
 source = ["magic_tiler"]
-omit = ["src/magic_tiler/interfaces.py"]
+omit = [
+    "src/magic_tiler/interfaces.py",
+    "src/magic_tiler/subprocess_runner.py",
+    "src/magic_tiler/sway.py",
+    "src/magic_tiler/main.py",
+]
 
 [tool.coverage.report]
 show_missing = true
 fail_under = 100
+
+[tool.isort]
+profile = "google"

src/magic_tiler/interfaces.py

@@ -17,3 +17,10 @@ def resize_width(self, window_title_regex: str, container_percentage: int) -> No
     @abc.abstractmethod
     def resize_height(self, window_title_regex: str, container_percentage: int) -> None:
         pass
+
+
+class Runner(object):
+    @abc.abstractmethod
+    def run_and_disown(self, command: str) -> None:
+        """Run a command and don't wait for it to finish"""
+        pass

src/magic_tiler/main.py

@@ -0,0 +1,10 @@
+import logging
+
+from magic_tiler import subprocess_runner
+from magic_tiler import sway
+
+logging.basicConfig(level=logging.INFO)
+swaywm = sway.Sway(subprocess_runner.SubprocessRunner())
+swaywm.make_horizontal_sibling("Alacritty:v", 'alacritty -e sh -c "ls | fzf"')
+# how do we make alacritty hang around after running the initial command?
+swaywm.make_horizontal_sibling("Alacritty:poetry", 'alacritty -e zsh -c "ls"')

src/magic_tiler/subprocess_runner.py

@@ -0,0 +1,12 @@
+import shlex
+import subprocess  # noqa: S404
+
+from magic_tiler import interfaces
+
+
+class SubprocessRunner(interfaces.Runner):
+    def run_and_disown(self, command: str) -> None:
+        """Run a command without waiting for it to exit"""
+        # split a command into tokens for the shell
+        args = shlex.split(command)
+        subprocess.Popen(args)  # noqa: S603

src/magic_tiler/sway.py

@@ -0,0 +1,47 @@
+import logging
+import time
+
+import i3ipc
+
+from magic_tiler import interfaces  # pragma: nocover
+
+"""We need to sleep for a short time since processes take time to start.
+If we don't sleep, then we may be focusing on a different window by the
+time that the current window is spawning, which would put it in the wrong
+split.
+"""
+SLEEP_TIME = 0.25
+
+
+class Sway(interfaces.TilingWindowManager):  # pragma: nocover
+    def __init__(self, runner: interfaces.Runner) -> None:
+        self._sway = i3ipc.Connection()
+        self._runner = runner
+
+    def make_horizontal_sibling(self, window_title_regex: str, command: str) -> None:
+        self._focus_window(window_title_regex)
+        self._runner.run_and_disown(command)
+        time.sleep(0.25)
+
+    def make_vertical_sibling(self, window_title_regex: str, command: str) -> None:
+        pass
+
+    def resize_width(self, window_title_regex: str, container_percentage: int) -> None:
+        pass
+
+    def resize_height(self, window_title_regex: str, container_percentage: int) -> None:
+        pass
+
+    def _focus_window(self, window_title_regex: str) -> None:
+        tree = self._sway.get_tree()
+        windows = tree.find_named(window_title_regex)
+        logging.debug(f"windows are {windows}")
+        if len(windows) > 1:
+            raise RuntimeError(
+                f'There is more than 1 window that matches the regex "{window_title_regex}"'
+            )
+        if len(windows) < 1:
+            raise RuntimeError(
+                f'There are no windows that matches the regex "{window_title_regex}"'
+            )
+        windows[0].command("focus")