#!/usr/bin/env python3 """WorkBuddy / CodeBuddy (Tencent) daily check-in client. Endpoints (reverse-engineered from the official desktop client, see community projects cpa-plugin/workbuddy and wb-trae-checkin): region CN base https://www.codebuddy.cn (daily check-in supported) region Global base https://www.workbuddy.ai (no daily check-in) status POST {base}/v2/billing/meter/checkin-activity-status claim POST {base}/v2/billing/meter/daily-checkin credits POST {base}/v2/billing/meter/get-user-resource refresh POST https://copilot.tencent.com/v2/plugin/auth/token/refresh (Global: https://www.workbuddy.ai/v2/plugin/auth/token/refresh) oauth GET https://copilot.tencent.com/v2/plugin/auth/state?platform=CLI GET https://copilot.tencent.com/v2/plugin/auth/token?state=... GET https://copilot.tencent.com/v2/plugin/login/account?state=... Result codes used across the app: ok / already / fail / no_session / inactive. """ import json import time from common import (http_post_json, http_get, log, now_ts, fmt_now) CN_BASE = "https://www.codebuddy.cn" GLOBAL_BASE = "https://www.workbuddy.ai" REFRESH_URL_CN = "https://copilot.tencent.com/v2/plugin/auth/token/refresh" REFRESH_URL_GLOBAL = "https://www.workbuddy.ai/v2/plugin/auth/token/refresh" AUTH_BASE = "https://copilot.tencent.com" EP_STATUS = "/v2/billing/meter/checkin-activity-status" EP_DAILY_CHECKIN = "/v2/billing/meter/daily-checkin" EP_USER_RESOURCE = "/v2/billing/meter/get-user-resource" REFRESH_MARGIN = 3 * 24 * 3600 # refresh 3 days before expiry SESSION_DEAD = ("12153", "Offline user session not found") def is_global(auth): return (auth.get("domain") or "").lower() in ("workbuddy.ai", "global") def billing_base(auth): return GLOBAL_BASE if is_global(auth) else CN_BASE def refresh_url(auth): return REFRESH_URL_GLOBAL if is_global(auth) else REFRESH_URL_CN def billing_headers(auth, account=None): account = account or {} headers = {"Authorization": "Bearer %s" % auth.get("accessToken", "")} if account.get("uid"): headers["X-User-Id"] = str(account["uid"]) if account.get("enterpriseId"): headers["X-Enterprise-Id"] = str(account["enterpriseId"]) headers["X-Tenant-Id"] = str(account["enterpriseId"]) if auth.get("domain"): headers["X-Domain"] = auth["domain"] base = billing_base(auth) headers["Referer"] = base + "/" headers["Origin"] = base return headers # --------------------------------------------------------------------------- # Token lifecycle # --------------------------------------------------------------------------- def _expires_at(auth): exp = auth.get("expiresAt") or 0 try: exp = int(exp) except (TypeError, ValueError): return 0 if exp > 10 ** 12: # milliseconds exp //= 1000 return exp def token_expiring(auth, margin=REFRESH_MARGIN): exp = _expires_at(auth) if not exp: return False return now_ts() + margin > exp def refresh_token(account): """Refresh access token in place. Returns (ok, message).""" auth = account.setdefault("auth", {}) rt = auth.get("refreshToken", "") if not rt: return False, "未配置 refreshToken，无法自动续期" headers = { "X-Refresh-Token": rt, "X-Auth-Refresh-Source": "workbuddy", } if (account.get("account") or {}).get("enterpriseId"): headers["X-Enterprise-Id"] = str(account["account"]["enterpriseId"]) data, status, err = http_post_json(refresh_url(auth), body={}, headers=headers) if err or not data: if status == 401 and any(m in (err or "") for m in SESSION_DEAD): account["disabled"] = True account["note"] = "会话已失效，需要重新授权" return False, "登录态已失效，请重新授权" return False, "续期失败（HTTP %s）：%s" % (status, err) if not isinstance(data, dict) or not data.get("accessToken"): return False, "续期响应缺少 accessToken" auth["accessToken"] = data["accessToken"] if data.get("refreshToken"): auth["refreshToken"] = data["refreshToken"] if data.get("domain"): auth["domain"] = data["domain"] if data.get("expiresAt") or data.get("expires_at"): auth["expiresAt"] = data.get("expiresAt") or data.get("expires_at") account["note"] = "" return True, "已续期" # --------------------------------------------------------------------------- # Billing API # --------------------------------------------------------------------------- def query_status(account): """Return (checked, detail, error). checked is True/False/None.""" auth = account.get("auth", {}) headers = billing_headers(auth, account.get("account")) data, status, err = http_post_json(billing_base(auth) + EP_STATUS, body={}, headers=headers) if err or data is None: return None, None, err or "empty response" if status in (401, 403) or any(m in (err or "") for m in SESSION_DEAD): return None, None, "no_session" d = data.get("data") if isinstance(data, dict) else None d = d if isinstance(d, dict) else {} for key in ("today_checked_in", "checked_in", "checkedIn", "hasCheckin", "isCheckedIn", "checked"): if isinstance(d.get(key), bool): return d[key], d, None if isinstance(data.get(key), bool): return data[key], d, None return None, d, None def perform_checkin(account): """Claim today's credits. Returns (status, message, points). status: ok / already / fail / no_session """ auth = account.get("auth", {}) headers = billing_headers(auth, account.get("account")) data, status, err = http_post_json(billing_base(auth) + EP_DAILY_CHECKIN, body={}, headers=headers) if status in (401, 403): return "no_session", "登录态已失效，请重新授权", 0 if data is None: return "fail", "请求失败（HTTP %s：%s）" % (status, err), 0 code = data.get("code") if isinstance(data, dict) else 0 if code in (10001,): return "already", "今天已签到", 0 if code in (0, 200) or code is None: d = data.get("data") if isinstance(data, dict) else {} d = d if isinstance(d, dict) else {} points = d.get("today_credit", d.get("daily_credit", d.get("points", 0))) return "ok", "签到成功", points or 0 if any(m in (err or "") for m in SESSION_DEAD): return "no_session", "登录态已失效，请重新授权", 0 return "fail", "领取返回 code=%s %s" % (code, (data.get("msg") or "")[:80]), 0 def fetch_credits(account): """Aggregate credit packages. Returns (summary, error).""" auth = account.get("auth", {}) headers = billing_headers(auth, account.get("account")) body = { "PageNumber": 1, "PageSize": 100, "ProductCode": "p_tcaca", "Status": [0, 3], "PackageEndTimeRangeBegin": time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()), "PackageEndTimeRangeEnd": "2126-12-31 23:59:59", } data, status, err = http_post_json(billing_base(auth) + EP_USER_RESOURCE, body=body, headers=headers) if err or data is None: if status in (401, 403): return None, "no_session" return None, err or "查询失败" resp = data if isinstance(data, dict) and "Response" in data: resp = data.get("Response") or {} resp_data = resp.get("Data", {}) if isinstance(resp, dict) else {} accounts = resp_data.get("Accounts", []) if isinstance(resp_data, dict) else [] if not isinstance(accounts, list): accounts = [] total_remain = total_used = total_size = 0 packages = [] for pkg in accounts: if not isinstance(pkg, dict): continue remain, used, size = _package_remain_used(pkg) total_remain += remain total_used += used total_size += size packages.append({ "name": pkg.get("PackageName", ""), "remain": remain, "used": used, "size": size, }) dosage = resp_data.get("TotalDosage", 0) if isinstance(dosage, (int, float)) and dosage > total_size: total_size = dosage derived = total_size - total_remain if derived > total_used: total_used = derived return { "remain": total_remain, "used": total_used, "size": total_size, "packages": packages, }, None def _package_remain_used(pkg): def num(v): return v if isinstance(v, (int, float)) else 0 cycle_size = num(pkg.get("CycleCapacitySize")) if cycle_size > 0: remain = max(num(pkg.get("CycleCapacityRemain")), 0) remain = min(remain, cycle_size) used = cycle_size - remain explicit = num(pkg.get("CycleCapacityUsed")) if explicit > used: used = explicit if cycle_size >= used: remain = cycle_size - used return remain, used, cycle_size cycle_remain = num(pkg.get("CycleCapacityRemain")) cycle_used = num(pkg.get("CycleCapacityUsed")) if cycle_remain > 0 or cycle_used > 0: remain = max(cycle_remain, 0) used = max(cycle_used, 0) size = remain + used cap = num(pkg.get("CapacitySize")) if cap > size: size = cap if size >= remain: used = size - remain return remain, used, size remain = max(num(pkg.get("CapacityRemain")), 0) used = max(num(pkg.get("CapacityUsed")), 0) size = num(pkg.get("CapacitySize")) if size <= 0: size = remain + used if used == 0 and size > remain: used = size - remain return remain, used, size # --------------------------------------------------------------------------- # OAuth web login (CLI-style device flow, driven from the NAS web UI) # --------------------------------------------------------------------------- def begin_oauth(): """Start a login session. Returns (state, login_url, error).""" data, status, err = http_post_json(AUTH_BASE + "/v2/plugin/auth/state?platform=CLI", body={}, timeout=15) if err or not isinstance(data, dict): return None, None, "获取授权状态失败（HTTP %s：%s）" % (status, err) d = data.get("data") if isinstance(data.get("data"), dict) else data state = d.get("state") or d.get("State") if not state: return None, None, "未能获取授权 state" login_url = "%s/login?state=%s&platform=CLI" % (AUTH_BASE, state) return state, login_url, None def poll_oauth(state): """Poll the token endpoint. Returns (status, payload, error). status: waiting / done / failed """ data, status_code, err = http_get( AUTH_BASE + "/v2/plugin/auth/token?state=%s" % state, timeout=15) if status_code == 200 and isinstance(data, dict): code = data.get("code", 0) d = data.get("data") if isinstance(data.get("data"), dict) else {} if code in (0, 200) and (d.get("accessToken") or d.get("access_token")): expires_at = d.get("expiresAt") or d.get("expires_at") or 0 if not expires_at and (d.get("expires_in") or d.get("expiresIn")): from common import now_ts expires_at = now_ts() + int(d.get("expires_in") or d.get("expiresIn")) payload = { "accessToken": d.get("accessToken") or d.get("access_token"), "refreshToken": d.get("refreshToken") or d.get("refresh_token"), "expiresAt": expires_at, "domain": d.get("domain") or "codebuddy.cn", } account = fetch_account_info(state, payload["accessToken"]) if account: payload["account"] = account return "done", payload, None # not logged in yet (or state expired) — keep waiting within the UI timeout return "waiting", None, None def fetch_account_info(state, access_token): """Best-effort account info for the OAuth session.""" headers = {"Authorization": "Bearer %s" % access_token} if access_token else None data, status, err = http_get( AUTH_BASE + "/v2/plugin/login/account?state=%s" % state, headers=headers, timeout=15) if not isinstance(data, dict): return None d = data.get("data") if isinstance(data.get("data"), dict) else data uid = d.get("uid") or d.get("userId") or d.get("Uid") nickname = d.get("nickname") or d.get("name") or d.get("Nickname") enterprise = d.get("enterpriseId") or d.get("EnterpriseId") or d.get("enterprise_id") if not (uid or nickname): return None return { "uid": str(uid) if uid else "", "nickname": str(nickname) if nickname else "", "enterpriseId": str(enterprise) if enterprise else "", } def account_display(account): acct = account.get("account") or {} return acct.get("nickname") or acct.get("uid") or account.get("name") or "WorkBuddy 账号"