Merge pull request #528 from HaveAGitGat/Ensure-Audio-Stream
Add Ensure Audio Streammake-only-subtitle-default
HaveAGitGat
2 years ago
committed by
GitHub
commit
ce4f0696bd
@ -0,0 +1,175 @@
|
|||||||
|
"use strict";
|
||||||
|
Object.defineProperty(exports, "__esModule", { value: true });
|
||||||
|
exports.plugin = exports.details = void 0;
|
||||||
|
/* eslint no-plusplus: ["error", { "allowForLoopAfterthoughts": true }] */
|
||||||
|
var details = function () { return ({
|
||||||
|
name: 'Ensure Audio Stream',
|
||||||
|
description: 'Ensure that the file has an audio stream with set codec and channel count',
|
||||||
|
style: {
|
||||||
|
borderColor: '#6efefc',
|
||||||
|
},
|
||||||
|
tags: 'video',
|
||||||
|
isStartPlugin: false,
|
||||||
|
pType: '',
|
||||||
|
requiresVersion: '2.11.01',
|
||||||
|
sidebarPosition: -1,
|
||||||
|
icon: '',
|
||||||
|
inputs: [
|
||||||
|
{
|
||||||
|
name: 'audioEncoder',
|
||||||
|
type: 'string',
|
||||||
|
defaultValue: 'aac',
|
||||||
|
inputUI: {
|
||||||
|
type: 'dropdown',
|
||||||
|
options: [
|
||||||
|
'aac',
|
||||||
|
'ac3',
|
||||||
|
'eac3',
|
||||||
|
'dca',
|
||||||
|
'flac',
|
||||||
|
'mp2',
|
||||||
|
'libmp3lame',
|
||||||
|
'truehd',
|
||||||
|
],
|
||||||
|
},
|
||||||
|
tooltip: 'Enter the desired audio codec',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: 'language',
|
||||||
|
type: 'string',
|
||||||
|
defaultValue: 'en',
|
||||||
|
inputUI: {
|
||||||
|
type: 'text',
|
||||||
|
},
|
||||||
|
tooltip: 'Tdarr will check to see if the stream language tag includes the tag you specify.'
|
||||||
|
+ ' Case-insensitive. One tag only',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: 'channels',
|
||||||
|
type: 'number',
|
||||||
|
defaultValue: '2',
|
||||||
|
inputUI: {
|
||||||
|
type: 'dropdown',
|
||||||
|
options: [
|
||||||
|
'1',
|
||||||
|
'2',
|
||||||
|
'6',
|
||||||
|
'8',
|
||||||
|
],
|
||||||
|
},
|
||||||
|
tooltip: 'Enter the desired number of channels',
|
||||||
|
},
|
||||||
|
],
|
||||||
|
outputs: [
|
||||||
|
{
|
||||||
|
number: 1,
|
||||||
|
tooltip: 'Continue to next plugin',
|
||||||
|
},
|
||||||
|
],
|
||||||
|
}); };
|
||||||
|
exports.details = details;
|
||||||
|
var getHighest = function (first, second) {
|
||||||
|
// @ts-expect-error channels
|
||||||
|
if ((first === null || first === void 0 ? void 0 : first.channels) > (second === null || second === void 0 ? void 0 : second.channels)) {
|
||||||
|
return first;
|
||||||
|
}
|
||||||
|
return second;
|
||||||
|
};
|
||||||
|
var attemptMakeStream = function (_a) {
|
||||||
|
var args = _a.args, langTag = _a.langTag, streams = _a.streams, audioCodec = _a.audioCodec, audioEncoder = _a.audioEncoder, wantedChannelCount = _a.wantedChannelCount;
|
||||||
|
var langMatch = function (stream) {
|
||||||
|
var _a;
|
||||||
|
return ((langTag === 'und'
|
||||||
|
&& (stream.tags === undefined || stream.tags.language === undefined))
|
||||||
|
|| (((_a = stream === null || stream === void 0 ? void 0 : stream.tags) === null || _a === void 0 ? void 0 : _a.language) && stream.tags.language.toLowerCase().includes(langTag)));
|
||||||
|
};
|
||||||
|
// filter streams to only include audio streams with the specified lang tag
|
||||||
|
var streamsWithLangTag = streams.filter(function (stream) {
|
||||||
|
if (stream.codec_type === 'audio'
|
||||||
|
&& langMatch(stream)) {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
});
|
||||||
|
if (streamsWithLangTag.length === 0) {
|
||||||
|
args.jobLog("No streams with language tag ".concat(langTag, " found. Skipping \n"));
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
// get the stream with the highest channel count
|
||||||
|
var streamWithHighestChannel = streamsWithLangTag.reduce(getHighest);
|
||||||
|
var highestChannelCount = Number(streamWithHighestChannel.channels);
|
||||||
|
var targetChannels = 0;
|
||||||
|
if (wantedChannelCount <= highestChannelCount) {
|
||||||
|
targetChannels = wantedChannelCount;
|
||||||
|
args.jobLog("The wanted channel count ".concat(wantedChannelCount, " is <= than the")
|
||||||
|
+ " highest available channel count (".concat(streamWithHighestChannel.channels, "). \n"));
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
targetChannels = highestChannelCount;
|
||||||
|
args.jobLog("The wanted channel count ".concat(wantedChannelCount, " is higher than the")
|
||||||
|
+ " highest available channel count (".concat(streamWithHighestChannel.channels, "). \n"));
|
||||||
|
}
|
||||||
|
var hasStreamAlready = streams.filter(function (stream) {
|
||||||
|
if (stream.codec_type === 'audio'
|
||||||
|
&& langMatch(stream)
|
||||||
|
&& stream.codec_name === audioCodec
|
||||||
|
&& stream.channels === targetChannels) {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
});
|
||||||
|
if (hasStreamAlready.length > 0) {
|
||||||
|
args.jobLog("File already has ".concat(langTag, " stream in ").concat(audioEncoder, ", ").concat(targetChannels, " channels \n"));
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
args.jobLog("Adding ".concat(langTag, " stream in ").concat(audioEncoder, ", ").concat(targetChannels, " channels \n"));
|
||||||
|
var streamCopy = JSON.parse(JSON.stringify(streamWithHighestChannel));
|
||||||
|
streamCopy.index = streams.length;
|
||||||
|
streamCopy.outputArgs.push('-c:{outputIndex}', audioEncoder);
|
||||||
|
streamCopy.outputArgs.push('-ac', "".concat(targetChannels));
|
||||||
|
// eslint-disable-next-line no-param-reassign
|
||||||
|
args.variables.ffmpegCommand.shouldProcess = true;
|
||||||
|
streams.push(streamCopy);
|
||||||
|
return true;
|
||||||
|
};
|
||||||
|
// eslint-disable-next-line @typescript-eslint/no-unused-vars
|
||||||
|
var plugin = function (args) {
|
||||||
|
var lib = require('../../../../../methods/lib')();
|
||||||
|
// eslint-disable-next-line @typescript-eslint/no-unused-vars,no-param-reassign
|
||||||
|
args.inputs = lib.loadDefaultValues(args.inputs, details);
|
||||||
|
var audioEncoder = String(args.inputs.audioEncoder);
|
||||||
|
var langTag = String(args.inputs.language).toLowerCase();
|
||||||
|
var wantedChannelCount = Number(args.inputs.channels);
|
||||||
|
var streams = args.variables.ffmpegCommand.streams;
|
||||||
|
var audioCodec = audioEncoder;
|
||||||
|
if (audioEncoder === 'dca') {
|
||||||
|
audioCodec = 'dts';
|
||||||
|
}
|
||||||
|
if (audioEncoder === 'libmp3lame') {
|
||||||
|
audioCodec = 'mp3';
|
||||||
|
}
|
||||||
|
var addedOrExists = attemptMakeStream({
|
||||||
|
args: args,
|
||||||
|
langTag: langTag,
|
||||||
|
streams: streams,
|
||||||
|
audioCodec: audioCodec,
|
||||||
|
audioEncoder: audioEncoder,
|
||||||
|
wantedChannelCount: wantedChannelCount,
|
||||||
|
});
|
||||||
|
if (!addedOrExists) {
|
||||||
|
attemptMakeStream({
|
||||||
|
args: args,
|
||||||
|
langTag: 'und',
|
||||||
|
streams: streams,
|
||||||
|
audioCodec: audioCodec,
|
||||||
|
audioEncoder: audioEncoder,
|
||||||
|
wantedChannelCount: wantedChannelCount,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
return {
|
||||||
|
outputFileObj: args.inputFileObj,
|
||||||
|
outputNumber: 1,
|
||||||
|
variables: args.variables,
|
||||||
|
};
|
||||||
|
};
|
||||||
|
exports.plugin = plugin;
|
||||||
@ -0,0 +1,224 @@
|
|||||||
|
import {
|
||||||
|
IffmpegCommandStream,
|
||||||
|
IpluginDetails,
|
||||||
|
IpluginInputArgs,
|
||||||
|
IpluginOutputArgs,
|
||||||
|
} from '../../../../FlowHelpers/1.0.0/interfaces/interfaces';
|
||||||
|
|
||||||
|
/* eslint no-plusplus: ["error", { "allowForLoopAfterthoughts": true }] */
|
||||||
|
const details = (): IpluginDetails => ({
|
||||||
|
name: 'Ensure Audio Stream',
|
||||||
|
description: 'Ensure that the file has an audio stream with set codec and channel count',
|
||||||
|
style: {
|
||||||
|
borderColor: '#6efefc',
|
||||||
|
},
|
||||||
|
tags: 'video',
|
||||||
|
isStartPlugin: false,
|
||||||
|
pType: '',
|
||||||
|
requiresVersion: '2.11.01',
|
||||||
|
sidebarPosition: -1,
|
||||||
|
icon: '',
|
||||||
|
inputs: [
|
||||||
|
{
|
||||||
|
name: 'audioEncoder',
|
||||||
|
type: 'string',
|
||||||
|
defaultValue: 'aac',
|
||||||
|
inputUI: {
|
||||||
|
type: 'dropdown',
|
||||||
|
options: [
|
||||||
|
'aac',
|
||||||
|
'ac3',
|
||||||
|
'eac3',
|
||||||
|
'dca',
|
||||||
|
'flac',
|
||||||
|
'mp2',
|
||||||
|
'libmp3lame',
|
||||||
|
'truehd',
|
||||||
|
],
|
||||||
|
},
|
||||||
|
tooltip:
|
||||||
|
'Enter the desired audio codec',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: 'language',
|
||||||
|
type: 'string',
|
||||||
|
defaultValue: 'en',
|
||||||
|
inputUI: {
|
||||||
|
type: 'text',
|
||||||
|
},
|
||||||
|
tooltip:
|
||||||
|
'Tdarr will check to see if the stream language tag includes the tag you specify.'
|
||||||
|
+ ' Case-insensitive. One tag only',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: 'channels',
|
||||||
|
type: 'number',
|
||||||
|
defaultValue: '2',
|
||||||
|
inputUI: {
|
||||||
|
type: 'dropdown',
|
||||||
|
options: [
|
||||||
|
'1',
|
||||||
|
'2',
|
||||||
|
'6',
|
||||||
|
'8',
|
||||||
|
],
|
||||||
|
},
|
||||||
|
tooltip:
|
||||||
|
'Enter the desired number of channels',
|
||||||
|
},
|
||||||
|
],
|
||||||
|
outputs: [
|
||||||
|
{
|
||||||
|
number: 1,
|
||||||
|
tooltip: 'Continue to next plugin',
|
||||||
|
},
|
||||||
|
],
|
||||||
|
});
|
||||||
|
|
||||||
|
const getHighest = (first: IffmpegCommandStream, second: IffmpegCommandStream) => {
|
||||||
|
// @ts-expect-error channels
|
||||||
|
if (first?.channels > second?.channels) {
|
||||||
|
return first;
|
||||||
|
}
|
||||||
|
return second;
|
||||||
|
};
|
||||||
|
|
||||||
|
const attemptMakeStream = ({
|
||||||
|
args,
|
||||||
|
langTag,
|
||||||
|
streams,
|
||||||
|
audioCodec,
|
||||||
|
audioEncoder,
|
||||||
|
wantedChannelCount,
|
||||||
|
}: {
|
||||||
|
args: IpluginInputArgs,
|
||||||
|
langTag: string
|
||||||
|
streams: IffmpegCommandStream[],
|
||||||
|
audioCodec: string,
|
||||||
|
audioEncoder: string,
|
||||||
|
wantedChannelCount: number,
|
||||||
|
}): boolean => {
|
||||||
|
const langMatch = (stream: IffmpegCommandStream) => (
|
||||||
|
(langTag === 'und'
|
||||||
|
&& (stream.tags === undefined || stream.tags.language === undefined))
|
||||||
|
|| (stream?.tags?.language && stream.tags.language.toLowerCase().includes(langTag)
|
||||||
|
)
|
||||||
|
);
|
||||||
|
|
||||||
|
// filter streams to only include audio streams with the specified lang tag
|
||||||
|
const streamsWithLangTag = streams.filter((stream) => {
|
||||||
|
if (
|
||||||
|
stream.codec_type === 'audio'
|
||||||
|
&& langMatch(stream)
|
||||||
|
) {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
});
|
||||||
|
|
||||||
|
if (streamsWithLangTag.length === 0) {
|
||||||
|
args.jobLog(`No streams with language tag ${langTag} found. Skipping \n`);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
// get the stream with the highest channel count
|
||||||
|
const streamWithHighestChannel = streamsWithLangTag.reduce(getHighest);
|
||||||
|
const highestChannelCount = Number(streamWithHighestChannel.channels);
|
||||||
|
|
||||||
|
let targetChannels = 0;
|
||||||
|
if (wantedChannelCount <= highestChannelCount) {
|
||||||
|
targetChannels = wantedChannelCount;
|
||||||
|
args.jobLog(`The wanted channel count ${wantedChannelCount} is <= than the`
|
||||||
|
+ ` highest available channel count (${streamWithHighestChannel.channels}). \n`);
|
||||||
|
} else {
|
||||||
|
targetChannels = highestChannelCount;
|
||||||
|
args.jobLog(`The wanted channel count ${wantedChannelCount} is higher than the`
|
||||||
|
+ ` highest available channel count (${streamWithHighestChannel.channels}). \n`);
|
||||||
|
}
|
||||||
|
|
||||||
|
const hasStreamAlready = streams.filter((stream) => {
|
||||||
|
if (
|
||||||
|
stream.codec_type === 'audio'
|
||||||
|
&& langMatch(stream)
|
||||||
|
&& stream.codec_name === audioCodec
|
||||||
|
&& stream.channels === targetChannels
|
||||||
|
) {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
});
|
||||||
|
|
||||||
|
if (hasStreamAlready.length > 0) {
|
||||||
|
args.jobLog(`File already has ${langTag} stream in ${audioEncoder}, ${targetChannels} channels \n`);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
args.jobLog(`Adding ${langTag} stream in ${audioEncoder}, ${targetChannels} channels \n`);
|
||||||
|
|
||||||
|
const streamCopy: IffmpegCommandStream = JSON.parse(JSON.stringify(streamWithHighestChannel));
|
||||||
|
streamCopy.index = streams.length;
|
||||||
|
streamCopy.outputArgs.push('-c:{outputIndex}', audioEncoder);
|
||||||
|
streamCopy.outputArgs.push('-ac', `${targetChannels}`);
|
||||||
|
|
||||||
|
// eslint-disable-next-line no-param-reassign
|
||||||
|
args.variables.ffmpegCommand.shouldProcess = true;
|
||||||
|
|
||||||
|
streams.push(streamCopy);
|
||||||
|
|
||||||
|
return true;
|
||||||
|
};
|
||||||
|
|
||||||
|
// 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);
|
||||||
|
|
||||||
|
const audioEncoder = String(args.inputs.audioEncoder);
|
||||||
|
const langTag = String(args.inputs.language).toLowerCase();
|
||||||
|
const wantedChannelCount = Number(args.inputs.channels);
|
||||||
|
|
||||||
|
const { streams } = args.variables.ffmpegCommand;
|
||||||
|
|
||||||
|
let audioCodec = audioEncoder;
|
||||||
|
|
||||||
|
if (audioEncoder === 'dca') {
|
||||||
|
audioCodec = 'dts';
|
||||||
|
}
|
||||||
|
|
||||||
|
if (audioEncoder === 'libmp3lame') {
|
||||||
|
audioCodec = 'mp3';
|
||||||
|
}
|
||||||
|
|
||||||
|
const addedOrExists = attemptMakeStream({
|
||||||
|
args,
|
||||||
|
langTag,
|
||||||
|
streams,
|
||||||
|
audioCodec,
|
||||||
|
audioEncoder,
|
||||||
|
wantedChannelCount,
|
||||||
|
});
|
||||||
|
|
||||||
|
if (!addedOrExists) {
|
||||||
|
attemptMakeStream({
|
||||||
|
args,
|
||||||
|
langTag: 'und',
|
||||||
|
streams,
|
||||||
|
audioCodec,
|
||||||
|
audioEncoder,
|
||||||
|
wantedChannelCount,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
return {
|
||||||
|
outputFileObj: args.inputFileObj,
|
||||||
|
outputNumber: 1,
|
||||||
|
variables: args.variables,
|
||||||
|
};
|
||||||
|
};
|
||||||
|
export {
|
||||||
|
details,
|
||||||
|
plugin,
|
||||||
|
};
|
||||||
Loading…
Reference in new issue