Tdarr_Plugins/FlowPluginsTs/CommunityFlowPlugins/ffmpegCommand/ffmpegCommandSetVideoBitrate/1.0.0/index.ts

import {
  IpluginDetails,
  IpluginInputArgs,
  IpluginOutputArgs,
} from '../../../../FlowHelpers/1.0.0/interfaces/interfaces';
import { getFfType } from '../../../../FlowHelpers/1.0.0/fileUtils';
import { checkFfmpegCommandInit } from '../../../../FlowHelpers/1.0.0/interfaces/flowUtils';

/* eslint no-plusplus: ["error", { "allowForLoopAfterthoughts": true }] */
const details = (): IpluginDetails => ({
  name: 'Set Video Bitrate',
  description: 'Set Video Bitrate',
  style: {
    borderColor: '#6efefc',
  },
  tags: 'video',
  isStartPlugin: false,
  pType: '',
  requiresVersion: '2.11.01',
  sidebarPosition: -1,
  icon: '',
  inputs: [
    {
      label: 'Use % of Input Bitrate',
      name: 'useInputBitrate',
      type: 'boolean',
      defaultValue: 'false',
      inputUI: {
        type: 'switch',
      },
      tooltip: 'Specify whether to use a % of input bitrate as the output bitrate',
    },

    {
      label: 'Target Bitrate %',
      name: 'targetBitratePercent',
      type: 'string',
      defaultValue: '50',
      inputUI: {
        type: 'text',
        displayConditions: {
          logic: 'AND',
          sets: [
            {
              logic: 'AND',
              inputs: [
                {
                  name: 'useInputBitrate',
                  value: 'true',
                  condition: '===',
                },
              ],
            },
          ],
        },
      },
      tooltip: 'Specify the target bitrate as a % of the input bitrate',
    },
    {
      label: 'Fallback Bitrate',
      name: 'fallbackBitrate',
      type: 'string',
      defaultValue: '4000',
      inputUI: {
        type: 'text',
        displayConditions: {
          logic: 'AND',
          sets: [
            {
              logic: 'AND',
              inputs: [
                {
                  name: 'useInputBitrate',
                  value: 'true',
                  condition: '===',
                },
              ],
            },
          ],
        },
      },
      tooltip: 'Specify fallback bitrate in kbps if input bitrate is not available',
    },
    {
      label: 'Bitrate',
      name: 'bitrate',
      type: 'string',
      defaultValue: '5000',
      inputUI: {
        type: 'text',
        displayConditions: {
          logic: 'AND',
          sets: [
            {
              logic: 'AND',
              inputs: [
                {
                  name: 'useInputBitrate',
                  value: 'true',
                  condition: '!==',
                },
              ],
            },
          ],
        },
      },
      tooltip: 'Specify bitrate in kbps',
    },
  ],
  outputs: [
    {
      number: 1,
      tooltip: 'Continue to next plugin',
    },
  ],
});

// eslint-disable-next-line @typescript-eslint/no-unused-vars
const plugin = (args: IpluginInputArgs): IpluginOutputArgs => {
  const lib = require('../../../../../methods/lib')();
  // eslint-disable-next-line @typescript-eslint/no-unused-vars,no-param-reassign
  args.inputs = lib.loadDefaultValues(args.inputs, details);

  checkFfmpegCommandInit(args);

  const { useInputBitrate } = args.inputs;
  const targetBitratePercent = String(args.inputs.targetBitratePercent);
  const fallbackBitrate = String(args.inputs.fallbackBitrate);
  const bitrate = String(args.inputs.bitrate);

  args.variables.ffmpegCommand.streams.forEach((stream) => {
    if (stream.codec_type === 'video') {
      const ffType = getFfType(stream.codec_type);
      if (useInputBitrate) {
        args.jobLog('Attempting to use % of input bitrate as output bitrate');
        // check if input bitrate is available
        const mediainfoIndex = stream.index + 1;

        let inputBitrate = args?.inputFileObj?.mediaInfo?.track?.[mediainfoIndex]?.BitRate;
        if (inputBitrate) {
          args.jobLog(`Found input bitrate: ${inputBitrate}`);
          // @ts-expect-error type
          inputBitrate = parseInt(inputBitrate, 10) / 1000;
          const targetBitrate = (inputBitrate * (parseInt(targetBitratePercent, 10) / 100));
          args.jobLog(`Setting video bitrate as ${targetBitrate}k`);
          stream.outputArgs.push(`-b:${ffType}:{outputTypeIndex}`, `${targetBitrate}k`);
        } else {
          args.jobLog(`Unable to find input bitrate, setting fallback bitrate as ${fallbackBitrate}k`);
          stream.outputArgs.push(`-b:${ffType}:{outputTypeIndex}`, `${fallbackBitrate}k`);
        }
      } else {
        args.jobLog(`Using fixed bitrate. Setting video bitrate as ${bitrate}k`);
        stream.outputArgs.push(`-b:${ffType}:{outputTypeIndex}`, `${bitrate}k`);
      }
    }
  });

  return {
    outputFileObj: args.inputFileObj,
    outputNumber: 1,
    variables: args.variables,
  };
};
export {
  details,
  plugin,
};