From d1d5d111420471d795ea3bf096a5239df3fe27b9 Mon Sep 17 00:00:00 2001 From: HaveAGitGat <43864057+HaveAGitGat@users.noreply.github.com> Date: Mon, 27 Nov 2023 07:46:28 +0000 Subject: [PATCH] HandBrake Custom Arguments --- .../handbrakeCustomArguments/1.0.0/index.js | 2 +- .../handbrakeCustomArguments/2.0.0/index.js | 223 ++++++++++++++++++ .../handbrakeCustomArguments/1.0.0/index.ts | 2 +- .../handbrakeCustomArguments/2.0.0/index.ts | 194 +++++++++++++++ 4 files changed, 419 insertions(+), 2 deletions(-) create mode 100644 FlowPlugins/CommunityFlowPlugins/handbrake/handbrakeCustomArguments/2.0.0/index.js create mode 100644 FlowPluginsTs/CommunityFlowPlugins/handbrake/handbrakeCustomArguments/2.0.0/index.ts diff --git a/FlowPlugins/CommunityFlowPlugins/handbrake/handbrakeCustomArguments/1.0.0/index.js b/FlowPlugins/CommunityFlowPlugins/handbrake/handbrakeCustomArguments/1.0.0/index.js index 31d6ac1f2..d1eedf894 100644 --- a/FlowPlugins/CommunityFlowPlugins/handbrake/handbrakeCustomArguments/1.0.0/index.js +++ b/FlowPlugins/CommunityFlowPlugins/handbrake/handbrakeCustomArguments/1.0.0/index.js @@ -92,7 +92,7 @@ var details = function () { return ({ 'mpeg', ], }, - tooltip: 'Specify HandBrake arguments', + tooltip: 'Specify output container', }, ], outputs: [ diff --git a/FlowPlugins/CommunityFlowPlugins/handbrake/handbrakeCustomArguments/2.0.0/index.js b/FlowPlugins/CommunityFlowPlugins/handbrake/handbrakeCustomArguments/2.0.0/index.js new file mode 100644 index 000000000..052e1f15f --- /dev/null +++ b/FlowPlugins/CommunityFlowPlugins/handbrake/handbrakeCustomArguments/2.0.0/index.js @@ -0,0 +1,223 @@ +"use strict"; +var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) { + function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); } + return new (P || (P = Promise))(function (resolve, reject) { + function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } } + function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } } + function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); } + step((generator = generator.apply(thisArg, _arguments || [])).next()); + }); +}; +var __generator = (this && this.__generator) || function (thisArg, body) { + var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g; + return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g; + function verb(n) { return function (v) { return step([n, v]); }; } + function step(op) { + if (f) throw new TypeError("Generator is already executing."); + while (g && (g = 0, op[0] && (_ = 0)), _) try { + if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t; + if (y = 0, t) op = [op[0] & 2, t.value]; + switch (op[0]) { + case 0: case 1: t = op; break; + case 4: _.label++; return { value: op[1], done: false }; + case 5: _.label++; y = op[1]; op = [0]; continue; + case 7: op = _.ops.pop(); _.trys.pop(); continue; + default: + if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; } + if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; } + if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; } + if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; } + if (t[2]) _.ops.pop(); + _.trys.pop(); continue; + } + op = body.call(thisArg, _); + } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; } + if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true }; + } +}; +Object.defineProperty(exports, "__esModule", { value: true }); +exports.plugin = exports.details = void 0; +var fs_1 = require("fs"); +var cliUtils_1 = require("../../../../FlowHelpers/1.0.0/cliUtils"); +var fileUtils_1 = require("../../../../FlowHelpers/1.0.0/fileUtils"); +/* eslint no-plusplus: ["error", { "allowForLoopAfterthoughts": true }] */ +var details = function () { return ({ + name: 'HandBrake Custom Arguments', + description: 'HandBrake Custom Arguments', + style: { + borderColor: 'green', + }, + tags: '', + isStartPlugin: false, + pType: '', + requiresVersion: '2.14.01', + sidebarPosition: -1, + icon: '', + inputs: [ + { + label: 'Use JSON Preset', + name: 'useJsonPreset', + type: 'boolean', + defaultValue: 'false', + inputUI: { + type: 'dropdown', + options: [ + 'false', + 'true', + ], + }, + tooltip: 'Specify whether to use a JSON preset or not', + }, + { + label: 'Custom Arguments', + name: 'customArguments', + type: 'string', + defaultValue: '-Z "Fast 1080p30" --all-subtitles', + inputUI: { + type: 'text', + displayConditions: { + logic: 'AND', + sets: [ + { + logic: 'AND', + inputs: [ + { + name: 'useJsonPreset', + value: 'true', + condition: '!==', + }, + ], + }, + ], + }, + }, + tooltip: 'Specify HandBrake arguments', + }, + { + label: 'Paste Contents of .json File Here', + name: 'jsonPreset', + type: 'string', + defaultValue: '', + inputUI: { + type: 'textarea', + style: { + height: '100px', + }, + displayConditions: { + logic: 'AND', + sets: [ + { + logic: 'AND', + inputs: [ + { + name: 'useJsonPreset', + value: 'true', + condition: '===', + }, + ], + }, + ], + }, + }, + tooltip: 'Paste a HandBrake JSON preset here.', + }, + { + label: 'Container', + name: 'container', + type: 'string', + defaultValue: 'mkv', + inputUI: { + type: 'dropdown', + options: [ + 'original', + 'mkv', + 'mp4', + 'm4v', + 'avi', + 'mov', + 'mpg', + 'mpeg', + ], + }, + tooltip: 'Specify output container', + }, + ], + outputs: [ + { + number: 1, + tooltip: 'Continue to next plugin', + }, + ], +}); }; +exports.details = details; +// eslint-disable-next-line @typescript-eslint/no-unused-vars +var plugin = function (args) { return __awaiter(void 0, void 0, void 0, function () { + var lib, customArguments, useJsonPreset, presetString, container, outputFilePath, cliArgs, presetPath, preset, cli, res; + return __generator(this, function (_a) { + switch (_a.label) { + case 0: + lib = require('../../../../../methods/lib')(); + // eslint-disable-next-line @typescript-eslint/no-unused-vars,no-param-reassign + args.inputs = lib.loadDefaultValues(args.inputs, details); + customArguments = String(args.inputs.customArguments); + useJsonPreset = args.inputs.useJsonPreset; + presetString = String(args.inputs.jsonPreset); + container = String(args.inputs.container); + if (container === 'original') { + container = (0, fileUtils_1.getContainer)(args.inputFileObj._id); + } + outputFilePath = "".concat((0, fileUtils_1.getPluginWorkDir)(args), "/").concat((0, fileUtils_1.getFileName)(args.inputFileObj._id), ".").concat(container); + cliArgs = [ + '-i', + "".concat(args.inputFileObj._id), + '-o', + "".concat(outputFilePath), + ]; + presetPath = "".concat(args.workDir, "/preset.json"); + if (!useJsonPreset) return [3 /*break*/, 2]; + preset = JSON.parse(presetString); + return [4 /*yield*/, fs_1.promises.writeFile(presetPath, JSON.stringify(preset, null, 2))]; + case 1: + _a.sent(); + cliArgs.push('--preset-import-file'); + cliArgs.push(presetPath); + cliArgs.push('-Z'); + cliArgs.push(preset.PresetList[0].PresetName); + return [3 /*break*/, 3]; + case 2: + cliArgs.push.apply(cliArgs, args.deps.parseArgsStringToArgv(customArguments, '', '')); + _a.label = 3; + case 3: + args.updateWorker({ + CLIType: args.handbrakePath, + preset: cliArgs.join(' '), + }); + cli = new cliUtils_1.CLI({ + cli: args.handbrakePath, + spawnArgs: cliArgs, + spawnOpts: {}, + jobLog: args.jobLog, + outputFilePath: outputFilePath, + inputFileObj: args.inputFileObj, + logFullCliOutput: args.logFullCliOutput, + updateWorker: args.updateWorker, + }); + return [4 /*yield*/, cli.runCli()]; + case 4: + res = _a.sent(); + if (res.cliExitCode !== 0) { + args.jobLog('Running HandBrake failed'); + throw new Error('Running HandBrake failed'); + } + args.logOutcome('tSuc'); + return [2 /*return*/, { + outputFileObj: { + _id: outputFilePath, + }, + outputNumber: 1, + variables: args.variables, + }]; + } + }); +}); }; +exports.plugin = plugin; diff --git a/FlowPluginsTs/CommunityFlowPlugins/handbrake/handbrakeCustomArguments/1.0.0/index.ts b/FlowPluginsTs/CommunityFlowPlugins/handbrake/handbrakeCustomArguments/1.0.0/index.ts index 76dae1d7b..1b97c8721 100644 --- a/FlowPluginsTs/CommunityFlowPlugins/handbrake/handbrakeCustomArguments/1.0.0/index.ts +++ b/FlowPluginsTs/CommunityFlowPlugins/handbrake/handbrakeCustomArguments/1.0.0/index.ts @@ -59,7 +59,7 @@ const details = ():IpluginDetails => ({ 'mpeg', ], }, - tooltip: 'Specify HandBrake arguments', + tooltip: 'Specify output container', }, ], outputs: [ diff --git a/FlowPluginsTs/CommunityFlowPlugins/handbrake/handbrakeCustomArguments/2.0.0/index.ts b/FlowPluginsTs/CommunityFlowPlugins/handbrake/handbrakeCustomArguments/2.0.0/index.ts new file mode 100644 index 000000000..1ca9ee627 --- /dev/null +++ b/FlowPluginsTs/CommunityFlowPlugins/handbrake/handbrakeCustomArguments/2.0.0/index.ts @@ -0,0 +1,194 @@ +import { promises as fs } from 'fs'; +import { CLI } from '../../../../FlowHelpers/1.0.0/cliUtils'; +import { + IpluginDetails, + IpluginInputArgs, + IpluginOutputArgs, +} from '../../../../FlowHelpers/1.0.0/interfaces/interfaces'; +import { getContainer, getFileName, getPluginWorkDir } from '../../../../FlowHelpers/1.0.0/fileUtils'; + +/* eslint no-plusplus: ["error", { "allowForLoopAfterthoughts": true }] */ +const details = (): IpluginDetails => ({ + name: 'HandBrake Custom Arguments', + description: 'HandBrake Custom Arguments', + style: { + borderColor: 'green', + }, + tags: '', + isStartPlugin: false, + pType: '', + requiresVersion: '2.14.01', + sidebarPosition: -1, + icon: '', + inputs: [ + { + label: 'Use JSON Preset', + name: 'useJsonPreset', + type: 'boolean', + defaultValue: 'false', + inputUI: { + type: 'dropdown', + options: [ + 'false', + 'true', + ], + }, + tooltip: 'Specify whether to use a JSON preset or not', + }, + { + label: 'Custom Arguments', + name: 'customArguments', + type: 'string', + defaultValue: '-Z "Fast 1080p30" --all-subtitles', + inputUI: { + type: 'text', + displayConditions: { + logic: 'AND', + sets: [ + { + logic: 'AND', + inputs: [ + { + name: 'useJsonPreset', + value: 'true', + condition: '!==', + }, + ], + }, + ], + }, + }, + tooltip: 'Specify HandBrake arguments', + }, + { + label: 'Paste Contents of .json File Here', + name: 'jsonPreset', + type: 'string', + defaultValue: '', + inputUI: { + type: 'textarea', + style: { + height: '100px', + }, + displayConditions: { + logic: 'AND', + sets: [ + { + logic: 'AND', + inputs: [ + { + name: 'useJsonPreset', + value: 'true', + condition: '===', + }, + ], + }, + ], + }, + }, + tooltip: 'Paste a HandBrake JSON preset here.', + }, + { + label: 'Container', + name: 'container', + type: 'string', + defaultValue: 'mkv', + inputUI: { + type: 'dropdown', + options: [ + 'original', + 'mkv', + 'mp4', + 'm4v', + 'avi', + 'mov', + 'mpg', + 'mpeg', + ], + }, + tooltip: 'Specify output container', + }, + ], + outputs: [ + { + number: 1, + tooltip: 'Continue to next plugin', + }, + ], +}); + +// eslint-disable-next-line @typescript-eslint/no-unused-vars +const plugin = async (args: IpluginInputArgs): Promise => { + const lib = require('../../../../../methods/lib')(); + // eslint-disable-next-line @typescript-eslint/no-unused-vars,no-param-reassign + args.inputs = lib.loadDefaultValues(args.inputs, details); + + const customArguments = String(args.inputs.customArguments); + const { useJsonPreset } = args.inputs; + const presetString = String(args.inputs.jsonPreset); + + let container = String(args.inputs.container); + + if (container === 'original') { + container = getContainer(args.inputFileObj._id); + } + + const outputFilePath = `${getPluginWorkDir(args)}/${getFileName(args.inputFileObj._id)}.${container}`; + + const cliArgs = [ + '-i', + `${args.inputFileObj._id}`, + '-o', + `${outputFilePath}`, + ]; + + const presetPath = `${args.workDir}/preset.json`; + + if (useJsonPreset) { + const preset = JSON.parse(presetString); + await fs.writeFile(presetPath, JSON.stringify(preset, null, 2)); + cliArgs.push('--preset-import-file'); + cliArgs.push(presetPath); + cliArgs.push('-Z'); + cliArgs.push(preset.PresetList[0].PresetName); + } else { + cliArgs.push(...args.deps.parseArgsStringToArgv(customArguments, '', '')); + } + + args.updateWorker({ + CLIType: args.handbrakePath, + preset: cliArgs.join(' '), + }); + + const cli = new CLI({ + cli: args.handbrakePath, + spawnArgs: cliArgs, + spawnOpts: {}, + jobLog: args.jobLog, + outputFilePath, + inputFileObj: args.inputFileObj, + logFullCliOutput: args.logFullCliOutput, + updateWorker: args.updateWorker, + }); + + const res = await cli.runCli(); + + if (res.cliExitCode !== 0) { + args.jobLog('Running HandBrake failed'); + throw new Error('Running HandBrake failed'); + } + + args.logOutcome('tSuc'); + + return { + outputFileObj: { + _id: outputFilePath, + }, + outputNumber: 1, + variables: args.variables, + }; +}; +export { + details, + plugin, +};