add cdn_cache

Author: codeskyblue

pyproject.toml

@@ -34,6 +34,7 @@ typer = "^0.9.0"
 rich = "^13.7.0"
 aiofiles = "^24.1.0"
 pydantic-settings = "^2.8.0"
+aiohttp = "^3.9.0"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^7.4.3"

servefs/cli.py

@@ -1,4 +1,3 @@
-import inspect
 import os
 from pathlib import Path
 from typing import Optional
@@ -8,6 +7,8 @@
 from rich.console import Console
 from rich.panel import Panel
 
+from servefs.utils.network import get_local_addresses
+
 app = typer.Typer(
     name="servefs",
     help="A modern HTTP file server with web UI",
@@ -89,9 +90,19 @@ def main(
         os.environ["SERVEFS_BASIC_AUTH"] = basic_auth
 
     # Display server information
+    if host == "0.0.0.0":
+        # Show all available addresses
+        addresses = get_local_addresses()
+        address_lines = []
+        for addr in addresses:
+            address_lines.append(f"[bold]http://{addr}:{port}[/bold]")
+        server_info = "\n".join(address_lines)
+    else:
+        server_info = f"[bold]http://{host}:{port}[/bold]"
+    
     console.print(Panel.fit(
         f"[bold green]Starting server at[/bold green]\n"
-        f"[bold]http://{host}:{port}[/bold]\n"
+        f"{server_info}\n"
         f"[bold blue]Root directory:[/bold blue] {os.environ['SERVEFS_ROOT']}\n"
         f"[bold yellow]Developer mode:[/bold yellow] {'enabled' if debug else 'disabled'}\n"
         "\n[dim]Press Ctrl+C to quit[/dim]",

servefs/config.py

@@ -1,11 +1,11 @@
 from pathlib import Path
-
+from typing import Union, Optional
 from pydantic_settings import BaseSettings
 
 
 class Settings(BaseSettings):
     DEBUG: bool = False
-    BASIC_AUTH: str | None = None
+    BASIC_AUTH: Optional[str] = None
     ROOT: Path = Path("./files")
 
     class Config:

servefs/routes/page.py

@@ -9,15 +9,19 @@
 
 import aiofiles
 from fastapi import APIRouter, HTTPException, Request
-from fastapi.responses import FileResponse, HTMLResponse, StreamingResponse
+from fastapi.responses import FileResponse, HTMLResponse, StreamingResponse, RedirectResponse
 
 from ..utils.static import ETaggedStaticFiles
+from ..utils.cdn_cache import CDNCacheManager
 
 logger = logging.getLogger(__name__)
 
 # Get current module path
 PACKAGE_DIR = Path(__file__).parent.parent
 
+# Initialize CDN cache manager
+cdn_cache_manager = CDNCacheManager(PACKAGE_DIR / "static" / "cdn_cache")
+
 router = APIRouter(tags=["page"])
 
 async def stream_file_range(file_path: Path, start: int, end: int) -> AsyncIterator[bytes]:
@@ -122,6 +126,17 @@ def init_static_files(app):
     """Initialize static file serving"""
     app.mount("/static", ETaggedStaticFiles(directory=PACKAGE_DIR / "static"), name="static")
 
+@router.get("/cdn/{url:path}")
+async def proxy_cdn(url: str, request: Request):
+    """Proxy requests to CDN with local caching"""
+    full_url = f'https://{url}'
+    response = await cdn_cache_manager.serve_cached_resource(full_url, request)
+    
+    if response:
+        return response
+    # If caching failed, redirect to original URL
+    return RedirectResponse(url=full_url, status_code=302)
+
 # Serve index.html for the root path
 @router.get("/", response_class=HTMLResponse)
 async def serve_root():