From 51340beaa7d024ba187d1b2a377b9d8bd226c7c8 Mon Sep 17 00:00:00 2001
From: phoenix <kundeng00@pm.me>
Date: Thu, 29 May 2025 19:40:42 -0400
Subject: [PATCH] Migrating code to icarus_envy

---
 src/lib.rs  | 11 +++++++++--
 src/main.rs |  6 +++---
 2 files changed, 12 insertions(+), 5 deletions(-)

diff --git a/src/lib.rs b/src/lib.rs
index ca334c5..99380ae 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -4,6 +4,7 @@ pub mod hashing;
 pub mod repo;
 pub mod token_stuff;
 
+/*
 pub mod keys {
     pub const DBURL: &str = "DATABASE_URL";
 
@@ -11,6 +12,7 @@ pub mod keys {
         pub const ERROR: &str = "DATABASE_URL must be set in .env";
     }
 }
+*/
 
 mod connection_settings {
     pub const MAXCONN: u32 = 5;
@@ -21,10 +23,13 @@ pub mod db {
     use sqlx::postgres::PgPoolOptions;
     use std::env;
 
-    use crate::{connection_settings, keys};
+    use icarus_envy::keys;
+
+    // use crate::{connection_settings, keys};
+    use crate::connection_settings;
 
     pub async fn create_pool() -> Result<sqlx::PgPool, sqlx::Error> {
-        let database_url = get_db_url().await;
+        let database_url = icarus_envy::environment::get_db_url().await;
         println!("Database url: {:?}", database_url);
 
         PgPoolOptions::new()
@@ -33,11 +38,13 @@ pub mod db {
             .await
     }
 
+    /*
     async fn get_db_url() -> String {
         #[cfg(debug_assertions)] // Example: Only load .env in debug builds
         dotenvy::dotenv().ok();
         env::var(keys::DBURL).expect(keys::error::ERROR)
     }
+    */
 
     pub async fn migrations(pool: &sqlx::PgPool) {
         // Run migrations using the sqlx::migrate! macro
diff --git a/src/main.rs b/src/main.rs
index 27b8bb1..68293ba 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -75,7 +75,7 @@ mod tests {
 
         pub async fn get_pool() -> Result<sqlx::PgPool, sqlx::Error> {
             dotenvy::dotenv().ok(); // Load .env file if it exists
-            let tm_db_url = std::env::var(keys::DBURL).expect("DATABASE_URL must be present");
+            let tm_db_url = std::env::var(icarus_envy::keys::DBURL).expect("DATABASE_URL must be present");
             let tm_options = sqlx::postgres::PgConnectOptions::from_str(&tm_db_url).unwrap();
             sqlx::PgPool::connect_with(tm_options).await
         }
@@ -87,7 +87,7 @@ mod tests {
         }
 
         pub async fn connect_to_db(db_name: &str) -> Result<sqlx::PgPool, sqlx::Error> {
-            let db_url = std::env::var(keys::DBURL).expect("DATABASE_URL must be set for tests");
+            let db_url = std::env::var(icarus_envy::keys::DBURL).expect("DATABASE_URL must be set for tests");
             let options = sqlx::postgres::PgConnectOptions::from_str(&db_url)?.database(db_name);
             sqlx::PgPool::connect_with(options).await
         }
@@ -116,7 +116,7 @@ mod tests {
         pub fn get_database_name() -> Result<String, Box<dyn std::error::Error>> {
             dotenvy::dotenv().ok(); // Load .env file if it exists
 
-            match std::env::var(keys::DBURL) {
+            match std::env::var(icarus_envy::keys::DBURL) {
                 Ok(database_url) => {
                     let parsed_url = url::Url::parse(&database_url)?;
                     if parsed_url.scheme() == "postgres" || parsed_url.scheme() == "postgresql" {