Spells endpoints

src/main.rs

@@ -6,6 +6,7 @@ use std::collections::{HashSet, HashMap};
 use std::env;
 use std::sync::Arc;
 
+use actix_web::{HttpServer, App};
 use commands::audio::{create_response, AudioConfig, AudioConfigs};
 
 use dotenv::dotenv;
@@ -111,60 +112,86 @@ impl EventHandler for Handler {
   }
 }
 
-#[tokio::main]
-async fn main() {
+#[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,siren=info"));
-
-  let token: String = env::var("DISCORD_TOKEN").expect("Expected a token in the environment");
-  let intents: GatewayIntents = GatewayIntents::all();
-
-  let http: Http = Http::new(&token);
-  let (owners, _bot_id) = match http.get_current_application_info().await {
-    Ok(info) => {
-      let mut owners: HashSet<serenity::model::id::UserId> = HashSet::new();
-      if let Some(team) = info.team {
-        owners.insert(team.owner_user_id);
-      } else {
-        owners.insert(info.owner.id);
-      }
-      match http.get_current_user().await {
-        Ok(bot) => (owners, bot.id),
-        Err(why) => panic!("Could not access the bot id: {:?}", why)
-      }
-    },
-    Err(why) => panic!("Could not access application info: {:?}", why)
-  };
-
   db::init();
-  
-  let handler = match env::var("OPENAI_API_KEY") {
-    Ok(token) => {
-      info!("Loaded OpenAI token");
-      Handler {
-        oai: Some(commands::oai::OAI { client: reqwest::Client::new(), base_url: "https://api.openai.com/v1".to_string(), max_attempts: 5, token , max_context_questions: 15 })
-      }
-    }
+  db::load_data();
+
+  // setup_discord_bot();
+
+  let host = env::var("SERVICE_HOST").unwrap_or("localhost".to_string());
+  let port = env::var("SERVICE_PORT").unwrap_or("5000".to_string());
+
+  match HttpServer::new(|| {
+    App::new()
+      .configure(db::spells::init_routes)
+  })
+  .bind(format!("{}:{}", host, port)) {
+    Ok(b) => {
+      info!("Binding server to {}:{}", host, port);
+      b
+    },
     Err(err) => {
-      warn!("Could not load OpenAI token: {}", err);
-      Handler { oai: None }
+      error!("Could not bind server: {}", err);
+      return Err(err);
     }
-  };
-
-  let mut client = Client::builder(token, intents)
-    .event_handler(handler)
-    .framework(StandardFramework::new()
-    .configure(|c| c.owners(owners)))
-    .register_songbird()
-    .await
-    .expect("Error creating client");
-
-  {
-    let mut data = client.data.write().await;
-    data.insert::<AudioConfigs>(Arc::new(RwLock::new(HashMap::default())));
-  }
-
-  if let Err(why) = client.start_autosharded().await {
-    error!("An error occurred while running the client: {:?}", why);
   }
+  .run()
+  .await
+}
+
+fn setup_discord_bot() {
+  tokio::spawn(async {
+    let token: String = env::var("DISCORD_TOKEN").expect("Expected a token in the environment");
+    let intents: GatewayIntents = GatewayIntents::all();
+
+    let http: Http = Http::new(&token);
+    let (owners, _bot_id) = match http.get_current_application_info().await {
+      Ok(info) => {
+        let mut owners: HashSet<serenity::model::id::UserId> = HashSet::new();
+        if let Some(team) = info.team {
+          owners.insert(team.owner_user_id);
+        } else {
+          owners.insert(info.owner.id);
+        }
+        match http.get_current_user().await {
+          Ok(bot) => (owners, bot.id),
+          Err(why) => panic!("Could not access the bot id: {:?}", why)
+        }
+      },
+      Err(why) => panic!("Could not access application info: {:?}", why)
+    };
+    
+    let handler = match env::var("OPENAI_API_KEY") {
+      Ok(token) => {
+        info!("Loaded OpenAI token");
+        Handler {
+          oai: Some(commands::oai::OAI { client: reqwest::Client::new(), base_url: "https://api.openai.com/v1".to_string(), max_attempts: 5, token , max_context_questions: 15 })
+        }
+      }
+      Err(err) => {
+        warn!("Could not load OpenAI token: {}", err);
+        Handler { oai: None }
+      }
+    };
+
+    let mut client = Client::builder(token, intents)
+      .event_handler(handler)
+      .framework(StandardFramework::new()
+      .configure(|c| c.owners(owners)))
+      .register_songbird()
+      .await
+      .expect("Error creating client");
+
+    {
+      let mut data = client.data.write().await;
+      data.insert::<AudioConfigs>(Arc::new(RwLock::new(HashMap::default())));
+    }
+
+    if let Err(why) = client.start_autosharded().await {
+      error!("An error occurred while running the client: {:?}", why);
+    }
+  });
 }