// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
 
using System;
using System.Threading;
using System.Threading.Tasks;
using System.Xml.Linq;
using Microsoft.CodeAnalysis;
using Microsoft.CodeAnalysis.CodeFixes.Configuration;
using Microsoft.CodeAnalysis.Diagnostics;
using Microsoft.Internal.VisualStudio.PlatformUI;
using Microsoft.VisualStudio.Imaging;
using Microsoft.VisualStudio.Imaging.Interop;
using Roslyn.Utilities;
 
namespace Microsoft.VisualStudio.LanguageServices.Implementation.SolutionExplorer;
 
internal sealed partial class DiagnosticItem(
    ProjectId projectId,
    AnalyzerReference analyzerReference,
    DiagnosticDescriptor descriptor,
    ReportDiagnostic effectiveSeverity,
    IAnalyzersCommandHandler commandHandler)
    : BaseItem(descriptor.Id + ": " + descriptor.Title), IEquatable<DiagnosticItem>
{
    private readonly AnalyzerReference _analyzerReference = analyzerReference;
    private readonly IAnalyzersCommandHandler _commandHandler = commandHandler;
 
    public ProjectId ProjectId { get; } = projectId;
    public DiagnosticDescriptor Descriptor { get; } = descriptor;
    private readonly ReportDiagnostic _effectiveSeverity = effectiveSeverity;
 
    public override ImageMoniker IconMoniker
        => MapEffectiveSeverityToIconMoniker(_effectiveSeverity);
 
    public override IContextMenuController ContextMenuController => _commandHandler.DiagnosticContextMenuController;
 
    public override object GetBrowseObject()
        => new BrowseObject(this);
 
    private static ImageMoniker MapEffectiveSeverityToIconMoniker(ReportDiagnostic effectiveSeverity)
        => effectiveSeverity switch
        {
            ReportDiagnostic.Error => KnownMonikers.CodeErrorRule,
            ReportDiagnostic.Warn => KnownMonikers.CodeWarningRule,
            ReportDiagnostic.Info => KnownMonikers.CodeInformationRule,
            ReportDiagnostic.Hidden => KnownMonikers.CodeHiddenRule,
            ReportDiagnostic.Suppress => KnownMonikers.CodeSuppressedRule,
            _ => default,
        };
 
    internal void SetRuleSetSeverity(ReportDiagnostic value, string pathToRuleSet)
    {
        var ruleSetDocument = XDocument.Load(pathToRuleSet);
 
        ruleSetDocument.SetSeverity(_analyzerReference.Display, Descriptor.Id, value);
 
        ruleSetDocument.Save(pathToRuleSet);
    }
 
    internal Task<Solution> GetSolutionWithUpdatedAnalyzerConfigSeverityAsync(ReportDiagnostic value, Project project, CancellationToken cancellationToken)
    {
        var effectiveSeverity = value.ToDiagnosticSeverity() ?? Descriptor.DefaultSeverity;
        var diagnostic = Diagnostic.Create(Descriptor, Location.None, effectiveSeverity, additionalLocations: null, properties: null);
        return ConfigurationUpdater.ConfigureSeverityAsync(value, diagnostic, project, cancellationToken);
    }
 
    public override int GetHashCode()
        => Hash.Combine(this.Name,
            Hash.Combine(this.ProjectId,
            Hash.Combine(this.Descriptor.GetHashCode(), (int)_effectiveSeverity)));
 
    public override bool Equals(object obj)
        => Equals(obj as DiagnosticItem);
 
    public bool Equals(DiagnosticItem? other)
    {
        if (this == other)
            return true;
 
        return other != null &&
            this.Name == other.Name &&
            this.ProjectId == other.ProjectId &&
            this.Descriptor.Equals(other.Descriptor) &&
            _effectiveSeverity == other._effectiveSeverity;
    }
}