using System.Collections.Generic;
using Microsoft.CodeAnalysis;

namespace Reacher;

internal class ExcludingMembersCollector(IReadOnlySet<IMethodSymbol> excludedMembers) : SymbolVisitor
{
    public IReadOnlySet<IMethodSymbol> Members => members;
    private readonly HashSet<IMethodSymbol> members = [];

    public override void VisitAssembly(IAssemblySymbol symbol)
    {
        symbol.GlobalNamespace.Accept(this);
    }

    public override void VisitNamespace(INamespaceSymbol symbol)
    {
        foreach (var namespaceOrType in symbol.GetMembers())
        {
            namespaceOrType.Accept(this);
        }
    }

    public override void VisitNamedType(INamedTypeSymbol type)
    {
        foreach (var member in type.GetMembers())
        {
            member.Accept(this);
        }

        foreach (var nestedType in type.GetTypeMembers())
        {
            nestedType.Accept(this);
        }
    }

    public override void VisitMethod(IMethodSymbol member)
    {
        if (excludedMembers.Contains(member))
        {
            return;
        }

        members.Add(member);
    }
}