From 372de640cad14d9e7577ac851edf6b239c8d830c Mon Sep 17 00:00:00 2001
From: reven <john@reven.nz>
Date: Mon, 27 Dec 2021 11:12:23 +1300
Subject: [PATCH] added audio track set language node

---
 BasicNodes/BasicNodes.csproj                  | Bin 2164 -> 2164 bytes
 BasicNodes/File/CopyFile.cs                   |   8 +++
 BasicNodes/File/MoveFile.cs                   |   8 ++-
 BasicNodes/Functions/Function.cs              |   2 +-
 VideoNodes/VideoNodes.csproj                  | Bin 2274 -> 2274 bytes
 VideoNodes/VideoNodes.en.json                 |   8 +++
 .../VideoNodes/AudioTrackSetLanguage.cs       |  68 ++++++++++++++++++
 7 files changed, 92 insertions(+), 2 deletions(-)
 create mode 100644 VideoNodes/VideoNodes/AudioTrackSetLanguage.cs

diff --git a/BasicNodes/BasicNodes.csproj b/BasicNodes/BasicNodes.csproj
index c51fe385dc9077fcc2dbf7bbed18d705cac96f56..7c90d3090755195400d5c16d2f06aa7f32536470 100644
GIT binary patch
delta 24
dcmew&@I_$53r1#B27}438ACzzW@jctHUM#v2Xz1d

delta 24
dcmew&@I_$53r1!W2J^|U8ACzzW@jctHUM%32Y~<p

diff --git a/BasicNodes/File/CopyFile.cs b/BasicNodes/File/CopyFile.cs
index d5c209a6..3f90f9fd 100644
--- a/BasicNodes/File/CopyFile.cs
+++ b/BasicNodes/File/CopyFile.cs
@@ -58,6 +58,13 @@ namespace FileFlows.BasicNodes.File
             else
                 dest = Path.Combine(dest, new FileInfo(args.FileName).Name);
 
+            var fiDest = new FileInfo(dest);
+            var fiWorking = new FileInfo(args.WorkingFile);
+            if (string.IsNullOrEmpty(fiDest.Extension) == false && fiDest.Extension != fiWorking.Extension)
+            {
+                dest = dest.Substring(0, dest.LastIndexOf(".")) + fiWorking.Extension;
+            }
+
             var destDir = new FileInfo(dest).DirectoryName;
             args.CreateDirectoryIfNotExists(destDir ?? String.Empty);
 
@@ -67,6 +74,7 @@ namespace FileFlows.BasicNodes.File
                 dest = Path.Combine(destDir!, destFile);
             }
 
+            args.Logger?.ILog($"Copying file '{args.WorkingFile}' to '{dest}");
             // have to use file streams so we can report progress
             int bufferSize = 1024 * 1024;
 
diff --git a/BasicNodes/File/MoveFile.cs b/BasicNodes/File/MoveFile.cs
index f9744571..b0eecf6c 100644
--- a/BasicNodes/File/MoveFile.cs
+++ b/BasicNodes/File/MoveFile.cs
@@ -45,6 +45,12 @@ namespace FileFlows.BasicNodes.File
             else
                 dest = Path.Combine(dest, new FileInfo(args.FileName).Name);
 
+            var fiDest = new FileInfo(dest);
+            var fiWorking = new FileInfo(args.WorkingFile);
+            if (string.IsNullOrEmpty(fiDest.Extension) == false && fiDest.Extension != fiWorking.Extension)
+            {
+                dest = dest.Substring(0, dest.LastIndexOf(".")) + fiWorking.Extension;
+            }
 
             if (string.IsNullOrEmpty(DestinationFile) == false)
             {
@@ -52,7 +58,7 @@ namespace FileFlows.BasicNodes.File
                 dest = Path.Combine(new FileInfo(dest).DirectoryName!, destFile);
             }
 
-            var fiDest = new FileInfo(dest);
+            fiDest = new FileInfo(dest);
             var fiWorkingFile = new FileInfo(args.WorkingFile);
             if (fiDest.Extension != fiWorkingFile.Extension)
             {
diff --git a/BasicNodes/Functions/Function.cs b/BasicNodes/Functions/Function.cs
index 714e1c8d..f1a24235 100644
--- a/BasicNodes/Functions/Function.cs
+++ b/BasicNodes/Functions/Function.cs
@@ -21,7 +21,7 @@ namespace FileFlows.BasicNodes.Functions
         public new int Outputs { get; set; }
 
         [Required]
-        [DefaultValue("// Variables contain variables available to this node from previous nodes.\n// Logger lets you log messages to the flow output.\n\n// return 0 to complete the flow.\n// return -1 to signal an error in the flow\n// return 1+ to select which output node will be processed next\n\nif(Variables.FileSize === 0)\n\treturn -1;\n\nreturn 0;")]
+        [DefaultValue("// Variables contain variables available to this node from previous nodes.\n// Logger lets you log messages to the flow output.\n\n// return 0 to complete the flow.\n// return -1 to signal an error in the flow\n// return 1+ to select which output node will be processed next\n\nif(Variables.file.Size === 0)\n\treturn -1;\n\nreturn 0;")]
         [Code(2)]
         public string Code { get; set; }
 
diff --git a/VideoNodes/VideoNodes.csproj b/VideoNodes/VideoNodes.csproj
index c521de88f48cbc37add2b79f6e05ad3feb0ffa71..4bd7898d194a5b6e24898ddbf27edd6e2765c18c 100644
GIT binary patch
delta 24
dcmaDP_(*WWJtk&T27}2DnL<JIW=rOktN?B@2d@AC

delta 24
dcmaDP_(*WWJtk%o2J^`enL<JIW=rOktN?DN2fF|O

diff --git a/VideoNodes/VideoNodes.en.json b/VideoNodes/VideoNodes.en.json
index 635b1948..bc3147f6 100644
--- a/VideoNodes/VideoNodes.en.json
+++ b/VideoNodes/VideoNodes.en.json
@@ -10,6 +10,14 @@
             "Languages-Help": "The order of languages to sort the audio tracks by.  This sorting is done before the codec."
           }
         },
+        "AudioTrackSetLanguage": {
+          "Label": "Audio: Set Language",
+          "Description": "Allows you to set the language for any audio tracks that have no language set.  If the audio track does have a language set, it will be skipped.\n\nOutput 1: Audio Tracks were updated\nOutput 2: No audio tracks were needing to be updated",
+          "Fields": {
+            "Language": "Language",
+            "Language-Help": "The ISO 639-2 language code to use. \nhttps://en.wikipedia.org/wiki/List_of_ISO_639-2_codes"
+          }
+        },
         "VideoFile": {
           "Description": "An input video file that has had its VideoInformation read and can be processed"
         },
diff --git a/VideoNodes/VideoNodes/AudioTrackSetLanguage.cs b/VideoNodes/VideoNodes/AudioTrackSetLanguage.cs
new file mode 100644
index 00000000..9f3490c7
--- /dev/null
+++ b/VideoNodes/VideoNodes/AudioTrackSetLanguage.cs
@@ -0,0 +1,68 @@
+namespace FileFlows.VideoNodes
+{
+    using FileFlows.Plugin;
+    using FileFlows.Plugin.Attributes;
+    using System;
+    using System.Collections.Generic;
+    using System.ComponentModel.DataAnnotations;
+
+    public class AudioTrackSetLanguage : EncodingNode
+    {
+        public override int Outputs => 2;
+
+        public override string Icon => "fas fa-comment-dots";
+
+        [Required]
+        [Text(1)]
+        public string Language { get; set; }
+
+        public override int Execute(NodeParameters args)
+        {
+            try
+            {
+                VideoInfo videoInfo = GetVideoInfo(args);
+                if (videoInfo == null)
+                    return -1;
+
+                string ffmpegExe = GetFFMpegExe(args);
+                if (string.IsNullOrEmpty(ffmpegExe))
+                    return -1;
+
+                List<string> ffArgs = new List<string>();
+
+                int index = 0;
+                foreach(var at in videoInfo.AudioStreams)
+                {
+                    if (string.IsNullOrEmpty(at.Language))
+                    {
+                        ffArgs.Add($"-metadata:s:a:{index} language={Language.ToLower()}");
+                    }
+                    ++index;
+                }
+                if (ffArgs.Count == 0)
+                    return 2; // nothing to do 
+
+
+                ffArgs.Insert(0, $"-map 0 -c copy");
+
+                string ffArgsLine = string.Join(" ", ffArgs);
+
+                string extension = new FileInfo(args.WorkingFile).Extension;
+                if(extension.StartsWith("."))
+                    extension = extension.Substring(1);
+                args.Logger?.DLog("Working file: " + args.WorkingFile);
+                args.Logger?.DLog("Extension: " + extension);
+
+                if (Encode(args, ffmpegExe, ffArgsLine, extension) == false)
+                    return -1;
+
+                return 1;
+            }
+            catch (Exception ex)
+            {
+                args.Logger?.ELog("Failed processing VideoFile: " + ex.Message);
+                return -1;
+            }
+}
+    }
+}