src/Workspaces/CoreTestUtilities/TestAnalyzerReferenceByLanguage.cs - main - githubmirror/roslyn

src/Workspaces/CoreTestUtilities/TestAnalyzerReferenceByLanguage.cs

58 строк2 KB

Tomáš Matoušek

Clean up KeyValuePair polyfills (#78872)

24 июн 2025, 03:17

24 июн 2025, 03:17f294360

// 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.Collections.Generic;
using System.Collections.Immutable;
using System.Linq;
using Microsoft.CodeAnalysis.Serialization;
using Roslyn.Utilities;

namespace Microsoft.CodeAnalysis.Diagnostics;

internal sealed class TestAnalyzerReferenceByLanguage :
    AnalyzerReference,
    SerializerService.TestAccessor.IAnalyzerReferenceWithGuid
{
    private readonly IReadOnlyDictionary<string, ImmutableArray<DiagnosticAnalyzer>> _analyzersMap;

    public TestAnalyzerReferenceByLanguage(IReadOnlyDictionary<string, ImmutableArray<DiagnosticAnalyzer>> analyzersMap, string? fullPath = null)
    {
        _analyzersMap = analyzersMap;
        FullPath = fullPath;

        // Make up a checksum so we can calculate Project checksums containing these references
        var checksumArray = Guid.NewGuid().ToByteArray();
        Array.Resize(ref checksumArray, Checksum.HashSize);
        this.Checksum = Checksum.From(checksumArray);
    }

    public override string? FullPath { get; }
    public override string Display => nameof(TestAnalyzerReferenceByLanguage);
    public override object Id => Display;
    public Guid Guid { get; } = Guid.NewGuid();

    public Checksum Checksum;

    public override ImmutableArray<DiagnosticAnalyzer> GetAnalyzersForAllLanguages()
        => _analyzersMap.SelectManyAsArray(kvp => kvp.Value);

    public override ImmutableArray<DiagnosticAnalyzer> GetAnalyzers(string language)
    {
        if (_analyzersMap.TryGetValue(language, out var analyzers))
        {
            return analyzers;
        }

        return [];
    }

    public TestAnalyzerReferenceByLanguage WithAdditionalAnalyzers(string language, IEnumerable<DiagnosticAnalyzer> analyzers)
    {
        var newAnalyzersMap = ImmutableDictionary.CreateRange(
            _analyzersMap.Select(kvp => KeyValuePair.Create(
                kvp.Key, kvp.Key == language ? kvp.Value.AddRange(analyzers) : kvp.Value)));
        return new(newAnalyzersMap);
    }
}