use std::env; use crate::{lang, Args, Context}; use poise::serenity_prelude::json::Value; use poise::{serenity_prelude::CreateMessage, CreateReply, ReplyHandle}; use poise::serenity_prelude::{ChannelId, Color, CreateActionRow, CreateButton, CreateEmbed, CreateEmbedAuthor, EditMessage, Http, Message, ReactionType, Timestamp, UserId}; #[derive(Clone)] pub struct Author { pub name: String, pub url: String, pub icon_url: String } #[derive(Clone)] pub struct EmbedOptions { pub desc: String, pub title: Option, pub col: Option, pub url: Option, pub ts: Option, pub ephemeral: bool, pub message: Option, pub author: Option, pub thumbnail: Option, pub actionrows: Option> } impl Default for EmbedOptions { fn default() -> Self { return EmbedOptions { desc: lang!("dc_msg_embed_default_embed_desc"), title: None, col: None, url: None, ts: None, ephemeral: false, message: None, author: None, thumbnail: None, actionrows: None }; } } static DEFAULT_DC_COL: u32 = 5793266; static REMOVED_DC_COL: u32 = 16716032; pub static JSON_TEXT_START: &str = "-# JSON: ||`"; pub static JSON_TEXT_END: &str = "`||"; fn none_to_empty(string: Option) -> String { return string.unwrap_or_default(); } pub async fn send_msg( ctx: Context<'_>, t: String, ephemeral: bool, reply: bool ) -> Option> { if reply { let r = CreateReply { content: Some(t), ephemeral: Some(ephemeral), ..Default::default() }; let msg = ctx.send(r).await; return Some(msg.unwrap()); } else { let _ = ctx.channel_id().say(ctx.http(), t).await; return None; } } #[allow(dead_code)] pub async fn http_send_msg( http: &Http, c_id: ChannelId, t: String ) -> Option { let r = CreateMessage::new().content(t); let msg = c_id.send_message(http, r).await; return msg.ok(); } pub async fn send_embed( ctx: Context<'_>, options: EmbedOptions, reply: bool ) -> Option> { let embed = embed_from_options(options.clone()); if reply { let r = CreateReply { embeds: vec![embed], content: options.message, ephemeral: Some(options.ephemeral), components: options.actionrows, ..Default::default() }; let msg = ctx.send(r).await; return Some(msg.unwrap()); } else { 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; } } pub async fn http_send_embed( http: &Http, c_id: ChannelId, options: EmbedOptions ) -> Option { let embed = embed_from_options(options.clone()); 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(); } pub fn embed_from_options(options: EmbedOptions) -> CreateEmbed { let mut author: Option = None; if let Some(o_author) = options.author { author = Some(CreateEmbedAuthor::new(o_author.name).url(o_author.url).icon_url(o_author.icon_url)) } let mut embed = CreateEmbed::new() .title (none_to_empty(options.title)) .description(options.desc) .colour (Color::new(options.col.unwrap_or(DEFAULT_DC_COL))) .url (none_to_empty(options.url)); if let Some(a) = author { embed = embed.author(a); } if options.thumbnail.is_some() { embed = embed.thumbnail(options.thumbnail.unwrap()); } if options.ts.is_some() { embed = embed.timestamp(options.ts.unwrap()); } return embed; } pub async fn edit_reply( ctx: Context<'_>, msg: ReplyHandle<'_>, new_text: String ) { let r = CreateReply { content: Some(new_text.clone()), ..Default::default() }; let _ = msg.edit(ctx, r).await; } #[allow(dead_code)] pub async fn http_edit_msg( http: &Http, mut msg: Message, new_msg: EditMessage ) { let _ = msg.edit(http, new_msg).await; } pub async fn send_dm(msg: String, args: Args, owners: Vec) { let token: String = if !args.test { env::var("ASSISTANT_TOKEN") .expect("Missing ASSISTANT_TOKEN env var!") } else { env::var("ASSISTANT_TOKEN_TEST").expect("Missing ASSISTANT_TOKEN_TEST env var!") }; let http = Http::new(&token); let c_msg = CreateMessage::new().content(msg); for uid in owners { let user = UserId::new(uid); let _ = user.dm(http.as_ref(), c_msg.clone()).await; } } pub fn make_post_embed(post_data: &Value, url: &str, ephemeral: bool) -> EmbedOptions { let media_type = &post_data["post_data"]["media_type"]; let desc_str = lang!( "dc_msg_embed_re_post", post_data["post_data"]["upvotes"].as_i64().unwrap(), post_data["votes"]["mod_voters"].as_array().unwrap().len(), if !media_type.is_null() { media_type.as_str().unwrap() } else { "None" }, url, if post_data["added"] ["by_human"].as_bool().unwrap() { "✅" } else { "❌" }, if post_data["added"] ["by_bot"].as_bool().unwrap() { "✅" } else { "❌" }, if post_data["approved"]["by_human"].as_bool().unwrap() { "✅" } else { "❌" } ); let trimmed = desc_str .lines() .map(|line| line.trim()) .collect::>() .join("\n"); let media_urls = post_data["post_data"]["media_urls"].as_array().unwrap(); let action_row = CreateActionRow::Buttons(vec![ CreateButton::new("vote_btn") .label("Vote") .emoji(ReactionType::Unicode("⬆️".to_string())), CreateButton::new("unvote_btn") .label("Un-vote"), 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\n{}{}{}", trimmed, JSON_TEXT_START, serde_json::to_string(&post_data).unwrap(), JSON_TEXT_END), col: Some(DEFAULT_DC_COL), url: Some(url.to_string()), ts: Some(Timestamp::from_unix_timestamp(post_data["post_data"]["date_unix"].as_i64().unwrap()).unwrap()), ephemeral, thumbnail: media_urls.first() .and_then(|url| url.as_str().map(|s| s.to_string())) .or(None), actionrows: Some(vec![action_row]), ..Default::default() }; } pub fn make_removed_embed(post_data: &Value, url: &str, ephemeral: bool) -> EmbedOptions { let action_row = CreateActionRow::Buttons(vec![ CreateButton::new("unremove_btn").label("Un-remove").emoji(ReactionType::Unicode("↩️".to_string())) ]); let desc = lang!( "dc_msg_embed_re_removed", post_data["removed"]["by"].as_str().unwrap(), if !post_data["removed"]["reason"].is_null() { post_data["removed"]["reason"].as_str().unwrap() } else { "None" }, url ); return EmbedOptions { title: Some(format!("[REMOVED] {}", post_data["post_data"]["title"])), desc: format!("{}\n\n{}{}{}", desc, JSON_TEXT_START, serde_json::to_string(&post_data).unwrap(), JSON_TEXT_END), col: Some(REMOVED_DC_COL), url: Some(url.to_string()), ts: Some(Timestamp::from_unix_timestamp(post_data["post_data"]["date_unix"].as_i64().unwrap()).unwrap()), ephemeral, actionrows: Some(vec![action_row]), ..Default::default() }; }