TotalFreedomMod/src/main/java/me/totalfreedom/totalfreedommod/discord/DiscordToMinecraftListener.java

package me.totalfreedom.totalfreedommod.discord;

import me.totalfreedom.totalfreedommod.TotalFreedomMod;
import me.totalfreedom.totalfreedommod.config.ConfigEntry;
import me.totalfreedom.totalfreedommod.discord.command.DiscordCommandManager;
import me.totalfreedom.totalfreedommod.rank.Rank;
import me.totalfreedom.totalfreedommod.rank.Title;
import me.totalfreedom.totalfreedommod.util.FLog;
import me.totalfreedom.totalfreedommod.util.FUtil;
import net.dv8tion.jda.api.entities.*;
import net.dv8tion.jda.api.events.message.MessageReceivedEvent;
import net.dv8tion.jda.api.hooks.ListenerAdapter;
import net.md_5.bungee.api.chat.BaseComponent;
import net.md_5.bungee.api.chat.ClickEvent;
import net.md_5.bungee.api.chat.ComponentBuilder;
import net.md_5.bungee.api.chat.HoverEvent;
import net.md_5.bungee.api.chat.TextComponent;
import net.md_5.bungee.api.chat.hover.content.Text;
import org.bukkit.Bukkit;
import org.bukkit.ChatColor;
import org.bukkit.entity.Player;

public class DiscordToMinecraftListener extends ListenerAdapter
{
    public void onMessageReceived(MessageReceivedEvent event)
    {
        final String chat_channel_id = ConfigEntry.DISCORD_CHAT_CHANNEL_ID.getString();
        final MessageChannel genericChannel = event.getChannel();

        if (event.getMember() == null)
        {
            return;
        }

        if (chat_channel_id.isEmpty())
        {
            return;
        }

        if (event.getAuthor().getId().equals(Discord.bot.getSelfUser().getId()))
        {
            return;
        }

        if (!genericChannel.getId().equals(chat_channel_id))
        {
            return;
        }

        if (!(genericChannel instanceof TextChannel))
        {
            return;
        }

        final TextChannel textChannel = (TextChannel) genericChannel;

        final Member member = event.getMember();
        final String tag = getDisplay(member);
        final Message msg = event.getMessage();
        final String content = msg.getContentStripped();

        if (content.startsWith(ConfigEntry.DISCORD_PREFIX.getString()))
        {
            Discord.DISCORD_COMMAND_MANAGER.parse(content, member, textChannel);
            return;
        }

        ComponentBuilder emsg = new ComponentBuilder();

        // Prefix
        emsg.append(ChatColor.DARK_GRAY + "[");
        TextComponent inviteLink = new TextComponent("Discord");
        inviteLink.setColor(ChatColor.DARK_AQUA.asBungee());
        inviteLink.setHoverEvent(new HoverEvent(HoverEvent.Action.SHOW_TEXT,
                new Text("Click here to get the invite link!")));
        inviteLink.setClickEvent(new ClickEvent(ClickEvent.Action.OPEN_URL,
                ConfigEntry.DISCORD_INVITE_LINK.getString()));
        emsg.append(inviteLink);
        emsg.append(ChatColor.DARK_GRAY + "] ", ComponentBuilder.FormatRetention.NONE);

        // Tag (if they have one)
        if (tag != null)
        {
            emsg.append(tag);
        }

        emsg.append(" ");

        // User
        TextComponent user = new TextComponent(FUtil.stripColors(member.getEffectiveName()));
        user.setColor(ChatColor.RED.asBungee());
        emsg.append(user);

        // Message
        emsg.append(ChatColor.DARK_GRAY + ": " + ChatColor.RESET
                + FUtil.stripColors(msg.getContentDisplay()), ComponentBuilder.FormatRetention.NONE);

        // Attachments
        if (!msg.getAttachments().isEmpty())
        {
            if (!msg.getContentDisplay().isEmpty())
                emsg.append(" ");

            for (Message.Attachment attachment : msg.getAttachments())
            {
                TextComponent media = new TextComponent("[Media] ");
                media.setColor(ChatColor.YELLOW.asBungee());
                media.setClickEvent(new ClickEvent(ClickEvent.Action.OPEN_URL, attachment.getUrl()));
                media.setHoverEvent(new HoverEvent(HoverEvent.Action.SHOW_TEXT, new Text(attachment.getUrl())));

                emsg.append(media, ComponentBuilder.FormatRetention.NONE);
            }
        }

        BaseComponent[] components = emsg.create();

        for (Player player : Bukkit.getOnlinePlayers())
        {
            if (TotalFreedomMod.getPlugin().pl.getData(player).doesDisplayDiscord())
            {
                player.sendMessage(components);
            }
        }

        FLog.info(TextComponent.toLegacyText(components), true);
    }


    public String getDisplay(Member member)
    {
        Guild server = Discord.bot.getGuildById(ConfigEntry.DISCORD_SERVER_ID.getString());
        // Server Owner
        assert server != null;
        if (member.getRoles().contains(server.getRoleById(ConfigEntry.DISCORD_SERVER_OWNER_ROLE_ID.getString())))
        {
            return Title.OWNER.getColoredTag();
        }
        // Developers
        else if (member.getRoles().contains(server.getRoleById(ConfigEntry.DISCORD_DEVELOPER_ROLE_ID.getString())))
        {
            return Title.DEVELOPER.getColoredTag();
        }
        // Executives
        else if (member.getRoles().contains(server.getRoleById(ConfigEntry.DISCORD_EXECUTIVE_ROLE_ID.getString())))
        {
            return Title.EXECUTIVE.getColoredTag();
        }
        // Assistant Executives
        else if (member.getRoles().contains(server.getRoleById(ConfigEntry.DISCORD_ASSISTANT_EXECUTIVE_ROLE_ID.getString())))
        {
            return Title.ASSTEXEC.getColoredTag();
        }
        // Senior Admins
        else if (member.getRoles().contains(server.getRoleById(ConfigEntry.DISCORD_SENIOR_ADMIN_ROLE_ID.getString())))
        {
            return Rank.SENIOR_ADMIN.getColoredTag();
        }
        // Admins
        else if (member.getRoles().contains(server.getRoleById(ConfigEntry.DISCORD_NEW_ADMIN_ROLE_ID.getString())))
        {
            return Rank.ADMIN.getColoredTag();
        }
        // Master Builders
        else if (member.getRoles().contains(server.getRoleById(ConfigEntry.DISCORD_MASTER_BUILDER_ROLE_ID.getString())))
        {
            return Title.MASTER_BUILDER.getColoredTag();
        }
        // None
        else
        {
            return null;
        }
    }
}