mirror of
https://github.com/supleed2/nanobot.git
synced 2024-09-19 21:46:18 +00:00
Split cmds.rs into submodules
This commit is contained in:
parent
8e40216174
commit
497b4576bc
SSH key fingerprint:
SHA256:CkbNRs0yVzXEiUp2zd0PSxsfRUMFF9bLlKXtE1xEbKM
803
src/cmds.rs
803
src/cmds.rs
|
|
@ -1,803 +0,0 @@
|
|||
use crate::{db, ACtx, Data, Error, ManualMember, Member, PendingMember};
|
||||
use poise::serenity_prelude as serenity;
|
||||
use poise::Modal;
|
||||
|
||||
/// Buttons to (de-)register application commands globally or by guild
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(prefix_command, owners_only)]
|
||||
pub(crate) async fn cmds(ctx: poise::Context<'_, Data, Error>) -> Result<(), Error> {
|
||||
tracing::info!("{}", ctx.author().name);
|
||||
poise::builtins::register_application_commands_buttons(ctx).await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Send (customisable) verification introduction message in specified channel
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn setup(
|
||||
ctx: ACtx<'_>,
|
||||
#[description = "Channel to send verification introduction message in"]
|
||||
#[channel_types("Text", "News")]
|
||||
channel: serenity::GuildChannel,
|
||||
) -> Result<(), Error> {
|
||||
#[derive(Modal)]
|
||||
struct Setup {
|
||||
#[name = "Contents of the verification intro message"]
|
||||
#[placeholder = "Welcome... Click the \"Begin\" button below to start verification"]
|
||||
#[paragraph]
|
||||
#[max_length = 500]
|
||||
message: String,
|
||||
#[name = "Emoji for the start button"]
|
||||
#[placeholder = "🚀"]
|
||||
#[max_length = 4]
|
||||
emoji: Option<String>,
|
||||
#[name = "Label for the start button"]
|
||||
#[placeholder = "Begin"]
|
||||
#[max_length = 80]
|
||||
text: Option<String>,
|
||||
}
|
||||
|
||||
tracing::info!("{} {}", ctx.author().name, channel.name());
|
||||
|
||||
if let Some(Setup {
|
||||
message,
|
||||
emoji,
|
||||
text,
|
||||
}) = Setup::execute(ctx).await?
|
||||
{
|
||||
ctx.say(format!("Sending intro message in {channel}"))
|
||||
.await?;
|
||||
let emoji = emoji.unwrap_or_default().chars().next().unwrap_or('🚀');
|
||||
channel
|
||||
.send_message(ctx.http(), |m| {
|
||||
m.content(message).components(|c| {
|
||||
c.create_action_row(|a| {
|
||||
a.create_button(|b| {
|
||||
b.style(serenity::ButtonStyle::Secondary)
|
||||
.emoji('📖')
|
||||
.label("More info")
|
||||
.custom_id("info")
|
||||
})
|
||||
.create_button(|b| {
|
||||
b.style(serenity::ButtonStyle::Primary)
|
||||
.emoji(emoji)
|
||||
.label(text.unwrap_or("Begin".to_string()))
|
||||
.custom_id("start")
|
||||
})
|
||||
})
|
||||
})
|
||||
})
|
||||
.await?;
|
||||
} else {
|
||||
ctx.say("Modal timed out, try again...").await?;
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Get the number of members in the members table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn count_members(ctx: ACtx<'_>) -> Result<(), Error> {
|
||||
tracing::info!("{}", ctx.author().name);
|
||||
let count = db::count_members(&ctx.data().db).await?;
|
||||
ctx.say(format!("There are {count} entries in the members table"))
|
||||
.await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Delete member info by Discord ID
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn delete_member(
|
||||
ctx: ACtx<'_>,
|
||||
id: serenity::Member,
|
||||
remove_roles: Option<bool>,
|
||||
) -> Result<(), Error> {
|
||||
tracing::info!("{} {}", ctx.author().name, id.user.name,);
|
||||
if db::delete_member_by_id(&ctx.data().db, id.user.id.into()).await? {
|
||||
if remove_roles.unwrap_or(true) {
|
||||
let mut m = id.clone();
|
||||
crate::verify::remove_role(ctx.serenity_context(), &mut m, ctx.data().member).await?;
|
||||
crate::verify::remove_role(ctx.serenity_context(), &mut m, ctx.data().fresher).await?;
|
||||
}
|
||||
ctx.say(format!("Successfully deleted member info for {id}"))
|
||||
.await?
|
||||
} else {
|
||||
ctx.say(format!("Failed to delete member info for {id}"))
|
||||
.await?
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Print all members in members table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn get_all_members(ctx: ACtx<'_>) -> Result<(), Error> {
|
||||
#[derive(Modal)]
|
||||
struct Confirm {
|
||||
#[name = "This will output the members db as text"]
|
||||
#[placeholder = "yes"]
|
||||
confirm: String,
|
||||
}
|
||||
|
||||
tracing::info!("{}", ctx.author().name);
|
||||
|
||||
if let Some(Confirm { confirm }) = Confirm::execute(ctx).await? {
|
||||
if confirm.to_lowercase().contains("yes") {
|
||||
let members = db::get_all_members(&ctx.data().db).await?;
|
||||
match tokio::fs::write("members.rs", format!("{members:#?}")).await {
|
||||
Ok(()) => {
|
||||
ctx.say("Sending members db data in followup message")
|
||||
.await?;
|
||||
ctx.channel_id()
|
||||
.send_files(&ctx.http(), vec!["members.rs"], |cm| {
|
||||
cm.content("File: members db")
|
||||
})
|
||||
.await?;
|
||||
}
|
||||
Err(e) => {
|
||||
tracing::error!("{e}");
|
||||
ctx.say("Failed to create members db file").await?;
|
||||
}
|
||||
}
|
||||
let _ = tokio::fs::remove_file("members.rs").await;
|
||||
Ok(())
|
||||
} else {
|
||||
ctx.say("Skipping members db output").await?;
|
||||
Ok(())
|
||||
}
|
||||
} else {
|
||||
ctx.say("Timed out").await?;
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
/// Unreachable, used to create get_member command folder
|
||||
#[poise::command(
|
||||
slash_command,
|
||||
subcommands(
|
||||
"get_member_by_id",
|
||||
"get_member_by_shortcode",
|
||||
"get_member_by_nickname",
|
||||
"get_member_by_realname",
|
||||
)
|
||||
)]
|
||||
pub(crate) async fn get_member(_ctx: ACtx<'_>) -> Result<(), Error> {
|
||||
unreachable!()
|
||||
}
|
||||
|
||||
/// Get member info by Discord ID
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command, rename = "id")]
|
||||
pub(crate) async fn get_member_by_id(ctx: ACtx<'_>, id: serenity::Member) -> Result<(), Error> {
|
||||
tracing::info!("{} {}", ctx.author().name, id.user.name,);
|
||||
match db::get_member_by_id(&ctx.data().db, id.user.id.into()).await? {
|
||||
Some(m) => {
|
||||
ctx.say(format!("Member info for {id}:\n```rust\n{m:#?}\n```"))
|
||||
.await?
|
||||
}
|
||||
None => ctx.say(format!("No member entry found for {id}")).await?,
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Get member info by Shortcode
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command, rename = "shortcode")]
|
||||
pub(crate) async fn get_member_by_shortcode(ctx: ACtx<'_>, shortcode: String) -> Result<(), Error> {
|
||||
tracing::info!("{} {shortcode}", ctx.author().name);
|
||||
match db::get_member_by_shortcode(&ctx.data().db, &shortcode).await? {
|
||||
Some(m) => {
|
||||
ctx.say(format!(
|
||||
"Member info for shortcode {shortcode}:\n```rust\n{m:#?}\n```"
|
||||
))
|
||||
.await?
|
||||
}
|
||||
None => {
|
||||
ctx.say(format!("No entry found for shortcode {shortcode}"))
|
||||
.await?
|
||||
}
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Get member info by Nickname
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command, rename = "nick")]
|
||||
pub(crate) async fn get_member_by_nickname(ctx: ACtx<'_>, nickname: String) -> Result<(), Error> {
|
||||
tracing::info!("{} {nickname}", ctx.author().name);
|
||||
match db::get_member_by_nickname(&ctx.data().db, &nickname).await? {
|
||||
Some(m) => {
|
||||
ctx.say(format!(
|
||||
"Member info for nickname {nickname}:\n```rust\n{m:#?}\n```"
|
||||
))
|
||||
.await?
|
||||
}
|
||||
None => {
|
||||
ctx.say(format!("No entry found for nickname {nickname}",))
|
||||
.await?
|
||||
}
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Get member info by Real Name
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command, rename = "name")]
|
||||
pub(crate) async fn get_member_by_realname(ctx: ACtx<'_>, realname: String) -> Result<(), Error> {
|
||||
tracing::info!("{} {realname}", ctx.author().name);
|
||||
match db::get_member_by_realname(&ctx.data().db, &realname).await? {
|
||||
Some(m) => {
|
||||
ctx.say(format!(
|
||||
"Member info for realname {realname}:\n```rust\n{m:#?}\n```"
|
||||
))
|
||||
.await?
|
||||
}
|
||||
None => {
|
||||
ctx.say(format!("No entry found for realname {realname}",))
|
||||
.await?
|
||||
}
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Add a member to the members table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn add_member(
|
||||
ctx: ACtx<'_>,
|
||||
id: serenity::Member,
|
||||
shortcode: String,
|
||||
nickname: String,
|
||||
realname: String,
|
||||
fresher: bool,
|
||||
) -> Result<(), Error> {
|
||||
tracing::info!(
|
||||
"{} {}, {shortcode}, {realname}, {nickname}",
|
||||
ctx.author().name,
|
||||
id.user.name,
|
||||
);
|
||||
db::insert_member(
|
||||
&ctx.data().db,
|
||||
Member {
|
||||
discord_id: id.user.id.into(),
|
||||
shortcode,
|
||||
nickname,
|
||||
realname,
|
||||
fresher,
|
||||
},
|
||||
)
|
||||
.await?;
|
||||
let mut m = id.clone();
|
||||
crate::verify::apply_role(ctx.serenity_context(), &mut m, ctx.data().member).await?;
|
||||
if fresher {
|
||||
crate::verify::apply_role(ctx.serenity_context(), &mut m, ctx.data().fresher).await?;
|
||||
}
|
||||
ctx.say(format!("Member added: {id}")).await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Manually add member to members table from pending table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn insert_member_from_pending(
|
||||
ctx: ACtx<'_>,
|
||||
id: serenity::Member,
|
||||
nickname: String,
|
||||
fresher: bool,
|
||||
) -> Result<(), Error> {
|
||||
tracing::info!("{} {}", ctx.author().name, id.user.name,);
|
||||
match db::insert_member_from_pending(&ctx.data().db, id.user.id.into(), &nickname, fresher)
|
||||
.await
|
||||
{
|
||||
Ok(()) => {
|
||||
ctx.say(format!("Member moved from pending to members table: {id}"))
|
||||
.await?
|
||||
}
|
||||
Err(e) => ctx.say(format!("Error: {e}")).await?,
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Manually add member to members table from manual table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn insert_member_from_manual(
|
||||
ctx: ACtx<'_>,
|
||||
id: serenity::Member,
|
||||
) -> Result<(), Error> {
|
||||
tracing::info!("{} {}", ctx.author().name, id.user.name,);
|
||||
match db::insert_member_from_manual(&ctx.data().db, id.user.id.into()).await {
|
||||
Ok(()) => {
|
||||
ctx.say(format!("Member moved from manual to members table: {id}"))
|
||||
.await?
|
||||
}
|
||||
Err(e) => ctx.say(format!("Error: {e}")).await?,
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Unreachable, used to create edit_member command folder
|
||||
#[poise::command(
|
||||
slash_command,
|
||||
subcommands(
|
||||
"edit_member_shortcode",
|
||||
"edit_member_nickname",
|
||||
"edit_member_realname",
|
||||
"edit_member_fresher",
|
||||
)
|
||||
)]
|
||||
pub(crate) async fn edit_member(_ctx: ACtx<'_>) -> Result<(), Error> {
|
||||
unreachable!()
|
||||
}
|
||||
|
||||
/// Edit member Shortcode
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command, rename = "shortcode")]
|
||||
pub(crate) async fn edit_member_shortcode(
|
||||
ctx: ACtx<'_>,
|
||||
id: serenity::Member,
|
||||
shortcode: String,
|
||||
) -> Result<(), Error> {
|
||||
tracing::info!("{} {shortcode}", ctx.author().name);
|
||||
if db::edit_member_shortcode(&ctx.data().db, id.user.id.into(), &shortcode).await? {
|
||||
ctx.say(format!("{id} Shortcode updated to {shortcode}"))
|
||||
.await?;
|
||||
} else {
|
||||
ctx.say(format!("Failed to update Shortcode for {id}"))
|
||||
.await?;
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Edit member Nickname
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command, rename = "nick")]
|
||||
pub(crate) async fn edit_member_nickname(
|
||||
ctx: ACtx<'_>,
|
||||
id: serenity::Member,
|
||||
nickname: String,
|
||||
) -> Result<(), Error> {
|
||||
tracing::info!("{} {nickname}", ctx.author().name);
|
||||
if db::edit_member_nickname(&ctx.data().db, id.user.id.into(), &nickname).await? {
|
||||
ctx.say(format!("{id} Nick updated to {nickname}")).await?;
|
||||
} else {
|
||||
ctx.say(format!("Failed to update Nick for {id}")).await?;
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Edit member Real Name
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command, rename = "name")]
|
||||
pub(crate) async fn edit_member_realname(
|
||||
ctx: ACtx<'_>,
|
||||
id: serenity::Member,
|
||||
realname: String,
|
||||
) -> Result<(), Error> {
|
||||
tracing::info!("{} {realname}", ctx.author().name);
|
||||
if db::edit_member_realname(&ctx.data().db, id.user.id.into(), &realname).await? {
|
||||
ctx.say(format!("{id} Name updated to {realname}")).await?;
|
||||
} else {
|
||||
ctx.say(format!("Failed to update Name for {id}")).await?;
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Edit member fresher status
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command, rename = "fresher")]
|
||||
pub(crate) async fn edit_member_fresher(
|
||||
ctx: ACtx<'_>,
|
||||
id: serenity::Member,
|
||||
fresher: bool,
|
||||
) -> Result<(), Error> {
|
||||
tracing::info!("{} {} {fresher}", ctx.author().name, id.user.name,);
|
||||
if db::edit_member_fresher(&ctx.data().db, id.user.id.into(), fresher).await? {
|
||||
ctx.say(format!("{id} Fresher status updated to {fresher}"))
|
||||
.await?;
|
||||
} else {
|
||||
ctx.say(format!("Failed to update Fresher status for {id}"))
|
||||
.await?;
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Set all members to non-freshers
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn set_members_non_fresher(ctx: ACtx<'_>) -> Result<(), Error> {
|
||||
tracing::info!("{}", ctx.author().name);
|
||||
let updated = db::set_members_non_fresher(&ctx.data().db).await?;
|
||||
ctx.say(format!("{updated} updated to non-fresher, removing roles"))
|
||||
.await?;
|
||||
for mut m in ctx.data().server.members(ctx.http(), None, None).await? {
|
||||
let _ = m.remove_role(ctx.http(), ctx.data().fresher).await;
|
||||
}
|
||||
ctx.say("Roles removed").await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Get the number of pending members in the pending table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn count_pending(ctx: ACtx<'_>) -> Result<(), Error> {
|
||||
tracing::info!("{}", ctx.author().name);
|
||||
let count = db::count_pending(&ctx.data().db).await?;
|
||||
ctx.say(format!("There are {count} entries in the pending table"))
|
||||
.await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Delete pending member info by Discord ID
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn delete_pending(ctx: ACtx<'_>, id: serenity::Member) -> Result<(), Error> {
|
||||
tracing::info!("{} {}", ctx.author().name, id.user.name,);
|
||||
if db::delete_pending_by_id(&ctx.data().db, id.user.id.into()).await? {
|
||||
ctx.say(format!("Successfully deleted pending member info for {id}"))
|
||||
.await?
|
||||
} else {
|
||||
ctx.say(format!("Failed to delete pending member info for {id}"))
|
||||
.await?
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Print all pending members in pending table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn get_all_pending(ctx: ACtx<'_>) -> Result<(), Error> {
|
||||
#[derive(Modal)]
|
||||
struct ConfirmPending {
|
||||
#[name = "This will output the pending db as text"]
|
||||
#[placeholder = "yes"]
|
||||
confirm: String,
|
||||
}
|
||||
|
||||
tracing::info!("{}", ctx.author().name);
|
||||
|
||||
if let Some(ConfirmPending { confirm }) = ConfirmPending::execute(ctx).await? {
|
||||
if confirm.to_lowercase().contains("yes") {
|
||||
let pending = db::get_all_pending(&ctx.data().db).await?;
|
||||
match tokio::fs::write("pending.rs", format!("{pending:#?}")).await {
|
||||
Ok(()) => {
|
||||
ctx.say("Sending pending db data in followup message")
|
||||
.await?;
|
||||
ctx.channel_id()
|
||||
.send_files(&ctx.http(), vec!["pending.rs"], |cm| {
|
||||
cm.content("File: pending db")
|
||||
})
|
||||
.await?;
|
||||
}
|
||||
Err(e) => {
|
||||
tracing::error!("{e}");
|
||||
ctx.say("Failed to create pending db file").await?;
|
||||
}
|
||||
}
|
||||
let _ = tokio::fs::remove_file("pending.rs").await;
|
||||
Ok(())
|
||||
} else {
|
||||
ctx.say("Skipping pending db output").await?;
|
||||
Ok(())
|
||||
}
|
||||
} else {
|
||||
ctx.say("Timed out").await?;
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
/// Get pending member info by Discord ID
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn get_pending(ctx: ACtx<'_>, id: serenity::Member) -> Result<(), Error> {
|
||||
tracing::info!("{} {}", ctx.author().name, id.user.name);
|
||||
match db::get_pending_by_id(&ctx.data().db, id.user.id.into()).await? {
|
||||
Some(p) => {
|
||||
ctx.say(format!("Pending info for {id}:\n```rust\n{p:#?}\n```"))
|
||||
.await?
|
||||
}
|
||||
None => ctx.say(format!("No pending entry found for {id}")).await?,
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Manually add pending member to pending table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn add_pending(
|
||||
ctx: ACtx<'_>,
|
||||
id: serenity::Member,
|
||||
shortcode: String,
|
||||
realname: String,
|
||||
) -> Result<(), Error> {
|
||||
tracing::info!(
|
||||
"{} {}, {shortcode}, {realname}",
|
||||
ctx.author().name,
|
||||
id.user.name,
|
||||
);
|
||||
db::insert_pending(
|
||||
&ctx.data().db,
|
||||
PendingMember {
|
||||
discord_id: id.user.id.into(),
|
||||
shortcode,
|
||||
realname,
|
||||
},
|
||||
)
|
||||
.await?;
|
||||
ctx.say(format!("Pending member added: {id}")).await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Delete all pending members in pending table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn delete_all_pending(ctx: ACtx<'_>) -> Result<(), Error> {
|
||||
#[derive(Modal)]
|
||||
struct ConfirmPurgePending {
|
||||
#[name = "This will wipe the pending db"]
|
||||
#[placeholder = "yes"]
|
||||
confirm: String,
|
||||
}
|
||||
|
||||
tracing::info!("{}", ctx.author().name);
|
||||
|
||||
if let Some(ConfirmPurgePending { confirm }) = ConfirmPurgePending::execute(ctx).await? {
|
||||
if confirm.to_lowercase().contains("yes") {
|
||||
let deleted = db::delete_all_pending(&ctx.data().db).await?;
|
||||
ctx.say(format!("Deleted {deleted} entries from the pending db"))
|
||||
.await?;
|
||||
Ok(())
|
||||
} else {
|
||||
ctx.say("Skipping pending db purge").await?;
|
||||
Ok(())
|
||||
}
|
||||
} else {
|
||||
ctx.say("Timed out").await?;
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
/// Get the number of manual members in the manual table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn count_manual(ctx: ACtx<'_>) -> Result<(), Error> {
|
||||
tracing::info!("{}", ctx.author().name);
|
||||
let count = db::count_manual(&ctx.data().db).await?;
|
||||
ctx.say(format!("There are {count} entries in the manual table"))
|
||||
.await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Delete manual member info by Discord ID
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn delete_manual(ctx: ACtx<'_>, id: serenity::Member) -> Result<(), Error> {
|
||||
tracing::info!("{} {}", ctx.author().name, id.user.name,);
|
||||
if db::delete_manual_by_id(&ctx.data().db, id.user.id.into()).await? {
|
||||
ctx.say(format!("Successfully deleted manual member info for {id}"))
|
||||
.await?
|
||||
} else {
|
||||
ctx.say(format!("Failed to delete manual member info for {id}"))
|
||||
.await?
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Print all manual members in manual table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn get_all_manual(ctx: ACtx<'_>) -> Result<(), Error> {
|
||||
#[derive(Modal)]
|
||||
struct ConfirmManual {
|
||||
#[name = "This will output the manual db as text"]
|
||||
#[placeholder = "yes"]
|
||||
confirm: String,
|
||||
}
|
||||
|
||||
tracing::info!("{}", ctx.author().name);
|
||||
|
||||
if let Some(ConfirmManual { confirm }) = ConfirmManual::execute(ctx).await? {
|
||||
if confirm.to_lowercase().contains("yes") {
|
||||
let manual = db::get_all_manual(&ctx.data().db).await?;
|
||||
match tokio::fs::write("manual.rs", format!("{manual:#?}")).await {
|
||||
Ok(()) => {
|
||||
ctx.say("Sending manual db data in followup message")
|
||||
.await?;
|
||||
ctx.channel_id()
|
||||
.send_files(&ctx.http(), vec!["manual.rs"], |cm| {
|
||||
cm.content("File: manual db")
|
||||
})
|
||||
.await?;
|
||||
}
|
||||
Err(e) => {
|
||||
tracing::error!("{e}");
|
||||
ctx.say("Failed to create manual db file").await?;
|
||||
}
|
||||
}
|
||||
let _ = tokio::fs::remove_file("manual.rs").await;
|
||||
Ok(())
|
||||
} else {
|
||||
ctx.say("Skipping manual db output").await?;
|
||||
Ok(())
|
||||
}
|
||||
} else {
|
||||
ctx.say("Timed out").await?;
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
/// Get manual member info by Discord ID
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn get_manual(ctx: ACtx<'_>, id: serenity::Member) -> Result<(), Error> {
|
||||
tracing::info!("{} {}", ctx.author().name, id.user.name);
|
||||
match db::get_manual_by_id(&ctx.data().db, id.user.id.into()).await? {
|
||||
Some(m) => {
|
||||
ctx.say(format!("Manual info for {id}:\n```rust\n{m:#?}\n```"))
|
||||
.await?
|
||||
}
|
||||
None => ctx.say(format!("No manual entry found for {id}")).await?,
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Manually add manual member to manual table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn add_manual(
|
||||
ctx: ACtx<'_>,
|
||||
id: serenity::Member,
|
||||
shortcode: String,
|
||||
nickname: String,
|
||||
realname: String,
|
||||
fresher: bool,
|
||||
) -> Result<(), Error> {
|
||||
tracing::info!(
|
||||
"{} {}, {shortcode}, {realname}, {nickname}",
|
||||
ctx.author().name,
|
||||
id.user.name,
|
||||
);
|
||||
db::insert_manual(
|
||||
&ctx.data().db,
|
||||
ManualMember {
|
||||
discord_id: id.user.id.into(),
|
||||
shortcode,
|
||||
nickname,
|
||||
realname,
|
||||
fresher,
|
||||
},
|
||||
)
|
||||
.await?;
|
||||
ctx.say(format!("Manual member added: {id}")).await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Delete all manual members in manual table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn delete_all_manual(ctx: ACtx<'_>) -> Result<(), Error> {
|
||||
#[derive(Modal)]
|
||||
struct ConfirmPurgeManual {
|
||||
#[name = "This will wipe the manual db"]
|
||||
#[placeholder = "yes"]
|
||||
confirm: String,
|
||||
}
|
||||
|
||||
tracing::info!("{}", ctx.author().name);
|
||||
|
||||
if let Some(ConfirmPurgeManual { confirm }) = ConfirmPurgeManual::execute(ctx).await? {
|
||||
if confirm.to_lowercase().contains("yes") {
|
||||
let deleted = db::delete_all_manual(&ctx.data().db).await?;
|
||||
ctx.say(format!("Deleted {deleted} entries from the manual db"))
|
||||
.await?;
|
||||
Ok(())
|
||||
} else {
|
||||
ctx.say("Skipping manual db purge").await?;
|
||||
Ok(())
|
||||
}
|
||||
} else {
|
||||
ctx.say("Timed out").await?;
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
/// Unreachable, used to create whois command folder
|
||||
#[poise::command(
|
||||
slash_command,
|
||||
subcommands("whois_by_id", "whois_by_nickname", "whois_by_realname")
|
||||
)]
|
||||
pub(crate) async fn whois(_ctx: ACtx<'_>) -> Result<(), Error> {
|
||||
unreachable!()
|
||||
}
|
||||
|
||||
/// (Public) Get member info by Discord ID
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command, rename = "id")]
|
||||
pub(crate) async fn whois_by_id(ctx: ACtx<'_>, id: serenity::Member) -> Result<(), Error> {
|
||||
tracing::info!("{} {}", ctx.author().name, id.user.name);
|
||||
match db::get_member_by_id(&ctx.data().db, id.user.id.into()).await? {
|
||||
Some(m) => {
|
||||
ctx.send(|c| c.content(format!("{id}: {}", m.nickname)).ephemeral(true))
|
||||
.await?
|
||||
}
|
||||
None => {
|
||||
ctx.send(|c| {
|
||||
c.content(format!("No member entry found for {id}"))
|
||||
.ephemeral(true)
|
||||
})
|
||||
.await?
|
||||
}
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// (Public) Get member info by Nickname (Exact)
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command, rename = "nick")]
|
||||
pub(crate) async fn whois_by_nickname(ctx: ACtx<'_>, nickname: String) -> Result<(), Error> {
|
||||
tracing::info!("{} {nickname}", ctx.author().name);
|
||||
if let Some(m) = db::get_member_by_nickname(&ctx.data().db, &nickname).await? {
|
||||
ctx.send(|c| {
|
||||
c.content(format!("{nickname}: <@{}>", m.discord_id))
|
||||
.ephemeral(true)
|
||||
})
|
||||
.await?;
|
||||
} else {
|
||||
let members = db::get_member_by_nickname_fuzzy(&ctx.data().db, &nickname, 3).await?;
|
||||
if members.is_empty() {
|
||||
ctx.send(|c| {
|
||||
c.content(format!("No member entry found for nickname {nickname}"))
|
||||
.ephemeral(true)
|
||||
})
|
||||
.await?;
|
||||
} else {
|
||||
ctx.send(|c| {
|
||||
c.ephemeral(true).content(format!(
|
||||
"Possible matches for {nickname}: {}",
|
||||
members
|
||||
.iter()
|
||||
.map(|m| format!(" <@{}>", m.discord_id))
|
||||
.collect::<String>()
|
||||
))
|
||||
})
|
||||
.await?;
|
||||
}
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// (Public) Get member info by Real Name (Exact)
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command, rename = "name")]
|
||||
pub(crate) async fn whois_by_realname(ctx: ACtx<'_>, realname: String) -> Result<(), Error> {
|
||||
tracing::info!("{} {realname}", ctx.author().name);
|
||||
if let Some(m) = db::get_member_by_realname(&ctx.data().db, &realname).await? {
|
||||
ctx.send(|c| {
|
||||
c.content(format!("{realname}: <@{}>", m.discord_id))
|
||||
.ephemeral(true)
|
||||
})
|
||||
.await?;
|
||||
} else {
|
||||
let members = db::get_member_by_realname_fuzzy(&ctx.data().db, &realname, 3).await?;
|
||||
if members.is_empty() {
|
||||
ctx.send(|c| {
|
||||
c.content(format!("No member entry found for realname {realname}"))
|
||||
.ephemeral(true)
|
||||
})
|
||||
.await?;
|
||||
} else {
|
||||
ctx.send(|c| {
|
||||
c.ephemeral(true).content(format!(
|
||||
"Possible matches for {realname}: {}",
|
||||
members
|
||||
.iter()
|
||||
.map(|m| format!(" <@{}>", m.discord_id))
|
||||
.collect::<String>()
|
||||
))
|
||||
})
|
||||
.await?;
|
||||
}
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
103
src/cmds/edit_members.rs
Normal file
103
src/cmds/edit_members.rs
Normal file
|
|
@ -0,0 +1,103 @@
|
|||
use crate::{db, ACtx, Error};
|
||||
use poise::serenity_prelude as serenity;
|
||||
|
||||
/// Unreachable, used to create edit_member command folder
|
||||
#[poise::command(
|
||||
slash_command,
|
||||
subcommands(
|
||||
"edit_member_shortcode",
|
||||
"edit_member_nickname",
|
||||
"edit_member_realname",
|
||||
"edit_member_fresher",
|
||||
)
|
||||
)]
|
||||
pub(crate) async fn edit_member(_ctx: ACtx<'_>) -> Result<(), Error> {
|
||||
unreachable!()
|
||||
}
|
||||
|
||||
/// Edit member Shortcode
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command, rename = "shortcode")]
|
||||
pub(crate) async fn edit_member_shortcode(
|
||||
ctx: ACtx<'_>,
|
||||
id: serenity::Member,
|
||||
shortcode: String,
|
||||
) -> Result<(), Error> {
|
||||
tracing::info!("{} {shortcode}", ctx.author().name);
|
||||
if db::edit_member_shortcode(&ctx.data().db, id.user.id.into(), &shortcode).await? {
|
||||
ctx.say(format!("{id} Shortcode updated to {shortcode}"))
|
||||
.await?;
|
||||
} else {
|
||||
ctx.say(format!("Failed to update Shortcode for {id}"))
|
||||
.await?;
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Edit member Nickname
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command, rename = "nick")]
|
||||
pub(crate) async fn edit_member_nickname(
|
||||
ctx: ACtx<'_>,
|
||||
id: serenity::Member,
|
||||
nickname: String,
|
||||
) -> Result<(), Error> {
|
||||
tracing::info!("{} {nickname}", ctx.author().name);
|
||||
if db::edit_member_nickname(&ctx.data().db, id.user.id.into(), &nickname).await? {
|
||||
ctx.say(format!("{id} Nick updated to {nickname}")).await?;
|
||||
} else {
|
||||
ctx.say(format!("Failed to update Nick for {id}")).await?;
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Edit member Real Name
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command, rename = "name")]
|
||||
pub(crate) async fn edit_member_realname(
|
||||
ctx: ACtx<'_>,
|
||||
id: serenity::Member,
|
||||
realname: String,
|
||||
) -> Result<(), Error> {
|
||||
tracing::info!("{} {realname}", ctx.author().name);
|
||||
if db::edit_member_realname(&ctx.data().db, id.user.id.into(), &realname).await? {
|
||||
ctx.say(format!("{id} Name updated to {realname}")).await?;
|
||||
} else {
|
||||
ctx.say(format!("Failed to update Name for {id}")).await?;
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Edit member fresher status
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command, rename = "fresher")]
|
||||
pub(crate) async fn edit_member_fresher(
|
||||
ctx: ACtx<'_>,
|
||||
id: serenity::Member,
|
||||
fresher: bool,
|
||||
) -> Result<(), Error> {
|
||||
tracing::info!("{} {} {fresher}", ctx.author().name, id.user.name,);
|
||||
if db::edit_member_fresher(&ctx.data().db, id.user.id.into(), fresher).await? {
|
||||
ctx.say(format!("{id} Fresher status updated to {fresher}"))
|
||||
.await?;
|
||||
} else {
|
||||
ctx.say(format!("Failed to update Fresher status for {id}"))
|
||||
.await?;
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Set all members to non-freshers
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn set_members_non_fresher(ctx: ACtx<'_>) -> Result<(), Error> {
|
||||
tracing::info!("{}", ctx.author().name);
|
||||
let updated = db::set_members_non_fresher(&ctx.data().db).await?;
|
||||
ctx.say(format!("{updated} updated to non-fresher, removing roles"))
|
||||
.await?;
|
||||
for mut m in ctx.data().server.members(ctx.http(), None, None).await? {
|
||||
let _ = m.remove_role(ctx.http(), ctx.data().fresher).await;
|
||||
}
|
||||
ctx.say("Roles removed").await?;
|
||||
Ok(())
|
||||
}
|
||||
147
src/cmds/manual.rs
Normal file
147
src/cmds/manual.rs
Normal file
|
|
@ -0,0 +1,147 @@
|
|||
use crate::{db, ACtx, Error, ManualMember};
|
||||
use poise::serenity_prelude as serenity;
|
||||
use poise::Modal;
|
||||
|
||||
/// Get the number of manual members in the manual table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn count_manual(ctx: ACtx<'_>) -> Result<(), Error> {
|
||||
tracing::info!("{}", ctx.author().name);
|
||||
let count = db::count_manual(&ctx.data().db).await?;
|
||||
ctx.say(format!("There are {count} entries in the manual table"))
|
||||
.await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Delete manual member info by Discord ID
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn delete_manual(ctx: ACtx<'_>, id: serenity::Member) -> Result<(), Error> {
|
||||
tracing::info!("{} {}", ctx.author().name, id.user.name,);
|
||||
if db::delete_manual_by_id(&ctx.data().db, id.user.id.into()).await? {
|
||||
ctx.say(format!("Successfully deleted manual member info for {id}"))
|
||||
.await?
|
||||
} else {
|
||||
ctx.say(format!("Failed to delete manual member info for {id}"))
|
||||
.await?
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Print all manual members in manual table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn get_all_manual(ctx: ACtx<'_>) -> Result<(), Error> {
|
||||
#[derive(Modal)]
|
||||
struct ConfirmManual {
|
||||
#[name = "This will output the manual db as text"]
|
||||
#[placeholder = "yes"]
|
||||
confirm: String,
|
||||
}
|
||||
|
||||
tracing::info!("{}", ctx.author().name);
|
||||
|
||||
if let Some(ConfirmManual { confirm }) = ConfirmManual::execute(ctx).await? {
|
||||
if confirm.to_lowercase().contains("yes") {
|
||||
let manual = db::get_all_manual(&ctx.data().db).await?;
|
||||
match tokio::fs::write("manual.rs", format!("{manual:#?}")).await {
|
||||
Ok(()) => {
|
||||
ctx.say("Sending manual db data in followup message")
|
||||
.await?;
|
||||
ctx.channel_id()
|
||||
.send_files(&ctx.http(), vec!["manual.rs"], |cm| {
|
||||
cm.content("File: manual db")
|
||||
})
|
||||
.await?;
|
||||
}
|
||||
Err(e) => {
|
||||
tracing::error!("{e}");
|
||||
ctx.say("Failed to create manual db file").await?;
|
||||
}
|
||||
}
|
||||
let _ = tokio::fs::remove_file("manual.rs").await;
|
||||
Ok(())
|
||||
} else {
|
||||
ctx.say("Skipping manual db output").await?;
|
||||
Ok(())
|
||||
}
|
||||
} else {
|
||||
ctx.say("Timed out").await?;
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
/// Get manual member info by Discord ID
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn get_manual(ctx: ACtx<'_>, id: serenity::Member) -> Result<(), Error> {
|
||||
tracing::info!("{} {}", ctx.author().name, id.user.name);
|
||||
match db::get_manual_by_id(&ctx.data().db, id.user.id.into()).await? {
|
||||
Some(m) => {
|
||||
ctx.say(format!("Manual info for {id}:\n```rust\n{m:#?}\n```"))
|
||||
.await?
|
||||
}
|
||||
None => ctx.say(format!("No manual entry found for {id}")).await?,
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Manually add manual member to manual table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn add_manual(
|
||||
ctx: ACtx<'_>,
|
||||
id: serenity::Member,
|
||||
shortcode: String,
|
||||
nickname: String,
|
||||
realname: String,
|
||||
fresher: bool,
|
||||
) -> Result<(), Error> {
|
||||
tracing::info!(
|
||||
"{} {}, {shortcode}, {realname}, {nickname}",
|
||||
ctx.author().name,
|
||||
id.user.name,
|
||||
);
|
||||
db::insert_manual(
|
||||
&ctx.data().db,
|
||||
ManualMember {
|
||||
discord_id: id.user.id.into(),
|
||||
shortcode,
|
||||
nickname,
|
||||
realname,
|
||||
fresher,
|
||||
},
|
||||
)
|
||||
.await?;
|
||||
ctx.say(format!("Manual member added: {id}")).await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Delete all manual members in manual table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn delete_all_manual(ctx: ACtx<'_>) -> Result<(), Error> {
|
||||
#[derive(Modal)]
|
||||
struct ConfirmPurgeManual {
|
||||
#[name = "This will wipe the manual db"]
|
||||
#[placeholder = "yes"]
|
||||
confirm: String,
|
||||
}
|
||||
|
||||
tracing::info!("{}", ctx.author().name);
|
||||
|
||||
if let Some(ConfirmPurgeManual { confirm }) = ConfirmPurgeManual::execute(ctx).await? {
|
||||
if confirm.to_lowercase().contains("yes") {
|
||||
let deleted = db::delete_all_manual(&ctx.data().db).await?;
|
||||
ctx.say(format!("Deleted {deleted} entries from the manual db"))
|
||||
.await?;
|
||||
Ok(())
|
||||
} else {
|
||||
ctx.say("Skipping manual db purge").await?;
|
||||
Ok(())
|
||||
}
|
||||
} else {
|
||||
ctx.say("Timed out").await?;
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
246
src/cmds/members.rs
Normal file
246
src/cmds/members.rs
Normal file
|
|
@ -0,0 +1,246 @@
|
|||
use crate::{db, ACtx, Error, Member};
|
||||
use poise::serenity_prelude as serenity;
|
||||
use poise::Modal;
|
||||
|
||||
/// Get the number of members in the members table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn count_members(ctx: ACtx<'_>) -> Result<(), Error> {
|
||||
tracing::info!("{}", ctx.author().name);
|
||||
let count = db::count_members(&ctx.data().db).await?;
|
||||
ctx.say(format!("There are {count} entries in the members table"))
|
||||
.await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Delete member info by Discord ID
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn delete_member(
|
||||
ctx: ACtx<'_>,
|
||||
id: serenity::Member,
|
||||
remove_roles: Option<bool>,
|
||||
) -> Result<(), Error> {
|
||||
tracing::info!("{} {}", ctx.author().name, id.user.name,);
|
||||
if db::delete_member_by_id(&ctx.data().db, id.user.id.into()).await? {
|
||||
if remove_roles.unwrap_or(true) {
|
||||
let mut m = id.clone();
|
||||
crate::verify::remove_role(ctx.serenity_context(), &mut m, ctx.data().member).await?;
|
||||
crate::verify::remove_role(ctx.serenity_context(), &mut m, ctx.data().fresher).await?;
|
||||
}
|
||||
ctx.say(format!("Successfully deleted member info for {id}"))
|
||||
.await?
|
||||
} else {
|
||||
ctx.say(format!("Failed to delete member info for {id}"))
|
||||
.await?
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Print all members in members table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn get_all_members(ctx: ACtx<'_>) -> Result<(), Error> {
|
||||
#[derive(Modal)]
|
||||
struct Confirm {
|
||||
#[name = "This will output the members db as text"]
|
||||
#[placeholder = "yes"]
|
||||
confirm: String,
|
||||
}
|
||||
|
||||
tracing::info!("{}", ctx.author().name);
|
||||
|
||||
if let Some(Confirm { confirm }) = Confirm::execute(ctx).await? {
|
||||
if confirm.to_lowercase().contains("yes") {
|
||||
let members = db::get_all_members(&ctx.data().db).await?;
|
||||
match tokio::fs::write("members.rs", format!("{members:#?}")).await {
|
||||
Ok(()) => {
|
||||
ctx.say("Sending members db data in followup message")
|
||||
.await?;
|
||||
ctx.channel_id()
|
||||
.send_files(&ctx.http(), vec!["members.rs"], |cm| {
|
||||
cm.content("File: members db")
|
||||
})
|
||||
.await?;
|
||||
}
|
||||
Err(e) => {
|
||||
tracing::error!("{e}");
|
||||
ctx.say("Failed to create members db file").await?;
|
||||
}
|
||||
}
|
||||
let _ = tokio::fs::remove_file("members.rs").await;
|
||||
Ok(())
|
||||
} else {
|
||||
ctx.say("Skipping members db output").await?;
|
||||
Ok(())
|
||||
}
|
||||
} else {
|
||||
ctx.say("Timed out").await?;
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
/// Unreachable, used to create get_member command folder
|
||||
#[poise::command(
|
||||
slash_command,
|
||||
subcommands(
|
||||
"get_member_by_id",
|
||||
"get_member_by_shortcode",
|
||||
"get_member_by_nickname",
|
||||
"get_member_by_realname",
|
||||
)
|
||||
)]
|
||||
pub(crate) async fn get_member(_ctx: ACtx<'_>) -> Result<(), Error> {
|
||||
unreachable!()
|
||||
}
|
||||
|
||||
/// Get member info by Discord ID
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command, rename = "id")]
|
||||
pub(crate) async fn get_member_by_id(ctx: ACtx<'_>, id: serenity::Member) -> Result<(), Error> {
|
||||
tracing::info!("{} {}", ctx.author().name, id.user.name,);
|
||||
match db::get_member_by_id(&ctx.data().db, id.user.id.into()).await? {
|
||||
Some(m) => {
|
||||
ctx.say(format!("Member info for {id}:\n```rust\n{m:#?}\n```"))
|
||||
.await?
|
||||
}
|
||||
None => ctx.say(format!("No member entry found for {id}")).await?,
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Get member info by Shortcode
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command, rename = "shortcode")]
|
||||
pub(crate) async fn get_member_by_shortcode(ctx: ACtx<'_>, shortcode: String) -> Result<(), Error> {
|
||||
tracing::info!("{} {shortcode}", ctx.author().name);
|
||||
match db::get_member_by_shortcode(&ctx.data().db, &shortcode).await? {
|
||||
Some(m) => {
|
||||
ctx.say(format!(
|
||||
"Member info for shortcode {shortcode}:\n```rust\n{m:#?}\n```"
|
||||
))
|
||||
.await?
|
||||
}
|
||||
None => {
|
||||
ctx.say(format!("No entry found for shortcode {shortcode}"))
|
||||
.await?
|
||||
}
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Get member info by Nickname
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command, rename = "nick")]
|
||||
pub(crate) async fn get_member_by_nickname(ctx: ACtx<'_>, nickname: String) -> Result<(), Error> {
|
||||
tracing::info!("{} {nickname}", ctx.author().name);
|
||||
match db::get_member_by_nickname(&ctx.data().db, &nickname).await? {
|
||||
Some(m) => {
|
||||
ctx.say(format!(
|
||||
"Member info for nickname {nickname}:\n```rust\n{m:#?}\n```"
|
||||
))
|
||||
.await?
|
||||
}
|
||||
None => {
|
||||
ctx.say(format!("No entry found for nickname {nickname}",))
|
||||
.await?
|
||||
}
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Get member info by Real Name
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command, rename = "name")]
|
||||
pub(crate) async fn get_member_by_realname(ctx: ACtx<'_>, realname: String) -> Result<(), Error> {
|
||||
tracing::info!("{} {realname}", ctx.author().name);
|
||||
match db::get_member_by_realname(&ctx.data().db, &realname).await? {
|
||||
Some(m) => {
|
||||
ctx.say(format!(
|
||||
"Member info for realname {realname}:\n```rust\n{m:#?}\n```"
|
||||
))
|
||||
.await?
|
||||
}
|
||||
None => {
|
||||
ctx.say(format!("No entry found for realname {realname}",))
|
||||
.await?
|
||||
}
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Add a member to the members table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn add_member(
|
||||
ctx: ACtx<'_>,
|
||||
id: serenity::Member,
|
||||
shortcode: String,
|
||||
nickname: String,
|
||||
realname: String,
|
||||
fresher: bool,
|
||||
) -> Result<(), Error> {
|
||||
tracing::info!(
|
||||
"{} {}, {shortcode}, {realname}, {nickname}",
|
||||
ctx.author().name,
|
||||
id.user.name,
|
||||
);
|
||||
db::insert_member(
|
||||
&ctx.data().db,
|
||||
Member {
|
||||
discord_id: id.user.id.into(),
|
||||
shortcode,
|
||||
nickname,
|
||||
realname,
|
||||
fresher,
|
||||
},
|
||||
)
|
||||
.await?;
|
||||
let mut m = id.clone();
|
||||
crate::verify::apply_role(ctx.serenity_context(), &mut m, ctx.data().member).await?;
|
||||
if fresher {
|
||||
crate::verify::apply_role(ctx.serenity_context(), &mut m, ctx.data().fresher).await?;
|
||||
}
|
||||
ctx.say(format!("Member added: {id}")).await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Manually add member to members table from pending table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn insert_member_from_pending(
|
||||
ctx: ACtx<'_>,
|
||||
id: serenity::Member,
|
||||
nickname: String,
|
||||
fresher: bool,
|
||||
) -> Result<(), Error> {
|
||||
tracing::info!("{} {}", ctx.author().name, id.user.name,);
|
||||
match db::insert_member_from_pending(&ctx.data().db, id.user.id.into(), &nickname, fresher)
|
||||
.await
|
||||
{
|
||||
Ok(()) => {
|
||||
ctx.say(format!("Member moved from pending to members table: {id}"))
|
||||
.await?
|
||||
}
|
||||
Err(e) => ctx.say(format!("Error: {e}")).await?,
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Manually add member to members table from manual table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn insert_member_from_manual(
|
||||
ctx: ACtx<'_>,
|
||||
id: serenity::Member,
|
||||
) -> Result<(), Error> {
|
||||
tracing::info!("{} {}", ctx.author().name, id.user.name,);
|
||||
match db::insert_member_from_manual(&ctx.data().db, id.user.id.into()).await {
|
||||
Ok(()) => {
|
||||
ctx.say(format!("Member moved from manual to members table: {id}"))
|
||||
.await?
|
||||
}
|
||||
Err(e) => ctx.say(format!("Error: {e}")).await?,
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
90
src/cmds/mod.rs
Normal file
90
src/cmds/mod.rs
Normal file
|
|
@ -0,0 +1,90 @@
|
|||
use crate::{ACtx, Data, Error};
|
||||
use poise::serenity_prelude as serenity;
|
||||
use poise::Modal;
|
||||
|
||||
pub(crate) mod members;
|
||||
pub(crate) use members::*;
|
||||
|
||||
pub(crate) mod pending;
|
||||
pub(crate) use pending::*;
|
||||
|
||||
pub(crate) mod manual;
|
||||
pub(crate) use manual::*;
|
||||
|
||||
pub(crate) mod edit_members;
|
||||
pub(crate) use edit_members::*;
|
||||
|
||||
pub(crate) mod whois;
|
||||
pub(crate) use whois::*;
|
||||
|
||||
/// Buttons to (de-)register application commands globally or by guild
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(prefix_command, owners_only)]
|
||||
pub(crate) async fn cmds(ctx: poise::Context<'_, Data, Error>) -> Result<(), Error> {
|
||||
tracing::info!("{}", ctx.author().name);
|
||||
poise::builtins::register_application_commands_buttons(ctx).await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Send (customisable) verification introduction message in specified channel
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn setup(
|
||||
ctx: ACtx<'_>,
|
||||
#[description = "Channel to send verification introduction message in"]
|
||||
#[channel_types("Text", "News")]
|
||||
channel: serenity::GuildChannel,
|
||||
) -> Result<(), Error> {
|
||||
#[derive(Modal)]
|
||||
struct Setup {
|
||||
#[name = "Contents of the verification intro message"]
|
||||
#[placeholder = "Welcome... Click the \"Begin\" button below to start verification"]
|
||||
#[paragraph]
|
||||
#[max_length = 500]
|
||||
message: String,
|
||||
#[name = "Emoji for the start button"]
|
||||
#[placeholder = "🚀"]
|
||||
#[max_length = 4]
|
||||
emoji: Option<String>,
|
||||
#[name = "Label for the start button"]
|
||||
#[placeholder = "Begin"]
|
||||
#[max_length = 80]
|
||||
text: Option<String>,
|
||||
}
|
||||
|
||||
tracing::info!("{} {}", ctx.author().name, channel.name());
|
||||
|
||||
if let Some(Setup {
|
||||
message,
|
||||
emoji,
|
||||
text,
|
||||
}) = Setup::execute(ctx).await?
|
||||
{
|
||||
ctx.say(format!("Sending intro message in {channel}"))
|
||||
.await?;
|
||||
let emoji = emoji.unwrap_or_default().chars().next().unwrap_or('🚀');
|
||||
channel
|
||||
.send_message(ctx.http(), |m| {
|
||||
m.content(message).components(|c| {
|
||||
c.create_action_row(|a| {
|
||||
a.create_button(|b| {
|
||||
b.style(serenity::ButtonStyle::Secondary)
|
||||
.emoji('📖')
|
||||
.label("More info")
|
||||
.custom_id("info")
|
||||
})
|
||||
.create_button(|b| {
|
||||
b.style(serenity::ButtonStyle::Primary)
|
||||
.emoji(emoji)
|
||||
.label(text.unwrap_or("Begin".to_string()))
|
||||
.custom_id("start")
|
||||
})
|
||||
})
|
||||
})
|
||||
})
|
||||
.await?;
|
||||
} else {
|
||||
ctx.say("Modal timed out, try again...").await?;
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
143
src/cmds/pending.rs
Normal file
143
src/cmds/pending.rs
Normal file
|
|
@ -0,0 +1,143 @@
|
|||
use crate::{db, ACtx, Error, PendingMember};
|
||||
use poise::serenity_prelude as serenity;
|
||||
use poise::Modal;
|
||||
|
||||
/// Get the number of pending members in the pending table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn count_pending(ctx: ACtx<'_>) -> Result<(), Error> {
|
||||
tracing::info!("{}", ctx.author().name);
|
||||
let count = db::count_pending(&ctx.data().db).await?;
|
||||
ctx.say(format!("There are {count} entries in the pending table"))
|
||||
.await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Delete pending member info by Discord ID
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn delete_pending(ctx: ACtx<'_>, id: serenity::Member) -> Result<(), Error> {
|
||||
tracing::info!("{} {}", ctx.author().name, id.user.name,);
|
||||
if db::delete_pending_by_id(&ctx.data().db, id.user.id.into()).await? {
|
||||
ctx.say(format!("Successfully deleted pending member info for {id}"))
|
||||
.await?
|
||||
} else {
|
||||
ctx.say(format!("Failed to delete pending member info for {id}"))
|
||||
.await?
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Print all pending members in pending table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn get_all_pending(ctx: ACtx<'_>) -> Result<(), Error> {
|
||||
#[derive(Modal)]
|
||||
struct ConfirmPending {
|
||||
#[name = "This will output the pending db as text"]
|
||||
#[placeholder = "yes"]
|
||||
confirm: String,
|
||||
}
|
||||
|
||||
tracing::info!("{}", ctx.author().name);
|
||||
|
||||
if let Some(ConfirmPending { confirm }) = ConfirmPending::execute(ctx).await? {
|
||||
if confirm.to_lowercase().contains("yes") {
|
||||
let pending = db::get_all_pending(&ctx.data().db).await?;
|
||||
match tokio::fs::write("pending.rs", format!("{pending:#?}")).await {
|
||||
Ok(()) => {
|
||||
ctx.say("Sending pending db data in followup message")
|
||||
.await?;
|
||||
ctx.channel_id()
|
||||
.send_files(&ctx.http(), vec!["pending.rs"], |cm| {
|
||||
cm.content("File: pending db")
|
||||
})
|
||||
.await?;
|
||||
}
|
||||
Err(e) => {
|
||||
tracing::error!("{e}");
|
||||
ctx.say("Failed to create pending db file").await?;
|
||||
}
|
||||
}
|
||||
let _ = tokio::fs::remove_file("pending.rs").await;
|
||||
Ok(())
|
||||
} else {
|
||||
ctx.say("Skipping pending db output").await?;
|
||||
Ok(())
|
||||
}
|
||||
} else {
|
||||
ctx.say("Timed out").await?;
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
/// Get pending member info by Discord ID
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn get_pending(ctx: ACtx<'_>, id: serenity::Member) -> Result<(), Error> {
|
||||
tracing::info!("{} {}", ctx.author().name, id.user.name);
|
||||
match db::get_pending_by_id(&ctx.data().db, id.user.id.into()).await? {
|
||||
Some(p) => {
|
||||
ctx.say(format!("Pending info for {id}:\n```rust\n{p:#?}\n```"))
|
||||
.await?
|
||||
}
|
||||
None => ctx.say(format!("No pending entry found for {id}")).await?,
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Manually add pending member to pending table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn add_pending(
|
||||
ctx: ACtx<'_>,
|
||||
id: serenity::Member,
|
||||
shortcode: String,
|
||||
realname: String,
|
||||
) -> Result<(), Error> {
|
||||
tracing::info!(
|
||||
"{} {}, {shortcode}, {realname}",
|
||||
ctx.author().name,
|
||||
id.user.name,
|
||||
);
|
||||
db::insert_pending(
|
||||
&ctx.data().db,
|
||||
PendingMember {
|
||||
discord_id: id.user.id.into(),
|
||||
shortcode,
|
||||
realname,
|
||||
},
|
||||
)
|
||||
.await?;
|
||||
ctx.say(format!("Pending member added: {id}")).await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Delete all pending members in pending table
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command)]
|
||||
pub(crate) async fn delete_all_pending(ctx: ACtx<'_>) -> Result<(), Error> {
|
||||
#[derive(Modal)]
|
||||
struct ConfirmPurgePending {
|
||||
#[name = "This will wipe the pending db"]
|
||||
#[placeholder = "yes"]
|
||||
confirm: String,
|
||||
}
|
||||
|
||||
tracing::info!("{}", ctx.author().name);
|
||||
|
||||
if let Some(ConfirmPurgePending { confirm }) = ConfirmPurgePending::execute(ctx).await? {
|
||||
if confirm.to_lowercase().contains("yes") {
|
||||
let deleted = db::delete_all_pending(&ctx.data().db).await?;
|
||||
ctx.say(format!("Deleted {deleted} entries from the pending db"))
|
||||
.await?;
|
||||
Ok(())
|
||||
} else {
|
||||
ctx.say("Skipping pending db purge").await?;
|
||||
Ok(())
|
||||
}
|
||||
} else {
|
||||
ctx.say("Timed out").await?;
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
102
src/cmds/whois.rs
Normal file
102
src/cmds/whois.rs
Normal file
|
|
@ -0,0 +1,102 @@
|
|||
use crate::{db, ACtx, Error};
|
||||
use poise::serenity_prelude as serenity;
|
||||
|
||||
/// Unreachable, used to create whois command folder
|
||||
#[poise::command(
|
||||
slash_command,
|
||||
subcommands("whois_by_id", "whois_by_nickname", "whois_by_realname")
|
||||
)]
|
||||
pub(crate) async fn whois(_ctx: ACtx<'_>) -> Result<(), Error> {
|
||||
unreachable!()
|
||||
}
|
||||
|
||||
/// (Public) Get member info by Discord ID
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command, rename = "id")]
|
||||
pub(crate) async fn whois_by_id(ctx: ACtx<'_>, id: serenity::Member) -> Result<(), Error> {
|
||||
tracing::info!("{} {}", ctx.author().name, id.user.name);
|
||||
match db::get_member_by_id(&ctx.data().db, id.user.id.into()).await? {
|
||||
Some(m) => {
|
||||
ctx.send(|c| c.content(format!("{id}: {}", m.nickname)).ephemeral(true))
|
||||
.await?
|
||||
}
|
||||
None => {
|
||||
ctx.send(|c| {
|
||||
c.content(format!("No member entry found for {id}"))
|
||||
.ephemeral(true)
|
||||
})
|
||||
.await?
|
||||
}
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// (Public) Get member info by Nickname (Exact)
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command, rename = "nick")]
|
||||
pub(crate) async fn whois_by_nickname(ctx: ACtx<'_>, nickname: String) -> Result<(), Error> {
|
||||
tracing::info!("{} {nickname}", ctx.author().name);
|
||||
if let Some(m) = db::get_member_by_nickname(&ctx.data().db, &nickname).await? {
|
||||
ctx.send(|c| {
|
||||
c.content(format!("{nickname}: <@{}>", m.discord_id))
|
||||
.ephemeral(true)
|
||||
})
|
||||
.await?;
|
||||
} else {
|
||||
let members = db::get_member_by_nickname_fuzzy(&ctx.data().db, &nickname, 3).await?;
|
||||
if members.is_empty() {
|
||||
ctx.send(|c| {
|
||||
c.content(format!("No member entry found for nickname {nickname}"))
|
||||
.ephemeral(true)
|
||||
})
|
||||
.await?;
|
||||
} else {
|
||||
ctx.send(|c| {
|
||||
c.ephemeral(true).content(format!(
|
||||
"Possible matches for {nickname}: {}",
|
||||
members
|
||||
.iter()
|
||||
.map(|m| format!(" <@{}>", m.discord_id))
|
||||
.collect::<String>()
|
||||
))
|
||||
})
|
||||
.await?;
|
||||
}
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// (Public) Get member info by Real Name (Exact)
|
||||
#[tracing::instrument(skip_all)]
|
||||
#[poise::command(slash_command, rename = "name")]
|
||||
pub(crate) async fn whois_by_realname(ctx: ACtx<'_>, realname: String) -> Result<(), Error> {
|
||||
tracing::info!("{} {realname}", ctx.author().name);
|
||||
if let Some(m) = db::get_member_by_realname(&ctx.data().db, &realname).await? {
|
||||
ctx.send(|c| {
|
||||
c.content(format!("{realname}: <@{}>", m.discord_id))
|
||||
.ephemeral(true)
|
||||
})
|
||||
.await?;
|
||||
} else {
|
||||
let members = db::get_member_by_realname_fuzzy(&ctx.data().db, &realname, 3).await?;
|
||||
if members.is_empty() {
|
||||
ctx.send(|c| {
|
||||
c.content(format!("No member entry found for realname {realname}"))
|
||||
.ephemeral(true)
|
||||
})
|
||||
.await?;
|
||||
} else {
|
||||
ctx.send(|c| {
|
||||
c.ephemeral(true).content(format!(
|
||||
"Possible matches for {realname}: {}",
|
||||
members
|
||||
.iter()
|
||||
.map(|m| format!(" <@{}>", m.discord_id))
|
||||
.collect::<String>()
|
||||
))
|
||||
})
|
||||
.await?;
|
||||
}
|
||||
};
|
||||
Ok(())
|
||||
}
|
||||
Loading…
Reference in a new issue