中控 Bridge Token 为空时自动生成,Origin 改为选填说明。
打开 env 配置或首次部署即可写入 Token,便于复制到中控登记。 Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
@@ -4,6 +4,7 @@ from __future__ import annotations
|
||||
import hashlib
|
||||
import hmac
|
||||
import os
|
||||
import secrets
|
||||
import time
|
||||
from typing import Optional
|
||||
from urllib.parse import urlencode
|
||||
@@ -94,3 +95,28 @@ def frame_ancestors_header_value() -> Optional[str]:
|
||||
return "*"
|
||||
parts = [p.strip() for p in raw.split() if p.strip()]
|
||||
return " ".join(parts) if parts else None
|
||||
|
||||
|
||||
def ensure_hub_bridge_token(env_path: str) -> str:
|
||||
"""若 .env 中 HUB_BRIDGE_TOKEN 为空则生成并写入;返回当前 token."""
|
||||
from lib.env.env_file_lib import apply_env_updates, env_get, read_env_lines
|
||||
|
||||
path = (env_path or "").strip()
|
||||
if not path or not os.path.isfile(path):
|
||||
existing = hub_bridge_token()
|
||||
if existing:
|
||||
return existing
|
||||
token = secrets.token_urlsafe(32)
|
||||
os.environ["HUB_BRIDGE_TOKEN"] = token
|
||||
return token
|
||||
|
||||
lines = read_env_lines(path)
|
||||
cur = (env_get(lines, "HUB_BRIDGE_TOKEN") or "").strip()
|
||||
if cur:
|
||||
os.environ["HUB_BRIDGE_TOKEN"] = cur
|
||||
return cur
|
||||
|
||||
token = secrets.token_urlsafe(32)
|
||||
apply_env_updates(path, {"HUB_BRIDGE_TOKEN": token})
|
||||
os.environ["HUB_BRIDGE_TOKEN"] = token
|
||||
return token
|
||||
|
||||
Reference in New Issue
Block a user