From e1dcb22fe0867f75ad0e1cefd5c9769ac39bd193 Mon Sep 17 00:00:00 2001
From: Jeremiah Parrack <jeremiahlukus1@gmail.com>
Date: Fri, 14 Jun 2024 09:21:03 -0400
Subject: [PATCH] revert to v1

---
 src/main.rs | 16 +++++-----------
 src/ssm.rs  | 53 +++++------------------------------------------------
 2 files changed, 10 insertions(+), 59 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 31da961..f43be6c 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -10,15 +10,10 @@ const ENV_FILE: &str = "/tmp/crypteia.json";
 #[tokio::main]
 async fn main() -> Result<(), Error> {
     log::cloudwatch_metric("main", "initialized", false, None);
-    
-    if std::env::var("AWS_LAMBDA_FUNCTION_NAME").is_err() {
-        log::cloudwatch_metric("main", "running_outside_lambda", false, None);
-        println!("Running outside of AWS Lambda environment.");
-    }
     let env_vars: HashMap<String, String> = std::env::vars().collect();
-    let env_map = ssm::get_envs(env_vars).await?;
+    let env_map = ssm::get_envs(env_vars).await.unwrap();
     log::cloudwatch_metric("main", "fetched", false, None);
-    write_envs_to_tmp_json(env_map)?;
+    write_envs_to_tmp_json(env_map);
     let func = service_fn(parameters_extension);
     lambda_extension::run(func).await
 }
@@ -33,9 +28,8 @@ async fn parameters_extension(event: LambdaEvent) -> Result<(), Error> {
     Ok(())
 }
 
-fn write_envs_to_tmp_json(env_map: HashMap<String, String>) -> Result<(), std::io::Error> {
+fn write_envs_to_tmp_json(env_map: HashMap<String, String>) {
     let envs_json = serde_json::to_string(&env_map).unwrap();
-    let mut file = File::create(ENV_FILE)?;
-    file.write_all(envs_json.as_bytes())?;
-    Ok(())
+    let mut file = File::create(ENV_FILE).unwrap();
+    file.write_all(envs_json.as_bytes()).unwrap();
 }
\ No newline at end of file
diff --git a/src/ssm.rs b/src/ssm.rs
index 953db3f..6ff97fc 100644
--- a/src/ssm.rs
+++ b/src/ssm.rs
@@ -4,7 +4,6 @@ use futures::future::join_all;
 use std::collections::HashMap;
 use tokio::{spawn, task::JoinHandle};
 
-
 pub async fn get_envs(env_vars: HashMap<String, String>) -> Result<HashMap<String, String>> {
     let sdk_config = aws_config::load_from_env().await;
     let ssm_client: aws_sdk_ssm::Client = aws_sdk_ssm::Client::new(&sdk_config);
@@ -32,9 +31,9 @@ pub async fn get_envs(env_vars: HashMap<String, String>) -> Result<HashMap<Strin
                         results.insert(key, value);
                     });
                 }
-                Err(error) => return Err(anyhow::anyhow!(format!("Parameter not found: {}", error))), // Return error if parameter is not found
+                Err(error) => log::cloudwatch_metric("ssm", "error", true, Some(error.to_string())),
             },
-            Err(error) => return Err(anyhow::anyhow!(error.to_string())), // Return error if task fails
+            Err(error) => log::cloudwatch_metric("ssm", "error", true, Some(error.to_string())),
         }
     }
     Ok(results)
@@ -56,8 +55,6 @@ async fn ssm_get_parameter(
         Ok(response) => {
             if let Some(parameter) = response.parameter {
                 items.insert(name, parameter.value.unwrap());
-            } else {
-                return Err(anyhow::anyhow!("Parameter not found: {}", path));
             }
         }
         Err(error) => {
@@ -70,7 +67,6 @@ async fn ssm_get_parameter(
                     name, path, error
                 )),
             );
-            return Err(anyhow::anyhow!(error.to_string())); // Return error
         }
     }
     Ok(items)
@@ -104,7 +100,7 @@ async fn ssm_get_parameters_by_path(
                         items.insert(env_name, parameter.value.unwrap());
                     }
                 }
-                if response.next_token.is_none() {
+                if response.next_token == None {
                     break;
                 }
                 token = response.next_token;
@@ -119,16 +115,14 @@ async fn ssm_get_parameters_by_path(
                         name, path, error
                     )),
                 );
-                return Err(anyhow::anyhow!(error.to_string())); // Return error
+                break;
             }
         }
     }
-    if items.is_empty() {
-        return Err(anyhow::anyhow!("Parameters not found for path: {}", path));
-    }
     Ok(items)
 }
 
+
 #[cfg(test)]
 mod test {
     use super::*;
@@ -208,41 +202,4 @@ mod test {
         assert_eq!(results, expected);
         Ok(())
     }
-
-    #[tokio::test]
-    async fn should_fail_if_param_not_found() -> Result<()> {
-        let sdk_config = aws_config::load_from_env().await;
-        let ssm_client = aws_sdk_ssm::Client::new(&sdk_config);
-        ssm_client
-            .put_parameter()
-            .name("/crypteia/v5/myapp/SECRET2".to_owned())
-            .value("1A2B3C4D5E6F".to_owned())
-            .r#type(ParameterType::SecureString)
-            .overwrite(true)
-            .send()
-            .await?;
-        let env_vars: HashMap<String, String> = HashMap::from([
-            ("EXISTING".to_string(), "existingvalue".to_string()),
-            (
-                "SECRET2".to_string(),
-                "x-crypteia-ssm:/crypteia/v5/myapp/SECRET".to_string(),
-            ),
-            ("NON_EXISTENT_PARAM".to_string(), "x-crypteia-ssm:/crypteia/v5/myapp/NON_EXISTENT_PARAM".to_string()),
-        ]);
-        let result = get_envs(env_vars).await;
-        assert!(result.is_err(), "Expected an error when parameter is not found");
-        Ok(())
-    }
-
- 
-
-    #[tokio::test]
-    async fn should_fail_if_param_not_found_in_path() -> Result<()> {
-        let sdk_config = aws_config::load_from_env().await;
-        let ssm_client = aws_sdk_ssm::Client::new(&sdk_config);
-        let result = ssm_get_parameters_by_path(&ssm_client, "NON_EXISTENT_PARAM".to_string(), "/crypteia/v5/myapp/non_existent_path".to_string()).await;
-        assert!(result.is_err(), "Expected an error when parameters are not found for the given path");
-        
-        Ok(())
-    }
 }
\ No newline at end of file