Merge pull request #7 from reminder-bot/poise-2

Poise 2

.prettierrc.toml

@@ -0,0 +1,2 @@
+printWidth = 90
+tabWidth = 4

Cargo.toml

@@ -5,8 +5,8 @@ authors = ["jellywx <judesouthworth@pm.me>"]
 edition = "2018"
 
 [dependencies]
+poise = "0.2"
 dotenv = "0.15"
-humantime = "2.1"
 tokio = { version = "1", features = ["process", "full"] }
 reqwest = "0.11"
 regex = "1.4"
@@ -25,28 +25,8 @@ levenshtein = "1.0"
 sqlx = { version = "0.5", features = ["runtime-tokio-rustls", "macros", "mysql", "bigdecimal", "chrono"]}
 base64 = "0.13.0"
 
-[dependencies.regex_command_attr]
-path = "command_attributes"
-
 [dependencies.postman]
 path = "postman"
 
 [dependencies.reminder_web]
 path = "web"
-
-[dependencies.serenity]
-git = "https://github.com/serenity-rs/serenity"
-branch = "next"
-default-features = false
-features = [
-    "builder",
-    "client",
-    "cache",
-    "gateway",
-    "http",
-    "model",
-    "utils",
-    "rustls_backend",
-    "collector",
-    "unstable_discord_api"
-]

Rocket.toml

@@ -1,7 +1,8 @@
 [default]
-address = "127.0.0.1"
+address = "0.0.0.0"
 port = 5000
 template_dir = "web/templates"
+limits = { json = "10MiB" }
 
 [debug]
 secret_key = "tR8krio5FXTnnyIZNiJDXPondz0kI1v6X6BXZcBGIRY="

command_attributes/Cargo.toml

@@ -1,16 +0,0 @@
-[package]
-name = "regex_command_attr"
-version = "0.3.6"
-authors = ["acdenisSK <acdenissk69@gmail.com>", "jellywx <judesouthworth@pm.me>"]
-edition = "2018"
-description = "Procedural macros for command creation for the Serenity library."
-license = "ISC"
-
-[lib]
-proc-macro = true
-
-[dependencies]
-quote = "^1.0"
-syn = { version = "^1.0", features = ["full", "derive", "extra-traits"] }
-proc-macro2 = "1.0"
-uuid = { version = "0.8", features = ["v4"] }

command_attributes/src/attributes.rs

@@ -1,351 +0,0 @@
-use std::fmt::{self, Write};
-
-use proc_macro2::Span;
-use syn::{
-    parse::{Error, Result},
-    spanned::Spanned,
-    Attribute, Ident, Lit, LitStr, Meta, NestedMeta, Path,
-};
-
-use crate::{
-    structures::{ApplicationCommandOptionType, Arg},
-    util::{AsOption, LitExt},
-};
-
-#[derive(Debug, Clone, Copy, PartialEq)]
-pub enum ValueKind {
-    // #[<name>]
-    Name,
-
-    // #[<name> = <value>]
-    Equals,
-
-    // #[<name>([<value>, <value>, <value>, ...])]
-    List,
-
-    // #[<name>([<prop> = <value>, <prop> = <value>, ...])]
-    EqualsList,
-
-    // #[<name>(<value>)]
-    SingleList,
-}
-
-impl fmt::Display for ValueKind {
-    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
-        match self {
-            ValueKind::Name => f.pad("`#[<name>]`"),
-            ValueKind::Equals => f.pad("`#[<name> = <value>]`"),
-            ValueKind::List => f.pad("`#[<name>([<value>, <value>, <value>, ...])]`"),
-            ValueKind::EqualsList => {
-                f.pad("`#[<name>([<prop> = <value>, <prop> = <value>, ...])]`")
-            }
-            ValueKind::SingleList => f.pad("`#[<name>(<value>)]`"),
-        }
-    }
-}
-
-fn to_ident(p: Path) -> Result<Ident> {
-    if p.segments.is_empty() {
-        return Err(Error::new(p.span(), "cannot convert an empty path to an identifier"));
-    }
-
-    if p.segments.len() > 1 {
-        return Err(Error::new(p.span(), "the path must not have more than one segment"));
-    }
-
-    if !p.segments[0].arguments.is_empty() {
-        return Err(Error::new(p.span(), "the singular path segment must not have any arguments"));
-    }
-
-    Ok(p.segments[0].ident.clone())
-}
-
-#[derive(Debug)]
-pub struct Values {
-    pub name: Ident,
-    pub literals: Vec<(Option<String>, Lit)>,
-    pub kind: ValueKind,
-    pub span: Span,
-}
-
-impl Values {
-    #[inline]
-    pub fn new(
-        name: Ident,
-        kind: ValueKind,
-        literals: Vec<(Option<String>, Lit)>,
-        span: Span,
-    ) -> Self {
-        Values { name, literals, kind, span }
-    }
-}
-
-pub fn parse_values(attr: &Attribute) -> Result<Values> {
-    fn is_list_or_named_list(meta: &NestedMeta) -> ValueKind {
-        match meta {
-            // catch if the nested value is a literal value
-            NestedMeta::Lit(_) => ValueKind::List,
-            // catch if the nested value is a meta value
-            NestedMeta::Meta(m) => match m {
-                // path => some quoted value
-                Meta::Path(_) => ValueKind::List,
-                Meta::List(_) | Meta::NameValue(_) => ValueKind::EqualsList,
-            },
-        }
-    }
-
-    let meta = attr.parse_meta()?;
-
-    match meta {
-        Meta::Path(path) => {
-            let name = to_ident(path)?;
-
-            Ok(Values::new(name, ValueKind::Name, Vec::new(), attr.span()))
-        }
-        Meta::List(meta) => {
-            let name = to_ident(meta.path)?;
-            let nested = meta.nested;
-
-            if nested.is_empty() {
-                return Err(Error::new(attr.span(), "list cannot be empty"));
-            }
-
-            if is_list_or_named_list(nested.first().unwrap()) == ValueKind::List {
-                let mut lits = Vec::with_capacity(nested.len());
-
-                for meta in nested {
-                    match meta {
-                        // catch if the nested value is a literal value
-                        NestedMeta::Lit(l) => lits.push((None, l)),
-                        // catch if the nested value is a meta value
-                        NestedMeta::Meta(m) => match m {
-                            // path => some quoted value
-                            Meta::Path(path) => {
-                                let i = to_ident(path)?;
-                                lits.push((None, Lit::Str(LitStr::new(&i.to_string(), i.span()))))
-                            }
-                            Meta::List(_) | Meta::NameValue(_) => {
-                                return Err(Error::new(attr.span(), "cannot nest a list; only accept literals and identifiers at this level"))
-                            }
-                        },
-                    }
-                }
-
-                let kind = if lits.len() == 1 { ValueKind::SingleList } else { ValueKind::List };
-
-                Ok(Values::new(name, kind, lits, attr.span()))
-            } else {
-                let mut lits = Vec::with_capacity(nested.len());
-
-                for meta in nested {
-                    match meta {
-                        // catch if the nested value is a literal value
-                        NestedMeta::Lit(_) => {
-                            return Err(Error::new(attr.span(), "key-value pairs expected"))
-                        }
-                        // catch if the nested value is a meta value
-                        NestedMeta::Meta(m) => match m {
-                            Meta::NameValue(n) => {
-                                let name = to_ident(n.path)?.to_string();
-                                let value = n.lit;
-
-                                lits.push((Some(name), value));
-                            }
-                            Meta::List(_) | Meta::Path(_) => {
-                                return Err(Error::new(attr.span(), "key-value pairs expected"))
-                            }
-                        },
-                    }
-                }
-
-                Ok(Values::new(name, ValueKind::EqualsList, lits, attr.span()))
-            }
-        }
-        Meta::NameValue(meta) => {
-            let name = to_ident(meta.path)?;
-            let lit = meta.lit;
-
-            Ok(Values::new(name, ValueKind::Equals, vec![(None, lit)], attr.span()))
-        }
-    }
-}
-
-#[derive(Debug, Clone)]
-struct DisplaySlice<'a, T>(&'a [T]);
-
-impl<'a, T: fmt::Display> fmt::Display for DisplaySlice<'a, T> {
-    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
-        let mut iter = self.0.iter().enumerate();
-
-        match iter.next() {
-            None => f.write_str("nothing")?,
-            Some((idx, elem)) => {
-                write!(f, "{}: {}", idx, elem)?;
-
-                for (idx, elem) in iter {
-                    f.write_char('\n')?;
-                    write!(f, "{}: {}", idx, elem)?;
-                }
-            }
-        }
-
-        Ok(())
-    }
-}
-
-#[inline]
-fn is_form_acceptable(expect: &[ValueKind], kind: ValueKind) -> bool {
-    if expect.contains(&ValueKind::List) && kind == ValueKind::SingleList {
-        true
-    } else {
-        expect.contains(&kind)
-    }
-}
-
-#[inline]
-fn validate(values: &Values, forms: &[ValueKind]) -> Result<()> {
-    if !is_form_acceptable(forms, values.kind) {
-        return Err(Error::new(
-            values.span,
-            // Using the `_args` version here to avoid an allocation.
-            format_args!("the attribute must be in of these forms:\n{}", DisplaySlice(forms)),
-        ));
-    }
-
-    Ok(())
-}
-
-#[inline]
-pub fn parse<T: AttributeOption>(values: Values) -> Result<T> {
-    T::parse(values)
-}
-
-pub trait AttributeOption: Sized {
-    fn parse(values: Values) -> Result<Self>;
-}
-
-impl AttributeOption for Vec<String> {
-    fn parse(values: Values) -> Result<Self> {
-        validate(&values, &[ValueKind::List])?;
-
-        Ok(values.literals.into_iter().map(|(_, l)| l.to_str()).collect())
-    }
-}
-
-impl AttributeOption for String {
-    #[inline]
-    fn parse(values: Values) -> Result<Self> {
-        validate(&values, &[ValueKind::Equals, ValueKind::SingleList])?;
-
-        Ok(values.literals[0].1.to_str())
-    }
-}
-
-impl AttributeOption for bool {
-    #[inline]
-    fn parse(values: Values) -> Result<Self> {
-        validate(&values, &[ValueKind::Name, ValueKind::SingleList])?;
-
-        Ok(values.literals.get(0).map_or(true, |(_, l)| l.to_bool()))
-    }
-}
-
-impl AttributeOption for Ident {
-    #[inline]
-    fn parse(values: Values) -> Result<Self> {
-        validate(&values, &[ValueKind::SingleList])?;
-
-        Ok(values.literals[0].1.to_ident())
-    }
-}
-
-impl AttributeOption for Vec<Ident> {
-    #[inline]
-    fn parse(values: Values) -> Result<Self> {
-        validate(&values, &[ValueKind::List])?;
-
-        Ok(values.literals.into_iter().map(|(_, l)| l.to_ident()).collect())
-    }
-}
-
-impl AttributeOption for Option<String> {
-    fn parse(values: Values) -> Result<Self> {
-        validate(&values, &[ValueKind::Name, ValueKind::Equals, ValueKind::SingleList])?;
-
-        Ok(values.literals.get(0).map(|(_, l)| l.to_str()))
-    }
-}
-
-impl AttributeOption for Arg {
-    fn parse(values: Values) -> Result<Self> {
-        validate(&values, &[ValueKind::EqualsList])?;
-
-        let mut arg: Arg = Default::default();
-
-        for (key, value) in &values.literals {
-            match key {
-                Some(s) => match s.as_str() {
-                    "name" => {
-                        arg.name = value.to_str();
-                    }
-                    "description" => {
-                        arg.description = value.to_str();
-                    }
-                    "required" => {
-                        arg.required = value.to_bool();
-                    }
-                    "kind" => arg.kind = ApplicationCommandOptionType::from_str(value.to_str()),
-                    _ => {
-                        return Err(Error::new(key.span(), "unexpected attribute"));
-                    }
-                },
-                _ => {
-                    return Err(Error::new(key.span(), "unnamed attribute"));
-                }
-            }
-        }
-
-        Ok(arg)
-    }
-}
-
-impl<T: AttributeOption> AttributeOption for AsOption<T> {
-    #[inline]
-    fn parse(values: Values) -> Result<Self> {
-        Ok(AsOption(Some(T::parse(values)?)))
-    }
-}
-
-macro_rules! attr_option_num {
-    ($($n:ty),*) => {
-        $(
-            impl AttributeOption for $n {
-                fn parse(values: Values) -> Result<Self> {
-                    validate(&values, &[ValueKind::SingleList])?;
-
-                    Ok(match &values.literals[0].1 {
-                        Lit::Int(l) => l.base10_parse::<$n>()?,
-                        l => {
-                            let s = l.to_str();
-                            // Use `as_str` to guide the compiler to use `&str`'s parse method.
-                            // We don't want to use our `parse` method here (`impl AttributeOption for String`).
-                            match s.as_str().parse::<$n>() {
-                                Ok(n) => n,
-                                Err(_) => return Err(Error::new(l.span(), "invalid integer")),
-                            }
-                        }
-                    })
-                }
-            }
-
-            impl AttributeOption for Option<$n> {
-                #[inline]
-                fn parse(values: Values) -> Result<Self> {
-                    <$n as AttributeOption>::parse(values).map(Some)
-                }
-            }
-        )*
-    }
-}
-
-attr_option_num!(u16, u32, usize);

command_attributes/src/consts.rs

@@ -1,10 +0,0 @@
-pub mod suffixes {
-    pub const COMMAND: &str = "COMMAND";
-    pub const ARG: &str = "ARG";
-    pub const SUBCOMMAND: &str = "SUBCOMMAND";
-    pub const SUBCOMMAND_GROUP: &str = "GROUP";
-    pub const CHECK: &str = "CHECK";
-    pub const HOOK: &str = "HOOK";
-}
-
-pub use self::suffixes::*;

command_attributes/src/lib.rs

@@ -1,321 +0,0 @@
-#![deny(rust_2018_idioms)]
-#![deny(broken_intra_doc_links)]
-
-use proc_macro::TokenStream;
-use proc_macro2::Ident;
-use quote::quote;
-use syn::{parse::Error, parse_macro_input, parse_quote, spanned::Spanned, Lit, Type};
-use uuid::Uuid;
-
-pub(crate) mod attributes;
-pub(crate) mod consts;
-pub(crate) mod structures;
-
-#[macro_use]
-pub(crate) mod util;
-
-use attributes::*;
-use consts::*;
-use structures::*;
-use util::*;
-
-macro_rules! match_options {
-    ($v:expr, $values:ident, $options:ident, $span:expr => [$($name:ident);*]) => {
-        match $v {
-            $(
-                stringify!($name) => $options.$name = propagate_err!($crate::attributes::parse($values)),
-            )*
-            _ => {
-                return Error::new($span, format_args!("invalid attribute: {:?}", $v))
-                    .to_compile_error()
-                    .into();
-            },
-        }
-    };
-}
-
-#[proc_macro_attribute]
-pub fn command(attr: TokenStream, input: TokenStream) -> TokenStream {
-    enum LastItem {
-        Fun,
-        SubFun,
-        SubGroup,
-        SubGroupFun,
-    }
-
-    let mut fun = parse_macro_input!(input as CommandFun);
-
-    let _name = if !attr.is_empty() {
-        parse_macro_input!(attr as Lit).to_str()
-    } else {
-        fun.name.to_string()
-    };
-
-    let mut hooks: Vec<Ident> = Vec::new();
-    let mut options = Options::new();
-    let mut last_desc = LastItem::Fun;
-
-    for attribute in &fun.attributes {
-        let span = attribute.span();
-        let values = propagate_err!(parse_values(attribute));
-
-        let name = values.name.to_string();
-        let name = &name[..];
-
-        match name {
-            "subcommand" => {
-                let new_subcommand = Subcommand::new(propagate_err!(attributes::parse(values)));
-
-                if let Some(subcommand_group) = options.subcommand_groups.last_mut() {
-                    last_desc = LastItem::SubGroupFun;
-                    subcommand_group.subcommands.push(new_subcommand);
-                } else {
-                    last_desc = LastItem::SubFun;
-                    options.subcommands.push(new_subcommand);
-                }
-            }
-            "subcommandgroup" => {
-                let new_group = SubcommandGroup::new(propagate_err!(attributes::parse(values)));
-                last_desc = LastItem::SubGroup;
-
-                options.subcommand_groups.push(new_group);
-            }
-            "arg" => {
-                let arg = propagate_err!(attributes::parse(values));
-
-                match last_desc {
-                    LastItem::Fun => {
-                        options.cmd_args.push(arg);
-                    }
-                    LastItem::SubFun => {
-                        options.subcommands.last_mut().unwrap().cmd_args.push(arg);
-                    }
-                    LastItem::SubGroup => {
-                        panic!("Argument not expected under subcommand group");
-                    }
-                    LastItem::SubGroupFun => {
-                        options
-                            .subcommand_groups
-                            .last_mut()
-                            .unwrap()
-                            .subcommands
-                            .last_mut()
-                            .unwrap()
-                            .cmd_args
-                            .push(arg);
-                    }
-                }
-            }
-            "example" => {
-                options.examples.push(propagate_err!(attributes::parse(values)));
-            }
-            "description" => {
-                let line: String = propagate_err!(attributes::parse(values));
-
-                match last_desc {
-                    LastItem::Fun => {
-                        util::append_line(&mut options.description, line);
-                    }
-                    LastItem::SubFun => {
-                        util::append_line(
-                            &mut options.subcommands.last_mut().unwrap().description,
-                            line,
-                        );
-                    }
-                    LastItem::SubGroup => {
-                        util::append_line(
-                            &mut options.subcommand_groups.last_mut().unwrap().description,
-                            line,
-                        );
-                    }
-                    LastItem::SubGroupFun => {
-                        util::append_line(
-                            &mut options
-                                .subcommand_groups
-                                .last_mut()
-                                .unwrap()
-                                .subcommands
-                                .last_mut()
-                                .unwrap()
-                                .description,
-                            line,
-                        );
-                    }
-                }
-            }
-            "hook" => {
-                hooks.push(propagate_err!(attributes::parse(values)));
-            }
-            _ => {
-                match_options!(name, values, options, span => [
-                    aliases;
-                    group;
-                    can_blacklist;
-                    supports_dm
-                ]);
-            }
-        }
-    }
-
-    let Options {
-        aliases,
-        description,
-        group,
-        examples,
-        can_blacklist,
-        supports_dm,
-        mut cmd_args,
-        mut subcommands,
-        mut subcommand_groups,
-    } = options;
-
-    let visibility = fun.visibility;
-    let name = fun.name.clone();
-    let body = fun.body;
-
-    let root_ident = name.with_suffix(COMMAND);
-
-    let command_path = quote!(crate::framework::Command);
-
-    populate_fut_lifetimes_on_refs(&mut fun.args);
-
-    let mut subcommand_group_idents = subcommand_groups
-        .iter()
-        .map(|subcommand| {
-            root_ident
-                .with_suffix(subcommand.name.replace("-", "_").as_str())
-                .with_suffix(SUBCOMMAND_GROUP)
-        })
-        .collect::<Vec<Ident>>();
-
-    let mut subcommand_idents = subcommands
-        .iter()
-        .map(|subcommand| {
-            root_ident
-                .with_suffix(subcommand.name.replace("-", "_").as_str())
-                .with_suffix(SUBCOMMAND)
-        })
-        .collect::<Vec<Ident>>();
-
-    let mut arg_idents = cmd_args
-        .iter()
-        .map(|arg| root_ident.with_suffix(arg.name.replace("-", "_").as_str()).with_suffix(ARG))
-        .collect::<Vec<Ident>>();
-
-    let mut tokens = quote! {};
-
-    tokens.extend(
-        subcommand_groups
-            .iter_mut()
-            .zip(subcommand_group_idents.iter())
-            .map(|(group, group_ident)| group.as_tokens(group_ident))
-            .fold(quote! {}, |mut a, b| {
-                a.extend(b);
-                a
-            }),
-    );
-
-    tokens.extend(
-        subcommands
-            .iter_mut()
-            .zip(subcommand_idents.iter())
-            .map(|(subcommand, sc_ident)| subcommand.as_tokens(sc_ident))
-            .fold(quote! {}, |mut a, b| {
-                a.extend(b);
-                a
-            }),
-    );
-
-    tokens.extend(
-        cmd_args.iter_mut().zip(arg_idents.iter()).map(|(arg, ident)| arg.as_tokens(ident)).fold(
-            quote! {},
-            |mut a, b| {
-                a.extend(b);
-                a
-            },
-        ),
-    );
-
-    arg_idents.append(&mut subcommand_group_idents);
-    arg_idents.append(&mut subcommand_idents);
-
-    let args = fun.args;
-
-    let variant = if args.len() == 2 {
-        quote!(crate::framework::CommandFnType::Multi)
-    } else {
-        let string: Type = parse_quote!(String);
-
-        let final_arg = args.get(2).unwrap();
-
-        if final_arg.kind == string {
-            quote!(crate::framework::CommandFnType::Text)
-        } else {
-            quote!(crate::framework::CommandFnType::Slash)
-        }
-    };
-
-    tokens.extend(quote! {
-        #[allow(missing_docs)]
-        pub static #root_ident: #command_path = #command_path {
-            fun: #variant(#name),
-            names: &[#_name, #(#aliases),*],
-            desc: #description,
-            group: #group,
-            examples: &[#(#examples),*],
-            can_blacklist: #can_blacklist,
-            supports_dm: #supports_dm,
-            args: &[#(&#arg_idents),*],
-            hooks: &[#(&#hooks),*],
-        };
-
-        #[allow(missing_docs)]
-        #visibility fn #name<'fut> (#(#args),*) -> ::serenity::futures::future::BoxFuture<'fut, ()> {
-            use ::serenity::futures::future::FutureExt;
-
-            async move {
-                #(#body)*;
-            }.boxed()
-        }
-    });
-
-    tokens.into()
-}
-
-#[proc_macro_attribute]
-pub fn check(_attr: TokenStream, input: TokenStream) -> TokenStream {
-    let mut fun = parse_macro_input!(input as CommandFun);
-
-    let n = fun.name.clone();
-    let name = n.with_suffix(HOOK);
-    let fn_name = n.with_suffix(CHECK);
-    let visibility = fun.visibility;
-
-    let body = fun.body;
-    let ret = fun.ret;
-    populate_fut_lifetimes_on_refs(&mut fun.args);
-    let args = fun.args;
-
-    let hook_path = quote!(crate::framework::Hook);
-    let uuid = Uuid::new_v4().as_u128();
-
-    (quote! {
-        #[allow(missing_docs)]
-        #visibility fn #fn_name<'fut>(#(#args),*) -> ::serenity::futures::future::BoxFuture<'fut, #ret> {
-            use ::serenity::futures::future::FutureExt;
-
-            async move {
-                let _output: #ret = { #(#body)* };
-                #[allow(unreachable_code)]
-                _output
-            }.boxed()
-        }
-
-        #[allow(missing_docs)]
-        pub static #name: #hook_path = #hook_path {
-            fun: #fn_name,
-            uuid: #uuid,
-        };
-    })
-    .into()
-}

command_attributes/src/structures.rs

@@ -1,331 +0,0 @@
-use proc_macro2::TokenStream as TokenStream2;
-use quote::{quote, ToTokens};
-use syn::{
-    braced,
-    parse::{Error, Parse, ParseStream, Result},
-    spanned::Spanned,
-    Attribute, Block, FnArg, Ident, Pat, ReturnType, Stmt, Token, Type, Visibility,
-};
-
-use crate::{
-    consts::{ARG, SUBCOMMAND},
-    util::{Argument, IdentExt2, Parenthesised},
-};
-
-fn parse_argument(arg: FnArg) -> Result<Argument> {
-    match arg {
-        FnArg::Typed(typed) => {
-            let pat = typed.pat;
-            let kind = typed.ty;
-
-            match *pat {
-                Pat::Ident(id) => {
-                    let name = id.ident;
-                    let mutable = id.mutability;
-
-                    Ok(Argument { mutable, name, kind: *kind })
-                }
-                Pat::Wild(wild) => {
-                    let token = wild.underscore_token;
-
-                    let name = Ident::new("_", token.spans[0]);
-
-                    Ok(Argument { mutable: None, name, kind: *kind })
-                }
-                _ => Err(Error::new(pat.span(), format_args!("unsupported pattern: {:?}", pat))),
-            }
-        }
-        FnArg::Receiver(_) => {
-            Err(Error::new(arg.span(), format_args!("`self` arguments are prohibited: {:?}", arg)))
-        }
-    }
-}
-
-#[derive(Debug)]
-pub struct CommandFun {
-    /// `#[...]`-style attributes.
-    pub attributes: Vec<Attribute>,
-    /// Populated cooked attributes. These are attributes outside of the realm of this crate's procedural macros
-    /// and will appear in generated output.
-    pub visibility: Visibility,
-    pub name: Ident,
-    pub args: Vec<Argument>,
-    pub ret: Type,
-    pub body: Vec<Stmt>,
-}
-
-impl Parse for CommandFun {
-    fn parse(input: ParseStream<'_>) -> Result<Self> {
-        let attributes = input.call(Attribute::parse_outer)?;
-
-        let visibility = input.parse::<Visibility>()?;
-
-        input.parse::<Token![async]>()?;
-
-        input.parse::<Token![fn]>()?;
-        let name = input.parse()?;
-
-        // (...)
-        let Parenthesised(args) = input.parse::<Parenthesised<FnArg>>()?;
-
-        let ret = match input.parse::<ReturnType>()? {
-            ReturnType::Type(_, t) => (*t).clone(),
-            ReturnType::Default => Type::Verbatim(quote!(())),
-        };
-
-        // { ... }
-        let bcont;
-        braced!(bcont in input);
-        let body = bcont.call(Block::parse_within)?;
-
-        let args = args.into_iter().map(parse_argument).collect::<Result<Vec<_>>>()?;
-
-        Ok(Self { attributes, visibility, name, args, ret, body })
-    }
-}
-
-impl ToTokens for CommandFun {
-    fn to_tokens(&self, stream: &mut TokenStream2) {
-        let Self { attributes: _, visibility, name, args, ret, body } = self;
-
-        stream.extend(quote! {
-            #visibility async fn #name (#(#args),*) -> #ret {
-                #(#body)*
-            }
-        });
-    }
-}
-
-#[derive(Debug)]
-pub(crate) enum ApplicationCommandOptionType {
-    SubCommand,
-    SubCommandGroup,
-    String,
-    Integer,
-    Boolean,
-    User,
-    Channel,
-    Role,
-    Mentionable,
-    Number,
-    Unknown,
-}
-
-impl ApplicationCommandOptionType {
-    pub fn from_str(s: String) -> Self {
-        match s.as_str() {
-            "SubCommand" => Self::SubCommand,
-            "SubCommandGroup" => Self::SubCommandGroup,
-            "String" => Self::String,
-            "Integer" => Self::Integer,
-            "Boolean" => Self::Boolean,
-            "User" => Self::User,
-            "Channel" => Self::Channel,
-            "Role" => Self::Role,
-            "Mentionable" => Self::Mentionable,
-            "Number" => Self::Number,
-            _ => Self::Unknown,
-        }
-    }
-}
-
-impl ToTokens for ApplicationCommandOptionType {
-    fn to_tokens(&self, stream: &mut TokenStream2) {
-        let path = quote!(
-            serenity::model::interactions::application_command::ApplicationCommandOptionType
-        );
-        let variant = match self {
-            ApplicationCommandOptionType::SubCommand => quote!(SubCommand),
-            ApplicationCommandOptionType::SubCommandGroup => quote!(SubCommandGroup),
-            ApplicationCommandOptionType::String => quote!(String),
-            ApplicationCommandOptionType::Integer => quote!(Integer),
-            ApplicationCommandOptionType::Boolean => quote!(Boolean),
-            ApplicationCommandOptionType::User => quote!(User),
-            ApplicationCommandOptionType::Channel => quote!(Channel),
-            ApplicationCommandOptionType::Role => quote!(Role),
-            ApplicationCommandOptionType::Mentionable => quote!(Mentionable),
-            ApplicationCommandOptionType::Number => quote!(Number),
-            ApplicationCommandOptionType::Unknown => quote!(Unknown),
-        };
-
-        stream.extend(quote! {
-            #path::#variant
-        });
-    }
-}
-
-#[derive(Debug)]
-pub(crate) struct Arg {
-    pub name: String,
-    pub description: String,
-    pub kind: ApplicationCommandOptionType,
-    pub required: bool,
-}
-
-impl Arg {
-    pub fn as_tokens(&self, ident: &Ident) -> TokenStream2 {
-        let arg_path = quote!(crate::framework::Arg);
-        let Arg { name, description, kind, required } = self;
-
-        quote! {
-            #[allow(missing_docs)]
-            pub static #ident: #arg_path = #arg_path {
-                name: #name,
-                description: #description,
-                kind: #kind,
-                required: #required,
-                options: &[]
-            };
-        }
-    }
-}
-
-impl Default for Arg {
-    fn default() -> Self {
-        Self {
-            name: String::new(),
-            description: String::new(),
-            kind: ApplicationCommandOptionType::String,
-            required: false,
-        }
-    }
-}
-
-#[derive(Debug)]
-pub(crate) struct Subcommand {
-    pub name: String,
-    pub description: String,
-    pub cmd_args: Vec<Arg>,
-}
-
-impl Subcommand {
-    pub fn as_tokens(&mut self, ident: &Ident) -> TokenStream2 {
-        let arg_path = quote!(crate::framework::Arg);
-        let subcommand_path = ApplicationCommandOptionType::SubCommand;
-
-        let arg_idents = self
-            .cmd_args
-            .iter()
-            .map(|arg| ident.with_suffix(arg.name.as_str()).with_suffix(ARG))
-            .collect::<Vec<Ident>>();
-
-        let mut tokens = self
-            .cmd_args
-            .iter_mut()
-            .zip(arg_idents.iter())
-            .map(|(arg, ident)| arg.as_tokens(ident))
-            .fold(quote! {}, |mut a, b| {
-                a.extend(b);
-                a
-            });
-
-        let Subcommand { name, description, .. } = self;
-
-        tokens.extend(quote! {
-            #[allow(missing_docs)]
-            pub static #ident: #arg_path = #arg_path {
-                name: #name,
-                description: #description,
-                kind: #subcommand_path,
-                required: false,
-                options: &[#(&#arg_idents),*],
-            };
-        });
-
-        tokens
-    }
-}
-
-impl Default for Subcommand {
-    fn default() -> Self {
-        Self { name: String::new(), description: String::new(), cmd_args: vec![] }
-    }
-}
-
-impl Subcommand {
-    pub(crate) fn new(name: String) -> Self {
-        Self { name, ..Default::default() }
-    }
-}
-
-#[derive(Debug)]
-pub(crate) struct SubcommandGroup {
-    pub name: String,
-    pub description: String,
-    pub subcommands: Vec<Subcommand>,
-}
-
-impl SubcommandGroup {
-    pub fn as_tokens(&mut self, ident: &Ident) -> TokenStream2 {
-        let arg_path = quote!(crate::framework::Arg);
-        let subcommand_group_path = ApplicationCommandOptionType::SubCommandGroup;
-
-        let arg_idents = self
-            .subcommands
-            .iter()
-            .map(|arg| {
-                ident
-                    .with_suffix(self.name.as_str())
-                    .with_suffix(arg.name.as_str())
-                    .with_suffix(SUBCOMMAND)
-            })
-            .collect::<Vec<Ident>>();
-
-        let mut tokens = self
-            .subcommands
-            .iter_mut()
-            .zip(arg_idents.iter())
-            .map(|(subcommand, ident)| subcommand.as_tokens(ident))
-            .fold(quote! {}, |mut a, b| {
-                a.extend(b);
-                a
-            });
-
-        let SubcommandGroup { name, description, .. } = self;
-
-        tokens.extend(quote! {
-            #[allow(missing_docs)]
-            pub static #ident: #arg_path = #arg_path {
-                name: #name,
-                description: #description,
-                kind: #subcommand_group_path,
-                required: false,
-                options: &[#(&#arg_idents),*],
-            };
-        });
-
-        tokens
-    }
-}
-
-impl Default for SubcommandGroup {
-    fn default() -> Self {
-        Self { name: String::new(), description: String::new(), subcommands: vec![] }
-    }
-}
-
-impl SubcommandGroup {
-    pub(crate) fn new(name: String) -> Self {
-        Self { name, ..Default::default() }
-    }
-}
-
-#[derive(Debug, Default)]
-pub(crate) struct Options {
-    pub aliases: Vec<String>,
-    pub description: String,
-    pub group: String,
-    pub examples: Vec<String>,
-    pub can_blacklist: bool,
-    pub supports_dm: bool,
-    pub cmd_args: Vec<Arg>,
-    pub subcommands: Vec<Subcommand>,
-    pub subcommand_groups: Vec<SubcommandGroup>,
-}
-
-impl Options {
-    #[inline]
-    pub fn new() -> Self {
-        Self { group: "None".to_string(), ..Default::default() }
-    }
-}

command_attributes/src/util.rs

@@ -1,176 +0,0 @@
-use proc_macro::TokenStream;
-use proc_macro2::{Span, TokenStream as TokenStream2};
-use quote::{format_ident, quote, ToTokens};
-use syn::{
-    braced, bracketed, parenthesized,
-    parse::{Error, Parse, ParseStream, Result as SynResult},
-    punctuated::Punctuated,
-    token::{Comma, Mut},
-    Ident, Lifetime, Lit, Type,
-};
-
-pub trait LitExt {
-    fn to_str(&self) -> String;
-    fn to_bool(&self) -> bool;
-    fn to_ident(&self) -> Ident;
-}
-
-impl LitExt for Lit {
-    fn to_str(&self) -> String {
-        match self {
-            Lit::Str(s) => s.value(),
-            Lit::ByteStr(s) => unsafe { String::from_utf8_unchecked(s.value()) },
-            Lit::Char(c) => c.value().to_string(),
-            Lit::Byte(b) => (b.value() as char).to_string(),
-            _ => panic!("values must be a (byte)string or a char"),
-        }
-    }
-
-    fn to_bool(&self) -> bool {
-        if let Lit::Bool(b) = self {
-            b.value
-        } else {
-            self.to_str()
-                .parse()
-                .unwrap_or_else(|_| panic!("expected bool from {:?}", self))
-        }
-    }
-
-    #[inline]
-    fn to_ident(&self) -> Ident {
-        Ident::new(&self.to_str(), self.span())
-    }
-}
-
-pub trait IdentExt2: Sized {
-    fn to_uppercase(&self) -> Self;
-    fn with_suffix(&self, suf: &str) -> Ident;
-}
-
-impl IdentExt2 for Ident {
-    #[inline]
-    fn to_uppercase(&self) -> Self {
-        format_ident!("{}", self.to_string().to_uppercase())
-    }
-
-    #[inline]
-    fn with_suffix(&self, suffix: &str) -> Ident {
-        format_ident!("{}_{}", self.to_string().to_uppercase(), suffix)
-    }
-}
-
-#[inline]
-pub fn into_stream(e: Error) -> TokenStream {
-    e.to_compile_error().into()
-}
-
-macro_rules! propagate_err {
-    ($res:expr) => {{
-        match $res {
-            Ok(v) => v,
-            Err(e) => return $crate::util::into_stream(e),
-        }
-    }};
-}
-
-#[derive(Debug)]
-pub struct Bracketed<T>(pub Punctuated<T, Comma>);
-
-impl<T: Parse> Parse for Bracketed<T> {
-    fn parse(input: ParseStream<'_>) -> SynResult<Self> {
-        let content;
-        bracketed!(content in input);
-
-        Ok(Bracketed(content.parse_terminated(T::parse)?))
-    }
-}
-
-#[derive(Debug)]
-pub struct Braced<T>(pub Punctuated<T, Comma>);
-
-impl<T: Parse> Parse for Braced<T> {
-    fn parse(input: ParseStream<'_>) -> SynResult<Self> {
-        let content;
-        braced!(content in input);
-
-        Ok(Braced(content.parse_terminated(T::parse)?))
-    }
-}
-
-#[derive(Debug)]
-pub struct Parenthesised<T>(pub Punctuated<T, Comma>);
-
-impl<T: Parse> Parse for Parenthesised<T> {
-    fn parse(input: ParseStream<'_>) -> SynResult<Self> {
-        let content;
-        parenthesized!(content in input);
-
-        Ok(Parenthesised(content.parse_terminated(T::parse)?))
-    }
-}
-
-#[derive(Debug)]
-pub struct AsOption<T>(pub Option<T>);
-
-impl<T: ToTokens> ToTokens for AsOption<T> {
-    fn to_tokens(&self, stream: &mut TokenStream2) {
-        match &self.0 {
-            Some(o) => stream.extend(quote!(Some(#o))),
-            None => stream.extend(quote!(None)),
-        }
-    }
-}
-
-impl<T> Default for AsOption<T> {
-    #[inline]
-    fn default() -> Self {
-        AsOption(None)
-    }
-}
-
-#[derive(Debug)]
-pub struct Argument {
-    pub mutable: Option<Mut>,
-    pub name: Ident,
-    pub kind: Type,
-}
-
-impl ToTokens for Argument {
-    fn to_tokens(&self, stream: &mut TokenStream2) {
-        let Argument {
-            mutable,
-            name,
-            kind,
-        } = self;
-
-        stream.extend(quote! {
-            #mutable #name: #kind
-        });
-    }
-}
-
-#[inline]
-pub fn populate_fut_lifetimes_on_refs(args: &mut Vec<Argument>) {
-    for arg in args {
-        if let Type::Reference(reference) = &mut arg.kind {
-            reference.lifetime = Some(Lifetime::new("'fut", Span::call_site()));
-        }
-    }
-}
-
-pub fn append_line(desc: &mut String, mut line: String) {
-    if line.starts_with(' ') {
-        line.remove(0);
-    }
-
-    match line.rfind("\\$") {
-        Some(i) => {
-            desc.push_str(line[..i].trim_end());
-            desc.push(' ');
-        }
-        None => {
-            desc.push_str(&line);
-            desc.push('\n');
-        }
-    }
-}

migration/04-reminder_templates.sql

@@ -24,8 +24,11 @@ CREATE TABLE reminder_template (
     `embed_author` VARCHAR(256) NOT NULL DEFAULT '',
     `embed_author_url` VARCHAR(512),
     `embed_color` INT UNSIGNED NOT NULL DEFAULT 0x0,
+    `embed_fields` JSON,
 
     PRIMARY KEY (id),
 
-    FOREIGN KEY (`guild_id`) REFERENCES channels (`id`) ON DELETE CASCADE,
+    FOREIGN KEY (`guild_id`) REFERENCES guilds (`id`) ON DELETE CASCADE
 );
+
+ALTER TABLE reminders ADD COLUMN embed_fields JSON;

postman/Cargo.toml

@@ -12,21 +12,7 @@ chrono = "0.4"
 chrono-tz = { version = "0.5", features = ["serde"] }
 lazy_static = "1.4"
 num-integer = "0.1"
-sqlx = { version = "0.5.10", features = ["runtime-tokio-rustls", "macros", "mysql", "bigdecimal", "chrono"]}
-
-[dependencies.serenity]
-git = "https://github.com/serenity-rs/serenity"
-branch = "next"
-default-features = false
-features = [
-    "builder",
-    "client",
-    "cache",
-    "gateway",
-    "http",
-    "model",
-    "utils",
-    "rustls_backend",
-    "collector",
-    "unstable_discord_api"
-]
+serde = "1.0"
+serde_json = "1.0"
+sqlx = { version = "0.5", features = ["runtime-tokio-rustls", "macros", "mysql", "bigdecimal", "chrono", "json"]}
+serenity = { version = "0.11.1", default-features = false, features = ["builder", "cache", "client", "gateway", "http", "model", "utils", "rustls_backend"] }

postman/src/lib.rs

@@ -1,23 +1,40 @@
 mod sender;
 
-use log::info;
+use std::env;
+
+use log::{info, warn};
 use serenity::client::Context;
 use sqlx::{Executor, MySql};
-use std::env;
-use tokio::time::sleep_until;
-use tokio::time::{Duration, Instant};
+use tokio::{
+    sync::broadcast::Receiver,
+    time::{sleep_until, Duration, Instant},
+};
 
 type Database = MySql;
 
-pub async fn initialize(ctx: Context, pool: impl Executor<'_, Database = Database> + Copy) {
-    let REMIND_INTERVAL = env::var("REMIND_INTERVAL")
+pub async fn initialize(
+    mut kill: Receiver<()>,
+    ctx: Context,
+    pool: impl Executor<'_, Database = Database> + Copy,
+) -> Result<(), &'static str> {
+    tokio::select! {
+        output = _initialize(ctx, pool) => Ok(output),
+        _ = kill.recv() => {
+            warn!("Received terminate signal. Goodbye");
+            Err("Received terminate signal. Goodbye")
+        }
+    }
+}
+
+async fn _initialize(ctx: Context, pool: impl Executor<'_, Database = Database> + Copy) {
+    let remind_interval = env::var("REMIND_INTERVAL")
         .map(|inner| inner.parse::<u64>().ok())
         .ok()
         .flatten()
         .unwrap_or(10);
 
     loop {
-        let sleep_to = Instant::now() + Duration::from_secs(REMIND_INTERVAL);
+        let sleep_to = Instant::now() + Duration::from_secs(remind_interval);
         let reminders = sender::Reminder::fetch_reminders(pool).await;
 
         if reminders.len() > 0 {

postman/src/sender.rs

@@ -1,10 +1,10 @@
-use crate::Database;
 use chrono::Duration;
 use chrono_tz::Tz;
 use lazy_static::lazy_static;
 use log::{error, info, warn};
 use num_integer::Integer;
 use regex::{Captures, Regex};
+use serde::Deserialize;
 use serenity::{
     builder::CreateEmbed,
     http::{CacheHttp, Http, StatusCode},
@@ -16,10 +16,15 @@ use serenity::{
     Error, Result,
 };
 use sqlx::{
-    types::chrono::{NaiveDateTime, Utc},
+    types::{
+        chrono::{NaiveDateTime, Utc},
+        Json,
+    },
     Executor,
 };
 
+use crate::Database;
+
 lazy_static! {
     pub static ref TIMEFROM_REGEX: Regex =
         Regex::new(r#"<<timefrom:(?P<time>\d+):(?P<format>.+)?>>"#).unwrap();
@@ -93,11 +98,6 @@ pub fn substitute(string: &str) -> String {
 }
 
 struct Embed {
-    inner: EmbedInner,
-    fields: Vec<EmbedField>,
-}
-
-struct EmbedInner {
     title: String,
     description: String,
     image_url: Option<String>,
@@ -107,8 +107,10 @@ struct EmbedInner {
     author: String,
     author_url: Option<String>,
     color: u32,
+    fields: Json<Vec<EmbedField>>,
 }
 
+#[derive(Deserialize)]
 struct EmbedField {
     title: String,
     value: String,
@@ -120,76 +122,54 @@ impl Embed {
         pool: impl Executor<'_, Database = Database> + Copy,
         id: u32,
     ) -> Option<Self> {
-        let mut inner = sqlx::query_as_unchecked!(
-            EmbedInner,
-            "
-SELECT
-    `embed_title` AS title,
-    `embed_description` AS description,
-    `embed_image_url` AS image_url,
-    `embed_thumbnail_url` AS thumbnail_url,
-    `embed_footer` AS footer,
-    `embed_footer_url` AS footer_url,
-    `embed_author` AS author,
-    `embed_author_url` AS author_url,
-    `embed_color` AS color
-FROM
-    reminders
-WHERE
-    `id` = ?
-            ",
+        let mut embed = sqlx::query_as!(
+            Self,
+            r#"
+            SELECT
+             `embed_title` AS title,
+             `embed_description` AS description,
+             `embed_image_url` AS image_url,
+             `embed_thumbnail_url` AS thumbnail_url,
+             `embed_footer` AS footer,
+             `embed_footer_url` AS footer_url,
+             `embed_author` AS author,
+             `embed_author_url` AS author_url,
+             `embed_color` AS color,
+             IFNULL(`embed_fields`, '[]') AS "fields:_"
+            FROM reminders
+            WHERE `id` = ?"#,
             id
         )
         .fetch_one(pool)
         .await
         .unwrap();
 
-        inner.title = substitute(&inner.title);
-        inner.description = substitute(&inner.description);
-        inner.footer = substitute(&inner.footer);
+        embed.title = substitute(&embed.title);
+        embed.description = substitute(&embed.description);
+        embed.footer = substitute(&embed.footer);
 
-        let mut fields = sqlx::query_as_unchecked!(
-            EmbedField,
-            "
-SELECT
-    title,
-    value,
-    inline
-FROM
-    embed_fields
-WHERE
-    reminder_id = ?
-            ",
-            id
-        )
-        .fetch_all(pool)
-        .await
-        .unwrap();
-
-        fields.iter_mut().for_each(|mut field| {
+        embed.fields.iter_mut().for_each(|mut field| {
             field.title = substitute(&field.title);
             field.value = substitute(&field.value);
         });
 
-        let e = Embed { inner, fields };
-
-        if e.has_content() {
-            Some(e)
+        if embed.has_content() {
+            Some(embed)
         } else {
             None
         }
     }
 
     pub fn has_content(&self) -> bool {
-        if self.inner.title.is_empty()
-            && self.inner.description.is_empty()
-            && self.inner.image_url.is_none()
-            && self.inner.thumbnail_url.is_none()
-            && self.inner.footer.is_empty()
-            && self.inner.footer_url.is_none()
-            && self.inner.author.is_empty()
-            && self.inner.author_url.is_none()
-            && self.fields.is_empty()
+        if self.title.is_empty()
+            && self.description.is_empty()
+            && self.image_url.is_none()
+            && self.thumbnail_url.is_none()
+            && self.footer.is_empty()
+            && self.footer_url.is_none()
+            && self.author.is_empty()
+            && self.author_url.is_none()
+            && self.fields.0.is_empty()
         {
             false
         } else {
@@ -202,37 +182,37 @@ impl Into<CreateEmbed> for Embed {
     fn into(self) -> CreateEmbed {
         let mut c = CreateEmbed::default();
 
-        c.title(&self.inner.title)
-            .description(&self.inner.description)
-            .color(self.inner.color)
+        c.title(&self.title)
+            .description(&self.description)
+            .color(self.color)
             .author(|a| {
-                a.name(&self.inner.author);
+                a.name(&self.author);
 
-                if let Some(author_icon) = &self.inner.author_url {
+                if let Some(author_icon) = &self.author_url {
                     a.icon_url(author_icon);
                 }
 
                 a
             })
             .footer(|f| {
-                f.text(&self.inner.footer);
+                f.text(&self.footer);
 
-                if let Some(footer_icon) = &self.inner.footer_url {
+                if let Some(footer_icon) = &self.footer_url {
                     f.icon_url(footer_icon);
                 }
 
                 f
             });
 
-        for field in &self.fields {
+        for field in &self.fields.0 {
             c.field(&field.title, &field.value, field.inline);
         }
 
-        if let Some(image_url) = &self.inner.image_url {
+        if let Some(image_url) = &self.image_url {
             c.image(image_url);
         }
 
-        if let Some(thumbnail_url) = &self.inner.thumbnail_url {
+        if let Some(thumbnail_url) = &self.thumbnail_url {
             c.thumbnail(thumbnail_url);
         }
 

src/commands/info_cmds.rs

@@ -1,16 +1,13 @@
 use chrono::offset::Utc;
-use regex_command_attr::command;
-use serenity::{builder::CreateEmbedFooter, client::Context};
+use poise::{serenity_prelude as serenity, serenity_prelude::Mentionable};
 
-use crate::{
-    framework::{CommandInvoke, CreateGenericResponse},
-    models::CtxData,
-    THEME_COLOR,
-};
+use crate::{models::CtxData, Context, Error, THEME_COLOR};
 
-fn footer(ctx: &Context) -> impl FnOnce(&mut CreateEmbedFooter) -> &mut CreateEmbedFooter {
-    let shard_count = ctx.cache.shard_count();
-    let shard = ctx.shard_id;
+fn footer(
+    ctx: Context<'_>,
+) -> impl FnOnce(&mut serenity::CreateEmbedFooter) -> &mut serenity::CreateEmbedFooter {
+    let shard_count = ctx.discord().cache.shard_count();
+    let shard = ctx.discord().shard_id;
 
     move |f| {
         f.text(format!(
@@ -22,19 +19,17 @@ fn footer(ctx: &Context) -> impl FnOnce(&mut CreateEmbedFooter) -> &mut CreateEm
     }
 }
 
-#[command]
-#[description("Get an overview of the bot commands")]
-async fn help(ctx: &Context, invoke: &mut CommandInvoke) {
+/// Get an overview of bot commands
+#[poise::command(slash_command)]
+pub async fn help(ctx: Context<'_>) -> Result<(), Error> {
     let footer = footer(ctx);
 
-    let _ = invoke
-        .respond(
-            &ctx,
-            CreateGenericResponse::new().embed(|e| {
-                e.title("Help")
-                    .color(*THEME_COLOR)
-                    .description(
-                        "__Info Commands__
+    ctx.send(|m| {
+        m.ephemeral(true).embed(|e| {
+            e.title("Help")
+                .color(*THEME_COLOR)
+                .description(
+                    "__Info Commands__
 `/help` `/info` `/donate` `/dashboard` `/clock`
 *run these commands with no options*
 
@@ -58,23 +53,23 @@ __Setup Commands__
 __Advanced Commands__
 `/macro` - Record and replay command sequences
                     ",
-                    )
-                    .footer(footer)
-            }),
-        )
-        .await;
+                )
+                .footer(footer)
+        })
+    })
+    .await?;
+
+    Ok(())
 }
 
-#[command]
-#[aliases("invite")]
-#[description("Get information about the bot")]
-async fn info(ctx: &Context, invoke: &mut CommandInvoke) {
+/// Get information about the bot
+#[poise::command(slash_command)]
+pub async fn info(ctx: Context<'_>) -> Result<(), Error> {
     let footer = footer(ctx);
 
-    let _ = invoke
-        .respond(
-            ctx.http.clone(),
-            CreateGenericResponse::new().embed(|e| {
+    let _ = ctx
+        .send(|m| {
+            m.ephemeral(true).embed(|e| {
                 e.title("Info")
                     .description(format!(
                         "Help: `/help`
@@ -89,23 +84,22 @@ Use our dashboard: https://reminder-bot.com/",
                     ))
                     .footer(footer)
                     .color(*THEME_COLOR)
-            }),
-        )
+            })
+        })
         .await;
+
+    Ok(())
 }
 
-#[command]
-#[description("Details on supporting the bot and Patreon benefits")]
-#[group("Info")]
-async fn donate(ctx: &Context, invoke: &mut CommandInvoke) {
+/// Details on supporting the bot and Patreon benefits
+#[poise::command(slash_command)]
+pub async fn donate(ctx: Context<'_>) -> Result<(), Error> {
     let footer = footer(ctx);
 
-    let _ = invoke
-        .respond(
-            ctx.http.clone(),
-            CreateGenericResponse::new().embed(|e| {
-                e.title("Donate")
-                    .description("Thinking of adding a monthly contribution? Click below for my Patreon and official bot server :)
+    ctx.send(|m| m.embed(|e| {
+        e.title("Donate")
+            .description("Thinking of adding a monthly contribution?
+Click below for my Patreon and official bot server :)
 
 **https://www.patreon.com/jellywx/**
 **https://discord.jellywx.com/**
@@ -120,43 +114,67 @@ With your new rank, you'll be able to:
 Just $2 USD/month!
 
 *Please note, you must be in the JellyWX Discord server to receive Patreon features*")
-                    .footer(footer)
-                    .color(*THEME_COLOR)
-            }),
-        )
-        .await;
+                .footer(footer)
+                .color(*THEME_COLOR)
+        }),
+    )
+    .await?;
+
+    Ok(())
 }
 
-#[command]
-#[description("Get the link to the online dashboard")]
-#[group("Info")]
-async fn dashboard(ctx: &Context, invoke: &mut CommandInvoke) {
+/// Get the link to the online dashboard
+#[poise::command(slash_command)]
+pub async fn dashboard(ctx: Context<'_>) -> Result<(), Error> {
     let footer = footer(ctx);
 
-    let _ = invoke
-        .respond(
-            ctx.http.clone(),
-            CreateGenericResponse::new().embed(|e| {
-                e.title("Dashboard")
-                    .description("**https://reminder-bot.com/dashboard**")
-                    .footer(footer)
-                    .color(*THEME_COLOR)
-            }),
-        )
-        .await;
+    ctx.send(|m| {
+        m.ephemeral(true).embed(|e| {
+            e.title("Dashboard")
+                .description("**https://reminder-bot.com/dashboard**")
+                .footer(footer)
+                .color(*THEME_COLOR)
+        })
+    })
+    .await?;
+
+    Ok(())
 }
 
-#[command]
-#[description("View the current time in your selected timezone")]
-#[group("Info")]
-async fn clock(ctx: &Context, invoke: &mut CommandInvoke) {
-    let ud = ctx.user_data(&invoke.author_id()).await.unwrap();
-    let now = Utc::now().with_timezone(&ud.timezone());
+/// View the current time in your selected timezone
+#[poise::command(slash_command)]
+pub async fn clock(ctx: Context<'_>) -> Result<(), Error> {
+    ctx.defer_ephemeral().await?;
 
-    let _ = invoke
-        .respond(
-            ctx.http.clone(),
-            CreateGenericResponse::new().content(format!("Current time: {}", now.format("%H:%M"))),
-        )
-        .await;
+    let tz = ctx.timezone().await;
+    let now = Utc::now().with_timezone(&tz);
+
+    ctx.send(|m| {
+        m.ephemeral(true).content(format!("Time in **{}**: `{}`", tz, now.format("%H:%M")))
+    })
+    .await?;
+
+    Ok(())
+}
+
+/// View the current time in a user's selected timezone
+#[poise::command(context_menu_command = "View Local Time")]
+pub async fn clock_context_menu(ctx: Context<'_>, user: serenity::User) -> Result<(), Error> {
+    ctx.defer_ephemeral().await?;
+
+    let user_data = ctx.user_data(user.id).await?;
+    let tz = user_data.timezone();
+
+    let now = Utc::now().with_timezone(&tz);
+
+    ctx.send(|m| {
+        m.ephemeral(true).content(format!(
+            "Time in {}'s timezone: `{}`",
+            user.mention(),
+            now.format("%H:%M")
+        ))
+    })
+    .await?;
+
+    Ok(())
 }

src/commands/moderation_cmds.rs

@@ -1,54 +1,63 @@
 use chrono::offset::Utc;
 use chrono_tz::{Tz, TZ_VARIANTS};
 use levenshtein::levenshtein;
-use regex_command_attr::command;
-use serenity::client::Context;
+use poise::CreateReply;
 
 use crate::{
     component_models::pager::{MacroPager, Pager},
     consts::{EMBED_DESCRIPTION_MAX_LENGTH, THEME_COLOR},
-    framework::{CommandInvoke, CommandOptions, CreateGenericResponse, OptionValue},
-    hooks::{CHECK_GUILD_PERMISSIONS_HOOK, GUILD_ONLY_HOOK},
-    models::{command_macro::CommandMacro, CtxData},
-    PopularTimezones, RecordingMacros, RegexFramework, SQLPool,
+    models::{
+        command_macro::{guild_command_macro, CommandMacro},
+        CtxData,
+    },
+    Context, Data, Error,
 };
 
-#[command("timezone")]
-#[description("Select your timezone")]
-#[arg(
-    name = "timezone",
-    description = "Timezone to use from this list: https://gist.github.com/JellyWX/913dfc8b63d45192ad6cb54c829324ee",
-    kind = "String",
-    required = false
-)]
-async fn timezone(ctx: &Context, invoke: &mut CommandInvoke, args: CommandOptions) {
-    let pool = ctx.data.read().await.get::<SQLPool>().cloned().unwrap();
-    let mut user_data = ctx.user_data(invoke.author_id()).await.unwrap();
+async fn timezone_autocomplete(ctx: Context<'_>, partial: String) -> Vec<String> {
+    if partial.is_empty() {
+        ctx.data().popular_timezones.iter().map(|t| t.to_string()).collect::<Vec<String>>()
+    } else {
+        TZ_VARIANTS
+            .iter()
+            .filter(|tz| tz.to_string().contains(&partial))
+            .take(25)
+            .map(|t| t.to_string())
+            .collect::<Vec<String>>()
+    }
+}
+
+/// Select your timezone
+#[poise::command(slash_command, identifying_name = "timezone")]
+pub async fn timezone(
+    ctx: Context<'_>,
+    #[description = "Timezone to use from this list: https://gist.github.com/JellyWX/913dfc8b63d45192ad6cb54c829324ee"]
+    #[autocomplete = "timezone_autocomplete"]
+    timezone: Option<String>,
+) -> Result<(), Error> {
+    let mut user_data = ctx.author_data().await.unwrap();
 
     let footer_text = format!("Current timezone: {}", user_data.timezone);
 
-    if let Some(OptionValue::String(timezone)) = args.get("timezone") {
+    if let Some(timezone) = timezone {
         match timezone.parse::<Tz>() {
             Ok(tz) => {
                 user_data.timezone = timezone.clone();
-                user_data.commit_changes(&pool).await;
+                user_data.commit_changes(&ctx.data().database).await;
 
                 let now = Utc::now().with_timezone(&tz);
 
-                let _ = invoke
-                    .respond(
-                        ctx.http.clone(),
-                        CreateGenericResponse::new().embed(|e| {
-                            e.title("Timezone Set")
-                                .description(format!(
-                                    "Timezone has been set to **{}**. Your current time should be `{}`",
-                                    timezone,
-                                    now.format("%H:%M").to_string()
-                                ))
-                                .color(*THEME_COLOR)
-                        }),
-                    )
-                    .await;
+                ctx.send(|m| {
+                    m.embed(|e| {
+                        e.title("Timezone Set")
+                            .description(format!(
+                                "Timezone has been set to **{}**. Your current time should be `{}`",
+                                timezone,
+                                now.format("%H:%M").to_string()
+                            ))
+                            .color(*THEME_COLOR)
+                    })
+                })
+                .await?;
             }
 
             Err(_) => {
@@ -56,8 +65,8 @@ async fn timezone(ctx: &Context, invoke: &mut CommandInvoke, args: CommandOption
                     .iter()
                     .filter(|tz| {
                         timezone.contains(&tz.to_string())
-                            || tz.to_string().contains(timezone)
-                            || levenshtein(&tz.to_string(), timezone) < 4
+                            || tz.to_string().contains(&timezone)
+                            || levenshtein(&tz.to_string(), &timezone) < 4
                     })
                     .take(25)
                     .map(|t| t.to_owned())
@@ -74,25 +83,21 @@ async fn timezone(ctx: &Context, invoke: &mut CommandInvoke, args: CommandOption
                     )
                 });
 
-                let _ = invoke
-                    .respond(
-                        ctx.http.clone(),
-                        CreateGenericResponse::new().embed(|e| {
-                            e.title("Timezone Not Recognized")
-                                .description("Possibly you meant one of the following timezones, otherwise click [here](https://gist.github.com/JellyWX/913dfc8b63d45192ad6cb54c829324ee):")
-                                .color(*THEME_COLOR)
-                                .fields(fields)
-                                .footer(|f| f.text(footer_text))
-                                .url("https://gist.github.com/JellyWX/913dfc8b63d45192ad6cb54c829324ee")
-                        }),
-                    )
-                    .await;
+                ctx.send(|m| {
+                    m.embed(|e| {
+                        e.title("Timezone Not Recognized")
+                            .description("Possibly you meant one of the following timezones, otherwise click [here](https://gist.github.com/JellyWX/913dfc8b63d45192ad6cb54c829324ee):")
+                            .color(*THEME_COLOR)
+                            .fields(fields)
+                            .footer(|f| f.text(footer_text))
+                            .url("https://gist.github.com/JellyWX/913dfc8b63d45192ad6cb54c829324ee")
+                    })
+                })
+                .await?;
             }
         }
     } else {
-        let popular_timezones = ctx.data.read().await.get::<PopularTimezones>().cloned().unwrap();
-
-        let popular_timezones_iter = popular_timezones.iter().map(|t| {
+        let popular_timezones_iter = ctx.data().popular_timezones.iter().map(|t| {
             (
                 t.to_string(),
                 format!("🕗 `{}`", Utc::now().with_timezone(t).format("%H:%M").to_string()),
@@ -100,279 +105,311 @@ async fn timezone(ctx: &Context, invoke: &mut CommandInvoke, args: CommandOption
             )
         });
 
-        let _ = invoke
-            .respond(
-                ctx.http.clone(),
-                CreateGenericResponse::new().embed(|e| {
-                    e.title("Timezone Usage")
-                        .description(
-                            "**Usage:**
+        ctx.send(|m| {
+            m.embed(|e| {
+                e.title("Timezone Usage")
+                    .description(
+                        "**Usage:**
 `/timezone Name`
 
 **Example:**
 `/timezone Europe/London`
 
 You may want to use one of the popular timezones below, otherwise click [here](https://gist.github.com/JellyWX/913dfc8b63d45192ad6cb54c829324ee):",
+                    )
+                    .color(*THEME_COLOR)
+                    .fields(popular_timezones_iter)
+                    .footer(|f| f.text(footer_text))
+                    .url("https://gist.github.com/JellyWX/913dfc8b63d45192ad6cb54c829324ee")
+            })
+        })
+        .await?;
+    }
+
+    Ok(())
+}
+
+async fn macro_name_autocomplete(ctx: Context<'_>, partial: String) -> Vec<String> {
+    sqlx::query!(
+        "
+SELECT name
+FROM macro
+WHERE
+    guild_id = (SELECT id FROM guilds WHERE guild = ?)
+    AND name LIKE CONCAT(?, '%')",
+        ctx.guild_id().unwrap().0,
+        partial,
+    )
+    .fetch_all(&ctx.data().database)
+    .await
+    .unwrap_or(vec![])
+    .iter()
+    .map(|s| s.name.clone())
+    .collect()
+}
+
+/// Record and replay command sequences
+#[poise::command(
+    slash_command,
+    rename = "macro",
+    guild_only = true,
+    default_member_permissions = "MANAGE_GUILD",
+    identifying_name = "macro_base"
+)]
+pub async fn macro_base(_ctx: Context<'_>) -> Result<(), Error> {
+    Ok(())
+}
+
+/// Start recording up to 5 commands to replay
+#[poise::command(
+    slash_command,
+    rename = "record",
+    guild_only = true,
+    default_member_permissions = "MANAGE_GUILD",
+    identifying_name = "record_macro"
+)]
+pub async fn record_macro(
+    ctx: Context<'_>,
+    #[description = "Name for the new macro"] name: String,
+    #[description = "Description for the new macro"] description: Option<String>,
+) -> Result<(), Error> {
+    let guild_id = ctx.guild_id().unwrap();
+
+    let row = sqlx::query!(
+        "
+SELECT 1 as _e FROM macro WHERE guild_id = (SELECT id FROM guilds WHERE guild = ?) AND name = ?",
+        guild_id.0,
+        name
+    )
+    .fetch_one(&ctx.data().database)
+    .await;
+
+    if row.is_ok() {
+        ctx.send(|m| {
+            m.ephemeral(true).embed(|e| {
+                e.title("Unique Name Required")
+                    .description(
+                        "A macro already exists under this name.
+Please select a unique name for your macro.",
+                    )
+                    .color(*THEME_COLOR)
+            })
+        })
+        .await?;
+    } else {
+        let okay = {
+            let mut lock = ctx.data().recording_macros.write().await;
+
+            if lock.contains_key(&(guild_id, ctx.author().id)) {
+                false
+            } else {
+                lock.insert(
+                    (guild_id, ctx.author().id),
+                    CommandMacro { guild_id, name, description, commands: vec![] },
+                );
+                true
+            }
+        };
+
+        if okay {
+            ctx.send(|m| {
+                m.ephemeral(true).embed(|e| {
+                    e.title("Macro Recording Started")
+                        .description(
+                            "Run up to 5 commands, or type `/macro finish` to stop at any point.
+Any commands ran as part of recording will be inconsequential",
                         )
                         .color(*THEME_COLOR)
-                        .fields(popular_timezones_iter)
-                        .footer(|f| f.text(footer_text))
-                        .url("https://gist.github.com/JellyWX/913dfc8b63d45192ad6cb54c829324ee")
-                }),
-            )
-            .await;
-    }
-}
-
-#[command("macro")]
-#[description("Record and replay command sequences")]
-#[subcommand("record")]
-#[description("Start recording up to 5 commands to replay")]
-#[arg(name = "name", description = "Name for the new macro", kind = "String", required = true)]
-#[arg(
-    name = "description",
-    description = "Description for the new macro",
-    kind = "String",
-    required = false
-)]
-#[subcommand("finish")]
-#[description("Finish current recording")]
-#[subcommand("list")]
-#[description("List recorded macros")]
-#[subcommand("run")]
-#[description("Run a recorded macro")]
-#[arg(name = "name", description = "Name of the macro to run", kind = "String", required = true)]
-#[subcommand("delete")]
-#[description("Delete a recorded macro")]
-#[arg(name = "name", description = "Name of the macro to delete", kind = "String", required = true)]
-#[supports_dm(false)]
-#[hook(GUILD_ONLY_HOOK)]
-#[hook(CHECK_GUILD_PERMISSIONS_HOOK)]
-async fn macro_cmd(ctx: &Context, invoke: &mut CommandInvoke, args: CommandOptions) {
-    let pool = ctx.data.read().await.get::<SQLPool>().cloned().unwrap();
-
-    match args.subcommand.clone().unwrap().as_str() {
-        "record" => {
-            let guild_id = invoke.guild_id().unwrap();
-
-            let name = args.get("name").unwrap().to_string();
-
-            let row = sqlx::query!(
-                "SELECT 1 as _e FROM macro WHERE guild_id = (SELECT id FROM guilds WHERE guild = ?) AND name = ?",
-                guild_id.0,
-                name
-            )
-            .fetch_one(&pool)
-            .await;
-
-            if row.is_ok() {
-                let _ = invoke
-                    .respond(
-                        &ctx,
-                        CreateGenericResponse::new().ephemeral().embed(|e| {
-                            e
-                            .title("Unique Name Required")
-                            .description("A macro already exists under this name. Please select a unique name for your macro.")
-                            .color(*THEME_COLOR)
-                        }),
-                    )
-                    .await;
-            } else {
-                let macro_buffer = ctx.data.read().await.get::<RecordingMacros>().cloned().unwrap();
-
-                let okay = {
-                    let mut lock = macro_buffer.write().await;
-
-                    if lock.contains_key(&(guild_id, invoke.author_id())) {
-                        false
-                    } else {
-                        lock.insert(
-                            (guild_id, invoke.author_id()),
-                            CommandMacro {
-                                guild_id,
-                                name,
-                                description: args.get("description").map(|d| d.to_string()),
-                                commands: vec![],
-                            },
-                        );
-                        true
-                    }
-                };
-
-                if okay {
-                    let _ = invoke
-                        .respond(
-                            &ctx,
-                            CreateGenericResponse::new().ephemeral().embed(|e| {
-                                e
-                                .title("Macro Recording Started")
-                                .description(
-"Run up to 5 commands, or type `/macro finish` to stop at any point.
-Any commands ran as part of recording will be inconsequential")
-                                .color(*THEME_COLOR)
-                            }),
-                        )
-                        .await;
-                } else {
-                    let _ = invoke
-                        .respond(
-                            &ctx,
-                            CreateGenericResponse::new().ephemeral().embed(|e| {
-                                e.title("Macro Already Recording")
-                                    .description(
-                                        "You are already recording a macro in this server.
+                })
+            })
+            .await?;
+        } else {
+            ctx.send(|m| {
+                m.ephemeral(true).embed(|e| {
+                    e.title("Macro Already Recording")
+                        .description(
+                            "You are already recording a macro in this server.
 Please use `/macro finish` to end this recording before starting another.",
-                                    )
-                                    .color(*THEME_COLOR)
-                            }),
                         )
-                        .await;
-                }
-            }
+                        .color(*THEME_COLOR)
+                })
+            })
+            .await?;
         }
-        "finish" => {
-            let key = (invoke.guild_id().unwrap(), invoke.author_id());
-            let macro_buffer = ctx.data.read().await.get::<RecordingMacros>().cloned().unwrap();
-
-            {
-                let lock = macro_buffer.read().await;
-                let contained = lock.get(&key);
-
-                if contained.map_or(true, |cmacro| cmacro.commands.is_empty()) {
-                    let _ = invoke
-                        .respond(
-                            &ctx,
-                            CreateGenericResponse::new().embed(|e| {
-                                e.title("No Macro Recorded")
-                                    .description("Use `/macro record` to start recording a macro")
-                                    .color(*THEME_COLOR)
-                            }),
-                        )
-                        .await;
-                } else {
-                    let command_macro = contained.unwrap();
-                    let json = serde_json::to_string(&command_macro.commands).unwrap();
-
-                    sqlx::query!(
-                        "INSERT INTO macro (guild_id, name, description, commands) VALUES ((SELECT id FROM guilds WHERE guild = ?), ?, ?, ?)",
-                        command_macro.guild_id.0,
-                        command_macro.name,
-                        command_macro.description,
-                        json
-                    )
-                        .execute(&pool)
-                        .await
-                        .unwrap();
-
-                    let _ = invoke
-                        .respond(
-                            &ctx,
-                            CreateGenericResponse::new().embed(|e| {
-                                e.title("Macro Recorded")
-                                    .description("Use `/macro run` to execute the macro")
-                                    .color(*THEME_COLOR)
-                            }),
-                        )
-                        .await;
-                }
-            }
-
-            {
-                let mut lock = macro_buffer.write().await;
-                lock.remove(&key);
-            }
-        }
-        "list" => {
-            let macros = CommandMacro::from_guild(ctx, invoke.guild_id().unwrap()).await;
-
-            let resp = show_macro_page(&macros, 0);
-
-            invoke.respond(&ctx, resp).await.unwrap();
-        }
-        "run" => {
-            let macro_name = args.get("name").unwrap().to_string();
-
-            match sqlx::query!(
-                "SELECT commands FROM macro WHERE guild_id = (SELECT id FROM guilds WHERE guild = ?) AND name = ?",
-                invoke.guild_id().unwrap().0,
-                macro_name
-            )
-            .fetch_one(&pool)
-            .await
-            {
-                Ok(row) => {
-                    invoke.defer(&ctx).await;
-
-                    let commands: Vec<CommandOptions> =
-                        serde_json::from_str(&row.commands).unwrap();
-                    let framework = ctx.data.read().await.get::<RegexFramework>().cloned().unwrap();
-
-                    for command in commands {
-                        framework.run_command_from_options(ctx, invoke, command).await;
-                    }
-                }
-
-                Err(sqlx::Error::RowNotFound) => {
-                    let _ = invoke
-                        .respond(
-                            &ctx,
-                            CreateGenericResponse::new()
-                                .content(format!("Macro \"{}\" not found", macro_name)),
-                        )
-                        .await;
-                }
-
-                Err(e) => {
-                    panic!("{}", e);
-                }
-            }
-        }
-        "delete" => {
-            let macro_name = args.get("name").unwrap().to_string();
-
-            match sqlx::query!(
-                "SELECT id FROM macro WHERE guild_id = (SELECT id FROM guilds WHERE guild = ?) AND name = ?",
-                invoke.guild_id().unwrap().0,
-                macro_name
-            )
-            .fetch_one(&pool)
-            .await
-            {
-                Ok(row) => {
-                    sqlx::query!("DELETE FROM macro WHERE id = ?", row.id)
-                        .execute(&pool)
-                        .await
-                        .unwrap();
-
-                    let _ = invoke
-                        .respond(
-                            &ctx,
-                            CreateGenericResponse::new()
-                                .content(format!("Macro \"{}\" deleted", macro_name)),
-                        )
-                        .await;
-                }
-
-                Err(sqlx::Error::RowNotFound) => {
-                    let _ = invoke
-                        .respond(
-                            &ctx,
-                            CreateGenericResponse::new()
-                                .content(format!("Macro \"{}\" not found", macro_name)),
-                        )
-                        .await;
-                }
-
-                Err(e) => {
-                    panic!("{}", e);
-                }
-            }
-        }
-        _ => {}
     }
+
+    Ok(())
 }
 
-pub fn max_macro_page(macros: &[CommandMacro]) -> usize {
+/// Finish current macro recording
+#[poise::command(
+    slash_command,
+    rename = "finish",
+    guild_only = true,
+    default_member_permissions = "MANAGE_GUILD",
+    identifying_name = "finish_macro"
+)]
+pub async fn finish_macro(ctx: Context<'_>) -> Result<(), Error> {
+    let key = (ctx.guild_id().unwrap(), ctx.author().id);
+
+    {
+        let lock = ctx.data().recording_macros.read().await;
+        let contained = lock.get(&key);
+
+        if contained.map_or(true, |cmacro| cmacro.commands.is_empty()) {
+            ctx.send(|m| {
+                m.embed(|e| {
+                    e.title("No Macro Recorded")
+                        .description("Use `/macro record` to start recording a macro")
+                        .color(*THEME_COLOR)
+                })
+            })
+            .await?;
+        } else {
+            let command_macro = contained.unwrap();
+            let json = serde_json::to_string(&command_macro.commands).unwrap();
+
+            sqlx::query!(
+                "INSERT INTO macro (guild_id, name, description, commands) VALUES ((SELECT id FROM guilds WHERE guild = ?), ?, ?, ?)",
+                command_macro.guild_id.0,
+                command_macro.name,
+                command_macro.description,
+                json
+            )
+                .execute(&ctx.data().database)
+                .await
+                .unwrap();
+
+            ctx.send(|m| {
+                m.embed(|e| {
+                    e.title("Macro Recorded")
+                        .description("Use `/macro run` to execute the macro")
+                        .color(*THEME_COLOR)
+                })
+            })
+            .await?;
+        }
+    }
+
+    {
+        let mut lock = ctx.data().recording_macros.write().await;
+        lock.remove(&key);
+    }
+
+    Ok(())
+}
+
+/// List recorded macros
+#[poise::command(
+    slash_command,
+    rename = "list",
+    guild_only = true,
+    default_member_permissions = "MANAGE_GUILD",
+    identifying_name = "list_macro"
+)]
+pub async fn list_macro(ctx: Context<'_>) -> Result<(), Error> {
+    let macros = ctx.command_macros().await?;
+
+    let resp = show_macro_page(&macros, 0);
+
+    ctx.send(|m| {
+        *m = resp;
+        m
+    })
+    .await?;
+
+    Ok(())
+}
+
+/// Run a recorded macro
+#[poise::command(
+    slash_command,
+    rename = "run",
+    guild_only = true,
+    default_member_permissions = "MANAGE_GUILD",
+    identifying_name = "run_macro"
+)]
+pub async fn run_macro(
+    ctx: poise::ApplicationContext<'_, Data, Error>,
+    #[description = "Name of macro to run"]
+    #[autocomplete = "macro_name_autocomplete"]
+    name: String,
+) -> Result<(), Error> {
+    match guild_command_macro(&Context::Application(ctx), &name).await {
+        Some(command_macro) => {
+            ctx.defer_response(false).await?;
+
+            for command in command_macro.commands {
+                if let Some(action) = command.action {
+                    match (action)(poise::ApplicationContext { args: &command.options, ..ctx })
+                        .await
+                    {
+                        Ok(()) => {}
+                        Err(e) => {
+                            println!("{:?}", e);
+                        }
+                    }
+                } else {
+                    Context::Application(ctx)
+                        .say(format!("Command \"{}\" not found", command.command_name))
+                        .await?;
+                }
+            }
+        }
+
+        None => {
+            Context::Application(ctx).say(format!("Macro \"{}\" not found", name)).await?;
+        }
+    }
+
+    Ok(())
+}
+
+/// Delete a recorded macro
+#[poise::command(
+    slash_command,
+    rename = "delete",
+    guild_only = true,
+    default_member_permissions = "MANAGE_GUILD",
+    identifying_name = "delete_macro"
+)]
+pub async fn delete_macro(
+    ctx: Context<'_>,
+    #[description = "Name of macro to delete"]
+    #[autocomplete = "macro_name_autocomplete"]
+    name: String,
+) -> Result<(), Error> {
+    match sqlx::query!(
+        "
+SELECT id FROM macro WHERE guild_id = (SELECT id FROM guilds WHERE guild = ?) AND name = ?",
+        ctx.guild_id().unwrap().0,
+        name
+    )
+    .fetch_one(&ctx.data().database)
+    .await
+    {
+        Ok(row) => {
+            sqlx::query!("DELETE FROM macro WHERE id = ?", row.id)
+                .execute(&ctx.data().database)
+                .await
+                .unwrap();
+
+            ctx.say(format!("Macro \"{}\" deleted", name)).await?;
+        }
+
+        Err(sqlx::Error::RowNotFound) => {
+            ctx.say(format!("Macro \"{}\" not found", name)).await?;
+        }
+
+        Err(e) => {
+            panic!("{}", e);
+        }
+    }
+
+    Ok(())
+}
+
+pub fn max_macro_page<U, E>(macros: &[CommandMacro<U, E>]) -> usize {
     let mut skipped_char_count = 0;
 
     macros
@@ -396,15 +433,19 @@ pub fn max_macro_page(macros: &[CommandMacro]) -> usize {
         })
 }
 
-pub fn show_macro_page(macros: &[CommandMacro], page: usize) -> CreateGenericResponse {
+pub fn show_macro_page<U, E>(macros: &[CommandMacro<U, E>], page: usize) -> CreateReply {
     let pager = MacroPager::new(page);
 
     if macros.is_empty() {
-        return CreateGenericResponse::new().embed(|e| {
+        let mut reply = CreateReply::default();
+
+        reply.embed(|e| {
             e.title("Macros")
                 .description("No Macros Set Up. Use `/macro record` to get started.")
                 .color(*THEME_COLOR)
         });
+
+        return reply;
     }
 
     let pages = max_macro_page(macros);
@@ -447,7 +488,9 @@ pub fn show_macro_page(macros: &[CommandMacro], page: usize) -> CreateGenericRes
 
     let display = display_vec.join("\n");
 
-    CreateGenericResponse::new()
+    let mut reply = CreateReply::default();
+
+    reply
         .embed(|e| {
             e.title("Macros")
                 .description(display)
@@ -458,5 +501,7 @@ pub fn show_macro_page(macros: &[CommandMacro], page: usize) -> CreateGenericRes
             pager.create_button_row(pages, comp);
 
             comp
-        })
+        });
+
+    reply
 }

src/commands/reminder_cmds.rs

@@ -7,17 +7,20 @@ use std::{
 use chrono::NaiveDateTime;
 use chrono_tz::Tz;
 use num_integer::Integer;
-use regex_command_attr::command;
-use serenity::{builder::CreateEmbed, client::Context, model::channel::Channel};
+use poise::{
+    serenity::{builder::CreateEmbed, model::channel::Channel},
+    CreateReply,
+};
 
 use crate::{
     component_models::{
         pager::{DelPager, LookPager, Pager},
         ComponentDataModel, DelSelector,
     },
-    consts::{EMBED_DESCRIPTION_MAX_LENGTH, REGEX_CHANNEL_USER, SELECT_MAX_ENTRIES, THEME_COLOR},
-    framework::{CommandInvoke, CommandOptions, CreateGenericResponse, OptionValue},
-    hooks::CHECK_GUILD_PERMISSIONS_HOOK,
+    consts::{
+        EMBED_DESCRIPTION_MAX_LENGTH, HOUR, MINUTE, REGEX_CHANNEL_USER, SELECT_MAX_ENTRIES,
+        THEME_COLOR,
+    },
     interval_parser::parse_duration,
     models::{
         reminder::{
@@ -28,34 +31,30 @@ use crate::{
             Reminder,
         },
         timer::Timer,
-        user_data::UserData,
         CtxData,
     },
     time_parser::natural_parser,
     utils::{check_guild_subscription, check_subscription},
-    SQLPool,
+    Context, Error,
 };
 
-#[command("pause")]
-#[description("Pause all reminders on the current channel until a certain time or indefinitely")]
-#[arg(
-    name = "until",
-    description = "When to pause until (hint: try 'next Wednesday', or '10 minutes')",
-    kind = "String",
-    required = false
+/// Pause all reminders on the current channel until a certain time or indefinitely
+#[poise::command(
+    slash_command,
+    identifying_name = "pause",
+    default_member_permissions = "MANAGE_GUILD"
 )]
-#[supports_dm(false)]
-#[hook(CHECK_GUILD_PERMISSIONS_HOOK)]
-async fn pause(ctx: &Context, invoke: &mut CommandInvoke, args: CommandOptions) {
-    let pool = ctx.data.read().await.get::<SQLPool>().cloned().unwrap();
+pub async fn pause(
+    ctx: Context<'_>,
+    #[description = "When to pause until"] until: Option<String>,
+) -> Result<(), Error> {
+    let timezone = ctx.timezone().await;
 
-    let timezone = UserData::timezone_of(&invoke.author_id(), &pool).await;
+    let mut channel = ctx.channel_data().await.unwrap();
 
-    let mut channel = ctx.channel_data(invoke.channel_id()).await.unwrap();
-
-    match args.get("until") {
-        Some(OptionValue::String(until)) => {
-            let parsed = natural_parser(until, &timezone.to_string()).await;
+    match until {
+        Some(until) => {
+            let parsed = natural_parser(&until, &timezone.to_string()).await;
 
             if let Some(timestamp) = parsed {
                 let dt = NaiveDateTime::from_timestamp(timestamp, 0);
@@ -63,92 +62,57 @@ async fn pause(ctx: &Context, invoke: &mut CommandInvoke, args: CommandOptions)
                 channel.paused = true;
                 channel.paused_until = Some(dt);
 
-                channel.commit_changes(&pool).await;
+                channel.commit_changes(&ctx.data().database).await;
 
-                let _ = invoke
-                    .respond(
-                        ctx.http.clone(),
-                        CreateGenericResponse::new().content(format!(
-                            "Reminders in this channel have been silenced until **<t:{}:D>**",
-                            timestamp
-                        )),
-                    )
-                    .await;
+                ctx.say(format!(
+                    "Reminders in this channel have been silenced until **<t:{}:D>**",
+                    timestamp
+                ))
+                .await?;
             } else {
-                let _ = invoke
-                    .respond(
-                        ctx.http.clone(),
-                        CreateGenericResponse::new()
-                            .content("Time could not be processed. Please write the time as clearly as possible"),
-                    )
-                    .await;
+                ctx.say(
+                    "Time could not be processed. Please write the time as clearly as possible",
+                )
+                .await?;
             }
         }
         _ => {
             channel.paused = !channel.paused;
             channel.paused_until = None;
 
-            channel.commit_changes(&pool).await;
+            channel.commit_changes(&ctx.data().database).await;
 
             if channel.paused {
-                let _ = invoke
-                    .respond(
-                        ctx.http.clone(),
-                        CreateGenericResponse::new()
-                            .content("Reminders in this channel have been silenced indefinitely"),
-                    )
-                    .await;
+                ctx.say("Reminders in this channel have been silenced indefinitely").await?;
             } else {
-                let _ = invoke
-                    .respond(
-                        ctx.http.clone(),
-                        CreateGenericResponse::new()
-                            .content("Reminders in this channel have been unsilenced"),
-                    )
-                    .await;
+                ctx.say("Reminders in this channel have been unsilenced").await?;
             }
         }
     }
+
+    Ok(())
 }
 
-#[command("offset")]
-#[description("Move all reminders in the current server by a certain amount of time. Times get added together")]
-#[arg(
-    name = "hours",
-    description = "Number of hours to offset by",
-    kind = "Integer",
-    required = false
+/// Move all reminders in the current server by a certain amount of time. Times get added together
+#[poise::command(
+    slash_command,
+    identifying_name = "offset",
+    default_member_permissions = "MANAGE_GUILD"
 )]
-#[arg(
-    name = "minutes",
-    description = "Number of minutes to offset by",
-    kind = "Integer",
-    required = false
-)]
-#[arg(
-    name = "seconds",
-    description = "Number of seconds to offset by",
-    kind = "Integer",
-    required = false
-)]
-#[hook(CHECK_GUILD_PERMISSIONS_HOOK)]
-async fn offset(ctx: &Context, invoke: &mut CommandInvoke, args: CommandOptions) {
-    let pool = ctx.data.read().await.get::<SQLPool>().cloned().unwrap();
-
-    let combined_time = args.get("hours").map_or(0, |h| h.as_i64().unwrap() * 3600)
-        + args.get("minutes").map_or(0, |m| m.as_i64().unwrap() * 60)
-        + args.get("seconds").map_or(0, |s| s.as_i64().unwrap());
+pub async fn offset(
+    ctx: Context<'_>,
+    #[description = "Number of hours to offset by"] hours: Option<isize>,
+    #[description = "Number of minutes to offset by"] minutes: Option<isize>,
+    #[description = "Number of seconds to offset by"] seconds: Option<isize>,
+) -> Result<(), Error> {
+    let combined_time = hours.map_or(0, |h| h * HOUR as isize)
+        + minutes.map_or(0, |m| m * MINUTE as isize)
+        + seconds.map_or(0, |s| s);
 
     if combined_time == 0 {
-        let _ = invoke
-            .respond(
-                ctx.http.clone(),
-                CreateGenericResponse::new()
-                    .content("Please specify one of `hours`, `minutes` or `seconds`"),
-            )
-            .await;
+        ctx.say("Please specify one of `hours`, `minutes` or `seconds`").await?;
     } else {
-        if let Some(guild) = invoke.guild(ctx.cache.clone()) {
+        if let Some(guild) = ctx.guild() {
             let channels = guild
                 .channels
                 .iter()
@@ -167,110 +131,75 @@ INNER JOIN
     `channels` ON `channels`.id = reminders.channel_id
 SET reminders.`utc_time` = DATE_ADD(reminders.`utc_time`, INTERVAL ? SECOND)
 WHERE FIND_IN_SET(channels.`channel`, ?)",
-                combined_time,
+                combined_time as i64,
                 channels
             )
-            .execute(&pool)
+            .execute(&ctx.data().database)
             .await
             .unwrap();
         } else {
             sqlx::query!(
                 "UPDATE reminders INNER JOIN `channels` ON `channels`.id = reminders.channel_id SET reminders.`utc_time` = reminders.`utc_time` + ? WHERE channels.`channel` = ?",
-                combined_time,
-                invoke.channel_id().0
+                combined_time as i64,
+                ctx.channel_id().0
             )
-            .execute(&pool)
+            .execute(&ctx.data().database)
             .await
             .unwrap();
         }
 
-        let _ = invoke
-            .respond(
-                ctx.http.clone(),
-                CreateGenericResponse::new()
-                    .content(format!("All reminders offset by {} seconds", combined_time)),
-            )
-            .await;
+        ctx.say(format!("All reminders offset by {} seconds", combined_time)).await?;
     }
+
+    Ok(())
 }
 
-#[command("nudge")]
-#[description("Nudge all future reminders on this channel by a certain amount (don't use for DST! See `/offset`)")]
-#[arg(
-    name = "minutes",
-    description = "Number of minutes to nudge new reminders by",
-    kind = "Integer",
-    required = false
+/// Nudge all future reminders on this channel by a certain amount (don't use for DST! See `/offset`)
+#[poise::command(
+    slash_command,
+    identifying_name = "nudge",
+    default_member_permissions = "MANAGE_GUILD"
 )]
-#[arg(
-    name = "seconds",
-    description = "Number of seconds to nudge new reminders by",
-    kind = "Integer",
-    required = false
-)]
-#[hook(CHECK_GUILD_PERMISSIONS_HOOK)]
-async fn nudge(ctx: &Context, invoke: &mut CommandInvoke, args: CommandOptions) {
-    let pool = ctx.data.read().await.get::<SQLPool>().cloned().unwrap();
+pub async fn nudge(
+    ctx: Context<'_>,
+    #[description = "Number of minutes to nudge new reminders by"] minutes: Option<isize>,
+    #[description = "Number of seconds to nudge new reminders by"] seconds: Option<isize>,
+) -> Result<(), Error> {
+    let combined_time = minutes.map_or(0, |m| m * MINUTE as isize) + seconds.map_or(0, |s| s);
 
-    let combined_time = args.get("minutes").map_or(0, |m| m.as_i64().unwrap() * 60)
-        + args.get("seconds").map_or(0, |s| s.as_i64().unwrap());
-
-    if combined_time < i16::MIN as i64 || combined_time > i16::MAX as i64 {
-        let _ = invoke
-            .respond(
-                ctx.http.clone(),
-                CreateGenericResponse::new().content("Nudge times must be less than 500 minutes"),
-            )
-            .await;
+    if combined_time < i16::MIN as isize || combined_time > i16::MAX as isize {
+        ctx.say("Nudge times must be less than 500 minutes").await?;
     } else {
-        let mut channel_data = ctx.channel_data(invoke.channel_id()).await.unwrap();
+        let mut channel_data = ctx.channel_data().await.unwrap();
 
         channel_data.nudge = combined_time as i16;
-        channel_data.commit_changes(&pool).await;
+        channel_data.commit_changes(&ctx.data().database).await;
 
-        let _ = invoke
-            .respond(
-                ctx.http.clone(),
-                CreateGenericResponse::new().content(format!(
-                    "Future reminders will be nudged by {} seconds",
-                    combined_time
-                )),
-            )
-            .await;
+        ctx.say(format!("Future reminders will be nudged by {} seconds", combined_time)).await?;
     }
+
+    Ok(())
 }
 
-#[command("look")]
-#[description("View reminders on a specific channel")]
-#[arg(
-    name = "channel",
-    description = "The channel to view reminders on",
-    kind = "Channel",
-    required = false
+/// View reminders on a specific channel
+#[poise::command(
+    slash_command,
+    identifying_name = "look",
+    default_member_permissions = "MANAGE_GUILD"
 )]
-#[arg(
-    name = "disabled",
-    description = "Whether to show disabled reminders or not",
-    kind = "Boolean",
-    required = false
-)]
-#[arg(
-    name = "relative",
-    description = "Whether to display times as relative or exact times",
-    kind = "Boolean",
-    required = false
-)]
-#[hook(CHECK_GUILD_PERMISSIONS_HOOK)]
-async fn look(ctx: &Context, invoke: &mut CommandInvoke, args: CommandOptions) {
-    let pool = ctx.data.read().await.get::<SQLPool>().cloned().unwrap();
-
-    let timezone = UserData::timezone_of(&invoke.author_id(), &pool).await;
+pub async fn look(
+    ctx: Context<'_>,
+    #[description = "Channel to view reminders on"] channel: Option<Channel>,
+    #[description = "Whether to show disabled reminders or not"] disabled: Option<bool>,
+    #[description = "Whether to display times as relative or exact times"] relative: Option<bool>,
+) -> Result<(), Error> {
+    let timezone = ctx.timezone().await;
 
     let flags = LookFlags {
-        show_disabled: args.get("disabled").map(|i| i.as_bool()).flatten().unwrap_or(true),
-        channel_id: args.get("channel").map(|i| i.as_channel_id()).flatten(),
-        time_display: args.get("relative").map_or(TimeDisplayType::Relative, |b| {
-            if b.as_bool() == Some(true) {
+        show_disabled: disabled.unwrap_or(true),
+        channel_id: channel.map(|c| c.id()),
+        time_display: relative.map_or(TimeDisplayType::Relative, |b| {
+            if b {
                 TimeDisplayType::Relative
             } else {
                 TimeDisplayType::Absolute
@@ -278,33 +207,29 @@ async fn look(ctx: &Context, invoke: &mut CommandInvoke, args: CommandOptions) {
         }),
     };
 
-    let channel_opt = invoke.channel_id().to_channel_cached(&ctx);
+    let channel_opt = ctx.channel_id().to_channel_cached(&ctx.discord());
 
     let channel_id = if let Some(Channel::Guild(channel)) = channel_opt {
-        if Some(channel.guild_id) == invoke.guild_id() {
-            flags.channel_id.unwrap_or_else(|| invoke.channel_id())
+        if Some(channel.guild_id) == ctx.guild_id() {
+            flags.channel_id.unwrap_or_else(|| ctx.channel_id())
         } else {
-            invoke.channel_id()
+            ctx.channel_id()
         }
     } else {
-        invoke.channel_id()
+        ctx.channel_id()
     };
 
-    let channel_name = if let Some(Channel::Guild(channel)) = channel_id.to_channel_cached(&ctx) {
-        Some(channel.name)
-    } else {
-        None
-    };
+    let channel_name =
+        if let Some(Channel::Guild(channel)) = channel_id.to_channel_cached(&ctx.discord()) {
+            Some(channel.name)
+        } else {
+            None
+        };
 
-    let reminders = Reminder::from_channel(ctx, channel_id, &flags).await;
+    let reminders = Reminder::from_channel(&ctx.data().database, channel_id, &flags).await;
 
     if reminders.is_empty() {
-        let _ = invoke
-            .respond(
-                ctx.http.clone(),
-                CreateGenericResponse::new().content("No reminders on specified channel"),
-            )
-            .await;
+        let _ = ctx.say("No reminders on specified channel").await;
     } else {
         let mut char_count = 0;
 
@@ -327,41 +252,52 @@ async fn look(ctx: &Context, invoke: &mut CommandInvoke, args: CommandOptions) {
 
         let pager = LookPager::new(flags, timezone);
 
-        invoke
-            .respond(
-                ctx.http.clone(),
-                CreateGenericResponse::new()
-                    .embed(|e| {
-                        e.title(format!(
-                            "Reminders{}",
-                            channel_name.map_or(String::new(), |n| format!(" on #{}", n))
-                        ))
-                        .description(display)
-                        .footer(|f| f.text(format!("Page {} of {}", 1, pages)))
-                        .color(*THEME_COLOR)
-                    })
-                    .components(|comp| {
-                        pager.create_button_row(pages, comp);
+        ctx.send(|r| {
+            r.ephemeral(true)
+                .embed(|e| {
+                    e.title(format!(
+                        "Reminders{}",
+                        channel_name.map_or(String::new(), |n| format!(" on #{}", n))
+                    ))
+                    .description(display)
+                    .footer(|f| f.text(format!("Page {} of {}", 1, pages)))
+                    .color(*THEME_COLOR)
+                })
+                .components(|comp| {
+                    pager.create_button_row(pages, comp);
 
-                        comp
-                    }),
-            )
-            .await
-            .unwrap();
+                    comp
+                })
+        })
+        .await?;
     }
+
+    Ok(())
 }
 
-#[command("del")]
-#[description("Delete reminders")]
-#[hook(CHECK_GUILD_PERMISSIONS_HOOK)]
-async fn delete(ctx: &Context, invoke: &mut CommandInvoke, _args: CommandOptions) {
-    let timezone = ctx.timezone(invoke.author_id()).await;
+/// Delete reminders
+#[poise::command(
+    slash_command,
+    rename = "del",
+    identifying_name = "delete",
+    default_member_permissions = "MANAGE_GUILD"
+)]
+pub async fn delete(ctx: Context<'_>) -> Result<(), Error> {
+    let timezone = ctx.timezone().await;
 
-    let reminders = Reminder::from_guild(ctx, invoke.guild_id(), invoke.author_id()).await;
+    let reminders =
+        Reminder::from_guild(&ctx.discord(), &ctx.data().database, ctx.guild_id(), ctx.author().id)
+            .await;
 
     let resp = show_delete_page(&reminders, 0, timezone);
 
-    let _ = invoke.respond(&ctx, resp).await;
+    ctx.send(|r| {
+        *r = resp;
+        r
+    })
+    .await?;
+
+    Ok(())
 }
 
 pub fn max_delete_page(reminders: &[Reminder], timezone: &Tz) -> usize {
@@ -386,20 +322,20 @@ pub fn max_delete_page(reminders: &[Reminder], timezone: &Tz) -> usize {
         })
 }
 
-pub fn show_delete_page(
-    reminders: &[Reminder],
-    page: usize,
-    timezone: Tz,
-) -> CreateGenericResponse {
+pub fn show_delete_page(reminders: &[Reminder], page: usize, timezone: Tz) -> CreateReply {
     let pager = DelPager::new(page, timezone);
 
     if reminders.is_empty() {
-        return CreateGenericResponse::new()
+        let mut reply = CreateReply::default();
+
+        reply
             .embed(|e| e.title("Delete Reminders").description("No Reminders").color(*THEME_COLOR))
             .components(|comp| {
                 pager.create_button_row(0, comp);
                 comp
             });
+
+        return reply;
     }
 
     let pages = max_delete_page(reminders, &timezone);
@@ -448,7 +384,9 @@ pub fn show_delete_page(
 
     let del_selector = ComponentDataModel::DelSelector(DelSelector { page, timezone });
 
-    CreateGenericResponse::new()
+    let mut reply = CreateReply::default();
+
+    reply
         .embed(|e| {
             e.title("Delete Reminders")
                 .description(display)
@@ -486,290 +424,230 @@ pub fn show_delete_page(
                     })
                 })
             })
+        });
+
+    reply
+}
+
+fn time_difference(start_time: NaiveDateTime) -> String {
+    let unix_time = SystemTime::now().duration_since(UNIX_EPOCH).unwrap().as_secs() as i64;
+    let now = NaiveDateTime::from_timestamp(unix_time, 0);
+
+    let delta = (now - start_time).num_seconds();
+
+    let (minutes, seconds) = delta.div_rem(&60);
+    let (hours, minutes) = minutes.div_rem(&60);
+    let (days, hours) = hours.div_rem(&24);
+
+    format!("{} days, {:02}:{:02}:{:02}", days, hours, minutes, seconds)
+}
+
+/// Manage timers
+#[poise::command(
+    slash_command,
+    rename = "timer",
+    identifying_name = "timer_base",
+    default_member_permissions = "MANAGE_GUILD"
+)]
+pub async fn timer_base(_ctx: Context<'_>) -> Result<(), Error> {
+    Ok(())
+}
+
+/// List the timers in this server or DM channel
+#[poise::command(
+    slash_command,
+    rename = "list",
+    identifying_name = "list_timer",
+    default_member_permissions = "MANAGE_GUILD"
+)]
+pub async fn list_timer(ctx: Context<'_>) -> Result<(), Error> {
+    let owner = ctx.guild_id().map(|g| g.0).unwrap_or_else(|| ctx.author().id.0);
+
+    let timers = Timer::from_owner(owner, &ctx.data().database).await;
+
+    if !timers.is_empty() {
+        ctx.send(|m| {
+            m.embed(|e| {
+                e.fields(timers.iter().map(|timer| {
+                    (&timer.name, format!("⌚ `{}`", time_difference(timer.start_time)), false)
+                }))
+                .color(*THEME_COLOR)
+            })
         })
+        .await?;
+    } else {
+        ctx.say("No timers currently. Use `/timer start` to create a new timer").await?;
+    }
+
+    Ok(())
 }
 
-#[command("timer")]
-#[description("Manage timers")]
-#[subcommand("list")]
-#[description("List the timers in this server or DM channel")]
-#[subcommand("start")]
-#[description("Start a new timer from now")]
-#[arg(name = "name", description = "Name for the new timer", kind = "String", required = true)]
-#[subcommand("delete")]
-#[description("Delete a timer")]
-#[arg(name = "name", description = "Name of the timer to delete", kind = "String", required = true)]
-#[hook(CHECK_GUILD_PERMISSIONS_HOOK)]
-async fn timer(ctx: &Context, invoke: &mut CommandInvoke, args: CommandOptions) {
-    fn time_difference(start_time: NaiveDateTime) -> String {
-        let unix_time = SystemTime::now().duration_since(UNIX_EPOCH).unwrap().as_secs() as i64;
-        let now = NaiveDateTime::from_timestamp(unix_time, 0);
+/// Start a new timer from now
+#[poise::command(
+    slash_command,
+    rename = "start",
+    identifying_name = "start_timer",
+    default_member_permissions = "MANAGE_GUILD"
+)]
+pub async fn start_timer(
+    ctx: Context<'_>,
+    #[description = "Name for the new timer"] name: String,
+) -> Result<(), Error> {
+    let owner = ctx.guild_id().map(|g| g.0).unwrap_or_else(|| ctx.author().id.0);
 
-        let delta = (now - start_time).num_seconds();
+    let count = Timer::count_from_owner(owner, &ctx.data().database).await;
 
-        let (minutes, seconds) = delta.div_rem(&60);
-        let (hours, minutes) = minutes.div_rem(&60);
-        let (days, hours) = hours.div_rem(&24);
+    if count >= 25 {
+        ctx.say("You already have 25 timers. Please delete some timers before creating a new one")
+            .await?;
+    } else {
+        if name.len() <= 32 {
+            Timer::create(&name, owner, &ctx.data().database).await;
 
-        format!("{} days, {:02}:{:02}:{:02}", days, hours, minutes, seconds)
+            ctx.say("Created a new timer").await?;
+        } else {
+            ctx.say(format!(
+                "Please name your timer something shorted (max. 32 characters, you used {})",
+                name.len()
+            ))
+            .await?;
+        }
     }
 
-    let pool = ctx.data.read().await.get::<SQLPool>().cloned().unwrap();
-
-    let owner = invoke.guild_id().map(|g| g.0).unwrap_or_else(|| invoke.author_id().0);
-
-    match args.subcommand.clone().unwrap().as_str() {
-        "start" => {
-            let count = Timer::count_from_owner(owner, &pool).await;
-
-            if count >= 25 {
-                let _ = invoke
-                    .respond(
-                        ctx.http.clone(),
-                        CreateGenericResponse::new()
-                            .content("You already have 25 timers. Please delete some timers before creating a new one"),
-                    )
-                    .await;
-            } else {
-                let name = args.get("name").unwrap().to_string();
-
-                if name.len() <= 32 {
-                    Timer::create(&name, owner, &pool).await;
-
-                    let _ = invoke
-                        .respond(
-                            ctx.http.clone(),
-                            CreateGenericResponse::new().content("Created a new timer"),
-                        )
-                        .await;
-                } else {
-                    let _ = invoke
-                        .respond(
-                            ctx.http.clone(),
-                            CreateGenericResponse::new()
-                                .content(format!("Please name your timer something shorted (max. 32 characters, you used {})", name.len())),
-                        )
-                        .await;
-                }
-            }
-        }
-        "delete" => {
-            let name = args.get("name").unwrap().to_string();
-
-            let exists = sqlx::query!(
-                "
-SELECT 1 as _r FROM timers WHERE owner = ? AND name = ?
-                    ",
-                owner,
-                name
-            )
-            .fetch_one(&pool)
-            .await;
-
-            if exists.is_ok() {
-                sqlx::query!(
-                    "
-DELETE FROM timers WHERE owner = ? AND name = ?
-                        ",
-                    owner,
-                    name
-                )
-                .execute(&pool)
-                .await
-                .unwrap();
-
-                let _ = invoke
-                    .respond(
-                        ctx.http.clone(),
-                        CreateGenericResponse::new().content("Deleted a timer"),
-                    )
-                    .await;
-            } else {
-                let _ = invoke
-                    .respond(
-                        ctx.http.clone(),
-                        CreateGenericResponse::new().content("Could not find a timer by that name"),
-                    )
-                    .await;
-            }
-        }
-        "list" => {
-            let timers = Timer::from_owner(owner, &pool).await;
-
-            if !timers.is_empty() {
-                let _ = invoke
-                    .respond(
-                        ctx.http.clone(),
-                        CreateGenericResponse::new().embed(|e| {
-                            e.fields(timers.iter().map(|timer| {
-                                (
-                                    &timer.name,
-                                    format!("⌚ `{}`", time_difference(timer.start_time)),
-                                    false,
-                                )
-                            }))
-                            .color(*THEME_COLOR)
-                        }),
-                    )
-                    .await;
-            } else {
-                let _ = invoke
-                    .respond(
-                        ctx.http.clone(),
-                        CreateGenericResponse::new().content(
-                            "No timers currently. Use `/timer start` to create a new timer",
-                        ),
-                    )
-                    .await;
-            }
-        }
-        _ => {}
-    }
+    Ok(())
 }
 
-#[command("remind")]
-#[description("Create a new reminder")]
-#[arg(
-    name = "time",
-    description = "A description of the time to set the reminder for",
-    kind = "String",
-    required = true
+/// Delete a timer
+#[poise::command(
+    slash_command,
+    rename = "delete",
+    identifying_name = "delete_timer",
+    default_member_permissions = "MANAGE_GUILD"
 )]
-#[arg(
-    name = "content",
-    description = "The message content to send",
-    kind = "String",
-    required = true
-)]
-#[arg(
-    name = "channels",
-    description = "Channel or user mentions to set the reminder for",
-    kind = "String",
-    required = false
-)]
-#[arg(
-    name = "interval",
-    description = "(Patreon only) Time to wait before repeating the reminder. Leave blank for one-shot reminder",
-    kind = "String",
-    required = false
-)]
-#[arg(
-    name = "expires",
-    description = "(Patreon only) For repeating reminders, the time at which the reminder will stop sending",
-    kind = "String",
-    required = false
-)]
-#[arg(
-    name = "tts",
-    description = "Set the TTS flag on the reminder message (like the /tts command)",
-    kind = "Boolean",
-    required = false
-)]
-#[hook(CHECK_GUILD_PERMISSIONS_HOOK)]
-async fn remind(ctx: &Context, invoke: &mut CommandInvoke, args: CommandOptions) {
-    if args.get("interval").is_none() && args.get("expires").is_some() {
-        let _ = invoke
-            .respond(
-                &ctx,
-                CreateGenericResponse::new().content("`expires` can only be used with `interval`"),
-            )
+pub async fn delete_timer(
+    ctx: Context<'_>,
+    #[description = "Name of timer to delete"] name: String,
+) -> Result<(), Error> {
+    let owner = ctx.guild_id().map(|g| g.0).unwrap_or_else(|| ctx.author().id.0);
+
+    let exists =
+        sqlx::query!("SELECT 1 as _r FROM timers WHERE owner = ? AND name = ?", owner, name)
+            .fetch_one(&ctx.data().database)
             .await;
 
-        return;
+    if exists.is_ok() {
+        sqlx::query!("DELETE FROM timers WHERE owner = ? AND name = ?", owner, name)
+            .execute(&ctx.data().database)
+            .await
+            .unwrap();
+
+        ctx.say("Deleted a timer").await?;
+    } else {
+        ctx.say("Could not find a timer by that name").await?;
     }
 
-    invoke.defer(&ctx).await;
+    Ok(())
+}
 
-    let user_data = ctx.user_data(invoke.author_id()).await.unwrap();
-    let timezone = user_data.timezone();
+/// Create a new reminder
+#[poise::command(
+    slash_command,
+    identifying_name = "remind",
+    default_member_permissions = "MANAGE_GUILD"
+)]
+pub async fn remind(
+    ctx: Context<'_>,
+    #[description = "A description of the time to set the reminder for"] time: String,
+    #[description = "The message content to send"] content: String,
+    #[description = "Channel or user mentions to set the reminder for"] channels: Option<String>,
+    #[description = "(Patreon only) Time to wait before repeating the reminder. Leave blank for one-shot reminder"]
+    interval: Option<String>,
+    #[description = "(Patreon only) For repeating reminders, the time at which the reminder will stop sending"]
+    expires: Option<String>,
+    #[description = "Set the TTS flag on the reminder message, similar to the /tts command"]
+    tts: Option<bool>,
+) -> Result<(), Error> {
+    if interval.is_none() && expires.is_some() {
+        ctx.say("`expires` can only be used with `interval`").await?;
 
-    let time = {
-        let time_str = args.get("time").unwrap().to_string();
+        return Ok(());
+    }
 
-        natural_parser(&time_str, &timezone.to_string()).await
-    };
+    ctx.defer().await?;
+
+    let user_data = ctx.author_data().await.unwrap();
+    let timezone = ctx.timezone().await;
+
+    let time = natural_parser(&time, &timezone.to_string()).await;
 
     match time {
         Some(time) => {
             let content = {
-                let content = args.get("content").unwrap().to_string();
-                let tts = args.get("tts").map_or(false, |arg| arg.as_bool().unwrap_or(false));
+                let tts = tts.unwrap_or(false);
 
                 Content { content, tts, attachment: None, attachment_name: None }
             };
 
             let scopes = {
-                let list = args
-                    .get("channels")
-                    .map(|arg| parse_mention_list(&arg.to_string()))
-                    .unwrap_or_default();
+                let list =
+                    channels.map(|arg| parse_mention_list(&arg.to_string())).unwrap_or_default();
 
                 if list.is_empty() {
-                    if invoke.guild_id().is_some() {
-                        vec![ReminderScope::Channel(invoke.channel_id().0)]
+                    if ctx.guild_id().is_some() {
+                        vec![ReminderScope::Channel(ctx.channel_id().0)]
                     } else {
-                        vec![ReminderScope::User(invoke.author_id().0)]
+                        vec![ReminderScope::User(ctx.author().id.0)]
                     }
                 } else {
                     list
                 }
             };
 
-            let (interval, expires) = if let Some(repeat) = args.get("interval") {
-                if check_subscription(&ctx, invoke.author_id()).await
-                    || (invoke.guild_id().is_some()
-                        && check_guild_subscription(&ctx, invoke.guild_id().unwrap()).await)
+            let (processed_interval, processed_expires) = if let Some(repeat) = &interval {
+                if check_subscription(&ctx.discord(), ctx.author().id).await
+                    || (ctx.guild_id().is_some()
+                        && check_guild_subscription(&ctx.discord(), ctx.guild_id().unwrap()).await)
                 {
                     (
-                        parse_duration(&repeat.to_string())
+                        parse_duration(repeat)
                             .or_else(|_| parse_duration(&format!("1 {}", repeat.to_string())))
                             .ok(),
                         {
-                            if let Some(arg) = args.get("expires") {
-                                natural_parser(&arg.to_string(), &timezone.to_string()).await
+                            if let Some(arg) = &expires {
+                                natural_parser(arg, &timezone.to_string()).await
                             } else {
                                 None
                             }
                         },
                     )
                 } else {
-                    let _ = invoke
-                        .respond(&ctx, CreateGenericResponse::new()
-                            .content("`repeat` is only available to Patreon subscribers or self-hosted users")
-                        ).await;
+                    ctx.say(
+                        "`repeat` is only available to Patreon subscribers or self-hosted users",
+                    )
+                    .await?;
 
-                    return;
+                    return Ok(());
                 }
             } else {
                 (None, None)
             };
 
-            if interval.is_none() && args.get("interval").is_some() {
-                let _ = invoke
-                    .respond(
-                        &ctx,
-                        CreateGenericResponse::new().content(
-                            "Repeat interval could not be processed. Try and format the repetition similar to `1 hour` or `4 days`",
-                        ),
-                    )
-                    .await;
-            } else if expires.is_none() && args.get("expires").is_some() {
-                let _ = invoke
-                    .respond(
-                        &ctx,
-                        CreateGenericResponse::new().content(
-                            "Expiry time failed to process. Please make it as clear as possible",
-                        ),
-                    )
-                    .await;
+            if processed_interval.is_none() && interval.is_some() {
+                ctx.say(
+                    "Repeat interval could not be processed. Try similar to `1 hour` or `4 days`",
+                )
+                .await?;
+            } else if processed_expires.is_none() && expires.is_some() {
+                ctx.say("Expiry time failed to process. Please make it as clear as possible")
+                    .await?;
             } else {
-                let mut builder = MultiReminderBuilder::new(ctx, invoke.guild_id())
+                let mut builder = MultiReminderBuilder::new(&ctx, ctx.guild_id())
                     .author(user_data)
                     .content(content)
                     .time(time)
                     .timezone(timezone)
-                    .expires(expires)
-                    .interval(interval);
+                    .expires(processed_expires)
+                    .interval(processed_interval);
 
                 builder.set_scopes(scopes);
 
@@ -777,23 +655,21 @@ async fn remind(ctx: &Context, invoke: &mut CommandInvoke, args: CommandOptions)
 
                 let embed = create_response(successes, errors, time);
 
-                let _ = invoke
-                    .respond(
-                        &ctx,
-                        CreateGenericResponse::new().embed(|c| {
-                            *c = embed;
-                            c
-                        }),
-                    )
-                    .await;
+                ctx.send(|m| {
+                    m.embed(|c| {
+                        *c = embed;
+                        c
+                    })
+                })
+                .await?;
             }
         }
         None => {
-            let _ = invoke
-                .respond(&ctx, CreateGenericResponse::new().content("Time could not be processed"))
-                .await;
+            ctx.say("Time could not be processed").await?;
         }
     }
+
+    Ok(())
 }
 
 fn create_response(

src/commands/todo_cmds.rs

@@ -1,5 +1,4 @@
-use regex_command_attr::command;
-use serenity::client::Context;
+use poise::CreateReply;
 
 use crate::{
     component_models::{
@@ -7,134 +6,218 @@ use crate::{
         ComponentDataModel, TodoSelector,
     },
     consts::{EMBED_DESCRIPTION_MAX_LENGTH, SELECT_MAX_ENTRIES, THEME_COLOR},
-    framework::{CommandInvoke, CommandOptions, CreateGenericResponse},
-    hooks::CHECK_GUILD_PERMISSIONS_HOOK,
-    SQLPool,
+    Context, Error,
 };
 
-#[command]
-#[description("Manage todo lists")]
-#[subcommandgroup("server")]
-#[description("Manage the server todo list")]
-#[subcommand("add")]
-#[description("Add an item to the server todo list")]
-#[arg(
-    name = "task",
-    description = "The task to add to the todo list",
-    kind = "String",
-    required = true
+/// Manage todo lists
+#[poise::command(
+    slash_command,
+    rename = "todo",
+    identifying_name = "todo_base",
+    default_member_permissions = "MANAGE_GUILD"
 )]
-#[subcommand("view")]
-#[description("View and remove from the server todo list")]
-#[subcommandgroup("channel")]
-#[description("Manage the channel todo list")]
-#[subcommand("add")]
-#[description("Add to the channel todo list")]
-#[arg(
-    name = "task",
-    description = "The task to add to the todo list",
-    kind = "String",
-    required = true
+pub async fn todo_base(_ctx: Context<'_>) -> Result<(), Error> {
+    Ok(())
+}
+
+/// Manage the server todo list
+#[poise::command(
+    slash_command,
+    rename = "server",
+    guild_only = true,
+    identifying_name = "todo_guild_base",
+    default_member_permissions = "MANAGE_GUILD"
 )]
-#[subcommand("view")]
-#[description("View and remove from the channel todo list")]
-#[subcommandgroup("user")]
-#[description("Manage your personal todo list")]
-#[subcommand("add")]
-#[description("Add to your personal todo list")]
-#[arg(
-    name = "task",
-    description = "The task to add to the todo list",
-    kind = "String",
-    required = true
+pub async fn todo_guild_base(_ctx: Context<'_>) -> Result<(), Error> {
+    Ok(())
+}
+
+/// Add an item to the server todo list
+#[poise::command(
+    slash_command,
+    rename = "add",
+    guild_only = true,
+    identifying_name = "todo_guild_add",
+    default_member_permissions = "MANAGE_GUILD"
 )]
-#[subcommand("view")]
-#[description("View and remove from your personal todo list")]
-#[hook(CHECK_GUILD_PERMISSIONS_HOOK)]
-async fn todo(ctx: &Context, invoke: &mut CommandInvoke, args: CommandOptions) {
-    if invoke.guild_id().is_none() && args.subcommand_group != Some("user".to_string()) {
-        let _ = invoke
-            .respond(
-                &ctx,
-                CreateGenericResponse::new().content("Please use `/todo user` in direct messages"),
-            )
-            .await;
-    } else {
-        let pool = ctx.data.read().await.get::<SQLPool>().cloned().unwrap();
+pub async fn todo_guild_add(
+    ctx: Context<'_>,
+    #[description = "The task to add to the todo list"] task: String,
+) -> Result<(), Error> {
+    sqlx::query!(
+        "INSERT INTO todos (guild_id, value)
+VALUES ((SELECT id FROM guilds WHERE guild = ?), ?)",
+        ctx.guild_id().unwrap().0,
+        task
+    )
+    .execute(&ctx.data().database)
+    .await
+    .unwrap();
 
-        let keys = match args.subcommand_group.as_ref().unwrap().as_str() {
-            "server" => (None, None, invoke.guild_id().map(|g| g.0)),
-            "channel" => (None, Some(invoke.channel_id().0), invoke.guild_id().map(|g| g.0)),
-            _ => (Some(invoke.author_id().0), None, None),
-        };
+    ctx.say("Item added to todo list").await?;
 
-        match args.get("task") {
-            Some(task) => {
-                let task = task.to_string();
+    Ok(())
+}
 
-                sqlx::query!(
-                    "INSERT INTO todos (user_id, channel_id, guild_id, value) VALUES ((SELECT id FROM users WHERE user = ?), (SELECT id FROM channels WHERE channel = ?), (SELECT id FROM guilds WHERE guild = ?), ?)",
-                    keys.0,
-                    keys.1,
-                    keys.2,
-                    task
-                )
-                .execute(&pool)
-                .await
-                .unwrap();
-
-                let _ = invoke
-                    .respond(&ctx, CreateGenericResponse::new().content("Item added to todo list"))
-                    .await;
-            }
-            None => {
-                let values = if let Some(uid) = keys.0 {
-                    sqlx::query!(
-                        "SELECT todos.id, value FROM todos
-INNER JOIN users ON todos.user_id = users.id
-WHERE users.user = ?",
-                        uid,
-                    )
-                    .fetch_all(&pool)
-                    .await
-                    .unwrap()
-                    .iter()
-                    .map(|row| (row.id as usize, row.value.clone()))
-                    .collect::<Vec<(usize, String)>>()
-                } else if let Some(cid) = keys.1 {
-                    sqlx::query!(
-                        "SELECT todos.id, value FROM todos
-INNER JOIN channels ON todos.channel_id = channels.id
-WHERE channels.channel = ?",
-                        cid,
-                    )
-                    .fetch_all(&pool)
-                    .await
-                    .unwrap()
-                    .iter()
-                    .map(|row| (row.id as usize, row.value.clone()))
-                    .collect::<Vec<(usize, String)>>()
-                } else {
-                    sqlx::query!(
-                        "SELECT todos.id, value FROM todos
+/// View and remove from the server todo list
+#[poise::command(
+    slash_command,
+    rename = "view",
+    guild_only = true,
+    identifying_name = "todo_guild_view",
+    default_member_permissions = "MANAGE_GUILD"
+)]
+pub async fn todo_guild_view(ctx: Context<'_>) -> Result<(), Error> {
+    let values = sqlx::query!(
+        "SELECT todos.id, value FROM todos
 INNER JOIN guilds ON todos.guild_id = guilds.id
 WHERE guilds.guild = ?",
-                        keys.2,
-                    )
-                    .fetch_all(&pool)
-                    .await
-                    .unwrap()
-                    .iter()
-                    .map(|row| (row.id as usize, row.value.clone()))
-                    .collect::<Vec<(usize, String)>>()
-                };
+        ctx.guild_id().unwrap().0,
+    )
+    .fetch_all(&ctx.data().database)
+    .await
+    .unwrap()
+    .iter()
+    .map(|row| (row.id as usize, row.value.clone()))
+    .collect::<Vec<(usize, String)>>();
 
-                let resp = show_todo_page(&values, 0, keys.0, keys.1, keys.2);
+    let resp = show_todo_page(&values, 0, None, None, ctx.guild_id().map(|g| g.0));
 
-                invoke.respond(&ctx, resp).await.unwrap();
-            }
-        }
-    }
+    ctx.send(|r| {
+        *r = resp;
+        r
+    })
+    .await?;
+
+    Ok(())
+}
+
+/// Manage the channel todo list
+#[poise::command(
+    slash_command,
+    rename = "channel",
+    guild_only = true,
+    identifying_name = "todo_channel_base",
+    default_member_permissions = "MANAGE_GUILD"
+)]
+pub async fn todo_channel_base(_ctx: Context<'_>) -> Result<(), Error> {
+    Ok(())
+}
+
+/// Add an item to the channel todo list
+#[poise::command(
+    slash_command,
+    rename = "add",
+    guild_only = true,
+    identifying_name = "todo_channel_add",
+    default_member_permissions = "MANAGE_GUILD"
+)]
+pub async fn todo_channel_add(
+    ctx: Context<'_>,
+    #[description = "The task to add to the todo list"] task: String,
+) -> Result<(), Error> {
+    sqlx::query!(
+        "INSERT INTO todos (guild_id, channel_id, value)
+VALUES ((SELECT id FROM guilds WHERE guild = ?), (SELECT id FROM channels WHERE channel = ?), ?)",
+        ctx.guild_id().unwrap().0,
+        ctx.channel_id().0,
+        task
+    )
+    .execute(&ctx.data().database)
+    .await
+    .unwrap();
+
+    ctx.say("Item added to todo list").await?;
+
+    Ok(())
+}
+
+/// View and remove from the channel todo list
+#[poise::command(
+    slash_command,
+    rename = "view",
+    guild_only = true,
+    identifying_name = "todo_channel_view",
+    default_member_permissions = "MANAGE_GUILD"
+)]
+pub async fn todo_channel_view(ctx: Context<'_>) -> Result<(), Error> {
+    let values = sqlx::query!(
+        "SELECT todos.id, value FROM todos
+INNER JOIN channels ON todos.channel_id = channels.id
+WHERE channels.channel = ?",
+        ctx.channel_id().0,
+    )
+    .fetch_all(&ctx.data().database)
+    .await
+    .unwrap()
+    .iter()
+    .map(|row| (row.id as usize, row.value.clone()))
+    .collect::<Vec<(usize, String)>>();
+
+    let resp =
+        show_todo_page(&values, 0, None, Some(ctx.channel_id().0), ctx.guild_id().map(|g| g.0));
+
+    ctx.send(|r| {
+        *r = resp;
+        r
+    })
+    .await?;
+
+    Ok(())
+}
+
+/// Manage your personal todo list
+#[poise::command(slash_command, rename = "user", identifying_name = "todo_user_base")]
+pub async fn todo_user_base(_ctx: Context<'_>) -> Result<(), Error> {
+    Ok(())
+}
+
+/// Add an item to your personal todo list
+#[poise::command(slash_command, rename = "add", identifying_name = "todo_user_add")]
+pub async fn todo_user_add(
+    ctx: Context<'_>,
+    #[description = "The task to add to the todo list"] task: String,
+) -> Result<(), Error> {
+    sqlx::query!(
+        "INSERT INTO todos (user_id, value)
+VALUES ((SELECT id FROM users WHERE user = ?), ?)",
+        ctx.author().id.0,
+        task
+    )
+    .execute(&ctx.data().database)
+    .await
+    .unwrap();
+
+    ctx.say("Item added to todo list").await?;
+
+    Ok(())
+}
+
+/// View and remove from your personal todo list
+#[poise::command(slash_command, rename = "view", identifying_name = "todo_user_view")]
+pub async fn todo_user_view(ctx: Context<'_>) -> Result<(), Error> {
+    let values = sqlx::query!(
+        "SELECT todos.id, value FROM todos
+INNER JOIN users ON todos.user_id = users.id
+WHERE users.user = ?",
+        ctx.author().id.0,
+    )
+    .fetch_all(&ctx.data().database)
+    .await
+    .unwrap()
+    .iter()
+    .map(|row| (row.id as usize, row.value.clone()))
+    .collect::<Vec<(usize, String)>>();
+
+    let resp = show_todo_page(&values, 0, Some(ctx.author().id.0), None, None);
+
+    ctx.send(|r| {
+        *r = resp;
+        r
+    })
+    .await?;
+
+    Ok(())
 }
 
 pub fn max_todo_page(todo_values: &[(usize, String)]) -> usize {
@@ -164,7 +247,7 @@ pub fn show_todo_page(
     user_id: Option<u64>,
     channel_id: Option<u64>,
     guild_id: Option<u64>,
-) -> CreateGenericResponse {
+) -> CreateReply {
     let pager = TodoPager::new(page, user_id, channel_id, guild_id);
 
     let pages = max_todo_page(todo_values);
@@ -219,17 +302,23 @@ pub fn show_todo_page(
     };
 
     if todo_ids.is_empty() {
-        CreateGenericResponse::new().embed(|e| {
+        let mut reply = CreateReply::default();
+
+        reply.embed(|e| {
             e.title(format!("{} Todo List", title))
                 .description("Todo List Empty!")
                 .footer(|f| f.text(format!("Page {} of {}", page + 1, pages)))
                 .color(*THEME_COLOR)
-        })
+        });
+
+        reply
     } else {
         let todo_selector =
             ComponentDataModel::TodoSelector(TodoSelector { page, user_id, channel_id, guild_id });
 
-        CreateGenericResponse::new()
+        let mut reply = CreateReply::default();
+
+        reply
             .embed(|e| {
                 e.title(format!("{} Todo List", title))
                     .description(display)
@@ -255,6 +344,8 @@ pub fn show_todo_page(
                         })
                     })
                 })
-            })
+            });
+
+        reply
     }
 }

src/component_models/mod.rs

@@ -3,9 +3,7 @@ pub(crate) mod pager;
 use std::io::Cursor;
 
 use chrono_tz::Tz;
-use rmp_serde::Serializer;
-use serde::{Deserialize, Serialize};
-use serenity::{
+use poise::serenity::{
     builder::CreateEmbed,
     client::Context,
     model::{
@@ -14,6 +12,8 @@ use serenity::{
         prelude::InteractionApplicationCommandCallbackDataFlags,
     },
 };
+use rmp_serde::Serializer;
+use serde::{Deserialize, Serialize};
 
 use crate::{
     commands::{
@@ -23,9 +23,9 @@ use crate::{
     },
     component_models::pager::{DelPager, LookPager, MacroPager, Pager, TodoPager},
     consts::{EMBED_DESCRIPTION_MAX_LENGTH, THEME_COLOR},
-    framework::CommandInvoke,
-    models::{command_macro::CommandMacro, reminder::Reminder},
-    SQLPool,
+    models::reminder::Reminder,
+    utils::send_as_initial_response,
+    Data,
 };
 
 #[derive(Deserialize, Serialize)]
@@ -55,7 +55,7 @@ impl ComponentDataModel {
         rmp_serde::from_read(cur).unwrap()
     }
 
-    pub async fn act(&self, ctx: &Context, component: MessageComponentInteraction) {
+    pub async fn act(&self, ctx: &Context, data: &Data, component: &MessageComponentInteraction) {
         match self {
             ComponentDataModel::LookPager(pager) => {
                 let flags = pager.flags;
@@ -72,7 +72,7 @@ impl ComponentDataModel {
                     component.channel_id
                 };
 
-                let reminders = Reminder::from_channel(ctx, channel_id, &flags).await;
+                let reminders = Reminder::from_channel(&data.database, channel_id, &flags).await;
 
                 let pages = reminders
                     .iter()
@@ -122,7 +122,7 @@ impl ComponentDataModel {
                     .create_interaction_response(&ctx, |r| {
                         r.kind(InteractionResponseType::UpdateMessage).interaction_response_data(
                             |response| {
-                                response.embeds(vec![embed]).components(|comp| {
+                                response.set_embeds(vec![embed]).components(|comp| {
                                     pager.create_button_row(pages, comp);
 
                                     comp
@@ -133,45 +133,68 @@ impl ComponentDataModel {
                     .await;
             }
             ComponentDataModel::DelPager(pager) => {
-                let reminders =
-                    Reminder::from_guild(ctx, component.guild_id, component.user.id).await;
+                let reminders = Reminder::from_guild(
+                    &ctx,
+                    &data.database,
+                    component.guild_id,
+                    component.user.id,
+                )
+                .await;
 
                 let max_pages = max_delete_page(&reminders, &pager.timezone);
 
                 let resp = show_delete_page(&reminders, pager.next_page(max_pages), pager.timezone);
 
-                let mut invoke = CommandInvoke::component(component);
-                let _ = invoke.respond(&ctx, resp).await;
+                let _ = component
+                    .create_interaction_response(&ctx, |f| {
+                        f.kind(InteractionResponseType::UpdateMessage).interaction_response_data(
+                            |d| {
+                                send_as_initial_response(resp, d);
+                                d
+                            },
+                        )
+                    })
+                    .await;
             }
             ComponentDataModel::DelSelector(selector) => {
-                let pool = ctx.data.read().await.get::<SQLPool>().cloned().unwrap();
                 let selected_id = component.data.values.join(",");
 
                 sqlx::query!("DELETE FROM reminders WHERE FIND_IN_SET(id, ?)", selected_id)
-                    .execute(&pool)
+                    .execute(&data.database)
                     .await
                     .unwrap();
 
-                let reminders =
-                    Reminder::from_guild(ctx, component.guild_id, component.user.id).await;
+                let reminders = Reminder::from_guild(
+                    &ctx,
+                    &data.database,
+                    component.guild_id,
+                    component.user.id,
+                )
+                .await;
 
                 let resp = show_delete_page(&reminders, selector.page, selector.timezone);
 
-                let mut invoke = CommandInvoke::component(component);
-                let _ = invoke.respond(&ctx, resp).await;
+                let _ = component
+                    .create_interaction_response(&ctx, |f| {
+                        f.kind(InteractionResponseType::UpdateMessage).interaction_response_data(
+                            |d| {
+                                send_as_initial_response(resp, d);
+                                d
+                            },
+                        )
+                    })
+                    .await;
             }
             ComponentDataModel::TodoPager(pager) => {
                 if Some(component.user.id.0) == pager.user_id || pager.user_id.is_none() {
-                    let pool = ctx.data.read().await.get::<SQLPool>().cloned().unwrap();
-
                     let values = if let Some(uid) = pager.user_id {
                         sqlx::query!(
                             "SELECT todos.id, value FROM todos
-    INNER JOIN users ON todos.user_id = users.id
-    WHERE users.user = ?",
+INNER JOIN users ON todos.user_id = users.id
+WHERE users.user = ?",
                             uid,
                         )
-                        .fetch_all(&pool)
+                        .fetch_all(&data.database)
                         .await
                         .unwrap()
                         .iter()
@@ -180,11 +203,11 @@ impl ComponentDataModel {
                     } else if let Some(cid) = pager.channel_id {
                         sqlx::query!(
                             "SELECT todos.id, value FROM todos
-    INNER JOIN channels ON todos.channel_id = channels.id
-    WHERE channels.channel = ?",
+INNER JOIN channels ON todos.channel_id = channels.id
+WHERE channels.channel = ?",
                             cid,
                         )
-                        .fetch_all(&pool)
+                        .fetch_all(&data.database)
                         .await
                         .unwrap()
                         .iter()
@@ -193,11 +216,11 @@ impl ComponentDataModel {
                     } else {
                         sqlx::query!(
                             "SELECT todos.id, value FROM todos
-    INNER JOIN guilds ON todos.guild_id = guilds.id
-    WHERE guilds.guild = ?",
+INNER JOIN guilds ON todos.guild_id = guilds.id
+WHERE guilds.guild = ?",
                             pager.guild_id,
                         )
-                        .fetch_all(&pool)
+                        .fetch_all(&data.database)
                         .await
                         .unwrap()
                         .iter()
@@ -215,8 +238,15 @@ impl ComponentDataModel {
                         pager.guild_id,
                     );
 
-                    let mut invoke = CommandInvoke::component(component);
-                    let _ = invoke.respond(&ctx, resp).await;
+                    let _ = component
+                        .create_interaction_response(&ctx, |f| {
+                            f.kind(InteractionResponseType::UpdateMessage)
+                                .interaction_response_data(|d| {
+                                    send_as_initial_response(resp, d);
+                                    d
+                                })
+                        })
+                        .await;
                 } else {
                     let _ = component
                         .create_interaction_response(&ctx, |r| {
@@ -233,11 +263,10 @@ impl ComponentDataModel {
             }
             ComponentDataModel::TodoSelector(selector) => {
                 if Some(component.user.id.0) == selector.user_id || selector.user_id.is_none() {
-                    let pool = ctx.data.read().await.get::<SQLPool>().cloned().unwrap();
                     let selected_id = component.data.values.join(",");
 
                     sqlx::query!("DELETE FROM todos WHERE FIND_IN_SET(id, ?)", selected_id)
-                        .execute(&pool)
+                        .execute(&data.database)
                         .await
                         .unwrap();
 
@@ -248,7 +277,7 @@ impl ComponentDataModel {
                     selector.channel_id,
                     selector.guild_id,
                 )
-                .fetch_all(&pool)
+                .fetch_all(&data.database)
                 .await
                 .unwrap()
                 .iter()
@@ -263,8 +292,15 @@ impl ComponentDataModel {
                         selector.guild_id,
                     );
 
-                    let mut invoke = CommandInvoke::component(component);
-                    let _ = invoke.respond(&ctx, resp).await;
+                    let _ = component
+                        .create_interaction_response(&ctx, |f| {
+                            f.kind(InteractionResponseType::UpdateMessage)
+                                .interaction_response_data(|d| {
+                                    send_as_initial_response(resp, d);
+                                    d
+                                })
+                        })
+                        .await;
                 } else {
                     let _ = component
                         .create_interaction_response(&ctx, |r| {
@@ -280,15 +316,23 @@ impl ComponentDataModel {
                 }
             }
             ComponentDataModel::MacroPager(pager) => {
-                let mut invoke = CommandInvoke::component(component);
-
-                let macros = CommandMacro::from_guild(ctx, invoke.guild_id().unwrap()).await;
+                let macros = data.command_macros(component.guild_id.unwrap()).await.unwrap();
 
                 let max_page = max_macro_page(&macros);
                 let page = pager.next_page(max_page);
 
                 let resp = show_macro_page(&macros, page);
-                let _ = invoke.respond(&ctx, resp).await;
+
+                let _ = component
+                    .create_interaction_response(&ctx, |f| {
+                        f.kind(InteractionResponseType::UpdateMessage).interaction_response_data(
+                            |d| {
+                                send_as_initial_response(resp, d);
+                                d
+                            },
+                        )
+                    })
+                    .await;
             }
         }
     }

src/component_models/pager.rs

@@ -1,8 +1,10 @@
 // todo split pager out into a single struct
 use chrono_tz::Tz;
+use poise::serenity::{
+    builder::CreateComponents, model::interactions::message_component::ButtonStyle,
+};
 use serde::{Deserialize, Serialize};
 use serde_repr::*;
-use serenity::{builder::CreateComponents, model::interactions::message_component::ButtonStyle};
 
 use crate::{component_models::ComponentDataModel, models::reminder::look_flags::LookFlags};
 

src/consts.rs

@@ -1,4 +1,6 @@
 pub const DAY: u64 = 86_400;
+pub const HOUR: u64 = 3_600;
+pub const MINUTE: u64 = 60;
 
 pub const EMBED_DESCRIPTION_MAX_LENGTH: usize = 4000;
 pub const SELECT_MAX_ENTRIES: usize = 25;
@@ -6,11 +8,12 @@ pub const SELECT_MAX_ENTRIES: usize = 25;
 pub const CHARACTERS: &str = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789_";
 
 const THEME_COLOR_FALLBACK: u32 = 0x8fb677;
+pub const MACRO_MAX_COMMANDS: usize = 5;
 
 use std::{collections::HashSet, env, iter::FromIterator};
 
+use poise::serenity::model::prelude::AttachmentType;
 use regex::Regex;
-use serenity::model::prelude::AttachmentType;
 
 lazy_static! {
     pub static ref DEFAULT_AVATAR: AttachmentType<'static> = (

src/event_handlers.rs

@@ -1,161 +1,126 @@
 use std::{collections::HashMap, env, sync::atomic::Ordering};
 
-use log::{info, warn};
-use serenity::{
-    async_trait,
-    client::{Context, EventHandler},
-    model::{
-        channel::GuildChannel,
-        gateway::{Activity, Ready},
-        guild::{Guild, UnavailableGuild},
-        id::GuildId,
-        interactions::Interaction,
-    },
-    utils::shard_id,
+use log::{error, info, warn};
+use poise::{
+    serenity::{model::interactions::Interaction, utils::shard_id},
+    serenity_prelude as serenity,
 };
 
-use crate::{ComponentDataModel, Handler, RegexFramework, ReqwestClient, SQLPool};
+use crate::{component_models::ComponentDataModel, Data, Error};
 
-#[async_trait]
-impl EventHandler for Handler {
-    async fn cache_ready(&self, ctx_base: Context, _guilds: Vec<GuildId>) {
-        info!("Cache Ready!");
-        info!("Preparing to send reminders");
+pub async fn listener(
+    ctx: &serenity::Context,
+    event: &poise::Event<'_>,
+    data: &Data,
+) -> Result<(), Error> {
+    match event {
+        poise::Event::CacheReady { .. } => {
+            info!("Cache Ready! Preparing extra processes");
 
-        if !self.is_loop_running.load(Ordering::Relaxed) {
-            let ctx1 = ctx_base.clone();
-            let ctx2 = ctx_base.clone();
+            if !data.is_loop_running.load(Ordering::Relaxed) {
+                let kill_tx = data.broadcast.clone();
+                let kill_recv = data.broadcast.subscribe();
 
-            let pool1 = ctx1.data.read().await.get::<SQLPool>().cloned().unwrap();
-            let pool2 = ctx2.data.read().await.get::<SQLPool>().cloned().unwrap();
+                let ctx1 = ctx.clone();
+                let ctx2 = ctx.clone();
 
-            let run_settings = env::var("DONTRUN").unwrap_or_else(|_| "".to_string());
+                let pool1 = data.database.clone();
+                let pool2 = data.database.clone();
 
-            if !run_settings.contains("postman") {
-                tokio::spawn(async move {
-                    postman::initialize(ctx1, &pool1).await;
-                });
-            } else {
-                warn!("Not running postman")
+                let run_settings = env::var("DONTRUN").unwrap_or_else(|_| "".to_string());
+
+                if !run_settings.contains("postman") {
+                    tokio::spawn(async move {
+                        match postman::initialize(kill_recv, ctx1, &pool1).await {
+                            Ok(_) => {}
+                            Err(e) => {
+                                error!("postman exiting: {}", e);
+                            }
+                        };
+                    });
+                } else {
+                    warn!("Not running postman")
+                }
+
+                if !run_settings.contains("web") {
+                    tokio::spawn(async move {
+                        reminder_web::initialize(kill_tx, ctx2, pool2).await.unwrap();
+                    });
+                } else {
+                    warn!("Not running web")
+                }
+
+                data.is_loop_running.swap(true, Ordering::Relaxed);
             }
-
-            if !run_settings.contains("web") {
-                tokio::spawn(async move {
-                    reminder_web::initialize(ctx2, pool2).await.unwrap();
-                });
-            } else {
-                warn!("Not running web")
-            }
-
-            self.is_loop_running.swap(true, Ordering::Relaxed);
         }
-    }
+        poise::Event::ChannelDelete { channel } => {
+            sqlx::query!("DELETE FROM channels WHERE channel = ?", channel.id.as_u64())
+                .execute(&data.database)
+                .await
+                .unwrap();
+        }
+        poise::Event::GuildCreate { guild, is_new } => {
+            if *is_new {
+                let guild_id = guild.id.as_u64().to_owned();
 
-    async fn channel_delete(&self, ctx: Context, channel: &GuildChannel) {
-        let pool = ctx
-            .data
-            .read()
-            .await
-            .get::<SQLPool>()
-            .cloned()
-            .expect("Could not get SQLPool from data");
-
-        sqlx::query!(
-            "
-DELETE FROM channels WHERE channel = ?
-            ",
-            channel.id.as_u64()
-        )
-        .execute(&pool)
-        .await
-        .unwrap();
-    }
-
-    async fn guild_create(&self, ctx: Context, guild: Guild, is_new: bool) {
-        if is_new {
-            let guild_id = guild.id.as_u64().to_owned();
-
-            {
-                let pool = ctx.data.read().await.get::<SQLPool>().cloned().unwrap();
-
-                let _ = sqlx::query!("INSERT INTO guilds (guild) VALUES (?)", guild_id)
-                    .execute(&pool)
-                    .await;
-            }
-
-            if let Ok(token) = env::var("DISCORDBOTS_TOKEN") {
-                let shard_count = ctx.cache.shard_count();
-                let current_shard_id = shard_id(guild_id, shard_count);
-
-                let guild_count = ctx
-                    .cache
-                    .guilds()
-                    .iter()
-                    .filter(|g| shard_id(g.as_u64().to_owned(), shard_count) == current_shard_id)
-                    .count() as u64;
-
-                let mut hm = HashMap::new();
-                hm.insert("server_count", guild_count);
-                hm.insert("shard_id", current_shard_id);
-                hm.insert("shard_count", shard_count);
-
-                let client = ctx
-                    .data
-                    .read()
+                sqlx::query!("INSERT INTO guilds (guild) VALUES (?)", guild_id)
+                    .execute(&data.database)
                     .await
-                    .get::<ReqwestClient>()
-                    .cloned()
-                    .expect("Could not get ReqwestClient from data");
+                    .unwrap();
 
-                let response = client
-                    .post(
-                        format!(
-                            "https://top.gg/api/bots/{}/stats",
-                            ctx.cache.current_user_id().as_u64()
+                if let Ok(token) = env::var("DISCORDBOTS_TOKEN") {
+                    let shard_count = ctx.cache.shard_count();
+                    let current_shard_id = shard_id(guild_id, shard_count);
+
+                    let guild_count = ctx
+                        .cache
+                        .guilds()
+                        .iter()
+                        .filter(|g| {
+                            shard_id(g.as_u64().to_owned(), shard_count) == current_shard_id
+                        })
+                        .count() as u64;
+
+                    let mut hm = HashMap::new();
+                    hm.insert("server_count", guild_count);
+                    hm.insert("shard_id", current_shard_id);
+                    hm.insert("shard_count", shard_count);
+
+                    let response = data
+                        .http
+                        .post(
+                            format!(
+                                "https://top.gg/api/bots/{}/stats",
+                                ctx.cache.current_user_id().as_u64()
+                            )
+                            .as_str(),
                         )
-                        .as_str(),
-                    )
-                    .header("Authorization", token)
-                    .json(&hm)
-                    .send()
-                    .await;
+                        .header("Authorization", token)
+                        .json(&hm)
+                        .send()
+                        .await;
 
-                if let Err(res) = response {
-                    println!("DiscordBots Response: {:?}", res);
+                    if let Err(res) = response {
+                        println!("DiscordBots Response: {:?}", res);
+                    }
                 }
             }
         }
-    }
-
-    async fn guild_delete(&self, ctx: Context, incomplete: UnavailableGuild, _full: Option<Guild>) {
-        let pool = ctx.data.read().await.get::<SQLPool>().cloned().unwrap();
-        let _ = sqlx::query!("DELETE FROM guilds WHERE guild = ?", incomplete.id.0)
-            .execute(&pool)
-            .await;
-    }
-
-    async fn ready(&self, ctx: Context, _: Ready) {
-        ctx.set_activity(Activity::watching("for /remind")).await;
-    }
-
-    async fn interaction_create(&self, ctx: Context, interaction: Interaction) {
-        match interaction {
-            Interaction::ApplicationCommand(application_command) => {
-                let framework = ctx
-                    .data
-                    .read()
-                    .await
-                    .get::<RegexFramework>()
-                    .cloned()
-                    .expect("RegexFramework not found in context");
-
-                framework.execute(ctx, application_command).await;
-            }
+        poise::Event::GuildDelete { incomplete, .. } => {
+            let _ = sqlx::query!("DELETE FROM guilds WHERE guild = ?", incomplete.id.0)
+                .execute(&data.database)
+                .await;
+        }
+        poise::Event::InteractionCreate { interaction } => match interaction {
             Interaction::MessageComponent(component) => {
                 let component_model = ComponentDataModel::from_custom_id(&component.data.custom_id);
-                component_model.act(&ctx, component).await;
+
+                component_model.act(ctx, data, component).await;
             }
             _ => {}
-        }
+        },
+        _ => {}
     }
+
+    Ok(())
 }

src/framework.rs

@@ -1,692 +0,0 @@
-// todo move framework to its own module, split out permission checks
-
-use std::{
-    collections::{HashMap, HashSet},
-    hash::{Hash, Hasher},
-    sync::Arc,
-};
-
-use log::info;
-use serde::{Deserialize, Serialize};
-use serenity::{
-    builder::{CreateApplicationCommands, CreateComponents, CreateEmbed},
-    cache::Cache,
-    client::Context,
-    futures::prelude::future::BoxFuture,
-    http::Http,
-    model::{
-        guild::Guild,
-        id::{ChannelId, GuildId, RoleId, UserId},
-        interactions::{
-            application_command::{
-                ApplicationCommand, ApplicationCommandInteraction, ApplicationCommandOptionType,
-            },
-            message_component::MessageComponentInteraction,
-            InteractionApplicationCommandCallbackDataFlags, InteractionResponseType,
-        },
-        prelude::application_command::ApplicationCommandInteractionDataOption,
-    },
-    prelude::TypeMapKey,
-    Result as SerenityResult,
-};
-
-use crate::SQLPool;
-
-pub struct CreateGenericResponse {
-    content: String,
-    embed: Option<CreateEmbed>,
-    components: Option<CreateComponents>,
-    flags: InteractionApplicationCommandCallbackDataFlags,
-}
-
-impl CreateGenericResponse {
-    pub fn new() -> Self {
-        Self {
-            content: "".to_string(),
-            embed: None,
-            components: None,
-            flags: InteractionApplicationCommandCallbackDataFlags::empty(),
-        }
-    }
-
-    pub fn ephemeral(mut self) -> Self {
-        self.flags.insert(InteractionApplicationCommandCallbackDataFlags::EPHEMERAL);
-
-        self
-    }
-
-    pub fn content<D: ToString>(mut self, content: D) -> Self {
-        self.content = content.to_string();
-
-        self
-    }
-
-    pub fn embed<F: FnOnce(&mut CreateEmbed) -> &mut CreateEmbed>(mut self, f: F) -> Self {
-        let mut embed = CreateEmbed::default();
-        f(&mut embed);
-
-        self.embed = Some(embed);
-        self
-    }
-
-    pub fn components<F: FnOnce(&mut CreateComponents) -> &mut CreateComponents>(
-        mut self,
-        f: F,
-    ) -> Self {
-        let mut components = CreateComponents::default();
-        f(&mut components);
-
-        self.components = Some(components);
-        self
-    }
-}
-
-#[derive(Clone)]
-enum InvokeModel {
-    Slash(ApplicationCommandInteraction),
-    Component(MessageComponentInteraction),
-}
-
-#[derive(Clone)]
-pub struct CommandInvoke {
-    model: InvokeModel,
-    already_responded: bool,
-    deferred: bool,
-}
-
-impl CommandInvoke {
-    pub fn component(component: MessageComponentInteraction) -> Self {
-        Self { model: InvokeModel::Component(component), already_responded: false, deferred: false }
-    }
-
-    fn slash(interaction: ApplicationCommandInteraction) -> Self {
-        Self { model: InvokeModel::Slash(interaction), already_responded: false, deferred: false }
-    }
-
-    pub async fn defer(&mut self, http: impl AsRef<Http>) {
-        if !self.deferred {
-            match &self.model {
-                InvokeModel::Slash(i) => {
-                    i.create_interaction_response(http, |r| {
-                        r.kind(InteractionResponseType::DeferredChannelMessageWithSource)
-                    })
-                    .await
-                    .unwrap();
-
-                    self.deferred = true;
-                }
-                InvokeModel::Component(i) => {
-                    i.create_interaction_response(http, |r| {
-                        r.kind(InteractionResponseType::DeferredChannelMessageWithSource)
-                    })
-                    .await
-                    .unwrap();
-
-                    self.deferred = true;
-                }
-            }
-        }
-    }
-
-    pub fn channel_id(&self) -> ChannelId {
-        match &self.model {
-            InvokeModel::Slash(i) => i.channel_id,
-            InvokeModel::Component(i) => i.channel_id,
-        }
-    }
-
-    pub fn guild_id(&self) -> Option<GuildId> {
-        match &self.model {
-            InvokeModel::Slash(i) => i.guild_id,
-            InvokeModel::Component(i) => i.guild_id,
-        }
-    }
-
-    pub fn guild(&self, cache: impl AsRef<Cache>) -> Option<Guild> {
-        self.guild_id().map(|id| id.to_guild_cached(cache)).flatten()
-    }
-
-    pub fn author_id(&self) -> UserId {
-        match &self.model {
-            InvokeModel::Slash(i) => i.user.id,
-            InvokeModel::Component(i) => i.user.id,
-        }
-    }
-
-    pub async fn respond(
-        &mut self,
-        http: impl AsRef<Http>,
-        generic_response: CreateGenericResponse,
-    ) -> SerenityResult<()> {
-        match &self.model {
-            InvokeModel::Slash(i) => {
-                if self.already_responded {
-                    i.create_followup_message(http, |d| {
-                        d.allowed_mentions(|m| m.empty_parse());
-                        d.content(generic_response.content);
-
-                        if let Some(embed) = generic_response.embed {
-                            d.add_embed(embed);
-                        }
-
-                        if let Some(components) = generic_response.components {
-                            d.components(|c| {
-                                *c = components;
-                                c
-                            });
-                        }
-
-                        d
-                    })
-                    .await
-                    .map(|_| ())
-                } else if self.deferred {
-                    i.edit_original_interaction_response(http, |d| {
-                        d.allowed_mentions(|m| m.empty_parse());
-                        d.content(generic_response.content);
-
-                        if let Some(embed) = generic_response.embed {
-                            d.add_embed(embed);
-                        }
-
-                        if let Some(components) = generic_response.components {
-                            d.components(|c| {
-                                *c = components;
-                                c
-                            });
-                        }
-
-                        d
-                    })
-                    .await
-                    .map(|_| ())
-                } else {
-                    i.create_interaction_response(http, |r| {
-                        r.kind(InteractionResponseType::ChannelMessageWithSource)
-                            .interaction_response_data(|d| {
-                                d.allowed_mentions(|m| m.empty_parse());
-                                d.content(generic_response.content);
-
-                                if let Some(embed) = generic_response.embed {
-                                    d.add_embed(embed);
-                                }
-
-                                if let Some(components) = generic_response.components {
-                                    d.components(|c| {
-                                        *c = components;
-                                        c
-                                    });
-                                }
-
-                                d
-                            })
-                    })
-                    .await
-                    .map(|_| ())
-                }
-            }
-            InvokeModel::Component(i) => i
-                .create_interaction_response(http, |r| {
-                    r.kind(InteractionResponseType::UpdateMessage).interaction_response_data(|d| {
-                        d.allowed_mentions(|m| m.empty_parse());
-                        d.content(generic_response.content);
-
-                        if let Some(embed) = generic_response.embed {
-                            d.add_embed(embed);
-                        }
-
-                        if let Some(components) = generic_response.components {
-                            d.components(|c| {
-                                *c = components;
-                                c
-                            });
-                        }
-
-                        d
-                    })
-                })
-                .await
-                .map(|_| ()),
-        }?;
-
-        self.already_responded = true;
-
-        Ok(())
-    }
-}
-
-#[derive(Debug)]
-pub struct Arg {
-    pub name: &'static str,
-    pub description: &'static str,
-    pub kind: ApplicationCommandOptionType,
-    pub required: bool,
-    pub options: &'static [&'static Self],
-}
-
-#[derive(Serialize, Deserialize, Clone)]
-pub enum OptionValue {
-    String(String),
-    Integer(i64),
-    Boolean(bool),
-    User(UserId),
-    Channel(ChannelId),
-    Role(RoleId),
-    Mentionable(u64),
-    Number(f64),
-}
-
-impl OptionValue {
-    pub fn as_i64(&self) -> Option<i64> {
-        match self {
-            OptionValue::Integer(i) => Some(*i),
-            _ => None,
-        }
-    }
-
-    pub fn as_bool(&self) -> Option<bool> {
-        match self {
-            OptionValue::Boolean(b) => Some(*b),
-            _ => None,
-        }
-    }
-
-    pub fn as_channel_id(&self) -> Option<ChannelId> {
-        match self {
-            OptionValue::Channel(c) => Some(*c),
-            _ => None,
-        }
-    }
-
-    pub fn to_string(&self) -> String {
-        match self {
-            OptionValue::String(s) => s.to_string(),
-            OptionValue::Integer(i) => i.to_string(),
-            OptionValue::Boolean(b) => b.to_string(),
-            OptionValue::User(u) => u.to_string(),
-            OptionValue::Channel(c) => c.to_string(),
-            OptionValue::Role(r) => r.to_string(),
-            OptionValue::Mentionable(m) => m.to_string(),
-            OptionValue::Number(n) => n.to_string(),
-        }
-    }
-}
-
-#[derive(Serialize, Deserialize, Clone)]
-pub struct CommandOptions {
-    pub command: String,
-    pub subcommand: Option<String>,
-    pub subcommand_group: Option<String>,
-    pub options: HashMap<String, OptionValue>,
-}
-
-impl CommandOptions {
-    pub fn get(&self, key: &str) -> Option<&OptionValue> {
-        self.options.get(key)
-    }
-}
-
-impl CommandOptions {
-    fn new(command: &'static Command) -> Self {
-        Self {
-            command: command.names[0].to_string(),
-            subcommand: None,
-            subcommand_group: None,
-            options: Default::default(),
-        }
-    }
-
-    fn populate(mut self, interaction: &ApplicationCommandInteraction) -> Self {
-        fn match_option(
-            option: ApplicationCommandInteractionDataOption,
-            cmd_opts: &mut CommandOptions,
-        ) {
-            match option.kind {
-                ApplicationCommandOptionType::SubCommand => {
-                    cmd_opts.subcommand = Some(option.name);
-
-                    for opt in option.options {
-                        match_option(opt, cmd_opts);
-                    }
-                }
-                ApplicationCommandOptionType::SubCommandGroup => {
-                    cmd_opts.subcommand_group = Some(option.name);
-
-                    for opt in option.options {
-                        match_option(opt, cmd_opts);
-                    }
-                }
-                ApplicationCommandOptionType::String => {
-                    cmd_opts.options.insert(
-                        option.name,
-                        OptionValue::String(option.value.unwrap().as_str().unwrap().to_string()),
-                    );
-                }
-                ApplicationCommandOptionType::Integer => {
-                    cmd_opts.options.insert(
-                        option.name,
-                        OptionValue::Integer(option.value.map(|m| m.as_i64()).flatten().unwrap()),
-                    );
-                }
-                ApplicationCommandOptionType::Boolean => {
-                    cmd_opts.options.insert(
-                        option.name,
-                        OptionValue::Boolean(option.value.map(|m| m.as_bool()).flatten().unwrap()),
-                    );
-                }
-                ApplicationCommandOptionType::User => {
-                    cmd_opts.options.insert(
-                        option.name,
-                        OptionValue::User(UserId(
-                            option
-                                .value
-                                .map(|m| m.as_str().map(|s| s.parse::<u64>().ok()))
-                                .flatten()
-                                .flatten()
-                                .unwrap(),
-                        )),
-                    );
-                }
-                ApplicationCommandOptionType::Channel => {
-                    cmd_opts.options.insert(
-                        option.name,
-                        OptionValue::Channel(ChannelId(
-                            option
-                                .value
-                                .map(|m| m.as_str().map(|s| s.parse::<u64>().ok()))
-                                .flatten()
-                                .flatten()
-                                .unwrap(),
-                        )),
-                    );
-                }
-                ApplicationCommandOptionType::Role => {
-                    cmd_opts.options.insert(
-                        option.name,
-                        OptionValue::Role(RoleId(
-                            option
-                                .value
-                                .map(|m| m.as_str().map(|s| s.parse::<u64>().ok()))
-                                .flatten()
-                                .flatten()
-                                .unwrap(),
-                        )),
-                    );
-                }
-                ApplicationCommandOptionType::Mentionable => {
-                    cmd_opts.options.insert(
-                        option.name,
-                        OptionValue::Mentionable(
-                            option.value.map(|m| m.as_u64()).flatten().unwrap(),
-                        ),
-                    );
-                }
-                ApplicationCommandOptionType::Number => {
-                    cmd_opts.options.insert(
-                        option.name,
-                        OptionValue::Number(option.value.map(|m| m.as_f64()).flatten().unwrap()),
-                    );
-                }
-                _ => {}
-            }
-        }
-
-        for option in &interaction.data.options {
-            match_option(option.clone(), &mut self)
-        }
-
-        self
-    }
-}
-
-pub enum HookResult {
-    Continue,
-    Halt,
-}
-
-type SlashCommandFn =
-    for<'fut> fn(&'fut Context, &'fut mut CommandInvoke, CommandOptions) -> BoxFuture<'fut, ()>;
-
-type MultiCommandFn = for<'fut> fn(&'fut Context, &'fut mut CommandInvoke) -> BoxFuture<'fut, ()>;
-
-pub type HookFn = for<'fut> fn(
-    &'fut Context,
-    &'fut mut CommandInvoke,
-    &'fut CommandOptions,
-) -> BoxFuture<'fut, HookResult>;
-
-pub enum CommandFnType {
-    Slash(SlashCommandFn),
-    Multi(MultiCommandFn),
-}
-
-pub struct Hook {
-    pub fun: HookFn,
-    pub uuid: u128,
-}
-
-impl PartialEq for Hook {
-    fn eq(&self, other: &Self) -> bool {
-        self.uuid == other.uuid
-    }
-}
-
-pub struct Command {
-    pub fun: CommandFnType,
-
-    pub names: &'static [&'static str],
-
-    pub desc: &'static str,
-    pub examples: &'static [&'static str],
-    pub group: &'static str,
-
-    pub args: &'static [&'static Arg],
-
-    pub can_blacklist: bool,
-    pub supports_dm: bool,
-
-    pub hooks: &'static [&'static Hook],
-}
-
-impl Hash for Command {
-    fn hash<H: Hasher>(&self, state: &mut H) {
-        self.names[0].hash(state)
-    }
-}
-
-impl PartialEq for Command {
-    fn eq(&self, other: &Self) -> bool {
-        self.names[0] == other.names[0]
-    }
-}
-
-impl Eq for Command {}
-
-pub struct RegexFramework {
-    pub commands_map: HashMap<String, &'static Command>,
-    pub commands: HashSet<&'static Command>,
-    ignore_bots: bool,
-    dm_enabled: bool,
-    debug_guild: Option<GuildId>,
-    hooks: Vec<&'static Hook>,
-}
-
-impl TypeMapKey for RegexFramework {
-    type Value = Arc<RegexFramework>;
-}
-
-impl RegexFramework {
-    pub fn new() -> Self {
-        Self {
-            commands_map: HashMap::new(),
-            commands: HashSet::new(),
-            ignore_bots: true,
-            dm_enabled: true,
-            debug_guild: None,
-            hooks: vec![],
-        }
-    }
-
-    pub fn ignore_bots(mut self, ignore_bots: bool) -> Self {
-        self.ignore_bots = ignore_bots;
-
-        self
-    }
-
-    pub fn dm_enabled(mut self, dm_enabled: bool) -> Self {
-        self.dm_enabled = dm_enabled;
-
-        self
-    }
-
-    pub fn add_hook(mut self, fun: &'static Hook) -> Self {
-        self.hooks.push(fun);
-
-        self
-    }
-
-    pub fn add_command(mut self, command: &'static Command) -> Self {
-        self.commands.insert(command);
-
-        for name in command.names {
-            self.commands_map.insert(name.to_string(), command);
-        }
-
-        self
-    }
-
-    pub fn debug_guild(mut self, guild_id: Option<GuildId>) -> Self {
-        self.debug_guild = guild_id;
-
-        self
-    }
-
-    fn _populate_commands<'a>(
-        &self,
-        commands: &'a mut CreateApplicationCommands,
-    ) -> &'a mut CreateApplicationCommands {
-        for command in &self.commands {
-            commands.create_application_command(|c| {
-                c.name(command.names[0]).description(command.desc);
-
-                for arg in command.args {
-                    c.create_option(|o| {
-                        o.name(arg.name)
-                            .description(arg.description)
-                            .kind(arg.kind)
-                            .required(arg.required);
-
-                        for option in arg.options {
-                            o.create_sub_option(|s| {
-                                s.name(option.name)
-                                    .description(option.description)
-                                    .kind(option.kind)
-                                    .required(option.required);
-
-                                for sub_option in option.options {
-                                    s.create_sub_option(|ss| {
-                                        ss.name(sub_option.name)
-                                            .description(sub_option.description)
-                                            .kind(sub_option.kind)
-                                            .required(sub_option.required)
-                                    });
-                                }
-
-                                s
-                            });
-                        }
-
-                        o
-                    });
-                }
-
-                c
-            });
-        }
-
-        commands
-    }
-
-    pub async fn build_slash(&self, http: impl AsRef<Http>) {
-        info!("Building slash commands...");
-
-        match self.debug_guild {
-            None => {
-                ApplicationCommand::set_global_application_commands(&http, |c| {
-                    self._populate_commands(c)
-                })
-                .await
-                .unwrap();
-            }
-            Some(debug_guild) => {
-                debug_guild
-                    .set_application_commands(&http, |c| self._populate_commands(c))
-                    .await
-                    .unwrap();
-            }
-        }
-
-        info!("Slash commands built!");
-    }
-
-    pub async fn execute(&self, ctx: Context, interaction: ApplicationCommandInteraction) {
-        {
-            if let Some(guild_id) = interaction.guild_id {
-                let pool = ctx.data.read().await.get::<SQLPool>().cloned().unwrap();
-                let _ = sqlx::query!("INSERT IGNORE INTO guilds (guild) VALUES (?)", guild_id.0)
-                    .execute(&pool)
-                    .await;
-            }
-        }
-
-        let command = {
-            self.commands_map
-                .get(&interaction.data.name)
-                .expect(&format!("Received invalid command: {}", interaction.data.name))
-        };
-
-        let args = CommandOptions::new(command).populate(&interaction);
-        let mut command_invoke = CommandInvoke::slash(interaction);
-
-        for hook in command.hooks {
-            match (hook.fun)(&ctx, &mut command_invoke, &args).await {
-                HookResult::Continue => {}
-                HookResult::Halt => {
-                    return;
-                }
-            }
-        }
-
-        for hook in &self.hooks {
-            match (hook.fun)(&ctx, &mut command_invoke, &args).await {
-                HookResult::Continue => {}
-                HookResult::Halt => {
-                    return;
-                }
-            }
-        }
-
-        match command.fun {
-            CommandFnType::Slash(t) => t(&ctx, &mut command_invoke, args).await,
-            CommandFnType::Multi(m) => m(&ctx, &mut command_invoke).await,
-        }
-    }
-
-    pub async fn run_command_from_options(
-        &self,
-        ctx: &Context,
-        command_invoke: &mut CommandInvoke,
-        command_options: CommandOptions,
-    ) {
-        let command = {
-            self.commands_map
-                .get(&command_options.command)
-                .expect(&format!("Received invalid command: {}", command_options.command))
-        };
-
-        match command.fun {
-            CommandFnType::Slash(t) => t(&ctx, command_invoke, command_options).await,
-            CommandFnType::Multi(m) => m(&ctx, command_invoke).await,
-        }
-    }
-}

src/hooks.rs

@@ -1,107 +1,79 @@
-use regex_command_attr::check;
-use serenity::{client::Context, model::channel::Channel};
+use poise::serenity::model::channel::Channel;
 
-use crate::{
-    framework::{CommandInvoke, CommandOptions, CreateGenericResponse, HookResult},
-    moderation_cmds, RecordingMacros,
-};
+use crate::{consts::MACRO_MAX_COMMANDS, models::command_macro::RecordedCommand, Context, Error};
 
-#[check]
-pub async fn guild_only(
-    ctx: &Context,
-    invoke: &mut CommandInvoke,
-    _args: &CommandOptions,
-) -> HookResult {
-    if invoke.guild_id().is_some() {
-        HookResult::Continue
-    } else {
-        let _ = invoke
-            .respond(
-                &ctx,
-                CreateGenericResponse::new().content("This command can only be used in servers"),
-            )
-            .await;
+async fn macro_check(ctx: Context<'_>) -> bool {
+    if let Context::Application(app_ctx) = ctx {
+        if let Some(guild_id) = ctx.guild_id() {
+            if ctx.command().identifying_name != "finish_macro" {
+                let mut lock = ctx.data().recording_macros.write().await;
 
-        HookResult::Halt
-    }
-}
+                if let Some(command_macro) = lock.get_mut(&(guild_id, ctx.author().id)) {
+                    if command_macro.commands.len() >= MACRO_MAX_COMMANDS {
+                        let _ = ctx.send(|m| {
+                                m.ephemeral(true).content(
+                                    format!("{} commands already recorded. Please use `/macro finish` to end recording.", MACRO_MAX_COMMANDS),
+                                )
+                            })
+                            .await;
+                    } else {
+                        let recorded = RecordedCommand {
+                            action: None,
+                            command_name: ctx.command().identifying_name.clone(),
+                            options: Vec::from(app_ctx.args),
+                        };
 
-#[check]
-pub async fn macro_check(
-    ctx: &Context,
-    invoke: &mut CommandInvoke,
-    args: &CommandOptions,
-) -> HookResult {
-    if let Some(guild_id) = invoke.guild_id() {
-        if args.command != moderation_cmds::MACRO_CMD_COMMAND.names[0] {
-            let active_recordings =
-                ctx.data.read().await.get::<RecordingMacros>().cloned().unwrap();
-            let mut lock = active_recordings.write().await;
+                        command_macro.commands.push(recorded);
 
-            if let Some(command_macro) = lock.get_mut(&(guild_id, invoke.author_id())) {
-                if command_macro.commands.len() >= 5 {
-                    let _ = invoke
-                        .respond(
-                            &ctx,
-                            CreateGenericResponse::new().content("5 commands already recorded. Please use `/macro finish` to end recording."),
-                        )
-                        .await;
+                        let _ = ctx
+                            .send(|m| m.ephemeral(true).content("Command recorded to macro"))
+                            .await;
+                    }
+
+                    false
                 } else {
-                    command_macro.commands.push(args.clone());
-
-                    let _ = invoke
-                        .respond(
-                            &ctx,
-                            CreateGenericResponse::new().content("Command recorded to macro"),
-                        )
-                        .await;
+                    true
                 }
-
-                HookResult::Halt
             } else {
-                HookResult::Continue
+                true
             }
         } else {
-            HookResult::Continue
+            true
         }
     } else {
-        HookResult::Continue
+        true
     }
 }
 
-#[check]
-pub async fn check_self_permissions(
-    ctx: &Context,
-    invoke: &mut CommandInvoke,
-    _args: &CommandOptions,
-) -> HookResult {
-    if let Some(guild) = invoke.guild(&ctx) {
-        let user_id = ctx.cache.current_user_id();
+async fn check_self_permissions(ctx: Context<'_>) -> bool {
+    if let Some(guild) = ctx.guild() {
+        let user_id = ctx.discord().cache.current_user_id();
 
-        let manage_webhooks =
-            guild.member_permissions(&ctx, user_id).await.map_or(false, |p| p.manage_webhooks());
-        let (view_channel, send_messages, embed_links) = invoke
+        let manage_webhooks = guild
+            .member_permissions(&ctx.discord(), user_id)
+            .await
+            .map_or(false, |p| p.manage_webhooks());
+        let (view_channel, send_messages, embed_links) = ctx
             .channel_id()
-            .to_channel_cached(&ctx)
+            .to_channel_cached(&ctx.discord())
             .map(|c| {
                 if let Channel::Guild(channel) = c {
-                    channel.permissions_for_user(ctx, user_id).ok()
+                    channel.permissions_for_user(&ctx.discord(), user_id).ok()
                 } else {
                     None
                 }
             })
             .flatten()
             .map_or((false, false, false), |p| {
-                (p.read_messages(), p.send_messages(), p.embed_links())
+                (p.view_channel(), p.send_messages(), p.embed_links())
             });
 
         if manage_webhooks && send_messages && embed_links {
-            HookResult::Continue
+            true
         } else {
-            let _ = invoke
-                .respond(
-                    &ctx,
-                    CreateGenericResponse::new().content(format!(
+            let _ = ctx
+                .send(|m| {
+                    m.content(format!(
                         "Please ensure the bot has the correct permissions:
 
 {}     **View Channel**
@@ -112,41 +84,17 @@ pub async fn check_self_permissions(
                         if send_messages { "✅" } else { "❌" },
                         if manage_webhooks { "✅" } else { "❌" },
                         if embed_links { "✅" } else { "❌" },
-                    )),
-                )
+                    ))
+                })
                 .await;
 
-            HookResult::Halt
+            false
         }
     } else {
-        HookResult::Continue
+        true
     }
 }
 
-#[check]
-pub async fn check_guild_permissions(
-    ctx: &Context,
-    invoke: &mut CommandInvoke,
-    _args: &CommandOptions,
-) -> HookResult {
-    if let Some(guild) = invoke.guild(&ctx) {
-        let permissions = guild.member_permissions(&ctx, invoke.author_id()).await.unwrap();
-
-        if !permissions.manage_guild() {
-            let _ = invoke
-                .respond(
-                    &ctx,
-                    CreateGenericResponse::new().content(
-                        "You must have the \"Manage Server\" permission to use this command",
-                    ),
-                )
-                .await;
-
-            HookResult::Halt
-        } else {
-            HookResult::Continue
-        }
-    } else {
-        HookResult::Continue
-    }
+pub async fn all_checks(ctx: Context<'_>) -> Result<bool, Error> {
+    Ok(macro_check(ctx).await && check_self_permissions(ctx).await)
 }

src/interval_parser.rs

@@ -1,5 +1,9 @@
 /*
-Copyright 2021 Paul Colomiets, 2022 Jude Southworth
+With modifications, 2022 Jude Southworth
+
+Original copyright notice:
+
+Copyright 2021 Paul Colomiets
 
 Permission is hereby granted, free of charge, to any person obtaining a copy of this software
 and associated documentation files (the "Software"), to deal in the Software without restriction,

src/main.rs

@@ -6,7 +6,6 @@ mod commands;
 mod component_models;
 mod consts;
 mod event_handlers;
-mod framework;
 mod hooks;
 mod interval_parser;
 mod models;
@@ -16,145 +15,188 @@ mod utils;
 use std::{
     collections::HashMap,
     env,
-    sync::{atomic::AtomicBool, Arc},
+    error::Error as StdError,
+    fmt::{Debug, Display, Formatter},
+    sync::atomic::AtomicBool,
 };
 
 use chrono_tz::Tz;
 use dotenv::dotenv;
-use log::info;
-use serenity::{
-    client::Client,
-    http::client::Http,
-    model::{
-        gateway::GatewayIntents,
-        id::{GuildId, UserId},
-    },
-    prelude::TypeMapKey,
+use poise::serenity::model::{
+    gateway::{Activity, GatewayIntents},
+    id::{GuildId, UserId},
 };
-use sqlx::mysql::MySqlPool;
-use tokio::sync::RwLock;
+use sqlx::{MySql, Pool};
+use tokio::sync::{broadcast, broadcast::Sender, RwLock};
 
 use crate::{
     commands::{info_cmds, moderation_cmds, reminder_cmds, todo_cmds},
-    component_models::ComponentDataModel,
     consts::THEME_COLOR,
-    framework::RegexFramework,
+    event_handlers::listener,
+    hooks::all_checks,
     models::command_macro::CommandMacro,
+    utils::register_application_commands,
 };
 
-struct SQLPool;
+type Database = MySql;
 
-impl TypeMapKey for SQLPool {
-    type Value = MySqlPool;
-}
+type Error = Box<dyn std::error::Error + Send + Sync>;
+type Context<'a> = poise::Context<'a, Data, Error>;
 
-struct ReqwestClient;
-
-impl TypeMapKey for ReqwestClient {
-    type Value = Arc<reqwest::Client>;
-}
-
-struct PopularTimezones;
-
-impl TypeMapKey for PopularTimezones {
-    type Value = Arc<Vec<Tz>>;
-}
-
-struct RecordingMacros;
-
-impl TypeMapKey for RecordingMacros {
-    type Value = Arc<RwLock<HashMap<(GuildId, UserId), CommandMacro>>>;
-}
-
-struct Handler {
+pub struct Data {
+    database: Pool<Database>,
+    http: reqwest::Client,
+    recording_macros: RwLock<HashMap<(GuildId, UserId), CommandMacro<Data, Error>>>,
+    popular_timezones: Vec<Tz>,
     is_loop_running: AtomicBool,
+    broadcast: Sender<()>,
 }
 
+impl std::fmt::Debug for Data {
+    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
+        write!(f, "Data {{ .. }}")
+    }
+}
+
+struct Ended;
+
+impl Debug for Ended {
+    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
+        f.write_str("Process ended.")
+    }
+}
+
+impl Display for Ended {
+    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
+        f.write_str("Process ended.")
+    }
+}
+
+impl StdError for Ended {}
+
 #[tokio::main]
-async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
+async fn main() -> Result<(), Box<dyn StdError + Send + Sync>> {
+    let (tx, mut rx) = broadcast::channel(16);
+
+    tokio::select! {
+        output = _main(tx) => output,
+        _ = rx.recv() => Err(Box::new(Ended) as Box<dyn StdError + Send + Sync>)
+    }
+}
+
+async fn _main(tx: Sender<()>) -> Result<(), Box<dyn StdError + Send + Sync>> {
     env_logger::init();
 
     dotenv()?;
 
-    let token = env::var("DISCORD_TOKEN").expect("Missing DISCORD_TOKEN from environment");
+    let discord_token = env::var("DISCORD_TOKEN").expect("Missing DISCORD_TOKEN from environment");
 
-    let application_id = {
-        let http = Http::new_with_token(&token);
-
-        http.get_current_application_info().await?.id
+    let options = poise::FrameworkOptions {
+        commands: vec![
+            info_cmds::help(),
+            info_cmds::info(),
+            info_cmds::donate(),
+            info_cmds::clock(),
+            info_cmds::clock_context_menu(),
+            info_cmds::dashboard(),
+            moderation_cmds::timezone(),
+            poise::Command {
+                subcommands: vec![
+                    moderation_cmds::delete_macro(),
+                    moderation_cmds::finish_macro(),
+                    moderation_cmds::list_macro(),
+                    moderation_cmds::record_macro(),
+                    moderation_cmds::run_macro(),
+                ],
+                ..moderation_cmds::macro_base()
+            },
+            reminder_cmds::pause(),
+            reminder_cmds::offset(),
+            reminder_cmds::nudge(),
+            reminder_cmds::look(),
+            reminder_cmds::delete(),
+            poise::Command {
+                subcommands: vec![
+                    reminder_cmds::list_timer(),
+                    reminder_cmds::start_timer(),
+                    reminder_cmds::delete_timer(),
+                ],
+                ..reminder_cmds::timer_base()
+            },
+            reminder_cmds::remind(),
+            poise::Command {
+                subcommands: vec![
+                    poise::Command {
+                        subcommands: vec![
+                            todo_cmds::todo_guild_add(),
+                            todo_cmds::todo_guild_view(),
+                        ],
+                        ..todo_cmds::todo_guild_base()
+                    },
+                    poise::Command {
+                        subcommands: vec![
+                            todo_cmds::todo_channel_add(),
+                            todo_cmds::todo_channel_view(),
+                        ],
+                        ..todo_cmds::todo_channel_base()
+                    },
+                    poise::Command {
+                        subcommands: vec![todo_cmds::todo_user_add(), todo_cmds::todo_user_view()],
+                        ..todo_cmds::todo_user_base()
+                    },
+                ],
+                ..todo_cmds::todo_base()
+            },
+        ],
+        allowed_mentions: None,
+        command_check: Some(|ctx| Box::pin(all_checks(ctx))),
+        listener: |ctx, event, _framework, data| Box::pin(listener(ctx, event, data)),
+        ..Default::default()
     };
 
-    let dm_enabled = env::var("DM_ENABLED").map_or(true, |var| var == "1");
+    let database =
+        Pool::connect(&env::var("DATABASE_URL").expect("No database URL provided")).await.unwrap();
 
-    let framework = RegexFramework::new()
-        .ignore_bots(env::var("IGNORE_BOTS").map_or(true, |var| var == "1"))
-        .debug_guild(env::var("DEBUG_GUILD").map_or(None, |g| {
-            Some(GuildId(g.parse::<u64>().expect("DEBUG_GUILD must be a guild ID")))
-        }))
-        .dm_enabled(dm_enabled)
-        // info commands
-        .add_command(&info_cmds::HELP_COMMAND)
-        .add_command(&info_cmds::INFO_COMMAND)
-        .add_command(&info_cmds::DONATE_COMMAND)
-        .add_command(&info_cmds::DASHBOARD_COMMAND)
-        .add_command(&info_cmds::CLOCK_COMMAND)
-        // reminder commands
-        .add_command(&reminder_cmds::TIMER_COMMAND)
-        .add_command(&reminder_cmds::REMIND_COMMAND)
-        // management commands
-        .add_command(&reminder_cmds::DELETE_COMMAND)
-        .add_command(&reminder_cmds::LOOK_COMMAND)
-        .add_command(&reminder_cmds::PAUSE_COMMAND)
-        .add_command(&reminder_cmds::OFFSET_COMMAND)
-        .add_command(&reminder_cmds::NUDGE_COMMAND)
-        // to-do commands
-        .add_command(&todo_cmds::TODO_COMMAND)
-        // moderation commands
-        .add_command(&moderation_cmds::TIMEZONE_COMMAND)
-        .add_command(&moderation_cmds::MACRO_CMD_COMMAND)
-        .add_hook(&hooks::CHECK_SELF_PERMISSIONS_HOOK)
-        .add_hook(&hooks::MACRO_CHECK_HOOK);
+    let popular_timezones = sqlx::query!(
+        "SELECT timezone FROM users GROUP BY timezone ORDER BY COUNT(timezone) DESC LIMIT 21"
+    )
+    .fetch_all(&database)
+    .await
+    .unwrap()
+    .iter()
+    .map(|t| t.timezone.parse::<Tz>().unwrap())
+    .collect::<Vec<Tz>>();
 
-    let framework_arc = Arc::new(framework);
+    poise::Framework::build()
+        .token(discord_token)
+        .user_data_setup(move |ctx, _bot, framework| {
+            Box::pin(async move {
+                ctx.set_activity(Activity::watching("for /remind")).await;
 
-    let mut client = Client::builder(&token)
+                register_application_commands(
+                    ctx,
+                    framework,
+                    env::var("DEBUG_GUILD")
+                        .map(|inner| GuildId(inner.parse().expect("DEBUG_GUILD not valid")))
+                        .ok(),
+                )
+                .await
+                .unwrap();
+
+                Ok(Data {
+                    http: reqwest::Client::new(),
+                    database,
+                    popular_timezones,
+                    recording_macros: Default::default(),
+                    is_loop_running: AtomicBool::new(false),
+                    broadcast: tx,
+                })
+            })
+        })
+        .options(options)
         .intents(GatewayIntents::GUILDS)
-        .application_id(application_id.0)
-        .event_handler(Handler { is_loop_running: AtomicBool::from(false) })
-        .await
-        .expect("Error occurred creating client");
-
-    {
-        let pool = MySqlPool::connect(
-            &env::var("DATABASE_URL").expect("Missing DATABASE_URL from environment"),
-        )
-        .await
-        .unwrap();
-
-        let popular_timezones = sqlx::query!(
-            "SELECT timezone FROM users GROUP BY timezone ORDER BY COUNT(timezone) DESC LIMIT 21"
-        )
-        .fetch_all(&pool)
-        .await
-        .unwrap()
-        .iter()
-        .map(|t| t.timezone.parse::<Tz>().unwrap())
-        .collect::<Vec<Tz>>();
-
-        let mut data = client.data.write().await;
-
-        data.insert::<SQLPool>(pool);
-        data.insert::<PopularTimezones>(Arc::new(popular_timezones));
-        data.insert::<ReqwestClient>(Arc::new(reqwest::Client::new()));
-        data.insert::<RegexFramework>(framework_arc.clone());
-        data.insert::<RecordingMacros>(Arc::new(RwLock::new(HashMap::new())));
-    }
-
-    framework_arc.build_slash(&client.cache_and_http.http).await;
-
-    info!("Starting client as autosharded");
-
-    client.start_autosharded().await?;
+        .run_autosharded()
+        .await?;
 
     Ok(())
 }

src/models/channel_data.rs

@@ -1,5 +1,5 @@
 use chrono::NaiveDateTime;
-use serenity::model::channel::Channel;
+use poise::serenity::model::channel::Channel;
 use sqlx::MySqlPool;
 
 pub struct ChannelData {

src/models/command_macro.rs

@@ -1,33 +1,73 @@
-use serenity::{client::Context, model::id::GuildId};
+use poise::serenity::model::{
+    id::GuildId, interactions::application_command::ApplicationCommandInteractionDataOption,
+};
+use serde::{Deserialize, Serialize};
 
-use crate::{framework::CommandOptions, SQLPool};
+use crate::{Context, Data, Error};
 
-pub struct CommandMacro {
+fn default_none<U, E>() -> Option<
+    for<'a> fn(
+        poise::ApplicationContext<'a, U, E>,
+    ) -> poise::BoxFuture<'a, Result<(), poise::FrameworkError<'a, U, E>>>,
+> {
+    None
+}
+
+#[derive(Serialize, Deserialize)]
+pub struct RecordedCommand<U, E> {
+    #[serde(skip)]
+    #[serde(default = "default_none::<U, E>")]
+    pub action: Option<
+        for<'a> fn(
+            poise::ApplicationContext<'a, U, E>,
+        ) -> poise::BoxFuture<'a, Result<(), poise::FrameworkError<'a, U, E>>>,
+    >,
+    pub command_name: String,
+    pub options: Vec<ApplicationCommandInteractionDataOption>,
+}
+
+pub struct CommandMacro<U, E> {
     pub guild_id: GuildId,
     pub name: String,
     pub description: Option<String>,
-    pub commands: Vec<CommandOptions>,
+    pub commands: Vec<RecordedCommand<U, E>>,
 }
 
-impl CommandMacro {
-    pub async fn from_guild(ctx: &Context, guild_id: impl Into<GuildId>) -> Vec<Self> {
-        let pool = ctx.data.read().await.get::<SQLPool>().cloned().unwrap();
-        let guild_id = guild_id.into();
+pub async fn guild_command_macro(
+    ctx: &Context<'_>,
+    name: &str,
+) -> Option<CommandMacro<Data, Error>> {
+    let row = sqlx::query!(
+        "
+SELECT * FROM macro WHERE guild_id = (SELECT id FROM guilds WHERE guild = ?) AND name = ?
+        ",
+        ctx.guild_id().unwrap().0,
+        name
+    )
+    .fetch_one(&ctx.data().database)
+    .await
+    .ok()?;
 
-        sqlx::query!(
-            "SELECT * FROM macro WHERE guild_id = (SELECT id FROM guilds WHERE guild = ?)",
-            guild_id.0
-        )
-        .fetch_all(&pool)
-        .await
-        .unwrap()
-        .iter()
-        .map(|row| Self {
-            guild_id,
-            name: row.name.clone(),
-            description: row.description.clone(),
-            commands: serde_json::from_str(&row.commands).unwrap(),
-        })
-        .collect::<Vec<Self>>()
+    let mut commands: Vec<RecordedCommand<Data, Error>> =
+        serde_json::from_str(&row.commands).unwrap();
+
+    for recorded_command in &mut commands {
+        let command = &ctx
+            .framework()
+            .options()
+            .commands
+            .iter()
+            .find(|c| c.identifying_name == recorded_command.command_name);
+
+        recorded_command.action = command.map(|c| c.slash_action).flatten().clone();
     }
+
+    let command_macro = CommandMacro {
+        guild_id: ctx.guild_id().unwrap(),
+        name: row.name,
+        description: row.description,
+        commands,
+    };
+
+    Some(command_macro)
 }

src/models/mod.rs

@@ -5,62 +5,71 @@ pub mod timer;
 pub mod user_data;
 
 use chrono_tz::Tz;
-use serenity::{
-    async_trait,
-    model::id::{ChannelId, UserId},
-    prelude::Context,
-};
+use poise::serenity::{async_trait, model::id::UserId};
 
 use crate::{
     models::{channel_data::ChannelData, user_data::UserData},
-    SQLPool,
+    CommandMacro, Context, Data, Error, GuildId,
 };
 
 #[async_trait]
 pub trait CtxData {
-    async fn user_data<U: Into<UserId> + Send + Sync>(
-        &self,
-        user_id: U,
-    ) -> Result<UserData, Box<dyn std::error::Error + Sync + Send>>;
+    async fn user_data<U: Into<UserId> + Send>(&self, user_id: U) -> Result<UserData, Error>;
 
-    async fn timezone<U: Into<UserId> + Send + Sync>(&self, user_id: U) -> Tz;
+    async fn author_data(&self) -> Result<UserData, Error>;
 
-    async fn channel_data<C: Into<ChannelId> + Send + Sync>(
-        &self,
-        channel_id: C,
-    ) -> Result<ChannelData, Box<dyn std::error::Error + Sync + Send>>;
+    async fn timezone(&self) -> Tz;
+
+    async fn channel_data(&self) -> Result<ChannelData, Error>;
+
+    async fn command_macros(&self) -> Result<Vec<CommandMacro<Data, Error>>, Error>;
 }
 
 #[async_trait]
-impl CtxData for Context {
-    async fn user_data<U: Into<UserId> + Send + Sync>(
+impl CtxData for Context<'_> {
+    async fn user_data<U: Into<UserId> + Send>(
         &self,
         user_id: U,
     ) -> Result<UserData, Box<dyn std::error::Error + Sync + Send>> {
-        let user_id = user_id.into();
-        let pool = self.data.read().await.get::<SQLPool>().cloned().unwrap();
-
-        let user = user_id.to_user(self).await.unwrap();
-
-        UserData::from_user(&user, &self, &pool).await
+        UserData::from_user(user_id, &self.discord(), &self.data().database).await
     }
 
-    async fn timezone<U: Into<UserId> + Send + Sync>(&self, user_id: U) -> Tz {
-        let user_id = user_id.into();
-        let pool = self.data.read().await.get::<SQLPool>().cloned().unwrap();
-
-        UserData::timezone_of(user_id, &pool).await
+    async fn author_data(&self) -> Result<UserData, Box<dyn std::error::Error + Sync + Send>> {
+        UserData::from_user(&self.author().id, &self.discord(), &self.data().database).await
     }
 
-    async fn channel_data<C: Into<ChannelId> + Send + Sync>(
-        &self,
-        channel_id: C,
-    ) -> Result<ChannelData, Box<dyn std::error::Error + Sync + Send>> {
-        let channel_id = channel_id.into();
-        let pool = self.data.read().await.get::<SQLPool>().cloned().unwrap();
+    async fn timezone(&self) -> Tz {
+        UserData::timezone_of(self.author().id, &self.data().database).await
+    }
 
-        let channel = channel_id.to_channel_cached(&self).unwrap();
+    async fn channel_data(&self) -> Result<ChannelData, Box<dyn std::error::Error + Sync + Send>> {
+        let channel = self.channel_id().to_channel_cached(&self.discord()).unwrap();
 
-        ChannelData::from_channel(&channel, &pool).await
+        ChannelData::from_channel(&channel, &self.data().database).await
+    }
+
+    async fn command_macros(&self) -> Result<Vec<CommandMacro<Data, Error>>, Error> {
+        self.data().command_macros(self.guild_id().unwrap()).await
+    }
+}
+
+impl Data {
+    pub(crate) async fn command_macros(
+        &self,
+        guild_id: GuildId,
+    ) -> Result<Vec<CommandMacro<Data, Error>>, Error> {
+        let rows = sqlx::query!(
+            "SELECT name, description, commands FROM macro WHERE guild_id = (SELECT id FROM guilds WHERE guild = ?)",
+            guild_id.0
+        )
+        .fetch_all(&self.database)
+        .await?.iter().map(|row| CommandMacro {
+            guild_id,
+            name: row.name.clone(),
+            description: row.description.clone(),
+            commands: serde_json::from_str(&row.commands).unwrap(),
+        }).collect();
+
+        Ok(rows)
     }
 }

src/models/reminder/builder.rs

@@ -2,8 +2,7 @@ use std::{collections::HashSet, fmt::Display};
 
 use chrono::{Duration, NaiveDateTime, Utc};
 use chrono_tz::Tz;
-use serenity::{
-    client::Context,
+use poise::serenity::{
     http::CacheHttp,
     model::{
         channel::GuildChannel,
@@ -15,15 +14,14 @@ use serenity::{
 use sqlx::MySqlPool;
 
 use crate::{
-    consts,
-    consts::{DAY, MAX_TIME, MIN_INTERVAL},
+    consts::{DAY, DEFAULT_AVATAR, MAX_TIME, MIN_INTERVAL},
     interval_parser::Interval,
     models::{
         channel_data::ChannelData,
         reminder::{content::Content, errors::ReminderError, helper::generate_uid, Reminder},
         user_data::UserData,
     },
-    SQLPool,
+    Context,
 };
 
 async fn create_webhook(
@@ -31,7 +29,7 @@ async fn create_webhook(
     channel: GuildChannel,
     name: impl Display,
 ) -> SerenityResult<Webhook> {
-    channel.create_webhook_with_avatar(ctx.http(), name, consts::DEFAULT_AVATAR.clone()).await
+    channel.create_webhook_with_avatar(ctx.http(), name, DEFAULT_AVATAR.clone()).await
 }
 
 #[derive(Hash, PartialEq, Eq)]
@@ -145,7 +143,7 @@ pub struct MultiReminderBuilder<'a> {
     expires: Option<NaiveDateTime>,
     content: Content,
     set_by: Option<u32>,
-    ctx: &'a Context,
+    ctx: &'a Context<'a>,
     guild_id: Option<GuildId>,
 }
 
@@ -210,8 +208,6 @@ impl<'a> MultiReminderBuilder<'a> {
     }
 
     pub async fn build(self) -> (HashSet<ReminderError>, HashSet<ReminderScope>) {
-        let pool = self.ctx.data.read().await.get::<SQLPool>().cloned().unwrap();
-
         let mut errors = HashSet::new();
 
         let mut ok_locs = HashSet::new();
@@ -225,12 +221,17 @@ impl<'a> MultiReminderBuilder<'a> {
             for scope in self.scopes {
                 let db_channel_id = match scope {
                     ReminderScope::User(user_id) => {
-                        if let Ok(user) = UserId(user_id).to_user(&self.ctx).await {
-                            let user_data =
-                                UserData::from_user(&user, &self.ctx, &pool).await.unwrap();
+                        if let Ok(user) = UserId(user_id).to_user(&self.ctx.discord()).await {
+                            let user_data = UserData::from_user(
+                                &user,
+                                &self.ctx.discord(),
+                                &self.ctx.data().database,
+                            )
+                            .await
+                            .unwrap();
 
                             if let Some(guild_id) = self.guild_id {
-                                if guild_id.member(&self.ctx, user).await.is_err() {
+                                if guild_id.member(&self.ctx.discord(), user).await.is_err() {
                                     Err(ReminderError::InvalidTag)
                                 } else {
                                     Ok(user_data.dm_channel)
@@ -243,26 +244,36 @@ impl<'a> MultiReminderBuilder<'a> {
                         }
                     }
                     ReminderScope::Channel(channel_id) => {
-                        let channel = ChannelId(channel_id).to_channel(&self.ctx).await.unwrap();
+                        let channel =
+                            ChannelId(channel_id).to_channel(&self.ctx.discord()).await.unwrap();
 
                         if let Some(guild_channel) = channel.clone().guild() {
                             if Some(guild_channel.guild_id) != self.guild_id {
                                 Err(ReminderError::InvalidTag)
                             } else {
                                 let mut channel_data =
-                                    ChannelData::from_channel(&channel, &pool).await.unwrap();
+                                    ChannelData::from_channel(&channel, &self.ctx.data().database)
+                                        .await
+                                        .unwrap();
 
                                 if channel_data.webhook_id.is_none()
                                     || channel_data.webhook_token.is_none()
                                 {
-                                    match create_webhook(&self.ctx, guild_channel, "Reminder").await
+                                    match create_webhook(
+                                        &self.ctx.discord(),
+                                        guild_channel,
+                                        "Reminder",
+                                    )
+                                    .await
                                     {
                                         Ok(webhook) => {
                                             channel_data.webhook_id =
                                                 Some(webhook.id.as_u64().to_owned());
                                             channel_data.webhook_token = webhook.token;
 
-                                            channel_data.commit_changes(&pool).await;
+                                            channel_data
+                                                .commit_changes(&self.ctx.data().database)
+                                                .await;
 
                                             Ok(channel_data.id)
                                         }
@@ -282,7 +293,7 @@ impl<'a> MultiReminderBuilder<'a> {
                 match db_channel_id {
                     Ok(c) => {
                         let builder = ReminderBuilder {
-                            pool: pool.clone(),
+                            pool: self.ctx.data().database.clone(),
                             uid: generate_uid(),
                             channel: c,
                             utc_time: self.utc_time,

src/models/reminder/look_flags.rs

@@ -1,6 +1,6 @@
+use poise::serenity::model::id::ChannelId;
 use serde::{Deserialize, Serialize};
 use serde_repr::*;
-use serenity::model::id::ChannelId;
 
 #[derive(Serialize_repr, Deserialize_repr, Copy, Clone, Debug)]
 #[repr(u8)]

src/models/reminder/mod.rs

@@ -6,15 +6,15 @@ pub mod look_flags;
 
 use chrono::{NaiveDateTime, TimeZone};
 use chrono_tz::Tz;
-use serenity::{
-    client::Context,
-    model::id::{ChannelId, GuildId, UserId},
+use poise::{
+    serenity::model::id::{ChannelId, GuildId, UserId},
+    serenity_prelude::Cache,
 };
-use sqlx::MySqlPool;
+use sqlx::Executor;
 
 use crate::{
     models::reminder::look_flags::{LookFlags, TimeDisplayType},
-    SQLPool,
+    Database,
 };
 
 #[derive(Debug, Clone)]
@@ -33,7 +33,10 @@ pub struct Reminder {
 }
 
 impl Reminder {
-    pub async fn from_uid(pool: &MySqlPool, uid: String) -> Option<Self> {
+    pub async fn from_uid(
+        pool: impl Executor<'_, Database = Database>,
+        uid: String,
+    ) -> Option<Self> {
         sqlx::query_as_unchecked!(
             Self,
             "
@@ -70,12 +73,10 @@ WHERE
     }
 
     pub async fn from_channel<C: Into<ChannelId>>(
-        ctx: &Context,
+        pool: impl Executor<'_, Database = Database>,
         channel_id: C,
         flags: &LookFlags,
     ) -> Vec<Self> {
-        let pool = ctx.data.read().await.get::<SQLPool>().cloned().unwrap();
-
         let enabled = if flags.show_disabled { "0,1" } else { "1" };
         let channel_id = channel_id.into();
 
@@ -113,16 +114,19 @@ ORDER BY
             channel_id.as_u64(),
             enabled,
         )
-        .fetch_all(&pool)
+        .fetch_all(pool)
         .await
         .unwrap()
     }
 
-    pub async fn from_guild(ctx: &Context, guild_id: Option<GuildId>, user: UserId) -> Vec<Self> {
-        let pool = ctx.data.read().await.get::<SQLPool>().cloned().unwrap();
-
+    pub async fn from_guild(
+        cache: impl AsRef<Cache>,
+        pool: impl Executor<'_, Database = Database>,
+        guild_id: Option<GuildId>,
+        user: UserId,
+    ) -> Vec<Self> {
         if let Some(guild_id) = guild_id {
-            let guild_opt = guild_id.to_guild_cached(&ctx);
+            let guild_opt = guild_id.to_guild_cached(cache);
 
             if let Some(guild) = guild_opt {
                 let channels = guild
@@ -163,7 +167,7 @@ WHERE
                 ",
                     channels
                 )
-                .fetch_all(&pool)
+                .fetch_all(pool)
                 .await
             } else {
                 sqlx::query_as_unchecked!(
@@ -196,7 +200,7 @@ WHERE
                 ",
                     guild_id.as_u64()
                 )
-                .fetch_all(&pool)
+                .fetch_all(pool)
                 .await
             }
         } else {
@@ -230,7 +234,7 @@ WHERE
             ",
                 user.as_u64()
             )
-            .fetch_all(&pool)
+            .fetch_all(pool)
             .await
         }
         .unwrap()

src/models/user_data.rs

@@ -1,9 +1,6 @@
 use chrono_tz::Tz;
 use log::error;
-use serenity::{
-    http::CacheHttp,
-    model::{id::UserId, user::User},
-};
+use poise::serenity::{http::CacheHttp, model::id::UserId};
 use sqlx::MySqlPool;
 
 use crate::consts::LOCAL_TIMEZONE;
@@ -11,7 +8,6 @@ use crate::consts::LOCAL_TIMEZONE;
 pub struct UserData {
     pub id: u32,
     pub user: u64,
-    pub name: String,
     pub dm_channel: u32,
     pub timezone: String,
 }
@@ -40,20 +36,20 @@ SELECT timezone FROM users WHERE user = ?
         .unwrap()
     }
 
-    pub async fn from_user(
-        user: &User,
+    pub async fn from_user<U: Into<UserId>>(
+        user: U,
         ctx: impl CacheHttp,
         pool: &MySqlPool,
     ) -> Result<Self, Box<dyn std::error::Error + Sync + Send>> {
-        let user_id = user.id.as_u64().to_owned();
+        let user_id = user.into();
 
         match sqlx::query_as_unchecked!(
             Self,
             "
-SELECT id, user, name, dm_channel, IF(timezone IS NULL, ?, timezone) AS timezone FROM users WHERE user = ?
+SELECT id, user, dm_channel, IF(timezone IS NULL, ?, timezone) AS timezone FROM users WHERE user = ?
             ",
             *LOCAL_TIMEZONE,
-            user_id
+            user_id.0
         )
         .fetch_one(pool)
         .await
@@ -61,27 +57,24 @@ SELECT id, user, name, dm_channel, IF(timezone IS NULL, ?, timezone) AS timezone
             Ok(c) => Ok(c),
 
             Err(sqlx::Error::RowNotFound) => {
-                let dm_channel = user.create_dm_channel(ctx).await?;
-                let dm_id = dm_channel.id.as_u64().to_owned();
-
+                let dm_channel = user_id.create_dm_channel(ctx).await?;
                 let pool_c = pool.clone();
 
                 sqlx::query!(
                     "
 INSERT IGNORE INTO channels (channel) VALUES (?)
                     ",
-                    dm_id
+                    dm_channel.id.0
                 )
                 .execute(&pool_c)
                 .await?;
 
                 sqlx::query!(
                     "
-INSERT INTO users (user, name, dm_channel, timezone) VALUES (?, ?, (SELECT id FROM channels WHERE channel = ?), ?)
+INSERT INTO users (user, dm_channel, timezone) VALUES (?, (SELECT id FROM channels WHERE channel = ?), ?)
                     ",
-                    user_id,
-                    user.name,
-                    dm_id,
+                    user_id.0,
+                    dm_channel.id.0,
                     *LOCAL_TIMEZONE
                 )
                 .execute(&pool_c)
@@ -90,9 +83,9 @@ INSERT INTO users (user, name, dm_channel, timezone) VALUES (?, ?, (SELECT id FR
                 Ok(sqlx::query_as_unchecked!(
                     Self,
                     "
-SELECT id, user, name, dm_channel, timezone FROM users WHERE user = ?
+SELECT id, user, dm_channel, timezone FROM users WHERE user = ?
                     ",
-                    user_id
+                    user_id.0
                 )
                 .fetch_one(pool)
                 .await?)
@@ -109,9 +102,8 @@ SELECT id, user, name, dm_channel, timezone FROM users WHERE user = ?
     pub async fn commit_changes(&self, pool: &MySqlPool) {
         sqlx::query!(
             "
-UPDATE users SET name = ?, timezone = ? WHERE id = ?
+UPDATE users SET timezone = ? WHERE id = ?
             ",
-            self.name,
             self.timezone,
             self.id
         )

src/utils.rs

@@ -1,9 +1,42 @@
-use serenity::{
-    http::CacheHttp,
-    model::id::{GuildId, UserId},
+use poise::{
+    serenity::{
+        builder::CreateApplicationCommands,
+        http::CacheHttp,
+        model::id::{GuildId, UserId},
+    },
+    serenity_prelude as serenity,
 };
 
-use crate::consts::{CNC_GUILD, SUBSCRIPTION_ROLES};
+use crate::{
+    consts::{CNC_GUILD, SUBSCRIPTION_ROLES},
+    Data, Error,
+};
+
+pub async fn register_application_commands(
+    ctx: &poise::serenity::client::Context,
+    framework: &poise::Framework<Data, Error>,
+    guild_id: Option<GuildId>,
+) -> Result<(), poise::serenity::Error> {
+    let mut commands_builder = CreateApplicationCommands::default();
+    let commands = &framework.options().commands;
+    for command in commands {
+        if let Some(slash_command) = command.create_as_slash_command() {
+            commands_builder.add_application_command(slash_command);
+        }
+        if let Some(context_menu_command) = command.create_as_context_menu_command() {
+            commands_builder.add_application_command(context_menu_command);
+        }
+    }
+    let commands_builder = poise::serenity::json::Value::Array(commands_builder.0);
+
+    if let Some(guild_id) = guild_id {
+        ctx.http.create_guild_application_commands(guild_id.0, &commands_builder).await?;
+    } else {
+        ctx.http.create_global_application_commands(&commands_builder).await?;
+    }
+
+    Ok(())
+}
 
 pub async fn check_subscription(cache_http: impl CacheHttp, user_id: impl Into<UserId>) -> bool {
     if let Some(subscription_guild) = *CNC_GUILD {
@@ -35,3 +68,40 @@ pub async fn check_guild_subscription(
         false
     }
 }
+
+/// Sends the message, specified via [`crate::CreateReply`], to the interaction initial response
+/// endpoint
+pub fn send_as_initial_response(
+    data: poise::CreateReply<'_>,
+    f: &mut serenity::CreateInteractionResponseData,
+) {
+    let poise::CreateReply {
+        content,
+        embeds,
+        attachments: _, // serenity doesn't support attachments in initial response yet
+        components,
+        ephemeral,
+        allowed_mentions,
+        reference_message: _, // can't reply to a message in interactions
+    } = data;
+
+    if let Some(content) = content {
+        f.content(content);
+    }
+    f.set_embeds(embeds);
+    if let Some(allowed_mentions) = allowed_mentions {
+        f.allowed_mentions(|f| {
+            *f = allowed_mentions.clone();
+            f
+        });
+    }
+    if let Some(components) = components {
+        f.components(|f| {
+            f.0 = components.0;
+            f
+        });
+    }
+    if ephemeral {
+        f.flags(serenity::InteractionApplicationCommandCallbackDataFlags::EPHEMERAL);
+    }
+}

web/Cargo.toml

@@ -7,11 +7,15 @@ edition = "2018"
 [dependencies]
 rocket = { git = "https://github.com/SergioBenitez/Rocket", branch = "master", features = ["tls", "secrets", "json"] }
 rocket_dyn_templates = { git = "https://github.com/SergioBenitez/Rocket", branch = "master", features = ["tera"] }
-serenity = { git = "https://github.com/serenity-rs/serenity", branch = "next", default-features = false, features = ["builder", "cache", "client", "gateway", "http", "model", "utils", "rustls_backend"] }
+serenity = { version = "0.11.1", default-features = false, features = ["builder", "cache", "client", "gateway", "http", "model", "utils", "rustls_backend"] }
 oauth2 = "4"
 log = "0.4"
 reqwest = "0.11"
 serde = { version = "1.0", features = ["derive"] }
-sqlx = { version = "0.5", features = ["runtime-tokio-rustls", "macros", "mysql", "chrono"] }
+serde_json = "1.0"
+sqlx = { version = "0.5", features = ["runtime-tokio-rustls", "macros", "mysql", "chrono", "json"] }
 chrono = "0.4"
 chrono-tz = "0.5"
+lazy_static = "1.4.0"
+rand = "0.7"
+base64 = "0.13"

web/src/consts.rs

@@ -1,4 +1,52 @@
 pub const DISCORD_OAUTH_TOKEN: &'static str = "https://discord.com/api/oauth2/token";
 pub const DISCORD_OAUTH_AUTHORIZE: &'static str = "https://discord.com/api/oauth2/authorize";
 pub const DISCORD_API: &'static str = "https://discord.com/api";
-pub const DISCORD_CDN: &'static str = "https://cdn.discordapp.com/avatars";
+
+pub const MAX_CONTENT_LENGTH: usize = 2000;
+pub const MAX_EMBED_DESCRIPTION_LENGTH: usize = 4096;
+pub const MAX_EMBED_TITLE_LENGTH: usize = 256;
+pub const MAX_EMBED_AUTHOR_LENGTH: usize = 256;
+pub const MAX_EMBED_FOOTER_LENGTH: usize = 2048;
+pub const MAX_URL_LENGTH: usize = 512;
+pub const MAX_USERNAME_LENGTH: usize = 100;
+pub const MAX_EMBED_FIELDS: usize = 25;
+pub const MAX_EMBED_FIELD_TITLE_LENGTH: usize = 256;
+pub const MAX_EMBED_FIELD_VALUE_LENGTH: usize = 1024;
+
+pub const MINUTE: usize = 60;
+pub const HOUR: usize = 60 * MINUTE;
+pub const DAY: usize = 24 * HOUR;
+
+pub const CHARACTERS: &str = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789_";
+
+use std::{collections::HashSet, env, iter::FromIterator};
+
+use lazy_static::lazy_static;
+use serenity::model::prelude::AttachmentType;
+
+lazy_static! {
+    pub static ref DEFAULT_AVATAR: AttachmentType<'static> = (
+        include_bytes!(concat!(
+            env!("CARGO_MANIFEST_DIR"),
+            "/../assets/",
+            env!("WEBHOOK_AVATAR", "WEBHOOK_AVATAR not provided for compilation")
+        )) as &[u8],
+        env!("WEBHOOK_AVATAR"),
+    )
+        .into();
+    pub static ref SUBSCRIPTION_ROLES: HashSet<u64> = HashSet::from_iter(
+        env::var("SUBSCRIPTION_ROLES")
+            .map(|var| var
+                .split(',')
+                .filter_map(|item| { item.parse::<u64>().ok() })
+                .collect::<Vec<u64>>())
+            .unwrap_or_else(|_| Vec::new())
+    );
+    pub static ref CNC_GUILD: Option<u64> =
+        env::var("CNC_GUILD").map(|var| var.parse::<u64>().ok()).ok().flatten();
+    pub static ref MIN_INTERVAL: u32 = env::var("MIN_INTERVAL")
+        .ok()
+        .map(|inner| inner.parse::<u32>().ok())
+        .flatten()
+        .unwrap_or(600);
+}

web/src/lib.rs

@@ -2,22 +2,36 @@
 extern crate rocket;
 
 mod consts;
+#[macro_use]
+mod macros;
 mod routes;
 
-use rocket::fs::{relative, FileServer};
-use std::collections::HashMap;
+use std::{collections::HashMap, env};
 
-use oauth2::basic::BasicClient;
-use oauth2::{AuthUrl, ClientId, ClientSecret, RedirectUrl, TokenUrl};
-
-use crate::consts::{DISCORD_OAUTH_AUTHORIZE, DISCORD_OAUTH_TOKEN};
+use oauth2::