figured out how to add buttons

This commit is contained in:
2025-06-06 00:30:59 +02:00
parent 71da587242
commit 53a1fdee19
9 changed files with 36 additions and 12 deletions
+2
View File
@@ -2,6 +2,8 @@
- [ ] Reddit bot that scrapes images with tag "Original Art" and posts them in Discord server
- [ ] Allow updating the data autonomously and via manual commands.
- [ ] Automatically approve posts that don't get caught by reverse image search (ris)
- [ ] Make buttons do stuff
- [ ] Add buttons to removed post embed
<!-- - [ ] 10-minute schedule for updating Discord channel (IMPOSSIBLE / REALLY FUCKING HARD) -->
### Medium priority:
+1 -1
View File
@@ -18,7 +18,7 @@ pub async fn cmd(
let r = dc_add_server(ctx.data(), ctx.guild_id().unwrap().into()).await;
if r.is_ok() {
send_msg(ctx, lang!("dc_msg_added_to_data"), true, true).await;
send_msg(ctx, lang!("dc_msg_add_to_data"), true, true).await;
}
else {
send_msg(ctx, lang!("dc_msg_corrupted_data"), true, true).await;
+2 -1
View File
@@ -39,7 +39,8 @@ pub async fn cmd(
ephemeral: ephemeral.unwrap_or(false),
message,
thumbnail,
author: if author.unwrap_or(false) { Some(Author { name: ctx.author().name.clone(), url: "".to_string(), icon_url: ctx.author().avatar_url().unwrap() }) } else { None }
author: if author.unwrap_or(false) { Some(Author { name: ctx.author().name.clone(), url: "".to_string(), icon_url: ctx.author().avatar_url().unwrap() }) } else { None },
..Default::default()
},
reply_unwrap
).await;
+1 -1
View File
@@ -91,7 +91,7 @@ async fn make_cmd_vec(data: &Data) -> Vec<Cmd> {
]);
// REDDIT
if cfg["RESTART_enabled"].as_bool().unwrap() {
if cfg["reddit"]["RESTART_enabled"].as_bool().unwrap() {
cmds.extend([
re_cmds::add::cmd(),
re_cmds::approve::cmd(),
+1 -1
View File
@@ -104,7 +104,7 @@ async fn main() {
rs_println!("Fetching language file...");
data::load_lang_data(args.clone().lang);
rs_println!("[IMPORTANT] The below message is a test message, it should be written in the language you've selected\n{}", lang!("log_lang_load_success"));
rs_println!("[IMPORTANT] The below message is a test message, it should be written in the language you've selected\nTest message: {}", lang!("log_lang_load_success"));
let own_env = std::env::var("ASSISTANT_OWNERS").unwrap_or("0".to_string());
let own_vec_str: Vec<String> = own_env.split(",").map(String::from).collect();
+26 -5
View File
@@ -4,7 +4,7 @@ use crate::{lang, Args, Context};
use poise::serenity_prelude::json::Value;
use poise::{serenity_prelude::CreateMessage, CreateReply, ReplyHandle};
use poise::serenity_prelude::{ChannelId, Color, CreateEmbed, CreateEmbedAuthor, EditMessage, Http, Message, Timestamp, UserId};
use poise::serenity_prelude::{ChannelId, Color, CreateActionRow, CreateButton, CreateEmbed, CreateEmbedAuthor, EditMessage, Http, Message, ReactionType, Timestamp, UserId};
use serde_json::json;
@@ -24,7 +24,8 @@ pub struct EmbedOptions {
pub ephemeral: bool,
pub message: Option<String>,
pub author: Option<Author>,
pub thumbnail: Option<String>
pub thumbnail: Option<String>,
pub actionrows: Option<Vec<CreateActionRow>>
}
impl Default for EmbedOptions {
fn default() -> Self {
@@ -37,7 +38,8 @@ impl Default for EmbedOptions {
ephemeral: false,
message: None,
author: None,
thumbnail: None
thumbnail: None,
actionrows: None
};
}
}
@@ -104,6 +106,7 @@ pub async fn send_embed(
embeds: vec![embed],
content: options.message,
ephemeral: Some(options.ephemeral),
components: options.actionrows,
..Default::default()
};
@@ -111,7 +114,12 @@ pub async fn send_embed(
return Some(msg.unwrap());
}
else {
let r = CreateMessage::new().embeds(vec![embed]);
let mut r = CreateMessage::new().embeds(vec![embed]);
if let Some(actionrows) = options.actionrows {
r = r.components(actionrows);
}
let _ = ctx.channel_id().send_message(ctx.http(), r).await;
return None;
}
@@ -126,7 +134,11 @@ pub async fn http_send_embed(
{
let embed = embed_from_options(options.clone());
let r = CreateMessage::new().embeds(vec![embed]);
let mut r = CreateMessage::new().embeds(vec![embed]);
if let Some(actionrows) = options.actionrows {
r = r.components(actionrows);
}
let msg = c_id.send_message(http, r).await;
return msg.ok();
@@ -222,6 +234,14 @@ pub fn make_post_embed(post_data: &Value, url: &str, ephemeral: bool) -> EmbedOp
);
let media_urls = post_data["post_data"]["media_urls"].as_array().unwrap();
let action_row = CreateActionRow::Buttons(vec![
CreateButton::new("upvote_btn") .label("Upvote") .emoji(ReactionType::Unicode("⬆️".to_string())),
CreateButton::new("unupvote_btn") .label("Un-upvote"),
CreateButton::new("approve_btn") .label("Approve") .emoji(ReactionType::Unicode("".to_string())),
CreateButton::new("unapprove_btn") .label("Disapprove") .emoji(ReactionType::Unicode("".to_string())),
CreateButton::new("remove_btn") .label("Remove") .emoji(ReactionType::Unicode("🗑️".to_string()))
]);
return EmbedOptions {
title: Some(post_data["post_data"]["title"].as_str().unwrap().to_string()),
desc: format!("{}\n\nJSON: ||`{}`||", trimmed, serde_json::to_string(&json_min).unwrap()),
@@ -232,6 +252,7 @@ pub fn make_post_embed(post_data: &Value, url: &str, ephemeral: bool) -> EmbedOp
thumbnail: media_urls.first()
.and_then(|url| url.as_str().map(|s| s.to_string()))
.or(None),
actionrows: Some(vec![action_row]),
..Default::default()
};
}
+1 -1
View File
@@ -21,7 +21,7 @@ pub async fn cmd(
send_msg(ctx, lang!("dc_msg_bound_channel", c_id), true, true).await;
}
else {
send_msg(ctx, lang!("dc_msg_server_data_404"), false, false).await;
send_msg(ctx, lang!("dc_msg_data_server_404"), true, true).await;
}
return Ok(());
+1 -1
View File
@@ -22,7 +22,7 @@ pub fn to_shorturl(url: &str) -> Result<String, &str> {
pub async fn send_embed_for_post(ctx: Context<'_>, post: Value, url: &str) -> Result<(), Error> {
send_embed(ctx, make_post_embed(&post, url, true), true).await;
Ok(())
return Ok(());
}
pub async fn send_embed_for_removed(ctx: Context<'_>, url: &str, post: &Value) {
+1 -1
View File
@@ -105,7 +105,7 @@ async fn update_progress(ctx: Context<'_>, p: ReplyHandle<'_>, t: String, added_
async fn get_c_id(ctx: Context<'_>) -> Option<ChannelId> {
if !data::dc_contains_server(ctx.data(), ctx.guild_id().unwrap().into()).await {
send_msg(ctx, lang!("dc_msg_server_data_404"), false, false).await;
send_msg(ctx, lang!("dc_msg_data_server_404"), true, true).await;
return None;
}