feat: only log change state

Author: daniel-sc

src/get_state.rs

@@ -0,0 +1,23 @@
+use active_win_pos_rs::get_active_window;
+use device_query::{DeviceQuery, DeviceState};
+
+#[derive(Debug, PartialEq)]
+pub struct State {
+    pub window_title: String,
+    pub cursor: (i32, i32),
+}
+
+pub(crate) fn get_state() -> State {
+    let mouse = DeviceState::new().get_mouse();
+    let window_title = match get_active_window() {
+        Ok(window) => format!("{}: {}", window.app_name, window.title),
+        Err(e) => {
+            eprintln!("Error getting active window: {:?}", e);
+            String::new()
+        }
+    };
+    State {
+        window_title,
+        cursor: mouse.coords,
+    }
+}

src/main.rs

@@ -1,97 +1,22 @@
-use active_win_pos_rs::get_active_window;
+mod get_state;
+mod write_csv;
+
 use chrono::Local;
-use device_query::{DeviceQuery, DeviceState};
-use std::fs::{create_dir_all, OpenOptions};
-use std::io::Write;
-use std::path::Path;
 use std::thread::sleep;
 use std::time::Duration;
 
-const SKIP_WRITING_TO_FILE: bool = false;
 const FREQUENCY: f64 = 10.0; // seconds
 
-/// Retrieves the current mouse cursor position.
-/// Returns a tuple (x, y). If an error occurs, returns None.
-fn get_mouse_cursor_position() -> Option<(i32, i32)> {
-    let device = DeviceState::new();
-    let mouse = device.get_mouse();
-    Some((mouse.coords.0, mouse.coords.1))
-}
-
-/// Retrieves the title of the active window.
-/// If no active window is found or an error occurs, returns an empty String.
-fn get_active_window_title() -> String {
-    match get_active_window() {
-        Ok(window) => format!("{}: {}", window.app_name, window.title),
-        Err(e) => {
-            eprintln!("Error getting active window: {:?}", e);
-            String::new()
-        }
-    }
-}
-
-/// Appends a log line to the CSV file.
-///
-/// The log line has the format:
-/// `YYYY-MM-DD HH:MM:SS.mmm;window_title_without_semicolon;x,y`
-fn add_to_csv(
-    csv_filename: &str,
-    current_time: &chrono::DateTime<Local>,
-    window_title: &str,
-    cursor: &str,
-) {
-    // Format the time to include milliseconds (as in the Python code)
-    let ms = current_time.timestamp_subsec_millis();
-    let time_str = format!("{}.{:03}", current_time.format("%Y-%m-%d %H:%M:%S"), ms);
-
-    // Remove any semicolons from the window title
-    let title_adapted = window_title.replace(";", "");
-    let line = format!("{};{};{}", time_str, title_adapted, cursor);
-
-    println!("{}", line);
-
-    if !SKIP_WRITING_TO_FILE {
-        // Ensure that the parent directory exists
-        if let Some(parent) = Path::new(csv_filename).parent() {
-            if !parent.exists() {
-                println!("creating {}", parent.display());
-                if let Err(e) = create_dir_all(parent) {
-                    eprintln!("Error creating directory {}: {}", parent.display(), e);
-                }
-            }
-        }
-
-        // Open the file in append mode (or create it if it does not exist)
-        let file_result = OpenOptions::new()
-            .append(true)
-            .create(true)
-            .open(csv_filename);
-        match file_result {
-            Ok(mut file) => {
-                if let Err(e) = writeln!(file, "{}", line) {
-                    eprintln!("Error writing to file {}: {}", csv_filename, e);
-                }
-            }
-            Err(e) => {
-                eprintln!("Error opening file {}: {}", csv_filename, e);
-            }
-        }
-    }
-}
-
 fn main() {
+    let mut prev_state = get_state::get_state();
     loop {
         let current_time = Local::now();
-
-        // Get the active window title and mouse cursor position
-        let current_window_title = get_active_window_title();
-        if let Some((x, y)) = get_mouse_cursor_position() {
-            let cursor = format!("{},{}", x, y);
-            // Build the filename: logs/activity_log_YYYY_MM_DD.csv
+        let state = get_state::get_state();
+        if state != prev_state {
+            let cursor = format!("{},{}", state.cursor.0, state.cursor.1);
             let csv_filename = format!("logs/activity_log_{}.csv", current_time.format("%Y_%m_%d"));
-            add_to_csv(&csv_filename, &current_time, &current_window_title, &cursor);
-        } else {
-            println!("Error getting mouse cursor position");
+            write_csv::add_to_csv(&csv_filename, &current_time, &state.window_title, &cursor);
+            prev_state = state;
         }
 
         sleep(Duration::from_secs_f64(FREQUENCY));

src/write_csv.rs

@@ -0,0 +1,51 @@
+use chrono::Local;
+use std::fs::{create_dir_all, OpenOptions};
+use std::path::Path;
+use std::io::Write;
+
+/// Appends a log line to the CSV file.
+///
+/// The log line has the format:
+/// `YYYY-MM-DD HH:MM:SS.mmm;window_title_without_semicolon;x,y`
+pub(crate) fn add_to_csv(
+    csv_filename: &str,
+    current_time: &chrono::DateTime<Local>,
+    window_title: &str,
+    cursor: &str,
+) {
+    // Format the time to include milliseconds (as in the Python code)
+    let ms = current_time.timestamp_subsec_millis();
+    let time_str = format!("{}.{:03}", current_time.format("%Y-%m-%d %H:%M:%S"), ms);
+
+    // Remove any semicolons from the window title
+    let title_adapted = window_title.replace(";", "");
+    let line = format!("{};{};{}", time_str, title_adapted, cursor);
+
+    println!("{}", line);
+
+    // Ensure that the parent directory exists
+    if let Some(parent) = Path::new(csv_filename).parent() {
+        if !parent.exists() {
+            println!("creating {}", parent.display());
+            if let Err(e) = create_dir_all(parent) {
+                eprintln!("Error creating directory {}: {}", parent.display(), e);
+            }
+        }
+    }
+
+    // Open the file in append mode (or create it if it does not exist)
+    let file_result = OpenOptions::new()
+        .append(true)
+        .create(true)
+        .open(csv_filename);
+    match file_result {
+        Ok(mut file) => {
+            if let Err(e) = writeln!(file, "{}", line) {
+                eprintln!("Error writing to file {}: {}", csv_filename, e);
+            }
+        }
+        Err(e) => {
+            eprintln!("Error opening file {}: {}", csv_filename, e);
+        }
+    }
+}