AI-Studio/app/MindWork AI Studio/Tools/PandocProcessBuilder.cs

using System.Diagnostics;
using System.Reflection;

using AIStudio.Tools.Metadata;
using AIStudio.Tools.Services;

using SharedTools;

namespace AIStudio.Tools;

public sealed class PandocProcessBuilder
{
    private static readonly Assembly ASSEMBLY = Assembly.GetExecutingAssembly();
    private static readonly MetaDataArchitectureAttribute META_DATA_ARCH = ASSEMBLY.GetCustomAttribute<MetaDataArchitectureAttribute>()!;

    // Use runtime detection instead of metadata to ensure correct RID on dev machines:
    private static readonly RID CPU_ARCHITECTURE = RIDExtensions.GetCurrentRID();
    private static readonly RID METADATA_ARCHITECTURE = META_DATA_ARCH.Architecture.ToRID();
    private static readonly ILogger LOGGER = Program.LOGGER_FACTORY.CreateLogger(nameof(PandocProcessBuilder));

    // Tracks whether the first log has been written to avoid log spam on repeated calls:
    private static bool HAS_LOGGED_ONCE;

    private string? providedInputFile;
    private string? providedOutputFile;
    private string? providedInputFormat;
    private string? providedOutputFormat;
    private bool useStandaloneMode;
    
    private readonly List<string> additionalArguments = new();
    
    private PandocProcessBuilder()
    {
    }
    
    public static PandocProcessBuilder Create() => new();
    
    public PandocProcessBuilder WithInputFile(string inputFile)
    {
        this.providedInputFile = inputFile;
        return this;
    }
    
    public PandocProcessBuilder WithOutputFile(string outputFile)
    {
        this.providedOutputFile = outputFile;
        return this;
    }
    
    public PandocProcessBuilder WithInputFormat(string inputFormat)
    {
        this.providedInputFormat = inputFormat;
        return this;
    }
    
    public PandocProcessBuilder WithOutputFormat(string outputFormat)
    {
        this.providedOutputFormat = outputFormat;
        return this;
    }
    
    public PandocProcessBuilder AddArgument(string argument)
    {
        this.additionalArguments.Add(argument);
        return this;
    }

    public PandocProcessBuilder UseStandaloneMode()
    {
        this.useStandaloneMode = true;
        return this;
    }
    
    public async Task<PandocPreparedProcess> BuildAsync(RustService rustService)
    {
        var pandocExecutable = await PandocExecutablePath(rustService);
        var startInfo = new ProcessStartInfo
        {
            FileName = pandocExecutable.Executable,
            RedirectStandardOutput = true,
            RedirectStandardError = true,
            UseShellExecute = false,
            CreateNoWindow = true
        };

        // Use argument tokens instead of a single command string so paths with spaces
        // or Unicode characters are passed to Pandoc unchanged on all platforms.
        if (this.useStandaloneMode)
            startInfo.ArgumentList.Add("--standalone");

        if (!string.IsNullOrWhiteSpace(this.providedInputFile))
            startInfo.ArgumentList.Add(this.providedInputFile);

        if (!string.IsNullOrWhiteSpace(this.providedInputFormat))
        {
            startInfo.ArgumentList.Add("-f");
            startInfo.ArgumentList.Add(this.providedInputFormat);
        }

        if (!string.IsNullOrWhiteSpace(this.providedOutputFormat))
        {
            startInfo.ArgumentList.Add("-t");
            startInfo.ArgumentList.Add(this.providedOutputFormat);
        }

        foreach (var additionalArgument in this.additionalArguments)
        {
            if (!string.IsNullOrWhiteSpace(additionalArgument))
                startInfo.ArgumentList.Add(additionalArgument);
        }

        if (!string.IsNullOrWhiteSpace(this.providedOutputFile))
        {
            startInfo.ArgumentList.Add("-o");
            startInfo.ArgumentList.Add(this.providedOutputFile);
        }

        return new(startInfo, pandocExecutable.IsLocalInstallation);
    }
    
    /// <summary>
    /// Returns the path to the pandoc executable.
    /// </summary>
    /// <remarks>
    /// Any local installation of pandoc will be preferred over the system-wide installation.
    /// When a local installation is found, its absolute path will be returned. In case no local
    /// installation is found, the name of the pandoc executable will be returned.
    /// </remarks>
    /// <param name="rustService">Global rust service to access file system and data dir.</param>
    /// <returns>Path to the pandoc executable.</returns>
    private static async Task<PandocExecutable> PandocExecutablePath(RustService rustService)
    {
        //
        // Determine if we should log (only on the first call):
        //
        var shouldLog = !HAS_LOGGED_ONCE;

        try
        {
            //
            // Log a warning if the runtime-detected RID differs from the metadata RID.
            // This can happen on dev machines where the metadata.txt contains stale values.
            // We always use the runtime-detected RID for correct behavior.
            //
            if (shouldLog && CPU_ARCHITECTURE != METADATA_ARCHITECTURE)
            {
                LOGGER.LogWarning(
                    "Runtime-detected RID '{RuntimeRID}' differs from metadata RID '{MetadataRID}'. Using runtime-detected RID. This is expected on dev machines where metadata.txt may be outdated.",
                    CPU_ARCHITECTURE.ToUserFriendlyName(),
                    METADATA_ARCHITECTURE.ToUserFriendlyName());
            }

            //
            // First, we try to find the pandoc executable in the data directory.
            // Any local installation should be preferred over the system-wide installation.
            //
            var localInstallationRootDirectory = await Pandoc.GetPandocDataFolder(rustService);

            //
            // Check if the data directory path is valid:
            //
            if (string.IsNullOrWhiteSpace(localInstallationRootDirectory))
            {
                if (shouldLog)
                    LOGGER.LogWarning("The local data directory path is empty or null. Cannot search for local Pandoc installation.");
            }
            else if (!Directory.Exists(localInstallationRootDirectory))
            {
                if (shouldLog)
                    LOGGER.LogWarning("The local Pandoc installation directory does not exist: '{LocalInstallationRootDirectory}'.", localInstallationRootDirectory);
            }
            else
            {
                //
                // The directory exists, search for the pandoc executable:
                //
                var executableName = PandocExecutableName;
                if (shouldLog)
                    LOGGER.LogInformation("Searching for Pandoc executable '{ExecutableName}' in: '{LocalInstallationRootDirectory}'.", executableName, localInstallationRootDirectory);

                try
                {
                    //
                    // First, check the root directory itself:
                    //
                    var rootExecutablePath = Path.Combine(localInstallationRootDirectory, executableName);
                    if (File.Exists(rootExecutablePath))
                    {
                        if (shouldLog)
                            LOGGER.LogInformation("Found local Pandoc installation at the root path: '{Path}'.", rootExecutablePath);

                        HAS_LOGGED_ONCE = true;
                        return new(rootExecutablePath, true);
                    }

                    //
                    // Then, search all subdirectories:
                    //
                    var subdirectories = Directory.GetDirectories(localInstallationRootDirectory, "*", SearchOption.AllDirectories);
                    foreach (var subdirectory in subdirectories)
                    {
                        var pandocPath = Path.Combine(subdirectory, executableName);
                        if (File.Exists(pandocPath))
                        {
                            if (shouldLog)
                                LOGGER.LogInformation("Found local Pandoc installation at: '{Path}'.", pandocPath);

                            HAS_LOGGED_ONCE = true;
                            return new(pandocPath, true);
                        }
                    }

                    if (shouldLog)
                        LOGGER.LogWarning("No Pandoc executable found in local installation directory or its subdirectories.");
                }
                catch (Exception ex)
                {
                    if (shouldLog)
                        LOGGER.LogWarning(ex, "Error while searching for a local Pandoc installation in: '{LocalInstallationRootDirectory}'.", localInstallationRootDirectory);
                }
            }

            foreach (var candidate in SystemPandocExecutableCandidates(PandocExecutableName))
            {
                if (!File.Exists(candidate))
                    continue;

                if (shouldLog)
                    LOGGER.LogInformation("Found system Pandoc installation at: '{Path}'.", candidate);

                return new(candidate, false);
            }

            //
            // When no local installation was found, we assume that the pandoc executable is in the system PATH:
            //
            if (shouldLog)
                LOGGER.LogWarning("Falling back to system PATH for the Pandoc executable: '{ExecutableName}'.", PandocExecutableName);
            
            return new(PandocExecutableName, false);
        }
        finally
        {
            HAS_LOGGED_ONCE = true;
        }
    }
    
    /// <summary>
    /// Reads the os platform to determine the used executable name.
    /// </summary>
    public static string PandocExecutableName => CPU_ARCHITECTURE is RID.WIN_ARM64 or RID.WIN_X64 ? "pandoc.exe" : "pandoc";

    private static IEnumerable<string> SystemPandocExecutableCandidates(string executableName)
    {
        var candidates = new List<string>();

        switch (CPU_ARCHITECTURE)
        {
            case RID.WIN_X64 or RID.WIN_ARM64:
                AddCandidate(candidates, Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData), "Pandoc", executableName);
                AddCandidate(candidates, Environment.GetFolderPath(Environment.SpecialFolder.ProgramFiles), "Pandoc", executableName);
                AddCandidate(candidates, Environment.GetFolderPath(Environment.SpecialFolder.ProgramFilesX86), "Pandoc", executableName);
                break;

            case RID.OSX_X64 or RID.OSX_ARM64:
                AddCandidate(candidates, "/opt/homebrew/bin", executableName);
                AddCandidate(candidates, "/usr/local/bin", executableName);
                AddCandidate(candidates, "/usr/bin", executableName);
                break;

            case RID.LINUX_X64 or RID.LINUX_ARM64:
                AddCandidate(candidates, "/usr/local/bin", executableName);
                AddCandidate(candidates, "/usr/bin", executableName);
                AddCandidate(candidates, "/snap/bin", executableName);

                var homeDirectory = Environment.GetFolderPath(Environment.SpecialFolder.UserProfile);
                AddCandidate(candidates, homeDirectory, ".local", "bin", executableName);
                break;
        }

        foreach (var pathDirectory in GetPathDirectories())
            AddCandidate(candidates, pathDirectory, executableName);

        var comparer = CPU_ARCHITECTURE is RID.WIN_X64 or RID.WIN_ARM64
            ? StringComparer.OrdinalIgnoreCase
            : StringComparer.Ordinal;
        return candidates.Distinct(comparer);
    }

    private static IEnumerable<string> GetPathDirectories()
    {
        var pathValue = Environment.GetEnvironmentVariable("PATH");
        if (string.IsNullOrWhiteSpace(pathValue))
            yield break;

        foreach (var pathDirectory in pathValue.Split(Path.PathSeparator, StringSplitOptions.RemoveEmptyEntries | StringSplitOptions.TrimEntries))
            yield return pathDirectory;
    }

    private static void AddCandidate(List<string> candidates, params string[] pathParts)
    {
        if (pathParts.Any(string.IsNullOrWhiteSpace))
            return;

        candidates.Add(Path.Combine(pathParts));
    }
}