// 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 = new RazorSourceDocumentProperties(resourcePath, resourcePath);
            return new StringSourceDocument(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 new StringSourceDocument(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 ?? 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 = new RazorSourceDocumentProperties(filePath, relativePath);
        return new StringSourceDocument(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 new StringSourceDocument(content, encoding ?? Encoding.UTF8, properties);
    }

    private static string NormalizeNewLines(string content)
    {
        return Regex.Replace(content, "(?<!\r)\n", "\r\n", RegexOptions.None, TimeSpan.FromSeconds(10));
    }
}
