82 lines
2.3 KiB
TypeScript
82 lines
2.3 KiB
TypeScript
/* eslint-disable no-await-in-loop, no-restricted-syntax */
|
|
import { Page, PageScreenshotOptions } from 'playwright';
|
|
import path from 'path';
|
|
|
|
import { EventEmitter } from 'events';
|
|
import {
|
|
Where, What, PageState, Workflow, WorkflowFile,
|
|
ParamType, SelectorArray, CustomFunctions,
|
|
} from './types/workflow';
|
|
|
|
import { operators, meta } from './types/logic';
|
|
import { arrayToObject } from './utils/utils';
|
|
import Concurrency from './utils/concurrency';
|
|
import Preprocessor from './preprocessor';
|
|
import log, { Level } from './utils/logger';
|
|
|
|
/**
|
|
* Defines optional intepreter options (passed in constructor)
|
|
*/
|
|
interface InterpreterOptions {
|
|
maxRepeats: number;
|
|
maxConcurrency: number;
|
|
serializableCallback: (output: any) => (void | Promise<void>);
|
|
binaryCallback: (output: any, mimeType: string) => (void | Promise<void>);
|
|
debug: boolean;
|
|
debugChannel: Partial<{
|
|
activeId: Function,
|
|
debugMessage: Function,
|
|
}>
|
|
}
|
|
|
|
/**
|
|
* Class for running the Smart Workflows.
|
|
*/
|
|
export default class Interpreter extends EventEmitter {
|
|
private workflow: Workflow;
|
|
|
|
private initializedWorkflow: Workflow | null;
|
|
|
|
private options: InterpreterOptions;
|
|
|
|
private concurrency : Concurrency;
|
|
|
|
private stopper : Function | null = null;
|
|
|
|
private log : typeof log;
|
|
|
|
constructor(workflow: WorkflowFile, options?: Partial<InterpreterOptions>) {
|
|
super();
|
|
this.workflow = workflow.workflow;
|
|
this.initializedWorkflow = null;
|
|
this.options = {
|
|
maxRepeats: 5,
|
|
maxConcurrency: 5,
|
|
serializableCallback: (data) => { log(JSON.stringify(data), Level.WARN); },
|
|
binaryCallback: () => { log('Received binary data, thrashing them.', Level.WARN); },
|
|
debug: false,
|
|
debugChannel: {},
|
|
...options,
|
|
};
|
|
this.concurrency = new Concurrency(this.options.maxConcurrency);
|
|
this.log = (...args) => log(...args);
|
|
|
|
const error = Preprocessor.validateWorkflow(workflow);
|
|
if (error) {
|
|
throw (error);
|
|
}
|
|
|
|
if (this.options.debugChannel?.debugMessage) {
|
|
const oldLog = this.log;
|
|
// @ts-ignore
|
|
this.log = (...args: Parameters<typeof oldLog>) => {
|
|
if (args[1] !== Level.LOG) {
|
|
this.options.debugChannel.debugMessage!(typeof args[0] === 'string' ? args[0] : args[0].message);
|
|
}
|
|
oldLog(...args);
|
|
};
|
|
}
|
|
}
|
|
|
|
|
|
} |