feat(wm): initial commit

One week of blissful, in-the-zone coding, applying all of the lessons learnt from the development of yatta.
2026-07-20 03:51:16 +02:00 · 2021-07-29 16:18:06 -07:00
commit 61cee458a1
32 changed files with 5593 additions and 0 deletions
@@ -0,0 +1,91 @@
+use clap::Clap;
+use serde::Deserialize;
+use serde::Serialize;
+use strum::Display;
+use strum::EnumString;
+
+use crate::Layout;
+
+#[derive(Clone, Copy, Debug, Serialize, Deserialize, Display, EnumString)]
+#[strum(serialize_all = "snake_case")]
+#[derive(Clap)]
+pub enum OperationDirection {
+    Left,
+    Right,
+    Up,
+    Down,
+}
+
+impl OperationDirection {
+    pub fn can_resize(&self, layout: Layout, idx: usize, len: usize) -> bool {
+        match layout {
+            Layout::BSP => match self {
+                Self::Left => len != 0 && idx != 0,
+                Self::Up => len > 2 && idx != 0 && idx != 1,
+                Self::Right => len > 1 && idx % 2 == 0 && idx != len - 1,
+                Self::Down => len > 2 && idx != len - 1 && idx % 2 != 0,
+            },
+            _ => false,
+        }
+    }
+
+    pub fn is_valid(&self, layout: Layout, idx: usize, len: usize) -> bool {
+        match self {
+            OperationDirection::Up => match layout {
+                Layout::BSP => len > 2 && idx != 0 && idx != 1,
+                Layout::Columns => false,
+                Layout::Rows => idx != 0,
+            },
+            OperationDirection::Down => match layout {
+                Layout::BSP => len > 2 && idx != len - 1 && idx % 2 != 0,
+                Layout::Columns => false,
+                Layout::Rows => idx != len - 1,
+            },
+            OperationDirection::Left => match layout {
+                Layout::BSP => len > 1 && idx != 0,
+                Layout::Columns => idx != 0,
+                Layout::Rows => false,
+            },
+            OperationDirection::Right => match layout {
+                Layout::BSP => len > 1 && idx % 2 == 0,
+                Layout::Columns => idx != len - 1,
+                Layout::Rows => false,
+            },
+        }
+    }
+
+    pub fn new_idx(&self, layout: Layout, idx: usize) -> usize {
+        match self {
+            OperationDirection::Up => match layout {
+                Layout::BSP => {
+                    if idx % 2 == 0 {
+                        idx - 1
+                    } else {
+                        idx - 2
+                    }
+                }
+                Layout::Columns => unreachable!(),
+                Layout::Rows => idx - 1,
+            },
+            OperationDirection::Down => match layout {
+                Layout::BSP | Layout::Rows => idx + 1,
+                Layout::Columns => unreachable!(),
+            },
+            OperationDirection::Left => match layout {
+                Layout::BSP => {
+                    if idx % 2 == 0 {
+                        idx - 2
+                    } else {
+                        idx - 1
+                    }
+                }
+                Layout::Columns => idx - 1,
+                Layout::Rows => unreachable!(),
+            },
+            OperationDirection::Right => match layout {
+                Layout::BSP | Layout::Columns => idx + 1,
+                Layout::Rows => unreachable!(),
+            },
+        }
+    }
+}