djoka/bentopdf
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

refactor pdf decrypt

Browse Source
This commit is contained in:
alam00000
2026-03-15 14:28:37 +05:30
parent 11368314dd
commit 47c1be6df1
5 changed files with 495 additions and 177 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
src/js/workflow/nodes/decrypt-node.ts
View File

@@ -4,7 +4,7 @@ import { pdfSocket } from '../sockets';
import type { SocketData } from '../types';
import { requirePdfInput, processBatch } from '../types';
import { PDFDocument } from 'pdf-lib';
import { initializeQpdf } from '../../utils/helpers.js';
import { decryptPdfBytes } from '../../utils/pdf-decrypt.js';
export class DecryptNode extends BaseWorkflowNode {
readonly category = 'Secure PDF' as const;
@@ -33,35 +33,13 @@ export class DecryptNode extends BaseWorkflowNode {
return {
pdf: await processBatch(pdfInputs, async (input) => {
const qpdf = await initializeQpdf();
const uid = `${Date.now()}_${Math.random().toString(36).slice(2, 9)}`;
const inputPath = `/tmp/input_decrypt_${uid}.pdf`;
const outputPath = `/tmp/output_decrypt_${uid}.pdf`;
let decryptedData: Uint8Array;
try {
qpdf.FS.writeFile(inputPath, input.bytes);
const args = password
? [inputPath, '--password=' + password, '--decrypt', outputPath]
: [inputPath, '--decrypt', outputPath];
qpdf.callMain(args);
decryptedData = qpdf.FS.readFile(outputPath, { encoding: 'binary' });
} finally {
try {
qpdf.FS.unlink(inputPath);
} catch {
/* cleanup */
}
try {
qpdf.FS.unlink(outputPath);
} catch {
/* cleanup */
}
}
const resultBytes = new Uint8Array(decryptedData);
const document = await PDFDocument.load(resultBytes);
const { bytes: resultBytes } = await decryptPdfBytes(
input.bytes,
password
);
const document = await PDFDocument.load(resultBytes, {
throwOnInvalidObject: false,
});
return {
type: 'pdf',

51
src/js/workflow/nodes/pdf-input-node.ts
View File

@@ -3,7 +3,8 @@ import { BaseWorkflowNode } from './base-node';
import { pdfSocket } from '../sockets';
import type { PDFData, SocketData, MultiPDFData } from '../types';
import { PDFDocument } from 'pdf-lib';
import { readFileAsArrayBuffer, initializeQpdf } from '../../utils/helpers.js';
import { readFileAsArrayBuffer } from '../../utils/helpers.js';
import { decryptPdfBytes } from '../../utils/pdf-decrypt.js';
export class EncryptedPDFError extends Error {
constructor(public readonly filename: string) {
@@ -63,44 +64,16 @@ export class PDFInputNode extends BaseWorkflowNode {
async addDecryptedFile(file: File, password: string): Promise<void> {
const arrayBuffer = await readFileAsArrayBuffer(file);
const bytes = new Uint8Array(arrayBuffer as ArrayBuffer);
const qpdf = await initializeQpdf();
const uid = `${Date.now()}_${Math.random().toString(36).slice(2, 9)}`;
const inputPath = `/tmp/input_decrypt_${uid}.pdf`;
const outputPath = `/tmp/output_decrypt_${uid}.pdf`;
try {
qpdf.FS.writeFile(inputPath, bytes);
qpdf.callMain([
inputPath,
'--password=' + password,
'--decrypt',
outputPath,
]);
const decryptedData = qpdf.FS.readFile(outputPath, {
encoding: 'binary',
});
const decryptedBytes = new Uint8Array(decryptedData);
const document = await PDFDocument.load(decryptedBytes, {
throwOnInvalidObject: false,
});
this.files.push({
type: 'pdf',
document,
bytes: decryptedBytes,
filename: file.name,
});
} finally {
try {
qpdf.FS.unlink(inputPath);
} catch {
/* cleanup */
}
try {
qpdf.FS.unlink(outputPath);
} catch {
/* cleanup */
}
}
const { bytes: decryptedBytes } = await decryptPdfBytes(bytes, password);
const document = await PDFDocument.load(decryptedBytes, {
throwOnInvalidObject: false,
});
this.files.push({
type: 'pdf',
document,
bytes: decryptedBytes,
filename: file.name,
});
}
async setFile(file: File): Promise<void> {