Move implementation to lib.rs

src/lib.rs

@@ -0,0 +1,92 @@
+mod checker;
+mod config;
+
+use std::process;
+
+pub use config::{CheckConfig, Config};
+use serde::Deserialize;
+use serde_json::json;
+use serde_repr::Deserialize_repr;
+
+use crate::checker::check_host;
+
+#[derive(Debug, Deserialize_repr, PartialEq)]
+#[repr(u8)]
+pub enum MouseButton {
+    Left = 1,
+    Middle = 2,
+    Right = 3,
+}
+
+#[derive(Debug, Deserialize, PartialEq)]
+pub struct ClickEvent {
+    pub name: String,
+    pub button: MouseButton,
+}
+
+pub async fn print_states(check_configs: &[CheckConfig]) {
+    print!("[");
+    let mut entries = vec![];
+    for check_config in check_configs {
+        entries.push(format!("{}", check_host(check_config).await));
+    }
+    entries.push(
+        json!({
+            "full_text": chrono::Local::now().format("%H:%M").to_string()
+        })
+        .to_string(),
+    );
+    println!("{}],", entries.join(","));
+}
+
+pub async fn get_click_cmd(name: String) -> Option<String> {
+    for check in Config::read().checks {
+        if check.name == name {
+            return check.click_cmd;
+        }
+    }
+    None
+}
+
+pub async fn run_click_cmd(cmd: String) {
+    if let Ok(mut child) = process::Command::new("sh")
+        .stdin(process::Stdio::piped())
+        .spawn()
+    {
+        use std::io::Write;
+        let _ = child.stdin.as_mut().unwrap().write_all(cmd.as_bytes());
+    };
+}
+
+pub fn read_click_event() -> Result<ClickEvent, ()> {
+    let stdin = std::io::stdin();
+    let mut input = String::new();
+
+    if stdin.read_line(&mut input).is_ok() {
+        // Return click event after removing leading comma
+        if let Ok(click_event) =
+            serde_json::from_str::<ClickEvent>(input.replace(",{", "{").as_str())
+        {
+            return Ok(click_event);
+        }
+    }
+    Err(())
+}
+
+#[cfg(test)]
+mod tests {
+    use crate::{ClickEvent, MouseButton};
+
+    #[test]
+    fn test_should_deserialize_click_event() {
+        let actual = serde_json::from_str::<ClickEvent>(r#"{"name": "test", "button": 1}"#);
+        let expected = ClickEvent {
+            name: "test".to_string(),
+            button: MouseButton::Left,
+        };
+
+        assert!(actual.is_ok());
+        let actual = actual.unwrap();
+        assert_eq!(actual, expected);
+    }
+}

src/main.rs

@@ -1,80 +1,8 @@
-mod checker;
+use checkbar::{get_click_cmd, print_states, read_click_event, run_click_cmd, Config, MouseButton};
-mod config;
-
-use std::process;
-use std::time::Duration;
-
-use serde::Deserialize;
 use serde_json::json;
-use serde_repr::Deserialize_repr;
+use std::time::Duration;
 use tokio::task;
 
-use crate::checker::check_host;
-use crate::config::{CheckConfig, Config};
-
-#[derive(Debug, Deserialize_repr, PartialEq)]
-#[repr(u8)]
-enum MouseButton {
-    Left = 1,
-    Middle = 2,
-    Right = 3,
-}
-
-#[derive(Debug, Deserialize, PartialEq)]
-struct ClickEvent {
-    name: String,
-    button: MouseButton,
-}
-
-async fn print_states(check_configs: &[CheckConfig]) {
-    print!("[");
-    let mut entries = vec![];
-    for check_config in check_configs {
-        entries.push(format!("{}", check_host(check_config).await));
-    }
-    entries.push(
-        json!({
-            "full_text": chrono::Local::now().format("%H:%M").to_string()
-        })
-        .to_string(),
-    );
-    println!("{}],", entries.join(","));
-}
-
-async fn get_click_cmd(name: String) -> Option<String> {
-    for check in Config::read().checks {
-        if check.name == name {
-            return check.click_cmd;
-        }
-    }
-    None
-}
-
-async fn run_click_cmd(cmd: String) {
-    if let Ok(mut child) = process::Command::new("sh")
-        .stdin(process::Stdio::piped())
-        .spawn()
-    {
-        use std::io::Write;
-        let _ = child.stdin.as_mut().unwrap().write_all(cmd.as_bytes());
-    };
-}
-
-fn read_click_event() -> Result<ClickEvent, ()> {
-    let stdin = std::io::stdin();
-    let mut input = String::new();
-
-    if stdin.read_line(&mut input).is_ok() {
-        // Return click event after removing leading comma
-        if let Ok(click_event) =
-            serde_json::from_str::<ClickEvent>(input.replace(",{", "{").as_str())
-        {
-            return Ok(click_event);
-        }
-    }
-    Err(())
-}
-
 #[tokio::main(flavor = "multi_thread", worker_threads = 2)]
 async fn main() {
     println!(
@@ -114,26 +42,3 @@ async fn main() {
 
     let _r = tokio::join!(inputs, checks);
 }
-
-#[cfg(test)]
-mod tests {
-    use crate::{ClickEvent, MouseButton};
-
-    #[test]
-    fn test_should_deserialize_click_event() {
-        let actual = serde_json::from_str::<ClickEvent>(r#"{"name": "test", "button": 1}"#);
-        let expected = ClickEvent {
-            name: "test".to_string(),
-            button: MouseButton::Left,
-        };
-
-        if actual.is_err() {
-            println!("{:?}", actual)
-        }
-
-        assert!(actual.is_ok());
-
-        let actual = actual.unwrap();
-        assert_eq!(actual, expected);
-    }
-}