's'

web2.py

@@ -0,0 +1,279 @@
+from fastapi import FastAPI, Query, Request
+from fastapi.responses import FileResponse, JSONResponse
+from pathlib import Path
+import asyncio
+import re
+from fastapi.middleware.cors import CORSMiddleware
+
+app = FastAPI(title="多进程录制控制台")
+
+# ---------------- 配置（根目录结构） ----------------
+BASE_DIR = Path(__file__).parent
+CONFIG_DIR = BASE_DIR / "config"
+DEFAULT_LOG_DIR = BASE_DIR / ".pm2" / "logs"
+
+MAX_LOG_LINES = 300
+
+# 本分支用文件名带「2」与另一套无后缀分支区分；典型部署为 Ubuntu + PM2。
+# 与 index.html 中 PROCESS_MONITOR 保持一致：pm2=PM2 应用名，script=入口脚本
+PROCESS_MONITOR = (
+    {"pm2": "tiktok", "script": "tiktok2.py", "label": "TikTok"},
+    {"pm2": "youtube", "script": "youtube2.py", "label": "YouTube"},
+    {"pm2": "obs", "script": "obs2.sh", "label": "OBS"},
+)
+VALID_PROCESSES = frozenset(p["pm2"] for p in PROCESS_MONITOR)
+# youtube.ini 仅对与 youtube2.py 绑定的 PM2 名开放
+YOUTUBE_PM2 = next(p["pm2"] for p in PROCESS_MONITOR if p["script"] == "youtube2.py")
+# URL_config.ini：tiktok2 / youtube2 两条入口共用
+URL_CONFIG_PM2 = frozenset(
+    p["pm2"] for p in PROCESS_MONITOR if p["script"] in ("tiktok2.py", "youtube2.py")
+)
+
+# CORS
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["GET", "POST"],
+    allow_headers=["*"],
+)
+
+# ---------------- 工具函数 ----------------
+def strip_ansi_codes(text: str) -> str:
+    return re.sub(r'\x1b\[[0-9;]*m', '', text)
+
+
+async def run_pm2(cmd: str, timeout: float = 15.0) -> str:
+    process = await asyncio.create_subprocess_shell(
+        f"pm2 {cmd}",
+        stdout=asyncio.subprocess.PIPE,
+        stderr=asyncio.subprocess.PIPE,
+    )
+    try:
+        stdout, stderr = await asyncio.wait_for(process.communicate(), timeout=timeout)
+        result = (stdout or stderr).decode(errors="ignore").strip()
+        return result if result else "命令执行完成（无输出）"
+    except asyncio.TimeoutError:
+        process.kill()
+        await process.wait()
+        return "ERROR: PM2 命令超时，已强制终止"
+
+
+async def force_kill_ffmpeg(process_name: str):
+    """尝试强杀与该进程相关的 ffmpeg"""
+    cmds = [
+        # 优先杀带有进程名的 ffmpeg（最精准）
+        f"pkill -f 'ffmpeg.*{process_name}' || true",
+        # 再杀所有 ffmpeg（兜底，比较暴力）
+        "killall -9 ffmpeg 2>/dev/null || true",
+    ]
+
+    for cmd in cmds:
+        proc = await asyncio.create_subprocess_shell(
+            cmd,
+            stdout=asyncio.subprocess.DEVNULL,
+            stderr=asyncio.subprocess.DEVNULL,
+        )
+        await proc.wait()
+        await asyncio.sleep(0.4)  # 稍微间隔，避免竞争
+
+
+async def read_log_path(path: str, lines: int) -> str:
+    if not path:
+        return "无日志路径\n"
+    if "/dev/null" in path:
+        return "日志输出被禁用（PM2 配置为无日志）\n"
+    log_file = Path(path)
+    if not log_file.exists():
+        return f"日志文件不存在: {path}\n"
+    try:
+        with open(log_file, "r", encoding="utf-8", errors="ignore") as f:
+            content_lines = f.readlines()
+            return "".join(content_lines[-lines:]) if content_lines else "无日志内容\n"
+    except Exception as e:
+        return f"读取日志失败 ({path}): {str(e)}\n"
+
+
+async def get_process_info(process: str) -> dict:
+    describe_raw = await run_pm2(f"describe {process}", timeout=10.0)
+    describe_clean = strip_ansi_codes(describe_raw)
+
+    if any(x in describe_clean.lower() for x in ["no such process", "unknown process", "not found"]) or not describe_clean.strip():
+        return {
+            "process_status": "not_found",
+            "out_path": None,
+            "err_path": None,
+        }
+
+    status = "unknown"
+    out_path = None
+    err_path = None
+
+    for line in describe_raw.splitlines():
+        line_clean = strip_ansi_codes(line)
+        if line_clean.count('│') < 2:
+            continue
+        parts = [p.strip() for p in line_clean.split('│') if p.strip()]
+        if len(parts) >= 2:
+            key = parts[0].lower()
+            value = parts[1]
+            if key == "status":
+                status = value.lower()
+            elif key == "out_log_path":
+                out_path = value
+            elif key == "err_log_path":
+                err_path = value
+
+    if not out_path:
+        out_path = str(DEFAULT_LOG_DIR / f"{process}-out.log")
+    if not err_path:
+        err_path = str(DEFAULT_LOG_DIR / f"{process}-error.log")
+
+    return {
+        "process_status": status,
+        "out_path": out_path,
+        "err_path": err_path,
+    }
+
+# ---------------- 配置路由（youtube.ini，仅 youtube） ----------------
+@app.get("/get_config")
+async def get_config(process: str = Query(..., description="进程名")):
+    if process != YOUTUBE_PM2:
+        return JSONResponse({"error": "仅 YouTube 入口（youtube2.py）对应进程支持此配置编辑"}, status_code=400)
+    
+    config_file = CONFIG_DIR / "youtube.ini"
+    if not config_file.exists():
+        return {"content": "# youtube.ini 文件不存在，将创建空文件\n"}
+    try:
+        content = config_file.read_text(encoding="utf-8")
+        return {"content": content}
+    except Exception as e:
+        return JSONResponse({"error": f"读取失败: {str(e)}"}, status_code=500)
+
+@app.post("/save_config")
+async def save_config(request: Request, process: str = Query(..., description="进程名")):
+    if process != YOUTUBE_PM2:
+        return JSONResponse({"error": "仅 YouTube 入口（youtube2.py）对应进程支持此配置编辑"}, status_code=400)
+    
+    try:
+        data = await request.json()
+        content = data.get("content", "")
+    except:
+        return JSONResponse({"error": "无效的 JSON 数据"}, status_code=400)
+    
+    config_file = CONFIG_DIR / "youtube.ini"
+    try:
+        CONFIG_DIR.mkdir(parents=True, exist_ok=True)
+        config_file.write_text(content, encoding="utf-8")
+        return {"message": "配置保存成功"}
+    except Exception as e:
+        return JSONResponse({"error": f"保存失败: {str(e)}"}, status_code=500)
+
+# ---------------- 配置路由（URL_config.ini，youtube 和 tiktok 共享） ----------------
+@app.get("/get_url_config")
+async def get_url_config(process: str = Query(..., description="进程名")):
+    if process not in URL_CONFIG_PM2:
+        return JSONResponse({"error": "仅 tiktok2.py / youtube2.py 对应进程支持 URL 配置编辑"}, status_code=400)
+    
+    config_file = CONFIG_DIR / "URL_config.ini"
+    if not config_file.exists():
+        return {"content": "# URL_config.ini 文件不存在，将创建空文件\n"}
+    try:
+        content = config_file.read_text(encoding="utf-8")
+        return {"content": content}
+    except Exception as e:
+        return JSONResponse({"error": f"读取失败: {str(e)}"}, status_code=500)
+
+@app.post("/save_url_config")
+async def save_url_config(request: Request, process: str = Query(..., description="进程名")):
+    if process not in URL_CONFIG_PM2:
+        return JSONResponse({"error": "仅 tiktok2.py / youtube2.py 对应进程支持 URL 配置编辑"}, status_code=400)
+    
+    try:
+        data = await request.json()
+        content = data.get("content", "")
+    except:
+        return JSONResponse({"error": "无效的 JSON 数据"}, status_code=400)
+    
+    config_file = CONFIG_DIR / "URL_config.ini"
+    try:
+        CONFIG_DIR.mkdir(parents=True, exist_ok=True)
+        config_file.write_text(content, encoding="utf-8")
+        return {"message": "配置保存成功"}
+    except Exception as e:
+        return JSONResponse({"error": f"保存失败: {str(e)}"}, status_code=500)
+
+# ---------------- PM2 控制路由 ----------------
+@app.get("/start")
+async def start(process: str = Query(..., description="进程名")):
+    if process not in VALID_PROCESSES:
+        return JSONResponse({"output": f"无效进程名: {process}"}, status_code=400)
+    output = await run_pm2(f"start {process}")
+    return JSONResponse({"output": output})
+
+
+@app.get("/stop")
+async def stop(process: str = Query(..., description="进程名")):
+    if process not in VALID_PROCESSES:
+        return JSONResponse({"output": f"无效进程名: {process}"}, status_code=400)
+
+    # 第一步：尝试优雅停止
+    pm2_output = await run_pm2(f"stop {process}")
+
+    # 第二步：等待 PM2 信号传播（通常 1~3 秒）
+    await asyncio.sleep(2.0)
+
+    # 第三步：主动查找并杀死 ffmpeg
+    await force_kill_ffmpeg(process)
+
+    # 第四步：再等一小会儿，确认
+    await asyncio.sleep(1.0)
+
+    return JSONResponse({
+        "message": "已执行 stop + 强杀 ffmpeg",
+        "pm2_output": pm2_output,
+        "note": "如果仍有残留，可多次点击停止或检查 pm2 logs"
+    })
+
+
+@app.get("/restart")
+async def restart(process: str = Query(..., description="进程名")):
+    if process not in VALID_PROCESSES:
+        return JSONResponse({"output": f"无效进程名: {process}"}, status_code=400)
+    output = await run_pm2(f"restart {process}")
+    # 重启后也建议清理一次残留（视情况可选）
+    await force_kill_ffmpeg(process)
+    return JSONResponse({"output": output})
+
+
+@app.get("/status")
+async def status(process: str = Query(..., description="进程名")):
+    if process not in VALID_PROCESSES:
+        return JSONResponse({
+            "raw_status": "",
+            "process_status": "invalid",
+            "recent_log": f"无效进程名: {process}",
+            "recent_error": ""
+        })
+
+    full_status_raw = await run_pm2("status", timeout=8.0)
+    full_status = strip_ansi_codes(full_status_raw)
+    info = await get_process_info(process)
+    recent_log = await read_log_path(info["out_path"], MAX_LOG_LINES)
+    recent_error = await read_log_path(info["err_path"], MAX_LOG_LINES)
+
+    if info["process_status"] == "not_found":
+        recent_log = "进程未注册到 PM2（可能被 pm2 delete 删除或从未保存）\n"
+        recent_error = ""
+
+    return JSONResponse({
+        "raw_status": full_status,
+        "process_status": info["process_status"],
+        "recent_log": recent_log,
+        "recent_error": recent_error,
+    })
+
+
+# ---------------- 首页 ----------------
+@app.get("/")
+async def index():
+    return FileResponse(BASE_DIR / "index.html")