feat(custom_layout): add yaml file support

This commit adds support for loading custom layouts from yaml files, and
also moves the custom layout loading and validating logic into the
komorebi-core crate.

re #50

komorebi-core/Cargo.toml

@@ -12,4 +12,5 @@ clap = "3.0.0-beta.4"
 color-eyre = "0.5"
 serde = { version = "1", features = ["derive"] }
 serde_json = "1"
+serde_yaml = "0.8"
 strum = { version = "0.21", features = ["derive"] }

komorebi-core/src/custom_layout.rs

@@ -1,6 +1,11 @@
 use std::collections::HashMap;
+use std::fs::File;
+use std::io::BufReader;
 use std::ops::Deref;
+use std::path::PathBuf;
 
+use color_eyre::eyre::anyhow;
+use color_eyre::Result;
 use serde::Deserialize;
 use serde::Serialize;
 
@@ -18,6 +23,28 @@ impl Deref for CustomLayout {
 }
 
 impl CustomLayout {
+    pub fn from_path_buf(path: PathBuf) -> Result<Self> {
+        let invalid_filetype = anyhow!("custom layouts must be json or yaml files");
+        let layout: Self = match path.extension() {
+            Some(extension) => {
+                if extension == "yaml" || extension == "yml" {
+                    serde_yaml::from_reader(BufReader::new(File::open(path)?))?
+                } else if extension == "json" {
+                    serde_json::from_reader(BufReader::new(File::open(path)?))?
+                } else {
+                    return Err(invalid_filetype);
+                }
+            }
+            None => return Err(invalid_filetype),
+        };
+
+        if !layout.is_valid() {
+            return Err(anyhow!("the layout file provided was invalid"));
+        }
+
+        Ok(layout)
+    }
+
     #[must_use]
     pub fn column_with_idx(&self, idx: usize) -> (usize, Option<&Column>) {
         let column_idx = self.column_for_container_idx(idx);

komorebi/src/window_manager.rs

@@ -1,6 +1,4 @@
 use std::collections::VecDeque;
-use std::fs::File;
-use std::io::BufReader;
 use std::io::ErrorKind;
 use std::num::NonZeroUsize;
 use std::path::PathBuf;
@@ -1059,11 +1057,8 @@ impl WindowManager {
     #[tracing::instrument(skip(self))]
     pub fn change_workspace_custom_layout(&mut self, path: PathBuf) -> Result<()> {
         tracing::info!("changing layout");
-        let layout: CustomLayout = serde_json::from_reader(BufReader::new(File::open(path)?))?;
-        if !layout.is_valid() {
-            return Err(anyhow!("the layout file provided was invalid"));
-        }
 
+        let layout = CustomLayout::from_path_buf(path)?;
         let workspace = self.focused_workspace_mut()?;
 
         match workspace.layout() {
@@ -1182,13 +1177,7 @@ impl WindowManager {
         path: PathBuf,
     ) -> Result<()> {
         tracing::info!("setting workspace layout");
-        let file = File::open(path)?;
-        let reader = BufReader::new(file);
-        let layout: CustomLayout = serde_json::from_reader(reader)?;
-        if !layout.is_valid() {
-            return Err(anyhow!("the layout file provided was invalid"));
-        }
-
+        let layout = CustomLayout::from_path_buf(path)?;
         let invisible_borders = self.invisible_borders;
         let offset = self.work_area_offset;
         let focused_monitor_idx = self.focused_monitor_idx();

komorebic/src/main.rs

@@ -299,7 +299,7 @@ struct Load {
 
 #[derive(Clap, AhkFunction)]
 struct LoadLayout {
-    /// File from which the custom layout definition should be loaded
+    /// JSON or YAML file from which the custom layout definition should be loaded
     path: String,
 }