feat(sandbox): add file extension attribute to SandboxFileNode and update related logic

Author: Mairuis

api/controllers/console/sandbox_files.py

@@ -36,6 +36,7 @@ class SandboxFileDownloadRequest(BaseModel):
     "is_dir": fields.Boolean,
     "size": fields.Raw,
     "mtime": fields.Raw,
+    "extension": fields.String,
 }
 
 
@@ -47,9 +48,7 @@ class SandboxFileDownloadRequest(BaseModel):
 
 
 sandbox_file_node_model = console_ns.model("SandboxFileNode", SANDBOX_FILE_NODE_FIELDS)
-sandbox_file_download_ticket_model = console_ns.model(
-    "SandboxFileDownloadTicket", SANDBOX_FILE_DOWNLOAD_TICKET_FIELDS
-)
+sandbox_file_download_ticket_model = console_ns.model("SandboxFileDownloadTicket", SANDBOX_FILE_DOWNLOAD_TICKET_FIELDS)
 
 
 @console_ns.route("/sandboxes/<string:sandbox_id>/files")

api/core/sandbox/entities/files.py

@@ -9,6 +9,7 @@ class SandboxFileNode:
     is_dir: bool
     size: int | None
     mtime: int | None
+    extension: str | None
 
 
 @dataclass(frozen=True)

api/core/sandbox/inspector.py

@@ -115,12 +115,19 @@ def stat_entry(full_path: str, rel_path: str) -> dict:
 
         entries: list[SandboxFileNode] = []
         for item in raw:
+            item_path = str(item.get("path"))
+            item_is_dir = bool(item.get("is_dir"))
+            extension = None
+            if not item_is_dir:
+                ext = os.path.splitext(item_path)[1]
+                extension = ext or None
             entries.append(
                 SandboxFileNode(
-                    path=str(item.get("path")),
-                    is_dir=bool(item.get("is_dir")),
+                    path=item_path,
+                    is_dir=item_is_dir,
                     size=item.get("size"),
                     mtime=item.get("mtime"),
+                    extension=extension,
                 )
             )
         return entries
@@ -243,7 +250,9 @@ def add_dir(dir_path: str) -> None:
                 if dir_path in ("", "."):
                     return
                 if dir_path not in entries_by_path:
-                    entries_by_path[dir_path] = SandboxFileNode(path=dir_path, is_dir=True, size=None, mtime=None)
+                    entries_by_path[dir_path] = SandboxFileNode(
+                        path=dir_path, is_dir=True, size=None, mtime=None, extension=None
+                    )
 
             def clean(member_name: str) -> str:
                 name = member_name.lstrip("./")
@@ -283,11 +292,16 @@ def clean(member_name: str) -> str:
                         parent = os.path.dirname(parent)
 
                     is_dir = m.isdir()
+                    extension = None
+                    if not is_dir:
+                        ext = os.path.splitext(mp)[1]
+                        extension = ext or None
                     entries_by_path[mp] = SandboxFileNode(
                         path=mp,
                         is_dir=is_dir,
                         size=None if is_dir else int(m.size),
                         mtime=int(m.mtime) if m.mtime else None,
+                        extension=extension,
                     )
 
             return sorted(entries_by_path.values(), key=lambda e: e.path)