1 files changed, 32 insertions, 4 deletions

diff --git a/src/util/mod.rs b/src/util/mod.rs
index a0105ac..f362ff7 100644
--- a/src/util/mod.rs
+++ b/src/util/mod.rs
@@ -3,7 +3,10 @@ use std::process::Stdio;
 use chrono::Duration;
 use lazy_static::lazy_static;
 use log::debug;
-use poise::CreateReply;
+use poise::{
+    CreateReply,
+    FrameworkError,
+};
 use regex::{
     Match,
     Regex,
@@ -35,7 +38,7 @@ use crate::{
 
 mod rest_vec;
 
-pub use rest_vec::RestVec;
+pub use rest_vec::*;
 
 pub async fn currently_playing(ctx: PoiseContext<'_>) -> bool {
     let (_sb, call) = songbird(ctx).await.expect("no songbird");
@@ -61,9 +64,9 @@ pub async fn users_listening(ctx: &Context) -> Result<bool> {
 }
 
 #[inline]
-pub fn msg(ctx: PoiseContext<'_>) -> Option<&Message> {
+pub fn msg<U, E>(ctx: poise::Context<'_, U, E>) -> Option<&Message> {
     match ctx {
-        PoiseContext::Prefix(poise::PrefixContext {
+        poise::Context::Prefix(poise::PrefixContext {
             msg,
             ..
         }) => Some(msg),
@@ -72,6 +75,31 @@ pub fn msg(ctx: PoiseContext<'_>) -> Option<&Message> {
 }
 
 #[inline]
+pub fn err_msg<'a, U, E>(err: &'a FrameworkError<U, E>) -> Option<&'a Message> {
+    use FrameworkError::*;
+
+    if let Some(ctx) = err.ctx() {
+        return msg(ctx);
+    }
+
+    match *err {
+        UnknownCommand {
+            msg,
+            ..
+        }
+        | NonCommandMessage {
+            msg,
+            ..
+        }
+        | DynamicPrefix {
+            msg,
+            ..
+        } => Some(msg),
+        _ => None,
+    }
+}
+
+#[inline]
 pub fn tts(ctx: PoiseContext<'_>) -> Option<bool> {
     msg(ctx).map(|msg| msg.tts)
 }