add CONFIG_PATH (#11)

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "boot_bot"
-version = "0.1.0"
+version = "0.1.1"
 edition = "2021"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
@@ -12,7 +12,6 @@ linfa = "0.7.0"
 linfa-trees = { version = "0.7.0", features = ["serde"] }
 ndarray = "0.15.6"
 reqwest = { version="0.11.24", features = ["blocking"] }
-openssl = { version = "0.10", features = ["vendored"] }
 serde_json = "1.0.113"
 serde = "1.0.196"
-bincode = "1.3.3"
+bincode = "1.3.3"

README.md

@@ -1,11 +1,20 @@
-requires environment variables:
+# requires environment variables:
 
 RIOT_TOKEN              The token for the Riot API
+
 DISCORD_TOKEN           The token for the Discord API
+
 VIP_USER                Snowflake of the user that may execute configuration commands
+
 VIP_GUILD               Snowflake of the guild that has configuration commands
+
 GAME_VERSION            The LoL version used for the embed downloads
 
-needs the following files:
+CONFIG_PATH             The base path for the files listed below
+
+
+# needs the following files:
+
+snowflake_puuid.txt     Maps Discord Snowflakes to Riot PUUIDs, each on a new line, separated by pipe symbols
 
-snowflake_puuid.txt     Maps Discord Snowflakes to Riot PUUIDs, each on a new line, separated by pipe symbols
+model.bin               The DecisionTree. Can be recreated by the bot

src/prediction.rs

@@ -58,12 +58,14 @@ fn train_decision_tree(data: &Array2<f32>, sq: SplitQuality) -> DecisionTree<f32
 }
 
 fn save_model(model: &DecisionTree<f32, String>, file_name: &str) {
-    let model_file = File::create(file_name).unwrap();
+    let base_path = env::var("CONFIG_PATH").unwrap_or(String::new());
+    let model_file = File::create(format!("{}{}", base_path, file_name)).unwrap();
     bincode::serialize_into(&model_file, &model).unwrap();
 }
 
 fn load_model(file_name: &str) -> Option<DecisionTree<f32, String>> {
-    if let Ok(mut model_file) = File::open(file_name) {
+    let base_path = env::var("CONFIG_PATH").unwrap_or(String::new());
+    if let Ok(mut model_file) = File::open(format!("{}{}", base_path, file_name)) {
         let mut buffer = Vec::new();
         model_file.read_to_end(&mut buffer).unwrap();
         let deserialized_model: DecisionTree<f32, String> = bincode::deserialize(&buffer).unwrap();
@@ -86,7 +88,8 @@ fn load_model(file_name: &str) -> Option<DecisionTree<f32, String>> {
 pub fn recreate_model(game_count: usize) {
     let token: &str = &env::var("RIOT_TOKEN")
         .expect("Could not fetch the Riot token");
-    let snowflake_map = create_snowflake_puuid_map(crate::constants::MAPPING_FILE);
+    let base_path = env::var("CONFIG_PATH").unwrap_or(String::new());
+    let snowflake_map = create_snowflake_puuid_map(&format!("{}{}", base_path, crate::constants::MAPPING_FILE));
     let test_puuid = snowflake_map.values().filter(|id| id.starts_with("f7Xz")).nth(0).unwrap().clone();
 
     // Train a new model
@@ -107,7 +110,8 @@ pub fn recreate_model(game_count: usize) {
 pub fn predict(snowflake: &str) -> Option<String> {
     let token: &str = &env::var("RIOT_TOKEN")
         .expect("Could not fetch the Riot token");
-    let snowflake_map = create_snowflake_puuid_map(crate::constants::MAPPING_FILE);
+    let base_path = env::var("CONFIG_PATH").unwrap_or(String::new());
+    let snowflake_map = create_snowflake_puuid_map(&format!("{}{}", base_path, crate::constants::MAPPING_FILE));
 
     if let Some(puuid) = snowflake_map.get(snowflake) {
         if let Some(model) = load_model(MODEL_FILE_NAME) {