1 files changed, 19 insertions, 0 deletions
diff --git a/src/EndpointFinder.cs b/src/EndpointFinder.cs
new file mode 100644
index 0000000..ce1b321
--- /dev/null
+++ b/src/EndpointFinder.cs
@@ -0,0 +1,19 @@
+using Microsoft.CodeAnalysis;
+using Microsoft.CodeAnalysis.CSharp.Syntax;
+
+namespace I2R.Endpoints;
+
+public class EndpointFinder : ISyntaxReceiver
+{
+    public HashSet<ClassDeclarationSyntax> AsyncEndpoints { get; } = new();
+    public HashSet<ClassDeclarationSyntax> SyncEndpoints { get; } = new();
+
+    public void OnVisitSyntaxNode(SyntaxNode syntaxNode) {
+        if (syntaxNode is not ClassDeclarationSyntax endpoint) return;
+        if (endpoint.BaseList?.Types.Any(c => c.ToString().StartsWith("AsyncEndpoint")) ?? false) {
+            AsyncEndpoints.Add(endpoint);
+        } else if (endpoint.BaseList?.Types.Any(c => c.ToString().StartsWith("SyncEndpoint")) ?? false) {
+            SyncEndpoints.Add(endpoint);
+        }
+    }
+}
+\ No newline at end of file