dsh-ssh-tui 0.3.6 → 0.3.8

package/lib/supergrok-token.js

@@ -0,0 +1,128 @@
+/**
+ * Read and refresh SuperGrok / xAI OAuth tokens from disk.
+ *
+ * dsh-ssh-tui does not depend on dsh-llm-xai-oauth. Keep this file's wire
+ * format identical to grok-bridge / that plugin: ~/.grok-bridge/auth.json
+ * with access_token, refresh_token, expires_at.
+ */
+import { mkdir, readFile, writeFile } from 'node:fs/promises';
+import { homedir } from 'node:os';
+import { dirname, join } from 'node:path';
+export const SUPERGROK_AUTH_PATHS = [
+    join(homedir(), '.grok-bridge', 'auth.json'),
+    join(homedir(), '.grok', 'auth.json'),
+];
+export const SUPERGROK_REFRESH_SKEW_MS = 5 * 60 * 1000;
+const DEFAULT_CLIENT_ID = 'b1a00492-073a-47ea-816f-4c329264a828';
+const DEFAULT_TOKEN_URL = 'https://auth.x.ai/oauth2/token';
+const DEFAULT_LIFETIME_SECONDS = 3600;
+function stringField(record, key) {
+    const value = record[key];
+    return typeof value === 'string' && value.trim() !== '' ? value.trim() : undefined;
+}
+function expiresAtMs(record) {
+    const raw = record.expires_at ?? record.expiresAt;
+    if (typeof raw === 'string')
+        return Date.parse(raw) || null;
+    if (typeof raw === 'number' && Number.isFinite(raw))
+        return raw < 1e12 ? raw * 1000 : raw;
+    return null;
+}
+export function parseSuperGrokAuthFile(raw, path) {
+    if (raw === null || typeof raw !== 'object' || Array.isArray(raw))
+        return undefined;
+    const record = raw;
+    const accessToken = stringField(record, 'access_token') ?? stringField(record, 'accessToken');
+    if (accessToken === undefined)
+        return undefined;
+    return {
+        accessToken,
+        refreshToken: stringField(record, 'refresh_token') ?? stringField(record, 'refreshToken') ?? null,
+        expiresAt: expiresAtMs(record),
+        clientId: stringField(record, 'oidc_client_id') ?? stringField(record, 'clientId') ?? null,
+        path,
+    };
+}
+export function superGrokTokenNeedsRefresh(token, now = Date.now(), skewMs = SUPERGROK_REFRESH_SKEW_MS) {
+    if (token.refreshToken === null)
+        return false;
+    if (token.expiresAt === null)
+        return false;
+    return token.expiresAt - now < skewMs;
+}
+async function readTokenFile(path) {
+    try {
+        return parseSuperGrokAuthFile(JSON.parse(await readFile(path, 'utf8')), path);
+    }
+    catch {
+        return undefined;
+    }
+}
+export async function loadSuperGrokToken(paths = SUPERGROK_AUTH_PATHS) {
+    for (const path of paths) {
+        const token = await readTokenFile(path);
+        if (token !== undefined)
+            return token;
+    }
+    return undefined;
+}
+async function postRefresh(refreshToken, clientId) {
+    const response = await fetch(DEFAULT_TOKEN_URL, {
+        method: 'POST',
+        headers: {
+            Accept: 'application/json',
+            'Content-Type': 'application/x-www-form-urlencoded',
+        },
+        body: new URLSearchParams({
+            grant_type: 'refresh_token',
+            refresh_token: refreshToken,
+            client_id: clientId,
+        }),
+        signal: AbortSignal.timeout(15_000),
+    });
+    const body = await response.json();
+    const access = stringField(body, 'access_token');
+    if (!response.ok || access === undefined) {
+        const error = stringField(body, 'error');
+        const description = stringField(body, 'error_description');
+        throw new Error(`SuperGrok token 刷新失败（HTTP ${response.status}）${error || description ? `：${[error, description].filter(Boolean).join(': ')}` : ''}`);
+    }
+    return {
+        access_token: access,
+        ...stringField(body, 'refresh_token') === undefined ? {} : { refresh_token: stringField(body, 'refresh_token') },
+        ...typeof body.expires_in === 'number' ? { expires_in: body.expires_in } : {},
+    };
+}
+export async function persistSuperGrokToken(path, tokenResponse, previousRefreshToken, now = Date.now()) {
+    const record = {
+        access_token: tokenResponse.access_token,
+        refresh_token: tokenResponse.refresh_token ?? previousRefreshToken,
+        expires_at: now + (tokenResponse.expires_in ?? DEFAULT_LIFETIME_SECONDS) * 1000,
+        saved_at: new Date(now).toISOString(),
+    };
+    await mkdir(dirname(path), { recursive: true, mode: 0o700 });
+    await writeFile(path, `${JSON.stringify(record, null, 2)}\n`, { mode: 0o600 });
+    const stored = parseSuperGrokAuthFile(record, path);
+    if (stored === undefined)
+        throw new Error(`无法写入 SuperGrok token（${path}）`);
+    return stored;
+}
+export async function resolveFreshSuperGrokToken(options = {}) {
+    const current = await loadSuperGrokToken(options.paths);
+    if (current === undefined)
+        return undefined;
+    const now = options.now ?? Date.now();
+    if (options.force !== true && !superGrokTokenNeedsRefresh(current, now)) {
+        return current.accessToken;
+    }
+    if (current.refreshToken === null) {
+        if (options.force === true || superGrokTokenNeedsRefresh({ ...current, refreshToken: 'x' }, now)) {
+            throw new Error(`SuperGrok token 没有 refresh_token，请运行 dsh-llm-xai-oauth login --force`);
+        }
+        return current.accessToken;
+    }
+    const next = await postRefresh(current.refreshToken, current.clientId ?? DEFAULT_CLIENT_ID);
+    const saved = await persistSuperGrokToken(current.path, next, current.refreshToken, now);
+    return saved.accessToken;
+}
+//# sourceMappingURL=supergrok-token.js.map

package/lib/supergrok-token.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"supergrok-token.js","sourceRoot":"","sources":["../src/supergrok-token.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAA;AAC7D,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAA;AACjC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAEzC,MAAM,CAAC,MAAM,oBAAoB,GAAG;IAClC,IAAI,CAAC,OAAO,EAAE,EAAE,cAAc,EAAE,WAAW,CAAC;IAC5C,IAAI,CAAC,OAAO,EAAE,EAAE,OAAO,EAAE,WAAW,CAAC;CAC7B,CAAA;AAEV,MAAM,CAAC,MAAM,yBAAyB,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,CAAA;AACtD,MAAM,iBAAiB,GAAG,sCAAsC,CAAA;AAChE,MAAM,iBAAiB,GAAG,gCAAgC,CAAA;AAC1D,MAAM,wBAAwB,GAAG,IAAI,CAAA;AAUrC,SAAS,WAAW,CAAC,MAA+B,EAAE,GAAW;IAC/D,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,CAAA;IACzB,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS,CAAA;AACpF,CAAC;AAED,SAAS,WAAW,CAAC,MAA+B;IAClD,MAAM,GAAG,GAAG,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,SAAS,CAAA;IACjD,IAAI,OAAO,GAAG,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,IAAI,CAAA;IAC3D,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC;QAAE,OAAO,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,CAAA;IACzF,OAAO,IAAI,CAAA;AACb,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,GAAY,EAAE,IAAY;IAC/D,IAAI,GAAG,KAAK,IAAI,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC;QAAE,OAAO,SAAS,CAAA;IACnF,MAAM,MAAM,GAAG,GAA8B,CAAA;IAC7C,MAAM,WAAW,GAAG,WAAW,CAAC,MAAM,EAAE,cAAc,CAAC,IAAI,WAAW,CAAC,MAAM,EAAE,aAAa,CAAC,CAAA;IAC7F,IAAI,WAAW,KAAK,SAAS;QAAE,OAAO,SAAS,CAAA;IAC/C,OAAO;QACL,WAAW;QACX,YAAY,EAAE,WAAW,CAAC,MAAM,EAAE,eAAe,CAAC,IAAI,WAAW,CAAC,MAAM,EAAE,cAAc,CAAC,IAAI,IAAI;QACjG,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC;QAC9B,QAAQ,EAAE,WAAW,CAAC,MAAM,EAAE,gBAAgB,CAAC,IAAI,WAAW,CAAC,MAAM,EAAE,UAAU,CAAC,IAAI,IAAI;QAC1F,IAAI;KACL,CAAA;AACH,CAAC;AAED,MAAM,UAAU,0BAA0B,CACxC,KAA2B,EAC3B,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,EAChB,MAAM,GAAG,yBAAyB;IAElC,IAAI,KAAK,CAAC,YAAY,KAAK,IAAI;QAAE,OAAO,KAAK,CAAA;IAC7C,IAAI,KAAK,CAAC,SAAS,KAAK,IAAI;QAAE,OAAO,KAAK,CAAA;IAC1C,OAAO,KAAK,CAAC,SAAS,GAAG,GAAG,GAAG,MAAM,CAAA;AACvC,CAAC;AAED,KAAK,UAAU,aAAa,CAAC,IAAY;IACvC,IAAI,CAAC;QACH,OAAO,sBAAsB,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAY,EAAE,IAAI,CAAC,CAAA;IAC1F,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAA;IAClB,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,kBAAkB,CAAC,QAA2B,oBAAoB;IACtF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,KAAK,GAAG,MAAM,aAAa,CAAC,IAAI,CAAC,CAAA;QACvC,IAAI,KAAK,KAAK,SAAS;YAAE,OAAO,KAAK,CAAA;IACvC,CAAC;IACD,OAAO,SAAS,CAAA;AAClB,CAAC;AAED,KAAK,UAAU,WAAW,CAAC,YAAoB,EAAE,QAAgB;IAK/D,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,iBAAiB,EAAE;QAC9C,MAAM,EAAE,MAAM;QACd,OAAO,EAAE;YACP,MAAM,EAAE,kBAAkB;YAC1B,cAAc,EAAE,mCAAmC;SACpD;QACD,IAAI,EAAE,IAAI,eAAe,CAAC;YACxB,UAAU,EAAE,eAAe;YAC3B,aAAa,EAAE,YAAY;YAC3B,SAAS,EAAE,QAAQ;SACpB,CAAC;QACF,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC;KACpC,CAAC,CAAA;IACF,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAA6B,CAAA;IAC7D,MAAM,MAAM,GAAG,WAAW,CAAC,IAAI,EAAE,cAAc,CAAC,CAAA;IAChD,IAAI,CAAC,QAAQ,CAAC,EAAE,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,WAAW,CAAC,IAAI,EAAE,OAAO,CAAC,CAAA;QACxC,MAAM,WAAW,GAAG,WAAW,CAAC,IAAI,EAAE,mBAAmB,CAAC,CAAA;QAC1D,MAAM,IAAI,KAAK,CACb,6BAA6B,QAAQ,CAAC,MAAM,IAAI,KAAK,IAAI,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CACpI,CAAA;IACH,CAAC;IACD,OAAO;QACL,YAAY,EAAE,MAAM;QACpB,GAAG,WAAW,CAAC,IAAI,EAAE,eAAe,CAAC,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,WAAW,CAAC,IAAI,EAAE,eAAe,CAAC,EAAE;QAChH,GAAG,OAAO,IAAI,CAAC,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE;KAC9E,CAAA;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,IAAY,EACZ,aAAoF,EACpF,oBAA4B,EAC5B,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE;IAEhB,MAAM,MAAM,GAAG;QACb,YAAY,EAAE,aAAa,CAAC,YAAY;QACxC,aAAa,EAAE,aAAa,CAAC,aAAa,IAAI,oBAAoB;QAClE,UAAU,EAAE,GAAG,GAAG,CAAC,aAAa,CAAC,UAAU,IAAI,wBAAwB,CAAC,GAAG,IAAI;QAC/E,QAAQ,EAAE,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE;KACtC,CAAA;IACD,MAAM,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAA;IAC5D,MAAM,SAAS,CAAC,IAAI,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAA;IAC9E,MAAM,MAAM,GAAG,sBAAsB,CAAC,MAAM,EAAE,IAAI,CAAC,CAAA;IACnD,IAAI,MAAM,KAAK,SAAS;QAAE,MAAM,IAAI,KAAK,CAAC,wBAAwB,IAAI,GAAG,CAAC,CAAA;IAC1E,OAAO,MAAM,CAAA;AACf,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAAC,UAI7C,EAAE;IACJ,MAAM,OAAO,GAAG,MAAM,kBAAkB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;IACvD,IAAI,OAAO,KAAK,SAAS;QAAE,OAAO,SAAS,CAAA;IAC3C,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,EAAE,CAAA;IACrC,IAAI,OAAO,CAAC,KAAK,KAAK,IAAI,IAAI,CAAC,0BAA0B,CAAC,OAAO,EAAE,GAAG,CAAC,EAAE,CAAC;QACxE,OAAO,OAAO,CAAC,WAAW,CAAA;IAC5B,CAAC;IACD,IAAI,OAAO,CAAC,YAAY,KAAK,IAAI,EAAE,CAAC;QAClC,IAAI,OAAO,CAAC,KAAK,KAAK,IAAI,IAAI,0BAA0B,CAAC,EAAE,GAAG,OAAO,EAAE,YAAY,EAAE,GAAG,EAAE,EAAE,GAAG,CAAC,EAAE,CAAC;YACjG,MAAM,IAAI,KAAK,CAAC,sEAAsE,CAAC,CAAA;QACzF,CAAC;QACD,OAAO,OAAO,CAAC,WAAW,CAAA;IAC5B,CAAC;IACD,MAAM,IAAI,GAAG,MAAM,WAAW,CAAC,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC,QAAQ,IAAI,iBAAiB,CAAC,CAAA;IAC3F,MAAM,KAAK,GAAG,MAAM,qBAAqB,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,YAAY,EAAE,GAAG,CAAC,CAAA;IACxF,OAAO,KAAK,CAAC,WAAW,CAAA;AAC1B,CAAC"}