// 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]);
        }
    }
}