Update app.py

app.py

@@ -1,635 +1,17 @@
-"""
-Service Orchestrator — Pro Linux Workspace
-
-Starts and health-monitors: Filebrowser, ttyd, code-server, Glances, Caddy.
-"""
-
-from __future__ import annotations
-
 import os
-import sys
-import time
-import signal
-import shutil
-import logging
 import subprocess
-from typing import Optional
-
-# ── Logging ──────────────────────────────────────────────────────────────────
-logging.basicConfig(
-    level=logging.INFO,
-    format="%(asctime)s - %(levelname)s - %(message)s",
-)
-log = logging.getLogger(__name__)
-
-# ── Constants ────────────────────────────────────────────────────────────────
-HEALTH_CHECK_INTERVAL = 10   # seconds
-DRIVE_DIR  = "/home/user/Drive"
-DB_DIR     = f"{DRIVE_DIR}/system_data"
-DB_PATH    = f"{DB_DIR}/filebrowser.db"
-CADDYFILE  = "/tmp/Caddyfile"
-
-# Service ports (localhost only)
-PORT_FB      = 8080
-PORT_TTYD    = 8081
-PORT_CODE    = 8082
-PORT_GLANCES = 8083
-PORT_CADDY   = 7860
-
-
-# ─────────────────────────────────────────────────────────────────────────────
-# Helpers
-# ─────────────────────────────────────────────────────────────────────────────
-
-def which(name: str) -> Optional[str]:
-    """Return full path of *name* if found on PATH, else None."""
-    return shutil.which(name)
-
-
-def run_checked(
-    cmd: list[str],
-    name: str,
-    *,
-    silent: bool = True,
-    critical: bool = False,
-    capture: bool = False,
-) -> subprocess.CompletedProcess:
-    """
-    Run a command synchronously and check its return code.
-    Calls sys.exit(1) if critical=True and command fails.
-    Returns the CompletedProcess object.
-    """
-    kwargs: dict = {}
-    if silent and not capture:
-        kwargs["stdout"] = subprocess.DEVNULL
-        kwargs["stderr"] = subprocess.DEVNULL
-    if capture:
-        kwargs["capture_output"] = True
-        kwargs["text"] = True
-
-    try:
-        result = subprocess.run(cmd, **kwargs)
-        if result.returncode != 0:
-            log.error(
-                "%s exited with code %d (cmd: %s)",
-                name,
-                result.returncode,
-                " ".join(cmd),
-            )
-            if critical:
-                sys.exit(1)
-        return result
-    except Exception as exc:
-        log.error("%s failed to run: %s", name, exc)
-        if critical:
-            sys.exit(1)
-        return subprocess.CompletedProcess(cmd, returncode=1, stdout="", stderr=str(exc))
-
-
-def launch(
-    cmd: Optional[list[str]],
-    name: str,
-    *,
-    env: Optional[dict] = None,
-    silent: bool = False,
-    critical: bool = False,
-) -> Optional[subprocess.Popen]:
-    """
-    Start *cmd* as a background process.
-
-    Parameters
-    ----------
-    cmd      : Command + arguments.  None -> skip gracefully.
-    name     : Human-readable service label (for logging).
-    env      : Override environment dict (defaults to inherited env).
-    silent   : Redirect stdout/stderr to /dev/null when True.
-    critical : Call sys.exit(1) if the process cannot be launched.
-    """
-    if cmd is None:
-        log.warning("%s: skipped (binary not found).", name)
-        return None
-
-    kwargs: dict = {}
-    if env is not None:
-        kwargs["env"] = env
-    if silent:
-        kwargs["stdout"] = subprocess.DEVNULL
-        kwargs["stderr"] = subprocess.DEVNULL
-
-    try:
-        proc = subprocess.Popen(cmd, **kwargs)
-        log.info("%s started (PID %d)", name, proc.pid)
-        return proc
-    except Exception as exc:
-        log.error("Failed to start %s: %s", name, exc)
-        if critical:
-            log.critical("Critical service '%s' failed — aborting.", name)
-            sys.exit(1)
-        return None
-
-
-# ─────────────────────────────────────────────────────────────────────────────
-# Setup
-# ─────────────────────────────────────────────────────────────────────────────
-
-def validate_auth(env: dict) -> tuple:
-    """
-    Parse and validate TERMINAL_AUTH from environment.
-    Returns (username, password) or exits the process.
-    """
-    auth = env.get("TERMINAL_AUTH", "").strip()
-    if not auth:
-        log.critical(
-            "TERMINAL_AUTH is not set. "
-            "Add it to your Space secrets as 'username:strongpassword'. "
-            "Refusing to start without authentication."
-        )
-        sys.exit(1)
-
-    if ":" not in auth:
-        log.critical(
-            "TERMINAL_AUTH format invalid. Expected 'username:password'."
-        )
-        sys.exit(1)
-
-    username, password = auth.split(":", 1)
-    username = username.strip()
-    password = password.strip()
-
-    if not username or not password:
-        log.critical(
-            "TERMINAL_AUTH has empty username or password. "
-            "Expected 'username:password'."
-        )
-        sys.exit(1)
-
-    return username, password
-
-
-def restore_cloud_data(env: dict) -> None:
-    """Restore previous session data from cloud backup."""
-    os.makedirs(DRIVE_DIR, exist_ok=True)
-    log.info("Restoring previous session from cloud…")
-    result = subprocess.run(
-        ["python3", "autobackup.py", "--restore"],
-        env=env,
-        capture_output=True,
-        text=True,
-    )
-    if result.returncode == 0:
-        log.info("Cloud data restored into %s", DRIVE_DIR)
-    else:
-        log.warning(
-            "Restore exited %d (first run or no credentials). Continuing…",
-            result.returncode,
-        )
-        if result.stderr:
-            log.debug("Restore stderr: %s", result.stderr.strip()[:500])
-
-
-def _parse_fb_user_table(output: str) -> Optional[dict]:
-    """
-    Parse the tabular output of 'filebrowser users ls'.
-
-    Example output:
-        ID  Username      Scope  Locale  ...
-         1  admin         /      en      ...
-
-    Returns dict with keys {id, username} of first data row,
-    or None if no user found.
-    """
-    lines = output.strip().splitlines()
-    if len(lines) < 2:
-        return None
-
-    # First line is always the header
-    header_line = lines[0]
-
-    # Find column positions from header
-    # We need "ID" and "Username" column start positions
-    id_start = header_line.find("ID")
-    username_start = header_line.find("Username")
-
-    if id_start == -1 or username_start == -1:
-        return None
-
-    # Find next column after Username to know its end boundary
-    # Look for the next column header after "Username"
-    remaining_header = header_line[username_start + len("Username"):]
-    # Find the start of the next non-space column
-    next_col_offset = 0
-    found_space = False
-    for i, ch in enumerate(remaining_header):
-        if ch == " ":
-            found_space = True
-        elif found_space:
-            next_col_offset = i
-            break
-    username_end = username_start + len("Username") + next_col_offset if next_col_offset else len(header_line)
-
-    # Parse first data row (line index 1)
-    data_line = lines[1]
-    if len(data_line) < username_start:
-        return None
-
-    # Extract ID (everything from id_start to username_start, stripped)
-    raw_id = data_line[id_start:username_start].strip()
-
-    # Extract Username (from username_start to username_end, stripped)
-    raw_username = data_line[username_start:username_end].strip()
-
-    if not raw_id or not raw_username:
-        return None
 
+def start_terminal():
+    print("🚀 Starting Professional Linux Web Terminal on Port 7860...")
+    
+    # ttyd কমান্ড: -p 7860 (পোর্ট), -W (টাইপ করার পারমিশন), bash (লিনাক্স শেল)
+    command = ["ttyd", "-p", "7860", "-W", "bash"]
+    
     try:
-        user_id = int(raw_id)
-    except ValueError:
-        return None
-
-    return {"id": user_id, "username": raw_username}
-
-
-def setup_filebrowser(username: str, password: str) -> None:
-    """
-    Initialise or update the Filebrowser database.
-
-    Handles all scenarios:
-    - Fresh install: create DB + user
-    - Same username: update password only (by numeric ID)
-    - Changed username: update both username and password (by numeric ID)
-    - Corrupted/unreadable DB: delete and recreate
-    """
-    fb_bin = which("filebrowser")
-    if not fb_bin:
-        log.critical("filebrowser binary not found — aborting.")
-        sys.exit(1)
-
-    os.makedirs(DB_DIR, exist_ok=True)
-
-    if not os.path.exists(DB_PATH):
-        # ── Fresh database ───────────────────────────────────────────────
-        _init_fresh_fb_db(username, password)
-        return
-
-    # ── Existing database — read current user ────────────────────────────
-    result = run_checked(
-        ["filebrowser", "users", "ls", "-d", DB_PATH],
-        "Filebrowser user list",
-        capture=True,
-    )
-
-    if result.returncode != 0 or not result.stdout or not result.stdout.strip():
-        # Cannot read DB — recreate from scratch
-        log.warning("Filebrowser: cannot read existing DB. Recreating…")
-        _recreate_fb_db(username, password)
-        return
-
-    existing = _parse_fb_user_table(result.stdout)
-
-    if existing is None:
-        # No users in DB or parse failure — recreate
-        log.warning("Filebrowser: no users found in DB. Recreating…")
-        _recreate_fb_db(username, password)
-        return
-
-    user_id = existing["id"]
-    existing_username = existing["username"]
-
-    if existing_username == username:
-        # Same username → update password only, using numeric ID
-        log.info("Filebrowser: updating password for '%s' (ID %d).", username, user_id)
-        run_checked(
-            [
-                "filebrowser", "users", "update", str(user_id),
-                "-p", password,
-                "-d", DB_PATH,
-            ],
-            "Filebrowser password update",
-        )
-    else:
-        # Username changed → update username + password using numeric ID
-        log.info(
-            "Filebrowser: username changed '%s' -> '%s' (ID %d). Updating…",
-            existing_username, username, user_id,
-        )
-        run_checked(
-            [
-                "filebrowser", "users", "update", str(user_id),
-                "--username", username,
-                "-p", password,
-                "-d", DB_PATH,
-            ],
-            "Filebrowser user update (username + password)",
-        )
-
-    log.info("Filebrowser: existing database loaded, credentials synced.")
-
-
-def _init_fresh_fb_db(username: str, password: str) -> None:
-    """Create a brand-new Filebrowser database with one admin user."""
-    run_checked(
-        [
-            "filebrowser", "config", "init",
-            "-d", DB_PATH,
-            "-a", "127.0.0.1",
-            "-p", str(PORT_FB),
-            "-r", "/home/user",
-        ],
-        "Filebrowser config init",
-        critical=True,
-    )
-    run_checked(
-        [
-            "filebrowser", "config", "set",
-            "--auth.method", "json",
-            "--auth.header", "",
-            "-d", DB_PATH,
-        ],
-        "Filebrowser config set",
-    )
-    run_checked(
-        [
-            "filebrowser", "users", "add",
-            username, password,
-            "--perm.admin",
-            "-d", DB_PATH,
-        ],
-        "Filebrowser user add",
-        critical=True,
-    )
-    log.info("Filebrowser: new persistent database initialised for '%s'.", username)
-
-
-def _recreate_fb_db(username: str, password: str) -> None:
-    """Delete the existing Filebrowser DB and create a fresh one."""
-    try:
-        os.remove(DB_PATH)
-        log.info("Filebrowser: removed corrupted database.")
-    except OSError as exc:
-        log.warning("Filebrowser: could not remove old DB: %s", exc)
-    _init_fresh_fb_db(username, password)
-
-
-def write_caddyfile() -> str:
-    """
-    Generate and write the Caddyfile.
-
-    Routing rules:
-    - /code/*     → handle_path (strips prefix, code-server expects /)
-    - /terminal/* → handle (keeps prefix, ttyd uses -b /terminal)
-    - /status/*   → handle_path (strips prefix for Glances)
-    - /           → Filebrowser (default)
-
-    flush_interval -1 disables response buffering for WebSocket services.
-    """
-    content = f"""\
-{{
-    admin off
-    log {{
-        level ERROR
-    }}
-}}
-
-:{PORT_CADDY} {{
-    # ── Suppress Filebrowser 401 noise on token renewal ──
-    @renew path /api/renew
-    handle @renew {{
-        reverse_proxy 127.0.0.1:{PORT_FB} {{
-            @err status 401
-            handle_response @err {{
-                respond "" 200
-            }}
-        }}
-    }}
-
-    # ── Health Dashboard (Glances) ──
-    redir /status /status/
-    handle_path /status/* {{
-        reverse_proxy 127.0.0.1:{PORT_GLANCES}
-    }}
-
-    # ── VS Code IDE ──
-    # handle_path strips /code/ so code-server receives / as expected
-    redir /code /code/
-    handle_path /code/* {{
-        reverse_proxy 127.0.0.1:{PORT_CODE} {{
-            flush_interval -1
-        }}
-    }}
-
-    # ── Web Terminal (ttyd) ──
-    # handle preserves /terminal/ because ttyd uses -b /terminal
-    redir /terminal /terminal/
-    handle /terminal/* {{
-        reverse_proxy 127.0.0.1:{PORT_TTYD} {{
-            flush_interval -1
-        }}
-    }}
-
-    # ── File Manager (default) ──
-    handle {{
-        reverse_proxy 127.0.0.1:{PORT_FB}
-    }}
-}}
-"""
-    with open(CADDYFILE, "w", encoding="utf-8") as fh:
-        fh.write(content)
-
-    log.info("Caddyfile written to %s", CADDYFILE)
-    return CADDYFILE
-
-
-# ─────────────────────────────────────────────────────────────────────────────
-# Service definitions
-# ─────────────────────────────────────────────────────────────────────────────
-
-def build_services(
-    username: str,
-    password: str,
-    env: dict,
-) -> dict:
-    """
-    Build command lists and environments for each service.
-    Returns dict: service_name -> (cmd, service_env, is_critical).
-    """
-    services: dict = {}
-
-    # ── Filebrowser ──
-    fb_bin = which("filebrowser")
-    if fb_bin:
-        services["File Manager"] = (
-            [fb_bin, "-d", DB_PATH, "-a", "127.0.0.1", "-p", str(PORT_FB)],
-            env,
-            True,
-        )
-    else:
-        log.critical("filebrowser not found — cannot start File Manager.")
-        sys.exit(1)
-
-    # ── ttyd (Web Terminal) ──
-    ttyd_bin = which("ttyd")
-    if ttyd_bin:
-        services["Web Terminal"] = (
-            [
-                ttyd_bin,
-                "-p", str(PORT_TTYD),
-                "-W",
-                "-c", f"{username}:{password}",
-                "-b", "/terminal",
-                "-t", "fontSize=14",
-                "tmux", "new-session", "-A", "-s", "main", "zsh",
-            ],
-            env,
-            False,
-        )
-    else:
-        log.warning("ttyd not found — Web Terminal will be unavailable.")
-
-    # ── code-server (VS Code IDE) ──
-    code_bin = which("code-server")
-    if code_bin:
-        code_env = env.copy()
-        code_env["PASSWORD"] = password
-        services["VS Code IDE"] = (
-            [
-                code_bin,
-                "--bind-addr", f"127.0.0.1:{PORT_CODE}",
-                "--auth", "password",
-                "--disable-telemetry",
-                "--disable-update-check",
-                "/home/user",
-            ],
-            code_env,
-            False,
-        )
-    else:
-        log.warning("code-server not found — VS Code IDE will be unavailable.")
-
-    # ── Glances (Health Dashboard) ──
-    glances_bin = which("glances")
-    if glances_bin:
-        services["Health Dashboard"] = (
-            [
-                glances_bin,
-                "-w",
-                "--port", str(PORT_GLANCES),
-                "--disable-plugin", "cloud",
-            ],
-            env,
-            False,
-        )
-    else:
-        log.warning("glances not found — Health Dashboard will be unavailable.")
-
-    # ── Auto-backup ──
-    services["Auto-backup"] = (
-        ["python3", "autobackup.py", "--backup"],
-        env,
-        False,
-    )
-
-    # ── Caddy (Router — critical) ──
-    caddy_bin = which("caddy")
-    if not caddy_bin:
-        log.critical("caddy not found — cannot start router. Aborting.")
-        sys.exit(1)
-    services["Caddy Router"] = (
-        [caddy_bin, "run", "--config", CADDYFILE],
-        env,
-        True,
-    )
-
-    return services
-
-
-# ─────────────────────────────────────────────────────────────────────────────
-# Main
-# ─────────────────────────────────────────────────────────────────────────────
-
-def main() -> None:
-    # ── 1. Build runtime environment ─────────────────────────────────────────
-    env = os.environ.copy()
-    env["PATH"] = f"/home/user/.local/bin:{env.get('PATH', '')}"
-
-    # ── 2. Validate authentication ───────────────────────────────────────────
-    username, password = validate_auth(env)
-    log.info("Starting Pro Linux Workspace — user: '%s'", username)
-
-    # ── 3. Restore cloud backup ──────────────────────────────────────────────
-    restore_cloud_data(env)
-
-    # ── 4. Initialise Filebrowser ────────────────────────────────────────────
-    setup_filebrowser(username, password)
-
-    # ── 5. Write Caddyfile ───────────────────────────────────────────────────
-    write_caddyfile()
-
-    # ── 6. Build and launch all services ─────────────────────────────────────
-    service_defs = build_services(username, password, env)
-
-    # Registry: name -> (proc, cmd, svc_env, is_critical)
-    registry: dict = {}
-    for svc_name, (cmd, svc_env, is_critical) in service_defs.items():
-        proc = launch(
-            cmd, svc_name,
-            env=svc_env,
-            silent=True,
-            critical=is_critical,
-        )
-        registry[svc_name] = (proc, cmd, svc_env, is_critical)
-
-    # ── 7. Graceful shutdown handler ─────────────────────────────────────────
-    def _shutdown(signum, _frame) -> None:
-        log.info("Shutting down all services…")
-        for svc_name, (proc, _, _svc_env, _crit) in registry.items():
-            if proc is None or proc.poll() is not None:
-                continue
-            try:
-                proc.terminate()
-                proc.wait(timeout=5)
-                log.info("  Stopped %s (PID %d)", svc_name, proc.pid)
-            except subprocess.TimeoutExpired:
-                log.warning("  Force-killing %s (PID %d)", svc_name, proc.pid)
-                proc.kill()
-                proc.wait(timeout=3)
-            except Exception as exc:
-                log.error("  Error stopping %s: %s", svc_name, exc)
-        sys.exit(0)
-
-    signal.signal(signal.SIGINT, _shutdown)
-    signal.signal(signal.SIGTERM, _shutdown)
-
-    # ── 8. Health-monitor loop ───────────────────────────────────────────────
-    log.info("All services launched. Entering health-monitor loop…")
-    try:
-        while True:
-            time.sleep(HEALTH_CHECK_INTERVAL)
-            for svc_name in list(registry.keys()):
-                proc, cmd, svc_env, is_critical = registry[svc_name]
-                if proc is None or proc.poll() is None:
-                    continue
-
-                log.warning(
-                    "%s (PID %d) exited with code %d — restarting…",
-                    svc_name,
-                    proc.pid,
-                    proc.returncode,
-                )
-                if cmd is None:
-                    log.warning("  No command for %s — skipping.", svc_name)
-                    continue
-
-                new_proc = launch(
-                    cmd, svc_name,
-                    env=svc_env,
-                    silent=True,
-                    critical=is_critical,
-                )
-                registry[svc_name] = (new_proc, cmd, svc_env, is_critical)
-
-    except KeyboardInterrupt:
-        _shutdown(None, None)
-
+        # প্রসেস রান করা
+        subprocess.run(command, check=True)
+    except Exception as e:
+        print(f"Error starting terminal: {e}")
 
 if __name__ == "__main__":
-    main()
+    start_terminal()