// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using System.Text;
using System.Text.RegularExpressions;
 
namespace Microsoft.Extensions.AI.Templates.Tests;
 
internal static class VerifyScrubbers
{
    /// <summary>
    /// Removes content after "Details: ".
    /// </summary>
    internal static void ScrubDetails(this StringBuilder output)
    {
        output.ScrubByRegex("(Details: )([^\\r\\n]*)", $"Details: %DETAILS%");
    }
 
    /// <summary>
    /// Removes table header delimiter.
    /// </summary>
    internal static void ScrubTableHeaderDelimiter(this StringBuilder output)
    {
        output.ScrubByRegex("---[- ]*", "%TABLE HEADER DELIMITER%");
    }
 
    /// <summary>
    /// Replaces Windows newlines (CRLF) with Unix style newlines (LF).
    /// </summary>
    /// <param name="output"></param>
    internal static StringBuilder UnixifyNewlines(this StringBuilder output)
    {
        return output.Replace("\r\n", "\n");
    }
 
    /// <summary>
    /// Replaces Windows Directory separator char (\) with Unix Directory separator char (/).
    /// </summary>
    /// <param name="output"></param>
    internal static string UnixifyDirSeparators(this string output)
    {
        return output.Replace('\\', '/');
    }
 
    /// <summary>
    /// Replaces content matching <paramref name="pattern"/> with <paramref name="replacement"/>.
    /// </summary>
    internal static void ScrubByRegex(this StringBuilder output, string pattern, string replacement, RegexOptions regexOptions = RegexOptions.None)
    {
        string finalOutput = Regex.Replace(output.ToString(), pattern, replacement, regexOptions);
        output.Clear();
        output.Append(finalOutput);
    }
 
    /// <summary>
    /// Replaces content matching <paramref name="textToReplace"/> with <paramref name="replacement"/>.
    /// </summary>
    internal static void ScrubAndReplace(this StringBuilder output, string textToReplace, string replacement)
    {
        string finalOutput = output.ToString().Replace(textToReplace, replacement);
        output.Clear();
        output.Append(finalOutput);
    }
}