langtrain 0.2.1 → 0.2.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "langtrain",
-  "version": "0.2.1",
+  "version": "0.2.3",
   "description": "Unified JavaScript SDK for Langtrain Ecosystem",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",
@@ -45,7 +45,6 @@
     "form-data": "^4.0.5",
     "gradient-string": "^3.0.0",
     "kleur": "^4.1.5",
-    "langtrain": "^0.1.15",
     "langtune": "^0.1.1",
     "langvision": "^0.1.1"
   }

package/src/cli/auth.ts

@@ -1,6 +1,25 @@
-import { password, isCancel, cancel, intro, green, yellow, red, bgMagenta, black, spinner, gray, cyan, dim, bold } from './ui';
+import { password, isCancel, cancel, intro, green, yellow, red, bgMagenta, black, spinner, gray, cyan, dim, bold, note } from './ui';
 import { getConfig, saveConfig } from './config';
 import { SubscriptionClient, SubscriptionInfo } from '../index';
+import axios from 'axios';
+import { exec } from 'child_process';
+import { promisify } from 'util';
+
+const execAsync = promisify(exec);
+const API_BASE = 'https://api.langtrain.ai/api/v1';
+
+async function openBrowser(url: string) {
+    try {
+        const command = process.platform === 'win32'
+            ? `start ${url}`
+            : process.platform === 'darwin'
+                ? `open "${url}"`
+                : `xdg-open "${url}"`;
+        await execAsync(command);
+    } catch {
+        // Silently fail if browser can't open
+    }
+}
 
 /**
  * Quick check if API key is stored (no network call).
@@ -28,9 +47,86 @@ export async function ensureAuth(): Promise<string> {
 }
 
 /**
- * Interactive login — Claude-style API key entry with immediate verification.
+ * Interactive login — browser-based OAuth flow with API key fallback.
  */
 export async function handleLogin() {
+    const s = spinner();
+    s.start('Connecting to Langtrain...');
+
+    try {
+        // 1. Request device code
+        const { data: codeData } = await axios.post(`${API_BASE}/auth/device/code`);
+        const { device_code, user_code, verification_url, expires_in, interval } = codeData;
+
+        s.stop(green('Connected.'));
+
+        console.log('\n  ' + bgMagenta(black(' AUTHENTICATION ')) + '\n');
+        console.log(`  To log in, please open your browser to:\n  ${cyan(verification_url + '?code=' + user_code)}\n`);
+
+        note(
+            `Confirmation Code:\n${bold(user_code)}`,
+            'Verify in Browser'
+        );
+
+        console.log(gray('  Opening browser automatically...'));
+        await openBrowser(`${verification_url}?code=${user_code}`);
+
+        const pollSpinner = spinner();
+        pollSpinner.start('Waiting for approval in browser...');
+
+        // 2. Poll for token
+        const startTime = Date.now();
+        const timeout = expires_in * 1000;
+
+        while (Date.now() - startTime < timeout) {
+            try {
+                const { data: tokenData } = await axios.post(`${API_BASE}/auth/device/token?device_code=${device_code}`);
+
+                if (tokenData.status === 'approved') {
+                    const apiKey = tokenData.api_key;
+                    pollSpinner.stop(green('Approved!'));
+
+                    const verifySpinner = spinner();
+                    verifySpinner.start('Verifying credentials...');
+
+                    const client = new SubscriptionClient({ apiKey });
+                    const info = await client.getStatus();
+
+                    const planBadge = info.plan === 'pro'
+                        ? bgMagenta(black(' PRO '))
+                        : info.plan === 'enterprise'
+                            ? bgMagenta(black(' ENTERPRISE '))
+                            : ' FREE ';
+
+                    verifySpinner.stop(green(`Authenticated ${planBadge}`));
+
+                    const config = getConfig();
+                    saveConfig({ ...config, apiKey });
+                    console.log(green('  ✔ Credentials saved to ~/.langtrain/config.json\n'));
+                    return;
+                }
+
+                if (tokenData.status === 'expired') {
+                    pollSpinner.stop(red('Device code expired.'));
+                    break;
+                }
+
+                // Wait for the requested interval
+                await new Promise(r => setTimeout(r, interval * 1000));
+            } catch (err) {
+                // Ignore network errors during polling
+                await new Promise(r => setTimeout(r, interval * 1000));
+            }
+        }
+
+        pollSpinner.stop(yellow('Login timed out.'));
+    } catch (err: any) {
+        s.stop(red('Could not reach Langtrain server.'));
+    }
+
+    // 3. Fallback to manual entry if browser flow fails
+    console.log(gray('\n  Browser login failed or timed out. Falling back to manual entry.'));
+
     while (true) {
         console.log(dim('  ─────────────────────────────────────'));
         console.log(gray('  Get your API Key at: ') + cyan('https://app.langtrain.xyz/home/api'));
@@ -49,8 +145,8 @@ export async function handleLogin() {
             process.exit(0);
         }
 
-        const s = spinner();
-        s.start('Verifying API Key...');
+        const verifySpinner = spinner();
+        verifySpinner.start('Verifying API Key...');
 
         try {
             const client = new SubscriptionClient({ apiKey: apiKey as string });
@@ -62,22 +158,14 @@ export async function handleLogin() {
                     ? bgMagenta(black(' ENTERPRISE '))
                     : ' FREE ';
 
-            s.stop(green(`Authenticated ${planBadge}`));
-
-            // Show initial token info if available
-            if (info.usage) {
-                const used = info.usage.tokensUsedThisMonth || 0;
-                const limit = info.usage.tokenLimit || 10000;
-                const pct = Math.round((used / limit) * 100);
-                console.log(dim(`  Tokens: ${used.toLocaleString()} / ${limit.toLocaleString()} (${pct}% used)`));
-            }
+            verifySpinner.stop(green(`Authenticated ${planBadge}`));
 
             const config = getConfig();
             saveConfig({ ...config, apiKey: apiKey as string });
             console.log(green('  ✔ Credentials saved to ~/.langtrain/config.json\n'));
             return;
         } catch (e: any) {
-            s.stop(red('Invalid API Key. Please try again.'));
+            verifySpinner.stop(red('Invalid API Key. Please try again.'));
         }
     }
 }

package/src/cli/index.ts

@@ -298,5 +298,7 @@ export async function main() {
         }
     });
 
-    main().catch(console.error);
+    program.parse(process.argv);
 }
+
+main().catch(console.error);