From ff0789904d3aedbac146854b547799125d0c3317 Mon Sep 17 00:00:00 2001
From: LittleSheep <littlesheep.code@hotmail.com>
Date: Thu, 10 Jul 2025 19:28:15 +0800
Subject: [PATCH] :sparkles: Improved open file endpoint

---
 DysonNetwork.Sphere/Connection/ClientTypeMiddleware.cs | 6 ++----
 DysonNetwork.Sphere/Storage/FileController.cs          | 4 ++++
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/DysonNetwork.Sphere/Connection/ClientTypeMiddleware.cs b/DysonNetwork.Sphere/Connection/ClientTypeMiddleware.cs
index cf0cbc3..8a305aa 100644
--- a/DysonNetwork.Sphere/Connection/ClientTypeMiddleware.cs
+++ b/DysonNetwork.Sphere/Connection/ClientTypeMiddleware.cs
@@ -1,5 +1,3 @@
-using Azure.Core;
-
 namespace DysonNetwork.Sphere.Connection;
 
 public class ClientTypeMiddleware(RequestDelegate next)
@@ -16,8 +14,8 @@ public class ClientTypeMiddleware(RequestDelegate next)
         }
         else
         {
-            var userAgent = headers["User-Agent"].ToString();
-            var accept = headers["Accept"].ToString();
+            var userAgent = headers.UserAgent.ToString();
+            var accept = headers.Accept.ToString();
 
             // Priority 2: Check known app User-Agent (backward compatibility)
             if (!string.IsNullOrEmpty(userAgent) && userAgent.Contains("Solian"))
diff --git a/DysonNetwork.Sphere/Storage/FileController.cs b/DysonNetwork.Sphere/Storage/FileController.cs
index d61bdd0..a9bf2f7 100644
--- a/DysonNetwork.Sphere/Storage/FileController.cs
+++ b/DysonNetwork.Sphere/Storage/FileController.cs
@@ -19,6 +19,10 @@ public class FileController(
     [HttpGet("{id}")]
     public async Task<ActionResult> OpenFile(string id, [FromQuery] bool original = false)
     {
+        // Support the file extension for client side data recognize
+        if (id.Contains("."))
+            id = id.Split('.').First();
+        
         var file = await fs.GetFileAsync(id);
         if (file is null) return NotFound();