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

using Microsoft.Build.Framework;
using Newtonsoft.Json;
using Newtonsoft.Json.Linq;
using NuGet.Frameworks;
using NuGet.Versioning;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using PropertyNames = NuGet.Client.ManagedCodeConventions.PropertyNames;

namespace Microsoft.DotNet.Build.Tasks.Packaging;

public class ValidateFrameworkPackage : ValidationTask
{
    [Required]
    public string Framework
    {
        get;
        set;
    }
    
    public string Runtime
    {
        get;
        set;
    }

    public override bool Execute()
    {
        InitializeValidationTask();

        var fx = NuGetFramework.Parse(Framework);

        var testAssets = GetTestAssets(fx, Runtime);
        var testAssetsByName = testAssets.Where(a => Path.GetExtension(a.PackagePath) == ".dll")
                                         .ToDictionary(a => Path.GetFileNameWithoutExtension(a.PackagePath), a => a);

        var permittedInbox = GetSuppressionValues(Suppression.PermitInbox) ?? new HashSet<string>();

        foreach (var testAssetByName in testAssetsByName)
        {
            var name = testAssetByName.Key;
            var testAsset = testAssetByName.Value;

            var logMissingInbox = permittedInbox.Contains(name) ? 
                new Action<string>(s => Log.LogMessage(MessageImportance.Low, $"Suppressed: {s}")) :
                new Action<string>(s => Log.LogError(s));

            PackageInfo packageInfo;
            if (!_index.Packages.TryGetValue(name, out packageInfo))
            {
                logMissingInbox($"File {name} was included framework package {_report.Id}/{_report.Version} but that file is missing from package index {string.Join(";", _index.IndexSources)}.  Please add it with appropriate {nameof(PackageInfo.InboxOn)} entry for {Framework} or suppress this message with {nameof(Suppression.PermitInbox)} suppression.");
                continue;
            }

            if (!packageInfo.InboxOn.IsInbox(fx, testAsset.Version))
            {
                logMissingInbox($"File {name}, version {testAsset.Version} was included framework package {_report.Id}/{_report.Version} but that version is not considered inbox in package index {string.Join(";", _index.IndexSources)}.  Please add it with appropriate {nameof(PackageInfo.InboxOn)} entry for {Framework} or suppress this message with {nameof(Suppression.PermitInbox)} suppression.");
                continue;
            }
        }

        var permittedMissingInbox = GetSuppressionValues(Suppression.PermitMissingInbox) ?? new HashSet<string>();
        
        var missingInboxAssemblies = _index.Packages.Where(packageInfo => packageInfo.Value.InboxOn.IsAnyVersionInbox(fx) && !testAssetsByName.ContainsKey(packageInfo.Key));

        foreach(var missingInboxAssembly in missingInboxAssemblies)
        {
            var logMissingPackage = permittedMissingInbox.Contains(missingInboxAssembly.Key) ?
                new Action<string>(s => Log.LogMessage(MessageImportance.Low, $"Suppressed: {s}")) :
                new Action<string>(s => Log.LogError(s));

            logMissingPackage($"File {missingInboxAssembly.Key}.dll is marked as inbox for framework {Framework} but was missing from framework package {_report.Id}/{_report.Version}.  Either add the file or update {nameof(PackageInfo.InboxOn)} entry in {string.Join(";", _index.IndexSources)}.   This may be suppressed with {nameof(Suppression.PermitMissingInbox)} suppression");
        }
        
        return !Log.HasLoggedErrors;
    }

    private IEnumerable<PackageAsset> GetTestAssets(NuGetFramework fx, string runtimeId)
    {

        var targetKey = string.IsNullOrEmpty(runtimeId) ? fx.ToString() : $"{fx}/{runtimeId}";

        Target target = null;
        if (!_report.Targets.TryGetValue(targetKey, out target))
        {
            Log.LogError($"Could not find target {targetKey} in {ReportFile}.");
            return Enumerable.Empty<PackageAsset>();
        }

        return string.IsNullOrEmpty(Runtime) ? target.CompileAssets : target.RuntimeAssets;
    }
}