Bug Fixes

Author: AhiVT

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "pit-tracker"
-version = "0.1.0"
+version = "1.0.0"
 authors = ["Joseph Zurowski <12567900+Dunklheit@users.noreply.github.com>"]
 edition = "2018"
 

Cargo.toml

@@ -26,11 +26,13 @@ impl BotHandler {
     let usr = punished.id.to_user(&ctx).await.unwrap();
     let _ = self.config.warn_channel.send_message(&ctx, |m| {
       m.allowed_mentions(|am| {
-        am.roles(vec![*(self.config.warn_role.id).as_u64()]);
+        am.roles(vec![*(self.config.warn_role).as_u64()]);
 
         am
       });
 
+      m.content(format!("<@&{}>", self.config.warn_role.as_u64()));
+
       m.embed(|e| {
         e.title("Pit Threshold Reached");
         e.description(format!(r#"
@@ -45,7 +47,9 @@ Active Strikes: `{}`
       });
 
       m
-    });
+    })
+      .await
+      .unwrap();
   }
 }
 
@@ -78,6 +82,8 @@ impl EventHandler for BotHandler {
           }
         }
 
+        state.pits += 1;
+
         if user.is_none() {
           state.users.push(PunishedUser {
             id: new.user.id,

src/handler.rs

@@ -5,15 +5,14 @@ use clokwerk::{Scheduler, TimeUnits};
 use serde::{Serialize, Deserialize};
 use serenity::{
   model::{
-    guild::Role,
     id::{
       ChannelId,
       RoleId,
       UserId
     },
   },
   prelude::*,
-};
+framework::StandardFramework};
 
 use std::error::Error;
 use std::fs::File;
@@ -29,7 +28,7 @@ pub struct BotConfig {
   pub settle_time: u16,
   pub(crate) token: String,
   pub warn_channel: ChannelId,
-  pub warn_role: Role,
+  pub warn_role: RoleId,
   pub warn_threshold: u16,
 }
 
@@ -158,13 +157,20 @@ async fn main() {
   let config = Arc::new(read_config().unwrap());
   let state = BotState::new();
 
+  let framework = StandardFramework::new()
+    .configure(|c| c
+      .with_whitespace(true)
+      .prefix("!")
+    );
+
   let mut client = Client::new(&config.token)
     .event_handler(
       handler::BotHandler::new(
         Arc::clone(&state),
         Arc::clone(&config),
       )
     )
+    .framework(framework)
     .await
     .expect("Error creating Discord client");