import path from "node:path";
import { CONFIG } from "../config.js";
import { ToolError } from "../errors.js";
import { jitteredBackoffMs, sleep } from "../utils.js";
import { doc2xRequestJson, normalizeUrl } from "./client.js";
type ExportFilenameMode = "auto" | "raw";
function normalizeExportFilename(filename: string, to: "md" | "tex" | "docx", mode: ExportFilenameMode): string {
const v = String(filename).trim();
if (!v) return v;
const base = path.basename(v);
if (mode === "raw") return base;
if (to === "md") return base.replace(/\.md$/i, "");
if (to === "tex") return base.replace(/\.tex$/i, "");
if (to === "docx") return base.replace(/\.docx$/i, "");
return base;
}
function exportUrlLooksLike(url: string, to: "md" | "tex" | "docx"): boolean {
const u = String(url || "").toLowerCase();
if (!u) return false;
if (to === "docx") return u.includes("convert_docx");
return u.includes(`convert_${to}_`);
}
export async function convertExportSubmit(args: {
uid: string;
to: "md" | "tex" | "docx";
formula_mode: "normal" | "dollar";
filename?: string;
merge_cross_page_forms?: boolean;
filename_mode?: ExportFilenameMode;
}) {
const body: any = {
uid: args.uid,
to: args.to,
formula_mode: args.formula_mode
};
if (args.merge_cross_page_forms != null) body.merge_cross_page_forms = args.merge_cross_page_forms;
if (args.filename != null) body.filename = normalizeExportFilename(args.filename, args.to, args.filename_mode ?? "auto");
const data = await doc2xRequestJson("POST", "/api/v2/convert/parse", { body });
return { uid: args.uid, status: String(data.status), url: String(data.url || "") };
}
export async function convertExportResult(uid: string) {
const data = await doc2xRequestJson("GET", "/api/v2/convert/parse/result", { query: { uid } });
return { uid, status: String(data.status), url: data.url ? normalizeUrl(String(data.url)) : "" };
}
export async function convertExportWaitByUid(args: {
uid: string;
to: "md" | "tex" | "docx";
poll_interval_ms?: number;
max_wait_ms?: number;
}) {
const pollInterval = args.poll_interval_ms ?? CONFIG.pollIntervalMs;
const maxWait = args.max_wait_ms ?? CONFIG.maxWaitMs;
const start = Date.now();
let attempt = 0;
while (true) {
if (Date.now() - start > maxWait) {
throw new ToolError({
code: "timeout",
message: `wait timeout after ${maxWait}ms (hint: exports for the same uid should be run sequentially, not in parallel)`,
retryable: true,
uid: args.uid
});
}
let st: Awaited<ReturnType<typeof convertExportResult>>;
try {
st = await convertExportResult(args.uid);
attempt = 0;
} catch (e) {
if (e instanceof ToolError && e.retryable) {
await sleep(jitteredBackoffMs(attempt++));
continue;
}
throw e;
}
if (st.status === "success") {
if (st.url && exportUrlLooksLike(st.url, args.to)) return st;
await sleep(pollInterval);
continue;
}
if (st.status === "failed") throw new ToolError({ code: "convert_failed", message: "convert failed", retryable: true, uid: args.uid });
await sleep(pollInterval);
}
}
export async function convertExportWait(args: {
uid: string;
to: "md" | "tex" | "docx";
formula_mode: "normal" | "dollar";
filename?: string;
merge_cross_page_forms?: boolean;
filename_mode?: ExportFilenameMode;
poll_interval_ms?: number;
max_wait_ms?: number;
}) {
const pollInterval = args.poll_interval_ms ?? CONFIG.pollIntervalMs;
const maxWait = args.max_wait_ms ?? CONFIG.maxWaitMs;
await convertExportSubmit({
uid: args.uid,
to: args.to,
formula_mode: args.formula_mode,
filename: args.filename,
filename_mode: args.filename_mode,
merge_cross_page_forms: args.merge_cross_page_forms
});
return convertExportWaitByUid({ uid: args.uid, to: args.to, poll_interval_ms: pollInterval, max_wait_ms: maxWait });
}
