feat: add store tracer methods

Author: alehander92

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "trace_types"
-version = "0.5.0"
+version = "0.5.1"
 edition = "2021"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

src/lib.rs

@@ -1,4 +1,3 @@
-use std::path::PathBuf;
 mod tracer;
 mod types;
 pub use crate::tracer::{Tracer, NONE_TYPE_ID, NONE_VALUE};
@@ -7,6 +6,7 @@ pub use crate::types::*;
 #[cfg(test)]
 mod tests {
     use super::*;
+    use std::path::PathBuf;
 
     #[test]
     fn simple_trace() {
@@ -22,13 +22,16 @@ mod tests {
 
         let int_value = ValueRecord::Int {
             i: 1,
-            type_id: tracer.ensure_type_id(TypeKind::Int, "Int")
+            type_id: tracer.ensure_type_id(TypeKind::Int, "Int"),
         };
         tracer.register_variable_with_full_value("test_variable", int_value);
         tracer.register_return(NONE_VALUE);
         assert_eq!(tracer.events.len(), 14);
         // visible with
         // cargo tets -- --nocapture
-        println!("{:#?}", tracer.events);
+        // println!("{:#?}", tracer.events);
+
+        // tracer.store_trace_metadata(&PathBuf::from("trace_metadata.json")).unwrap();
+        // tracer.store_trace_events(&PathBuf::from("trace.json")).unwrap();
     }
 }

src/tracer.rs

@@ -1,11 +1,14 @@
-use crate::types::{
-    ArgRecord, CallRecord, EventLogKind, FullValueRecord, FunctionId, FunctionRecord, Line, PathId, RecordEvent, ReturnRecord, StepRecord,
-    TraceLowLevelEvent, TypeId, ValueRecord, VariableId, TypeSpecificInfo, TypeRecord, TypeKind,
-};
 use std::collections::HashMap;
 use std::env;
+use std::error::Error;
+use std::fs;
 use std::path::{Path, PathBuf};
 
+use crate::types::{
+    ArgRecord, CallRecord, EventLogKind, FullValueRecord, FunctionId, FunctionRecord, Line, PathId, RecordEvent, ReturnRecord, StepRecord,
+    TraceLowLevelEvent, TraceMetadata, TypeId, TypeKind, TypeRecord, TypeSpecificInfo, ValueRecord, VariableId,
+};
+
 pub struct Tracer {
     // trace metadata:
     workdir: PathBuf,
@@ -33,7 +36,7 @@ pub const NONE_VALUE: ValueRecord = ValueRecord::None { type_id: NONE_TYPE_ID };
 impl Tracer {
     pub fn new(program: &str, args: &[String]) -> Self {
         Tracer {
-            workdir: env::current_dir().unwrap(),
+            workdir: env::current_dir().expect("can access the current dir"),
             program: program.to_string(),
             args: args.to_vec(),
             events: vec![],
@@ -46,7 +49,6 @@ impl Tracer {
     }
 
     pub fn start(&mut self, path: &PathBuf, line: Line) {
-        let path_id = self.ensure_path_id(path);
         let function_id = self.ensure_function_id("<toplevel>", path, line);
         self.register_call(function_id, vec![]);
 
@@ -144,12 +146,24 @@ impl Tracer {
     }
 
     pub fn register_full_value(&mut self, variable_id: VariableId, value: ValueRecord) {
-        self.events.push(
-            TraceLowLevelEvent::Value(FullValueRecord {
-                variable_id,
-                value,
-            })
-        );
+        self.events.push(TraceLowLevelEvent::Value(FullValueRecord { variable_id, value }));
+    }
+
+    pub fn store_trace_metadata(&mut self, path: &PathBuf) -> Result<(), Box<dyn Error>> {
+        let trace_metadata = TraceMetadata {
+            program: self.program.clone(),
+            args: self.args.clone(),
+            workdir: self.workdir.clone(),
+        };
+        let json = serde_json::to_string(&trace_metadata)?;
+        fs::write(path, json)?;
+        Ok(())
     }
 
+    pub fn store_trace_events(&mut self, path: &PathBuf) -> Result<(), Box<dyn Error>> {
+        // TODO: probably change format
+        let json = serde_json::to_string(&self.events)?;
+        fs::write(path, json)?;
+        Ok(())
+    }
 }