Use API for welcome modal

src/main.rs

@@ -8,6 +8,7 @@ use sass_rocket_fairing::SassFairing;
 use std::env;
 
 mod models;
+use models::Settings;
 
 mod utils;
 
@@ -30,7 +31,9 @@ async fn rocket() -> _ {
         env::var("SECRET")
             .expect("Must have SECRET defined in .env for private cookie persistence"),
     ));
+
     rocket::custom(config)
+        .manage(Settings::new(&http).await.unwrap())
         .manage(http)
         .mount(
             "/",

src/models/mod.rs

@@ -3,3 +3,6 @@ pub use user::User;
 
 mod challenge;
 pub use challenge::Challenge;
+
+mod settings;
+pub use settings::Settings;

src/models/settings.rs

@@ -0,0 +1,56 @@
+use derive_more::From;
+use poise::serenity_prelude::Http;
+use serde::{Deserialize, Serialize};
+use std::fs;
+
+#[derive(Serialize, Deserialize)]
+pub struct Settings {
+    pub guilds: Vec<Guild>,
+}
+
+impl Settings {
+    pub async fn new(http: &Http) -> Result<Self, GetSettingsError> {
+        let file = fs::read_to_string("settings.yaml").unwrap();
+        let mut settings: Self = serde_yaml::from_str(&file)?;
+        settings.load(http).await?;
+        Ok(settings)
+    }
+
+    pub async fn load(&mut self, http: &Http) -> poise::serenity_prelude::Result<()> {
+        for guild in self.guilds.iter_mut() {
+            guild.load(http).await?;
+        }
+        Ok(())
+    }
+}
+
+#[derive(From, Debug)]
+pub enum GetSettingsError {
+    Io(std::io::Error),
+    Deserialize(serde_yaml::Error),
+    Serenity(poise::serenity_prelude::Error),
+}
+
+#[derive(Serialize, Deserialize)]
+pub struct Guild {
+    pub id: u64,
+    #[serde(default)]
+    pub hidden: bool,
+    #[serde(default)]
+    pub recommended: bool,
+    pub channel: u64,
+    // None case means guild is not yet loaded
+    #[serde(default)]
+    pub icon: Option<Option<String>>,
+    #[serde(default)]
+    pub name: Option<String>,
+}
+
+impl Guild {
+    pub async fn load(&mut self, http: &Http) -> poise::serenity_prelude::Result<()> {
+        let server = http.get_guild(self.id).await?;
+        self.icon = Some(server.icon);
+        self.name = Some(server.name);
+        Ok(())
+    }
+}

src/routes/get_challenge.rs

@@ -1,10 +1,12 @@
-use rocket::http::CookieJar;
+use std::ops::Deref;
+
+use rocket::{http::CookieJar, State};
 use rocket_dyn_templates::{context, Template};
 
-use crate::models::{Challenge, User};
+use crate::models::{Challenge, User, Settings};
 
 #[get("/<challenge>")]
-pub async fn get_challenge(challenge: u32, cookies: &CookieJar<'_>) -> Template {
+pub async fn get_challenge(challenge: u32, cookies: &CookieJar<'_>, settings: &State<Settings>) -> Template {
     println!(
         "{:?}",
         cookies
@@ -15,6 +17,7 @@ pub async fn get_challenge(challenge: u32, cookies: &CookieJar<'_>) -> Template
         "index",
         context! {
             challenge,
+            settings: settings.deref(),
             user: User::get(cookies).await.unwrap(),
             content: Challenge::get(challenge),
         },