@leo000001/opencode-quota-sidebar 3.0.10 → 4.0.0

package/dist/providers/third_party/xyai.js

@@ -1,348 +0,0 @@
-import { debugError, isRecord, swallow } from "../../helpers.js";
-import { asNumber, configuredProviderEnabled, fetchWithTimeout, sanitizeBaseURL, toIso, } from "../common.js";
-const XYAI_BASE_URL = "https://new.xychatai.com";
-function resolveSiteOrigin(value) {
-    const normalized = sanitizeBaseURL(value);
-    if (!normalized)
-        return XYAI_BASE_URL;
-    try {
-        return new URL(normalized).origin;
-    }
-    catch {
-        return XYAI_BASE_URL;
-    }
-}
-function isXyaiBaseURL(value) {
-    const normalized = sanitizeBaseURL(value);
-    if (!normalized)
-        return false;
-    try {
-        const parsed = new URL(normalized);
-        return parsed.protocol === "https:" && parsed.host === "new.xychatai.com";
-    }
-    catch {
-        return false;
-    }
-}
-function providerConfigFor(config, providerIDs) {
-    for (const providerID of providerIDs) {
-        if (!providerID)
-            continue;
-        const value = config.quota.providers?.[providerID];
-        if (value && typeof value === "object" && !Array.isArray(value)) {
-            return value;
-        }
-    }
-    return undefined;
-}
-function xyaiConfigCandidates(runtimeProviderID, providerID) {
-    const candidates = [];
-    const push = (value) => {
-        if (!value)
-            return;
-        if (!candidates.includes(value))
-            candidates.push(value);
-    };
-    if (runtimeProviderID === "xyai" || runtimeProviderID === "xyai-vibe") {
-        push("xyai");
-    }
-    push(runtimeProviderID);
-    push(providerID);
-    push("xyai");
-    push("xyai-vibe");
-    return candidates;
-}
-function resolveSessionCookie(auth, providerConfig) {
-    if (typeof providerConfig?.sessionCookie === "string" &&
-        providerConfig.sessionCookie) {
-        return providerConfig.sessionCookie;
-    }
-    if (!auth)
-        return undefined;
-    if (auth.type === "wellknown") {
-        if (typeof auth.token === "string" && auth.token)
-            return auth.token;
-    }
-    return undefined;
-}
-function resolveServiceType(providerConfig) {
-    return providerConfig?.serviceType === "claudecode" ? "claudecode" : "codex";
-}
-function resolveLogin(providerConfig) {
-    const login = providerConfig?.login;
-    if (!login || typeof login !== "object" || Array.isArray(login)) {
-        return undefined;
-    }
-    const username = typeof login.username === "string" ? login.username.trim() : "";
-    const password = typeof login.password === "string" ? login.password : "";
-    if (!username || !password)
-        return undefined;
-    return { username, password };
-}
-function headerCookies(response) {
-    const headers = response.headers;
-    if (typeof headers.getSetCookie === "function") {
-        return headers.getSetCookie().filter(Boolean);
-    }
-    const joined = response.headers.get("set-cookie");
-    return joined ? [joined] : [];
-}
-function extractShareSession(response) {
-    for (const value of headerCookies(response)) {
-        const match = value.match(/share-session=("?)([^;"]+)\1/);
-        if (match?.[2])
-            return match[2];
-    }
-    return undefined;
-}
-async function loginAndPersistSession(siteOrigin, login, providerID, updateAuth, timeoutMs) {
-    const response = await fetchWithTimeout(`${siteOrigin}/frontend-api/login`, {
-        method: "POST",
-        headers: {
-            Accept: "application/json",
-            "Content-Type": "application/json",
-            "User-Agent": "opencode-quota-sidebar",
-        },
-        body: JSON.stringify({
-            userToken: login.username,
-            password: login.password,
-            token: "",
-        }),
-    }, timeoutMs).catch(swallow("fetchXyaiQuota:login"));
-    if (!response) {
-        return { error: "login request failed" };
-    }
-    if (!response.ok) {
-        return { error: `login http ${response.status}` };
-    }
-    const payload = await response
-        .json()
-        .catch(swallow("fetchXyaiQuota:loginJson"));
-    if (!isRecord(payload)) {
-        return { error: "invalid login response" };
-    }
-    if (payload.code !== 1) {
-        const msg = typeof payload.msg === "string" && payload.msg
-            ? payload.msg
-            : "login failed";
-        return { error: msg };
-    }
-    const session = extractShareSession(response);
-    if (!session) {
-        return { error: "missing share-session cookie" };
-    }
-    if (updateAuth) {
-        try {
-            await updateAuth(providerID, { type: "wellknown", token: session });
-        }
-        catch (error) {
-            debugError("updateAuth:xyai", error);
-            return {
-                session,
-                warning: "session refreshed but failed to persist; using in-memory session",
-            };
-        }
-    }
-    return { session };
-}
-async function fetchQuotaPayload(siteOrigin, session, timeoutMs) {
-    const response = await fetchWithTimeout(`${siteOrigin}/frontend-api/vibe-code/quota`, {
-        headers: {
-            Accept: "application/json",
-            Cookie: `share-session=${session}`,
-            "User-Agent": "opencode-quota-sidebar",
-        },
-    }, timeoutMs).catch(swallow("fetchXyaiQuota:quota"));
-    if (!response)
-        return { error: "network request failed" };
-    if (!response.ok)
-        return { error: `http ${response.status}` };
-    const payload = await response
-        .json()
-        .catch(swallow("fetchXyaiQuota:quotaJson"));
-    if (!isRecord(payload))
-        return { error: "invalid response" };
-    return { payload };
-}
-function isAuthFailure(payload) {
-    return payload.code === -1 || payload.msg === "认证失败，请重新登录";
-}
-function formatAmount(value) {
-    if (!Number.isFinite(value))
-        return "0";
-    if (Math.abs(value) >= 10) {
-        const one = value.toFixed(1);
-        return one.endsWith(".0") ? one.slice(0, -2) : one;
-    }
-    return value.toFixed(2);
-}
-function pickServicePayload(payload, preferred) {
-    const source = isRecord(payload.data) ? payload.data : payload;
-    const ordered = preferred === "claudecode"
-        ? ["claudecode", "codex"]
-        : ["codex", "claudecode"];
-    for (const key of ordered) {
-        const value = source[key];
-        if (isRecord(value))
-            return { serviceType: key, value };
-    }
-    return undefined;
-}
-function parseQuotaSnapshot(args) {
-    const base = {
-        providerID: args.providerID,
-        adapterID: "xyai",
-        label: "XYAI",
-        shortLabel: "XYAI",
-        sortOrder: 7,
-    };
-    const subscriptions = isRecord(args.payload.subscriptions)
-        ? args.payload.subscriptions
-        : undefined;
-    const usage = isRecord(args.payload.currentUsage)
-        ? args.payload.currentUsage
-        : undefined;
-    const amountLimit = asNumber(subscriptions?.amountLimit);
-    const remainingAmount = asNumber(subscriptions?.remainingAmount);
-    const periodResetTime = toIso(subscriptions?.periodResetTime);
-    const expireTime = toIso(subscriptions?.expireTime);
-    if (amountLimit === undefined || remainingAmount === undefined) {
-        return {
-            ...base,
-            status: "error",
-            checkedAt: args.checkedAt,
-            note: "missing quota fields",
-        };
-    }
-    const remainingPercent = amountLimit > 0
-        ? Math.max(0, Math.min(100, (remainingAmount / amountLimit) * 100))
-        : undefined;
-    const windows = [
-        {
-            label: `Daily $${formatAmount(remainingAmount)}/$${formatAmount(amountLimit)}`,
-            showPercent: false,
-            remainingPercent,
-            resetAt: periodResetTime,
-            resetLabel: "Rst",
-        },
-    ];
-    const noteParts = [
-        expireTime ? `exp ${expireTime.slice(5, 10)}` : undefined,
-        args.serviceType === "claudecode" ? "service=claudecode" : undefined,
-        args.warning,
-    ].filter((value) => Boolean(value));
-    return {
-        ...base,
-        status: "ok",
-        checkedAt: args.checkedAt,
-        remainingPercent,
-        resetAt: periodResetTime,
-        expiresAt: expireTime,
-        note: noteParts.join(" | ") || undefined,
-        windows,
-    };
-}
-async function fetchXyaiQuota(ctx) {
-    const checkedAt = Date.now();
-    const runtimeProviderID = typeof ctx.sourceProviderID === "string" && ctx.sourceProviderID
-        ? ctx.sourceProviderID
-        : ctx.providerID;
-    const providerConfig = providerConfigFor(ctx.config, xyaiConfigCandidates(runtimeProviderID, ctx.providerID));
-    const siteOrigin = resolveSiteOrigin(providerConfig?.baseURL ?? ctx.providerOptions?.baseURL);
-    const serviceType = resolveServiceType(providerConfig);
-    const base = {
-        providerID: runtimeProviderID,
-        adapterID: "xyai",
-        label: "XYAI",
-        shortLabel: "XYAI",
-        sortOrder: 7,
-    };
-    let session = resolveSessionCookie(ctx.auth, providerConfig);
-    const login = resolveLogin(providerConfig);
-    let warning;
-    if (!session && login) {
-        const loginResult = await loginAndPersistSession(siteOrigin, login, ctx.providerID, ctx.updateAuth, ctx.config.quota.requestTimeoutMs);
-        if ("error" in loginResult) {
-            return {
-                ...base,
-                status: "unavailable",
-                checkedAt,
-                note: loginResult.error,
-            };
-        }
-        session = loginResult.session;
-        warning = loginResult.warning;
-    }
-    if (!session) {
-        return {
-            ...base,
-            status: "unavailable",
-            checkedAt,
-            note: "missing share-session or login credentials",
-        };
-    }
-    let quotaResult = await fetchQuotaPayload(siteOrigin, session, ctx.config.quota.requestTimeoutMs);
-    if (!("error" in quotaResult) &&
-        isAuthFailure(quotaResult.payload) &&
-        login) {
-        const loginResult = await loginAndPersistSession(siteOrigin, login, ctx.providerID, ctx.updateAuth, ctx.config.quota.requestTimeoutMs);
-        if (!("error" in loginResult)) {
-            session = loginResult.session;
-            warning = loginResult.warning ?? warning;
-            quotaResult = await fetchQuotaPayload(siteOrigin, session, ctx.config.quota.requestTimeoutMs);
-        }
-    }
-    if ("error" in quotaResult) {
-        return {
-            ...base,
-            status: "error",
-            checkedAt,
-            note: quotaResult.error,
-        };
-    }
-    if (isAuthFailure(quotaResult.payload)) {
-        return {
-            ...base,
-            status: "unavailable",
-            checkedAt,
-            note: "auth expired",
-        };
-    }
-    const service = pickServicePayload(quotaResult.payload, serviceType);
-    if (!service) {
-        return {
-            ...base,
-            status: "error",
-            checkedAt,
-            note: "missing service payload",
-        };
-    }
-    return parseQuotaSnapshot({
-        providerID: runtimeProviderID,
-        serviceType: service.serviceType,
-        payload: service.value,
-        checkedAt,
-        warning,
-    });
-}
-export const xyaiAdapter = {
-    id: "xyai",
-    label: "XYAI",
-    shortLabel: "XYAI",
-    sortOrder: 7,
-    normalizeID: (providerID) => providerID === "xyai" || providerID === "xyai-vibe" ? "xyai" : undefined,
-    matchScore: ({ providerID, providerOptions }) => {
-        if (providerID === "xyai")
-            return 100;
-        if (providerID === "xyai-vibe")
-            return 99;
-        return isXyaiBaseURL(providerOptions?.baseURL) ? 95 : 0;
-    },
-    isEnabled: (config) => {
-        const renamed = config.quota.providers?.xyai?.enabled;
-        if (typeof renamed === "boolean")
-            return renamed;
-        return configuredProviderEnabled(config.quota, "xyai-vibe", false);
-    },
-    fetch: fetchXyaiQuota,
-};