// 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 Microsoft.DotNet.Cli.Commands.Workload.Install.WorkloadInstallRecords;
using Microsoft.DotNet.Cli.ToolPackage;
using Microsoft.Extensions.EnvironmentAbstractions;
using Microsoft.NET.Sdk.WorkloadManifestReader;

namespace Microsoft.DotNet.Cli.Commands.Workload.Install;

internal interface IInstaller : IWorkloadManifestInstaller
{
    int ExitCode { get; }

    WorkloadSet GetWorkloadSetContents(string workloadVersion);

    void InstallWorkloads(IEnumerable<WorkloadId> workloadIds, SdkFeatureBand sdkFeatureBand, ITransactionContext transactionContext, DirectoryPath? offlineCache = null);

    void RepairWorkloads(IEnumerable<WorkloadId> workloadIds, SdkFeatureBand sdkFeatureBand, DirectoryPath? offlineCache = null);

    void GarbageCollect(Func<string, IWorkloadResolver> getResolverForWorkloadSet, DirectoryPath? offlineCache = null, bool cleanAllPacks = false);

    WorkloadSet InstallWorkloadSet(ITransactionContext context, string workloadSetVersion, DirectoryPath? offlineCache = null);

    void InstallWorkloadManifest(ManifestVersionUpdate manifestUpdate, ITransactionContext transactionContext, DirectoryPath? offlineCache = null);

    IWorkloadInstallationRecordRepository GetWorkloadInstallationRecordRepository();

    IEnumerable<WorkloadDownload> GetDownloads(IEnumerable<WorkloadId> workloadIds, SdkFeatureBand sdkFeatureBand, bool includeInstalledItems);

    void AdjustWorkloadSetInInstallState(SdkFeatureBand sdkFeatureBand, string workloadVersion);

    void WriteWorkloadHistoryRecord(WorkloadHistoryRecord workloadHistoryRecord, string sdkFeatureBand);

    IEnumerable<WorkloadHistoryRecord> GetWorkloadHistoryRecords(string sdkFeatureBand);

    /// <summary>
    /// Replace the workload resolver used by this installer. Typically used to call <see cref="GetDownloads(IEnumerable{WorkloadId}, SdkFeatureBand, bool)"/>
    /// for a set of workload manifests that isn't currently installed
    /// </summary>
    /// <param name="workloadResolver">A new workload resolver to use</param>
    void ReplaceWorkloadResolver(IWorkloadResolver workloadResolver);

    void Shutdown();

    /// <summary>
    /// Delete the install state file at the specified path.
    /// </summary>
    /// <param name="sdkFeatureBand">The SDK feature band of the install state file.</param>
    void RemoveManifestsFromInstallState(SdkFeatureBand sdkFeatureBand);

    /// <summary>
    /// Writes the specified JSON contents to the install state file.
    /// </summary>
    /// <param name="sdkFeatureBand">The SDK feature band of the install state file.</param>
    /// <param name="manifestContents">The JSON contents describing the install state.</param>
    void SaveInstallStateManifestVersions(SdkFeatureBand sdkFeatureBand, Dictionary<string, string> manifestContents);

    void UpdateInstallMode(SdkFeatureBand sdkFeatureBand, bool? newMode);

    void RecordWorkloadSetInGlobalJson(SdkFeatureBand sdkFeatureBand, string globalJsonPath, string workloadSetVersion);
}

// Interface to pass to workload manifest updater
internal interface IWorkloadManifestInstaller
{
    PackageId GetManifestPackageId(ManifestId manifestId, SdkFeatureBand featureBand);

    //  Extract the contents of the manifest (IE what's in the data directory in the file-based NuGet package) to the targetPath
    Task ExtractManifestAsync(string nupkgPath, string targetPath);
}

public class WorkloadDownload(string id, string nuGetPackageId, string nuGetPackageVersion)
{
    /// <summary>
    /// The ID of the workload pack or manifest to be downloaded (not necessarily the same as the <see cref="NuGetPackageId"/>)
    /// </summary>
    public string Id { get; } = id;

    public string NuGetPackageId { get; } = nuGetPackageId;

    public string NuGetPackageVersion { get; } = nuGetPackageVersion;
}