sub2api 开通授初始余额:余额为 0 的 key 在网关侧直接 403

This commit is contained in:
host committed 2026-10-27 17:22:54 +08:00
1 parent ec47328d3c
commit dbfa026a4e
3 files changed
+85 -4

No files matched your search

+4
View File
@@ -1033,6 +1033,10 @@ interrupted 且终态任务 cancel 返回 400、心跳后 sweep 不扫刚心跳
任务直接 failed,绝不跑错网关。协调员子任务无 created_by 时
跟父任务属主走。key 明文存 PG,与 GitHub App 私钥同一信任
边界(库只在宿主 loopback 可达);at-rest 加密是后续项。
开通同时授初始余额:`CLOUD_SUB2API_OPENING_BALANCE_USD`
(默认 10 美元,显式配 0 表示不授;非法/负值回落默认)——
网关鉴权层在余额 ≤ 0 时直接 403,不授余额的 key 立刻不可
用;授余额失败时 key 与账号一并清理再抛错,不留孤儿。
- 注入:worker 在三条执行路径(Local popen、Local 同步、Docker
同步)都注入任务级 env:`SUB2API_API_KEY`(属主的 key)、
`SUB2API_BASE_URL`,并把网关 host 追加进 NO_PROXY(否则
+36 -4
View File
@@ -73,6 +73,24 @@ def gateway_group_name():
or "codex-system"
def opening_balance():
"""开通时授的初始余额(美元):默认 10,显式值优先。
新用户余额为 0 时网关鉴权层直接 403(余额门在任何上游路由
之前),所以开通必须同时授余额,否则发出的 key 立刻不可用。
显式配 0 表示不授(调用方自理);非法/负值回落默认 10。
"""
raw = os.environ.get(
"CLOUD_SUB2API_OPENING_BALANCE_USD", "").strip()
if not raw:
return 10.0
try:
value = float(raw)
except ValueError:
return 10.0
return value if value >= 0 else 10.0
def ensure_gateway_keys_table(conn):
conn.execute(GATEWAY_TABLE_SQL)
@@ -181,6 +199,16 @@ class Sub2apiAdmin:
"DELETE", "/api/v1/admin/users/%s" % sub_user_id)
return st in (200, 204)
def set_balance(self, sub_user_id, amount):
st, data = self._call(
"POST", "/api/v1/admin/users/%s/balance" % sub_user_id,
{"balance": amount, "operation": "set",
"notes": "codex provisioning opening balance"})
if st != 200:
raise Sub2apiError(
"sub2api 授余额失败: %s"
% ((data or {}).get("message") or st), status=st)
def list_groups(self):
st, data = self._call("GET", "/api/v1/admin/groups")
payload = (data or {}).get("data")
@@ -277,11 +305,12 @@ def list_gateway_models(origin, api_key, request=None):
def ensure_user_gateway_key(conn, user_id, admin=None, request=None):
"""幂等开通:已有行直接返回 key;否则建号->登录->发 key->入库。
"""幂等开通:已有行直接返回 key;否则建号->登录->发 key->授余额->入库。
入库失败时尽力清理刚建的 key 与账号(避免孤儿 key),再抛原错。
调用方负责 conn 的提交语义:本函数在入库后 commit(开通是独立
事务,不跟调用方的业务事务绑在一起)。
入库失败时尽力清理刚建的 key 与账号(避免孤儿 key),再抛原错;
授余额失败同样走这条清理(余额为 0 的 key 在网关侧直接 403,
留着没有意义)。调用方负责 conn 的提交语义:本函数在入库后
commit(开通是独立事务,不跟调用方的业务事务绑在一起)。
"""
ensure_gateway_keys_table(conn)
row = get_user_gateway_key(conn, user_id)
@@ -304,6 +333,9 @@ def ensure_user_gateway_key(conn, user_id, admin=None, request=None):
key_id, key = user_create_key(origin, user_token,
"codex-task-key", group_id,
request=req)
amount = opening_balance()
if amount > 0:
admin.set_balance(sub_id, amount)
now = time.time()
conn.execute(
"INSERT INTO user_gateway_keys(user_id, sub2api_user_id, "
+45
View File
@@ -36,6 +36,8 @@ class FakeApi:
self.next_user_id = 10
self.next_key_id = 20
self.fail_refresh_once = False
self.balance_calls = []
self.fail_balance = False
def __call__(self, method, url, body=None, token=None):
path = url.split("28084", 1)[-1] if "28084" in url else url
@@ -52,6 +54,12 @@ class FakeApi:
return 200, {"data": {"access_token": "new-tok",
"refresh_token": "new-ref",
"expires_in": 3600}}
if path.startswith("/api/v1/admin/users/") \
and path.endswith("/balance") and method == "POST":
self.balance_calls.append(body)
if self.fail_balance:
return 400, {"message": "bad balance"}
return 200, {"data": {"balance": body.get("balance")}}
if path == "/api/v1/admin/users" and method == "POST":
self.next_user_id += 1
return 200, {"data": {"id": self.next_user_id}}
@@ -135,6 +143,43 @@ class TestEnsureGatewayKey(unittest.TestCase):
request=self.api)
self.assertEqual(models, ["gpt-5", "claude-x"])
def test_provision_grants_opening_balance(self):
conn = _conn()
key = ensure_user_gateway_key(conn, "user-bal", admin=self.admin,
request=self.api)
self.assertTrue(key.startswith("sk-probe-"))
self.assertEqual(len(self.api.balance_calls), 1)
self.assertEqual(self.api.balance_calls[0]["balance"], 10.0)
self.assertEqual(self.api.balance_calls[0]["operation"], "set")
def test_provision_balance_failure_cleans_up(self):
self.api.fail_balance = True
conn = _conn()
with self.assertRaises(sub2api.Sub2apiError):
ensure_user_gateway_key(conn, "user-fail", admin=self.admin,
request=self.api)
paths = [(m, p) for (m, p, _b, _t) in self.api.calls]
self.assertIn(("DELETE", "/api/v1/keys/21"), paths)
self.assertIn(("DELETE", "/api/v1/admin/users/11"), paths)
row = conn.execute(
"SELECT COUNT(*) FROM user_gateway_keys"
" WHERE user_id = ?", ("user-fail",)).fetchone()
self.assertEqual(row[0], 0)
def test_provision_zero_opening_balance_skips_grant(self):
old = os.environ.get("CLOUD_SUB2API_OPENING_BALANCE_USD")
os.environ["CLOUD_SUB2API_OPENING_BALANCE_USD"] = "0"
try:
conn = _conn()
ensure_user_gateway_key(conn, "user-zero", admin=self.admin,
request=self.api)
finally:
if old is None:
os.environ.pop("CLOUD_SUB2API_OPENING_BALANCE_USD", None)
else:
os.environ["CLOUD_SUB2API_OPENING_BALANCE_USD"] = old
self.assertEqual(self.api.balance_calls, [])
class TestProviderCommand(unittest.TestCase):
def test_server_command_sub2api(self):