added support for language files

src/gen.rs

@@ -0,0 +1,96 @@
+use std::collections::HashSet;
+
+use poise::serenity_prelude::UserId;
+use poise::serenity_prelude as serenity;
+use poise::serenity_prelude::Client;
+
+use crate::{cmds, data, events, re_cmds, rs_println, Args, Data};
+
+
+pub async fn gen_data(args: Args, owners: Vec<u64>) -> Data {
+  let ball_classic_str = std::fs::read_to_string("./data/8-ball_classic.txt").unwrap();
+  let ball_quirk_str   = std::fs::read_to_string("./data/8-ball_quirky.txt").unwrap();
+
+  let ball_classic: Vec<String> = ball_classic_str.lines().map(String::from).collect();
+  let ball_quirk:   Vec<String> = ball_quirk_str  .lines().map(String::from).collect();
+  
+  let mods_env = std::env::var("ASSISTANT_BK_MODS").unwrap_or("0".to_string());
+  let mods_vec_str: Vec<String> = mods_env.split(",").map(String::from).collect();
+  let mods_vec_u64: Vec<u64> = mods_vec_str
+    .iter()
+    .map(|s| s.parse::<u64>().expect("Failed to parse ASSISTANT_BK_MODS. Invalid syntax."))
+    .collect();
+
+  let data = Data {
+    owners,
+    ball_prompts: [ball_classic, ball_quirk],
+    bk_mods:      mods_vec_u64,
+    reddit_data:  None.into(),
+    discord_data: None.into(),
+    cfg:          None.into(),
+    args:         args.clone()
+  };
+
+  data::read_dc_data (&data, args.clone().wipe).await;
+  data::read_re_data (&data, args.clone().wipe).await;
+  data::read_cfg_data(&data, args.clone().wipe).await;
+
+  return data;
+}
+
+
+pub async fn gen_bot(data: Data, args: Args) -> Client {
+  let token =
+    if !args.test { std::env::var("ASSISTANT_TOKEN").expect("Missing ASSISTANT_TOKEN env var!") }
+    else { std::env::var("ASSISTANT_TOKEN_TEST").expect("Missing ASSISTANT_TOKEN_TEST env var!") };
+
+  let intents = serenity::GatewayIntents::all();
+
+  let peek_len = 27;
+  let token_peek = &token[..peek_len];
+  let token_end_len = token[peek_len..].len();
+  rs_println!("Token: {}{}", token_peek, "*".repeat(token_end_len));
+
+  let own: HashSet<UserId> = data.owners.clone().into_iter().map(UserId::from).collect();
+
+  let framework = poise::Framework::builder()
+    .options(poise::FrameworkOptions {
+      owners: own,
+      commands: vec![
+        cmds::help(),
+        cmds::ping(),
+        cmds::embed(),
+        cmds::send(),
+        cmds::stop(),
+        cmds::eight_ball(),
+        cmds::re_shorturl(),
+        cmds::add_server(),
+        // reddit
+        re_cmds::add::cmd(),
+        re_cmds::approve::cmd(),
+        re_cmds::get::cmd(),
+        re_cmds::remove::cmd(),
+        re_cmds::top::cmd(),
+        re_cmds::update::cmd(),
+        re_cmds::vote::cmd(),
+        // reddit admin
+        re_cmds::admin_bind::cmd(),
+        // cfg
+        cmds::reload_cfg()
+      ],
+      event_handler: events::event_handler,
+      ..Default::default()
+    })
+    .setup(|ctx, _ready, framework| {
+      Box::pin(async move {
+        poise::builtins::register_globally(ctx, &framework.options().commands).await?;
+        return Ok(data);
+      })
+    })
+    .build();
+
+  return serenity::ClientBuilder::new(token, intents)
+    .framework(framework)
+    .await
+    .unwrap();
+}