// Copyright (c) .NET Foundation. All rights reserved.
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.

#nullable disable

using System.Collections.Generic;
using System.Collections.Immutable;
using System.Linq;
using NuGet.Common;
using NuGet.Frameworks;
using NuGet.Shared;

namespace NuGet.Commands
{
    public static class AssetTargetFallbackUtility
    {
        public static readonly string AssetTargetFallback = nameof(AssetTargetFallback);

        /// <summary>
        /// Throw if an invalid combination exists.
        /// </summary>
        public static void EnsureValidFallback(IEnumerable<NuGetFramework> packageTargetFallback, IEnumerable<NuGetFramework> assetTargetFallback, string filePath)
        {
            if (packageTargetFallback?.Any() == true && assetTargetFallback?.Any() == true)
            {
                var error = GetInvalidFallbackCombinationMessage(filePath);
                throw new RestoreCommandException(error);
            }
        }

        public static RestoreLogMessage GetInvalidFallbackCombinationMessage(string path)
        {
            var error = RestoreLogMessage.CreateError(NuGetLogCode.NU1003, Strings.Error_InvalidATF);
            error.ProjectPath = path;
            error.FilePath = path;

            return error;
        }

        /// <summary>
        /// Returns the fallback framework or the original.
        /// </summary>
        public static NuGetFramework GetFallbackFramework(NuGetFramework projectFramework, IEnumerable<NuGetFramework> packageTargetFallback, IEnumerable<NuGetFramework> assetTargetFallback)
        {
            if (assetTargetFallback?.Any() == true)
            {
                // AssetTargetFallback
                return new AssetTargetFallbackFramework(projectFramework, assetTargetFallback.AsList());
            }
            else if (packageTargetFallback?.Any() == true)
            {
                // PackageTargetFallback
                return new FallbackFramework(projectFramework, packageTargetFallback.AsList());
            }

            return projectFramework;
        }

        /// <summary>
        /// Update TargetFrameworkInformation properties.
        /// </summary>
        public static (NuGetFramework frameworkName, ImmutableArray<NuGetFramework> imports, bool assetTargetFallback, bool warn) GetFallbackFrameworkInformation(NuGetFramework originalFrameworkName, IEnumerable<NuGetFramework> packageTargetFallback, IEnumerable<NuGetFramework> assetTargetFallbackEnum)
        {
            // Update the framework appropriately
            var frameworkName = GetFallbackFramework(
                originalFrameworkName,
                packageTargetFallback,
                assetTargetFallbackEnum);

            ImmutableArray<NuGetFramework> imports = [];
            bool assetTargetFallback = false;
            bool warn = false;

            if (assetTargetFallbackEnum?.Any() == true)
            {
                // AssetTargetFallback
                imports = assetTargetFallbackEnum.ToImmutableArray();
                assetTargetFallback = true;
                warn = true;
            }
            else if (packageTargetFallback?.Any() == true)
            {
                // PackageTargetFallback
                imports = packageTargetFallback.ToImmutableArray();
            }

            return (frameworkName, imports, assetTargetFallback, warn);
        }
    }
}