vibehacker 4.1.0

package/src/providers.js

@@ -0,0 +1,387 @@
+'use strict';
+
+// ── Provider Definitions ──────────────────────────────────────────────────────
+
+const VH_GATEWAY_URL = process.env.VH_GATEWAY_URL || 'https://api.vibsecurity.com/v1';
+
+const PROVIDERS = {
+  vibehacker: {
+    type:      'vibehacker',
+    name:      'Vibe Hacker',
+    shortName: 'Vibe Hacker',
+    baseURL:   VH_GATEWAY_URL,
+    freeNote:  '100 req/day free',
+    tier:      'free',
+    detectKey: k => k.startsWith('vh_'),
+    getKey:    () => 'Sign up at https://vibsecurity.com',
+    models: [
+      { id: 'meta-llama/llama-3.3-70b-instruct:free',       name: 'Vibe Model',       contextWindow: 131072, maxTokens: 8192, provider: 'Vibe Hacker', free: true },
+      { id: 'mistralai/mistral-small-3.1-24b-instruct:free',name: 'Vibe Model',       contextWindow: 131072, maxTokens: 8192, provider: 'Vibe Hacker', free: true },
+      { id: 'google/gemma-3-27b-it:free',                   name: 'Vibe Model',       contextWindow: 131072, maxTokens: 8192, provider: 'Vibe Hacker', free: true },
+      { id: 'qwen/qwen3-235b-a22b:free',                    name: 'Vibe Model',       contextWindow: 32768,  maxTokens: 8192, provider: 'Vibe Hacker', free: true },
+      { id: 'qwen/qwen3-coder-480b-a35b-instruct:free',     name: 'Vibe Model',       contextWindow: 32768,  maxTokens: 8192, provider: 'Vibe Hacker', free: true },
+      { id: 'deepseek/deepseek-r1:free',                    name: 'Vibe Model',       contextWindow: 163840, maxTokens: 8192, provider: 'Vibe Hacker', free: true },
+    ],
+  },
+
+  openrouter: {
+    type:      'openrouter',
+    name:      'Vibe Hacker',
+    shortName: 'Vibe Hacker',
+    baseURL:   'https://openrouter.ai/api/v1',
+    freeNote:  '100 req/day free',
+    tier:      'free',
+    detectKey: k => k.startsWith('sk-or-v1-'),
+    getKey:    () => 'vibsecurity.com/login',
+    models: [
+      { id: 'meta-llama/llama-3.3-70b-instruct:free',       name: 'Vibe Model',       contextWindow: 131072, maxTokens: 8192, provider: 'Vibe Hacker', free: true },
+      { id: 'mistralai/mistral-small-3.1-24b-instruct:free',name: 'Vibe Model',       contextWindow: 131072, maxTokens: 8192, provider: 'Vibe Hacker', free: true },
+      { id: 'google/gemma-3-27b-it:free',                   name: 'Vibe Model',       contextWindow: 131072, maxTokens: 8192, provider: 'Vibe Hacker', free: true },
+      { id: 'qwen/qwen3-235b-a22b:free',                    name: 'Vibe Model',       contextWindow: 32768,  maxTokens: 8192, provider: 'Vibe Hacker', free: true },
+      { id: 'qwen/qwen3-coder-480b-a35b-instruct:free',     name: 'Vibe Model',       contextWindow: 32768,  maxTokens: 8192, provider: 'Vibe Hacker', free: true },
+      { id: 'nvidia/llama-3.1-nemotron-ultra-253b-v1:free',  name: 'Vibe Model',       contextWindow: 131072, maxTokens: 8192, provider: 'Vibe Hacker', free: true },
+      { id: 'deepseek/deepseek-r1:free',                    name: 'Vibe Model',       contextWindow: 163840, maxTokens: 8192, provider: 'Vibe Hacker', free: true },
+    ],
+  },
+
+  groq: {
+    type:      'groq',
+    name:      'Groq',
+    shortName: 'Groq',
+    baseURL:   'https://api.groq.com/openai/v1',
+    freeNote:  '14,400 req/day free',
+    tier:      'free',
+    detectKey: k => k.startsWith('gsk_'),
+    getKey:    () => 'console.groq.com → API Keys → Create Key',
+    models: [
+      { id: 'llama-3.1-8b-instant',           name: 'Llama 3.1 8B',       contextWindow: 131072, maxTokens: 8192, provider: 'Groq', free: true },
+      { id: 'llama-3.3-70b-versatile',         name: 'Llama 3.3 70B',      contextWindow: 131072, maxTokens: 8192, provider: 'Groq', free: true },
+      { id: 'llama-4-scout-17b-16e-instruct',  name: 'Llama 4 Scout 17B',  contextWindow: 131072, maxTokens: 8192, provider: 'Groq', free: true },
+      { id: 'moonshotai/kimi-k2-instruct',     name: 'Kimi K2',             contextWindow: 131072, maxTokens: 8192, provider: 'Groq', free: true },
+      { id: 'qwen/qwen3-32b',                  name: 'Qwen3 32B',           contextWindow: 32768,  maxTokens: 8192, provider: 'Groq', free: true },
+    ],
+  },
+
+  cerebras: {
+    type:      'cerebras',
+    name:      'Cerebras',
+    shortName: 'Cbrs',
+    baseURL:   'https://api.cerebras.ai/v1',
+    freeNote:  '1M tokens/day free',
+    tier:      'free',
+    detectKey: k => k.startsWith('csk-'),
+    getKey:    () => 'inference.cerebras.ai → API Keys → Create',
+    models: [
+      { id: 'qwen-3-235b-a22b', name: 'Qwen3 235B',    contextWindow: 32768,  maxTokens: 8192, provider: 'Cerebras', free: true },
+      { id: 'llama3.1-8b',      name: 'Llama 3.1 8B',  contextWindow: 131072, maxTokens: 8192, provider: 'Cerebras', free: true },
+    ],
+  },
+
+  gemini: {
+    type:      'gemini',
+    name:      'Gemini',
+    shortName: 'Gem',
+    baseURL:   'https://generativelanguage.googleapis.com/v1beta/openai/',
+    freeNote:  '1000 req/day free',
+    tier:      'free',
+    detectKey: k => k.startsWith('AIza'),
+    getKey:    () => 'aistudio.google.com → Get API key → Create',
+    models: [
+      { id: 'gemini-2.5-flash-lite', name: 'Gemini 2.5 Flash Lite', contextWindow: 1000000, maxTokens: 8192, provider: 'Google', free: true },
+      { id: 'gemini-2.5-flash',      name: 'Gemini 2.5 Flash',      contextWindow: 1000000, maxTokens: 8192, provider: 'Google', free: true },
+      { id: 'gemini-2.5-pro',        name: 'Gemini 2.5 Pro',        contextWindow: 1000000, maxTokens: 8192, provider: 'Google', free: true },
+    ],
+  },
+
+  mistral: {
+    type:      'mistral',
+    name:      'Mistral',
+    shortName: 'Mist',
+    baseURL:   'https://api.mistral.ai/v1',
+    freeNote:  '~1B tokens/month free',
+    tier:      'free',
+    detectKey: k => k.startsWith('mistral:'),
+    getKey:    () => 'console.mistral.ai → API Keys → then: /addkey mistral:<key>',
+    models: [
+      { id: 'codestral-latest',     name: 'Codestral',     contextWindow: 256000, maxTokens: 8192, provider: 'Mistral', free: true },
+      { id: 'mistral-large-latest', name: 'Mistral Large', contextWindow: 131072, maxTokens: 8192, provider: 'Mistral', free: true },
+    ],
+  },
+
+  anthropic: {
+    type:      'anthropic',
+    name:      'Anthropic',
+    shortName: 'Ant',
+    baseURL:   'https://api.anthropic.com/v1',
+    freeNote:  'Pay-per-use',
+    tier:      'paid',
+    detectKey: k => k.startsWith('sk-ant-'),
+    getKey:    () => 'console.anthropic.com → API Keys → Create Key',
+    models: [
+      { id: 'claude-opus-4-5',   name: 'Claude Opus 4.5',   contextWindow: 200000, maxTokens: 8192, provider: 'Anthropic', free: false },
+      { id: 'claude-sonnet-4-5', name: 'Claude Sonnet 4.5', contextWindow: 200000, maxTokens: 8192, provider: 'Anthropic', free: false },
+      { id: 'claude-haiku-3-5',  name: 'Claude Haiku 3.5',  contextWindow: 200000, maxTokens: 8192, provider: 'Anthropic', free: false },
+    ],
+  },
+
+  openai: {
+    type:      'openai',
+    name:      'OpenAI',
+    shortName: 'OAI',
+    baseURL:   'https://api.openai.com/v1',
+    freeNote:  'Pay-per-use',
+    tier:      'paid',
+    detectKey: k => k.startsWith('sk-') && !k.startsWith('sk-or-v1-') && !k.startsWith('sk-ant-'),
+    getKey:    () => 'platform.openai.com → API Keys → Create Key',
+    models: [
+      { id: 'gpt-4.1',       name: 'GPT-4.1',       contextWindow: 128000, maxTokens: 8192, provider: 'OpenAI', free: false },
+      { id: 'gpt-4o',        name: 'GPT-4o',         contextWindow: 128000, maxTokens: 8192, provider: 'OpenAI', free: false },
+      { id: 'gpt-4o-mini',   name: 'GPT-4o Mini',    contextWindow: 128000, maxTokens: 8192, provider: 'OpenAI', free: false },
+      { id: 'o4-mini',       name: 'o4-mini',         contextWindow: 128000, maxTokens: 8192, provider: 'OpenAI', free: false },
+    ],
+  },
+
+  deepseek: {
+    type:      'deepseek',
+    name:      'DeepSeek',
+    shortName: 'DS',
+    baseURL:   'https://api.deepseek.com/v1',
+    freeNote:  'Pay-per-use (cheap)',
+    tier:      'paid',
+    detectKey: k => k.startsWith('sk-ds-') || k.startsWith('dsk-'),
+    getKey:    () => 'platform.deepseek.com → API Keys → Create',
+    models: [
+      { id: 'deepseek-chat',     name: 'DeepSeek V3', contextWindow: 131072, maxTokens: 8192, provider: 'DeepSeek', free: false },
+      { id: 'deepseek-reasoner', name: 'DeepSeek R1', contextWindow: 131072, maxTokens: 8192, provider: 'DeepSeek', free: false },
+    ],
+  },
+
+  xai: {
+    type:      'xai',
+    name:      'xAI (Grok)',
+    shortName: 'xAI',
+    baseURL:   'https://api.x.ai/v1',
+    freeNote:  '$25/mo free credits',
+    tier:      'free',
+    detectKey: k => k.startsWith('xai-'),
+    getKey:    () => 'console.x.ai → API Keys → Create Key',
+    models: [
+      { id: 'grok-3',      name: 'Grok 3',      contextWindow: 131072, maxTokens: 8192, provider: 'xAI', free: true },
+      { id: 'grok-3-mini', name: 'Grok 3 Mini', contextWindow: 131072, maxTokens: 8192, provider: 'xAI', free: true },
+    ],
+  },
+
+  together: {
+    type:      'together',
+    name:      'Together',
+    shortName: 'Togth',
+    baseURL:   'https://api.together.xyz/v1',
+    freeNote:  '$1 free credit',
+    tier:      'free',
+    detectKey: k => k.startsWith('tog-') || k.startsWith('together-'),
+    getKey:    () => 'api.together.xyz → Settings → API Keys',
+    models: [
+      { id: 'meta-llama/Llama-3.3-70B-Instruct-Turbo', name: 'Llama 3.3 70B', contextWindow: 131072, maxTokens: 8192, provider: 'Together', free: true },
+      { id: 'Qwen/Qwen2.5-72B-Instruct-Turbo',         name: 'Qwen 2.5 72B',  contextWindow: 131072, maxTokens: 8192, provider: 'Together', free: true },
+      { id: 'deepseek-ai/DeepSeek-R1',                  name: 'DeepSeek R1',    contextWindow: 131072, maxTokens: 8192, provider: 'Together', free: true },
+    ],
+  },
+};
+
+// ── ProviderManager ───────────────────────────────────────────────────────────
+
+class ProviderManager {
+  constructor(modelManager) {
+    this._mm      = modelManager;
+    this._list    = [];
+    this._idx     = 0;
+    this._modelIdx = {};
+    this._limited  = new Set();
+    this._load();
+  }
+
+  _load() {
+    const config = require('./config');
+    const saved  = config.providers || [];
+
+    // Detect primary type from stored key
+    const primaryType = config.apiKey && config.apiKey.startsWith('vh_')
+      ? 'vibehacker'
+      : 'openrouter';
+
+    this._list = [{ type: primaryType, apiKey: config.apiKey }];
+    for (const p of saved) {
+      if (p.type && p.apiKey && p.type !== primaryType) {
+        this._list.push({ type: p.type, apiKey: p.apiKey });
+      }
+    }
+    this._applyCurrent();
+  }
+
+  _save() {
+    const toSave = this._list
+      .filter(p => p.type !== 'vibehacker' && p.type !== 'openrouter')
+      .map(p => ({ type: p.type, apiKey: p.apiKey }));
+    const fs   = require('fs');
+    const path = require('path');
+    const os   = require('os');
+    const dir  = path.join(os.homedir(), '.vibehacker');
+    const file = path.join(dir, 'config.json');
+    try {
+      if (!fs.existsSync(dir)) fs.mkdirSync(dir, { recursive: true });
+      let ex = {};
+      try { ex = JSON.parse(fs.readFileSync(file, 'utf8')); } catch (_) {}
+      fs.writeFileSync(file, JSON.stringify({ ...ex, providers: toSave }, null, 2));
+    } catch (_) {}
+  }
+
+  _applyCurrent() {
+    const config = require('./config');
+    const p   = this._list[this._idx] || this._list[0];
+    if (!p) return;
+    const def = PROVIDERS[p.type];
+    if (!def) return;
+    config.apiKey  = p.apiKey;
+    config.baseURL = def.baseURL;
+  }
+
+  current() {
+    const p   = this._list[this._idx] || this._list[0];
+    const def = PROVIDERS[p.type] || PROVIDERS.openrouter;
+    return { ...def, apiKey: p.apiKey };
+  }
+
+  currentModel() {
+    const p = this.current();
+    if (p.type === 'openrouter') return this._mm.current();
+    const models = p.models || [];
+    const idx    = Math.min(this._modelIdx[p.type] || 0, models.length - 1);
+    return models[Math.max(0, idx)];
+  }
+
+  currentModels() {
+    const p = this.current();
+    if (p.type === 'openrouter') return this._mm.list();
+    return p.models || [];
+  }
+
+  list() {
+    return this._list.map((p, i) => ({
+      ...(PROVIDERS[p.type] || {}),
+      apiKey:  p.apiKey,
+      active:  i === this._idx,
+      limited: this._limited.has(p.type),
+      index:   i,
+    }));
+  }
+
+  currentIdx()  { return this._idx; }
+  allLimited()  { return this._list.every(p => this._limited.has(p.type)); }
+  isCurrentFree() {
+    const m = this.currentModel();
+    if (m && typeof m.free === 'boolean') return m.free;
+    return (PROVIDERS[this.current().type] || {}).tier === 'free';
+  }
+
+  selectModelById(id) {
+    const p = this.current();
+    if (p.type === 'openrouter') { this._mm.selectById(id); return; }
+    const idx = (p.models || []).findIndex(m => m.id === id);
+    if (idx >= 0) this._modelIdx[p.type] = idx;
+  }
+
+  selectByType(type) {
+    const idx = this._list.findIndex(p => p.type === type);
+    if (idx >= 0) { this._idx = idx; this._applyCurrent(); }
+  }
+
+  // Silent model rotation on 429 — pick next model in current provider
+  rotateModel() {
+    const p = this.current();
+    const models = p.type === 'openrouter' ? this._mm.list() : (p.models || []);
+    const curIdx = this._modelIdx[p.type] || 0;
+    const nextIdx = (curIdx + 1) % models.length;
+    this._modelIdx[p.type] = nextIdx;
+    if (p.type === 'openrouter') {
+      this._mm.selectById(models[nextIdx].id);
+    }
+    return models[nextIdx];
+  }
+
+  add(apiKey) {
+    let strippedKey = apiKey;
+    const colonIdx = apiKey.indexOf(':');
+    if (colonIdx > 0 && colonIdx < 12) {
+      const prefix = apiKey.substring(0, colonIdx).toLowerCase();
+      if (PROVIDERS[prefix]) {
+        strippedKey = apiKey.substring(colonIdx + 1).trim();
+        apiKey      = prefix + ':' + strippedKey;
+      }
+    }
+    const type = this._detectType(apiKey);
+    const actualKey = type === 'mistral' ? strippedKey : apiKey;
+
+    if (type === 'vibehacker' || type === 'openrouter') {
+      const config = require('./config');
+      config.setApiKey(actualKey);
+      this._list = this._list.filter(p => p.type !== 'vibehacker' && p.type !== 'openrouter');
+      this._list.unshift({ type, apiKey: actualKey });
+      this._idx = 0;
+      this._applyCurrent();
+    } else {
+      const existing = this._list.findIndex(p => p.type === type);
+      if (existing >= 0) this._list[existing].apiKey = strippedKey || actualKey;
+      else               this._list.push({ type, apiKey: strippedKey || actualKey });
+    }
+
+    this._limited.delete(type);
+    this._save();
+    return type;
+  }
+
+  remove(type) {
+    if (type === 'openrouter') return;
+    this._list = this._list.filter(p => p.type !== type);
+    if (this._idx >= this._list.length) this._idx = 0;
+    this._limited.delete(type);
+    this._save();
+    this._applyCurrent();
+  }
+
+  markRateLimited() {
+    const cur = this.current();
+    this._limited.add(cur.type);
+    const startIdx = this._idx;
+    for (let i = 1; i < this._list.length; i++) {
+      const nextIdx  = (startIdx + i) % this._list.length;
+      const nextType = this._list[nextIdx].type;
+      if (!this._limited.has(nextType)) {
+        this._idx = nextIdx;
+        this._applyCurrent();
+        return true;
+      }
+    }
+    return false;
+  }
+
+  clearLimits() {
+    this._limited.clear();
+    this._applyCurrent();
+  }
+
+  _detectType(key) {
+    for (const [type, def] of Object.entries(PROVIDERS)) {
+      if (def.detectKey && def.detectKey(key)) return type;
+    }
+    return 'openrouter';
+  }
+
+  static detectType(key) {
+    for (const [type, def] of Object.entries(PROVIDERS)) {
+      if (def.detectKey && def.detectKey(key)) return type;
+    }
+    return 'openrouter';
+  }
+}
+
+module.exports = { ProviderManager, PROVIDERS };

package/src/setup.js

@@ -0,0 +1,95 @@
+'use strict';
+
+const blessed = require('blessed');
+
+function ew(s) {
+  return String(s).replace(/\{/g, '\\{').replace(/\}/g, '\\}');
+}
+
+// ── First-run setup — simple API key paste ─────────────────────────────────
+
+function showSetup(screen) {
+  return new Promise((resolve) => {
+    const W = Math.min(screen.width - 4, 60);
+    const H = Math.min(screen.height - 2, 22);
+
+    const overlay = blessed.box({
+      parent: screen,
+      top: 'center', left: 'center',
+      width: W, height: H,
+      tags: true,
+      border: { type: 'line' },
+      style: { fg: '#00ff88', bg: '#000a00', border: { fg: '#00aa44' } },
+      padding: { left: 2, right: 2 },
+    });
+
+    overlay.setContent([
+      '',
+      '{center}{green-fg}{bold}◈ Vibe Hacker — Get Started{/bold}{/green-fg}{/center}',
+      '',
+      '{center}{#888888-fg}Enter your API key to begin.{/#888888-fg}{/center}',
+      '{center}{#888888-fg}Get a free key at vibsecurity.com{/#888888-fg}{/center}',
+      '',
+      '{#003300-fg}  ════════════════════════════════════════════{/#003300-fg}',
+      '',
+      '  {#555555-fg}Supported keys:{/#555555-fg}',
+      '  {#44ff88-fg}Vibe Hacker{/#44ff88-fg}  {#666666-fg}vh_… / sk-or-v1-…{/#666666-fg}',
+      '  {#44ff88-fg}Groq{/#44ff88-fg}         {#666666-fg}gsk_…{/#666666-fg}',
+      '  {#44ff88-fg}Gemini{/#44ff88-fg}       {#666666-fg}AIza…{/#666666-fg}',
+      '  {#44ff88-fg}Cerebras{/#44ff88-fg}     {#666666-fg}csk-…{/#666666-fg}',
+      '  {#44ff88-fg}OpenAI{/#44ff88-fg}       {#666666-fg}sk-…{/#666666-fg}',
+      '  {#44ff88-fg}Anthropic{/#44ff88-fg}    {#666666-fg}sk-ant-…{/#666666-fg}',
+      '',
+    ].join('\n'));
+
+    const input = blessed.textbox({
+      parent: overlay,
+      bottom: 1, left: 2, right: 2, height: 3,
+      border: { type: 'line' },
+      label: ' {#00ff88-fg}Paste API Key{/#00ff88-fg} ',
+      style: { fg: '#00ff88', bg: '#001100', border: { fg: '#00aa44' } },
+      inputOnFocus: false,
+      mouse: true,
+    });
+
+    const statusLine = blessed.text({
+      parent: overlay,
+      bottom: 0, left: 2, right: 2, height: 1,
+      tags: true,
+      content: '',
+      style: { fg: '#00aa44', bg: 'black' },
+    });
+
+    overlay.focus();
+    screen.render();
+
+    setTimeout(() => {
+      input.focus();
+      input.readInput();
+      screen.render();
+    }, 50);
+
+    input.on('submit', (value) => {
+      const key = (value || '').trim();
+      if (!key || key.length < 10) {
+        statusLine.setContent('  {red-fg}Key too short — paste a valid API key{/red-fg}');
+        input.clearValue();
+        input.focus();
+        input.readInput();
+        screen.render();
+        return;
+      }
+      try { input.destroy(); overlay.destroy(); } catch (_) {}
+      screen.render();
+      resolve(key);
+    });
+
+    input.on('cancel', () => {
+      try { input.destroy(); overlay.destroy(); } catch (_) {}
+      screen.render();
+      resolve(null);
+    });
+  });
+}
+
+module.exports = { showSetup };