feat: impl save to file

Author: c-git

.gitignore

@@ -2,3 +2,4 @@
 .shuttle-storage
 Secrets*.toml
 .shuttle*
+KV/

src/config.rs

@@ -48,7 +48,6 @@ impl StartupConfig {
 }
 
 impl SharedConfig {
-    const KEY_VALUE_STORE_FOLDER: &str = "KV";
     pub fn try_new(secret_store: &SecretStore) -> anyhow::Result<&'static Self> {
         let auth_role_id = KeyName::AuthRoleId.get_non_secret_parse(secret_store)?;
         let channel_unranked = KeyName::ChannelUnrankedId.get_non_secret_parse(secret_store)?;

src/db.rs

@@ -1,48 +1,67 @@
+use std::{fs, io::Write as _, path::PathBuf};
+
+use tracing::error;
+
+const KEY_VALUE_STORE_FOLDER: &str = "KV";
+
+/// Returns a path if able to be created else logs error and returns `None`
+fn get_file_path(key: &str) -> Option<PathBuf> {
+    let mut result = PathBuf::from(KEY_VALUE_STORE_FOLDER);
+    match std::fs::create_dir_all(&result) {
+        Ok(()) => {}
+        Err(err_msg) => {
+            error!(
+                ?err_msg,
+                "Failed to create parent directory for key value store: {result:?}"
+            );
+            return None;
+        }
+    };
+    result = result.join(key);
+    if !result.add_extension("json") {
+        error!("Unable to add json extension to path: {result:?}");
+    }
+    Some(result)
+}
+
 pub async fn save_kv(key: &str, value: String) {
-    todo!("save to disk")
-    //     let query = sqlx::query!(
-    //         "\
-    // INSERT INTO kv_store (id, content)
-    //     VALUES ($1, $2)
-    //     ON CONFLICT(id)
-    //     DO UPDATE SET
-    //     content = EXCLUDED.content;",
-    //         key,
-    //         value
-    //     );
-    //     match query.execute(pool).await {
-    //         Ok(query_result) => {
-    //             if query_result.rows_affected() == 1 {
-    //                 debug!("Save completed for key: {key}");
-    //             } else {
-    //                 error!(
-    //                     ?key,
-    //                     "Expected 1 row to be affected by save but got: {}",
-    //                     query_result.rows_affected()
-    //                 )
-    //             }
-    //         }
-    //         Err(err_msg) => error!(
-    //             ?err_msg,
-    //             "Failed to save content for key: {key} to kv store"
-    //         ),
-    //     }
+    let Some(path) = get_file_path(key) else {
+        return;
+    };
+
+    let mut file = match std::fs::OpenOptions::new()
+        .write(true)
+        .create(true)
+        .truncate(true)
+        .open(path)
+    {
+        Ok(file) => file,
+        Err(err_msg) => {
+            error!(
+                ?err_msg,
+                "Failed to save content for key: {key} to kv store (file creation failed)"
+            );
+            return;
+        }
+    };
+    match file.write_all(value.as_bytes()) {
+        Ok(_) => {}
+        Err(err_msg) => {
+            error!(
+                ?err_msg,
+                "Failed to save content for key: {key} to kv store (write failed)"
+            );
+        }
+    };
 }
 
 pub async fn load_kv(key: &str) -> Option<String> {
-    todo!("load from disk")
-    // match sqlx::query!("SELECT content FROM kv_store where id = $1", key)
-    //     .fetch_optional(pool)
-    //     .await
-    // {
-    //     Ok(Some(record)) => Some(record.content),
-    //     Ok(None) => {
-    //         info!("No content found in DB for key: {key}");
-    //         None
-    //     }
-    //     Err(err_msg) => {
-    //         error!(?err_msg, "Failed to get content for key: {key}");
-    //         None
-    //     }
-    // }
+    let path = get_file_path(key)?;
+    match fs::read_to_string(path) {
+        Ok(content) => Some(content),
+        Err(err_msg) => {
+            error!(?err_msg, "Failed to get content for key: {key}");
+            None
+        }
+    }
 }