dircord/src/irc_discord.rs

use irc::{client::Client as IrcClient, proto::Command};

use std::{collections::HashMap, sync::Arc, time::Instant};

use tokio::sync::{mpsc::unbounded_channel, Mutex};

use tokio_stream::wrappers::UnboundedReceiverStream;

use serenity::{
    cache::Cache,
    futures::StreamExt,
    http::Http,
    model::{
        guild::Emoji,
        id::ChannelId,
        prelude::{GuildChannel, Member, UserId},
        webhook::Webhook,
    },
    prelude::*,
    utils::{content_safe, ContentSafeOptions},
};

use crate::{regex, OptionReplacer};

use fancy_regex::{Captures, Replacer};

macro_rules! unwrap_or_continue {
    ($opt:expr) => {
        match $opt {
            ::core::option::Option::Some(v) => v,
            ::core::option::Option::None => continue,
        }
    };
}

#[allow(clippy::too_many_lines)] // missing, fight me
pub async fn irc_loop(
    mut client: IrcClient,
    http: Arc<Http>,
    cache: Arc<Cache>,
    mapping: Arc<HashMap<String, u64>>,
    webhooks: HashMap<String, Webhook>,
    members: Arc<Mutex<Vec<Member>>>,
    cache_ttl: Option<u64>,
) -> anyhow::Result<()> {
    let (send, recv) = unbounded_channel();
    tokio::spawn(msg_task(UnboundedReceiverStream::new(recv)));

    let mut avatar_cache: HashMap<String, Option<String>> = HashMap::new();
    let mut id_cache: HashMap<String, Option<u64>> = HashMap::new();
    let mut emoji_cache: Vec<Emoji> = Vec::new();
    let mut channel_users: HashMap<String, Vec<String>> = HashMap::new();

    let mut ttl = Instant::now();

    client.identify()?;
    let mut stream = client.stream()?;

    for k in mapping.keys() {
        client.send(Command::NAMES(Some(k.clone()), None))?;
        client.send_topic(k, "")?;
    }

    let mut channels_cache = None;
    let mut guild = None;

    while let Some(orig_message) = stream.next().await.transpose()? {
        if ttl.elapsed().as_secs() > cache_ttl.unwrap_or(1800) {
            avatar_cache.clear();
            channels_cache = None;
            guild = None;
            emoji_cache.clear();
            ttl = Instant::now();
        }

        if let Command::Response(response, args) = orig_message.command {
            use irc::client::prelude::Response;

            if response == Response::RPL_NAMREPLY {
                let channel = args[2].to_string();
                let users = args[3]
                    .split(' ')
                    .map(ToOwned::to_owned)
                    .collect::<Vec<String>>();

                channel_users.insert(channel, users);
            } else if response == Response::RPL_TOPIC {
                let channel = &args[1];
                let topic = &args[2];

                let channel = ChannelId::from(*unwrap_or_continue!(mapping.get(channel)));
                channel.edit(&http, |c| c.topic(topic)).await?;
            }

            continue;
        };

        let mut nickname = unwrap_or_continue!(orig_message.source_nickname());

        if option_env!("DIRCORD_POLARIAN_MODE").is_some() {
            nickname = "polarbear";
        }
        if let Command::PRIVMSG(ref channel, ref message) = orig_message.command {
            let channel_id = ChannelId::from(*unwrap_or_continue!(mapping.get(channel)));

            if channels_cache.is_none() || guild.is_none() || emoji_cache.is_empty() {
                let (cc, g, es) = {
                    let guild = channel_id
                        .to_channel(&http)
                        .await?
                        .guild()
                        .unwrap()
                        .guild_id;

                    let chans = guild.channels(&http).await?;
                    let emojis = guild.emojis(&http).await?;

                    (chans, guild, emojis)
                };
                channels_cache = Some(cc);
                guild = Some(g);
                emoji_cache = es;
            }
            let channels = channels_cache.as_ref().unwrap();

            let members_lock = members.lock().await;

            let mut computed = irc_to_discord_processing(
                message,
                &*members_lock,
                &mut id_cache,
                channels,
                &emoji_cache,
            );

            computed = {
                let opts = ContentSafeOptions::new()
                    .clean_role(false)
                    .clean_user(false)
                    .clean_channel(false)
                    .show_discriminator(false)
                    .clean_here(true) // setting these to true explicitly isn't needed,
                    .clean_everyone(true); // but i did it anyway for readability

                content_safe(&cache, computed, &opts, &[])
            };

            if let Some(webhook) = webhooks.get(channel) {
                let avatar = &*avatar_cache.entry(nickname.to_owned()).or_insert_with(|| {
                    members_lock.iter().find_map(|member| {
                        (*member.display_name() == nickname)
                            .then(|| member.user.avatar_url())
                            .flatten()
                    })
                });

                send.send(QueuedMessage::Webhook {
                    webhook: webhook.clone(),
                    http: http.clone(),
                    avatar_url: avatar.clone(),
                    content: computed,
                    nickname: nickname.to_string(),
                })?;
            } else {
                send.send(QueuedMessage::Raw {
                    channel_id,
                    http: http.clone(),
                    message: format!("<{}>, {}", nickname, computed),
                })?;
            }
        } else if let Command::JOIN(ref channel, _, _) = orig_message.command {
            let channel_id = ChannelId::from(*unwrap_or_continue!(mapping.get(channel)));
            let users = unwrap_or_continue!(channel_users.get_mut(channel));

            users.push(nickname.to_string());

            send.send(QueuedMessage::Raw {
                channel_id,
                http: http.clone(),
                message: format!("*{}* has joined the channel", nickname),
            })?;
        } else if let Command::PART(ref channel, ref reason) = orig_message.command {
            let users = unwrap_or_continue!(channel_users.get_mut(channel));
            let channel_id = ChannelId::from(*unwrap_or_continue!(mapping.get(channel)));
            let pos = unwrap_or_continue!(users.iter().position(|u| u == nickname));

            users.swap_remove(pos);

            let reason = reason.as_deref().unwrap_or("Connection closed");

            send.send(QueuedMessage::Raw {
                channel_id,
                http: http.clone(),
                message: format!("*{}* has quit ({})", nickname, reason),
            })?;
        } else if let Command::QUIT(ref reason) = orig_message.command {
            for (channel, users) in &mut channel_users {
                let channel_id = ChannelId::from(*unwrap_or_continue!(mapping.get(channel)));
                let pos = unwrap_or_continue!(users.iter().position(|u| u == nickname));

                users.swap_remove(pos);

                let reason = reason.as_deref().unwrap_or("Connection closed");
                if reason == "Ping timeout: 2m30s"
                {
                    break;
                }
                send.send(QueuedMessage::Raw {
                    channel_id,
                    http: http.clone(),
                    message: format!("*{}* has quit ({})", nickname, reason),
                })?;
            }
        } else if let Command::NICK(ref new_nick) = orig_message.command {
            for (channel, users) in &mut channel_users {
                let channel_id = ChannelId::from(*unwrap_or_continue!(mapping.get(channel)));
                let pos = unwrap_or_continue!(users.iter().position(|u| u == nickname));

                users[pos] = new_nick.to_string();

                send.send(QueuedMessage::Raw {
                    channel_id,
                    http: http.clone(),
                    message: format!("*{}* is now known as *{}*", nickname, new_nick),
                })?;
            }
        } else if let Command::TOPIC(ref channel, ref topic) = orig_message.command {
            let topic = unwrap_or_continue!(topic.as_ref());
            let channel_id = ChannelId::from(*unwrap_or_continue!(mapping.get(channel)));
            channel_id.edit(&http, |c| c.topic(topic)).await?;
        }
    }
    Ok(())
}

fn irc_to_discord_processing(
    message: &str,
    members: &[Member],
    id_cache: &mut HashMap<String, Option<u64>>,
    channels: &HashMap<ChannelId, GuildChannel>,
    emojis: &[Emoji],
) -> String {
    struct MemberReplacer<'a> {
        id_cache: &'a mut HashMap<String, Option<u64>>,
        members: &'a [Member],
    }

    impl<'a> Replacer for MemberReplacer<'a> {
        fn replace_append(&mut self, caps: &Captures<'_>, dst: &mut String) {
            let slice = &caps[1];

            let id = self
                .id_cache
                .entry(slice.to_owned())
                .or_insert_with(|| {
                    self.members.iter().find_map(|member| {
                        (slice == member.display_name().as_str()).then(|| member.user.id.0)
                    })
                })
                .map(UserId);

            if let Some(id) = id {
                dst.push_str(&id.mention().to_string());
            } else {
                dst.push_str(caps.get(0).unwrap().as_str());
            }
        }
    }

    regex! {
        static PING_NICK_1 = r"^([\w+]+)(?::|,)";
        static PING_RE_2 = r"(?<=\s|^)@([\w\S]+)";
        static CONTROL_CHAR_RE = r"\x1f|\x02|\x12|\x0f|\x16|\x03(?:\d{1,2}(?:,\d{1,2})?)?";
        static WHITESPACE_RE = r"^\s";
        static CHANNEL_RE = r"#([A-Za-z-*]+)";
        static EMOJI_RE = r":(\w+):";
    }

    if WHITESPACE_RE.is_match(message).unwrap() && !PING_RE_2.is_match(message).unwrap() {
        return format!("`{}`", message);
    }

    let mut computed = message.to_owned();

    computed = PING_NICK_1
        .replace_all(&computed, MemberReplacer { id_cache, members })
        .into_owned();

    computed = PING_RE_2
        .replace_all(&computed, MemberReplacer { id_cache, members })
        .into_owned();

    computed = CHANNEL_RE
        .replace_all(
            &computed,
            OptionReplacer(|caps: &Captures| {
                channels
                    .iter()
                    .find_map(|(id, c)| (c.name == caps[1]).then(|| format!("<#{}>", id.0)))
            }),
        )
        .into_owned();

    computed = EMOJI_RE
        .replace_all(
            &computed,
            OptionReplacer(|caps: &Captures| {
                emojis
                    .iter()
                    .find_map(|e| (e.name == caps[1]).then(|| format!("<:{}:{}>", e.name, e.id.0)))
            }),
        )
        .into_owned();

    #[allow(clippy::map_unwrap_or)]
    {
        computed = computed
            .strip_prefix("\x01ACTION ")
            .and_then(|s| s.strip_suffix('\x01'))
            .map(|s| format!("*{}*", s))
            .unwrap_or_else(|| computed); // if any step in the way fails, fall back to using computed
    }

    computed = {
        let mut new = String::with_capacity(computed.len());

        let mut has_opened_bold = false;
        let mut has_opened_italic = false;

        for c in computed.chars() {
            if c == '\x02' || (c == '\x0F' && has_opened_bold) {
                new.push_str("**");
                has_opened_bold = !has_opened_bold;
            } else if c == '\x1D' || (c == '\x0F' && has_opened_italic) {
                new.push('*');
                has_opened_italic = !has_opened_italic;
            } else {
                new.push(c);
            }
        }

        if has_opened_italic {
            new.push('*');
        }

        if has_opened_bold {
            new.push_str("**");
        }

        CONTROL_CHAR_RE.replace_all(&new, "").into_owned()
    };

    computed
}

#[allow(clippy::large_enum_variant)] // lmao
#[derive(Debug)]
enum QueuedMessage {
    Webhook {
        webhook: Webhook,
        http: Arc<Http>,
        avatar_url: Option<String>,
        content: String,
        nickname: String,
    },
    Raw {
        channel_id: ChannelId,
        http: Arc<Http>,
        message: String,
    },
}

async fn msg_task(mut recv: UnboundedReceiverStream<QueuedMessage>) -> anyhow::Result<()> {
    while let Some(msg) = recv.next().await {
        match msg {
            QueuedMessage::Webhook {
                webhook,
                http,
                avatar_url,
                content,
                nickname,
            } => {
                if content.is_empty() {
                    continue;
                }
                webhook
                    .execute(&http, true, |w| {
                        if let Some(ref url) = avatar_url {
                            w.avatar_url(url);
                        }

                        w.username(nickname).content(content)
                    })
                    .await?;
            }
            QueuedMessage::Raw {
                channel_id,
                http,
                message,
            } => {
                if message.is_empty() {
                    continue;
                }
                channel_id.say(&http, message).await?;
            }
        }
    }
    Ok(())
}