coze_lab 0.1.44 → 0.1.45

package/scripts/shared/cozeloop_refresh.py

@@ -1,57 +0,0 @@
-#!/usr/bin/env python3
-"""
-CozeLoop token refresh hook — runs at SessionStart.
-Checks credentials.json and refreshes the token if expiring soon.
-Writes the fresh token back so subsequent Stop hooks pick it up.
-"""
-import json, os, sys, time, urllib.request
-from pathlib import Path
-
-CLIENT_ID = "08972682140163281554629748278108.app.coze"
-COZE_API  = "https://api.coze.cn"
-THRESHOLD = 10 * 60  # 10 minutes
-CREDS     = Path.home() / ".cozeloop" / "credentials.json"
-
-def load():
-    try: return json.loads(CREDS.read_text())
-    except: return None
-
-def save(c):
-    CREDS.parent.mkdir(parents=True, exist_ok=True)
-    CREDS.write_text(json.dumps(c, indent=2))
-    os.chmod(CREDS, 0o600)
-
-def refresh(rt):
-    try:
-        body = json.dumps({"grant_type":"refresh_token","client_id":CLIENT_ID,"refresh_token":rt}).encode()
-        req = urllib.request.Request(f"{COZE_API}/api/permission/oauth2/token",
-            data=body, headers={
-                "Content-Type":"application/json",
-            })
-        with urllib.request.urlopen(req, timeout=10) as r:
-            d = json.loads(r.read())
-        if d.get("access_token"):
-            existing = load() or {}
-            save({"access_token":d["access_token"],
-                  "refresh_token":d.get("refresh_token",rt),
-                  "expires_at":d.get("expires_in",0)*1000,
-                  "workspace_id":existing.get("workspace_id","")})
-            return True
-    except Exception as e:
-        print(f"[cozeloop_refresh] refresh failed: {e}", file=sys.stderr)
-    return False
-
-def main():
-    creds = load()
-    if not creds: return
-    # expires_at 由 index.js 以毫秒存储；先减去当前毫秒时间再换算成秒，与 THRESHOLD(秒) 比较。
-    remaining = (creds.get("expires_at",0) - time.time()*1000)/1000
-    if remaining > THRESHOLD: return  # still fresh
-    print(f"[cozeloop_refresh] token expiring in {int(remaining)}s, refreshing...", file=sys.stderr)
-    if creds.get("refresh_token"):
-        if refresh(creds["refresh_token"]):
-            print("[cozeloop_refresh] token refreshed OK", file=sys.stderr)
-        else:
-            print("[cozeloop_refresh] refresh failed, token may expire soon", file=sys.stderr)
-
-main()