Feature

closes #37

airlock_libs/.gitignore

@@ -1,3 +1,4 @@
 /target
 build.sh
 pythontest.py
+changelog.md

airlock_libs/Cargo.lock

@@ -26,7 +26,7 @@ dependencies = [
 
 [[package]]
 name = "airlock_libs"
-version = "3.1.2"
+version = "4.0.1"
 dependencies = [
  "chrono",
  "indicatif",

airlock_libs/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "airlock_libs"
-version = "3.1.2"
+version = "4.0.1"
 edition = "2024"
 
 [lib]

airlock_libs/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "maturin"
 
 [project]
 name = "airlock_libs"
-version = "3.1.2"
+version = "4.0.1"
 description = "Airlock Digital API Wrapper"
 readme = "README.md"
 license = { text = "AGPL-3.0-only" }

airlock_libs/src/services.rs

@@ -25,12 +25,34 @@ use std::{
     str::FromStr,
 };
 
+#[allow(non_snake_case)]
 #[derive(Deserialize, Debug)]
 struct TelemetryConfig {
     TELEMETRY: bool,
     TELEM_URL: Option<String>,
 }
 
+impl TelemetryConfig {
+    pub fn load() -> Self {
+        let cfg_path = get_base_directory().join("config\\user_config.json");
+        if !cfg_path.exists() {
+            return Self {
+                TELEMETRY: false,
+                TELEM_URL: None,
+            };
+        }
+        match fs::read_to_string(&cfg_path) {
+            Ok(contents) => serde_json::from_str::<Self>(&contents).unwrap_or(Self {
+                TELEMETRY: false,
+                TELEM_URL: None,
+            }),
+            Err(_) => Self {
+                TELEMETRY: false,
+                TELEM_URL: None,
+            },
+        }
+    }
+}
 #[derive(Debug, Deserialize, Serialize)]
 struct ApiResponse {
     error: String,
@@ -86,13 +108,25 @@ pub fn pull_policy_exec_histories(
     )
     .into();
     let writeable_filepath = file_path.clone();
-    if !file_path.exists() {
+    if !&file_path.exists() {
-        if let Some(parent_dir) = file_path.parent()
+        if let Some(parent_dir) = &file_path.parent()
             && !parent_dir.exists()
         {
-            fs::create_dir_all(parent_dir).unwrap();
+            match fs::create_dir_all(parent_dir) {
+                Ok(_) => {}
+                Err(e) => {
+                    println!("Failed to Create Directory {:?}: {}", parent_dir, e);
+                    std::process::abort();
+                }
+            }
+        }
+        match fs::File::create(&file_path) {
+            Ok(_) => {}
+            Err(e) => {
+                println!("Failed to Create Directory {:?}: {}", &file_path, e);
+                std::process::abort();
+            }
         }
-        fs::File::create(file_path).unwrap();
     }
     let data = ApiResponse {
         error: "Success".to_string(),
@@ -135,18 +169,32 @@ pub fn pull_policy_exec_histories(
                 );
                 cx.span()
                     .set_status(Status::error("Client Failed to Build"));
-                panic!("Failed to Build Client: {:?}", client_result);
+                println!("Failed to Build Client: {:?}", client_result);
+                std::process::abort();
             }
         }
     });
     let api: Py<PyAny> = py_self;
     let cutoff = Local::now().naive_local() - Duration::days(days);
-    let mut f = File::open(&writeable_filepath).unwrap();
+    let mut f = match File::open(&writeable_filepath) {
+        Ok(f) => f,
+        Err(e) => {
+            println!("Failed to Access {:?}: {}", &writeable_filepath, e);
+            std::process::abort();
+        }
+    };
     tracer.in_span("Airlock Data Retreival", |cx| {
         let span = cx.span();
         span.set_attribute(Key::new("Days").string(days.to_string().to_string()));
+        span.set_attribute(KeyValue::new("Policy Name", policy_names.clone()));
         loop {
-            f.seek(SeekFrom::Start(0)).unwrap();
+            match f.seek(SeekFrom::Start(0)) {
+                Ok(_) => {}
+                Err(e) => {
+                    println!("Failed to seek start of {:?}: {}", f, e);
+                    std::process::abort();
+                }
+            }
             let execution_histories = tracer.in_span(checkpoint_number.to_string(), |cx| {
                 let results: ApiResponse = history_logging(
                     py,
@@ -334,30 +382,8 @@ fn skipback(days: i64) -> ObjectId {
     ObjectId::parse_str(&objectid_hex).expect("Invalid ObjectId hex")
 }
 
-fn load_telemetry_config() -> TelemetryConfig {
-    let cfg_path = get_base_directory().join("config\\user_config.json");
-    if !cfg_path.exists() {
-        return TelemetryConfig {
-            TELEMETRY: false,
-            TELEM_URL: None,
-        };
-    }
-    match fs::read_to_string(&cfg_path) {
-        Ok(contents) => {
-            serde_json::from_str::<TelemetryConfig>(&contents).unwrap_or(TelemetryConfig {
-                TELEMETRY: false,
-                TELEM_URL: None,
-            })
-        }
-        Err(_) => TelemetryConfig {
-            TELEMETRY: false,
-            TELEM_URL: None,
-        },
-    }
-}
-
 fn init_tracer() -> Result<Option<sdktrace::Tracer>, TraceError> {
-    let cfg = load_telemetry_config();
+    let cfg = TelemetryConfig::load();
     if !cfg.TELEMETRY {
         global::set_tracer_provider(NoopTracerProvider::new());
         return Ok(None);

requirements.txt

@@ -11,4 +11,4 @@ urllib3==2.5.0
 pyperclip==1.11.0
 
 --extra-index-url https://git.racooncity.org/api/packages/brotoskyj/pypi/simple/
-airlock_libs==3.1.2
+airlock_libs==4.0.1