dsh-qoder-connect 0.1.2 → 0.1.5

package/lib/index.d.ts

@@ -141,7 +141,7 @@ interface QoderProbeAction {
    * All are writes, which is why they share this route's in-process key
    * and loopback guards rather than the read-only status GET.
    */
-  action: 'probe' | 'clear' | 'refresh' | 'set-maximum-context-window';
+  action: 'probe' | 'clear' | 'refresh' | 'set-maximum-context-window' | 'clear-checkin-logs' | 'checkin';
   /** Target model id; required for `probe`. */
   model?: string;
   /** Requested value for `set-maximum-context-window`. */
@@ -259,6 +259,25 @@ type QoderWebStatus = {
    * loopback guard); it is never persisted and rotates per process.
    */
   probeKey?: string;
+  /**
+   * Daily check-in status record for this variant.
+   */
+  checkIn?: {
+    lastDate: string;
+    lastAt: number;
+    status: 'claimed' | 'already-claimed' | 'no-campaign' | 'error';
+    amount?: number | undefined;
+    message?: string | undefined;
+    logs?: readonly {
+      id: string;
+      date: string;
+      timestamp: number;
+      status: 'claimed' | 'already-claimed' | 'no-campaign' | 'error';
+      amount?: number | undefined;
+      campaignKey?: string | undefined;
+      message?: string | undefined;
+    }[] | undefined;
+  };
   /**
    * In-process key authorizing PAT writes, including clearing. Travels with
    * the document for the same reason `probeKey` does.
@@ -692,6 +711,17 @@ interface QoderAccountInfo {
   updatedAt: string;
 }
 //#endregion
+//#region src/qoder/transport/checkin.d.ts
+interface QoderCheckInResult {
+  variantId: string;
+  date: string;
+  timestamp: number;
+  status: 'claimed' | 'already-claimed' | 'no-campaign' | 'error';
+  amount?: number | undefined;
+  campaignKey?: string | undefined;
+  message?: string | undefined;
+}
+//#endregion
 //#region src/qoder/transport/index.d.ts
 interface QoderTransport {
   stream(options: GenerateOptions, model?: QoderCatalogModel): AsyncIterable<StreamChunk>;
@@ -700,6 +730,7 @@ interface QoderTransport {
     force?: boolean | undefined;
     signal?: AbortSignal | undefined;
   }): Promise<QoderAccountInfo>;
+  checkIn(signal?: AbortSignal): Promise<QoderCheckInResult>;
 }
 //#endregion
 //#region src/upstream.d.ts
@@ -840,6 +871,10 @@ declare class QoderUpstreamClient {
    * account's "unlimited" state.
    */
   fetchCredits(signal?: AbortSignal): Promise<QoderCredits>;
+  /**
+   * Run daily benefit check-in for this client's variant.
+   */
+  checkIn(signal?: AbortSignal): Promise<QoderCheckInResult>;
   /**
    * Stream one translated chat-completions request.
    *
@@ -1345,6 +1380,10 @@ interface Config {
   sidebarQuotaCN?: boolean;
   /** Show the global variant's sidebar quota card. */
   sidebarQuotaGlobal?: boolean;
+  /** Automatically check in daily at 10:00 (UTC+8) to claim credits for China variant. */
+  autoCheckInCN?: boolean;
+  /** Automatically check in daily at 10:00 (UTC+8) to claim credits for Global variant. */
+  autoCheckInGlobal?: boolean;
   /**
    * Sidebar quota refresh interval in milliseconds. One shared value (both
    * cards poll on it) because the two widgets hit the same rate-limited

package/lib/index.js

@@ -1,4 +1,4 @@
-import { A as QODER_GLOBAL_AUTH_PATH, B as qoderCredentialIdentity, C as probeModel, D as FALLBACK_QODER_MODELS, E as qoderCatalogPath, F as QODER_AUTH_FILENAME, G as qoderPluginDataDir, H as QODER_DATA_DIR_ENV, I as QODER_PAT_ENV_CN, K as qoderStateDir, L as QODER_PAT_ENV_GLOBAL, M as QODER_GLOBAL_STATUS_PATH, N as QODER_PROBE_PATH, O as QoderCatalog, P as QODER_STATUS_PATH, R as QoderCredentialStore, S as PROBE_EFFORT_CANDIDATES, T as QoderCatalogStore, U as QODER_DATA_DIR_NAME, V as qoderOwnAuthPath, W as qoderMachineIdPath, _ as modelInfoOf, a as QODER_HOST_HEARTBEAT_FILENAME, b as createQoderTransport, c as processStartTimeMs, d as writeHostHeartbeat, f as QODER_CONNECT_VERSION, g as kindFromQoderFailure, h as classifyUpstreamError, i as variantFor, j as QODER_GLOBAL_PROBE_PATH, k as QODER_AUTH_PATH, l as qoderHostHeartbeatPath, m as QoderUpstreamClient, n as GLOBAL_VARIANT, o as clearHostHeartbeat, p as KIND_STATUS, r as QODER_VARIANTS, s as isHeartbeatProcessAlive, t as CHINA_VARIANT, u as readHostHeartbeat, v as normalizeCredits, w as randomSentinel, x as getMachineId, y as validateApiKey, z as patTail } from "./variants-C0iZD8AD.js";
+import { A as QODER_GLOBAL_AUTH_PATH, B as qoderCredentialIdentity, C as probeModel, D as FALLBACK_QODER_MODELS, E as qoderCatalogPath, F as QODER_AUTH_FILENAME, G as qoderPluginDataDir, H as QODER_DATA_DIR_ENV, I as QODER_PAT_ENV_CN, K as qoderStateDir, L as QODER_PAT_ENV_GLOBAL, M as QODER_GLOBAL_STATUS_PATH, N as QODER_PROBE_PATH, O as QoderCatalog, P as QODER_STATUS_PATH, R as QoderCredentialStore, S as PROBE_EFFORT_CANDIDATES, T as QoderCatalogStore, U as QODER_DATA_DIR_NAME, V as qoderOwnAuthPath, W as qoderMachineIdPath, _ as modelInfoOf, a as QODER_HOST_HEARTBEAT_FILENAME, b as createQoderTransport, c as processStartTimeMs, d as writeHostHeartbeat, f as QODER_CONNECT_VERSION, g as kindFromQoderFailure, h as classifyUpstreamError, i as variantFor, j as QODER_GLOBAL_PROBE_PATH, k as QODER_AUTH_PATH, l as qoderHostHeartbeatPath, m as QoderUpstreamClient, n as GLOBAL_VARIANT, o as clearHostHeartbeat, p as KIND_STATUS, r as QODER_VARIANTS, s as isHeartbeatProcessAlive, t as CHINA_VARIANT, u as readHostHeartbeat, v as normalizeCredits, w as randomSentinel, x as getMachineId, y as validateApiKey, z as patTail } from "./variants-BEiQve5M.js";
 import z from "@deepseek-ai/schemastery";
 import { createHash, randomBytes, timingSafeEqual } from "node:crypto";
 import { dirname, join, resolve } from "node:path";
@@ -1056,15 +1056,20 @@ async function qoderWebStatus(deps) {
 		...probed,
 		jobTokenRefreshedAt: refreshedAt
 	};
+	const checkInRecord = deps.checkIn?.();
+	const withCheckIn = checkInRecord === void 0 ? withRefreshNotice : {
+		...withRefreshNotice,
+		checkIn: checkInRecord
+	};
 	try {
 		const credits = await deps.client.fetchCredits();
 		return {
-			...withRefreshNotice,
+			...withCheckIn,
 			credits
 		};
 	} catch (error) {
 		return {
-			...withRefreshNotice,
+			...withCheckIn,
 			creditsError: safeMessage(error)
 		};
 	}
@@ -1169,6 +1174,8 @@ function parseAction(text) {
 	const wrapped = parsed;
 	const action = wrapped["action"];
 	if (action === "clear") return { action: "clear" };
+	if (action === "clear-checkin-logs") return { action: "clear-checkin-logs" };
+	if (action === "checkin") return { action: "checkin" };
 	if (action === "refresh") return { action: "refresh" };
 	if (action === "set-maximum-context-window") return typeof wrapped["enabled"] === "boolean" ? {
 		action: "set-maximum-context-window",
@@ -1212,6 +1219,19 @@ function qoderProbeHandler(deps, key) {
 			return;
 		}
 		try {
+			if (action.action === "clear-checkin-logs") {
+				deps.clearCheckInLogs?.();
+				json(res, 200, { state: "cleared" });
+				return;
+			}
+			if (action.action === "checkin") {
+				if (deps.checkIn === void 0) {
+					json(res, 404, { error: "checkin-not-supported" });
+					return;
+				}
+				json(res, 200, await deps.checkIn());
+				return;
+			}
 			if (action.action === "clear") {
 				deps.clear();
 				json(res, 200, { state: "cleared" });
@@ -1254,6 +1274,162 @@ function registerQoderProbeRoute(ctx, deps, key) {
 	}, "dsh-qoder-connect: probe control route");
 }
 //#endregion
+//#region src/checkin-scheduler.ts
+/** Scheduling and catch-up orchestration for daily 10:00 (UTC+8) check-in. */
+var JsonFileCheckInStore = class {
+	filePath;
+	constructor(filePath) {
+		this.filePath = filePath ?? join(qoderPluginDataDir(), "checkin-status.json");
+	}
+	readAll() {
+		try {
+			if (!existsSync(this.filePath)) return {};
+			const raw = readFileSync(this.filePath, "utf-8");
+			return JSON.parse(raw);
+		} catch {
+			return {};
+		}
+	}
+	read(variantId) {
+		return this.readAll()[variantId];
+	}
+	clearLogs(variantId) {
+		try {
+			const all = this.readAll();
+			if (all[variantId]) {
+				all[variantId] = {
+					...all[variantId],
+					logs: []
+				};
+				mkdirSync(dirname(this.filePath), { recursive: true });
+				writeFileSync(this.filePath, JSON.stringify(all, null, 2), "utf-8");
+			}
+		} catch {}
+	}
+	write(variantId, record) {
+		try {
+			const all = this.readAll();
+			const existingLogs = all[variantId]?.logs ?? [];
+			const newLog = {
+				id: `${record.lastDate}-${record.lastAt}`,
+				date: record.lastDate,
+				timestamp: record.lastAt,
+				status: record.status,
+				...record.amount === void 0 ? {} : { amount: record.amount },
+				...record.message === void 0 ? {} : { message: record.message }
+			};
+			const updatedLogs = [newLog, ...existingLogs.filter((l) => l.id !== newLog.id)].slice(0, 30);
+			all[variantId] = {
+				...record,
+				logs: updatedLogs
+			};
+			mkdirSync(dirname(this.filePath), { recursive: true });
+			writeFileSync(this.filePath, JSON.stringify(all, null, 2), "utf-8");
+		} catch {}
+	}
+};
+/**
+* Returns the current date in YYYY-MM-DD standardized on UTC+8 (Beijing Time).
+*/
+function getUtc8DateString(nowMs = Date.now()) {
+	const d = new Date(nowMs);
+	const utc8 = new Date(d.getTime() + (d.getTimezoneOffset() + 480) * 6e4);
+	return `${utc8.getFullYear()}-${String(utc8.getMonth() + 1).padStart(2, "0")}-${String(utc8.getDate()).padStart(2, "0")}`;
+}
+/**
+* Calculates milliseconds until the next 10:00:05 AM in UTC+8.
+*/
+function msUntilNext10amUtc8(nowMs = Date.now()) {
+	const d = new Date(nowMs);
+	const utc8Time = new Date(d.getTime() + (d.getTimezoneOffset() + 480) * 6e4);
+	const targetUtc8 = new Date(utc8Time.getTime());
+	targetUtc8.setHours(10, 0, 5, 0);
+	let diff = targetUtc8.getTime() - utc8Time.getTime();
+	if (diff <= 0) {
+		targetUtc8.setDate(targetUtc8.getDate() + 1);
+		diff = targetUtc8.getTime() - utc8Time.getTime();
+	}
+	return diff;
+}
+/**
+* Checks whether catch-up is needed today:
+* Current time is past today's 10:00:00 AM (UTC+8) and today has not yet settled a check-in.
+*/
+function shouldCatchUp(today, lastDate, nowMs = Date.now()) {
+	if (lastDate === today) return false;
+	const d = new Date(nowMs);
+	return new Date(d.getTime() + (d.getTimezoneOffset() + 480) * 6e4).getHours() >= 10;
+}
+var CheckInScheduler = class {
+	targets;
+	isEnabled;
+	store;
+	onResult;
+	now;
+	timer;
+	disposed = false;
+	constructor(options) {
+		this.targets = options.targets;
+		this.isEnabled = options.isEnabled;
+		this.store = options.store ?? new JsonFileCheckInStore();
+		this.onResult = options.onResult;
+		this.now = options.now ?? Date.now;
+	}
+	start() {
+		if (this.disposed) return;
+		this.sweepAll(true);
+		this.armNextTimer();
+	}
+	dispose() {
+		this.disposed = true;
+		if (this.timer) {
+			clearTimeout(this.timer);
+			this.timer = void 0;
+		}
+	}
+	armNextTimer() {
+		if (this.disposed) return;
+		const delay = msUntilNext10amUtc8(this.now());
+		this.timer = setTimeout(() => {
+			this.sweepAll(false);
+			this.armNextTimer();
+		}, delay);
+		this.timer.unref?.();
+	}
+	async sweepAll(isCatchUp) {
+		if (this.disposed) return;
+		const nowMs = this.now();
+		const today = getUtc8DateString(nowMs);
+		for (const target of this.targets) {
+			if (!this.isEnabled(target.variantId)) continue;
+			const record = this.store.read(target.variantId);
+			if (isCatchUp && !shouldCatchUp(today, record?.lastDate, nowMs)) continue;
+			if (!isCatchUp && record?.lastDate === today && (record.status === "claimed" || record.status === "already-claimed")) continue;
+			let pat;
+			try {
+				pat = await target.getPat();
+			} catch {
+				continue;
+			}
+			if (!pat) continue;
+			try {
+				const result = await target.service.checkIn(pat);
+				if (result.status !== "error") {
+					this.store.write(target.variantId, {
+						lastDate: result.date,
+						lastAt: result.timestamp,
+						status: result.status,
+						amount: result.amount,
+						message: result.message
+					});
+					if (result.status === "claimed") target.onClaimed?.();
+				}
+				this.onResult?.(result);
+			} catch {}
+		}
+	}
+};
+//#endregion
 //#region src/job-token-hint.ts
 /** The name recorded on the hint row (and therefore its displayed title). */
 const JOB_TOKEN_HINT_NAME = "qoder";
@@ -1290,16 +1466,42 @@ function emitJobTokenHint(at, text) {
 	const agent = lastRunningAgent;
 	if (agent?.session === void 0) return;
 	lastRefreshAt.set(String(agent.id), at);
+	appendHintRow(agent, "success", text);
+}
+/**
+* Print the failed-heal row into the conversation whose request triggered it.
+*
+* The mirror of {@link emitJobTokenHint}: the self-heal ran, exchanged a fresh
+* job token, retried, and the upstream still refused. Without this row that
+* outcome was silent — the user saw only the resulting authorization failure
+* and could not tell that recovery had already been attempted and lost.
+*
+* Rate limited by the transport (once per unresolved outage), not here: this
+* module stays a pure printer so the dedupe policy lives with the state that
+* knows when an outage ends.
+*
+* @param at - When the failed heal was recorded.
+* @param text - The row's summary line.
+*/
+function emitJobTokenRefreshFailedHint(at, text) {
+	const agent = lastRunningAgent;
+	if (agent?.session === void 0) return;
+	appendHintRow(agent, "error", text);
+}
+/** Append one log-only command pair carrying `text` as its outcome. */
+function appendHintRow(agent, kind, text) {
+	const session = agent.session;
+	if (session === void 0) return;
 	const commandId = `cmd-qoder-hint-${Date.now().toString(36)}-${++hintSeq}`;
 	try {
-		agent.session.append("command/run", {
+		session.append("command/run", {
 			commandId,
 			name: JOB_TOKEN_HINT_NAME,
 			source: { kind: "user" }
 		});
-		agent.session.append("command/done", {
+		session.append("command/done", {
 			commandId,
-			kind: "success",
+			kind,
 			text
 		});
 	} catch {}
@@ -1397,6 +1599,8 @@ const MAXIMUM_CONTEXT_WINDOW_CN_FIELD = z.boolean().default(true).description("U
 const MODEL_CONTEXT_WINDOWS_FIELD = z.dict(z.number().min(1), z.string()).description("Per-model context-window overrides for Qoder Global (model id → tokens)");
 /** Sidebar quota toggle (one per variant; both live on the shared quota card). */
 const QUOTA_TOGGLE_FIELD = z.boolean().default(false).description("Show this variant’s remaining-credit card in the sidebar footer (off by default)");
+/** Automatic check-in toggle. */
+const AUTO_CHECK_IN_FIELD = z.boolean().default(false).description("每天 10:00 (UTC+8) 自动签到领取算力额度（默认关闭）");
 /**
 * Quota poll interval: default 5 minutes, floor 1 minute. The status route
 * performs a live upstream billing call per request with no cache, so an
@@ -1415,6 +1619,8 @@ const Config = z.object({
 	modelContextWindowsCN: MODEL_CONTEXT_WINDOWS_FIELD,
 	sidebarQuotaCN: QUOTA_TOGGLE_FIELD,
 	sidebarQuotaGlobal: QUOTA_TOGGLE_FIELD,
+	autoCheckInCN: AUTO_CHECK_IN_FIELD,
+	autoCheckInGlobal: AUTO_CHECK_IN_FIELD,
 	quotaPollMs: QUOTA_POLL_FIELD
 });
 /**
@@ -1446,6 +1652,8 @@ const GLOBAL_SECTION = z.object({
 const QUOTA_SECTION = z.object({
 	sidebarQuotaCN: QUOTA_TOGGLE_FIELD,
 	sidebarQuotaGlobal: QUOTA_TOGGLE_FIELD,
+	autoCheckInCN: AUTO_CHECK_IN_FIELD,
+	autoCheckInGlobal: AUTO_CHECK_IN_FIELD,
 	quotaPollMs: QUOTA_POLL_FIELD
 });
 /** Stable identity key used by credentials, probe records, and catalog entries. */
@@ -1475,6 +1683,17 @@ function fallbackFor(_variant) {
 function jobTokenHintText(at) {
 	return `jobToken 已自动刷新（${new Date(at).toLocaleTimeString("zh-CN", { hour12: false })}）— 旧令牌被上游拒绝，已自动重换并恢复`;
 }
+/**
+* The failed-heal row's summary line.
+*
+* Deliberately does not name a cause: the upstream rejection that survived a
+* fresh token is not necessarily an authorization problem at all, and the real
+* reason travels in the failure message itself (see the SSE envelope body).
+* Claiming "quota" or "revoked" here would be a guess presented as a finding.
+*/
+function jobTokenRefreshFailedHintText(at) {
+	return `jobToken 已重换但仍被上游拒绝（${new Date(at).toLocaleTimeString("zh-CN", { hour12: false })}）— 自愈未恢复，请查看上方错误详情`;
+}
 /** Build one variant's stores, transport, and probe state. */
 function createVariantRuntime(ctx, config, variant, current, identityOf) {
 	const store = new QoderCredentialStore({
@@ -1490,6 +1709,10 @@ function createVariantRuntime(ctx, config, variant, current, identityOf) {
 			jobTokenRefreshedAt = info.at;
 			ctx.logger.warn(`dsh-qoder-connect: ${variant.displayName} job token was auto-refreshed after an upstream rejection`);
 			emitJobTokenHint(info.at, jobTokenHintText(info.at));
+		},
+		onJobTokenRefreshFailed: (info) => {
+			ctx.logger.warn(`dsh-qoder-connect: ${variant.displayName} job token refresh did not recover the chat (upstream status ${info.status ?? "unknown"})`);
+			emitJobTokenRefreshFailedHint(info.at, jobTokenRefreshFailedHintText(info.at));
 		}
 	});
 	const client = new QoderUpstreamClient({
@@ -1699,6 +1922,24 @@ function apply(ctx, config) {
 	*/
 	const lastIdentities = /* @__PURE__ */ new Map();
 	const runtimes = QODER_VARIANTS.map((variant) => createVariantRuntime(ctx, config, variant, () => current(), (id) => lastIdentities.get(id)));
+	const checkInStore = new JsonFileCheckInStore();
+	const checkInScheduler = new CheckInScheduler({
+		targets: runtimes.map((runtime) => ({
+			variantId: runtime.variant.id,
+			service: runtime.transport.checkInService,
+			getPat: async () => runtime.store.patPromise(),
+			onClaimed: () => {
+				runtime.client.fetchCredits().catch(() => void 0);
+			}
+		})),
+		isEnabled: (variantId) => {
+			const cfg = current();
+			if (variantId === CHINA_VARIANT.id) return cfg.autoCheckInCN === true;
+			return cfg.autoCheckInGlobal === true;
+		},
+		store: checkInStore
+	});
+	checkInScheduler.start();
 	installJobTokenHint(ctx);
 	const probeKey = createProbeKey();
 	/**
@@ -1782,7 +2023,8 @@ function apply(ctx, config) {
 				probeKey,
 				authKey,
 				...runtime.variant.id === CHINA_VARIANT.id ? { useMaximumContextWindow: () => current().useMaximumContextWindowCN === true } : { useMaximumContextWindow: () => current().useMaximumContextWindow === true },
-				jobTokenRefreshedAt: runtime.jobTokenRefreshedAt
+				jobTokenRefreshedAt: runtime.jobTokenRefreshedAt,
+				checkIn: () => checkInStore.read(runtime.variant.id)
 			});
 			registerQoderAuthRoute(webCtx, {
 				path: runtime.variant.authPath,
@@ -1845,6 +2087,36 @@ function apply(ctx, config) {
 						reason: runtime.catalogError
 					};
 				},
+				clearCheckInLogs: () => {
+					checkInStore.clearLogs(runtime.variant.id);
+				},
+				checkIn: async () => {
+					let pat;
+					try {
+						pat = await runtime.store.patPromise();
+					} catch {
+						return {
+							state: "failed",
+							reason: "No PAT available"
+						};
+					}
+					const result = await runtime.transport.checkInService.checkIn(pat);
+					if (result.status !== "error") {
+						checkInStore.write(runtime.variant.id, {
+							lastDate: result.date,
+							lastAt: result.timestamp,
+							status: result.status,
+							...result.amount === void 0 ? {} : { amount: result.amount },
+							...result.message === void 0 ? {} : { message: result.message }
+						});
+						if (result.status === "claimed") runtime.client.fetchCredits().catch(() => void 0);
+					}
+					return {
+						state: result.status,
+						...result.amount === void 0 ? {} : { amount: result.amount },
+						...result.message === void 0 ? {} : { reason: result.message }
+					};
+				},
 				...runtime.variant.id === CHINA_VARIANT.id ? { setMaximumContextWindow: async (enabled) => {
 					if (setMaximumContextWindowCN === void 0) return {
 						state: "failed",
@@ -1925,6 +2197,7 @@ function apply(ctx, config) {
 	});
 	ctx.effect(() => () => {
 		stopped = true;
+		checkInScheduler.dispose();
 		for (const timer of timers) clearInterval(timer);
 		timers.length = 0;
 		detach(ctx, clearHostHeartbeat(), "host heartbeat cleanup");