BungeeCore/src/de/steamwar/messages/ChatSender.java

/*
    This file is a part of the SteamWar software.

    Copyright (C) 2022  SteamWar.de-Serverteam

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU Affero General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    GNU Affero General Public License for more details.

    You should have received a copy of the GNU Affero General Public License
    along with this program.  If not, see <https://www.gnu.org/licenses/>.
*/

package de.steamwar.messages;

import de.steamwar.bungeecore.BungeeCore;
import de.steamwar.bungeecore.Message;
import de.steamwar.bungeecore.bot.SteamwarDiscordBot;
import de.steamwar.bungeecore.sql.SteamwarUser;
import net.md_5.bungee.api.ChatMessageType;
import net.md_5.bungee.api.CommandSender;
import net.md_5.bungee.api.chat.BaseComponent;
import net.md_5.bungee.api.chat.ClickEvent;
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 net.md_5.bungee.api.connection.ProxiedPlayer;
import net.md_5.bungee.command.ConsoleCommandSender;

import java.text.DateFormat;
import java.text.MessageFormat;
import java.util.Date;
import java.util.Locale;
import java.util.ResourceBundle;
import java.util.logging.Level;

public interface ChatSender {

    SteamwarUser user();

    Locale getLocale();
    boolean chatShown();
    void sendMessage(ChatMessageType type, BaseComponent... msg);

    default void chat(Message message) {
        send(false, ChatMessageType.CHAT, null, null, message);
    }

    default void system(String message, Object... params) {
        send(true, ChatMessageType.SYSTEM, null, null, new Message(message, params));
    }

    default void system(String message, Message onHover, ClickEvent onClick, Object... params) {
        send(true, ChatMessageType.SYSTEM, onHover, onClick, new Message(message, params));
    }

    default void prefixless(String message, Object... params) {
        send(false, ChatMessageType.SYSTEM, null, null, new Message(message, params));
    }

    default void send(boolean prefixed, ChatMessageType type, Message onHover, ClickEvent onClick, Message message) {
        TextComponent msg = new TextComponent(parse(prefixed, message));
        if(onHover != null)
            msg.setHoverEvent(new HoverEvent(HoverEvent.Action.SHOW_TEXT, new Text(parse(false, onHover))));
        if(onClick != null)
            msg.setClickEvent(onClick);
        sendMessage(type, msg);
    }

    default BaseComponent[] parse(boolean prefixed, Message message) {
        Locale locale = getLocale();
        ResourceBundle resourceBundle = SteamwarResourceBundle.getResourceBundle(locale);
        String pattern = "";
        if(prefixed)
            pattern = resourceBundle.getObject("PREFIX") + " ";
        pattern += (String)resourceBundle.getObject(message.getMessage());

        MessageFormat format = new MessageFormat(pattern, locale);
        Object[] params = message.getParams();
        for (int i = 0; i < params.length; i++) {
            if(params[i] instanceof Message) {
                params[i] = parse(false, (Message) params[i]);
            } else if(params[i] instanceof Date) {
                params[i] = DateFormat.getDateTimeInstance(DateFormat.SHORT, DateFormat.SHORT, locale).format((Date) params[i]);
            } else if(params[i] instanceof SteamwarUser) {
                params[i] = ((SteamwarUser) params[i]).getUserName();
            } else if(params[i] instanceof ProxiedPlayer) {
                params[i] = ((ProxiedPlayer) params[i]).getName();
            } else if(params[i] instanceof ChatSender) {
                params[i] = ((ChatSender) params[i]).user().getUserName();
            }
        }
        return TextComponent.fromLegacyText(format.format(params));
    }

    static ChatSender of(ProxiedPlayer player) {
        return new ChatSender() {
            @Override
            public SteamwarUser user() {
                return SteamwarUser.get(player.getUniqueId());
            }

            @Override
            public Locale getLocale() {
                Locale locale = player.getLocale();
                if(locale == null)
                    locale = Locale.getDefault();
                return locale;
            }

            @Override
            public boolean chatShown() {
                return player.getChatMode() == ProxiedPlayer.ChatMode.SHOWN;
            }

            @Override
            public void sendMessage(ChatMessageType type, BaseComponent... msg) {
                player.sendMessage(type, msg);
            }
        };
    }

    static ChatSender of(CommandSender sender) {
        if(sender instanceof ProxiedPlayer)
            return of((ProxiedPlayer) sender);

        //Console
        return new ChatSender() {
            @Override
            public SteamwarUser user() {
                return SteamwarUser.get(-1);
            }

            @Override
            public Locale getLocale() {
                return Locale.getDefault();
            }

            @Override
            public boolean chatShown() {
                return true;
            }

            @Override
            public void sendMessage(ChatMessageType type, BaseComponent... msg) {
                sender.sendMessage(msg);
            }
        };
    }

    static ChatSender console() {
        return of(ConsoleCommandSender.getInstance());
    }

    static ChatSender discordConsole() {
        return new ChatSender() {
            @Override
            public SteamwarUser user() {
                return SteamwarUser.get(-1);
            }

            @Override
            public Locale getLocale() {
                return Locale.getDefault();
            }

            @Override
            public boolean chatShown() {
                return true;
            }

            @Override
            public void sendMessage(ChatMessageType type, BaseComponent... msg) {
                try {
                    SteamwarDiscordBot.instance().getIngameChatListener().send("<t:" + (System.currentTimeMillis() / 1000) + "> " + new TextComponent(msg).toPlainText());
                } catch (Exception e) {
                    BungeeCore.get().getLogger().log(Level.SEVERE, "Could not send chat message to discord", e);
                }
            }
        };
    }

    static ChatSender of(net.dv8tion.jda.api.entities.Message message, SteamwarUser user) {
        return new ChatSender() {
            @Override
            public SteamwarUser user() {
                return user;
            }

            @Override
            public Locale getLocale() {
                return Locale.getDefault();
            }

            @Override
            public boolean chatShown() {
                return true;
            }

            @Override
            public void sendMessage(ChatMessageType type, BaseComponent... msg) {
                try {
                    //TODO SteamwarDiscordBot.instance().getIngameChatListener().send("<t:" + (System.currentTimeMillis() / 1000) + "> " + new TextComponent(msg).toPlainText());
                } catch (Exception e) {
                    BungeeCore.get().getLogger().log(Level.SEVERE, "Could not send chat reply to discord user", e);
                }
            }
        };
    }
}