Add csv logging from daemon

.gitignore

@@ -2,3 +2,9 @@ debug/
 target/
 .idea/
 pkg/
+
+
+# Auto-generated by autoignore:
+# --------------- #
+test.csv
+# ----------------#

src/args.rs

@@ -186,12 +186,16 @@ enum ACSCommand {
         no_animation: bool,
 
         /// Graph
-        #[structopt(short = "g", long = "--graph")]
+        #[structopt(short = "g", long = "graph")]
         graph_type: Option<String>,
 
         /// Commit hash
         #[structopt(short, long)]
         commit: bool,
+
+        /// Write to a csv file
+        #[structopt(long = "csv")]
+        csv_file: Option<String>,
     },
 
     /// Monitor each cpu, it's min, max, and current speed, along with the governor
@@ -220,6 +224,10 @@ enum ACSCommand {
         /// Commit hash
         #[structopt(short, long)]
         commit: bool,
+
+        /// Write to a csv file
+        #[structopt(long = "csv")]
+        csv_file: Option<String>,
     },
 }
 
@@ -234,6 +242,7 @@ pub fn parse_args(config: config::Config) {
         graph: GraphType::Hidden,
         commit: false,
         testing: false,
+        csv_file: None,
     };
 
     let int = Interface {
@@ -280,6 +289,7 @@ pub fn parse_args(config: config::Config) {
             no_animation,
             graph_type,
             commit,
+            csv_file,
         } => {
             if !config_dir_exists() {
                 warn_user!("Config directory '/etc/acs' does not exist!");
@@ -315,6 +325,7 @@ pub fn parse_args(config: config::Config) {
                 graph: parsed_graph_type,
                 commit,
                 testing: false,
+                csv_file,
             };
 
             match daemon_init(settings, config) {
@@ -333,6 +344,7 @@ pub fn parse_args(config: config::Config) {
             graph_type,
             hook,
             commit,
+            csv_file,
         } => {
             if !config_dir_exists() {
                 warn_user!("Config directory '/etc/acs' does not exist!");
@@ -367,6 +379,7 @@ pub fn parse_args(config: config::Config) {
                 graph: parsed_graph_type,
                 commit,
                 testing: false,
+                csv_file,
             };
 
             match daemon_init(settings, config) {

src/cpu.rs

@@ -23,6 +23,7 @@ pub trait Speed {
     fn get_temp(&mut self) -> Result<(), Error>;
     fn get_gov(&mut self) -> Result<(), Error>;
     fn set_gov(&mut self, gov: String) -> Result<(), Error>;
+    fn to_csv(&self) -> String;
     fn random() -> CPU;
 }
 
@@ -205,6 +206,20 @@ impl Speed for CPU {
             },
         }
     }
+
+    fn to_csv(&self) -> String {
+        format!(
+            "{},{},{},{},{},{},{},{}\n",
+            self.name,
+            self.number,
+            self.max_freq,
+            self.min_freq,
+            self.cur_freq,
+            self.cur_temp,
+            self.cur_usage,
+            self.gov
+        )
+    }
 }
 
 impl fmt::Display for CPU {

src/daemon.rs

@@ -1,5 +1,8 @@
 use crate::power::battery::{has_battery, Battery};
 use std::convert::TryInto;
+use std::fs::{File, OpenOptions};
+use std::io::Write;
+use std::path::Path;
 use std::sync::{Arc, Mutex};
 use std::time::SystemTime;
 use std::{thread, time};
@@ -69,6 +72,7 @@ pub trait Checker {
     fn update_all(&mut self) -> Result<(), Error>;
 
     fn run_state_machine(&mut self) -> State;
+    fn write_csv(&self);
 
     fn preprint_render(&mut self) -> String;
     fn postprint_render(&mut self) -> String;
@@ -184,6 +188,23 @@ impl Checker for Daemon {
         state
     }
 
+    fn write_csv(&self) {
+        let lines = &self.cpus.iter().map(|c| c.to_csv()).collect::<String>();
+
+        if let Some(name) = &self.settings.csv_file {
+            let mut file = OpenOptions::new()
+                .write(true)
+                .append(true) // This is needed to append to file
+                .open(name)
+                .unwrap();
+
+            match write!(file, "{}", lines) {
+                Ok(_) => {}
+                Err(..) => warn_user!("CSV Writer not working."),
+            };
+        }
+    }
+
     fn init(&mut self) {
         // Get the commit hash from the compile time env variable
         if self.settings.commit {
@@ -192,6 +213,12 @@ impl Checker for Daemon {
 
         self.timeout_battery = time::Duration::from_millis(self.settings.delay_battery);
         self.timeout = time::Duration::from_millis(self.settings.delay);
+
+        if let Some(name) = &self.settings.csv_file {
+            if !Path::new(name).exists() {
+                File::create(name).expect("CSV file could not be created.");
+            }
+        }
     }
 
     fn start_loop(&mut self) -> Result<(), Error> {
@@ -204,6 +231,8 @@ impl Checker for Daemon {
         self.lid_state = read_lid_state()?;
         self.usage = calculate_average_usage(&self.cpus) * 100.0;
 
+        self.write_csv();
+
         Ok(())
     }
 
@@ -484,6 +513,7 @@ pub fn daemon_init(settings: Settings, config: Config) -> Result<Arc<Mutex<Daemo
         graph: settings.graph,
         commit: settings.commit,
         testing: settings.testing,
+        csv_file: settings.csv_file,
     };
 
     // Attempt to create battery object
@@ -640,6 +670,7 @@ mod tests {
             graph: GraphType::Hidden,
             commit: false,
             testing: true,
+            csv_file: None,
         };
 
         let config = default_config();
@@ -668,6 +699,7 @@ mod tests {
             graph: GraphType::Hidden,
             commit: false,
             testing: true,
+            csv_file: None,
         };
 
         let config = default_config();
@@ -699,6 +731,7 @@ mod tests {
             graph: GraphType::Hidden,
             commit: false,
             testing: true,
+            csv_file: None,
         };
 
         let config = default_config();

src/interactive.rs

@@ -58,6 +58,7 @@ pub fn interactive() {
         graph: GraphType::Hidden,
         commit: false,
         testing: false,
+        csv_file: None,
     };
 
     loop {

src/settings.rs

@@ -43,4 +43,5 @@ pub struct Settings {
     pub graph: GraphType,
     pub commit: bool,
     pub testing: bool,
+    pub csv_file: Option<String>,
 }