From 16afcf09bfe7fdaf3b68b78ce08163dd6e6db036 Mon Sep 17 00:00:00 2001
From: HaveAGitGat <43864057+HaveAGitGat@users.noreply.github.com>
Date: Mon, 22 Aug 2022 02:05:55 +0100
Subject: [PATCH] Add 10 bit video filter

---
 .../Tdarr_Plugin_00td_filter_10_bit_video.js  | 56 ++++++++++++++++++
 .../Tdarr_Plugin_00td_filter_10_bit_video.js  | 58 +++++++++++++++++++
 2 files changed, 114 insertions(+)
 create mode 100644 Community/Tdarr_Plugin_00td_filter_10_bit_video.js
 create mode 100644 tests/Community/Tdarr_Plugin_00td_filter_10_bit_video.js

diff --git a/Community/Tdarr_Plugin_00td_filter_10_bit_video.js b/Community/Tdarr_Plugin_00td_filter_10_bit_video.js
new file mode 100644
index 0000000..89ff8d1
--- /dev/null
+++ b/Community/Tdarr_Plugin_00td_filter_10_bit_video.js
@@ -0,0 +1,56 @@
+const details = () => ({
+  id: 'Tdarr_Plugin_00td_filter_10_bit_video',
+  Stage: 'Pre-processing',
+  Name: 'Filter 10 bit video',
+  Type: 'Video',
+  Operation: 'Filter',
+  Description: 'Allow/disallow 10 bit video to be processed.',
+  Version: '1.00',
+  Tags: 'filter',
+  Inputs: [
+    {
+      name: 'process10BitVideo',
+      type: 'boolean',
+      defaultValue: false,
+      inputUI: {
+        type: 'dropdown',
+        options: [
+          'false',
+          'true',
+        ],
+      },
+      tooltip: 'Set true to allow 10 bit video to be processed.',
+    },
+  ],
+});
+
+// eslint-disable-next-line no-unused-vars
+const plugin = (file, librarySettings, inputs, otherArguments) => {
+  const lib = require('../methods/lib')();
+  // eslint-disable-next-line no-unused-vars,no-param-reassign
+  inputs = lib.loadDefaultValues(inputs, details);
+  const response = {
+    processFile: true,
+    infoLog: 'File will be processed.',
+  };
+
+  try {
+    const streams10Bit = file.ffProbeData.streams.filter((row) => row.profile === 'Main 10');
+    if (inputs.process10BitVideo === false && streams10Bit.length > 0) {
+      response.processFile = false;
+      response.infoLog = 'File video is 10 bit but 10 bit video processing is not allowed. Skipping plugins.';
+    } else if (inputs.process10BitVideo === true && streams10Bit.length > 0) {
+      response.infoLog = 'File video is 10 bit and 10 bit video processing is allowed. Continuing to plugins';
+    } else if (streams10Bit.length === 0) {
+      response.infoLog += 'File is not 10 bit.';
+    }
+  } catch (err) {
+    // eslint-disable-next-line no-console
+    console.log(err);
+  }
+
+  return response;
+};
+
+module.exports.details = details;
+module.exports.plugin = plugin;
diff --git a/tests/Community/Tdarr_Plugin_00td_filter_10_bit_video.js b/tests/Community/Tdarr_Plugin_00td_filter_10_bit_video.js
new file mode 100644
index 0000000..d5f9cb7
--- /dev/null
+++ b/tests/Community/Tdarr_Plugin_00td_filter_10_bit_video.js
@@ -0,0 +1,58 @@
+/* eslint max-len: 0 */
+const _ = require('lodash');
+const run = require('../helpers/run');
+
+const tests = [
+  {
+    input: {
+      file: require('../sampleData/media/sampleH264_1.json'),
+      librarySettings: {},
+      inputs: {},
+      otherArguments: {},
+    },
+    output: {
+      processFile: true,
+      infoLog: 'File will be processed.File is not 10 bit.',
+    },
+  },
+  {
+    input: {
+      file: (() => {
+        const file = _.cloneDeep(require('../sampleData/media/sampleH265_1.json'));
+        file.ffProbeData.streams[0].profile = 'Main 10';
+        return file;
+      })(),
+      librarySettings: {},
+      inputs: {
+        process10BitVideo: false,
+      },
+      otherArguments: {},
+    },
+    output: {
+      processFile: false,
+      infoLog: 'File video is 10 bit but 10 bit video processing is not allowed. Skipping plugins.',
+    },
+  },
+
+  {
+    input: {
+      file: (() => {
+        const file = _.cloneDeep(require('../sampleData/media/sampleH265_1.json'));
+        file.ffProbeData.streams[0].profile = 'Main 10';
+        return file;
+      })(),
+      librarySettings: {},
+      inputs: {
+        process10BitVideo: true,
+      },
+      otherArguments: {},
+    },
+    output: {
+      processFile: true,
+      infoLog: 'File video is 10 bit and 10 bit video processing is allowed. Continuing to plugins',
+    },
+  },
+
+];
+
+run(tests);