FileFlowsPlugins/AudioNodes/Helpers/IFFmpegHelper.cs

using System.Diagnostics;

namespace FileFlows.AudioNodes.Helpers;

/// <summary>
/// Interface for FFmpeg Helper
/// </summary>
public interface IFFmpegHelper
{
    /// <summary>
    /// Reads the info for a file
    /// </summary>
    /// <param name="file">the path to the file to read</param>
    /// <returns>the file output</returns>
    Result<string> ReadFFmpeg(string file);
    
    /// <summary>
    /// Reads the info for a file from FFprobe
    /// </summary>
    /// <param name="file">the path to the file to read</param>
    /// <returns>the file output</returns>
    Result<string> ReadFFprobe(string file);
}

/// <summary>
/// FFmpeg Helper
/// </summary>
/// <param name="ffmpeg">the FFmpeg executable</param>
/// <param name="ffprobe">the FFprobe executable</param>
public class FFmpegHelper(string ffmpeg, string ffprobe) : IFFmpegHelper
{
    /// <inheritdoc />
    public Result<string> ReadFFmpeg(string file)
    {
        try
        {
            using (var process = new Process())
            {
                process.StartInfo = new ProcessStartInfo
                {
                    FileName = ffmpeg,
                    UseShellExecute = false,
                    RedirectStandardOutput = true,
                    RedirectStandardError = true,
                    CreateNoWindow = true,
                    ArgumentList = {
                        "-hide_banner", 
                        "-i",
                        file
                    }
                };
                process.Start();

                bool exited = process.WaitForExit(60000);
                if (exited == false)
                {
                    process.Kill();
                    string pkOutput = process.StandardError.ReadToEnd()?.EmptyAsNull() ?? process.StandardOutput.ReadToEnd();
                    return Result<string>.Fail("Process timed out." + Environment.NewLine + pkOutput);
                }

                // we use error here, since we're not specify an output file, FFmpeg will report it as an error, but we don't care
                string output = process.StandardOutput.ReadToEnd();
                string error = process.StandardError.ReadToEnd();

                return output?.EmptyAsNull() ?? error;
            }
        }
        catch (Exception ex)
        {
            return Result<string>.Fail($"An error occurred: {ex.Message}");
        }
    }

    /// <inheritdoc />
    public Result<string> ReadFFprobe(string file)
    {
        try
        {
            using var process = new Process();
            process.StartInfo = new ProcessStartInfo
            {
                FileName = ffprobe,
                UseShellExecute = false,
                RedirectStandardOutput = true,
                RedirectStandardError = true,
                CreateNoWindow = true,
                ArgumentList = {
                    "-v", "error",
                    "-select_streams", "a:0",
                    "-show_format",
                    "-of", "json",
                    file
                }
            };
            process.Start();

            bool exited = process.WaitForExit(60000);
            if (exited == false)
            {
                process.Kill();
                string pkOutput = process.StandardError.ReadToEnd()?.EmptyAsNull() ?? process.StandardOutput.ReadToEnd();
                return Result<string>.Fail("Process timed out." + Environment.NewLine + pkOutput);
            }

            string output = process.StandardOutput.ReadToEnd();
            string error = process.StandardError.ReadToEnd();

            if (string.IsNullOrEmpty(error) == false)
                return Result<string>.Fail($"Failed reading ffmpeg info: {error}");

            return output;
        }
        catch (Exception ex)
        {
            return Result<string>.Fail($"An error occurred: {ex.Message}");
        }
    }
}