splitPdfByIndex

This commit is contained in:
Felix Kaspar 2024-05-18 22:01:19 +02:00
parent 771e66100f
commit edd4e0c39a

View File

@ -1,25 +1,52 @@
import { PdfFile, RepresentationType } from "../wrappers/PdfFile";
import { Operator, Progress, oneToN } from ".";
import Joi from "@stirling-tools/joi";
import { JoiPDFFileSchema } from "../wrappers/PdfFileJoi";
import i18next from "i18next";
import CommaArrayJoiExt from "../wrappers/CommaArrayJoiExt";
import { PdfFile } from "../wrappers/PdfFile";
import { parsePageIndexSpecification } from "./common/pageIndexesUtils";
import { splitPagesByIndex } from "./common/splitPagesByIndex";
export interface SplitPdfByIndexParamsType {
file: PdfFile;
pageIndexes: string | number[];
}
export async function splitPdfByIndex(params: SplitPdfByIndexParamsType): Promise<PdfFile[]> {
const { file, pageIndexes } = params;
const pdfLibDocument = await file.pdfLibDocument;
export class SplitPdfByIndex extends Operator {
static type = "splitPdfByIndex";
let indexes = pageIndexes;
/**
* Validation & Localisation
*/
if (!Array.isArray(indexes)) {
indexes = parsePageIndexSpecification(indexes, pdfLibDocument.getPageCount());
protected static inputSchema = JoiPDFFileSchema.label(i18next.t("inputs.pdffile.name")).description(i18next.t("inputs.pdffile.description"));
protected static valueSchema = Joi.object({
pageIndexes: CommaArrayJoiExt.comma_array().items(Joi.number().integer()).required()
.label(i18next.t("values.pageIndexes.friendlyName", { ns: "splitPdfByIndex" })).description(i18next.t("values.pageIndexes.description", { ns: "splitPdfByIndex" }))
.example("1").example("1, 2, 3, 4").example("4, 2, 4, 3").required()
});
protected static outputSchema = JoiPDFFileSchema.label(i18next.t("outputs.pdffile.name")).description(i18next.t("outputs.pdffile.description"));
static schema = Joi.object({
input: SplitPdfByIndex.inputSchema,
values: SplitPdfByIndex.valueSchema.required(),
output: SplitPdfByIndex.outputSchema
}).label(i18next.t("friendlyName", { ns: "splitPdfByIndex" })).description(i18next.t("description", { ns: "splitPdfByIndex" }));
/**
* Logic
*/
/** PDF extraction, specify pages from one pdf and output them to a new pdf */
async run(input: PdfFile[], progressCallback: (state: Progress) => void): Promise<PdfFile[]> {
return oneToN<PdfFile, PdfFile>(input, async (input, index, max) => {
const newFiles = await splitPagesByIndex(input, this.actionValues.pageIndexes);
for (let i = 0; i < newFiles.length; i++) {
newFiles[i].filename += "_split-" + i;
}
progressCallback({ curFileProgress: 1, operationProgress: index/max });
return newFiles;
});
}
const newFiles = await splitPagesByIndex(file, indexes);
for (let i = 0; i < newFiles.length; i++) {
newFiles[i].filename += "_split-"+i;
}
return newFiles;
}
}