cciwon-code-review-cli 2.0.9 → 2.1.0

package/lib/api-client.js

@@ -63,6 +63,40 @@ class CodeReviewClient {
     }
   }
 
+  /**
+   * 대화형 코드 리뷰
+   * @param {string} sessionId - 세션 ID
+   * @param {string} message - 사용자 메시지
+   * @param {string} fileContent - 파일 내용 (최초 1회)
+   * @param {string} filePath - 파일 경로 (최초 1회)
+   */
+  async chat(sessionId, message, fileContent = null, filePath = null) {
+    try {
+      const response = await this.client.post('/chat', {
+        session_id: sessionId,
+        message,
+        file_content: fileContent,
+        file_path: filePath
+      });
+      return response.data;
+    } catch (error) {
+      this.handleError(error);
+    }
+  }
+
+  /**
+   * 대화 세션 초기화
+   * @param {string} sessionId - 세션 ID
+   */
+  async clearChat(sessionId) {
+    try {
+      const response = await this.client.delete(`/chat/${sessionId}`);
+      return response.data;
+    } catch (error) {
+      this.handleError(error);
+    }
+  }
+
   /**
    * 에러 핸들링
    */

package/lib/chat-mode.js

@@ -3,17 +3,18 @@ const readline = require('readline');
 const chalk = require('chalk');
 const FileReader = require('./file-reader');
 const CodeReviewClient = require('./api-client');
-const { parseDiffAndShowInVSCode } = require('./diff-viewer');
+const crypto = require('crypto');
 
 class ChatMode {
   constructor() {
     this.fileReader = new FileReader();
-    // 환경변수가 있으면 사용, 없으면 기본값 사용
     const serverUrl = process.env.CODE_REVIEW_SERVER;
     this.client = serverUrl ? new CodeReviewClient(serverUrl) : new CodeReviewClient();
     this.projectFiles = [];
-    this.filesContent = {};
     this.rl = null;
+    this.sessionId = crypto.randomBytes(16).toString('hex');
+    this.currentFile = null;  // 현재 리뷰 중인 파일
+    this.currentFileContent = null;
   }
 
   /**
@@ -68,7 +69,8 @@ class ChatMode {
       prompt: chalk.bold.cyan('> ')
     });
 
-    console.log(chalk.gray('무엇을 도와드릴까요? (종료하려면 "exit" 입력)\n'));
+    console.log(chalk.gray('파일을 선택하여 대화를 시작하세요. 예: "api.py 리뷰해줘"'));
+    console.log(chalk.gray('종료하려면 "exit" 입력\n'));
     this.rl.prompt();
 
     this.rl.on('line', async (input) => {
@@ -81,10 +83,17 @@ class ChatMode {
 
       if (command === 'exit' || command === 'quit') {
         console.log(chalk.green('\n👋 안녕히 가세요!'));
+        await this.cleanup();
         this.rl.close();
         process.exit(0);
       }
 
+      if (command === 'clear' || command === '초기화') {
+        await this.clearCurrentFile();
+        this.rl.prompt();
+        return;
+      }
+
       await this.handleCommand(command);
       this.rl.prompt();
     });
@@ -100,10 +109,10 @@ class ChatMode {
   async handleCommand(command) {
     const lowerCommand = command.toLowerCase();
 
-    // 파일 리뷰 요청 감지
-    if (lowerCommand.includes('리뷰') || lowerCommand.includes('review') ||
-        lowerCommand.includes('검사') || lowerCommand.includes('check')) {
-      await this.handleReviewRequest(command);
+    // 파일 선택 또는 리뷰 요청 감지
+    if (lowerCommand.includes('리뷰') || lowerCommand.includes('review')) {
+      // 파일 선택 및 대화 시작
+      await this.selectFileAndStartChat(command);
     }
     // 파일 목록 요청
     else if (lowerCommand.includes('목록') || lowerCommand.includes('list') ||
@@ -114,17 +123,22 @@ class ChatMode {
     else if (lowerCommand.includes('help') || lowerCommand.includes('도움말')) {
       this.showHelp();
     }
-    // 기타 - 일반 질문으로 처리
+    // 기타 - 현재 파일에 대한 대화
     else {
-      console.log(chalk.yellow('\n💡 파일을 리뷰하려면 "파일명 리뷰해줘" 형식으로 입력해주세요.'));
-      console.log(chalk.gray('   예: api.py 리뷰해줘, 모든 Python 파일 검사해줘\n'));
+      if (this.currentFile) {
+        // 현재 파일이 선택되어 있으면 대화 계속
+        await this.continueChat(command);
+      } else {
+        console.log(chalk.yellow('\n💡 먼저 파일을 선택해주세요.'));
+        console.log(chalk.gray('   예: api.py 리뷰해줘\n'));
+      }
     }
   }
 
   /**
-   * 리뷰 요청 처리
+   * 파일 선택 및 대화 시작
    */
-  async handleReviewRequest(command) {
+  async selectFileAndStartChat(command) {
     // 파일 매칭
     const matchedFiles = this.fileReader.matchFiles(command, this.projectFiles);
 
@@ -134,44 +148,94 @@ class ChatMode {
       return;
     }
 
-    console.log(chalk.cyan(`\n🔍 ${matchedFiles.length}개 파일 리뷰 중...\n`));
+    if (matchedFiles.length > 1) {
+      console.log(chalk.yellow('\n⚠️  여러 파일이 매칭됩니다. 하나를 선택해주세요:'));
+      matchedFiles.forEach(file => {
+        console.log(chalk.gray(`  - ${file}`));
+      });
+      console.log('');
+      return;
+    }
 
-    for (const filePath of matchedFiles) {
-      await this.reviewFile(filePath);
+    const filePath = matchedFiles[0];
+
+    // 파일 내용 읽기
+    try {
+      console.log(chalk.cyan(`\n📖 ${filePath} 파일을 읽는 중...`));
+      const content = await this.fileReader.readFile(filePath);
+
+      // 파일이 바뀌면 세션 초기화
+      if (this.currentFile && this.currentFile !== filePath) {
+        await this.client.clearChat(this.sessionId);
+        this.sessionId = crypto.randomBytes(16).toString('hex');
+      }
+
+      this.currentFile = filePath;
+      this.currentFileContent = content;
+
+      console.log(chalk.green(`✅ 파일 로드 완료 (${content.length}자)`));
+      console.log(chalk.cyan(`\n💬 "${this.currentFile}" 에 대해 질문하세요!\n`));
+
+      // 초기 메시지 전송
+      const initialMessage = "이 파일을 리뷰해주세요. 주요 문제점과 개선사항을 알려주세요.";
+      await this.sendMessage(initialMessage, true);
+
+    } catch (error) {
+      console.error(chalk.red(`❌ 파일 읽기 실패: ${error.message}\n`));
     }
   }
 
   /**
-   * 단일 파일 리뷰
+   * 대화 계속하기
    */
-  async reviewFile(filePath) {
-    try {
-      console.log(chalk.bold(`\n=== ${filePath} ===\n`));
-
-      // 파일 내용 읽기
-      console.log(chalk.gray(`📖 파일 읽는 중...`));
-      const content = await this.fileReader.readFile(filePath);
-      console.log(chalk.gray(`✓ 파일 읽기 완료 (${content.length}자)`));
-
-      // 언어 감지
-      const language = this.detectLanguage(filePath);
-      console.log(chalk.gray(`🔍 언어: ${language}`));
+  async continueChat(message) {
+    await this.sendMessage(message, false);
+  }
 
-      // 서버에 리뷰 요청
-      console.log(chalk.gray(`🚀 서버로 리뷰 요청 중...`));
-      const result = await this.client.reviewCode(content, language);
-      console.log(chalk.gray(`✓ 리뷰 완료\n`));
+  /**
+   * 메시지 전송
+   */
+  async sendMessage(message, isFirst) {
+    try {
+      console.log(chalk.gray('🤔 생각 중...\n'));
 
-      // diff를 파싱하고 VSCode로 표시, 리뷰 텍스트만 반환
-      const reviewTextOnly = parseDiffAndShowInVSCode(result.review, filePath);
+      const response = await this.client.chat(
+        this.sessionId,
+        message,
+        isFirst ? this.currentFileContent : null,
+        isFirst ? this.currentFile : null
+      );
 
-      // 결과 출력 (리뷰 텍스트만)
-      console.log(chalk.bold.cyan('=== 코드 리뷰 결과 ==='));
-      console.log(reviewTextOnly);
+      console.log(chalk.bold.green('코드가드:'));
+      console.log(response.response);
       console.log('');
 
     } catch (error) {
-      console.error(chalk.red(`❌ 리뷰 실패 (${filePath}): ${error.message}\n`));
+      console.error(chalk.red(`❌ 에러: ${error.message}\n`));
+    }
+  }
+
+  /**
+   * 현재 파일 초기화
+   */
+  async clearCurrentFile() {
+    if (this.currentFile) {
+      await this.client.clearChat(this.sessionId);
+      console.log(chalk.yellow(`\n🔄 "${this.currentFile}" 대화를 초기화했습니다.\n`));
+      this.currentFile = null;
+      this.currentFileContent = null;
+      this.sessionId = crypto.randomBytes(16).toString('hex');
+    } else {
+      console.log(chalk.yellow('\n⚠️  활성화된 파일이 없습니다.\n'));
+    }
+  }
+
+  /**
+   * 정리
+   */
+  async cleanup() {
+    if (this.currentFile) {
+      await this.client.clearChat(this.sessionId);
     }
   }
 
@@ -196,35 +260,15 @@ class ChatMode {
    */
   showHelp() {
     console.log(chalk.bold('\n📖 사용 가능한 명령어:\n'));
-    console.log(chalk.cyan('  파일명 리뷰해줘') + chalk.gray('      - 특정 파일 리뷰'));
-    console.log(chalk.cyan('  모든 Python 파일 검사') + chalk.gray(' - 확장자별 리뷰'));
+    console.log(chalk.cyan('  파일명 리뷰해줘') + chalk.gray('      - 파일 선택 및 대화 시작'));
+    console.log(chalk.cyan('  이 부분 설명해줘') + chalk.gray('     - 선택된 파일에 대해 추가 질문'));
+    console.log(chalk.cyan('  보안 문제는?') + chalk.gray('         - 선택된 파일에 대해 추가 질문'));
     console.log(chalk.cyan('  파일 목록') + chalk.gray('            - 분석 가능한 파일 목록'));
+    console.log(chalk.cyan('  clear / 초기화') + chalk.gray('        - 현재 대화 초기화'));
     console.log(chalk.cyan('  help') + chalk.gray('                 - 이 도움말'));
     console.log(chalk.cyan('  exit') + chalk.gray('                 - 종료\n'));
   }
 
-  /**
-   * 파일 확장자로 언어 감지
-   */
-  detectLanguage(filePath) {
-    const ext = filePath.split('.').pop().toLowerCase();
-    const langMap = {
-      'py': 'python',
-      'js': 'javascript',
-      'jsx': 'javascript',
-      'ts': 'typescript',
-      'tsx': 'typescript',
-      'java': 'java',
-      'cpp': 'cpp',
-      'c': 'c',
-      'go': 'go',
-      'rs': 'rust',
-      'rb': 'ruby',
-      'php': 'php'
-    };
-    return langMap[ext] || 'text';
-  }
-
   /**
    * 사용자에게 질문하고 답변 받기
    */

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "cciwon-code-review-cli",
-  "version": "2.0.9",
+  "version": "2.1.0",
   "description": "AI-powered code review CLI tool using Qwen3-Coder-30B model with IP whitelist support",
   "main": "lib/api-client.js",
   "bin": {

package/.ipynb_checkpoints/README-checkpoint.md

@@ -1,233 +0,0 @@
-# cciwon-code-review-cli
-
-AI 기반 코드 리뷰 CLI 도구 (Qwen3-Coder-30B DPO 파인튜닝 모델 사용)
-
-## 특징
-
-- 🔒 **폐쇄망 지원**: IP 화이트리스트 기반 (192.168.10.152)
-- 📁 **폴더 전체 리뷰**: 프로젝트 전체를 한 번에 분석
-- 📄 **파일 단위 리뷰**: 개별 파일 심층 분석
-- 🛡️ **보안 중심**: SQL Injection, XSS 등 보안 취약점 탐지
-- ⚡ **빠른 응답**: vLLM 기반 고성능 추론
-
-## 설치
-
-### 1. 서버 설정
-
-먼저 vLLM 서버를 시작해야 합니다:
-
-```bash
-# 필수 패키지 설치
-pip install vllm fastapi uvicorn
-
-# 서버 실행
-cd /home/cwk317/codereivew
-python vllm_server.py
-```
-
-서버가 `192.168.10.152:8000`에서 실행됩니다.
-
-### 2. CLI 설치
-
-```bash
-cd code-review-cli
-npm install
-npm link  # 전역 설치
-```
-
-또는 로컬에서 직접 사용:
-
-```bash
-node bin/code-review.js <command>
-```
-
-## 사용법
-
-### 서버 상태 확인
-
-```bash
-code-review health
-```
-
-출력 예시:
-```
-✅ 서버 정상 작동 중
-
-서버 정보:
-  - 상태: healthy
-  - 모델 로드: 완료
-  - 서버 URL: http://192.168.10.152:8000
-```
-
-### 폴더 전체 리뷰
-
-```bash
-code-review folder /path/to/project
-```
-
-옵션:
-- `-i, --include <patterns>`: 포함할 파일 패턴 (기본: `*.py,*.js,*.ts,*.java,*.cpp,*.go`)
-- `-e, --exclude <patterns>`: 제외할 패턴 (기본: `node_modules,__pycache__,.git,*.pyc`)
-- `-m, --max-files <number>`: 최대 파일 수 (기본: 50)
-
-예시:
-```bash
-# Python 프로젝트 리뷰
-code-review folder ./my-python-project -i "*.py" -e "__pycache__,*.pyc,venv"
-
-# JavaScript 프로젝트 리뷰
-code-review folder ./my-js-project -i "*.js,*.jsx" -e "node_modules,dist,build"
-
-# 모든 코드 파일 리뷰 (최대 100개)
-code-review folder ./my-project -m 100
-```
-
-### 단일 파일 리뷰
-
-```bash
-code-review file /path/to/file.py
-```
-
-옵션:
-- `-l, --language <lang>`: 프로그래밍 언어 (기본: python)
-- `-c, --checklist <path>`: 체크리스트 파일 경로
-
-예시:
-```bash
-# Python 파일 리뷰
-code-review file ./app.py -l python
-
-# TypeScript 파일 + 체크리스트
-code-review file ./api.ts -l typescript -c ./checklist.txt
-
-# Java 파일 리뷰
-code-review file ./Main.java -l java
-```
-
-## VSCode 통합
-
-### 방법 1: Tasks 사용
-
-`.vscode/tasks.json` 파일 생성:
-
-```json
-{
-  "version": "2.0.0",
-  "tasks": [
-    {
-      "label": "Code Review: Current Folder",
-      "type": "shell",
-      "command": "code-review folder ${workspaceFolder}",
-      "problemMatcher": [],
-      "presentation": {
-        "reveal": "always",
-        "panel": "new"
-      }
-    },
-    {
-      "label": "Code Review: Current File",
-      "type": "shell",
-      "command": "code-review file ${file}",
-      "problemMatcher": [],
-      "presentation": {
-        "reveal": "always",
-        "panel": "new"
-      }
-    }
-  ]
-}
-```
-
-사용법:
-1. `Ctrl+Shift+P` → `Tasks: Run Task`
-2. `Code Review: Current Folder` 또는 `Code Review: Current File` 선택
-
-### 방법 2: 터미널에서 직접 사용
-
-VSCode 내장 터미널에서:
-
-```bash
-# 현재 워크스페이스 리뷰
-code-review folder .
-
-# 특정 파일 리뷰
-code-review file src/main.py
-```
-
-## 환경 변수
-
-서버 URL을 변경하려면 환경 변수 설정:
-
-```bash
-export CODE_REVIEW_SERVER=http://192.168.10.152:8000
-code-review health
-```
-
-## IP 화이트리스트
-
-현재 허용된 IP:
-- `192.168.10.152` (서버 IP)
-- `127.0.0.1` (로컬 테스트용)
-
-다른 IP에서 접근 시:
-```
-❌ 허용된 아이피가 아닙니다
-현재 IP: 192.168.10.xxx
-허용된 IP: 192.168.10.152, 127.0.0.1
-```
-
-IP 추가하려면 `vllm_server.py`의 `ALLOWED_IPS` 수정:
-
-```python
-ALLOWED_IPS = ["192.168.10.152", "127.0.0.1", "192.168.10.100"]  # 새 IP 추가
-```
-
-## 트러블슈팅
-
-### 서버 연결 실패
-
-```
-❌ 서버 연결 실패
-서버 연결 실패: http://192.168.10.152:8000
-서버가 실행 중인지 확인하세요.
-```
-
-해결:
-1. vLLM 서버가 실행 중인지 확인: `ps aux | grep vllm_server`
-2. 서버 로그 확인: `tail -f /path/to/server.log`
-3. 방화벽 확인: `sudo ufw status`
-
-### IP 차단
-
-```
-❌ 허용된 아이피가 아닙니다
-```
-
-해결:
-1. 현재 IP 확인: `curl ifconfig.me`
-2. `vllm_server.py`의 `ALLOWED_IPS`에 추가
-3. 서버 재시작
-
-### 메모리 부족
-
-vLLM 서버가 OOM으로 죽는 경우:
-
-`vllm_server.py` 수정:
-```python
-llm = LLM(
-    model=OUTPUT_DIR,
-    tensor_parallel_size=2,
-    gpu_memory_utilization=0.7,  # 0.9 → 0.7로 감소
-    max_model_len=1024,  # 2048 → 1024로 감소
-)
-```
-
-## 성능
-
-- 평균 응답 시간: ~5초 (폴더 리뷰, 10개 파일 기준)
-- 동시 처리: vLLM 배치 처리로 여러 요청 동시 처리 가능
-- GPU 사용량: 2x RTX 5090 (50-60GB)
-
-## 라이선스
-
-MIT

package/.ipynb_checkpoints/package-checkpoint.json

@@ -1,36 +0,0 @@
-{
-  "name": "cciwon-code-review-cli",
-  "version": "2.0.6",
-  "description": "AI-powered code review CLI tool using Qwen3-Coder-30B model with IP whitelist support",
-  "main": "lib/api-client.js",
-  "bin": {
-    "code-review": "bin/code-review.js"
-  },
-  "scripts": {
-    "test": "echo \"Error: no test specified\" && exit 1"
-  },
-  "keywords": [
-    "code-review",
-    "ai",
-    "qwen",
-    "qwen3-coder",
-    "code-analysis",
-    "security",
-    "llm",
-    "code-quality",
-    "static-analysis"
-  ],
-  "author": "cciwon",
-  "license": "MIT",
-  "dependencies": {
-    "axios": "^1.6.0",
-    "commander": "^11.1.0",
-    "chalk": "^4.1.2",
-    "ora": "^5.4.1",
-    "fast-glob": "^3.3.2",
-    "readline": "^1.3.0"
-  },
-  "engines": {
-    "node": ">=14.0.0"
-  }
-}