ai-warden 0.9.2 → 0.9.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-warden",
-  "version": "0.9.2",
+  "version": "0.9.4",
   "description": "AI security scanner - Detect prompt injection attacks and PII with user settings",
   "main": "src/index.js",
   "bin": {

package/src/cli.js

@@ -7,11 +7,14 @@
 
 const fs = require('fs');
 const path = require('path');
+const { execSync } = require('child_process');
+const readline = require('readline');
 const scanner = require('./scanner');
 const fileClassifier = require('./fileClassifier');
 const IgnoreParser = require('./ignoreParser');
 const InteractivePrompt = require('./interactivePrompt');
 const AIWarden = require('./index');
+const config = require('./config');
 
 const args = process.argv.slice(2);
 const command = args[0];
@@ -30,6 +33,9 @@ Usage: aiwarden <command> [options]
 Commands:
   scan <path>              Scan file or directory (offline mode)
   validate <text>          Validate text via API (requires API key)
+  login                    Sign up or log in and save API key
+  logout                   Remove saved API key
+  whoami                   Show current API key status
   version                  Show version
   help                     Show this help
 
@@ -271,17 +277,23 @@ async function validateCommand(textOrPath, options = {}) {
     }
   }
 
-  // Get API key from --api-key flag or environment variable
-  const apiKey = options.apiKey || process.env.AI_WARDEN_API_KEY;
+  // Get API key from:
+  // 1. --api-key flag
+  // 2. Environment variable
+  // 3. Saved config file
+  const apiKey = options.apiKey || process.env.AI_WARDEN_API_KEY || config.getApiKey();
 
   if (!apiKey) {
     console.error('❌ Error: API key required for validate command');
     console.log('');
     console.log('Options:');
-    console.log('  1. Pass --api-key flag:');
+    console.log('  1. Log in (saves key for future use):');
+    console.log('     aiwarden login');
+    console.log('');
+    console.log('  2. Pass --api-key flag:');
     console.log('     aiwarden validate "text" --api-key sk_live_xxx');
     console.log('');
-    console.log('  2. Set environment variable:');
+    console.log('  3. Set environment variable:');
     console.log('     export AI_WARDEN_API_KEY=sk_live_xxx');
     console.log('     aiwarden validate "text"');
     console.log('');
@@ -383,6 +395,100 @@ function parseArgs() {
   return options;
 }
 
+// Login command
+async function loginCommand() {
+  console.log('🔐 AI-Warden Login\n');
+  
+  console.log('Opening browser to get your API key...');
+  console.log('→ https://prompt-shield.se/signup\n');
+  
+  // Open browser
+  const url = 'https://prompt-shield.se/signup';
+  try {
+    const platform = process.platform;
+    if (platform === 'darwin') {
+      execSync(`open "${url}"`);
+    } else if (platform === 'win32') {
+      // Windows: start "" "URL" (first empty string is window title)
+      execSync(`start "" "${url}"`, { shell: true });
+    } else {
+      execSync(`xdg-open "${url}"`);
+    }
+  } catch (error) {
+    console.log('⚠️  Could not open browser automatically.');
+    console.log(`Please visit: ${url}\n`);
+  }
+  
+  console.log('After signing up:');
+  console.log('1. Copy your API key from the dashboard');
+  console.log('2. Paste it below\n');
+  
+  // Read API key from stdin
+  const rl = readline.createInterface({
+    input: process.stdin,
+    output: process.stdout
+  });
+  
+  rl.question('API Key: ', (apiKey) => {
+    rl.close();
+    
+    apiKey = apiKey.trim();
+    
+    if (!apiKey) {
+      console.error('❌ No API key provided');
+      process.exit(1);
+    }
+    
+    // Validate format (sk_live_... or sk_test_...)
+    if (!apiKey.startsWith('sk_live_') && !apiKey.startsWith('sk_test_')) {
+      console.error('❌ Invalid API key format');
+      console.error('Expected format: sk_live_... or sk_test_...');
+      process.exit(1);
+    }
+    
+    // Save to config
+    config.setApiKey(apiKey);
+    
+    const keyType = apiKey.startsWith('sk_live_') ? 'Live' : 'Test';
+    console.log(`\n✅ ${keyType} API key saved successfully!`);
+    console.log(`Config: ${config.getConfigPath()}`);
+    console.log('\nYou can now use: aiwarden validate "text"');
+  });
+}
+
+// Logout command
+function logoutCommand() {
+  const currentKey = config.getApiKey();
+  
+  if (!currentKey) {
+    console.log('ℹ️  No API key configured');
+    return;
+  }
+  
+  config.removeApiKey();
+  console.log('✅ API key removed');
+  console.log(`Config file: ${config.getConfigPath()}`);
+}
+
+// Whoami command
+function whoamiCommand() {
+  const apiKey = config.getApiKey();
+  
+  if (!apiKey) {
+    console.log('❌ Not logged in');
+    console.log('Run: aiwarden login');
+    process.exit(1);
+  }
+  
+  const keyType = apiKey.startsWith('sk_live_') ? 'Live' : 'Test';
+  const masked = apiKey.substring(0, 12) + '...' + apiKey.substring(apiKey.length - 4);
+  
+  console.log('✅ Logged in');
+  console.log(`Type: ${keyType}`);
+  console.log(`Key:  ${masked}`);
+  console.log(`Config: ${config.getConfigPath()}`);
+}
+
 // Main
 (async () => {
   if (!command || command === 'help' || command === '--help' || command === '-h') {
@@ -403,6 +509,12 @@ function parseArgs() {
     const text = args[1];
     const options = parseArgs();
     await validateCommand(text, options);
+  } else if (command === 'login') {
+    await loginCommand();
+  } else if (command === 'logout') {
+    logoutCommand();
+  } else if (command === 'whoami') {
+    whoamiCommand();
   } else {
     console.error(`❌ Unknown command: ${command}`);
     console.log('Run "aiwarden help" for usage information');

package/src/config.js

@@ -0,0 +1,98 @@
+/**
+ * AI-Warden Config Manager
+ * Handles API key storage in ~/.config/aiwarden/config.json
+ */
+
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+
+const CONFIG_DIR = path.join(os.homedir(), '.config', 'aiwarden');
+const CONFIG_FILE = path.join(CONFIG_DIR, 'config.json');
+
+/**
+ * Ensure config directory exists
+ */
+function ensureConfigDir() {
+  if (!fs.existsSync(CONFIG_DIR)) {
+    fs.mkdirSync(CONFIG_DIR, { recursive: true });
+  }
+}
+
+/**
+ * Load config from file
+ * @returns {object} Config object (empty if file doesn't exist)
+ */
+function loadConfig() {
+  if (!fs.existsSync(CONFIG_FILE)) {
+    return {};
+  }
+  
+  try {
+    const data = fs.readFileSync(CONFIG_FILE, 'utf8');
+    return JSON.parse(data);
+  } catch (error) {
+    console.error('⚠️  Warning: Could not read config file:', error.message);
+    return {};
+  }
+}
+
+/**
+ * Save config to file
+ * @param {object} config - Config object to save
+ */
+function saveConfig(config) {
+  ensureConfigDir();
+  
+  try {
+    fs.writeFileSync(CONFIG_FILE, JSON.stringify(config, null, 2), 'utf8');
+  } catch (error) {
+    console.error('❌ Error: Could not save config:', error.message);
+    process.exit(1);
+  }
+}
+
+/**
+ * Get stored API key
+ * @returns {string|null} API key or null if not set
+ */
+function getApiKey() {
+  const config = loadConfig();
+  return config.apiKey || null;
+}
+
+/**
+ * Set API key
+ * @param {string} apiKey - API key to store
+ */
+function setApiKey(apiKey) {
+  const config = loadConfig();
+  config.apiKey = apiKey;
+  saveConfig(config);
+}
+
+/**
+ * Remove API key
+ */
+function removeApiKey() {
+  const config = loadConfig();
+  delete config.apiKey;
+  saveConfig(config);
+}
+
+/**
+ * Get config file path
+ * @returns {string} Path to config file
+ */
+function getConfigPath() {
+  return CONFIG_FILE;
+}
+
+module.exports = {
+  getApiKey,
+  setApiKey,
+  removeApiKey,
+  getConfigPath,
+  loadConfig,
+  saveConfig
+};