// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.

using System.Globalization;
using Microsoft.AspNetCore.Razor.Hosting;
using Microsoft.AspNetCore.Razor.Language;

namespace Umbraco.Cms.DevelopmentMode.Backoffice.InMemoryAuto;

/*
 * This is a clone of Microsofts implementation.
 * https://github.com/dotnet/aspnetcore/blob/main/src/Mvc/Mvc.Razor.RuntimeCompilation/src/ChecksumValidator.cs
 * The purpose of this class is to check if compiled views has changed and needs to be recompiled.
 */
internal static class ChecksumValidator
{
    public static bool IsRecompilationSupported(RazorCompiledItem item)
    {
        if (item == null)
        {
            throw new ArgumentNullException(nameof(item));
        }

        // A Razor item only supports recompilation if its primary source file has a checksum.
        //
        // Other files (view imports) may or may not have existed at the time of compilation,
        // so we may not have checksums for them.
        IReadOnlyList<IRazorSourceChecksumMetadata> checksums = item.GetChecksumMetadata();
        return checksums.Any(c => string.Equals(item.Identifier, c.Identifier, StringComparison.OrdinalIgnoreCase));
    }

    // Validates that we can use an existing precompiled view by comparing checksums with files on
    // disk.
    public static bool IsItemValid(RazorProjectFileSystem fileSystem, RazorCompiledItem item)
    {
        if (fileSystem == null)
        {
            throw new ArgumentNullException(nameof(fileSystem));
        }

        if (item == null)
        {
            throw new ArgumentNullException(nameof(item));
        }

        IReadOnlyList<IRazorSourceChecksumMetadata> checksums = item.GetChecksumMetadata();

        // The checksum that matches 'Item.Identity' in this list is significant. That represents the main file.
        //
        // We don't really care about the validation unless the main file exists. This is because we expect
        // most sites to have some _ViewImports in common location. That means that in the case you're
        // using views from a 3rd party library, you'll always have **some** conflicts.
        //
        // The presence of the main file with the same content is a very strong signal that you're in a
        // development scenario.
        IRazorSourceChecksumMetadata? primaryChecksum = checksums
            .FirstOrDefault(c => string.Equals(item.Identifier, c.Identifier, StringComparison.OrdinalIgnoreCase));
        if (primaryChecksum == null)
        {
            // No primary checksum, assume valid.
            return true;
        }

        RazorProjectItem projectItem = fileSystem.GetItem(primaryChecksum.Identifier, fileKind: null);
        if (!projectItem.Exists)
        {
            // Main file doesn't exist - assume valid.
            return true;
        }

        var sourceDocument = RazorSourceDocument.ReadFrom(projectItem);
        if (!string.Equals(sourceDocument.GetChecksumAlgorithm(), primaryChecksum.ChecksumAlgorithm) ||
            !ChecksumsEqual(primaryChecksum.Checksum, sourceDocument.GetChecksum()))
        {
            // Main file exists, but checksums not equal.
            return false;
        }

        for (var i = 0; i < checksums.Count; i++)
        {
            IRazorSourceChecksumMetadata checksum = checksums[i];
            if (string.Equals(item.Identifier, checksum.Identifier, StringComparison.OrdinalIgnoreCase))
            {
                // Ignore primary checksum on this pass.
                continue;
            }

            RazorProjectItem importItem = fileSystem.GetItem(checksum.Identifier, fileKind: null);
            if (!importItem.Exists)
            {
                // Import file doesn't exist - assume invalid.
                return false;
            }

            sourceDocument = RazorSourceDocument.ReadFrom(importItem);
            if (!string.Equals(sourceDocument.GetChecksumAlgorithm(), checksum.ChecksumAlgorithm) ||
                !ChecksumsEqual(checksum.Checksum, sourceDocument.GetChecksum()))
            {
                // Import file exists, but checksums not equal.
                return false;
            }
        }

        return true;
    }

    private static bool ChecksumsEqual(string checksum, byte[] bytes)
    {
        if (bytes.Length * 2 != checksum.Length)
        {
            return false;
        }

        for (var i = 0; i < bytes.Length; i++)
        {
            var text = bytes[i].ToString("x2", CultureInfo.InvariantCulture);
            if (checksum[i * 2] != text[0] || checksum[(i * 2) + 1] != text[1])
            {
                return false;
            }
        }

        return true;
    }
}
