Overhaul refactor. Still things in progress

api/src/main.rs

@@ -1,12 +1,10 @@
-extern crate diesel;
-#[macro_use]
-extern crate diesel_migrations;
-
 use std::env;
 
 use actix_cors::Cors;
 use actix_web::{App, HttpServer, middleware::Logger};
-use dotenv::dotenv;
+use dotenv::from_filename;
+use crate::auth::hash;
+use crate::users::{User, ADMIN_ROLE};
 
 mod airports;
 mod auth;
@@ -17,15 +15,41 @@ mod scheduler;
 mod users;
 
 #[actix_web::main]
-async fn main() -> std::io::Result<()> {
-  dotenv().ok();
-  env_logger::init_from_env(env_logger::Env::default().filter_or("RUST_LOG", "warn,api=info"));
-  db::init().await;
+async fn main() -> Result<(), Box<dyn std::error::Error>> {
+  initialize_environment()?;
+  db::initialize().await?;
   // scheduler::update_airports();
 
   let host = env::var("API_HOST").unwrap_or("localhost".to_string());
   let port = env::var("API_PORT").unwrap_or("5000".to_string());
 
+  // Initialize admin user
+  let admin_username = env::var("ADMIN_USERNAME");
+  let admin_password = env::var("ADMIN_PASSWORD");
+  if admin_username.is_ok() && admin_password.is_ok() {
+    let username = admin_username.unwrap();
+    if User::select(&username).await.is_none() {
+      log::debug!("Creating default administrator");
+      let password = admin_password.unwrap();
+      let password_hash = hash(&password)?;
+      let admin_user = User {
+        email: username,
+        password_hash,
+        role: ADMIN_ROLE.to_string(),
+        first_name: "Admin".to_string(),
+        last_name: "".to_string(),
+        updated_at: Default::default(),
+        created_at: Default::default(),
+      };
+      match admin_user.insert().await {
+        Ok(_) => log::debug!("Default administrator was successfully created"),
+        Err(err) => {
+          log::warn!("{}", err);
+        }
+      };
+    }
+  }
+
   let server = match HttpServer::new(move || {
     let cors = Cors::default()
       .allow_any_origin()
@@ -49,9 +73,35 @@ async fn main() -> std::io::Result<()> {
     }
     Err(err) => {
       log::error!("Could not bind server: {}", err);
-      return Err(err);
+      return Err(err.into());
     }
   };
 
-  server.run().await
+  if let Err(err) = server.run().await {
+    return Err(err.into());
+  }
+  Ok(())
+}
+
+fn initialize_environment() -> std::io::Result<()> {
+  // Iterate over files in the current directory
+  for entry in std::fs::read_dir(".")? {
+    let entry = entry?;
+    let path = entry.path();
+
+    // Check if the file name starts with ".env" and is a file
+    if let Some(file_name) = path.file_name().and_then(|n| n.to_str()) {
+      if file_name.starts_with(".env") && path.is_file() {
+        // Try to load the file
+        if let Err(err) = from_filename(&file_name) {
+          eprintln!("Failed to load {}: {}", file_name, err);
+        } else {
+          println!("Loaded: {}", file_name);
+        }
+      }
+    }
+  }
+
+  env_logger::init_from_env(env_logger::Env::default().filter_or("RUST_LOG", "warn,api=info"));
+  Ok(())
 }