supports config loading

Cargo.toml

@@ -5,7 +5,9 @@ edition = "2021"
 
 [dependencies]
 anyhow = "1.0.102"
+poise = "0.6.2"
 reqwest = {version = "0.13.3", features = ["json"]}
 serde = {version = "1.0.215", features = ["derive", "serde_derive"]}
+serde_json = "1.0.149"
 tokio = {version = "1.41.1", features = ["full"]}
 url = "2.5.3"

config.json

@@ -0,0 +1,10 @@
+{
+    "version": "0.1.0",
+    "entries": [
+        {
+            "handler_type": "minecraft",
+            "address": "https://dawn.shibedrill.site",
+            "token": "foobar"
+        }
+    ]
+}

src/bot_runner.rs

@@ -0,0 +1,7 @@
+use crate::types::*;
+use poise;
+
+struct BotRunner {
+    client: poise::serenity_prelude::Client,
+    server_data: dyn ServerInfo,
+}

src/config_parser.rs

@@ -0,0 +1,35 @@
+use std::{fs::File, path::Path};
+use crate::{handlers, types::ServerInfo};
+use serde::Deserialize;
+use url::Url;
+
+#[derive(Deserialize)]
+pub struct ConfigEntry {
+    handler_type: String,
+    address: Url,
+    token: String,
+}
+
+#[derive(Deserialize)]
+pub struct Config {
+    version: String,
+    entries: Vec<ConfigEntry>,
+}
+
+pub fn parse_configs(path: &Path) -> Result<Config, anyhow::Error> {
+    let file_handle = File::open(path)?;
+    Ok(serde_json::from_reader::<File, Config>(file_handle)?)
+}
+
+pub fn build_handlers(conf: Config) -> Vec<Box<dyn ServerInfo>> {
+    let mut results: Vec<Box<dyn ServerInfo>> = vec!();
+    for item in conf.entries {
+        match item.handler_type.as_str() {
+                "minecraft" => {
+                    results.push(Box::new(handlers::minecraft::Server::new(item.token, item.address)))
+                },
+                _ => {}
+        }
+    }
+    results
+}

src/handlers/minecraft.rs

@@ -1,8 +1,8 @@
-use reqwest::{header::HeaderValue, Client};
 use serde::Deserialize;
+use serde_json;
 use url::Url;
 
-use crate::types::{PlayerList, ServerInfo, ServerOnlineResponse, ServerResponse};
+use crate::types::{ServerInfo, ServerOnlineResponse, ServerResponse};
 
 #[derive(Deserialize, Debug)]
 struct ApiResponse {
@@ -46,34 +46,6 @@ pub struct OnlineResponse {
     players: Vec<String>,
 }
 
-impl ServerOnlineResponse for OnlineResponse {
-    fn players_online(&self) -> u64 {
-        self.players_online
-    }
-
-    fn player_limit(&self) -> u64 {
-        self.player_limit
-    }
-
-    fn version(&self) -> &String {
-        &self.version
-    }
-
-    fn readable_name(&self) -> &String {
-        &self.clean_name
-    }
-
-    fn searchable_name(&self) -> &String {
-        &self.searchable_name
-    }
-}
-
-impl PlayerList for OnlineResponse {
-    fn players(&self) -> &Vec<String> {
-        &self.players
-    }
-}
-
 pub struct Server {
     #[allow(dead_code)]
     token: String,
@@ -85,56 +57,42 @@ impl ServerInfo for Server {
         Server { token, addr }
     }
 
-    type OnlineResponse = OnlineResponse;
-
-    type AddressableName = Url;
-
-    fn addressable_name(&self) -> &Self::AddressableName {
-        &self.addr
+    fn addressable_name(&self) -> String {
+        self.addr.clone().into()
     }
 
     fn app_token(&self) -> String {
         self.token.clone()
     }
 
-    async fn poll(&self) -> ServerResponse<Self::OnlineResponse> {
-        use reqwest;
+    fn api_address(&self) -> Url {
         let url_string = format!("https://api.mcsrvstat.us/3/{}", self.addr.host().unwrap());
-        let url = Url::try_from(url_string.as_str()).unwrap();
-        let mut request = reqwest::Request::new(reqwest::Method::GET, url);
-        request.headers_mut().append("User-Agent", HeaderValue::from_str("Mozilla/5.0 (Linux; Android 10; K) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/143.0.0.0 Mobile Safari/537.36").unwrap());
+        Url::parse(&url_string).unwrap()
+    }
 
-        let client = Client::new();
-
-        match client.execute(request).await {
-            Err(error) => ServerResponse::Error(anyhow::anyhow!(error)),
-            Ok(answer) => {
-                let json = answer.json::<ApiResponse>().await.unwrap();
-                //println!("{}", answer.text().await.unwrap());
-                //ServerResponse::Offline
-
-                match json.online {
-                    true => {
-                        let players = json.players.unwrap();
-                        let motd = json.motd.as_ref().unwrap();
-                        self::ServerResponse::Online(OnlineResponse {
-                            searchable_name: self.addr.host().unwrap().to_string(),
-                            reported_name: motd.clean.first().unwrap().into(),
-                            clean_name: motd.clean.first().unwrap().into(),
-                            players_online: players.online,
-                            player_limit: players.max,
-                            version: json.version.unwrap(),
-                            players: players
-                                .list
-                                .unwrap_or(Vec::<ApiResponsePlayerEntry>::new())
-                                .iter()
-                                .map(|e| e.name.clone())
-                                .collect(),
-                        })
-                    }
-                    false => ServerResponse::Offline,
-                }
+    fn parse(&self, response: String) -> ServerResponse {
+        let parsed_data: ApiResponse = serde_json::from_str(&response).unwrap();
+        match parsed_data.online {
+            true => {
+                let players = parsed_data.players.unwrap();
+                let motd = parsed_data.motd.as_ref().unwrap();
+                self::ServerResponse::Online(ServerOnlineResponse {
+                    searchable_name: self.addr.host().unwrap().to_string(),
+                    readable_name: motd.clean.first().unwrap().into(),
+                    reported_name: motd.clean.first().unwrap().into(),
+                    clean_name: motd.clean.first().unwrap().into(),
+                    players_online: players.online,
+                    player_limit: players.max,
+                    version: parsed_data.version.unwrap(),
+                    players: Some(players
+                        .list
+                        .unwrap_or_default()
+                        .iter()
+                        .map(|e| e.name.clone())
+                        .collect()),
+                })
             }
+            false => ServerResponse::Offline,
         }
     }
 }

src/main.rs

@@ -1,31 +1,35 @@
+mod bot_runner;
+mod config_parser;
 mod handlers;
 mod types;
+mod request;
+
+use std::path::Path;
 
 use crate::{
-    handlers::minecraft,
-    types::{PlayerList, ServerInfo, ServerOnlineResponse, ServerResponse},
+    request::request, types::ServerResponse
 };
 
 #[tokio::main]
 async fn main() {
-    let mc = minecraft::Server::new(
-        "foo".into(),
-        "http://dawn.shibedrill.site".try_into().unwrap(),
-    );
-    let results = mc.poll().await;
+    
+    let config = config_parser::parse_configs(Path::new("config.json")).unwrap();
+    let handlers = config_parser::build_handlers(config);
+
+    let http_response = request(handlers[0].api_address()).await.unwrap();
+    let results = handlers[0].parse(http_response.text().await.unwrap());
     match results {
-        ServerResponse::Error(err) => println!("Error: {}", err),
         ServerResponse::Offline => println!("Offline"),
         ServerResponse::Online(online_info) => {
             println!(
                 "Name: {}\nAddress: {}\nVersion: {}\nPlayers: {}/{}",
-                online_info.readable_name(),
-                online_info.searchable_name(),
-                online_info.version(),
-                online_info.players_online(),
-                online_info.player_limit()
+                online_info.readable_name,
+                online_info.searchable_name,
+                online_info.version,
+                online_info.players_online,
+                online_info.player_limit
             );
-            println!("Players online: {}", online_info.players().join(", "));
+            println!("Players online: {}", online_info.players.unwrap().join(", "));
         }
     }
 }

src/request.rs

@@ -0,0 +1,9 @@
+use url::Url;
+use reqwest::{Response, header::HeaderValue};
+
+pub async fn request(url: Url) -> Result<Response, reqwest::Error> {
+    let client = reqwest::Client::new();
+    let mut request = reqwest::Request::new(reqwest::Method::GET, url);
+    request.headers_mut().append("User-Agent", HeaderValue::from_str("Mozilla/5.0 (Linux; Android 10; K) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/143.0.0.0 Mobile Safari/537.36").unwrap());
+    client.execute(request).await
+}

src/types.rs

@@ -1,37 +1,38 @@
 use url::Url;
 
 #[derive(Debug)]
-pub enum ServerResponse<T: ServerOnlineResponse> {
+pub enum ServerResponse {
     Offline,
-    Online(T),
-    Error(anyhow::Error),
+    Online(ServerOnlineResponse),
 }
 
-pub trait ServerOnlineResponse {
-    fn players_online(&self) -> u64;
-    fn player_limit(&self) -> u64;
-    fn version(&self) -> &String;
-    fn searchable_name(&self) -> &String;
-    fn readable_name(&self) -> &String;
-}
-
-pub trait PlayerList {
-    fn players(&self) -> &Vec<String>;
+#[derive(Debug)]
+pub struct ServerOnlineResponse {
+    pub players_online: u64,
+    pub player_limit: u64,
+    pub version: String,
+    pub searchable_name: String,
+    pub readable_name: String,
+    pub clean_name: String,
+    pub reported_name: String,
+    pub players: Option<Vec<String>>,
 }
 
 #[allow(dead_code)]
 pub trait ServerInfo {
-    fn new(token: String, addr: Url) -> Self;
+    fn new(token: String, addr: Url) -> Self
+    where
+        Self: Sized;
 
     /// An addressable, unique identifier, used for its API.
     /// If it is a Minecraft server, this can be a fully qualified domain name or URL.
     /// If this is an SCP secret laboratory server, it can be a server ID.
-    type AddressableName;
-    fn addressable_name(&self) -> &Self::AddressableName;
+    fn addressable_name(&self) -> String;
+    /// The address wrapped in the API URL.
+    fn api_address(&self) -> Url;
+    fn parse(&self, response: String) -> ServerResponse;
 
     /// The app token used to send status updates.
     fn app_token(&self) -> String;
 
-    type OnlineResponse: ServerOnlineResponse;
-    async fn poll(&self) -> ServerResponse<Self::OnlineResponse>;
 }