can1357
60 lines
2.1 KiB
TypeScript
60 lines
2.1 KiB
TypeScript
// Adapted from markit-ai (MIT). See ./NOTICE.
|
|
import * as path from "node:path";
|
|
import { DocxConverter } from "./converters/docx";
|
|
import { EpubConverter } from "./converters/epub";
|
|
import { PdfConverter } from "./converters/pdf";
|
|
import { PptxConverter } from "./converters/pptx";
|
|
import { XlsxConverter } from "./converters/xlsx";
|
|
import type { ConversionResult, Converter, MarkitOptions, StreamInfo } from "./types";
|
|
|
|
/**
|
|
* In-house document → markdown engine (replaces the `markit-ai` package).
|
|
*
|
|
* Only the document converters omp routes are registered (pdf, docx, pptx,
|
|
* xlsx, epub). The first converter whose `accepts()` returns true and whose
|
|
* `convert()` succeeds wins.
|
|
*/
|
|
export class Markit {
|
|
readonly #converters: readonly Converter[];
|
|
readonly #options: MarkitOptions;
|
|
|
|
constructor(options: MarkitOptions = {}) {
|
|
this.#options = options;
|
|
this.#converters = [
|
|
new PdfConverter(),
|
|
new DocxConverter(),
|
|
new PptxConverter(),
|
|
new XlsxConverter(),
|
|
new EpubConverter(),
|
|
];
|
|
}
|
|
|
|
async convertFile(filePath: string, extra?: { imageDir?: string }): Promise<ConversionResult> {
|
|
const buffer = Buffer.from(await Bun.file(filePath).arrayBuffer());
|
|
const streamInfo: StreamInfo = {
|
|
localPath: filePath,
|
|
extension: path.extname(filePath).toLowerCase(),
|
|
filename: path.basename(filePath),
|
|
...extra,
|
|
};
|
|
return this.convert(buffer, streamInfo);
|
|
}
|
|
|
|
async convert(input: Buffer, streamInfo: StreamInfo): Promise<ConversionResult> {
|
|
const errors: { converter: string; error: Error }[] = [];
|
|
for (const converter of this.#converters) {
|
|
if (!converter.accepts(streamInfo)) continue;
|
|
try {
|
|
return await converter.convert(input, streamInfo, this.#options);
|
|
} catch (err) {
|
|
errors.push({ converter: converter.name, error: err instanceof Error ? err : new Error(String(err)) });
|
|
}
|
|
}
|
|
if (errors.length > 0) {
|
|
const details = errors.map(e => ` ${e.converter}: ${e.error.message}`).join("\n");
|
|
throw new Error(`Conversion failed:\n${details}`);
|
|
}
|
|
throw new Error(`Unsupported format: ${streamInfo.extension || streamInfo.mimetype || "unknown"}`);
|
|
}
|
|
}
|