mirror of
https://github.com/Frooodle/Stirling-PDF.git
synced 2025-06-02 01:16:44 +02:00
Started migrating workflow controller
This commit is contained in:
parent
57142381ca
commit
5ae8cb77ac
@ -1,13 +0,0 @@
|
|||||||
|
|
||||||
declare module '@stirling-pdf/shared-operations/wasm/pdfcpu/pdfcpu-wrapper-node.js' {
|
|
||||||
export async function oneToOne(wasmArray: any, snapshot: any): Promise<Uint8Array>;
|
|
||||||
}
|
|
||||||
|
|
||||||
declare module '@stirling-pdf/shared-operations/workflow/traverseOperations.js' {
|
|
||||||
export type PDF = {
|
|
||||||
originalFileName: string;
|
|
||||||
fileName: string;
|
|
||||||
buffer: Uint8Array;
|
|
||||||
}
|
|
||||||
export async function * traverseOperations(operations: any, input: PDF|PDF[], Operations: any);
|
|
||||||
}
|
|
@ -1,6 +1,6 @@
|
|||||||
import express, { Request, Response } from 'express';
|
import express, { Request, Response } from 'express';
|
||||||
|
|
||||||
//import workflow from './workflow-controller';
|
import workflow from './workflow-controller';
|
||||||
import operations from './operations-controller';
|
import operations from './operations-controller';
|
||||||
|
|
||||||
const router = express.Router();
|
const router = express.Router();
|
||||||
@ -11,6 +11,6 @@ router.get("/", (req: Request, res: Response) => {
|
|||||||
});
|
});
|
||||||
|
|
||||||
router.use("/operations", operations);
|
router.use("/operations", operations);
|
||||||
//router.use("/workflow", workflow);
|
router.use("/workflow", workflow);
|
||||||
|
|
||||||
export default router;
|
export default router;
|
@ -1,7 +1,7 @@
|
|||||||
|
|
||||||
import Operations from '../../utils/pdf-operations';
|
import Operations from '../../utils/pdf-operations';
|
||||||
import { respondWithPdfFile, response_mustHaveExactlyOneFile } from '../../utils/endpoint-utils';
|
import { respondWithPdfFile, response_mustHaveExactlyOneFile } from '../../utils/endpoint-utils';
|
||||||
import { PdfFile, fromMulterFile } from '@stirling-pdf/shared-operations/wrappers/PdfFile'
|
import { PdfFile, fromMulterFile } from '@stirling-pdf/shared-operations/src/wrappers/PdfFile'
|
||||||
|
|
||||||
import express, { Request, Response } from 'express';
|
import express, { Request, Response } from 'express';
|
||||||
const router = express.Router();
|
const router = express.Router();
|
||||||
|
@ -5,100 +5,100 @@ import Archiver from 'archiver';
|
|||||||
import multer from 'multer'
|
import multer from 'multer'
|
||||||
const upload = multer();
|
const upload = multer();
|
||||||
|
|
||||||
//import Operations from "../../utils/pdf-operations";
|
import Operations from "../../utils/pdf-operations";
|
||||||
//import { traverseOperations } from "@stirling-pdf/shared-operations/workflow/traverseOperations.js";
|
import { traverseOperations } from "@stirling-pdf/shared-operations/src/workflow/traverseOperations";
|
||||||
|
|
||||||
const activeWorkflows: any = {};
|
const activeWorkflows: any = {};
|
||||||
|
|
||||||
const router = express.Router();
|
const router = express.Router();
|
||||||
/*
|
|
||||||
router.post("/:workflowUuid?", [
|
router.post("/:workflowUuid?", [
|
||||||
upload.any(),
|
upload.array("files"),
|
||||||
async (req: Request, res: Response) => {
|
async (req: Request, res: Response) => {
|
||||||
if(req.files == null) {
|
// TODO: Maybe replace with another validator
|
||||||
|
if(req.files?.length == 0) {
|
||||||
res.status(400).json({"error": "No files were uploaded."});
|
res.status(400).json({"error": "No files were uploaded."});
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
const filesArr = Array.isArray(req.files.files) ? req.files.files : [req.files.files];
|
// TODO: Validate input further (json may be invalid or not be in workflow format)
|
||||||
|
|
||||||
const workflow = JSON.parse(req.body.workflow);
|
const workflow = JSON.parse(req.body.workflow);
|
||||||
// TODO: Validate input further (json may fail or not be a valid workflow)
|
|
||||||
|
|
||||||
const inputs = await Promise.all(filesArr.map(async file => {
|
const inputs = await Promise.all((req.files as Express.Multer.File[]).map(async file => {
|
||||||
|
console.log(file);
|
||||||
return {
|
return {
|
||||||
originalFileName: file.name.replace(/\.[^/.]+$/, ""),
|
originalFileName: file.originalname.replace(/\.[^/.]+$/, ""),
|
||||||
fileName: file.name.replace(/\.[^/.]+$/, ""),
|
fileName: file.originalname.replace(/\.[^/.]+$/, ""),
|
||||||
buffer: new Uint8Array(await file.data)
|
buffer: new Uint8Array(await file.buffer)
|
||||||
}
|
}
|
||||||
}));
|
}));
|
||||||
|
|
||||||
// Allow option to do it synchronously and just make a long request
|
// // Allow option to do it synchronously and just make a long request
|
||||||
if(req.body.async === "false") {
|
// if(req.body.async === "false") {
|
||||||
console.log("Don't do async");
|
// console.log("Don't do async");
|
||||||
|
|
||||||
const traverse = traverseOperations(workflow.operations, inputs, Operations);
|
// const traverse = traverseOperations(workflow.operations, inputs, Operations);
|
||||||
|
|
||||||
let pdfResults;
|
// let pdfResults;
|
||||||
let iteration;
|
// let iteration;
|
||||||
while (true) {
|
// while (true) {
|
||||||
iteration = await traverse.next();
|
// iteration = await traverse.next();
|
||||||
if (iteration.done) {
|
// if (iteration.done) {
|
||||||
pdfResults = iteration.value;
|
// pdfResults = iteration.value;
|
||||||
console.log("Done");
|
// console.log("Done");
|
||||||
break;
|
// break;
|
||||||
}
|
// }
|
||||||
console.log(iteration.value);
|
// console.log(iteration.value);
|
||||||
}
|
// }
|
||||||
|
|
||||||
console.log("Download");
|
// console.log("Download");
|
||||||
downloadHandler(res, pdfResults);
|
// downloadHandler(res, pdfResults);
|
||||||
}
|
// }
|
||||||
else {
|
// else {
|
||||||
console.log("Start Aync Workflow");
|
// console.log("Start Aync Workflow");
|
||||||
// TODO: UUID collision checks
|
// // TODO: UUID collision checks
|
||||||
let workflowID = req.params.workflowUuid
|
// let workflowID = req.params.workflowUuid
|
||||||
if(!workflowID)
|
// if(!workflowID)
|
||||||
workflowID = generateWorkflowID();
|
// workflowID = generateWorkflowID();
|
||||||
|
|
||||||
activeWorkflows[workflowID] = {
|
// activeWorkflows[workflowID] = {
|
||||||
createdAt: Date.now(),
|
// createdAt: Date.now(),
|
||||||
finished: false,
|
// finished: false,
|
||||||
eventStream: null,
|
// eventStream: null,
|
||||||
result: null,
|
// result: null,
|
||||||
// TODO: When auth is implemented: owner
|
// // TODO: When auth is implemented: owner
|
||||||
}
|
// }
|
||||||
const activeWorkflow = activeWorkflows[workflowID];
|
// const activeWorkflow = activeWorkflows[workflowID];
|
||||||
|
|
||||||
res.status(200).json({
|
// res.status(200).json({
|
||||||
"workflowID": workflowID,
|
// "workflowID": workflowID,
|
||||||
"data-recieved": {
|
// "data-recieved": {
|
||||||
"fileCount": filesArr.length,
|
// "fileCount": filesArr.length,
|
||||||
"workflow": workflow
|
// "workflow": workflow
|
||||||
}
|
// }
|
||||||
});
|
// });
|
||||||
|
|
||||||
const traverse = traverseOperations(workflow.operations, inputs, Operations);
|
// const traverse = traverseOperations(workflow.operations, inputs, Operations);
|
||||||
|
|
||||||
let pdfResults;
|
// let pdfResults;
|
||||||
let iteration;
|
// let iteration;
|
||||||
while (true) {
|
// while (true) {
|
||||||
iteration = await traverse.next();
|
// iteration = await traverse.next();
|
||||||
if (iteration.done) {
|
// if (iteration.done) {
|
||||||
pdfResults = iteration.value;
|
// pdfResults = iteration.value;
|
||||||
if(activeWorkflow.eventStream) {
|
// if(activeWorkflow.eventStream) {
|
||||||
activeWorkflow.eventStream.write(`data: processing done\n\n`);
|
// activeWorkflow.eventStream.write(`data: processing done\n\n`);
|
||||||
activeWorkflow.eventStream.end();
|
// activeWorkflow.eventStream.end();
|
||||||
}
|
// }
|
||||||
break;
|
// break;
|
||||||
}
|
// }
|
||||||
if(activeWorkflow.eventStream)
|
// if(activeWorkflow.eventStream)
|
||||||
activeWorkflow.eventStream.write(`data: ${iteration.value}\n\n`);
|
// activeWorkflow.eventStream.write(`data: ${iteration.value}\n\n`);
|
||||||
}
|
// }
|
||||||
|
|
||||||
activeWorkflow.result = pdfResults;
|
// activeWorkflow.result = pdfResults;
|
||||||
activeWorkflow.finished = true;
|
// activeWorkflow.finished = true;
|
||||||
}
|
// }
|
||||||
}
|
}
|
||||||
]);
|
]);
|
||||||
|
|
||||||
@ -159,7 +159,7 @@ router.get("/result/:workflowUuid", (req: Request, res: Response) => {
|
|||||||
* If workflow isn't done return error
|
* If workflow isn't done return error
|
||||||
* Send file, TODO: if there are multiple outputs return as zip
|
* Send file, TODO: if there are multiple outputs return as zip
|
||||||
* If download is done, delete results / allow deletion within the next 5-60 mins
|
* If download is done, delete results / allow deletion within the next 5-60 mins
|
||||||
*
|
*/
|
||||||
const workflow = activeWorkflows[req.params.workflowUuid];
|
const workflow = activeWorkflows[req.params.workflowUuid];
|
||||||
if(!workflow.finished) {
|
if(!workflow.finished) {
|
||||||
res.status(202).json({ message: "Workflow hasn't finished yet. Check progress or connect to progress-steam to get notified when its done." });
|
res.status(202).json({ message: "Workflow hasn't finished yet. Check progress or connect to progress-steam to get notified when its done." });
|
||||||
@ -227,5 +227,5 @@ function downloadHandler(res: Response, pdfResults: any) {
|
|||||||
readStream.pipe(res);
|
readStream.pipe(res);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
*/
|
|
||||||
export default router;
|
export default router;
|
@ -1,6 +1,6 @@
|
|||||||
|
|
||||||
import { Response } from 'express';
|
import { Response } from 'express';
|
||||||
import { PdfFile } from '@stirling-pdf/shared-operations/wrappers/PdfFile'
|
import { PdfFile } from '@stirling-pdf/shared-operations/src/wrappers/PdfFile'
|
||||||
|
|
||||||
export async function respondWithPdfFile(res: Response, file: PdfFile): Promise<void> {
|
export async function respondWithPdfFile(res: Response, file: PdfFile): Promise<void> {
|
||||||
const byteFile = await file.convertToByteArrayFile();
|
const byteFile = await file.convertToByteArrayFile();
|
||||||
|
@ -1,5 +1,5 @@
|
|||||||
|
|
||||||
import SharedOperations from "@stirling-pdf/shared-operations";
|
import SharedOperations from "@stirling-pdf/shared-operations/src";
|
||||||
|
|
||||||
// Import injected libraries here!
|
// Import injected libraries here!
|
||||||
//import * as pdfcpuWrapper from "@stirling-pdf/shared-operations/wasm/pdfcpu/pdfcpu-wrapper-node.js";
|
//import * as pdfcpuWrapper from "@stirling-pdf/shared-operations/wasm/pdfcpu/pdfcpu-wrapper-node.js";
|
||||||
|
5
shared-operations/declarations/Operation.d.ts
vendored
Normal file
5
shared-operations/declarations/Operation.d.ts
vendored
Normal file
@ -0,0 +1,5 @@
|
|||||||
|
export interface Operation {
|
||||||
|
values: {id:any};
|
||||||
|
type: string;
|
||||||
|
operations?: Operation[];
|
||||||
|
}
|
5
shared-operations/declarations/PDF.d.ts
vendored
Normal file
5
shared-operations/declarations/PDF.d.ts
vendored
Normal file
@ -0,0 +1,5 @@
|
|||||||
|
export interface PDF {
|
||||||
|
originalFileName: string;
|
||||||
|
fileName: string;
|
||||||
|
buffer: Uint8Array;
|
||||||
|
}
|
@ -1,9 +1,4 @@
|
|||||||
|
import { Operation } from "../../declarations/Operation";
|
||||||
export interface Operation {
|
|
||||||
values: {id:any};
|
|
||||||
type: string;
|
|
||||||
operations?: Operation[];
|
|
||||||
}
|
|
||||||
|
|
||||||
export function organizeWaitOperations(operations: Operation[]) {
|
export function organizeWaitOperations(operations: Operation[]) {
|
||||||
|
|
@ -1,34 +1,17 @@
|
|||||||
import { organizeWaitOperations } from "./organizeWaitOperations.js";
|
import { organizeWaitOperations } from "./organizeWaitOperations.js";
|
||||||
|
import { Operation } from "../../declarations/Operation.js";
|
||||||
|
import { PDF } from "../../declarations/PDF.js";
|
||||||
|
|
||||||
/**
|
export async function * traverseOperations(operations: Operation[], input: PDF[] | PDF, Operations: AllOperations) {
|
||||||
* @typedef PDF
|
|
||||||
* @property {string} originalFileName
|
|
||||||
* @property {string} fileName
|
|
||||||
* @property {Uint8Array} buffer
|
|
||||||
*/
|
|
||||||
|
|
||||||
/**
|
|
||||||
*
|
|
||||||
* @param {JSON} operations
|
|
||||||
* @param {PDF|PDF[]} input
|
|
||||||
* @returns {}
|
|
||||||
*/
|
|
||||||
export async function * traverseOperations(operations, input, Operations) {
|
|
||||||
const waitOperations = organizeWaitOperations(operations);
|
const waitOperations = organizeWaitOperations(operations);
|
||||||
/** @type {PDF[]} */ let results = [];
|
let results: PDF[] = [];
|
||||||
yield* nextOperation(operations, input);
|
yield* nextOperation(operations, input);
|
||||||
return results;
|
return results;
|
||||||
|
|
||||||
/**
|
async function * nextOperation(operations: Operation[], input: PDF[] | PDF) {
|
||||||
*
|
|
||||||
* @param {JSON} operations
|
|
||||||
* @param {PDF|PDF[]} input
|
|
||||||
* @returns {undefined}
|
|
||||||
*/
|
|
||||||
async function * nextOperation(operations, input) {
|
|
||||||
if(Array.isArray(operations) && operations.length == 0) { // isEmpty
|
if(Array.isArray(operations) && operations.length == 0) { // isEmpty
|
||||||
if(Array.isArray(input)) {
|
if(Array.isArray(input)) {
|
||||||
console.log("operation done: " + input[0].fileName + input.length > 1 ? "+" : "");
|
console.log("operation done: " + input[0].fileName + (input.length > 1 ? "+" : ""));
|
||||||
results = results.concat(input);
|
results = results.concat(input);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
@ -44,13 +27,7 @@ export async function * traverseOperations(operations, input, Operations) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
async function * computeOperation(operation: Operation, input: PDF|PDF[]) {
|
||||||
*
|
|
||||||
* @param {JSON} operation
|
|
||||||
* @param {PDF|PDF[]} input
|
|
||||||
* @returns {undefined}
|
|
||||||
*/
|
|
||||||
async function * computeOperation(operation, input) {
|
|
||||||
yield "Starting: " + operation.type;
|
yield "Starting: " + operation.type;
|
||||||
switch (operation.type) {
|
switch (operation.type) {
|
||||||
case "done": // Skip this, because it is a valid node.
|
case "done": // Skip this, because it is a valid node.
|
Loading…
Reference in New Issue
Block a user