feat: Remove excepts from secrets, drop into idle loop when local (#342)

* feat: Remove excepts from secrets, drop into idle loop when local

* feat: fmt

* fix: use unwrap_or_default, remove ref of option as_ref handles it

Author: astuyve

bottlecap/src/bin/bottlecap/main.rs

@@ -78,7 +78,7 @@ struct RegisterResponse {
     // body, but as a header. Header is extracted and set manually.
     #[serde(skip_deserializing)]
     extension_id: String,
-    account_id: String,
+    account_id: Option<String>,
 }
 
 #[derive(Deserialize)]
@@ -190,13 +190,11 @@ async fn main() -> Result<()> {
 fn load_configs() -> (AwsConfig, Arc<Config>) {
     // First load the configuration
     let aws_config = AwsConfig {
-        region: env::var("AWS_DEFAULT_REGION").expect("AWS_DEFAULT_REGION not set"),
-        aws_access_key_id: env::var("AWS_ACCESS_KEY_ID").expect("AWS_ACCESS_KEY_ID not set"),
-        aws_secret_access_key: env::var("AWS_SECRET_ACCESS_KEY")
-            .expect("AWS_SECRET_ACCESS_KEY not set"),
-        aws_session_token: env::var("AWS_SESSION_TOKEN").expect("AWS_SESSION_TOKEN not set"),
-        function_name: env::var("AWS_LAMBDA_FUNCTION_NAME")
-            .expect("AWS_LAMBDA_FUNCTION_NAME not set"),
+        region: env::var("AWS_DEFAULT_REGION").unwrap_or("us-east-1".to_string()),
+        aws_access_key_id: env::var("AWS_ACCESS_KEY_ID").unwrap_or_default(),
+        aws_secret_access_key: env::var("AWS_SECRET_ACCESS_KEY").unwrap_or_default(),
+        aws_session_token: env::var("AWS_SESSION_TOKEN").unwrap_or_default(),
+        function_name: env::var("AWS_LAMBDA_FUNCTION_NAME").unwrap_or_default(),
     };
     let lambda_directory = env::var("LAMBDA_TASK_ROOT").unwrap_or_else(|_| "/var/task".to_string());
     let config = match config::get_config(Path::new(&lambda_directory)) {
@@ -256,7 +254,11 @@ async fn extension_loop_active(
 ) -> Result<()> {
     let mut event_bus = EventBus::run();
 
-    let tags_provider = setup_tag_provider(aws_config, config, &r.account_id);
+    let tags_provider = setup_tag_provider(
+        aws_config,
+        config,
+        r.account_id.as_ref().unwrap_or(&"none".to_string()),
+    );
     let (logs_agent_channel, logs_flusher) = start_logs_agent(
         config,
         resolved_api_key.clone(),