import { writeFile } from "node:fs/promises";
import { extname, join } from "node:path";
import { convertDocument } from "@snapotter/doc-engine";
import type { FastifyInstance } from "fastify";
import { z } from "zod";
import { env } from "../../config.js";
import { InputValidationError } from "../../modality/contract.js";
import { createToolRoute } from "../tool-factory.js";

const CONTENT_TYPES: Record<string, string> = {
  docx: "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
  odt: "application/vnd.oasis.opendocument.text",
  rtf: "application/rtf",
  txt: "text/plain",
};

const settingsSchema = z.object({
  format: z.enum(["docx", "odt", "rtf", "txt"]),
});

export function registerConvertDocument(app: FastifyInstance) {
  createToolRoute(app, {
    toolId: "convert-document",
    settingsSchema,
    process: async () => {
      throw new Error("convert-document is v2-only");
    },
    processV2: async (ctx) => {
      const settings = ctx.settings as z.infer<typeof settingsSchema>;
      const input = ctx.inputs[0];

      // Reject same-format no-ops
      const inputExt = extname(input.filename).toLowerCase();
      if (inputExt === `.${settings.format}`) {
        throw new InputValidationError("The file is already in that format", 422);
      }

      const base = input.filename.replace(/\.[^.]+$/, "");
      // Preserve the real extension so LibreOffice can sniff the input format.
      const sanitized = input.filename.replace(/[^A-Za-z0-9._-]/g, "_");
      const inPath = join(ctx.scratchDir, `in-${sanitized}`);
      await writeFile(inPath, input.buffer);

      ctx.report(10, "Converting");
      const outPath = await convertDocument(inPath, ctx.scratchDir, settings.format, {
        timeoutMs: (env.LIBREOFFICE_TIMEOUT_S || 120) * 1000,
      });
      ctx.report(90, "Done");

      return {
        scratchPath: outPath,
        filename: `${base}.${settings.format}`,
        contentType: CONTENT_TYPES[settings.format],
      };
    },
  });
}