From d952b88bc1c1e86be29808bd808bbb3e592630bb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=A7=8B=E9=9B=A8=E8=90=BD?= <i@rain.cx>
Date: Tue, 12 Sep 2023 12:14:54 +0800
Subject: [PATCH] :sparkles: Websocket Server.
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: 秋雨落 <i@rain.cx>
---
 CrystalConnector/CrystalConnector.csproj |  4 ++++
 CrystalConnector/Startup.cs              | 12 ++++++++++++
 2 files changed, 16 insertions(+)

diff --git a/CrystalConnector/CrystalConnector.csproj b/CrystalConnector/CrystalConnector.csproj
index b29c029..a3c16b2 100644
--- a/CrystalConnector/CrystalConnector.csproj
+++ b/CrystalConnector/CrystalConnector.csproj
@@ -33,4 +33,8 @@
     </ItemGroup>
 
 
+    <ItemGroup>
+      <Folder Include="Web\" />
+    </ItemGroup>
+
 </Project>
diff --git a/CrystalConnector/Startup.cs b/CrystalConnector/Startup.cs
index 2ea873d..773d7a0 100644
--- a/CrystalConnector/Startup.cs
+++ b/CrystalConnector/Startup.cs
@@ -22,5 +22,17 @@ public class Startup
 
     public void Configure(IApplicationBuilder app)
     {
+        app.UseWebSockets();
+        
+        app.Use(async (context, next) =>
+        {
+            if (context.WebSockets.IsWebSocketRequest)
+            {
+                await context.WebSockets.AcceptWebSocketAsync();
+                Logger.Info("Got WebSocket connection from {Ip}:{Port}", context.Connection.RemoteIpAddress?.ToString(), context.Connection.RemotePort);
+            }
+
+            await next(context);
+        });
     }
 }
\ No newline at end of file