dsh-plugin-subscription-oauth 0.15.0 → 0.16.1

package/lib/protocol.d.ts

@@ -12,7 +12,7 @@
 /** Prefix of every RPC method this plugin serves. */
 export declare const RPC_PREFIX = "subscription-oauth.";
 /** Endpoints the Settings page may call. */
-export declare const RPC_ENDPOINTS: readonly ["status", "login", "cancel", "logout", "setDefault", "refresh", "discover", "setModelVisible", "setModelOrder", "addModel", "removeModel", "usage", "reload", "auto-state", "setAutoOn", "checkin", "deepseek-balance", "ccswitch-balance", "model-variants"];
+export declare const RPC_ENDPOINTS: readonly ["status", "login", "cancel", "logout", "setDefault", "refresh", "discover", "setModelVisible", "setModelOrder", "addModel", "removeModel", "usage", "reload", "auto-state", "setAutoOn", "checkin", "tasks", "deepseek-balance", "ccswitch-balance", "model-variants"];
 /** One endpoint name. */
 export type RpcEndpoint = typeof RPC_ENDPOINTS[number];
 /** Client → host envelope. */
@@ -334,6 +334,10 @@ export interface RpcMap {
         payload: CheckinRequest;
         value: CheckinResult;
     };
+    tasks: {
+        payload: TasksRequest;
+        value: TasksResult;
+    };
     'deepseek-balance': {
         payload: DeepSeekBalanceRequest;
         value: DeepSeekBalanceResult;
@@ -420,6 +424,29 @@ export interface CheckinResult {
     streakDays: number;
     already: boolean;
 }
+/** Payload of `tasks`: which account to read and auto-claim. */
+export interface TasksRequest {
+    provider: string;
+    account: string;
+    /** Claim every claimable reward; false only lists. */
+    claim?: boolean;
+}
+/** One task's state for the card. */
+export interface WorkBuddyTaskState {
+    taskCode: string;
+    title: string;
+    credit: number;
+    target: number;
+    current: number;
+    state: 'pending' | 'claimable' | 'claimed' | 'locked';
+}
+/** Result of `tasks`: the task list and what auto-claiming earned. */
+export interface TasksResult {
+    tasks: WorkBuddyTaskState[];
+    claimed: number;
+    claimable: number;
+    creditEarned: number;
+}
 /** Payload of `deepseek-balance`. */
 export interface DeepSeekBalanceRequest {
     /** Client metadata the host's account service expects. */

package/lib/protocol.js

@@ -29,6 +29,7 @@ export const RPC_ENDPOINTS = [
     'auto-state',
     'setAutoOn',
     'checkin',
+    'tasks',
     'deepseek-balance',
     'ccswitch-balance',
     'model-variants',

package/lib/providers/workbuddy-provider.js

@@ -10,7 +10,7 @@ import { TokenManager } from '../auth/tokens.js';
 import { LOGIN_CANCELLED_MESSAGE } from '../auth/device-flow.js';
 import path from 'node:path';
 import { readSpendLedger } from './workbuddy-credits.js';
-import { finishWorkBuddySignIn, parseWorkBuddySession, readWorkBuddyUsage, refreshWorkBuddyToken, startWorkBuddySignIn, workbuddyRefreshable, workbuddyStale, claimWorkBuddyCheckin, WORKBUDDY_EARLY_MS, } from '../auth/workbuddy-flow.js';
+import { finishWorkBuddySignIn, parseWorkBuddySession, readWorkBuddyUsage, refreshWorkBuddyToken, startWorkBuddySignIn, workbuddyRefreshable, workbuddyStale, claimWorkBuddyCheckin, listWorkBuddyTasks, acceptWorkBuddyTasks, claimWorkBuddyTask, WORKBUDDY_EARLY_MS, } from '../auth/workbuddy-flow.js';
 import { WorkBuddyAdapter, WORKBUDDY_CODE_URL, WORKBUDDY_CATALOG, WORKBUDDY_DEFAULT_MODEL, WORKBUDDY_PROVIDER_ID, WORKBUDDY_PROVIDER_NAME, } from './workbuddy.js';
 import { ProviderController } from '../auth/provider-controller.js';
 import { proxiedFetch } from '../http.js';
@@ -223,5 +223,50 @@ export function createWorkBuddyProvider(options) {
                 throw new ProviderError(`WorkBuddy check-in failed: ${message}`, REAUTH_REQUIRED);
             }
         },
+        tasks: async (account, claim) => {
+            const held = await store.get(account);
+            if (held === undefined)
+                throw new ProviderError('WorkBuddy account not signed in', REAUTH_REQUIRED);
+            const token = tokenOf(held);
+            // Enrol in every open task first: accepting is the sign-up half, and replaying it is
+            // a safe no-op, so progress a chat or an activity report lights is never missed.
+            const tasks = await listWorkBuddyTasks(token, fetchFn);
+            const open = tasks.filter((task) => !task.claimed && !task.locked && task.acceptStatus !== 'claimed').map((task) => task.taskCode);
+            try {
+                await acceptWorkBuddyTasks(token, open, fetchFn);
+            }
+            catch {
+                // Accepting is best-effort; a failure here does not block reading or claiming.
+            }
+            const fresh = await listWorkBuddyTasks(token, fetchFn);
+            let creditEarned = 0;
+            let claimedCount = 0;
+            if (claim) {
+                for (const task of fresh.filter((entry) => entry.claimable && !entry.locked)) {
+                    try {
+                        const reward = await claimWorkBuddyTask(token, task.taskCode, fetchFn);
+                        creditEarned += reward.credit;
+                        claimedCount += 1;
+                    }
+                    catch {
+                        // One task's refusal does not stop the others.
+                    }
+                }
+            }
+            const stateOf = (task) => task.claimed ? 'claimed' : task.locked ? 'locked' : task.claimable ? 'claimable' : 'pending';
+            return {
+                tasks: fresh.map((task) => ({
+                    taskCode: task.taskCode,
+                    title: task.title,
+                    credit: task.credit,
+                    target: task.target,
+                    current: task.current,
+                    state: stateOf(task),
+                })),
+                claimed: claimedCount,
+                claimable: fresh.filter((task) => task.claimable && !task.locked).length,
+                creditEarned,
+            };
+        },
     };
 }

package/lib/providers/workbuddy.js

@@ -12,7 +12,7 @@ import { LlmAdapter, errorChain, ReasoningEffortId, ToolCallId, } from '@deepsee
 import { idleWatchdog, mapFetchFailure } from './common.js';
 import { proxiedFetch } from '../http.js';
 import { findEntry } from './catalog.js';
-import { WORKBUDDY_ENDPOINT, WORKBUDDY_UA_VERSION, workbuddyHeaders } from '../auth/workbuddy-flow.js';
+import { WORKBUDDY_ENDPOINT, workbuddyHeaders } from '../auth/workbuddy-flow.js';
 import { recordSpend } from './workbuddy-credits.js';
 /** The `workbuddy` provider route key (China build). */
 export const WORKBUDDY_PROVIDER_ID = 'workbuddy';
@@ -80,12 +80,6 @@ function unflagMessages(messages) {
     }
     return messages;
 }
-/** 32 hex digits, for the chat ids WorkBuddy's own client makes. */
-function randomHex() {
-    const bytes = new Uint8Array(16);
-    crypto.getRandomValues(bytes);
-    return [...bytes].map((byte) => byte.toString(16).padStart(2, '0')).join('');
-}
 /** A tool's name may hold letters, digits, `_`, `-`; another character is refused. */
 const NAME_OK = /^[A-Za-z0-9_-]+$/;
 /** A name with each character WorkBuddy refuses as `_`, kept apart from `taken`. */
@@ -285,18 +279,7 @@ export class WorkBuddyAdapter extends LlmAdapter {
             const body = this.requestBody(options);
             const response = await this.fetchFn(WORKBUDDY_API, {
                 method: 'POST',
-                headers: {
-                    'Content-Type': 'application/json',
-                    'Accept': 'text/event-stream',
-                    ...workbuddyHeaders(token),
-                    // Name the chat the way WorkBuddy's own client does: the conversation, the
-                    // turn, and the message. Without these the gateway counts the request under no
-                    // client and cuts the stream, which is why the reply stopped after a few seconds.
-                    'X-Conversation-Message-ID': randomHex(),
-                    'X-Request-ID': randomHex(),
-                    'X-Conversation-Request-ID': randomHex(),
-                    'X-Conversation-ID': randomHex(),
-                },
+                headers: workbuddyHeaders(token),
                 body: JSON.stringify(body),
                 signal: watchdog.signal,
             });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dsh-plugin-subscription-oauth",
-  "version": "0.15.0",
+  "version": "0.16.1",
   "description": "Use subscription accounts as DeepSeek Harness LLM providers — Kimi Code (OAuth device flow), Grok / X Premium (OIDC + PKCE), Google Antigravity (Google OAuth + Cloud Code Assist) and GLM / BigModel (ZCode's browser sign-in) — with per-model visibility, drag-to-reorder Auto fallback and plan quotas from the web Settings page",
   "license": "MIT",
   "author": "EricFL1998",