@ayurak/aribot-cli 1.0.2 → 1.0.3

package/dist/cli.js

@@ -37,7 +37,7 @@ function getHeaders() {
         process.exit(1);
     }
     return {
-        'Authorization': `Bearer ${apiKey}`,
+        'X-API-Key': apiKey,
         'Content-Type': 'application/json'
     };
 }
@@ -64,19 +64,28 @@ program
             message: 'Enter your Aribot API key:',
             mask: '*'
         }]);
-    const spinner = (0, ora_1.default)('Validating API key...').start();
+    const spinner = (0, ora_1.default)('Exchanging API key for token...').start();
     try {
         const fetch = (await import('node-fetch')).default;
-        const response = await fetch(`${API_BASE}/v1/users/me/`, {
-            headers: { 'Authorization': `Bearer ${apiKey}` }
+        // Exchange API key for JWT token
+        const response = await fetch(`${API_BASE}/v1/developer/token/`, {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify({ api_key: apiKey })
         });
         if (response.ok) {
+            const data = await response.json();
+            config.set('accessToken', data.access);
+            config.set('refreshToken', data.refresh);
             config.set('apiKey', apiKey);
+            config.set('userEmail', data.user?.email);
+            config.set('company', data.user?.company);
             spinner.succeed(chalk_1.default.green('Authenticated successfully!'));
-            console.log(chalk_1.default.dim('API key stored securely.'));
+            console.log(chalk_1.default.dim(`Logged in as ${data.user?.email}`));
         }
         else {
-            spinner.fail(chalk_1.default.red('Invalid API key'));
+            const error = await response.json();
+            spinner.fail(chalk_1.default.red(error.message || 'Invalid API key'));
         }
     }
     catch (error) {
@@ -108,8 +117,9 @@ program
             return;
         }
         data.results.forEach((d) => {
-            const status = d.status === 'completed' ? chalk_1.default.green('✓') : chalk_1.default.yellow('⋯');
-            console.log(`${status} ${chalk_1.default.cyan(d.id.slice(0, 8))}  ${d.name}  ${chalk_1.default.dim(d.threats_count + ' threats')}`);
+            const status = d.stage === 'completed' ? chalk_1.default.green('✓') : chalk_1.default.yellow('⋯');
+            const name = d.name || d.filename || 'Unnamed';
+            console.log(`${status} ${chalk_1.default.cyan(d.id.slice(0, 8))}  ${name}  ${chalk_1.default.dim(d.threats_count + ' threats')}`);
         });
         console.log(chalk_1.default.dim(`\nShowing ${data.results.length} of ${data.count} diagrams`));
     }
@@ -193,13 +203,14 @@ program
             medium: chalk_1.default.blue,
             low: chalk_1.default.dim
         };
-        data.results?.forEach((t) => {
+        const threats = data.threats || data.results || [];
+        threats.forEach((t) => {
             const color = severityColors[t.severity] || chalk_1.default.white;
-            console.log(`${color(`[${t.severity.toUpperCase()}]`)} ${t.title}`);
-            console.log(chalk_1.default.dim(`  STRIDE: ${t.stride_category} | ${t.id.slice(0, 8)}`));
+            console.log(`${color(`[${t.severity?.toUpperCase() || 'UNKNOWN'}]`)} ${t.title || t.name}`);
+            console.log(chalk_1.default.dim(`  Category: ${t.category || t.stride_category || 'N/A'} | ID: ${String(t.id).slice(0, 8)}`));
             console.log();
         });
-        console.log(chalk_1.default.dim(`Total: ${data.count} threats`));
+        console.log(chalk_1.default.dim(`Total: ${threats.length} threats`));
     }
     catch (error) {
         spinner.fail('Failed to fetch threats');

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ayurak/aribot-cli",
-  "version": "1.0.2",
+  "version": "1.0.3",
   "description": "AI-powered threat modeling CLI by Ayurak. Automatically analyze diagrams, generate STRIDE threats, and get security recommendations.",
   "main": "dist/index.js",
   "bin": {

package/src/cli.ts

@@ -37,7 +37,7 @@ function getHeaders(): Record<string, string> {
     process.exit(1);
   }
   return {
-    'Authorization': `Bearer ${apiKey}`,
+    'X-API-Key': apiKey,
     'Content-Type': 'application/json'
   };
 }
@@ -69,20 +69,29 @@ program
       mask: '*'
     }]);
 
-    const spinner = ora('Validating API key...').start();
+    const spinner = ora('Exchanging API key for token...').start();
 
     try {
       const fetch = (await import('node-fetch')).default;
-      const response = await fetch(`${API_BASE}/v1/users/me/`, {
-        headers: { 'Authorization': `Bearer ${apiKey}` }
+      // Exchange API key for JWT token
+      const response = await fetch(`${API_BASE}/v1/developer/token/`, {
+        method: 'POST',
+        headers: { 'Content-Type': 'application/json' },
+        body: JSON.stringify({ api_key: apiKey })
       });
 
       if (response.ok) {
+        const data = await response.json() as any;
+        config.set('accessToken', data.access);
+        config.set('refreshToken', data.refresh);
         config.set('apiKey', apiKey);
+        config.set('userEmail', data.user?.email);
+        config.set('company', data.user?.company);
         spinner.succeed(chalk.green('Authenticated successfully!'));
-        console.log(chalk.dim('API key stored securely.'));
+        console.log(chalk.dim(`Logged in as ${data.user?.email}`));
       } else {
-        spinner.fail(chalk.red('Invalid API key'));
+        const error = await response.json() as any;
+        spinner.fail(chalk.red(error.message || 'Invalid API key'));
       }
     } catch (error) {
       spinner.fail(chalk.red('Authentication failed'));
@@ -119,8 +128,9 @@ program
       }
 
       data.results.forEach((d: any) => {
-        const status = d.status === 'completed' ? chalk.green('✓') : chalk.yellow('⋯');
-        console.log(`${status} ${chalk.cyan(d.id.slice(0, 8))}  ${d.name}  ${chalk.dim(d.threats_count + ' threats')}`);
+        const status = d.stage === 'completed' ? chalk.green('✓') : chalk.yellow('⋯');
+        const name = d.name || d.filename || 'Unnamed';
+        console.log(`${status} ${chalk.cyan(d.id.slice(0, 8))}  ${name}  ${chalk.dim(d.threats_count + ' threats')}`);
       });
 
       console.log(chalk.dim(`\nShowing ${data.results.length} of ${data.count} diagrams`));
@@ -220,14 +230,15 @@ program
         low: chalk.dim
       };
 
-      data.results?.forEach((t: any) => {
+      const threats = data.threats || data.results || [];
+      threats.forEach((t: any) => {
         const color = severityColors[t.severity] || chalk.white;
-        console.log(`${color(`[${t.severity.toUpperCase()}]`)} ${t.title}`);
-        console.log(chalk.dim(`  STRIDE: ${t.stride_category} | ${t.id.slice(0, 8)}`));
+        console.log(`${color(`[${t.severity?.toUpperCase() || 'UNKNOWN'}]`)} ${t.title || t.name}`);
+        console.log(chalk.dim(`  Category: ${t.category || t.stride_category || 'N/A'} | ID: ${String(t.id).slice(0, 8)}`));
         console.log();
       });
 
-      console.log(chalk.dim(`Total: ${data.count} threats`));
+      console.log(chalk.dim(`Total: ${threats.length} threats`));
     } catch (error) {
       spinner.fail('Failed to fetch threats');
       console.error(error);