initial commit

Signed-off-by: Sean Cross <sean@xobs.io>

src/main.rs

@@ -0,0 +1,123 @@
+fn make_holidays() -> std::collections::BTreeMap<chrono::Date<chrono::Local>, &'static str> {
+    use chrono::offset::TimeZone;
+    use chrono::Local;
+    let mut holidays = std::collections::BTreeMap::new();
+
+    holidays.insert(Local.ymd(2021, 1, 1), "New Year's Day");
+    holidays.insert(Local.ymd(2021, 2, 12), "Chinese New Year");
+    holidays.insert(Local.ymd(2021, 2, 13), "Chinese New Year");
+    holidays.insert(Local.ymd(2021, 4, 2), "Good Friday");
+    holidays.insert(Local.ymd(2021, 5, 1), "Labour Day");
+    holidays.insert(Local.ymd(2021, 5, 13), "Hari Raya Puasa");
+    holidays.insert(Local.ymd(2021, 7, 20), "Vesak Day");
+    holidays.insert(Local.ymd(2021, 8, 9), "National Day");
+    holidays.insert(Local.ymd(2021, 11, 4), "Deepavali");
+    holidays.insert(Local.ymd(2021, 12, 25), "Christmas");
+
+    holidays.insert(Local.ymd(2022, 1, 1), "New Year's Day");
+    holidays.insert(Local.ymd(2022, 2, 1), "Chinese New Year");
+    holidays.insert(Local.ymd(2022, 2, 2), "Chinese New Year");
+    holidays.insert(Local.ymd(2022, 4, 15), "Good Friday");
+    holidays.insert(Local.ymd(2022, 5, 1), "Labour Day");
+    holidays.insert(Local.ymd(2022, 5, 2), "Hari Raya Puasa");
+    holidays.insert(Local.ymd(2022, 5, 3), "Labour Day (Observed)");
+    holidays.insert(Local.ymd(2022, 5, 15), "Vesak Day");
+    holidays.insert(Local.ymd(2022, 7, 9), "Hari Raya Haji");
+    holidays.insert(Local.ymd(2022, 8, 9), "National Day");
+    holidays.insert(Local.ymd(2022, 10, 24), "Deepavali");
+    holidays.insert(Local.ymd(2022, 12, 25), "Christmas");
+    holidays
+}
+
+fn process_is_running(name: &str) -> bool {
+    use sysinfo::{ProcessExt, SystemExt};
+    // First we update all information of our system struct.
+    let mut system = sysinfo::System::new_all();
+    system.refresh_all();
+    for (_pid, proc_) in system.get_processes() {
+        if proc_.name() == name {
+            return true;
+        }
+    }
+    false
+}
+
+fn try_terminating(name: &str) -> Option<std::path::PathBuf> {
+    let mut process_path = None;
+    use sysinfo::{ProcessExt, SystemExt};
+
+    // First we update all information of our system struct.
+    let mut system = sysinfo::System::new_all();
+    system.refresh_all();
+
+    // Now let's print every process' id and name:
+    for (pid, proc_) in system.get_processes() {
+        // println!("{}:{} => status: {:?}", pid, proc_.name(), proc_.status());
+        if proc_.name() == name {
+            println!("LARK FOUND!!!  PID: {}, status: {:?}", pid, proc_.status());
+            println!("Original exe: {}", proc_.exe().display());
+            process_path = Some(proc_.exe().to_owned());
+            proc_.kill(sysinfo::Signal::Term);
+        }
+    }
+    process_path
+}
+
+fn is_working_hours<T: chrono::Datelike + chrono::Timelike>(date_time: &T) -> bool {
+    let holidays = make_holidays();
+
+    if let Some(holiday) = holidays.get(&chrono::Local::today()) {
+        println!("Happy {}! No work today.", holiday);
+        return false;
+    }
+
+    match date_time.weekday() {
+        chrono::Weekday::Sun | chrono::Weekday::Sat => return false,
+        _ => (),
+    }
+
+    date_time.hour() >= 9 && date_time.hour() <= 18
+}
+
+fn main() {
+    use chrono::Local;
+    use winrt_notification::{Duration, Sound, Toast};
+    let mut process_path: Option<std::path::PathBuf> = None;
+    let process_name = "Lark.exe";
+
+    loop {
+        let local = Local::now(); // e.g. `2014-11-28T21:45:59.324310806+09:00`
+        let should_be_running = is_working_hours(&local);
+        if let Some(p) = &process_path {
+            if should_be_running && !process_is_running(process_name) {
+                println!("Running Lark!");
+                std::process::Command::new(p)
+                    .spawn()
+                    .expect("Couldn't run Lark!");
+
+                Toast::new(Toast::POWERSHELL_APP_ID)
+                    .title("Started Lark")
+                    .text1("Running Lark because it's time to work")
+                    .sound(Some(Sound::SMS))
+                    .duration(Duration::Short)
+                    .show()
+                    .expect("unable to toast");
+            }
+        } else {
+            if !should_be_running && process_is_running(process_name) {
+                println!("Terminating Lark, since it should't be running");
+                process_path = try_terminating(process_name);
+                if process_path.is_some() {
+                    Toast::new(Toast::POWERSHELL_APP_ID)
+                        .title("Terminated Lark")
+                        .text1("Lark was terminated because it is outside working hours")
+                        .sound(Some(Sound::SMS))
+                        .duration(Duration::Short)
+                        .show()
+                        .expect("unable to toast");
+                }
+            }
+        }
+        std::thread::sleep(std::time::Duration::from_secs(100));
+    }
+}