// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
#nullable disable
 
using System;
using System.IO;
using System.Reflection;
using System.Text;
using System.Text.RegularExpressions;
 
namespace Microsoft.AspNetCore.Razor.Language;
 
public static class TestRazorSourceDocument
{
    public static RazorSourceDocument CreateResource(string resourcePath, Type type, Encoding encoding = null, bool normalizeNewLines = false)
    {
        return CreateResource(resourcePath, type.GetTypeInfo().Assembly, encoding, normalizeNewLines);
    }
 
    public static RazorSourceDocument CreateResource(string resourcePath, Assembly assembly, Encoding encoding = null, bool normalizeNewLines = false)
    {
        var file = TestFile.Create(resourcePath, assembly);
 
        using (var input = file.OpenRead())
        using (var reader = new StreamReader(input))
        {
            var content = reader.ReadToEnd();
            if (normalizeNewLines)
            {
                content = NormalizeNewLines(content);
            }
 
            var properties = RazorSourceDocumentProperties.Create(resourcePath, resourcePath);
            return RazorSourceDocument.Create(content, encoding ?? Encoding.UTF8, properties);
        }
    }
 
    public static RazorSourceDocument CreateResource(
        string path,
        Assembly assembly,
        Encoding encoding,
        RazorSourceDocumentProperties properties,
        bool normalizeNewLines = false)
    {
        var file = TestFile.Create(path, assembly);
 
        using (var input = file.OpenRead())
        using (var reader = new StreamReader(input))
        {
            var content = reader.ReadToEnd();
            if (normalizeNewLines)
            {
                content = NormalizeNewLines(content);
            }
 
            return RazorSourceDocument.Create(content, encoding ?? Encoding.UTF8, properties);
        }
    }
 
    public static MemoryStream CreateStreamContent(string content = "Hello, World!", Encoding encoding = null, bool normalizeNewLines = false)
    {
        var stream = new MemoryStream();
        encoding ??= Encoding.UTF8;
        using (var writer = new StreamWriter(stream, encoding, bufferSize: 1024, leaveOpen: true))
        {
            if (normalizeNewLines)
            {
                content = NormalizeNewLines(content);
            }
 
            writer.Write(content);
        }
 
        stream.Seek(0L, SeekOrigin.Begin);
 
        return stream;
    }
 
    public static RazorSourceDocument Create(
        string content = "Hello, world!",
        Encoding encoding = null,
        bool normalizeNewLines = false,
        string filePath = "test.cshtml",
        string relativePath = "test.cshtml")
    {
        if (normalizeNewLines)
        {
            content = NormalizeNewLines(content);
        }
 
        var properties = RazorSourceDocumentProperties.Create(filePath, relativePath);
        return RazorSourceDocument.Create(content, encoding ?? Encoding.UTF8, properties);
    }
 
    public static RazorSourceDocument Create(
        string content,
        RazorSourceDocumentProperties properties,
        Encoding encoding = null,
        bool normalizeNewLines = false)
    {
        if (normalizeNewLines)
        {
            content = NormalizeNewLines(content);
        }
 
        return RazorSourceDocument.Create(content, encoding ?? Encoding.UTF8, properties);
    }
 
    private static string NormalizeNewLines(string content)
    {
        return Regex.Replace(content, "(?<!\r)\n", "\r\n", RegexOptions.None, TimeSpan.FromSeconds(10));
    }
}