import {
  ApplicationCommandType,
  ContextMenuCommandBuilder,
  DiscordAPIError,
  DiscordjsError,
  DiscordjsErrorCodes,
} from "discord.js";
import { defineCommand } from "..";
import { translateImpl } from "../language/translate";
import { ocrImpl } from "./ocr";
import { buildTranslateModal } from "../language/translateMenu";
import { abort } from "../../utils/error";
import { findFuzzyLanguage } from "../../utils/deepl";
import { defer, getImageUrlFromMessage } from "../../utils/functions";

export default defineCommand({
  data: new ContextMenuCommandBuilder()
    .setName("OCR and translate...")
    .setType(ApplicationCommandType.Message),

  async execute(interaction) {
    if (!interaction.isMessageContextMenuCommand()) return;

    const imageUrl = getImageUrlFromMessage(interaction.targetMessage);

    const modal = buildTranslateModal();
    await interaction.showModal(modal);

    await interaction
      .awaitModalSubmit({
        filter: i => i.customId === "translate-modal",
        time: 60000 * 5,
      })
      .then(async interaction => {
        await defer(interaction);

        const sourceField =
          interaction.fields.getTextInputValue("source") || null;
        const targetField =
          interaction.fields.getTextInputValue("target") || "en-US";

        const source =
          sourceField === null
            ? sourceField
            : await findFuzzyLanguage(sourceField, "source").then(l => l?.code);
        const target =
          targetField === "en-US"
            ? targetField
            : await findFuzzyLanguage(targetField, "target").then(l => l?.code);

        if (source === undefined) {
          abort("Source language not found");
        }
        if (!target) {
          abort("Target language not found");
        }

        const { text, model } = await ocrImpl(imageUrl);
        const payload = await translateImpl(text, source, target, model);
        await interaction.editReply(payload);
      })
      .catch(err => {
        if (
          (err instanceof DiscordjsError &&
            err.code === DiscordjsErrorCodes.InteractionCollectorError) ||
          (err instanceof DiscordAPIError &&
            err.message.includes("Unknown interaction"))
        )
          return;
        throw err;
      });
  },
});