Add resizing functionality (#10)

* Add functionality to get the current state of windows

* Add screenshot to readme

* Add resizing width and height

* Add window size query script

* Add example with 25-50-25 split

Author: abstractlyZach

README.md

@@ -49,6 +49,7 @@ Sure, I could just use `tmux`, but that came with some issues:
 * allow different IDEs to be defined through a readable file (like TOML)
 * run arbitrary commands (not just shells and TUIs!)
 
+[screenshot](screenshots/early_magic_tile.png)
 
 ## Alternatives
 * [tmux](https://github.com/tmux/tmux)

pyproject.toml

@@ -39,6 +39,7 @@ omit = [
     "src/magic_tiler/subprocess_runner.py",
     "src/magic_tiler/sway.py",
     "src/magic_tiler/main.py",
+    "src/magic_tiler/get_window_sizes.py",
 ]
 
 [tool.coverage.report]
@@ -50,3 +51,4 @@ profile = "google"
 
 [tool.poetry.scripts]
 magic-tiler = "magic_tiler.main:main"
+get-window-sizes = "magic_tiler.get_window_sizes:main"

screenshots/early_magic_tile.png

@@ -0,0 +1,13 @@
+import pprint
+
+import click
+
+from magic_tiler import subprocess_runner
+from magic_tiler import sway
+
+
+@click.command()
+def main() -> None:
+    runner = subprocess_runner.SubprocessRunner()
+    window_manager = sway.Sway(runner)
+    pprint.pprint(window_manager.get_window_sizes())

src/magic_tiler/get_window_sizes.py

@@ -1,4 +1,5 @@
 import logging
+import pprint
 
 import click
 
@@ -21,13 +22,12 @@ def main() -> None:
     if swaywm.num_workspace_windows > 1:
         raise RuntimeError("There are multiple windows open in the current workspace.")
     swaywm.make_window('alacritty --title "first window"')
-    swaywm.make_horizontal_sibling(
-        "^first window$", 'alacritty --title "second window"'
-    )
-    swaywm.make_vertical_sibling("^second window$", 'alacritty --title "small window"')
-    swaywm.make_vertical_sibling(
-        "^first window$", 'alacritty --title "another small window"'
-    )
+    swaywm.make_horizontal_sibling("^first window$", 'alacritty --title "big boy"')
+    swaywm.resize_width("first window", 25)
+    swaywm.make_horizontal_sibling("^big boy$", 'alacritty --title "side window"')
+    swaywm.resize_width("side window", 33)
+    window_sizes = swaywm.get_window_sizes()
+    logging.info(pprint.pformat(window_sizes))
 
 
 def woops() -> None:

src/magic_tiler/main.py

@@ -1,5 +1,6 @@
 import logging
 import time
+from typing import Dict, List
 
 import i3ipc
 
@@ -35,10 +36,23 @@ def make_window(self, command: str) -> None:
         time.sleep(0.25)
 
     def resize_width(self, window_title_regex: str, container_percentage: int) -> None:
-        pass
+        window = self._get_window(window_title_regex)
+        window.command("focus")
+        window.command(f"resize set width {container_percentage} ppt")
 
     def resize_height(self, window_title_regex: str, container_percentage: int) -> None:
-        pass
+        window = self._get_window(window_title_regex)
+        window.command("focus")
+        window.command(f"resize set height {container_percentage} ppt")
+
+    def get_window_sizes(self) -> Dict[str, Dict[str, float]]:
+        return {
+            window.name: {
+                "width": window.window_rect.width,
+                "height": window.window_rect.height,
+            }
+            for window in self._get_windows_in_current_workspace()
+        }
 
     def _get_window(self, window_title_regex: str) -> i3ipc.Con:
         tree = self._sway.get_tree()
@@ -54,13 +68,7 @@ def _get_window(self, window_title_regex: str) -> i3ipc.Con:
             )
         return windows[0]
 
-    @property
-    def num_workspace_windows(self) -> int:
-        """Get the number of windows open on the current workspace
-
-        The current workspace must not be a "named workspace"
-        https://i3ipc-python.readthedocs.io/en/latest/replies.html#i3ipc.WorkspaceReply
-        """
+    def _get_windows_in_current_workspace(self) -> List[i3ipc.Con]:
         workspaces = self._sway.get_workspaces()
         for workspace in workspaces:
             if workspace.focused:
@@ -70,11 +78,20 @@ def num_workspace_windows(self) -> int:
                 break
         else:
             raise RuntimeError("There is no current workspace")
-        num_windows = 0
+        windows_in_current_workspace = []
         for container in self._sway.get_tree().leaves():
             logging.debug(
                 f'"{container.name}" is in workspace {container.workspace().num}'
             )
             if container.workspace().num == current_workspace_num:
-                num_windows += 1
-        return num_windows
+                windows_in_current_workspace.append(container)
+        return windows_in_current_workspace
+
+    @property
+    def num_workspace_windows(self) -> int:
+        """Get the number of windows open on the current workspace
+
+        The current workspace must not be a "named workspace"
+        https://i3ipc-python.readthedocs.io/en/latest/replies.html#i3ipc.WorkspaceReply
+        """
+        return len(self._get_windows_in_current_workspace())