feat: server side aspects of adding minecraft whitelist added

src/commands/add_minecraft.rs

@@ -0,0 +1,180 @@
+use serenity::{
+  builder::CreateApplicationCommand,
+  client::Context,
+  model::{
+    application::interaction::application_command::ApplicationCommandInteraction,
+    prelude::{command::CommandOptionType, interaction::application_command::CommandDataOptionValue},
+  },
+};
+
+use skynet_discord_bot::{get_server_config, DataBase, Servers};
+use sqlx::{Pool, Sqlite};
+
+pub(crate) mod user {}
+
+pub(crate) mod server {
+  use serde::{Deserialize, Serialize};
+  use serenity::model::id::GuildId;
+  use sqlx::Error;
+  // this is to managfe the server side of commands related to minecraft
+  use super::*;
+  use skynet_discord_bot::set_roles::get_server_member_bulk;
+  use skynet_discord_bot::{is_admin, Config};
+
+  pub fn register(command: &mut CreateApplicationCommand) -> &mut CreateApplicationCommand {
+    command
+      .name("add_minecraft")
+      .description("Enable the bot for this discord")
+      .create_option(|option| {
+        option
+          .name("server_id")
+          .description("ID of the Minecraft server hosted by the Computer Society")
+          .kind(CommandOptionType::String)
+          .required(true)
+      })
+  }
+
+  pub async fn run(command: &ApplicationCommandInteraction, ctx: &Context) -> String {
+    // check if user has high enough permisssions
+    if let Some(msg) = is_admin(command, ctx).await {
+      return msg;
+    }
+    let g_id = match command.guild_id {
+      None => return "Not in a server".to_string(),
+      Some(x) => x,
+    };
+
+    let server_minecraft = if let CommandDataOptionValue::String(id) = command
+      .data
+      .options
+      .get(0)
+      .expect("Expected server_id option")
+      .resolved
+      .as_ref()
+      .expect("Expected server_id object")
+    {
+      Some(id.to_owned())
+    } else {
+      None
+    };
+
+    let db_lock = {
+      let data_read = ctx.data.read().await;
+      data_read.get::<DataBase>().expect("Expected Databse in TypeMap.").clone()
+    };
+    let db = db_lock.read().await;
+
+    let server_data = match get_server_config(&db, &g_id).await {
+      None => {
+        return "No existing server config, have you used ``/add``?".to_string();
+      }
+      Some(mut x) => {
+        x.server_minecraft = server_minecraft.clone();
+        x
+      }
+    };
+
+    match add_server(&db, &server_data).await {
+      Ok(_) => {}
+      Err(e) => {
+        println!("{:?}", e);
+        return format!("Failure to insert into Servers {:?}", server_data);
+      }
+    }
+
+    let config_lock = {
+      let data_read = ctx.data.read().await;
+      data_read.get::<Config>().expect("Expected Config in TypeMap.").clone()
+    };
+    let config = config_lock.read().await;
+
+    update_server(server_minecraft, &db, &g_id, &config).await;
+
+    "Added/Updated minecraft_server info".to_string()
+  }
+
+  async fn add_server(db: &Pool<Sqlite>, server: &Servers) -> Result<Option<Servers>, Error> {
+    sqlx::query_as::<_, Servers>(
+      "
+      UPDATE servers
+      SET server_minecraft = ?2
+      WHERE server = ?1; 
+    ",
+    )
+    .bind(*server.server.as_u64() as i64)
+    .bind(&server.server_minecraft)
+    .fetch_optional(db)
+    .await
+  }
+
+  /**
+  loop through all members of server
+  get a list of folks with mc accounts that are members
+  and a list that arent members
+  */
+  async fn update_server(server_minecraft: Option<String>, db: &Pool<Sqlite>, g_id: &GuildId, config: &Config) {
+    if let Some(server_id) = server_minecraft {
+      let mut usernames = vec![];
+      for member in get_server_member_bulk(db, g_id).await {
+        if let Some(x) = member.minecraft {
+          usernames.push(x);
+        }
+      }
+      if !usernames.is_empty() {
+        update_whitelist(&usernames, &server_id, &config.discord_minecraft).await;
+      }
+    }
+  }
+
+  pub async fn update_whitelist(add: &Vec<String>, server: &str, token: &str) {
+    let url_base = format!("http://panel.games.skynet.ie/api/client/servers/{server}");
+    let bearer = format!("Bearer {token}");
+
+    async fn post<T: Serialize>(url: &str, bearer: &str, data: &T) {
+      match surf::post(url)
+        .header("Authorization", bearer)
+        .header("Content-Type", "application/json")
+        .header("Accept", "Application/vnd.pterodactyl.v1+json")
+        .body_json(&data)
+      {
+        Ok(req) => {
+          req.await.ok();
+        }
+        Err(e) => {
+          dbg!(e);
+        }
+      }
+    }
+
+    #[derive(Deserialize, Serialize, Debug)]
+    struct BodyCommand {
+      command: String,
+    }
+
+    #[derive(Deserialize, Serialize, Debug)]
+    struct BodyDelete {
+      root: String,
+      files: Vec<String>,
+    }
+
+    // delete whitelist
+    let deletion = BodyDelete {
+      root: "/".to_string(),
+      files: vec!["whitelist.json".to_string()],
+    };
+    post(&format!("{url_base}/files/delete"), &bearer, &deletion).await;
+
+    // reload the whitelist
+    let data = BodyCommand {
+      command: "whitelist reload".to_string(),
+    };
+    post(&format!("{url_base}/command"), &bearer, &data).await;
+
+    for name in add {
+      let data = BodyCommand {
+        command: format!("whitelist add {name}"),
+      };
+      post(&format!("{url_base}/command"), &bearer, &data).await;
+    }
+  }
+}

src/commands/add_server.rs

@@ -7,12 +7,12 @@ use serenity::{
   },
 };
 use skynet_discord_bot::get_data::get_wolves;
-use skynet_discord_bot::{get_server_config, set_roles::update_server, DataBase, Servers, is_admin};
+use skynet_discord_bot::{get_server_config, is_admin, set_roles::update_server, DataBase, Servers};
 use sqlx::{Error, Pool, Sqlite};
 
 pub async fn run(command: &ApplicationCommandInteraction, ctx: &Context) -> String {
   // check if user has high enough permisssions
-  if let Some(msg) = is_admin(command, ctx).await{
+  if let Some(msg) = is_admin(command, ctx).await {
     return msg;
   }
 
@@ -64,6 +64,8 @@ pub async fn run(command: &ApplicationCommandInteraction, ctx: &Context) -> Stri
     role_current,
     member_past: 0,
     member_current: 0,
+    // this gets added later
+    server_minecraft: None,
   };
 
   match add_server(&db, ctx, &server_data).await {
@@ -109,16 +111,22 @@ async fn add_server(db: &Pool<Sqlite>, ctx: &Context, server: &Servers) -> Resul
   let role_past = server.role_past.map(|x| *x.as_u64() as i64);
   let role_current = server.role_current.map(|x| *x.as_u64() as i64);
 
+  let server_minecraft = match get_server_config(db, &server.server).await {
+    None => None,
+    Some(x) => x.server_minecraft,
+  };
+
   let insert = sqlx::query_as::<_, Servers>(
     "
-      INSERT OR REPLACE INTO servers (server, wolves_api, role_past, role_current)
-      VALUES (?1, ?2, ?3, ?4)
+      INSERT OR REPLACE INTO servers (server, wolves_api, role_past, role_current, server_minecraft)
+      VALUES (?1, ?2, ?3, ?4, ?5)
     ",
   )
   .bind(*server.server.as_u64() as i64)
   .bind(&server.wolves_api)
   .bind(role_past)
   .bind(role_current)
+  .bind(server_minecraft)
   .fetch_optional(db)
   .await;
 

src/commands/mod.rs

@@ -1,2 +1,3 @@
+pub mod add_minecraft;
 pub mod add_server;
 pub mod link_email;