mirror of
https://github.com/snapotter-hq/SnapOtter.git
synced 2026-08-03 07:46:42 +02:00
feat(tools): 2.0 phase 5 wave 4 - office, ebooks, data, archives (14 tools) (#224)
This commit is contained in:
@@ -0,0 +1,57 @@
|
||||
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],
|
||||
};
|
||||
},
|
||||
});
|
||||
}
|
||||
Reference in New Issue
Block a user