aisubs 0.1.0 → 0.3.0

package/dist/auth.d.ts

@@ -66,6 +66,8 @@ export declare class SubscriptionAuth {
     details(provider: ProviderId, account?: string, signal?: AbortSignal): Promise<SubscriptionAccountDetails>;
     fetch(provider: ProviderId, input: string | URL | Request, init?: RequestInit, account?: string): Promise<Response>;
     proxy(provider: ProviderId, account: string, path: string, init?: RequestInit): Promise<Response>;
+    /** Build an authorized direct-provider request for transports such as WebSocket. */
+    authorizeProxyRequest(provider: ProviderId, account: string, path: string, init?: RequestInit): Promise<Request>;
     getUsage(provider: ProviderId, account?: string, callerSignal?: AbortSignal): Promise<ProviderUsage | null>;
     getModels(provider: ProviderId, account?: string, callerSignal?: AbortSignal): Promise<ProviderModels | null>;
     account(provider: ProviderId, account: string): SubscriptionAccount;

package/dist/auth.js

@@ -19,6 +19,11 @@ function normalizeAccountKey(value) {
     }
     return account;
 }
+function createRequest(input, init) {
+    return new Request(input, init?.body instanceof ReadableStream
+        ? { ...init, duplex: "half" }
+        : init);
+}
 function credentialKey(provider, accountKey) {
     if (accountKey === DEFAULT_ACCOUNT)
         return provider;
@@ -375,7 +380,7 @@ export class SubscriptionAuth {
     }
     async fetch(provider, input, init, account = DEFAULT_ACCOUNT) {
         const adapter = this.adapter(provider);
-        const original = new Request(input, init);
+        const original = createRequest(input, init);
         const accountKey = normalizeAccountKey(account);
         const send = async (credential) => {
             const authorized = await adapter.authorize(original.clone(), credential);
@@ -400,7 +405,7 @@ export class SubscriptionAuth {
         const accountKey = normalizeAccountKey(account);
         const credential = await this.credential(provider, accountKey);
         if (adapter.proxy) {
-            const local = new Request(`http://aisubs.local/${path.replace(/^\//, "")}`, init);
+            const local = createRequest(`http://aisubs.local/${path.replace(/^\//, "")}`, init);
             let response = await adapter.proxy(local.clone(), credential);
             if (response.status === 401) {
                 await response.body?.cancel();
@@ -416,6 +421,22 @@ export class SubscriptionAuth {
         const base = baseUrl.endsWith("/") ? baseUrl : `${baseUrl}/`;
         return this.fetch(provider, new URL(path, base), init, accountKey);
     }
+    /** Build an authorized direct-provider request for transports such as WebSocket. */
+    async authorizeProxyRequest(provider, account, path, init) {
+        const adapter = this.adapter(provider);
+        const accountKey = normalizeAccountKey(account);
+        if (adapter.proxy) {
+            throw new Error(`${provider} does not expose a direct transport endpoint`);
+        }
+        const credential = await this.credential(provider, accountKey);
+        const baseUrl = typeof adapter.proxyBaseUrl === "function"
+            ? adapter.proxyBaseUrl(credential)
+            : adapter.proxyBaseUrl;
+        if (!baseUrl)
+            throw new Error(`${provider} does not expose a direct API endpoint`);
+        const base = baseUrl.endsWith("/") ? baseUrl : `${baseUrl}/`;
+        return adapter.authorize(createRequest(new URL(path, base), init), credential);
+    }
     async getUsage(provider, account = DEFAULT_ACCOUNT, callerSignal) {
         const adapter = this.adapter(provider);
         if (!adapter.getUsage)

package/dist/cli.js

@@ -8,7 +8,7 @@ import { claudeProvider } from "./providers/claude.js";
 import { copilotProvider } from "./providers/copilot.js";
 import { grokProvider } from "./providers/grok.js";
 import { openCodeGoProvider, openCodeZenProvider } from "./providers/opencode.js";
-import { defaultAiSubsDataDir, FileCredentialStore } from "./store.js";
+import { defaultAiSubsDataDir, FileApiKeyStore, FileCredentialStore } from "./store.js";
 const DEFAULT_DASHBOARD_PORT = 4319;
 function usage() {
     console.log(`AI Subs
@@ -62,6 +62,7 @@ async function main() {
     if (!Number.isInteger(port) || port < 0 || port > 65535)
         throw new Error("Invalid port");
     const store = new FileCredentialStore(join(dataDirectory, "credentials.json"));
+    const apiKeys = new FileApiKeyStore(join(dataDirectory, "api-key"));
     const auth = createSubscriptionAuth({
         store,
         providers: [
@@ -73,14 +74,16 @@ async function main() {
             openCodeZenProvider(),
         ],
     });
-    const dashboard = await createSubscriptionAuthDashboardServer({ auth, port });
-    console.log(`AI Subs is running at ${dashboard.url}`);
-    console.log(`Credentials: ${store.file}`);
-    console.log(`Control API key: ${dashboard.apiKey}`);
-    console.log(`Dashboard link: ${dashboard.bootstrapUrl}`);
+    const dashboard = await createSubscriptionAuthDashboardServer({
+        auth,
+        apiKey: await apiKeys.readOrCreate(),
+        regenerateApiKey: () => apiKeys.regenerate(),
+        port,
+    });
+    console.log(`AI Subs is running at \u001b]8;;${dashboard.url}\u0007${dashboard.url}\u001b]8;;\u0007`);
     console.log("Press Ctrl+C to stop.");
     if (shouldOpen)
-        openBrowser(dashboard.bootstrapUrl);
+        openBrowser(dashboard.url);
     let closing = false;
     const close = async () => {
         if (closing)

package/dist/compatibility.d.ts

@@ -0,0 +1,14 @@
+import type { SubscriptionAuth } from "./auth.js";
+import type { ProviderId } from "./types.js";
+export type WireProtocol = "responses" | "chat/completions" | "messages" | "google";
+export declare class CompatibilityError extends Error {
+    readonly code: string;
+    readonly status: number;
+    constructor(message: string, code?: string, status?: number);
+}
+export declare function requestProtocol(path: string): {
+    protocol: WireProtocol;
+    model?: string;
+    stream?: boolean;
+} | null;
+export declare function proxyCompatible(auth: SubscriptionAuth, provider: ProviderId, account: string, path: string, body: Buffer, headers: Headers, signal?: AbortSignal): Promise<Response | null>;