forked from karx/dircord
Split Discord->IRC code into separate file
This commit is contained in:
parent
8a676c1919
commit
0a911c1b20
298
src/discord_irc.rs
Normal file
298
src/discord_irc.rs
Normal file
|
@ -0,0 +1,298 @@
|
||||||
|
use crate::{
|
||||||
|
regex, ChannelMappingKey, MembersKey, OptionReplacer, OptionStringKey, SenderKey, StrChunks,
|
||||||
|
UserIdKey,
|
||||||
|
};
|
||||||
|
use fancy_regex::{Captures, Replacer};
|
||||||
|
use pulldown_cmark::Parser;
|
||||||
|
use serenity::{
|
||||||
|
async_trait,
|
||||||
|
client::Context,
|
||||||
|
http::CacheHttp,
|
||||||
|
model::{
|
||||||
|
channel::{Channel, Message, MessageReference, MessageType},
|
||||||
|
guild::Member,
|
||||||
|
prelude::{ChannelId, GuildId, Ready, Role, RoleId},
|
||||||
|
},
|
||||||
|
prelude::*,
|
||||||
|
};
|
||||||
|
use std::borrow::Cow;
|
||||||
|
use std::collections::HashMap;
|
||||||
|
|
||||||
|
async fn create_prefix(msg: &Message, is_reply: bool, http: impl CacheHttp) -> (String, usize) {
|
||||||
|
let nick = match msg.member(http).await {
|
||||||
|
Ok(Member {
|
||||||
|
nick: Some(nick), ..
|
||||||
|
}) => Cow::Owned(nick),
|
||||||
|
_ => Cow::Borrowed(&msg.author.name),
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut chars = nick.char_indices();
|
||||||
|
let first_char = chars.next().unwrap().1;
|
||||||
|
let second_char_offset = chars.next().unwrap().0;
|
||||||
|
|
||||||
|
let colour_index = (first_char as usize + nick.len()) % 12;
|
||||||
|
|
||||||
|
let prefix = format!(
|
||||||
|
"{}<\x03{:02}{}\u{200B}{}\x0F> ",
|
||||||
|
if is_reply { "(reply to) " } else { "" },
|
||||||
|
colour_index,
|
||||||
|
&nick[..second_char_offset],
|
||||||
|
&nick[second_char_offset..]
|
||||||
|
);
|
||||||
|
let content_limit = 510 - prefix.len();
|
||||||
|
|
||||||
|
(prefix, content_limit)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct Handler;
|
||||||
|
|
||||||
|
#[async_trait]
|
||||||
|
impl EventHandler for Handler {
|
||||||
|
async fn message(&self, ctx: Context, msg: Message) {
|
||||||
|
match msg.kind {
|
||||||
|
MessageType::Regular | MessageType::InlineReply => {}
|
||||||
|
_ => return,
|
||||||
|
}
|
||||||
|
|
||||||
|
let ctx_data = ctx.data.read().await;
|
||||||
|
|
||||||
|
let user_id = ctx_data.get::<UserIdKey>().copied().unwrap();
|
||||||
|
let sender = ctx_data.get::<SenderKey>().unwrap();
|
||||||
|
let members = ctx_data.get::<MembersKey>().unwrap();
|
||||||
|
let raw_prefix = ctx_data
|
||||||
|
.get::<OptionStringKey>()
|
||||||
|
.unwrap()
|
||||||
|
.as_deref()
|
||||||
|
.unwrap_or("++");
|
||||||
|
let mapping = ctx_data.get::<ChannelMappingKey>().unwrap().clone();
|
||||||
|
|
||||||
|
if user_id == msg.author.id || msg.author.bot {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
let (prefix, content_limit) = create_prefix(&msg, false, &ctx).await;
|
||||||
|
|
||||||
|
let (channel, channel_id) = match mapping.iter().find(|(_, &v)| v == msg.channel_id.0) {
|
||||||
|
Some((k, v)) => (k.as_str(), ChannelId::from(*v)),
|
||||||
|
None => return,
|
||||||
|
};
|
||||||
|
|
||||||
|
let attachments: Vec<&str> = msg.attachments.iter().map(|a| a.url.as_str()).collect();
|
||||||
|
|
||||||
|
let roles = channel_id
|
||||||
|
.to_channel(&ctx)
|
||||||
|
.await
|
||||||
|
.unwrap()
|
||||||
|
.guild()
|
||||||
|
.unwrap()
|
||||||
|
.guild_id
|
||||||
|
.roles(&ctx)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
let members_lock = members.lock().await;
|
||||||
|
|
||||||
|
let computed = discord_to_irc_processing(&msg.content, &**members_lock, &ctx, &roles).await;
|
||||||
|
|
||||||
|
if let Some(MessageReference {
|
||||||
|
guild_id,
|
||||||
|
channel_id,
|
||||||
|
message_id: Some(message_id),
|
||||||
|
..
|
||||||
|
}) = msg.message_reference
|
||||||
|
{
|
||||||
|
if let Ok(mut reply) = channel_id.message(&ctx, message_id).await {
|
||||||
|
reply.guild_id = guild_id; // lmao
|
||||||
|
let (reply_prefix, reply_content_limit) = create_prefix(&reply, true, &ctx).await;
|
||||||
|
|
||||||
|
let mut content = reply.content;
|
||||||
|
content = content.replace("\r\n", " "); // just in case
|
||||||
|
content = content.replace('\n', " ");
|
||||||
|
content = format!(
|
||||||
|
"{} {}",
|
||||||
|
content,
|
||||||
|
reply
|
||||||
|
.attachments
|
||||||
|
.iter()
|
||||||
|
.map(|a| &*a.url)
|
||||||
|
.collect::<String>()
|
||||||
|
);
|
||||||
|
|
||||||
|
let to_send = if content.len() > reply_content_limit {
|
||||||
|
format!("{}...", &content[..reply_content_limit - 3])
|
||||||
|
} else {
|
||||||
|
content
|
||||||
|
};
|
||||||
|
|
||||||
|
sender
|
||||||
|
.send_privmsg(channel, &format!("{}{}", reply_prefix, to_send))
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some((stripped, false)) = computed
|
||||||
|
.strip_prefix(&raw_prefix)
|
||||||
|
.map(str::trim)
|
||||||
|
.and_then(|v| v.strip_suffix('\x0F'))
|
||||||
|
.map(|v| (v, v.is_empty()))
|
||||||
|
{
|
||||||
|
sender.send_privmsg(channel, &prefix).unwrap();
|
||||||
|
sender.send_privmsg(channel, stripped).unwrap();
|
||||||
|
} else {
|
||||||
|
for line in computed.lines() {
|
||||||
|
for chunk in StrChunks::new(line, content_limit) {
|
||||||
|
sender
|
||||||
|
.send_privmsg(channel, &format!("{}{}", prefix, chunk))
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for attachment in attachments {
|
||||||
|
sender
|
||||||
|
.send_privmsg(channel, &format!("{}{}", prefix, attachment))
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn ready(&self, ctx: Context, info: Ready) {
|
||||||
|
let id = info.user.id;
|
||||||
|
|
||||||
|
let mut data = ctx.data.write().await;
|
||||||
|
data.insert::<UserIdKey>(id);
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn guild_member_addition(&self, ctx: Context, _: GuildId, new_member: Member) {
|
||||||
|
let ctx_data = ctx.data.read().await;
|
||||||
|
let mut members = ctx_data.get::<MembersKey>().unwrap().lock().await;
|
||||||
|
members.push(new_member);
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn guild_member_update(&self, ctx: Context, _: Option<Member>, new: Member) {
|
||||||
|
let ctx_data = ctx.data.read().await;
|
||||||
|
let mut members = ctx_data.get::<MembersKey>().unwrap().lock().await;
|
||||||
|
|
||||||
|
let x = members
|
||||||
|
.iter()
|
||||||
|
.position(|m| m.user.id == new.user.id)
|
||||||
|
.unwrap();
|
||||||
|
members.remove(x);
|
||||||
|
members.push(new);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn discord_to_irc_processing(
|
||||||
|
message: &str,
|
||||||
|
members: &[Member],
|
||||||
|
ctx: &Context,
|
||||||
|
roles: &HashMap<RoleId, Role>,
|
||||||
|
) -> String {
|
||||||
|
struct MemberReplacer<'a> {
|
||||||
|
members: &'a [Member],
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a> Replacer for MemberReplacer<'a> {
|
||||||
|
fn replace_append(&mut self, caps: &Captures<'_>, dst: &mut String) {
|
||||||
|
let id = caps[1].parse::<u64>().unwrap();
|
||||||
|
|
||||||
|
let display_name = self.members.iter().find_map(|member| {
|
||||||
|
(id == member.user.id.0).then(|| member.display_name().into_owned())
|
||||||
|
});
|
||||||
|
|
||||||
|
if let Some(display_name) = display_name {
|
||||||
|
dst.push_str(&format!("@{}", display_name));
|
||||||
|
} else {
|
||||||
|
dst.push_str(caps.get(0).unwrap().as_str());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
regex! {
|
||||||
|
static PING_RE_1 = r"<@([0-9]+)>";
|
||||||
|
static PING_RE_2 = r"<@!([0-9]+)>";
|
||||||
|
static EMOJI_RE = r"<:(\w+):[0-9]+>";
|
||||||
|
static CHANNEL_RE = r"<#([0-9]+)>";
|
||||||
|
static ROLE_RE = r"<@&([0-9]+)>";
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut computed = message.to_owned();
|
||||||
|
|
||||||
|
computed = PING_RE_1
|
||||||
|
.replace_all(&computed, MemberReplacer { members })
|
||||||
|
.into_owned();
|
||||||
|
|
||||||
|
computed = PING_RE_2
|
||||||
|
.replace_all(&computed, MemberReplacer { members })
|
||||||
|
.into_owned();
|
||||||
|
|
||||||
|
computed = EMOJI_RE.replace_all(&computed, "$1").into_owned();
|
||||||
|
|
||||||
|
// FIXME: the await makes it impossible to use `replace_all`, idk how to fix this
|
||||||
|
for caps in CHANNEL_RE.captures_iter(&computed.clone()) {
|
||||||
|
let replacement = match ChannelId(caps.unwrap()[1].parse().unwrap())
|
||||||
|
.to_channel(&ctx)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
Ok(Channel::Guild(gc)) => Cow::Owned(format!("#{}", gc.name)),
|
||||||
|
Ok(Channel::Category(cat)) => Cow::Owned(format!("#{}", cat.name)),
|
||||||
|
_ => Cow::Borrowed("#deleted-channel"),
|
||||||
|
};
|
||||||
|
|
||||||
|
computed = CHANNEL_RE.replace(&computed, replacement).to_string();
|
||||||
|
}
|
||||||
|
|
||||||
|
computed = ROLE_RE
|
||||||
|
.replace_all(
|
||||||
|
&computed,
|
||||||
|
OptionReplacer(|caps: &Captures| {
|
||||||
|
roles
|
||||||
|
.get(&RoleId(caps[1].parse().unwrap()))
|
||||||
|
.map(|role| format!("@{}", role.name))
|
||||||
|
}),
|
||||||
|
)
|
||||||
|
.into_owned();
|
||||||
|
|
||||||
|
computed = {
|
||||||
|
#[allow(clippy::enum_glob_use)]
|
||||||
|
use pulldown_cmark::{Event::*, Tag::*};
|
||||||
|
|
||||||
|
let mut new = String::with_capacity(computed.len());
|
||||||
|
|
||||||
|
for line in computed.lines() {
|
||||||
|
let parser = Parser::new(line);
|
||||||
|
|
||||||
|
let mut computed_line = String::with_capacity(line.len());
|
||||||
|
|
||||||
|
for event in parser {
|
||||||
|
match event {
|
||||||
|
Text(t) | Html(t) => computed_line.push_str(&t),
|
||||||
|
Code(t) => computed_line.push_str(&format!("`{}`", t)),
|
||||||
|
End(_) => computed_line.push('\x0F'),
|
||||||
|
Start(Emphasis) => computed_line.push('\x1D'),
|
||||||
|
Start(Strong) => computed_line.push('\x02'),
|
||||||
|
Start(Link(_, dest, _)) => {
|
||||||
|
computed_line.push_str(&dest);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
Start(List(num)) => {
|
||||||
|
if let Some(num) = num {
|
||||||
|
computed_line.push_str(&format!("{}. ", num));
|
||||||
|
} else {
|
||||||
|
computed_line.push_str("- ");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Start(BlockQuote) => computed_line.push_str("> "),
|
||||||
|
_ => {}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
computed_line.push('\n');
|
||||||
|
|
||||||
|
new.push_str(&computed_line);
|
||||||
|
}
|
||||||
|
|
||||||
|
new
|
||||||
|
};
|
||||||
|
|
||||||
|
computed
|
||||||
|
}
|
282
src/main.rs
282
src/main.rs
|
@ -1,5 +1,8 @@
|
||||||
#![warn(clippy::pedantic)]
|
#![warn(clippy::pedantic)]
|
||||||
|
|
||||||
|
mod discord_irc;
|
||||||
|
mod irc_discord;
|
||||||
|
|
||||||
use std::{borrow::Cow, collections::HashMap, env, fs::File, io::Read, sync::Arc};
|
use std::{borrow::Cow, collections::HashMap, env, fs::File, io::Read, sync::Arc};
|
||||||
|
|
||||||
use serenity::{
|
use serenity::{
|
||||||
|
@ -43,7 +46,7 @@ struct DircordConfig {
|
||||||
webhooks: Option<HashMap<String, String>>,
|
webhooks: Option<HashMap<String, String>>,
|
||||||
}
|
}
|
||||||
|
|
||||||
struct StrChunks<'a> {
|
pub(crate) struct StrChunks<'a> {
|
||||||
v: &'a str,
|
v: &'a str,
|
||||||
size: usize,
|
size: usize,
|
||||||
}
|
}
|
||||||
|
@ -84,169 +87,6 @@ impl<'a> StrChunks<'a> {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn create_prefix(msg: &Message, is_reply: bool, http: impl CacheHttp) -> (String, usize) {
|
|
||||||
let nick = match msg.member(http).await {
|
|
||||||
Ok(Member {
|
|
||||||
nick: Some(nick), ..
|
|
||||||
}) => Cow::Owned(nick),
|
|
||||||
_ => Cow::Borrowed(&msg.author.name),
|
|
||||||
};
|
|
||||||
|
|
||||||
let mut chars = nick.char_indices();
|
|
||||||
let first_char = chars.next().unwrap().1;
|
|
||||||
let second_char_offset = chars.next().unwrap().0;
|
|
||||||
|
|
||||||
let colour_index = (first_char as usize + nick.len()) % 12;
|
|
||||||
|
|
||||||
let prefix = format!(
|
|
||||||
"{}<\x03{:02}{}\u{200B}{}\x0F> ",
|
|
||||||
if is_reply { "(reply to) " } else { "" },
|
|
||||||
colour_index,
|
|
||||||
&nick[..second_char_offset],
|
|
||||||
&nick[second_char_offset..]
|
|
||||||
);
|
|
||||||
let content_limit = 510 - prefix.len();
|
|
||||||
|
|
||||||
(prefix, content_limit)
|
|
||||||
}
|
|
||||||
|
|
||||||
struct Handler;
|
|
||||||
|
|
||||||
#[async_trait]
|
|
||||||
impl EventHandler for Handler {
|
|
||||||
async fn message(&self, ctx: Context, msg: Message) {
|
|
||||||
match msg.kind {
|
|
||||||
MessageType::Regular | MessageType::InlineReply => {}
|
|
||||||
_ => return,
|
|
||||||
}
|
|
||||||
|
|
||||||
let ctx_data = ctx.data.read().await;
|
|
||||||
|
|
||||||
let user_id = ctx_data.get::<UserIdKey>().copied().unwrap();
|
|
||||||
let sender = ctx_data.get::<SenderKey>().unwrap();
|
|
||||||
let members = ctx_data.get::<MembersKey>().unwrap();
|
|
||||||
let raw_prefix = ctx_data
|
|
||||||
.get::<OptionStringKey>()
|
|
||||||
.unwrap()
|
|
||||||
.as_deref()
|
|
||||||
.unwrap_or("++");
|
|
||||||
let mapping = ctx_data.get::<ChannelMappingKey>().unwrap().clone();
|
|
||||||
|
|
||||||
if user_id == msg.author.id || msg.author.bot {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
let (prefix, content_limit) = create_prefix(&msg, false, &ctx).await;
|
|
||||||
|
|
||||||
let (channel, channel_id) = match mapping.iter().find(|(_, &v)| v == msg.channel_id.0) {
|
|
||||||
Some((k, v)) => (k.as_str(), ChannelId::from(*v)),
|
|
||||||
None => return,
|
|
||||||
};
|
|
||||||
|
|
||||||
let attachments: Vec<&str> = msg.attachments.iter().map(|a| a.url.as_str()).collect();
|
|
||||||
|
|
||||||
let roles = channel_id
|
|
||||||
.to_channel(&ctx)
|
|
||||||
.await
|
|
||||||
.unwrap()
|
|
||||||
.guild()
|
|
||||||
.unwrap()
|
|
||||||
.guild_id
|
|
||||||
.roles(&ctx)
|
|
||||||
.await
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
let members_lock = members.lock().await;
|
|
||||||
|
|
||||||
let computed = discord_to_irc_processing(&msg.content, &**members_lock, &ctx, &roles).await;
|
|
||||||
|
|
||||||
if let Some(MessageReference {
|
|
||||||
guild_id,
|
|
||||||
channel_id,
|
|
||||||
message_id: Some(message_id),
|
|
||||||
..
|
|
||||||
}) = msg.message_reference
|
|
||||||
{
|
|
||||||
if let Ok(mut reply) = channel_id.message(&ctx, message_id).await {
|
|
||||||
reply.guild_id = guild_id; // lmao
|
|
||||||
let (reply_prefix, reply_content_limit) = create_prefix(&reply, true, &ctx).await;
|
|
||||||
|
|
||||||
let mut content = reply.content;
|
|
||||||
content = content.replace("\r\n", " "); // just in case
|
|
||||||
content = content.replace('\n', " ");
|
|
||||||
content = format!(
|
|
||||||
"{} {}",
|
|
||||||
content,
|
|
||||||
reply
|
|
||||||
.attachments
|
|
||||||
.iter()
|
|
||||||
.map(|a| &*a.url)
|
|
||||||
.collect::<String>()
|
|
||||||
);
|
|
||||||
|
|
||||||
let to_send = if content.len() > reply_content_limit {
|
|
||||||
format!("{}...", &content[..reply_content_limit - 3])
|
|
||||||
} else {
|
|
||||||
content
|
|
||||||
};
|
|
||||||
|
|
||||||
sender
|
|
||||||
.send_privmsg(channel, &format!("{}{}", reply_prefix, to_send))
|
|
||||||
.unwrap();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if let Some((stripped, false)) = computed
|
|
||||||
.strip_prefix(&raw_prefix)
|
|
||||||
.map(str::trim)
|
|
||||||
.and_then(|v| v.strip_suffix('\x0F'))
|
|
||||||
.map(|v| (v, v.is_empty()))
|
|
||||||
{
|
|
||||||
sender.send_privmsg(channel, &prefix).unwrap();
|
|
||||||
sender.send_privmsg(channel, stripped).unwrap();
|
|
||||||
} else {
|
|
||||||
for line in computed.lines() {
|
|
||||||
for chunk in StrChunks::new(line, content_limit) {
|
|
||||||
sender
|
|
||||||
.send_privmsg(channel, &format!("{}{}", prefix, chunk))
|
|
||||||
.unwrap();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
for attachment in attachments {
|
|
||||||
sender
|
|
||||||
.send_privmsg(channel, &format!("{}{}", prefix, attachment))
|
|
||||||
.unwrap();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn ready(&self, ctx: Context, info: Ready) {
|
|
||||||
let id = info.user.id;
|
|
||||||
|
|
||||||
let mut data = ctx.data.write().await;
|
|
||||||
data.insert::<UserIdKey>(id);
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn guild_member_addition(&self, ctx: Context, _: GuildId, new_member: Member) {
|
|
||||||
let ctx_data = ctx.data.read().await;
|
|
||||||
let mut members = ctx_data.get::<MembersKey>().unwrap().lock().await;
|
|
||||||
members.push(new_member);
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn guild_member_update(&self, ctx: Context, _: Option<Member>, new: Member) {
|
|
||||||
let ctx_data = ctx.data.read().await;
|
|
||||||
let mut members = ctx_data.get::<MembersKey>().unwrap().lock().await;
|
|
||||||
|
|
||||||
let x = members
|
|
||||||
.iter()
|
|
||||||
.position(|m| m.user.id == new.user.id)
|
|
||||||
.unwrap();
|
|
||||||
members.remove(x);
|
|
||||||
members.push(new);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
macro_rules! type_map_key {
|
macro_rules! type_map_key {
|
||||||
($($name:ident => $value:ty),* $(,)?) => {
|
($($name:ident => $value:ty),* $(,)?) => {
|
||||||
$(
|
$(
|
||||||
|
@ -514,6 +354,7 @@ impl<T: AsRef<str>, F: for<'r, 't> FnMut(&'r Captures<'t>) -> Option<T>> Replace
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[macro_export]
|
||||||
macro_rules! regex {
|
macro_rules! regex {
|
||||||
($(static $name:ident = $regex:literal;)*) => {
|
($(static $name:ident = $regex:literal;)*) => {
|
||||||
::lazy_static::lazy_static! {
|
::lazy_static::lazy_static! {
|
||||||
|
@ -621,119 +462,6 @@ fn irc_to_discord_processing(
|
||||||
computed
|
computed
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn discord_to_irc_processing(
|
|
||||||
message: &str,
|
|
||||||
members: &[Member],
|
|
||||||
ctx: &Context,
|
|
||||||
roles: &HashMap<RoleId, Role>,
|
|
||||||
) -> String {
|
|
||||||
struct MemberReplacer<'a> {
|
|
||||||
members: &'a [Member],
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<'a> Replacer for MemberReplacer<'a> {
|
|
||||||
fn replace_append(&mut self, caps: &Captures<'_>, dst: &mut String) {
|
|
||||||
let id = caps[1].parse::<u64>().unwrap();
|
|
||||||
|
|
||||||
let display_name = self.members.iter().find_map(|member| {
|
|
||||||
(id == member.user.id.0).then(|| member.display_name().into_owned())
|
|
||||||
});
|
|
||||||
|
|
||||||
if let Some(display_name) = display_name {
|
|
||||||
dst.push_str(&format!("@{}", display_name));
|
|
||||||
} else {
|
|
||||||
dst.push_str(caps.get(0).unwrap().as_str());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
regex! {
|
|
||||||
static PING_RE_1 = r"<@([0-9]+)>";
|
|
||||||
static PING_RE_2 = r"<@!([0-9]+)>";
|
|
||||||
static EMOJI_RE = r"<:(\w+):[0-9]+>";
|
|
||||||
static CHANNEL_RE = r"<#([0-9]+)>";
|
|
||||||
static ROLE_RE = r"<@&([0-9]+)>";
|
|
||||||
}
|
|
||||||
|
|
||||||
let mut computed = message.to_owned();
|
|
||||||
|
|
||||||
computed = PING_RE_1
|
|
||||||
.replace_all(&computed, MemberReplacer { members })
|
|
||||||
.into_owned();
|
|
||||||
|
|
||||||
computed = PING_RE_2
|
|
||||||
.replace_all(&computed, MemberReplacer { members })
|
|
||||||
.into_owned();
|
|
||||||
|
|
||||||
computed = EMOJI_RE.replace_all(&computed, "$1").into_owned();
|
|
||||||
|
|
||||||
// FIXME: the await makes it impossible to use `replace_all`, idk how to fix this
|
|
||||||
for caps in CHANNEL_RE.captures_iter(&computed.clone()) {
|
|
||||||
let replacement = match ChannelId(caps.unwrap()[1].parse().unwrap()).to_channel(&ctx).await {
|
|
||||||
Ok(Channel::Guild(gc)) => Cow::Owned(format!("#{}", gc.name)),
|
|
||||||
Ok(Channel::Category(cat)) => Cow::Owned(format!("#{}", cat.name)),
|
|
||||||
_ => Cow::Borrowed("#deleted-channel"),
|
|
||||||
};
|
|
||||||
|
|
||||||
computed = CHANNEL_RE.replace(&computed, replacement).to_string();
|
|
||||||
}
|
|
||||||
|
|
||||||
computed = ROLE_RE
|
|
||||||
.replace_all(
|
|
||||||
&computed,
|
|
||||||
OptionReplacer(|caps: &Captures| {
|
|
||||||
roles
|
|
||||||
.get(&RoleId(caps[1].parse().unwrap()))
|
|
||||||
.map(|role| format!("@{}", role.name))
|
|
||||||
}),
|
|
||||||
)
|
|
||||||
.into_owned();
|
|
||||||
|
|
||||||
computed = {
|
|
||||||
#[allow(clippy::enum_glob_use)]
|
|
||||||
use pulldown_cmark::{Event::*, Tag::*};
|
|
||||||
|
|
||||||
let mut new = String::with_capacity(computed.len());
|
|
||||||
|
|
||||||
for line in computed.lines() {
|
|
||||||
let parser = Parser::new(line);
|
|
||||||
|
|
||||||
let mut computed_line = String::with_capacity(line.len());
|
|
||||||
|
|
||||||
for event in parser {
|
|
||||||
match event {
|
|
||||||
Text(t) | Html(t) => computed_line.push_str(&t),
|
|
||||||
Code(t) => computed_line.push_str(&format!("`{}`", t)),
|
|
||||||
End(_) => computed_line.push('\x0F'),
|
|
||||||
Start(Emphasis) => computed_line.push('\x1D'),
|
|
||||||
Start(Strong) => computed_line.push('\x02'),
|
|
||||||
Start(Link(_, dest, _)) => {
|
|
||||||
computed_line.push_str(&dest);
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
Start(List(num)) => {
|
|
||||||
if let Some(num) = num {
|
|
||||||
computed_line.push_str(&format!("{}. ", num));
|
|
||||||
} else {
|
|
||||||
computed_line.push_str("- ");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Start(BlockQuote) => computed_line.push_str("> "),
|
|
||||||
_ => {}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
computed_line.push('\n');
|
|
||||||
|
|
||||||
new.push_str(&computed_line);
|
|
||||||
}
|
|
||||||
|
|
||||||
new
|
|
||||||
};
|
|
||||||
|
|
||||||
computed
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn parse_webhook_url(http: Arc<Http>, url: String) -> anyhow::Result<Webhook> {
|
async fn parse_webhook_url(http: Arc<Http>, url: String) -> anyhow::Result<Webhook> {
|
||||||
let url = url.trim_start_matches("https://discord.com/api/webhooks/");
|
let url = url.trim_start_matches("https://discord.com/api/webhooks/");
|
||||||
let split = url.split('/').collect::<Vec<&str>>();
|
let split = url.split('/').collect::<Vec<&str>>();
|
||||||
|
|
Loading…
Reference in a new issue