Index forum support

.env.example

@@ -14,6 +14,9 @@ GITHUB_CLIENT_SECRET=
 # Discord
 
 DISCORD_WEBHOOK_URL=
+DISCORD_BOT_TOKEN=
+DISCORD_GUILD_ID=
+DISCORD_CHANNEL_ID=
 
 # Config
 

src/config.rs

@@ -5,6 +5,7 @@ pub struct AppData {
     front_url: String,
     github: GitHubClientData,
     webhook_url: String,
+    discord: DiscordForumData,
     disable_downloads: bool,
     max_download_mb: u32,
     port: u16,
@@ -17,6 +18,13 @@ pub struct GitHubClientData {
     client_secret: String,
 }
 
+#[derive(Clone)]
+pub struct DiscordForumData {
+    guild_id: u64,
+    channel_id: u64,
+    bot_token: String,
+}
+
 pub async fn build_config() -> anyhow::Result<AppData> {
     let env_url = dotenvy::var("DATABASE_URL")?;
 
@@ -31,6 +39,15 @@ pub async fn build_config() -> anyhow::Result<AppData> {
     let github_client = dotenvy::var("GITHUB_CLIENT_ID").unwrap_or("".to_string());
     let github_secret = dotenvy::var("GITHUB_CLIENT_SECRET").unwrap_or("".to_string());
     let webhook_url = dotenvy::var("DISCORD_WEBHOOK_URL").unwrap_or("".to_string());
+    let guild_id = dotenvy::var("DISCORD_GUILD_ID")
+        .unwrap_or("0".to_string())
+        .parse::<u64>()
+        .unwrap_or(0);
+    let channel_id = dotenvy::var("DISCORD_CHANNEL_ID")
+        .unwrap_or("0".to_string())
+        .parse::<u64>()
+        .unwrap_or(0);
+    let bot_token = dotenvy::var("DISCORD_BOT_TOKEN").unwrap_or("".to_string());
     let disable_downloads =
         dotenvy::var("DISABLE_DOWNLOAD_COUNTS").unwrap_or("0".to_string()) == "1";
     let max_download_mb = dotenvy::var("MAX_MOD_FILESIZE_MB")
@@ -47,6 +64,11 @@ pub async fn build_config() -> anyhow::Result<AppData> {
             client_secret: github_secret,
         },
         webhook_url,
+        discord: DiscordForumData {
+            guild_id,
+            channel_id,
+            bot_token,
+        },
         disable_downloads,
         max_download_mb,
         port,
@@ -64,6 +86,24 @@ impl GitHubClientData {
     }
 }
 
+impl DiscordForumData {
+    pub fn is_valid(&self) -> bool {
+        self.guild_id != 0 && self.channel_id != 0 && !self.bot_token.is_empty()
+    }
+
+    pub fn guild_id(&self) -> u64 {
+        self.guild_id
+    }
+
+    pub fn channel_id(&self) -> u64 {
+        self.channel_id
+    }
+
+    pub fn bot_auth(&self) -> String {
+        format!("Bot {}", self.bot_token)
+    }
+}
+
 impl AppData {
     pub fn db(&self) -> &sqlx::postgres::PgPool {
         &self.db
@@ -85,6 +125,10 @@ impl AppData {
         &self.webhook_url
     }
 
+    pub fn discord(&self) -> &DiscordForumData {
+        &self.discord
+    }
+
     pub fn disable_downloads(&self) -> bool {
         self.disable_downloads
     }

src/endpoints/mod_versions.rs

@@ -11,8 +11,7 @@ use crate::events::mod_created::{
 };
 use crate::webhook::discord::DiscordWebhook;
 use crate::{
-    extractors::auth::Auth,
-    types::{
+    extractors::auth::Auth, forum::discord::create_or_update_thread, types::{
         api::{ApiError, ApiResponse},
         mod_json::{split_version_and_compare, ModJson},
         models::{
@@ -316,6 +315,26 @@ pub async fn create_version(
         }
         .to_discord_webhook()
         .send(&data.webhook_url());
+    } else {
+        tokio::spawn(async move {
+            if !data.discord().is_valid() {
+                log::error!("Discord configuration is not set up. Not creating forum threads.");
+                return;
+            }
+
+            let version_res = ModVersion::get_one(&path.id, &json.version, true, false, &mut pool).await;
+            if version_res.is_err() {
+                return;
+            }
+            create_or_update_thread(
+                None,
+                &data.discord(),
+                &fetched_mod.unwrap(),
+                &version_res.unwrap(),
+                "",
+                &data.app_url()
+            ).await;
+        });
     }
     Ok(HttpResponse::NoContent())
 }
@@ -396,7 +415,7 @@ pub async fn update_version(
                 name: version.name.clone(),
                 version: version.version.clone(),
                 owner,
-                verified_by: dev,
+                verified_by: dev.clone(),
                 base_url: data.app_url().to_string(),
             }
             .to_discord_webhook()
@@ -407,13 +426,39 @@ pub async fn update_version(
                 name: version.name.clone(),
                 version: version.version.clone(),
                 owner,
-                verified: NewModVersionVerification::Admin(dev),
+                verified: NewModVersionVerification::Admin(dev.clone()),
                 base_url: data.app_url().to_string(),
             }
             .to_discord_webhook()
             .send(&data.webhook_url());
         }
     }
 
+    if payload.status == ModVersionStatusEnum::Accepted || payload.status == ModVersionStatusEnum::Rejected {
+        tokio::spawn(async move {
+            if !data.discord().is_valid() {
+                log::error!("Discord configuration is not set up. Not creating forum threads.");
+                return;
+            }
+
+            let mod_res = Mod::get_one(&path.id, false, &mut pool).await.ok().flatten();
+            if mod_res.is_none() {
+                return;
+            }
+            let version_res = ModVersion::get_one(&path.id, &path.version, true, false, &mut pool).await;
+            if version_res.is_err() {
+                return;
+            }
+            create_or_update_thread(
+                None,
+                &data.discord(),
+                &mod_res.unwrap(),
+                &version_res.unwrap(),
+                &dev.display_name,
+                &data.app_url()
+            ).await;
+        });
+    }
+
     Ok(HttpResponse::NoContent())
 }

src/endpoints/mods.rs

@@ -3,11 +3,13 @@ use crate::database::repository::developers;
 use crate::database::repository::mods;
 use crate::events::mod_feature::ModFeaturedEvent;
 use crate::extractors::auth::Auth;
+use crate::forum::discord::create_or_update_thread;
 use crate::types::api::{create_download_link, ApiError, ApiResponse};
 use crate::types::mod_json::ModJson;
 use crate::types::models::incompatibility::Incompatibility;
 use crate::types::models::mod_entity::{download_geode_file, Mod, ModUpdate};
 use crate::types::models::mod_gd_version::{GDVersionEnum, VerPlatform};
+use crate::types::models::mod_version::ModVersion;
 use crate::types::models::mod_version_status::ModVersionStatusEnum;
 use crate::webhook::discord::DiscordWebhook;
 use actix_web::{get, post, put, web, HttpResponse, Responder};
@@ -147,6 +149,30 @@ pub async fn create(
         .await
         .or(Err(ApiError::TransactionError))?;
 
+    tokio::spawn(async move {
+        if !data.discord().is_valid() {
+            log::error!("Discord configuration is not set up. Not creating forum threads.");
+            return;
+        }
+
+        let mod_res = Mod::get_one(&json.id, false, &mut pool).await.ok().flatten();
+        if mod_res.is_none() {
+            return;
+        }
+        let version_res = ModVersion::get_one(&json.id, &json.version, true, false, &mut pool).await;
+        if version_res.is_err() {
+            return;
+        }
+        create_or_update_thread(
+            None,
+            &data.discord(),
+            &mod_res.unwrap(),
+            &version_res.unwrap(),
+            "",
+            &data.app_url()
+        ).await;
+    });
+
     Ok(HttpResponse::NoContent())
 }