refactor(credential): STS 刷新中间件默认启用，去掉 FC 环境特殊判断

之前默认仅在 FC 环境（FC_REGION 存在）才启用。按需求改为**默认启用**，不再
特殊判断 FC 环境；仅在环境变量 AGENTRUN_STS_REFRESH_ENABLED 设为假值
（0/false/no/off）或构造参数 enabled=False 时关闭。中间件在 x-fc-* 头不齐全
时本就透传、无副作用，故默认开是安全的。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Signed-off-by: OhYee <oyohyee@oyohyee.com>

Author: OhYee

agentrun/server/server.py

@@ -124,9 +124,10 @@ def __init__(
         """
         self.app = FastAPI(title="AgentRun Server")
 
-        # 注入 STS 刷新中间件：从每次请求的 FC 头解析最新 STS 临时凭证，写入
+        # 注入 STS 刷新中间件：从每次请求的 x-fc-* 头解析最新 STS 临时凭证，写入
         # 请求级 overlay，使本次请求内所有 Config/client 静默使用最新凭证。
-        # 无条件启用：未携带相关头时（本地/非 FC）中间件不产生任何副作用。
+        # 默认启用；未携带相关头时不产生任何副作用。如需关闭设环境变量
+        # AGENTRUN_STS_REFRESH_ENABLED=false。
         self.app.add_middleware(StsRefreshMiddleware)
 
         # 如果启用了 memory，包装 invoke_agent

agentrun/server/sts_middleware.py

@@ -14,13 +14,17 @@
 头名可配置 / Configurable header names:
     构造参数 > 环境变量 > 默认值(``x-fc-*``)。头名大小写不敏感。
 
+启用开关 / Enable switch:
+    中间件**默认启用**。仅在特定情况下关闭：构造参数 ``enabled=False``，或
+    环境变量 ``AGENTRUN_STS_REFRESH_ENABLED`` 设为假值（``0`` / ``false`` /
+    ``no`` / ``off``）。
+
 信任边界 / Trust boundary:
-    overlay 会覆盖运维方在环境变量里配置的凭证，因此只在**可信来源**提供
-    ``x-fc-*`` 头时才能启用。函数计算（FC）拥有该头命名空间并会剥离客户端伪造的
-    同名头，故默认仅在检测到 FC 环境（``FC_REGION`` 存在）时生效；非 FC 环境
-    （如裸 uvicorn / 自有网关）默认关闭，避免不可信客户端注入 ``x-fc-*`` 头冒用
-    身份。可用构造参数 ``enabled`` 或环境变量 ``AGENTRUN_STS_REFRESH_ENABLED``
-    显式开关（开启前请确保已有前置鉴权或网关剥离这些头）。
+    overlay 仅在 ``x-fc-*`` 头**齐全**时注入（覆盖运维方 env 凭证），否则透传。
+    函数计算（FC）拥有该头命名空间并会剥离客户端伪造的同名头，FC 内安全。
+    **注意**：若部署在非 FC 环境（裸 uvicorn / 自有网关）且服务可被不可信客户端
+    直达，攻击者可注入 ``x-fc-*`` 头冒用身份——此类场景请前置鉴权 / 由网关剥离
+    这些头，或按上面的开关关闭本中间件。
 """
 
 from __future__ import annotations
@@ -37,12 +41,15 @@
 
 
 def _detect_enabled() -> bool:
-    """决定是否启用 overlay：显式环境变量优先，否则按是否在 FC 环境自动判断。"""
+    """是否启用 overlay：**默认启用**，仅环境变量显式设为假值时关闭。
+
+    ``AGENTRUN_STS_REFRESH_ENABLED`` 未设置 -> 启用；设为
+    ``0`` / ``false`` / ``no`` / ``off`` -> 关闭；其余真值 -> 启用。
+    """
     flag = os.getenv("AGENTRUN_STS_REFRESH_ENABLED")
-    if flag is not None:
-        return flag.strip().lower() in ("1", "true", "yes", "on")
-    # FC 环境下 FC_REGION 必然存在；非 FC 默认关闭以防 x-fc-* 头注入。
-    return os.getenv("FC_REGION") is not None
+    if flag is None:
+        return True
+    return flag.strip().lower() in ("1", "true", "yes", "on")
 
 
 class StsRefreshMiddleware(BaseHTTPMiddleware):
@@ -58,7 +65,8 @@ def __init__(
         security_token_header: Optional[str] = None,
     ) -> None:
         super().__init__(app)
-        # enabled=None 时自动探测（FC 环境或显式环境变量开关）。
+        # enabled=None 时按环境变量决定（默认启用，
+        # AGENTRUN_STS_REFRESH_ENABLED 设为假值时关闭）。
         self._enabled = _detect_enabled() if enabled is None else enabled
         # 头名解析（参数 > 环境变量 > 默认）交由 sts_from_headers 处理，这里只存原值。
         self._ak_header = access_key_id_header

tests/unittests/test_sts_refresh.py

@@ -173,7 +173,7 @@ def _build_app():
     from agentrun.server.sts_middleware import StsRefreshMiddleware
 
     app = FastAPI()
-    # 测试环境无 FC_REGION，需显式启用（非 FC 默认关闭以防头注入）。
+    # 显式启用，避免受 dev 环境 AGENTRUN_STS_REFRESH_ENABLED 影响。
     app.add_middleware(StsRefreshMiddleware, enabled=True)
 
     @app.get("/async")
@@ -334,49 +334,68 @@ def test_middleware_partial_headers_ignored(monkeypatch):
     assert resp == {"ak": "ENV_AK", "sts": ""}
 
 
-def test_middleware_disabled_off_fc(monkeypatch):
-    """H2：非 FC（无 FC_REGION、未显式 enable）时不应应用 overlay。"""
+def test_middleware_enabled_by_default(monkeypatch):
+    """默认启用（不看 FC 环境，无需任何开关）。"""
     from fastapi import FastAPI
     from fastapi.testclient import TestClient
 
     from agentrun.server.sts_middleware import StsRefreshMiddleware
 
     monkeypatch.delenv("FC_REGION", raising=False)
     monkeypatch.delenv("AGENTRUN_STS_REFRESH_ENABLED", raising=False)
+
+    app = FastAPI()
+    app.add_middleware(StsRefreshMiddleware)  # enabled=None → 默认启用
+
+    @app.get("/x")
+    async def _x():
+        return {"ak": Config().get_access_key_id()}
+
+    client = TestClient(app)
+    # 默认开：携带完整 x-fc-* 头即生效。
+    assert client.get("/x", headers=_HEADERS).json() == {"ak": "H_AK"}
+
+
+def test_middleware_disabled_via_env(monkeypatch):
+    """仅在 AGENTRUN_STS_REFRESH_ENABLED 设为假值时关闭。"""
+    from fastapi import FastAPI
+    from fastapi.testclient import TestClient
+
+    from agentrun.server.sts_middleware import StsRefreshMiddleware
+
+    monkeypatch.setenv("AGENTRUN_STS_REFRESH_ENABLED", "false")
     monkeypatch.setenv("AGENTRUN_ACCESS_KEY_ID", "ENV_AK")
 
     app = FastAPI()
-    app.add_middleware(StsRefreshMiddleware)  # enabled=None → auto-detect
+    app.add_middleware(StsRefreshMiddleware)  # enabled=None → 读环境变量 -> 关闭
 
     @app.get("/x")
     async def _x():
         return {"ak": Config().get_access_key_id()}
 
     client = TestClient(app)
-    # 即便携带完整 x-fc-* 头，非 FC 环境也不应覆盖 env 凭证（防注入）。
-    resp = client.get("/x", headers=_HEADERS).json()
-    assert resp == {"ak": "ENV_AK"}
+    # 已关闭：即便携带完整 x-fc-* 头也不覆盖 env 凭证。
+    assert client.get("/x", headers=_HEADERS).json() == {"ak": "ENV_AK"}
 
 
-def test_middleware_enabled_on_fc(monkeypatch):
-    """H2：检测到 FC_REGION 时自动启用。"""
+def test_middleware_disabled_via_constructor(monkeypatch):
+    """构造参数 enabled=False 显式关闭。"""
     from fastapi import FastAPI
     from fastapi.testclient import TestClient
 
     from agentrun.server.sts_middleware import StsRefreshMiddleware
 
-    monkeypatch.setenv("FC_REGION", "cn-hangzhou")
-    monkeypatch.delenv("AGENTRUN_STS_REFRESH_ENABLED", raising=False)
+    monkeypatch.setenv("AGENTRUN_ACCESS_KEY_ID", "ENV_AK")
 
     app = FastAPI()
-    app.add_middleware(StsRefreshMiddleware)  # enabled=None → auto-detect FC
+    app.add_middleware(StsRefreshMiddleware, enabled=False)
 
     @app.get("/x")
     async def _x():
         return {"ak": Config().get_access_key_id()}
 
     client = TestClient(app)
-    assert client.get("/x", headers=_HEADERS).json() == {"ak": "H_AK"}
+    assert client.get("/x", headers=_HEADERS).json() == {"ak": "ENV_AK"}
 
 
 def test_invoker_sync_path_sees_overlay():