src/EditorFeatures/TestUtilities/PullMemberUp/TestPullMemberUpService.cs - main - githubmirror/roslyn

src/EditorFeatures/TestUtilities/PullMemberUp/TestPullMemberUpService.cs
79 строк3 KB
Cyrus Najmabadi
Use file scoped namespaces. (#77854)
27 мар 2025, 09:55
27 мар 2025, 09:5598d41b8
// 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.

#nullable disable

using System;
using System.Collections.Generic;
using System.Collections.Immutable;
using System.Linq;
using Microsoft.CodeAnalysis.CodeRefactorings.PullMemberUp;
using Microsoft.CodeAnalysis.CodeRefactorings.PullMemberUp.Dialog;
using Microsoft.CodeAnalysis.PullMemberUp;

namespace Microsoft.CodeAnalysis.Test.Utilities.PullMemberUp;

internal sealed class TestPullMemberUpService : IPullMemberUpOptionsService
{
    private readonly IEnumerable<(string member, bool makeAbstract)> _selectedMembers;

    private string DestinationName { get; }

    public TestPullMemberUpService(IEnumerable<(string member, bool makeAbstract)> selectedMembers, string destinationName)
    {
        _selectedMembers = selectedMembers;
        DestinationName = destinationName;
    }

    public PullMembersUpOptions GetPullMemberUpOptions(Document document, ImmutableArray<ISymbol> selectedNodeSymbols)
    {
        var containingType = selectedNodeSymbols[0].ContainingType;
        var members = containingType.GetMembers().Where(member => MemberAndDestinationValidator.IsMemberValid(member));

        var selectedMember = _selectedMembers == null
            ? members.Select(member => (member, false))
            : _selectedMembers.Select(selection => (members.Single(symbol => symbol.Name == selection.member), selection.makeAbstract));

        var allInterfaces = containingType.AllInterfaces;
        var baseClass = containingType.BaseType;

        INamedTypeSymbol destination = null;
        if (DestinationName == null)
        {
            destination = allInterfaces.FirstOrDefault() ?? baseClass;

            if (destination == null)
            {
                throw new ArgumentException($"No target base type for {containingType}");
            }
        }
        else
        {
            if (allInterfaces != null)
            {
                destination = allInterfaces.SingleOrDefault(@interface => @interface.Name == DestinationName);
            }

            if (baseClass != null && destination == null)
            {
                for (var i = baseClass; i != null; i = i.BaseType)
                {
                    if (i.Name == DestinationName)
                    {
                        return PullMembersUpOptionsBuilder.BuildPullMembersUpOptions(i, [.. selectedMember]);
                    }
                }
            }
        }

        if (destination == null)
        {
            throw new ArgumentException($"No Matching target base type for {DestinationName}");
        }
        else
        {
            return PullMembersUpOptionsBuilder.BuildPullMembersUpOptions(destination, [.. selectedMember]);
        }
    }
}