feat: basic filemanager query API

lib/workload/stateless/stacks/filemanager/filemanager-build/src/error.rs

@@ -1,15 +1,17 @@
 //! This module contains the crate's error types.
 //!
 
-use crate::error::ErrorKind::{IoError, LoadingEnvironment};
-use crate::workspace_path;
-use miette::{diagnostic, Diagnostic, NamedSource, SourceOffset};
 use std::fmt::{Display, Formatter};
 use std::fs::read_to_string;
 use std::panic::Location;
 use std::{io, result};
+
+use miette::{diagnostic, Diagnostic, NamedSource, SourceOffset};
 use thiserror::Error;
 
+use crate::error::ErrorKind::{IoError, LoadingEnvironment};
+use crate::workspace_path;
+
 pub type Result<T> = result::Result<T, Error>;
 
 /// Error types for the filemanager.

lib/workload/stateless/stacks/filemanager/filemanager-build/src/gen_entities.rs

@@ -2,14 +2,16 @@
 //! database schema.
 //!
 
-use crate::error::ErrorKind::EntityGeneration;
-use crate::error::{Error, Result};
-use crate::Config;
+use std::ffi::OsStr;
+use std::fs::write;
+
 use clap_builder::Parser;
 use quote::quote;
 use sea_orm_cli::{run_generate_command, Cli, Commands};
-use std::ffi::OsStr;
-use std::fs::write;
+
+use crate::error::ErrorKind::EntityGeneration;
+use crate::error::{Error, Result};
+use crate::Config;
 
 pub async fn generate_entities() -> Result<()> {
     let config = Config::load()?;
@@ -19,6 +21,14 @@ pub async fn generate_entities() -> Result<()> {
         "sea-orm-cli",
         "generate",
         "entity",
+        "--with-serde",
+        "both",
+        "--enum-extra-derives",
+        "strum::FromRepr",
+        "--enum-extra-derives",
+        "strum::EnumCount",
+        "--enum-extra-attributes",
+        "repr(u8)",
         "-u",
         &config.database_url,
         "-o",
@@ -44,6 +54,5 @@ pub async fn generate_entities() -> Result<()> {
 
     let out_file = out_dir.join("generated.rs");
     write(out_file, format!("{}\n\n{}", generated_comment, generated))?;
-
     Ok(())
 }

lib/workload/stateless/stacks/filemanager/filemanager-build/src/lib.rs

@@ -1,21 +1,26 @@
-use crate::error::Error;
-use crate::error::ErrorKind::LoadingEnvironment;
-use envy::from_env;
-use error::Result;
-use serde::Deserialize;
 use std::env::var;
 use std::path::{Path, PathBuf};
 
+use envy::from_env;
+use serde::Deserialize;
+
+use error::Result;
+
+use crate::error::Error;
+use crate::error::ErrorKind::LoadingEnvironment;
+
 pub mod error;
 pub mod gen_entities;
 
+/// Configuration environment variables for the build process.
 #[derive(Debug, Deserialize)]
 pub struct Config {
     database_url: String,
     out_dir: PathBuf,
 }
 
 impl Config {
+    /// Load environment variables into a `Config` struct.
     #[track_caller]
     pub fn load() -> Result<Self> {
         Ok(from_env::<Config>()?)

lib/workload/stateless/stacks/filemanager/filemanager-http-lambda/src/main.rs

@@ -1,3 +1,4 @@
+use filemanager::env::Config;
 use lambda_http::Error;
 use lambda_runtime::{run, service_fn, LambdaEvent};
 
@@ -11,15 +12,17 @@ use filemanager::handlers::init_tracing;
 async fn main() -> Result<(), Error> {
     init_tracing();
 
-    let options = &create_database_pool().await?;
+    let config = &Config::load()?;
+    let options = &create_database_pool(config).await?;
     run(service_fn(|_: LambdaEvent<()>| async move {
-        update_credentials(options).await?;
+        update_credentials(options, config).await?;
 
         receive_and_ingest(
             S3Client::with_defaults().await,
             SQSClient::with_defaults().await,
             None::<String>,
-            DbClient::from_ref(options),
+            DbClient::new(options.clone()),
+            config,
         )
         .await?;
 

lib/workload/stateless/stacks/filemanager/filemanager-ingest-lambda/src/main.rs

@@ -3,21 +3,24 @@ use lambda_runtime::{run, service_fn, Error, LambdaEvent};
 
 use filemanager::clients::aws::s3::Client;
 use filemanager::database::Client as DbClient;
+use filemanager::env::Config;
 use filemanager::handlers::aws::{create_database_pool, ingest_event, update_credentials};
 use filemanager::handlers::init_tracing;
 
 #[tokio::main]
 async fn main() -> Result<(), Error> {
     init_tracing();
 
-    let options = &create_database_pool().await?;
+    let config = &Config::load()?;
+    let options = &create_database_pool(config).await?;
     run(service_fn(|event: LambdaEvent<SqsEvent>| async move {
-        update_credentials(options).await?;
+        update_credentials(options, config).await?;
 
         ingest_event(
             event.payload,
             Client::with_defaults().await,
-            DbClient::from_ref(options),
+            DbClient::new(options.clone()),
+            config,
         )
         .await?;
 

lib/workload/stateless/stacks/filemanager/filemanager-inventory-lambda/src/main.rs

@@ -3,6 +3,7 @@ use lambda_runtime::{run, service_fn, Error, LambdaEvent};
 use serde::Deserialize;
 
 use filemanager::database::Client as DbClient;
+use filemanager::env::Config;
 use filemanager::events::aws::inventory::Manifest;
 use filemanager::handlers::aws::{create_database_pool, ingest_s3_inventory, update_credentials};
 use filemanager::handlers::init_tracing;
@@ -26,12 +27,13 @@ pub struct BucketKey {
 async fn main() -> Result<(), Error> {
     init_tracing();
 
-    let options = &create_database_pool().await?;
+    let config = &Config::load()?;
+    let options = &create_database_pool(config).await?;
     run(service_fn(|event: LambdaEvent<Request>| async move {
-        update_credentials(options).await?;
+        update_credentials(options, config).await?;
 
         let client = Client::with_defaults().await;
-        let database = DbClient::from_ref(options);
+        let database = DbClient::new(options.clone());
 
         match event.payload {
             Request::BucketKey(bucket_key) => {
@@ -41,11 +43,12 @@ async fn main() -> Result<(), Error> {
                     Some(bucket_key.bucket),
                     Some(bucket_key.key),
                     None,
+                    config,
                 )
                 .await?
             }
             Request::Manifest(manifest) => {
-                ingest_s3_inventory(client, database, None, None, Some(manifest)).await?
+                ingest_s3_inventory(client, database, None, None, Some(manifest), config).await?
             }
         };
 

lib/workload/stateless/stacks/filemanager/filemanager-migrate-lambda/src/main.rs

@@ -7,6 +7,7 @@ use crate::Event::Provider;
 use filemanager::database::aws::migration::Migration;
 use filemanager::database::Client as DbClient;
 use filemanager::database::Migrate;
+use filemanager::env::Config;
 use filemanager::handlers::aws::{create_database_pool, update_credentials};
 use filemanager::handlers::init_tracing;
 
@@ -32,9 +33,10 @@ pub enum CloudFormationRequest {
 async fn main() -> Result<(), Error> {
     init_tracing();
 
-    let options = &create_database_pool().await?;
+    let config = &Config::load()?;
+    let options = &create_database_pool(config).await?;
     run(service_fn(|event: LambdaEvent<Event>| async move {
-        update_credentials(options).await?;
+        update_credentials(options, config).await?;
 
         // Migrate depending on the type of lifecycle event using the CDK provider framework:
         // https://docs.aws.amazon.com/cdk/api/v2/docs/aws-cdk-lib.custom_resources-readme.html#provider-framework
@@ -47,7 +49,7 @@ async fn main() -> Result<(), Error> {
             _ => {
                 // If there's nothing to migrate, then this will just return Ok.
                 Ok::<_, Error>(
-                    Migration::new(DbClient::from_ref(options))
+                    Migration::new(DbClient::new(options.clone()))
                         .migrate()
                         .await?,
                 )