feat!: implemented answer buttons
- Also you can /init without restarting bot BREAKING CHANGE: Changed type of Data::discord to Arc<Mutex<DiscordConfig>, removed field pending_answers from DiscordConfig
This commit is contained in:
parent
3f7e6313b0
commit
1ae57ad358
4 changed files with 70 additions and 23 deletions
|
|
@ -1,4 +1,4 @@
|
||||||
use poise::serenity_prelude::{Attachment, CreateAttachment, CreateMessage, Mentionable};
|
use poise::serenity_prelude::{Attachment, ComponentInteractionCollector, CreateActionRow, CreateAttachment, CreateButton, CreateMessage, EditMessage, Mentionable};
|
||||||
|
|
||||||
use crate::{Context, Error};
|
use crate::{Context, Error};
|
||||||
|
|
||||||
|
|
@ -14,7 +14,11 @@ pub async fn answer(
|
||||||
#[description = "Text answer to the quest"]
|
#[description = "Text answer to the quest"]
|
||||||
text: Option<String>,
|
text: Option<String>,
|
||||||
#[description = "Attachment answer to the quest"]
|
#[description = "Attachment answer to the quest"]
|
||||||
files: Vec<Attachment>,
|
file1: Option<Attachment>,
|
||||||
|
#[description = "Attachment answer to the quest"]
|
||||||
|
file2: Option<Attachment>,
|
||||||
|
#[description = "Attachment answer to the quest"]
|
||||||
|
file3: Option<Attachment>,
|
||||||
) -> Result<(), Error> {
|
) -> Result<(), Error> {
|
||||||
let quests = ctx.data().config.load_quests();
|
let quests = ctx.data().config.load_quests();
|
||||||
let Some(quest) = quests.iter().find(|q| q.id == quest_id) else {
|
let Some(quest) = quests.iter().find(|q| q.id == quest_id) else {
|
||||||
|
|
@ -22,6 +26,13 @@ pub async fn answer(
|
||||||
ctx.reply(reply_string).await?;
|
ctx.reply(reply_string).await?;
|
||||||
return Ok(());
|
return Ok(());
|
||||||
};
|
};
|
||||||
|
|
||||||
|
let mut files: Vec<Attachment> = Vec::new();
|
||||||
|
for file in [file1, file2, file3] {
|
||||||
|
if let Some(f) = file {
|
||||||
|
files.push(f);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
if text.is_none() && files.len() == 0 {
|
if text.is_none() && files.len() == 0 {
|
||||||
let reply_string = "Please specify text or at least one attachment.".to_string();
|
let reply_string = "Please specify text or at least one attachment.".to_string();
|
||||||
|
|
@ -38,7 +49,7 @@ pub async fn answer(
|
||||||
"\nPassed answer has attachments.".to_string()
|
"\nPassed answer has attachments.".to_string()
|
||||||
};
|
};
|
||||||
|
|
||||||
let content = format!("# From: {user}\n\
|
let content = format!("## From: {user}\n\
|
||||||
### Quest #{quest_id}: {quest_name}\n\
|
### Quest #{quest_id}: {quest_name}\n\
|
||||||
### Expected answer:\n\
|
### Expected answer:\n\
|
||||||
||{quest_answer}||{text_ans}{attachment_notice}",
|
||{quest_answer}||{text_ans}{attachment_notice}",
|
||||||
|
|
@ -53,16 +64,47 @@ pub async fn answer(
|
||||||
let attachment = CreateAttachment::url(ctx, &file.url).await?;
|
let attachment = CreateAttachment::url(ctx, &file.url).await?;
|
||||||
attachments.push(attachment);
|
attachments.push(attachment);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
let ctx_id = ctx.id();
|
||||||
|
let approve_id = format!("{ctx_id}approve");
|
||||||
|
let reject_id = format!("{ctx_id}reject");
|
||||||
|
|
||||||
let ans_channel = ctx.data().discord.answers_channel;
|
let components = CreateActionRow::Buttons(vec![
|
||||||
let message = CreateMessage::new()
|
CreateButton::new(&approve_id).label("Approve".to_string()),
|
||||||
.content(content)
|
CreateButton::new(&reject_id).label("Reject".to_string()),
|
||||||
.files(attachments);
|
]);
|
||||||
|
|
||||||
ans_channel.send_message(ctx, message).await?;
|
let ans_channel = {
|
||||||
|
let discord = ctx.data().discord.clone();
|
||||||
|
let guard = discord.lock().expect("should not be locked");
|
||||||
|
guard.answers_channel
|
||||||
|
};
|
||||||
|
let builder = CreateMessage::new()
|
||||||
|
.content(content.clone())
|
||||||
|
.files(attachments)
|
||||||
|
.components(vec![components]);
|
||||||
|
|
||||||
|
let mut message = ans_channel.send_message(ctx, builder).await?;
|
||||||
|
|
||||||
let reply_string = "Your answer has been posted.".to_string();
|
let reply_string = "Your answer has been posted.".to_string();
|
||||||
ctx.reply(reply_string).await?;
|
ctx.reply(reply_string).await?;
|
||||||
|
|
||||||
|
while let Some(press) = ComponentInteractionCollector::new(ctx)
|
||||||
|
.filter(move |press| press.data.custom_id.starts_with(&ctx_id.to_string()))
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
let admin = press.user.mention();
|
||||||
|
let is_approved = press.data.custom_id == approve_id;
|
||||||
|
let content = if is_approved {
|
||||||
|
format!("{content}\nApproved by: {admin}")
|
||||||
|
} else {
|
||||||
|
format!("~~{content}~~\nRejected by: {admin}")
|
||||||
|
};
|
||||||
|
|
||||||
|
let builder = EditMessage::new().content(content).components(Vec::new());
|
||||||
|
message.edit(ctx, builder).await?;
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -19,18 +19,22 @@ pub async fn init(
|
||||||
#[description = "Channel to post answers to check"]
|
#[description = "Channel to post answers to check"]
|
||||||
answers_channel: ChannelId,
|
answers_channel: ChannelId,
|
||||||
) -> Result<(), Error> {
|
) -> Result<(), Error> {
|
||||||
let mut dc = ctx.data().discord.clone();
|
let reply_string = {
|
||||||
let guild = ctx.guild_id().unwrap();
|
let dc = ctx.data().discord.clone();
|
||||||
dc.quests_channel = quests_channel;
|
let mut guard = dc.lock().expect("shouldn't be locked");
|
||||||
dc.answers_channel = answers_channel;
|
let guild = ctx.guild_id().unwrap();
|
||||||
dc.guild = guild;
|
guard.quests_channel = quests_channel;
|
||||||
let path = &ctx.data().config.full_impl_path().unwrap();
|
guard.answers_channel = answers_channel;
|
||||||
let reply_string = match dc.save(path.parent().unwrap_or(Path::new("")).to_owned()) {
|
guard.guild = guild;
|
||||||
Ok(_) => "Settings updated, please restart bot to apply changes.".to_string(),
|
|
||||||
Err(error) => {
|
let path = &ctx.data().config.full_impl_path().unwrap();
|
||||||
eprintln!("{error}");
|
match guard.save(path.parent().unwrap_or(Path::new("")).to_owned()) {
|
||||||
ERROR_MSG.to_string()
|
Ok(_) => "Settings updated.".to_string(),
|
||||||
},
|
Err(error) => {
|
||||||
|
eprintln!("{error}");
|
||||||
|
ERROR_MSG.to_string()
|
||||||
|
},
|
||||||
|
}
|
||||||
};
|
};
|
||||||
ctx.reply(reply_string).await?;
|
ctx.reply(reply_string).await?;
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -10,7 +10,6 @@ pub struct DiscordConfig {
|
||||||
pub quests_channel: ChannelId,
|
pub quests_channel: ChannelId,
|
||||||
pub answers_channel: ChannelId,
|
pub answers_channel: ChannelId,
|
||||||
pub quests_messages: Vec<MessageId>,
|
pub quests_messages: Vec<MessageId>,
|
||||||
pub pending_answers: Vec<MessageId>,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub trait ConfigImpl {
|
pub trait ConfigImpl {
|
||||||
|
|
|
||||||
|
|
@ -1,3 +1,5 @@
|
||||||
|
use std::sync::{Arc, Mutex};
|
||||||
|
|
||||||
use clap::Parser;
|
use clap::Parser;
|
||||||
use dotenvy::dotenv;
|
use dotenvy::dotenv;
|
||||||
use poise::serenity_prelude as serenity;
|
use poise::serenity_prelude as serenity;
|
||||||
|
|
@ -11,7 +13,7 @@ mod config;
|
||||||
|
|
||||||
struct Data {
|
struct Data {
|
||||||
pub config: Config,
|
pub config: Config,
|
||||||
pub discord: DiscordConfig,
|
pub discord: Arc<Mutex<DiscordConfig>>,
|
||||||
}
|
}
|
||||||
type Error = Box<dyn std::error::Error + Send + Sync>;
|
type Error = Box<dyn std::error::Error + Send + Sync>;
|
||||||
type Context<'a> = poise::Context<'a, Data, Error>;
|
type Context<'a> = poise::Context<'a, Data, Error>;
|
||||||
|
|
@ -45,7 +47,7 @@ async fn main() {
|
||||||
poise::builtins::register_globally(ctx, &framework.options().commands).await?;
|
poise::builtins::register_globally(ctx, &framework.options().commands).await?;
|
||||||
Ok(Data {
|
Ok(Data {
|
||||||
config,
|
config,
|
||||||
discord,
|
discord: Arc::new(Mutex::new(discord)),
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue