dyadmisha
/
dune-weaver
-ын хуулбар https://github.com/tuanchris/dune-weaver.git


			
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180
							"""
FastAPI router for WiFi management endpoints.
"""

from fastapi import APIRouter, HTTPException, Request
from fastapi.responses import HTMLResponse
from pydantic import BaseModel
from typing import Optional
import logging

from . import manager

logger = logging.getLogger(__name__)

router = APIRouter(prefix="/api/wifi", tags=["wifi"])

# Separate router for the captive portal handler (no prefix)
captive_portal_router = APIRouter(tags=["wifi"])


class WiFiConnectRequest(BaseModel):
    ssid: str
    password: Optional[str] = ""


class WiFiForgetRequest(BaseModel):
    ssid: str


class HotspotPasswordRequest(BaseModel):
    password: Optional[str] = ""


@router.get("/status")
async def wifi_status():
    """Get current WiFi mode, connection, and IP."""
    return manager.get_wifi_status()


@router.get("/networks")
async def wifi_networks():
    """Scan for available WiFi networks."""
    return manager.scan_networks()


@router.post("/connect")
async def wifi_connect(req: WiFiConnectRequest):
    """Connect to a WiFi network and reboot."""
    if not req.ssid:
        raise HTTPException(status_code=400, detail="SSID is required")

    result = await manager.connect_to_network(req.ssid, req.password or "")
    if not result["success"]:
        raise HTTPException(status_code=400, detail=result["message"])
    return result


@router.post("/save")
async def wifi_save(req: WiFiConnectRequest):
    """Save a WiFi network without connecting."""
    if not req.ssid:
        raise HTTPException(status_code=400, detail="SSID is required")

    result = manager.save_network(req.ssid, req.password or "")
    if not result["success"]:
        raise HTTPException(status_code=400, detail=result["message"])
    return result


@router.post("/forget")
async def wifi_forget(req: WiFiForgetRequest):
    """Forget a saved WiFi network."""
    if not req.ssid:
        raise HTTPException(status_code=400, detail="SSID is required")

    result = manager.forget_network(req.ssid)
    if not result["success"]:
        raise HTTPException(status_code=400, detail=result["message"])
    return result


@router.get("/saved")
async def wifi_saved():
    """Get list of saved WiFi connections."""
    return manager.get_saved_connections()


@router.get("/hotspot/password")
async def get_hotspot_password():
    """Get the current hotspot password."""
    return manager.get_hotspot_password()


@router.post("/hotspot/password")
async def set_hotspot_password(req: HotspotPasswordRequest):
    """Set or remove the hotspot password."""
    result = manager.set_hotspot_password(req.password or "")
    if not result["success"]:
        raise HTTPException(status_code=400, detail=result["message"])
    return result


# --- Captive portal detection endpoints ---
# These handle the well-known URLs that phones/tablets probe to detect captive portals.
# In hotspot mode, DNS redirects all domains to the Pi, so these probes arrive here.
# We serve a minimal HTML page that redirects to the WiFi setup page.

CAPTIVE_PORTAL_HTML = """<!DOCTYPE html>
<html>
<head>
    <meta charset="utf-8">
    <meta name="viewport" content="width=device-width, initial-scale=1">
    <title>Dune Weaver</title>
    <style>
        * { margin: 0; padding: 0; box-sizing: border-box; }
        body { font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', sans-serif;
               display: flex; justify-content: center; align-items: center;
               min-height: 100vh; background: #f8fafc; color: #0f172a; padding: 1rem; }
        .card { background: #ffffff; border: 1px solid #e2e8f0; border-radius: 12px;
                padding: 2rem; max-width: 400px; width: 100%; text-align: center;
                box-shadow: 0 1px 3px rgba(0,0,0,0.1); }
        h1 { font-size: 1.5rem; margin-bottom: 0.5rem; font-weight: 600; }
        p { color: #64748b; margin-bottom: 1.5rem; font-size: 0.9rem; }
        .buttons { display: flex; flex-direction: column; gap: 0.75rem; }
        a, button { display: block; padding: 0.75rem 2rem; border-radius: 8px;
            text-decoration: none; font-weight: 500; text-align: center;
            width: 100%; font-size: 1rem; cursor: pointer; border: none; }
        .btn-primary { background: #2563eb; color: white; }
        .btn-primary:hover { background: #1d4ed8; }
        .btn-secondary { background: #f1f5f9; color: #0f172a; border: 1px solid #e2e8f0; }
        .btn-secondary:hover { background: #e2e8f0; }
        .url-hint { display: none; margin-top: 1rem; padding: 0.75rem; background: #f1f5f9;
                    border-radius: 8px; font-size: 0.85rem; color: #475569; }
        .url-hint code { background: #e2e8f0; padding: 0.15rem 0.4rem; border-radius: 4px;
                         font-size: 0.9rem; user-select: all; }
    </style>
</head>
<body>
    <div class="card">
        <h1>Welcome to Dune Weaver</h1>
        <p>What would you like to do?</p>
        <div class="buttons">
            <a href="/wifi-setup" class="btn-primary">Connect to WiFi</a>
            <button class="btn-secondary" onclick="openInBrowser()">Control Table</button>
        </div>
        <div class="url-hint" id="url-hint">
            Open your browser and go to:<br>
            <code>http://10.42.0.1</code>
        </div>
    </div>
    <script>
        function openInBrowser() {
            // Try to open in real browser (outside captive portal webview)
            var url = 'http://10.42.0.1';
            var w = window.open(url, '_blank');
            if (!w || w.closed) {
                // window.open blocked — show URL for user to open manually
                document.getElementById('url-hint').style.display = 'block';
            }
        }
    </script>
</body>
</html>"""


@captive_portal_router.get("/hotspot-detect.html", response_class=HTMLResponse)
@captive_portal_router.get("/generate_204", response_class=HTMLResponse)
@captive_portal_router.get("/connecttest.txt", response_class=HTMLResponse)
@captive_portal_router.get("/ncsi.txt", response_class=HTMLResponse)
@captive_portal_router.get("/redirect", response_class=HTMLResponse)
@captive_portal_router.get("/canonical.html", response_class=HTMLResponse)
async def captive_portal_detect():
    """Handle captive portal detection probes.

    Phones and tablets check these well-known URLs after connecting to WiFi.
    In hotspot mode, DNS resolves all domains to the Pi, so these probes
    arrive at our server. Returning anything other than the expected response
    triggers the OS to show a captive portal browser.
    """
    return HTMLResponse(content=CAPTIVE_PORTAL_HTML, status_code=200)