From f48f101eb0dd6bbdca91d5f50b4ce4194d7369ab Mon Sep 17 00:00:00 2001
From: ivarlovlie <git@ivarlovlie.no>
Date: Sat, 19 Nov 2022 13:52:13 +0700
Subject: feat: Initial commit

---
 src/EndpointFinder.cs | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)
 create mode 100644 src/EndpointFinder.cs

(limited to 'src/EndpointFinder.cs')

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
-- 
cgit v1.3