// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
#nullable disable
 
namespace Microsoft.DotNet.Cli.Commands.Workload;
 
internal class WorkloadHistoryRecord
{
    public DateTimeOffset TimeStarted { get; set; }
    public DateTimeOffset TimeCompleted { get; set; }
 
    public string CommandName { get; set; }
 
    public Dictionary<string, string> RollbackFileContents { get; set; }
 
    public string[] CommandLineArgs { get; set; }
 
    public bool Succeeded { get; set; }
 
    public string ErrorMessage { get; set; }
 
    public WorkloadHistoryState StateBeforeCommand { get; set; }
 
    public WorkloadHistoryState StateAfterCommand { get; set; }
 
    public string GlobalJsonVersion { get; set; }
}
 
internal class WorkloadHistoryState
{
    public Dictionary<string, string> ManifestVersions { get; set; }
 
    public string WorkloadSetVersion { get; set; }
 
    public List<string> InstalledWorkloads { get; set; }
 
    public bool Equals(WorkloadHistoryState other)
    {
        if (ManifestVersions.Count != other.ManifestVersions.Count)
        {
            return false;
        }
 
        foreach (var manifestId in ManifestVersions.Keys)
        {
            if (!other.ManifestVersions.TryGetValue(manifestId, out string otherManifestVersion) ||
                ManifestVersions[manifestId] != otherManifestVersion)
            {
                return false;
            }
        }
 
        if (WorkloadSetVersion is not null && !WorkloadSetVersion.Equals(other.WorkloadSetVersion) ||
            WorkloadSetVersion is null && other.WorkloadSetVersion is not null)
        {
            return false;
        }
 
        return new HashSet<string>(InstalledWorkloads).SetEquals(other.InstalledWorkloads);
    }
 
    public override bool Equals(object other)
    {
        if (other is WorkloadHistoryState otherState)
        {
            return Equals(otherState);
        }
 
        return false;
    }
 
    public override int GetHashCode()
    {
        HashCode hc = new HashCode();
        foreach (var kvp in ManifestVersions)
        {
            hc.Add(kvp.Key);
            hc.Add(kvp.Value);
        }
 
        hc.Add(WorkloadSetVersion);
 
        foreach (var workload in InstalledWorkloads)
        {
            hc.Add(workload);
        }
 
        return hc.ToHashCode();
    }
}