package dev.plutorocks.mixin;

import dev.plutorocks.MinecraftIRC;
import net.minecraft.client.MinecraftClient;
import net.minecraft.client.gui.screen.ChatScreen;
import net.minecraft.text.MutableText;
import net.minecraft.text.Text;
import net.minecraft.util.Formatting;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.injection.At;
import org.spongepowered.asm.mixin.injection.Inject;
import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;

@Mixin(ChatScreen.class)
public abstract class ChatScreenMixin {

    @Inject(method = "sendMessage", at = @At("HEAD"), cancellable = true)
    private void minecraftirc$interceptIrcMessages(String chatText, boolean addToHistory, CallbackInfo ci) {
        if (chatText == null) {
            return;
        }

        String trimmed = chatText.trim();
        if (!trimmed.startsWith("!")) {
            return;
        }

        String ircMessage = trimmed.substring(1);

        MinecraftClient client = MinecraftClient.getInstance();

        if (MinecraftIRC.IRC_CLIENT != null && MinecraftIRC.IRC_CLIENT.isConnected()) {
            if (!ircMessage.isEmpty()) {
                MinecraftIRC.IRC_CLIENT.sendChannelMessage(ircMessage);
            }

            if (client != null && client.inGameHud != null) {
                MutableText prefix = Text.literal("[IRC] ").formatted(Formatting.AQUA);
                MutableText nick = Text.literal("<" + MinecraftIRC.ircNick + "> ")
                        .formatted(Formatting.WHITE);
                MutableText body = Text.literal(ircMessage)
                        .formatted(Formatting.GRAY);

                client.inGameHud.getChatHud().addMessage(
                        prefix.append(nick).append(body)
                );
            }
        } else {
            if (client != null && client.inGameHud != null) {
                MutableText prefix = Text.literal("[IRC] ").formatted(Formatting.AQUA);
                MutableText body = Text.literal("Not connected.")
                        .formatted(Formatting.GRAY);
                client.inGameHud.getChatHud().addMessage(
                        prefix.append(body)
                );
            }
        }

        ci.cancel();
    }
}