You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tdarr_Plugins/FlowPluginsTs/FlowHelpers/1.0.0/classicPlugins.ts

165 lines
5.2 KiB

import { promises as fs } from 'fs';
import {
getContainer, getFileName, getPluginWorkDir, getScanTypes,
} from './fileUtils';
import { IpluginInputArgs } from './interfaces/interfaces';
import { IFileObject } from './interfaces/synced/IFileObject';
export interface IrunClassicPlugin {
result:{
processFile: boolean,
handBrakeMode?: boolean,
handbrakeMode?: boolean,
FFmpegMode?: boolean,
ffmpegMode?: boolean,
cliToUse?: string,
custom?: {
cliPath?: string,
args: string[],
outputPath: string,
},
workerLog?: string,
transcodeSettingsLog?: string,
error?: string,
container: string,
preset: string,
};
cacheFilePath:string;
absolutePath:string;
}
export const runClassicPlugin = async (args:IpluginInputArgs, type:'filter'|'transcode'):Promise<IrunClassicPlugin> => {
const path = require('path');
const pluginSourceId = String(args.inputs.pluginSourceId);
const parts = pluginSourceId.split(':');
const pluginSource = parts[0];
const pluginId = parts[1];
const relativePluginPath = `../../../${pluginSource}/${pluginId}.js`;
const absolutePath = path.resolve(__dirname, relativePluginPath);
let classicPlugin;
let pluginSrcStr = '';
if (pluginSource === 'Community') {
classicPlugin = args.deps.importFresh(relativePluginPath);
pluginSrcStr = await fs.readFile(absolutePath, 'utf8');
} else {
// eslint-disable-next-line no-await-in-loop
const res = await args.deps.axiosMiddleware('api/v2/read-plugin', {
plugin: {
id: pluginId,
source: pluginSource,
},
});
classicPlugin = args.deps.requireFromString(res.pluginRaw, absolutePath);
pluginSrcStr = res.pluginRaw;
}
if (type === 'filter' && classicPlugin.details().Operation !== 'Filter') {
throw new Error(
`${'This plugin is meant for classic plugins that have '
+ 'Operation: Filter. This classic plugin has Operation: '}${classicPlugin.details().Operation}`
+ '. Please use the Run Classic Transcode Flow Plugin plugin instead.'
,
);
}
if (type !== 'filter' && classicPlugin.details().Operation === 'Filter') {
throw new Error(
`${'This plugin is meant for classic plugins that have '
+ 'Operation: Transcode. This classic plugin has Operation: '}${classicPlugin.details().Operation}`
+ 'Please use the Run Classic Filter Flow Plugin plugin instead.'
,
);
}
if (Array.isArray(classicPlugin.dependencies)) {
if (args.installClassicPluginDeps) {
args.jobLog(`Installing dependencies for ${pluginSourceId}`);
await args.installClassicPluginDeps(classicPlugin.dependencies);
} else {
args.jobLog(`Not installing dependencies for ${pluginSourceId}, please update Tdarr`);
}
} else {
args.jobLog(`No depedencies to install for ${pluginSourceId}`);
}
const container = getContainer(args.inputFileObj._id);
const cacheFilePath = `${getPluginWorkDir(args)}/${getFileName(args.inputFileObj._id)}.${container}`;
const scanTypes = getScanTypes([pluginSrcStr]);
let pluginInputFileObj:IFileObject;
let originalLibraryFile:IFileObject;
const inputFileScanArgs = {
_id: args.inputFileObj._id,
file: args.inputFileObj.file,
DB: args.inputFileObj.DB,
footprintId: args.inputFileObj.footprintId,
};
const originalLibraryFileScanArgs = {
_id: args.originalLibraryFile._id,
file: args.originalLibraryFile.file,
DB: args.originalLibraryFile.DB,
footprintId: args.originalLibraryFile.footprintId,
};
// added in 2.19.01
if (typeof args.scanIndividualFile !== 'undefined') {
args.jobLog('Scanning files using Node');
pluginInputFileObj = await args.scanIndividualFile(inputFileScanArgs, scanTypes);
originalLibraryFile = await args.scanIndividualFile(originalLibraryFileScanArgs, scanTypes);
} else {
args.jobLog('Scanning files using Server API');
pluginInputFileObj = await args.deps.axiosMiddleware('api/v2/scan-individual-file', {
file: inputFileScanArgs,
scanTypes,
});
originalLibraryFile = await args.deps.axiosMiddleware('api/v2/scan-individual-file', {
file: originalLibraryFileScanArgs,
scanTypes,
});
}
const otherArguments = {
handbrakePath: args.handbrakePath,
ffmpegPath: args.ffmpegPath,
mkvpropeditPath: args.mkvpropeditPath,
originalLibraryFile,
nodeHardwareType: args.nodeHardwareType,
pluginCycle: 0,
workerType: args.workerType,
version: args.config.version,
platform_arch_isdocker: args.platform_arch_isdocker,
cacheFilePath,
job: args.job,
};
const result = await classicPlugin.plugin(
pluginInputFileObj,
args.librarySettings,
args.inputs,
otherArguments,
);
if (result?.file?._id && args.inputFileObj._id !== result.file._id) {
// eslint-disable-next-line no-param-reassign
args.inputFileObj._id = result.file._id;
// eslint-disable-next-line no-param-reassign
args.inputFileObj.file = result.file.file;
args.jobLog(`File ID changed from ${args.inputFileObj._id} to ${result.file._id}`);
}
return {
result,
cacheFilePath,
absolutePath,
};
};