InterfaceGenerator/Speckle.InterfaceGenerator/SyntaxReceiver.cs

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

namespace Speckle.InterfaceGenerator;

internal class SyntaxReceiver : ISyntaxReceiver
{
    public IList<TypeDeclarationSyntax> CandidateTypes { get; } = new List<TypeDeclarationSyntax>();

    public void OnVisitSyntaxNode(SyntaxNode syntaxNode)
    {
        if (
            syntaxNode is TypeDeclarationSyntax typeDeclarationSyntax
            && IsClassOrRecord(typeDeclarationSyntax)
            && typeDeclarationSyntax.AttributeLists.Count > 0
        )
        {
            CandidateTypes.Add(typeDeclarationSyntax);
        }
    }

    private static bool IsClassOrRecord(TypeDeclarationSyntax typeDeclarationSyntax)
    {
        return typeDeclarationSyntax is ClassDeclarationSyntax
            || typeDeclarationSyntax is RecordDeclarationSyntax;
    }
}