krasavacode 0.2.2 → 0.3.0

package/README.md

@@ -67,6 +67,28 @@ chmod +x ~/krasavacode
 
 ---
 
+## Хочешь модель помощнее? (опционально, 60 секунд)
+
+По дефолту работает простая модель — её хватит на первые проекты. Если хочешь качество как у настоящего Claude — подключи бесплатный **Google Gemini 2.5 Flash** (1500 запросов в день, без карты).
+
+**Как:**
+
+1. На значке **«ВАЙБКОДИНГ»** на Рабочем столе — пока не дабл-кликай.
+2. Открой Терминал (Mac: Spotlight → «Терминал»; Win: меню «Пуск» → «Командная строка»).
+3. Скопируй и вставь одну строку:
+
+   ```bash
+   npx krasavacode setup-gemini
+   ```
+4. Программа откроет в браузере страницу Google AI Studio. Войди через свой Gmail.
+5. Нажми **«Create API key»** — появится длинная строка-ключ.
+6. Скопируй ключ, вернись в окно Терминала и **вставь** (Mac: ⌘+V, Win: правая кнопка → Paste).
+7. Программа сама проверит ключ и переключит вайбкодинг на Gemini.
+
+После этого обычный значок **«ВАЙБКОДИНГ»** работает уже на Gemini. Качество — ощутимо выше.
+
+---
+
 ## Что-то пошло не так?
 
 | Проблема | Что делать |

package/bin/krasavacode.js

@@ -5,15 +5,21 @@ import { ensurePreset } from '../src/preset.js';
 import { launchClaude } from '../src/launch.js';
 import { runUpgrade } from '../src/upgrade.js';
 import { runDoctor } from '../src/doctor.js';
+import { runSetupGemini } from '../src/setup-gemini.js';
 
 // Hardcoded so it works inside Bun --compile (no FS access to package.json)
-const VERSION = '0.2.2';
+const VERSION = '0.3.0';
 
 const cmd = process.argv[2];
 
 async function main() {
   if (cmd === 'doctor') return runDoctor();
   if (cmd === 'upgrade') return runUpgrade();
+  if (cmd === 'setup-gemini' || cmd === 'gemini') {
+    const result = await runSetupGemini();
+    if (!result?.launchAfter) return;
+    // fall through to normal launch flow below
+  }
   if (cmd === '--version' || cmd === '-v') {
     console.log(`KRASAVACODE v${VERSION}`);
     return;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "krasavacode",
-  "version": "0.2.2",
+  "version": "0.3.0",
   "description": "KRASAVACODE — однокнопочный бесплатный вайбкодинг для учеников. Claude Code на бесплатных провайдерах через локальный gateway. Сам ставит Node при необходимости.",
   "type": "module",
   "bin": {

package/src/hub.js

@@ -1,6 +1,7 @@
 import { spawn } from 'node:child_process';
 import { setTimeout as sleep } from 'node:timers/promises';
 import { CCR_PORT } from './preset.js';
+import { loadGeminiKey } from './setup-gemini.js';
 
 const HOST = '127.0.0.1';
 const PORT = CCR_PORT;
@@ -38,10 +39,16 @@ export async function startHub(paths) {
 
   process.stdout.write(`🚀 Поднимаю локальный gateway на порту ${PORT}… `);
 
+  // Inject GEMINI_API_KEY into ccr env if user has configured Gemini.
+  // ccr's config.json references it as $GEMINI_API_KEY (env-interpolation).
+  const ccrEnv = { ...paths.env };
+  const geminiKey = await loadGeminiKey();
+  if (geminiKey) ccrEnv.GEMINI_API_KEY = geminiKey;
+
   const child = spawn(paths.ccrBin, ['start'], {
     stdio: process.env.KRASAVACODE_DEBUG ? 'inherit' : 'pipe',
     detached: false,
-    env: paths.env,
+    env: ccrEnv,
   });
 
   let stderrTail = '';

package/src/launch.js

@@ -1,8 +1,11 @@
 import { spawn } from 'node:child_process';
+import { isGeminiConfigured } from './setup-gemini.js';
 
 const PLACEHOLDER_TOKEN = 'sk-krasavacode-local';
 
 export async function launchClaude(paths, hub /*, detection */) {
+  const geminiOn = await isGeminiConfigured();
+
   const env = {
     ...process.env,
     ANTHROPIC_BASE_URL: hub.baseUrl,
@@ -21,7 +24,12 @@ export async function launchClaude(paths, hub /*, detection */) {
   console.log('');
   console.log('━'.repeat(58));
   console.log('  KRASAVACODE — вайбкодинг через локальный hub');
-  console.log('  Hub: ' + hub.baseUrl + '  (claude-code-router → Pollinations)');
+  if (geminiOn) {
+    console.log('  Модель: Google Gemini 2.5 Flash  (1500 запросов в день)');
+  } else {
+    console.log('  Модель: gpt-oss-20b через Pollinations  (бесплатно, без логина)');
+    console.log('  💡 Хочешь модель посильнее бесплатно? → krasavacode setup-gemini');
+  }
   console.log('  Пиши задачу обычным языком, ИИ сделает.');
   console.log('━'.repeat(58));
   console.log('');

package/src/preset.js

@@ -1,6 +1,7 @@
 import { mkdir, readFile, writeFile, copyFile, access } from 'node:fs/promises';
 import { homedir } from 'node:os';
-import { join, dirname } from 'node:path';
+import { join } from 'node:path';
+import { isGeminiConfigured } from './setup-gemini.js';
 
 const CCR_DIR = join(homedir(), '.claude-code-router');
 const CCR_CONFIG = join(CCR_DIR, 'config.json');
@@ -8,52 +9,76 @@ const STATE_FILE = join(homedir(), '.krasavacode', 'state.json');
 
 const KRASAVACODE_MARKER = 'krasavacode/managed';
 
-const FREE_CONFIG = {
-  HOST: '127.0.0.1',
-  PORT: 3456,
-  LOG: false,
-  API_TIMEOUT_MS: 600000,
-  Providers: [
-    {
-      name: 'pollinations',
-      api_base_url: 'https://text.pollinations.ai/openai/chat/completions',
-      api_key: 'public',
-      models: ['openai', 'openai-fast', 'gpt-oss-20b'],
-    },
-  ],
-  Router: {
-    default: 'pollinations,openai',
-    background: 'pollinations,openai-fast',
-    think: 'pollinations,openai',
-    longContext: 'pollinations,openai',
-    longContextThreshold: 60000,
-  },
-  // Marker so future runs know this config is ours and we may overwrite it.
-  // If a user has manually edited config (no marker), we leave it alone.
-  _krasavacode: KRASAVACODE_MARKER,
-};
+function pollinationsProvider() {
+  return {
+    name: 'pollinations',
+    api_base_url: 'https://text.pollinations.ai/openai/chat/completions',
+    api_key: 'public',
+    models: ['openai', 'openai-fast', 'gpt-oss-20b'],
+  };
+}
 
-async function readState() {
-  try { return JSON.parse(await readFile(STATE_FILE, 'utf8')); }
-  catch { return {}; }
+function geminiProvider() {
+  return {
+    name: 'gemini',
+    api_base_url: 'https://generativelanguage.googleapis.com/v1beta/models/',
+    api_key: '$GEMINI_API_KEY',
+    models: ['gemini-2.5-flash', 'gemini-2.5-pro', 'gemini-flash-latest'],
+    transformer: { use: ['gemini'] },
+  };
 }
-async function writeState(s) {
-  await writeFile(STATE_FILE, JSON.stringify(s, null, 2));
+
+function buildConfig({ withGemini }) {
+  const Providers = withGemini
+    ? [geminiProvider(), pollinationsProvider()]
+    : [pollinationsProvider()];
+
+  const Router = withGemini
+    ? {
+        default: 'gemini,gemini-2.5-flash',
+        background: 'gemini,gemini-2.5-flash',
+        think: 'gemini,gemini-2.5-pro',
+        longContext: 'gemini,gemini-2.5-pro',
+        longContextThreshold: 60000,
+      }
+    : {
+        default: 'pollinations,openai',
+        background: 'pollinations,openai-fast',
+        think: 'pollinations,openai',
+        longContext: 'pollinations,openai',
+        longContextThreshold: 60000,
+      };
+
+  return {
+    HOST: '127.0.0.1',
+    PORT: 3456,
+    LOG: false,
+    API_TIMEOUT_MS: 600000,
+    Providers,
+    Router,
+    _krasavacode: KRASAVACODE_MARKER,
+  };
 }
 
-async function exists(p) {
-  return access(p).then(() => true).catch(() => false);
+async function readState() {
+  try { return JSON.parse(await readFile(STATE_FILE, 'utf8')); }
+  catch { return {}; }
 }
+async function writeState(s) { await writeFile(STATE_FILE, JSON.stringify(s, null, 2)); }
+async function exists(p) { return access(p).then(() => true).catch(() => false); }
 
 /**
- * Writes a default claude-code-router config pointing at Pollinations
- * (free, no-API-key). Backs up any existing user config before overwriting.
+ * Generates ~/.claude-code-router/config.json:
+ *   - If user has run setup-gemini → Gemini first, Pollinations as fallback Provider
+ *   - Else → Pollinations only
  *
- * Returns { mode: 'anthropic-direct' } so launch.js stays generic.
+ * Returns { withGemini: boolean }.
  */
-export async function ensurePreset(/* hub unused in CCR mode */) {
+export async function ensurePreset() {
   await mkdir(CCR_DIR, { recursive: true });
   const state = await readState();
+  const withGemini = await isGeminiConfigured();
+  const config = buildConfig({ withGemini });
 
   if (await exists(CCR_CONFIG)) {
     let existing;
@@ -69,16 +94,10 @@ export async function ensurePreset(/* hub unused in CCR mode */) {
       await writeState(state);
       console.log(`💾 Найден свой config.json у claude-code-router — сохранил резервную копию: ${backupPath}`);
     }
-
-    if (isOurs) {
-      // Already managed by us; rewrite each time so updates roll out.
-      await writeFile(CCR_CONFIG, JSON.stringify(FREE_CONFIG, null, 2));
-      return { mode: 'anthropic-direct' };
-    }
   }
 
-  await writeFile(CCR_CONFIG, JSON.stringify(FREE_CONFIG, null, 2));
-  return { mode: 'anthropic-direct' };
+  await writeFile(CCR_CONFIG, JSON.stringify(config, null, 2));
+  return { withGemini };
 }
 
-export const CCR_PORT = FREE_CONFIG.PORT;
+export const CCR_PORT = 3456;

package/src/setup-gemini.js

@@ -0,0 +1,169 @@
+import { spawn } from 'node:child_process';
+import { mkdir, writeFile, chmod, readFile, access } from 'node:fs/promises';
+import { homedir, platform } from 'node:os';
+import { join } from 'node:path';
+import { createInterface } from 'node:readline';
+
+const ROOT = join(homedir(), '.krasavacode');
+const ENV_FILE = join(ROOT, 'gemini.env');
+const STATE_FILE = join(ROOT, 'state.json');
+
+const CONSOLE_URL = 'https://aistudio.google.com/apikey';
+
+function openBrowser(url) {
+  const cmd = platform() === 'darwin' ? 'open'
+    : platform() === 'win32' ? 'start'
+    : 'xdg-open';
+  const args = platform() === 'win32' ? ['', url] : [url];
+  spawn(cmd, args, { detached: true, stdio: 'ignore', shell: platform() === 'win32' }).unref();
+}
+
+function readState() {
+  return readFile(STATE_FILE, 'utf8').then(JSON.parse).catch(() => ({}));
+}
+async function writeState(s) { await writeFile(STATE_FILE, JSON.stringify(s, null, 2)); }
+
+function prompt(question) {
+  const rl = createInterface({ input: process.stdin, output: process.stdout });
+  return new Promise(resolve => rl.question(question, ans => { rl.close(); resolve(ans); }));
+}
+
+function isValidKeyFormat(key) {
+  // Google API keys look like AIza followed by 35 chars [A-Za-z0-9_-]
+  return /^AIza[A-Za-z0-9_-]{35}$/.test(key);
+}
+
+/** Sanity-check the key with a tiny Gemini API call. */
+async function verifyKey(key) {
+  const url = `https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:generateContent?key=${encodeURIComponent(key)}`;
+  const t0 = Date.now();
+  const res = await fetch(url, {
+    method: 'POST',
+    headers: { 'content-type': 'application/json' },
+    body: JSON.stringify({
+      contents: [{ parts: [{ text: 'Say only the word: ok' }] }],
+      generationConfig: { maxOutputTokens: 20 },
+    }),
+    signal: AbortSignal.timeout(15000),
+  });
+  const ms = Date.now() - t0;
+  if (!res.ok) {
+    let msg = `HTTP ${res.status}`;
+    try { const j = await res.json(); msg = j.error?.message || msg; } catch {}
+    return { ok: false, error: msg, ms };
+  }
+  const data = await res.json();
+  const text = data.candidates?.[0]?.content?.parts?.[0]?.text || '';
+  return { ok: true, text: text.trim(), ms };
+}
+
+function header(text) {
+  const line = '━'.repeat(58);
+  console.log('');
+  console.log(line);
+  console.log('  ' + text);
+  console.log(line);
+  console.log('');
+}
+
+export async function runSetupGemini() {
+  await mkdir(ROOT, { recursive: true });
+
+  console.log('');
+  console.log('  ╔══════════════════════════════════════════════════╗');
+  console.log('  ║  KRASAVACODE — апгрейд на Gemini 2.5 Flash       ║');
+  console.log('  ╚══════════════════════════════════════════════════╝');
+  console.log('');
+  console.log('  Сейчас вайбкодинг работает на простой модели.');
+  console.log('  Подключив бесплатный Google Gemini, получишь:');
+  console.log('    ✓ Качество в разы выше');
+  console.log('    ✓ 1500 запросов в день бесплатно');
+  console.log('    ✓ Без банковской карты');
+  console.log('');
+  console.log('  Это займёт 60 секунд.');
+
+  header('Открываю в браузере: ' + CONSOLE_URL);
+  openBrowser(CONSOLE_URL);
+
+  console.log('  ШАГ 1. Войди через Google.');
+  console.log('         (если у тебя Gmail, YouTube или Android — это он)');
+  console.log('');
+  console.log('  ШАГ 2. Нажми кнопку «Create API key» наверху страницы.');
+  console.log('         Если попросят выбрать проект — оставь предложенный.');
+  console.log('');
+  console.log('  ШАГ 3. Появится длинная строка-ключ. Нажми «Copy».');
+  console.log('         Ключ начинается с «AIza».');
+  console.log('');
+  console.log('  ШАГ 4. Вернись сюда в это окно и вставь ключ ниже.');
+  console.log('         Mac: ⌘+V    Windows/Linux: Ctrl+V    или правая кнопка → Paste');
+  console.log('');
+
+  let key;
+  for (let attempt = 0; attempt < 3; attempt++) {
+    key = (await prompt('  Вставь ключ Gemini сюда: ')).trim();
+    if (!key) {
+      console.log('  ⚠️  Пусто. Скопируй ключ и попробуй ещё раз.\n');
+      continue;
+    }
+    if (!isValidKeyFormat(key)) {
+      console.log('  ⚠️  Не похоже на ключ Gemini.');
+      console.log('      Должно быть AIza + 35 символов (всего 39).');
+      console.log('      Скопируй ещё раз внимательно.\n');
+      continue;
+    }
+    break;
+  }
+  if (!isValidKeyFormat(key)) {
+    console.log('\n  ❌ Не удалось получить ключ. Запусти `krasavacode setup-gemini` ещё раз.');
+    return;
+  }
+
+  console.log('\n  ⏳ Проверяю ключ через тестовый запрос…');
+  const result = await verifyKey(key);
+  if (!result.ok) {
+    console.log(`  ❌ Ключ не работает: ${result.error}`);
+    console.log('     Проверь, что скопировал целиком, без пробелов.');
+    console.log('     Иногда Google требует подождать ~30 секунд после создания ключа.');
+    console.log('     Запусти `krasavacode setup-gemini` ещё раз.');
+    return;
+  }
+  console.log(`  ✅ Работает! Gemini ответил «${result.text}» за ${(result.ms / 1000).toFixed(1)} сек.`);
+
+  // Save the key in a private env file (chmod 600)
+  const envContent = `GEMINI_API_KEY=${key}\n`;
+  await writeFile(ENV_FILE, envContent);
+  try { await chmod(ENV_FILE, 0o600); } catch {}
+
+  // Mark in state.json so future `krasavacode` runs know to use Gemini
+  const state = await readState();
+  state.geminiConfigured = true;
+  state.geminiConfiguredAt = new Date().toISOString();
+  await writeState(state);
+
+  console.log('');
+  console.log('  💾 Ключ сохранён в ' + ENV_FILE);
+  console.log('     (он остаётся только у тебя на компьютере, никуда не отправляется)');
+  console.log('');
+  console.log('  ✅ Готово! Теперь твой вайбкодинг — на Gemini 2.5 Flash.');
+  console.log('');
+
+  const launch = (await prompt('  Запустить вайбкодинг прямо сейчас? [Enter — да, n — позже]: ')).trim().toLowerCase();
+  if (launch === '' || launch === 'y' || launch === 'yes' || launch === 'д' || launch === 'да') {
+    return { launchAfter: true };
+  }
+  console.log('\n  ОК. Когда захочешь — запусти `krasavacode`.');
+  return { launchAfter: false };
+}
+
+/** Read GEMINI_API_KEY from gemini.env. Returns null if not configured. */
+export async function loadGeminiKey() {
+  try {
+    const content = await readFile(ENV_FILE, 'utf8');
+    const m = content.match(/^GEMINI_API_KEY=(.+)$/m);
+    return m ? m[1].trim() : null;
+  } catch { return null; }
+}
+
+export async function isGeminiConfigured() {
+  return access(ENV_FILE).then(() => true).catch(() => false);
+}