@su-record/vibe 2.6.28 → 2.6.31

package/hooks/scripts/llm-orchestrate.js

@@ -1,171 +1,245 @@
-/**
- * UserPromptSubmit Hook - LLM 오케스트레이션 (GPT/Gemini)
- *
- * Usage:
- *   node llm-orchestrate.js <provider> <mode> "prompt"
- *   node llm-orchestrate.js <provider> <mode> "systemPrompt" "prompt"
- *
- *   provider: gpt | gemini
- *   mode: orchestrate | orchestrate-json
- *
- * Features:
- *   - Exponential backoff retry (3 attempts)
- *   - Auto fallback: gemini → gpt, gpt → gemini
- *   - Overload/rate-limit detection
- *
- * Input: JSON from stdin with { prompt: string } (when no CLI args)
- */
-import { getLibBaseUrl } from './utils.js';
-
-const LIB_URL = getLibBaseUrl();
-const DEFAULT_SYSTEM_PROMPT = 'You are a helpful assistant.';
-
-const provider = process.argv[2] || 'gemini';
-const mode = process.argv[3] || 'orchestrate';
-
-// Retry configuration
-const MAX_RETRIES = 3;
-const INITIAL_DELAY_MS = 2000;
-
-// Errors that should skip retry and go to fallback immediately
-const SKIP_RETRY_PATTERNS = [
-  /rate.?limit/i,
-  /quota/i,
-  /unauthorized/i,
-  /forbidden/i,
-  /401/,
-  /403/,
-  /429/,
-];
-
-// Errors that should trigger retry with backoff
-const RETRY_PATTERNS = [
-  /overload/i,
-  /503/,
-  /5\d\d/,
-  /network/i,
-  /timeout/i,
-  /ECONNRESET/i,
-  /ETIMEDOUT/i,
-];
-
-function shouldSkipRetry(errorMsg) {
-  return SKIP_RETRY_PATTERNS.some(pattern => pattern.test(errorMsg));
-}
-
-function shouldRetry(errorMsg) {
-  return RETRY_PATTERNS.some(pattern => pattern.test(errorMsg));
-}
-
-function sleep(ms) {
-  return new Promise(resolve => setTimeout(resolve, ms));
-}
-
-async function callProvider(providerName, prompt, sysPrompt, jsonMode) {
-  const modulePath = `${LIB_URL}${providerName}-api.js`;
-  const module = await import(modulePath);
-
-  const orchestrateFn = providerName === 'gpt'
-    ? module.vibeGptOrchestrate
-    : module.vibeGeminiOrchestrate;
-
-  return await orchestrateFn(prompt, sysPrompt, { jsonMode });
-}
-
-async function callWithRetry(providerName, prompt, sysPrompt, jsonMode) {
-  let lastError;
-
-  for (let attempt = 1; attempt <= MAX_RETRIES; attempt++) {
-    try {
-      return { success: true, result: await callProvider(providerName, prompt, sysPrompt, jsonMode) };
-    } catch (e) {
-      lastError = e;
-      const errorMsg = e.message || String(e);
-
-      // Skip retry for auth/quota errors - go to fallback immediately
-      if (shouldSkipRetry(errorMsg)) {
-        return { success: false, error: errorMsg, skipToFallback: true };
-      }
-
-      // Retry with backoff for transient errors
-      if (shouldRetry(errorMsg) && attempt < MAX_RETRIES) {
-        const delay = INITIAL_DELAY_MS * Math.pow(2, attempt - 1);
-        console.error(`[${providerName.toUpperCase()}] Retry ${attempt}/${MAX_RETRIES} after ${delay}ms...`);
-        await sleep(delay);
-        continue;
-      }
-
-      // Unknown error or max retries reached
-      return { success: false, error: errorMsg, skipToFallback: false };
-    }
-  }
-
-  return { success: false, error: lastError?.message || 'Max retries exceeded', skipToFallback: false };
-}
-
-async function main() {
-  let prompt;
-  let systemPrompt = DEFAULT_SYSTEM_PROMPT;
-
-  // CLI argument가 있으면 사용
-  // Usage 1: node script.js gpt orchestrate "system prompt" "user prompt"
-  // Usage 2: node script.js gpt orchestrate "user prompt" (uses default system prompt)
-  const arg4 = process.argv[4]?.trim();
-  const arg5 = process.argv.slice(5).join(' ').trim();
-
-  if (arg5) {
-    // 5번째 인자가 있으면: arg4=시스템 프롬프트, arg5=사용자 프롬프트
-    systemPrompt = arg4;
-    prompt = arg5;
-  } else if (arg4) {
-    // 4번째 인자만 있으면: arg4=사용자 프롬프트 (시스템 프롬프트는 기본값)
-    prompt = arg4;
-  } else {
-    // Hook에서 호출: stdin으로 JSON 입력
-    let inputData = '';
-    for await (const chunk of process.stdin) {
-      inputData += chunk;
-    }
-
-    try {
-      const parsed = JSON.parse(inputData);
-      prompt = parsed.prompt;
-    } catch {
-      console.log(`[${provider.toUpperCase()}] Error: Invalid JSON input`);
-      return;
-    }
-  }
-
-  // 접두사 제거
-  const prefixPatterns = {
-    gpt: /^(gpt[-.\s]|지피티-|vibe-gpt-)\s*/i,
-    gemini: /^(gemini[-.\s]|제미나이-|vibe-gemini-)\s*/i,
-  };
-  const cleanPrompt = prompt.replace(prefixPatterns[provider] || /^/, '').trim();
-  const jsonMode = mode === 'orchestrate-json';
-
-  // Provider chain: primary → fallback
-  const fallbackProvider = provider === 'gpt' ? 'gemini' : 'gpt';
-  const providerChain = [provider, fallbackProvider];
-
-  for (const currentProvider of providerChain) {
-    const label = currentProvider === 'gpt' ? 'GPT-5.2' : 'Gemini-3';
-    const result = await callWithRetry(currentProvider, cleanPrompt, systemPrompt, jsonMode);
-
-    if (result.success) {
-      console.log(`${label} response: ${result.result}`);
-      return;
-    }
-
-    // Log failure and try fallback
-    if (currentProvider !== providerChain[providerChain.length - 1]) {
-      const fallbackLabel = fallbackProvider === 'gpt' ? 'GPT' : 'Gemini';
-      console.error(`[${currentProvider.toUpperCase()}] Failed: ${result.error}. Falling back to ${fallbackLabel}...`);
-    } else {
-      // All providers failed
-      console.log(`[LLM] Error: All providers failed. Last error: ${result.error}`);
-    }
-  }
-}
-
-main();
+/**
+ * UserPromptSubmit Hook - LLM 오케스트레이션 (GPT/Gemini)
+ *
+ * Usage:
+ *   node llm-orchestrate.js <provider> <mode> "prompt"
+ *   node llm-orchestrate.js <provider> <mode> "systemPrompt" "prompt"
+ *
+ *   provider: gpt | gemini
+ *   mode: orchestrate | orchestrate-json | image
+ *
+ * Image Mode:
+ *   node llm-orchestrate.js gemini image "prompt" --output "./image.png"
+ *   node llm-orchestrate.js gemini image "prompt" --output "./image.png" --size "1920x1080"
+ *
+ * Features:
+ *   - Exponential backoff retry (3 attempts)
+ *   - Auto fallback: gemini → gpt, gpt → gemini
+ *   - Overload/rate-limit detection
+ *   - Image generation (Gemini only, Nano Banana model)
+ *
+ * Input: JSON from stdin with { prompt: string } (when no CLI args)
+ */
+import { getLibBaseUrl } from './utils.js';
+import fs from 'fs';
+import path from 'path';
+
+const LIB_URL = getLibBaseUrl();
+const DEFAULT_SYSTEM_PROMPT = 'You are a helpful assistant.';
+
+const provider = process.argv[2] || 'gemini';
+const mode = process.argv[3] || 'orchestrate';
+
+// Retry configuration
+const MAX_RETRIES = 3;
+const INITIAL_DELAY_MS = 2000;
+
+// Errors that should skip retry and go to fallback immediately
+const SKIP_RETRY_PATTERNS = [
+  /rate.?limit/i,
+  /quota/i,
+  /unauthorized/i,
+  /forbidden/i,
+  /401/,
+  /403/,
+  /429/,
+];
+
+// Errors that should trigger retry with backoff
+const RETRY_PATTERNS = [
+  /overload/i,
+  /503/,
+  /5\d\d/,
+  /network/i,
+  /timeout/i,
+  /ECONNRESET/i,
+  /ETIMEDOUT/i,
+];
+
+function shouldSkipRetry(errorMsg) {
+  return SKIP_RETRY_PATTERNS.some(pattern => pattern.test(errorMsg));
+}
+
+function shouldRetry(errorMsg) {
+  return RETRY_PATTERNS.some(pattern => pattern.test(errorMsg));
+}
+
+function sleep(ms) {
+  return new Promise(resolve => setTimeout(resolve, ms));
+}
+
+// ============================================
+// Image Generation (delegates to gemini-api)
+// ============================================
+
+function parseImageArgs(args) {
+  const result = { prompt: null, output: './generated-image.png', size: '1024x1024' };
+  for (let i = 0; i < args.length; i++) {
+    if (args[i] === '--output' || args[i] === '-o') {
+      result.output = args[++i];
+    } else if (args[i] === '--size' || args[i] === '-s') {
+      result.size = args[++i];
+    } else if (!args[i].startsWith('-') && !result.prompt) {
+      result.prompt = args[i];
+    }
+  }
+  return result;
+}
+
+async function callProvider(providerName, prompt, sysPrompt, jsonMode) {
+  const modulePath = `${LIB_URL}${providerName}-api.js`;
+  const module = await import(modulePath);
+
+  const orchestrateFn = providerName === 'gpt'
+    ? module.vibeGptOrchestrate
+    : module.vibeGeminiOrchestrate;
+
+  return await orchestrateFn(prompt, sysPrompt, { jsonMode });
+}
+
+async function callWithRetry(providerName, prompt, sysPrompt, jsonMode) {
+  let lastError;
+
+  for (let attempt = 1; attempt <= MAX_RETRIES; attempt++) {
+    try {
+      return { success: true, result: await callProvider(providerName, prompt, sysPrompt, jsonMode) };
+    } catch (e) {
+      lastError = e;
+      const errorMsg = e.message || String(e);
+
+      // Skip retry for auth/quota errors - go to fallback immediately
+      if (shouldSkipRetry(errorMsg)) {
+        return { success: false, error: errorMsg, skipToFallback: true };
+      }
+
+      // Retry with backoff for transient errors
+      if (shouldRetry(errorMsg) && attempt < MAX_RETRIES) {
+        const delay = INITIAL_DELAY_MS * Math.pow(2, attempt - 1);
+        console.error(`[${providerName.toUpperCase()}] Retry ${attempt}/${MAX_RETRIES} after ${delay}ms...`);
+        await sleep(delay);
+        continue;
+      }
+
+      // Unknown error or max retries reached
+      return { success: false, error: errorMsg, skipToFallback: false };
+    }
+  }
+
+  return { success: false, error: lastError?.message || 'Max retries exceeded', skipToFallback: false };
+}
+
+async function main() {
+  // Image mode handling
+  if (mode === 'image') {
+    if (provider !== 'gemini') {
+      console.log('[IMAGE] Error: Image generation only supports gemini provider');
+      return;
+    }
+
+    const imageArgs = parseImageArgs(process.argv.slice(4));
+    if (!imageArgs.prompt) {
+      console.log('[IMAGE] Error: --prompt is required');
+      return;
+    }
+
+    // Ensure output directory exists
+    const outputDir = path.dirname(imageArgs.output);
+    if (outputDir && outputDir !== '.' && !fs.existsSync(outputDir)) {
+      fs.mkdirSync(outputDir, { recursive: true });
+    }
+
+    console.error(`[IMAGE] Generating with Nano Banana...`);
+    console.error(`  Prompt: ${imageArgs.prompt}`);
+    console.error(`  Size: ${imageArgs.size}`);
+    console.error(`  Output: ${imageArgs.output}`);
+
+    try {
+      const geminiApi = await import(`${LIB_URL}gemini-api.js`);
+      const image = await geminiApi.generateImage(imageArgs.prompt, { size: imageArgs.size });
+      fs.writeFileSync(imageArgs.output, image.data);
+      const stats = fs.statSync(imageArgs.output);
+      const sizeKB = (stats.size / 1024).toFixed(1);
+
+      console.log(JSON.stringify({
+        success: true,
+        path: path.resolve(imageArgs.output),
+        size: stats.size,
+        sizeKB: `${sizeKB} KB`,
+        prompt: imageArgs.prompt
+      }));
+    } catch (err) {
+      console.log(JSON.stringify({
+        success: false,
+        error: err.message,
+        prompt: imageArgs.prompt
+      }));
+    }
+    return;
+  }
+
+  // Text generation mode (orchestrate / orchestrate-json)
+  let prompt;
+  let systemPrompt = DEFAULT_SYSTEM_PROMPT;
+
+  // CLI argument가 있으면 사용
+  // Usage 1: node script.js gpt orchestrate "system prompt" "user prompt"
+  // Usage 2: node script.js gpt orchestrate "user prompt" (uses default system prompt)
+  const arg4 = process.argv[4]?.trim();
+  const arg5 = process.argv.slice(5).join(' ').trim();
+
+  if (arg5) {
+    // 5번째 인자가 있으면: arg4=시스템 프롬프트, arg5=사용자 프롬프트
+    systemPrompt = arg4;
+    prompt = arg5;
+  } else if (arg4) {
+    // 4번째 인자만 있으면: arg4=사용자 프롬프트 (시스템 프롬프트는 기본값)
+    prompt = arg4;
+  } else {
+    // Hook에서 호출: stdin으로 JSON 입력
+    let inputData = '';
+    for await (const chunk of process.stdin) {
+      inputData += chunk;
+    }
+
+    try {
+      const parsed = JSON.parse(inputData);
+      prompt = parsed.prompt;
+    } catch {
+      console.log(`[${provider.toUpperCase()}] Error: Invalid JSON input`);
+      return;
+    }
+  }
+
+  // 접두사 제거
+  const prefixPatterns = {
+    gpt: /^(gpt[-.\s]|지피티-|vibe-gpt-)\s*/i,
+    gemini: /^(gemini[-.\s]|제미나이-|vibe-gemini-)\s*/i,
+  };
+  const cleanPrompt = prompt.replace(prefixPatterns[provider] || /^/, '').trim();
+  const jsonMode = mode === 'orchestrate-json';
+
+  // Provider chain: primary → fallback
+  const fallbackProvider = provider === 'gpt' ? 'gemini' : 'gpt';
+  const providerChain = [provider, fallbackProvider];
+
+  for (const currentProvider of providerChain) {
+    const label = currentProvider === 'gpt' ? 'GPT-5.2' : 'Gemini-3';
+    const result = await callWithRetry(currentProvider, cleanPrompt, systemPrompt, jsonMode);
+
+    if (result.success) {
+      console.log(`${label} response: ${result.result}`);
+      return;
+    }
+
+    // Log failure and try fallback
+    if (currentProvider !== providerChain[providerChain.length - 1]) {
+      const fallbackLabel = fallbackProvider === 'gpt' ? 'GPT' : 'Gemini';
+      console.error(`[${currentProvider.toUpperCase()}] Failed: ${result.error}. Falling back to ${fallbackLabel}...`);
+    } else {
+      // All providers failed
+      console.log(`[LLM] Error: All providers failed. Last error: ${result.error}`);
+    }
+  }
+}
+
+main();

package/hooks/scripts/post-edit.js

@@ -1,97 +1,97 @@
-/**
- * PostToolUse Hook - Edit 후 자동 포맷팅 및 TypeScript 체크
- *
- * 기능:
- * - TypeScript/JavaScript 파일 수정 시 자동 Prettier 포맷팅
- * - TypeScript 파일 수정 시 타입 체크
- * - console.log 감지 및 경고
- */
-import { execSync } from 'child_process';
-import { existsSync } from 'fs';
-import { PROJECT_DIR } from './utils.js';
-
-// Claude Code에서 전달받는 환경변수에서 파일 경로 추출
-const toolInput = process.env.TOOL_INPUT || '{}';
-
-function main() {
-  try {
-    const input = JSON.parse(toolInput);
-    const filePath = input.file_path || input.path || '';
-
-    if (!filePath) {
-      return;
-    }
-
-    // TypeScript/JavaScript 파일인지 확인
-    const isTs = /\.(ts|tsx)$/.test(filePath);
-    const isJs = /\.(js|jsx|mjs|cjs)$/.test(filePath);
-
-    if (!isTs && !isJs) {
-      return;
-    }
-
-    const results = [];
-
-    // Prettier 포맷팅 (prettier가 설치된 경우만)
-    const prettierConfig = [
-      `${PROJECT_DIR}/.prettierrc`,
-      `${PROJECT_DIR}/.prettierrc.json`,
-      `${PROJECT_DIR}/.prettierrc.js`,
-      `${PROJECT_DIR}/prettier.config.js`,
-    ].find(p => existsSync(p));
-
-    if (prettierConfig || existsSync(`${PROJECT_DIR}/node_modules/.bin/prettier`)) {
-      try {
-        execSync(`npx prettier --write "${filePath}" 2>/dev/null`, {
-          cwd: PROJECT_DIR,
-          stdio: 'pipe',
-          timeout: 10000,
-        });
-        results.push('formatted');
-      } catch {
-        // Prettier 실패는 무시
-      }
-    }
-
-    // TypeScript 타입 체크 (tsconfig가 있는 경우만)
-    if (isTs && existsSync(`${PROJECT_DIR}/tsconfig.json`)) {
-      try {
-        execSync(`npx tsc --noEmit --skipLibCheck 2>&1 | head -5`, {
-          cwd: PROJECT_DIR,
-          stdio: 'pipe',
-          timeout: 30000,
-        });
-        results.push('types OK');
-      } catch (e) {
-        const output = e.stdout?.toString() || '';
-        const errorCount = (output.match(/error TS/g) || []).length;
-        if (errorCount > 0) {
-          results.push(`${errorCount} type errors`);
-        }
-      }
-    }
-
-    // console.log 감지
-    try {
-      const grepResult = execSync(
-        `grep -n "console\\.log" "${filePath}" 2>/dev/null | head -3`,
-        { cwd: PROJECT_DIR, stdio: 'pipe', encoding: 'utf-8' }
-      );
-      if (grepResult.trim()) {
-        const lines = grepResult.trim().split('\n').map(l => l.split(':')[0]).join(',');
-        results.push(`console.log at line ${lines}`);
-      }
-    } catch {
-      // grep 실패는 console.log 없음을 의미
-    }
-
-    if (results.length > 0) {
-      console.log(`[POST-EDIT] ${filePath.split('/').pop()}: ${results.join(' | ')}`);
-    }
-
-  } catch {
-    // 조용히 실패
-  }
-}
-
-main();
+/**
+ * PostToolUse Hook - Edit 후 자동 포맷팅 및 TypeScript 체크
+ *
+ * 기능:
+ * - TypeScript/JavaScript 파일 수정 시 자동 Prettier 포맷팅
+ * - TypeScript 파일 수정 시 타입 체크
+ * - console.log 감지 및 경고
+ */
+import { execSync } from 'child_process';
+import { existsSync } from 'fs';
+import { PROJECT_DIR } from './utils.js';
+
+// Claude Code에서 전달받는 환경변수에서 파일 경로 추출
+const toolInput = process.env.TOOL_INPUT || '{}';
+
+function main() {
+  try {
+    const input = JSON.parse(toolInput);
+    const filePath = input.file_path || input.path || '';
+
+    if (!filePath) {
+      return;
+    }
+
+    // TypeScript/JavaScript 파일인지 확인
+    const isTs = /\.(ts|tsx)$/.test(filePath);
+    const isJs = /\.(js|jsx|mjs|cjs)$/.test(filePath);
+
+    if (!isTs && !isJs) {
+      return;
+    }
+
+    const results = [];
+
+    // Prettier 포맷팅 (prettier가 설치된 경우만)
+    const prettierConfig = [
+      `${PROJECT_DIR}/.prettierrc`,
+      `${PROJECT_DIR}/.prettierrc.json`,
+      `${PROJECT_DIR}/.prettierrc.js`,
+      `${PROJECT_DIR}/prettier.config.js`,
+    ].find(p => existsSync(p));
+
+    if (prettierConfig || existsSync(`${PROJECT_DIR}/node_modules/.bin/prettier`)) {
+      try {
+        execSync(`npx prettier --write "${filePath}" 2>/dev/null`, {
+          cwd: PROJECT_DIR,
+          stdio: 'pipe',
+          timeout: 10000,
+        });
+        results.push('formatted');
+      } catch {
+        // Prettier 실패는 무시
+      }
+    }
+
+    // TypeScript 타입 체크 (tsconfig가 있는 경우만)
+    if (isTs && existsSync(`${PROJECT_DIR}/tsconfig.json`)) {
+      try {
+        execSync(`npx tsc --noEmit --skipLibCheck 2>&1 | head -5`, {
+          cwd: PROJECT_DIR,
+          stdio: 'pipe',
+          timeout: 30000,
+        });
+        results.push('types OK');
+      } catch (e) {
+        const output = e.stdout?.toString() || '';
+        const errorCount = (output.match(/error TS/g) || []).length;
+        if (errorCount > 0) {
+          results.push(`${errorCount} type errors`);
+        }
+      }
+    }
+
+    // console.log 감지
+    try {
+      const grepResult = execSync(
+        `grep -n "console\\.log" "${filePath}" 2>/dev/null | head -3`,
+        { cwd: PROJECT_DIR, stdio: 'pipe', encoding: 'utf-8' }
+      );
+      if (grepResult.trim()) {
+        const lines = grepResult.trim().split('\n').map(l => l.split(':')[0]).join(',');
+        results.push(`console.log at line ${lines}`);
+      }
+    } catch {
+      // grep 실패는 console.log 없음을 의미
+    }
+
+    if (results.length > 0) {
+      console.log(`[POST-EDIT] ${filePath.split('/').pop()}: ${results.join(' | ')}`);
+    }
+
+  } catch {
+    // 조용히 실패
+  }
+}
+
+main();