codereviewerai 1.0.0

package/dist/ui/home.js

@@ -0,0 +1,54 @@
+import chalk from 'chalk';
+import figlet from 'figlet';
+import gradient from 'gradient-string';
+export const homePage = {
+    /**
+     * Displays the high-tech matrix-style header
+     */
+    displayWelcome: async () => {
+        // Ensure the terminal is clear for the professional 'splash' effect
+        console.clear();
+        // Fix: Store the color in a constant for cleaner reuse
+        const matrixGreen = chalk.hex('#00FF41');
+        // Your specific ASCII Art - Wrapped in a raw string to protect backslashes
+        console.log(matrixGreen.bold(`
+    ██████╗ ██████╗ ██████╗ ███████╗██████╗ ███████╗██╗   ██╗██╗███████╗██╗    ██╗███████╗██████╗ 
+    ██╔════╝██╔═══██╗██╔══██╗██╔════╝██╔══██╗██╔════╝██║   ██║██║██╔════╝██║    ██║██╔════╝██╔══██╗
+    ██║     ██║   ██║██║  ██║█████╗  ██████╔╝█████╗  ██║   ██║██║█████╗  ██║ █╗ ██║█████╗  ██████╔╝
+    ██║     ██║   ██║██║  ██║██╔══╝  ██╔══██╗██╔══╝  ╚██╗ ██╔╝██║██╔══╝  ██║███╗██║██╔══╝  ██╔══██╗
+    ╚██████╗╚██████╔╝██████╔╝███████╗██║  ██║███████╗ ╚████╔╝ ██║███████╗╚███╔███╔╝███████╗██║  ██║
+     ╚═════╝ ╚═════╝ ╚═════╝ ╚══════╝╚═╝  ╚═╝╚══════╝  ╚═══╝  ╚═╝╚══════╝ ╚══╝╚══╝ ╚══════╝╚═╝  ╚═╝.ai
+        `));
+        // Adding a subtle subtitle using gradient
+        try {
+            const subHeader = figlet.textSync('CODEREVIEWER', {
+                font: 'Small',
+                horizontalLayout: 'default',
+                verticalLayout: 'default',
+            });
+            console.log(gradient(['#00FF41', '#008F11']).multiline(subHeader));
+        }
+        catch (err) {
+            // Fallback if figlet fails
+            console.log(matrixGreen.bold('      --- CODEREVIEWER.AI ---      '));
+        }
+        console.log(chalk.gray('————————————————————————————————————————————————————————————————————'));
+        console.log(chalk.white('  Advanced AI Analysis • Performance Optimization • Security Audit'));
+        console.log(chalk.gray('————————————————————————————————————————————————————————————————————\n'));
+    },
+    /**
+     * Shows a clean list of available commands
+     */
+    displayQuickHelp: () => {
+        console.log(chalk.bold.white('📂 GETTING STARTED'));
+        console.log(`  ${chalk.green('awd init')}        ${chalk.gray('→ Setup your API keys and provider')}`);
+        console.log(`  ${chalk.green('awd review')}      ${chalk.gray('→ Review staged changes in Git')}`);
+        console.log(`  ${chalk.green('awd watch')}       ${chalk.gray('→ Enable real-time auto-review mode')}`);
+        console.log(`  ${chalk.green('awd chat')}        ${chalk.gray('→ Discuss results with the AI assistant')}\n`);
+        console.log(chalk.bold.white('⚙️  SYSTEM'));
+        console.log(`  ${chalk.green('awd status')}      ${chalk.gray('→ Show current configuration')}`);
+        console.log(`  ${chalk.green('awd history')}     ${chalk.gray('→ View previous review scores')}\n`);
+        console.log(chalk.hex('#00FF41')('Ready for input...'));
+    }
+};
+//# sourceMappingURL=home.js.map

package/dist/ui/home.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"home.js","sourceRoot":"","sources":["../../src/ui/home.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,QAAQ,MAAM,iBAAiB,CAAC;AAEvC,MAAM,CAAC,MAAM,QAAQ,GAAG;IACpB;;OAEG;IACH,cAAc,EAAE,KAAK,IAAI,EAAE;QACvB,oEAAoE;QACpE,OAAO,CAAC,KAAK,EAAE,CAAC;QAEhB,uDAAuD;QACvD,MAAM,WAAW,GAAG,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAEzC,2EAA2E;QAC3E,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC;;;;;;;SAO5B,CAAC,CAAC,CAAC;QAEJ,0CAA0C;QAC1C,IAAI,CAAC;YACD,MAAM,SAAS,GAAG,MAAM,CAAC,QAAQ,CAAC,cAAc,EAAE;gBAC9C,IAAI,EAAE,OAAO;gBACb,gBAAgB,EAAE,SAAS;gBAC3B,cAAc,EAAE,SAAS;aAC5B,CAAC,CAAC;YACH,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC;QACvE,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACX,2BAA2B;YAC3B,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAC,CAAC;QACzE,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,sEAAsE,CAAC,CAAC,CAAC;QAChG,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,oEAAoE,CAAC,CAAC,CAAC;QAC/F,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,wEAAwE,CAAC,CAAC,CAAC;IACtG,CAAC;IAED;;OAEG;IACH,gBAAgB,EAAE,GAAG,EAAE;QACnB,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC,CAAC;QACpD,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC,WAAW,KAAK,CAAC,IAAI,CAAC,oCAAoC,CAAC,EAAE,CAAC,CAAC;QACvG,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,SAAS,KAAK,CAAC,IAAI,CAAC,gCAAgC,CAAC,EAAE,CAAC,CAAC;QACnG,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC,UAAU,KAAK,CAAC,IAAI,CAAC,qCAAqC,CAAC,EAAE,CAAC,CAAC;QACxG,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC,WAAW,KAAK,CAAC,IAAI,CAAC,yCAAyC,CAAC,IAAI,CAAC,CAAC;QAE9G,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC;QAC5C,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,SAAS,KAAK,CAAC,IAAI,CAAC,8BAA8B,CAAC,EAAE,CAAC,CAAC;QACjG,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,KAAK,CAAC,aAAa,CAAC,QAAQ,KAAK,CAAC,IAAI,CAAC,+BAA+B,CAAC,IAAI,CAAC,CAAC;QAEpG,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC;IAC5D,CAAC;CACJ,CAAC"}

package/package.json

@@ -0,0 +1,52 @@
+{
+  "name": "codereviewerai",
+  "version": "1.0.0",
+  "description": "codereviewer.ai - AI-powered code review tool for developers",
+  "type": "module",
+  "main": "dist/index.js",
+  "bin": {
+    "awd": "./bin/awesomediagns.js",
+    "awesomediagns": "./bin/awesomediagns.js"
+  },
+  "scripts": {
+    "build": "tsc",
+    "dev": "ts-node src/index.ts",
+    "start": "node dist/index.js",
+    "watch": "tsc --watch",
+    "prepublishOnly": "npm run build"
+  },
+  "keywords": [
+    "code-review",
+    "ai",
+    "cli",
+    "productivity",
+    "git",
+    "llm",
+    "automation"
+  ],
+  "author": "CodeReviewer.ai Team",
+  "license": "MIT",
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "dependencies": {
+    "axios": "^1.6.2",
+    "boxen": "^8.0.1",
+    "chalk": "^5.3.0",
+    "chokidar": "^4.0.1",
+    "commander": "^12.1.0",
+    "conf": "^13.0.2",
+    "dotenv": "^16.4.5",
+    "figlet": "^1.8.0",
+    "gradient-string": "^3.0.0",
+    "inquirer": "^12.0.0",
+    "ora": "^8.1.0",
+    "simple-git": "^3.27.0"
+  },
+  "devDependencies": {
+    "@types/figlet": "^1.7.0",
+    "@types/node": "^22.0.0",
+    "ts-node": "^10.9.2",
+    "typescript": "^5.6.0"
+  }
+}

package/src/ai/provider.ts

@@ -0,0 +1,170 @@
+import axios from 'axios';
+import chalk from 'chalk';
+import { ChatMessage } from '../core/context.js';
+
+export class AIProvider {
+    private config: any;
+
+    constructor(configManager: any) {
+        this.config = configManager.getConfig();
+    }
+
+    /**
+     * The main function to send code to the AI
+     */
+    async reviewCode(code: string, filePath: string, context?: any): Promise<any> {
+        const prompt = this.buildReviewPrompt(code, filePath, context);
+
+        try {
+            switch (this.config.provider) {
+                case 'gemini':
+                    return await this.reviewWithGemini(prompt);
+                case 'openai':
+                    return await this.reviewWithOpenAI(prompt);
+                case 'claude':
+                    return await this.reviewWithClaude(prompt);
+                default:
+                    throw new Error('No AI provider configured. Run "awd init"');
+            }
+        } catch (error: any) {
+            throw new Error(`AI Request Failed: ${error.response?.data?.error?.message || error.message}`);
+        }
+    }
+
+    /**
+     * Chat with the AI using conversation history
+     */
+    async chat(message: string, history: ChatMessage[]): Promise<string> {
+        try {
+            switch (this.config.provider) {
+                case 'gemini':
+                    return await this.chatWithGemini(message, history);
+                case 'openai':
+                    return await this.chatWithOpenAI(message, history);
+                case 'claude':
+                    return await this.chatWithClaude(message, history);
+                default:
+                    throw new Error('No AI provider configured. Run "awd init"');
+            }
+        } catch (error: any) {
+            throw new Error(`AI Chat Failed: ${error.response?.data?.error?.message || error.message}`);
+        }
+    }
+
+    private buildReviewPrompt(code: string, filePath: string, context?: any): string {
+        return `
+        You are "codereviewer.ai", an expert senior developer. 
+        Review the following code for file: ${filePath}
+
+        CRITERIA:
+        1. Identify bugs or security flaws.
+        2. Suggest performance optimizations.
+        3. Check for best practices and readability.
+
+        CONTEXT FROM PREVIOUS CHATS:
+        ${JSON.stringify(context || "None")}
+
+        CODE TO REVIEW:
+        \`\`\`
+        ${code}
+        \`\`\`
+
+        RESPONSE FORMAT:
+        You MUST respond ONLY with a valid JSON object. Do not include markdown formatting or prose.
+        {
+          "summary": "Brief overall thought",
+          "score": 1-10,
+          "issues": [{"line": number, "type": "bug|style|security", "msg": "description", "fix": "suggested code"}],
+          "optimizations": ["list of tips"]
+        }
+        `;
+    }
+
+    private async reviewWithGemini(prompt: string) {
+        const url = `https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key=${this.config.apiKey}`;
+        const response = await axios.post(url, {
+            contents: [{ parts: [{ text: prompt }] }]
+        });
+        const text = response.data.candidates[0].content.parts[0].text;
+        return JSON.parse(this.cleanJSON(text));
+    }
+
+    private async reviewWithOpenAI(prompt: string) {
+        const response = await axios.post('https://api.openai.com/v1/chat/completions', {
+            model: 'gpt-4-turbo-preview',
+            messages: [{ role: 'user', content: prompt }],
+            response_format: { type: "json_object" }
+        }, {
+            headers: { Authorization: `Bearer ${this.config.apiKey}` }
+        });
+        return JSON.parse(response.data.choices[0].message.content);
+    }
+
+    private async reviewWithClaude(prompt: string) {
+        // Logic for Claude API (Anthropic)
+        const response = await axios.post('https://api.anthropic.com/v1/messages', {
+            model: 'claude-3-sonnet-20240229',
+            max_tokens: 1024,
+            messages: [{ role: 'user', content: prompt }]
+        }, {
+            headers: { 
+                'x-api-key': this.config.apiKey,
+                'anthropic-version': '2023-06-01'
+            }
+        });
+        return JSON.parse(this.cleanJSON(response.data.content[0].text));
+    }
+
+    private async chatWithGemini(message: string, history: ChatMessage[]): Promise<string> {
+        const prompt = this.buildChatPrompt(message, history);
+        const url = `https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key=${this.config.apiKey}`;
+        const response = await axios.post(url, {
+            contents: [{ parts: [{ text: prompt }] }]
+        });
+        return response.data.candidates[0].content.parts[0].text;
+    }
+
+    private async chatWithOpenAI(message: string, history: ChatMessage[]): Promise<string> {
+        const messages = history.map(h => ({ role: h.role, content: h.content }));
+        messages.push({ role: 'user', content: message });
+        const response = await axios.post('https://api.openai.com/v1/chat/completions', {
+            model: 'gpt-4-turbo-preview',
+            messages: messages
+        }, {
+            headers: { Authorization: `Bearer ${this.config.apiKey}` }
+        });
+        return response.data.choices[0].message.content;
+    }
+
+    private async chatWithClaude(message: string, history: ChatMessage[]): Promise<string> {
+        const messages = history.map(h => ({ role: h.role, content: h.content }));
+        messages.push({ role: 'user', content: message });
+        const response = await axios.post('https://api.anthropic.com/v1/messages', {
+            model: 'claude-3-sonnet-20240229',
+            max_tokens: 1024,
+            messages: messages
+        }, {
+            headers: { 
+                'x-api-key': this.config.apiKey,
+                'anthropic-version': '2023-06-01'
+            }
+        });
+        return response.data.content[0].text;
+    }
+
+    private buildChatPrompt(message: string, history: ChatMessage[]): string {
+        let prompt = "You are codereviewer.ai, an expert developer assistant. Help with code-related questions.\n\n";
+        for (const msg of history) {
+            prompt += `${msg.role}: ${msg.content}\n`;
+        }
+        prompt += `user: ${message}\nassistant:`;
+        return prompt;
+    }
+
+    /**
+     * Removes Markdown code blocks if the AI accidentally includes them
+     */
+    private cleanJSON(text: string): string {
+        return text.replace(/```json|```/g, "").trim();
+    }
+}

package/src/config/manager.ts

@@ -0,0 +1,120 @@
+import Conf from 'conf';
+import chalk from 'chalk';
+import path from 'path';
+import fs from 'fs';
+
+// 1. Defined a strict schema for the config to prevent type errors
+export interface AppConfig {
+    provider: 'openai' | 'gemini' | 'claude';
+    apiKey: string;
+    model: string;
+    reviewDepth: 'quick' | 'deep';
+    mode: 'manual' | 'auto';
+    includeContext: boolean;
+    maxContextMessages: number;
+    projectPath?: string;
+}
+
+export class ConfigManager {
+    // We cast the Conf instance to allow better flexibility with the 'conf' library versions
+    private config: Conf<AppConfig>;
+
+    constructor() {
+        this.config = new Conf<AppConfig>({
+            projectName: 'codereviewerai',
+            // Default settings if the user hasn't run 'init'
+            defaults: {
+                provider: 'gemini',
+                apiKey: '',
+                model: 'gemini-1.5-pro',
+                reviewDepth: 'deep',
+                mode: 'manual',
+                includeContext: true,
+                maxContextMessages: 10
+            }
+        });
+    }
+
+    isConfigured(): boolean {
+        const key = this.config.get('apiKey');
+        return typeof key === 'string' && key.length > 0;
+    }
+
+    // Fixed: Explicitly return AppConfig to ensure UI elements can read keys safely
+    getConfig(): AppConfig {
+        return this.config.store as AppConfig;
+    }
+
+    // Fixed: Standardized update method
+    setKey<K extends keyof AppConfig>(key: K, value: AppConfig[K]): void {
+        this.config.set(key, value);
+    }
+
+    // Helper for saving multiple answers from Inquirer at once
+    setFullConfig(answers: Partial<AppConfig>): void {
+        for (const [key, value] of Object.entries(answers)) {
+            this.config.set(key as keyof AppConfig, value);
+        }
+    }
+
+    displayConfig(): void {
+        const store = this.getConfig();
+        console.log(chalk.cyan.bold('\n🛠️  Current Configuration:'));
+        
+        console.log(`
+  ${chalk.yellow('Provider:')} ${store.provider}
+  ${chalk.yellow('Model:')}    ${store.model}
+  ${chalk.yellow('Mode:')}     ${store.mode}
+  ${chalk.yellow('Depth:')}    ${store.reviewDepth}
+  ${chalk.yellow('API Key:')}  ${store.apiKey ? '********' + store.apiKey.slice(-4) : chalk.red('Not Set')}
+        `);
+    }
+
+    /**
+     * LOCAL PROJECT LOGIC
+     * This creates the .awesomediagns folder in the CURRENT working directory
+     */
+    initProject(projectPath: string): void {
+        this.setKey('projectPath', projectPath);
+        
+        const localDir = path.join(projectPath, '.awesomediagns');
+        const historyDir = path.join(localDir, 'history');
+
+        if (!fs.existsSync(localDir)) {
+            fs.mkdirSync(localDir, { recursive: true });
+            fs.mkdirSync(historyDir, { recursive: true });
+            
+            // Auto-ignore the local metadata from Git
+            const gitignore = path.join(projectPath, '.gitignore');
+            const ignoreEntry = '\n# AwesomeDiagns metadata\n.awesomediagns/\n';
+            
+            if (fs.existsSync(gitignore)) {
+                const content = fs.readFileSync(gitignore, 'utf8');
+                if (!content.includes('.awesomediagns')) {
+                    fs.appendFileSync(gitignore, ignoreEntry);
+                }
+            } else {
+                fs.writeFileSync(gitignore, ignoreEntry);
+            }
+            console.log(chalk.green('✅ Project initialized with local context folder.'));
+        }
+    }
+
+    // Check if the current folder is an AWD project
+    isProjectInitialized(currentPath: string = process.cwd()): boolean {
+        return fs.existsSync(path.join(currentPath, '.awesomediagns'));
+    }
+
+    // Useful for a fallback to find files if needed
+    getProjectConfig() {
+        return {
+            watchedFiles: ['*.ts', '*.js', '*.jsx', '*.tsx', '*.py', '*.go'],
+            ignoredPatterns: ['node_modules', 'dist', '.git']
+        };
+    }
+
+    clearConfig(): void {
+        this.config.clear();
+        console.log(chalk.red('🗑️  Global configuration wiped.'));
+    }
+}

package/src/config/wizard.ts

@@ -0,0 +1,93 @@
+import inquirer from 'inquirer';
+import chalk from 'chalk';
+import { ConfigManager, AppConfig } from './manager.js';
+
+export class ConfigWizard {
+    private manager: ConfigManager;
+
+    constructor() {
+        this.manager = new ConfigManager();
+    }
+
+    /**
+     * The main setup flow for a new user
+     */
+    async runSetup(): Promise<void> {
+        console.log(chalk.cyan.bold('\n✨ codereviewer.ai Setup Wizard ✨'));
+        console.log(chalk.gray('Let\'s configure your AI preferences.\n'));
+
+        const answers = await inquirer.prompt([
+            {
+                type: 'list',
+                name: 'provider',
+                message: 'Select your preferred AI Provider:',
+                choices: [
+                    { name: 'Google Gemini (Fast & Large Context)', value: 'gemini' },
+                    { name: 'OpenAI (GPT-4o)', value: 'openai' },
+                    { name: 'Anthropic (Claude 3.5 Sonnet)', value: 'claude' }
+                ]
+            },
+            {
+                type: 'password',
+                name: 'apiKey',
+                message: 'Enter your API Key:',
+                mask: '*',
+                validate: (input: string) => input.length > 0 || 'API Key is required.'
+            },
+            {
+                type: 'list',
+                name: 'reviewDepth',
+                message: 'Default Review Depth:',
+                choices: [
+                    { name: 'Quick (Focus on critical bugs)', value: 'quick' },
+                    { name: 'Deep (Full architectural analysis)', value: 'deep' }
+                ]
+            },
+            {
+                type: 'list',
+                name: 'mode',
+                message: 'Operation Mode:',
+                choices: [
+                    { name: 'Manual (Review when I run the command)', value: 'manual' },
+                    { name: 'Auto (Review every time I save a file)', value: 'auto' }
+                ]
+            }
+        ]);
+
+        // Map models based on the provider chosen
+        let defaultModel = 'gemini-1.5-pro';
+        if (answers.provider === 'openai') defaultModel = 'gpt-4o';
+        if (answers.provider === 'claude') defaultModel = 'claude-3-5-sonnet-20240620';
+
+        // Save everything to the manager
+        this.manager.setFullConfig({
+            ...answers,
+            model: defaultModel,
+            includeContext: true,
+            maxContextMessages: 10
+        });
+
+        console.log(chalk.green.bold('\n✅ Configuration successful!'));
+        console.log(chalk.gray('You can now run ') + chalk.white.bold('awd review') + chalk.gray(' to start your first review.\n'));
+    }
+
+    /**
+     * Allows reconfiguring a specific setting without a full reset
+     */
+    async reconfigure(type: 'provider' | 'mode'): Promise<void> {
+        if (type === 'provider') {
+            const { provider, apiKey } = await inquirer.prompt([
+                { type: 'list', name: 'provider', message: 'New Provider:', choices: ['gemini', 'openai', 'claude'] },
+                { type: 'password', name: 'apiKey', message: 'New API Key:' }
+            ]);
+            this.manager.setKey('provider', provider);
+            this.manager.setKey('apiKey', apiKey);
+        } else {
+            const { mode } = await inquirer.prompt([
+                { type: 'list', name: 'mode', message: 'Switch Mode:', choices: ['manual', 'auto'] }
+            ]);
+            this.manager.setKey('mode', mode);
+        }
+        console.log(chalk.green(`✅ ${type} updated successfully.`));
+    }
+}

package/src/core/context.ts

@@ -0,0 +1,139 @@
+import fs from 'fs';
+import path from 'path';
+import chalk from 'chalk';
+
+export interface ReviewEntry {
+    file: string;
+    timestamp: string;
+    code: string;
+    review: any;
+    score: number;
+}
+
+export interface ChatMessage {
+    role: 'user' | 'assistant' | 'system';
+    content: string;
+}
+
+export class ContextManager {
+    private contextDir: string;
+    private historyFile: string;
+    private chatFile: string;
+
+    constructor() {
+        // This looks for the local project folder created by ConfigManager
+        this.contextDir = path.join(process.cwd(), '.awesomediagns');
+        this.historyFile = path.join(this.contextDir, 'history', 'reviews.json');
+        this.chatFile = path.join(this.contextDir, 'history', 'chat.json');
+        this.ensureDirectories();
+    }
+
+    /**
+     * Ensures the local .awesomediagns folder exists before writing
+     */
+    private ensureDirectories(): void {
+        const historyDir = path.join(this.contextDir, 'history');
+        
+        if (!fs.existsSync(historyDir)) {
+            try {
+                fs.mkdirSync(historyDir, { recursive: true });
+            } catch (err) {
+                // If not in a project, we don't crash, but we can't save context
+                return;
+            }
+        }
+
+        if (!fs.existsSync(this.historyFile)) {
+            fs.writeFileSync(this.historyFile, JSON.stringify([], null, 2));
+        }
+        if (!fs.existsSync(this.chatFile)) {
+            fs.writeFileSync(this.chatFile, JSON.stringify([], null, 2));
+        }
+    }
+
+    /**
+     * Gets previous review data for a specific file to give AI "memory"
+     */
+    async getContext(filePath: string): Promise<any> {
+        try {
+            const history = this.loadHistory();
+            const fileHistory = history
+                .filter((entry: ReviewEntry) => entry.file === filePath)
+                .slice(-3); // Last 3 reviews for this specific file
+
+            return {
+                previousReviews: fileHistory.map((entry: ReviewEntry) => ({
+                    timestamp: entry.timestamp,
+                    score: entry.score,
+                    summary: entry.review.summary
+                }))
+            };
+        } catch (error) {
+            return { previousReviews: [] };
+        }
+    }
+
+    /**
+     * Saves a new review into the local project history
+     */
+    async addToContext(filePath: string, code: string, review: any): Promise<void> {
+        try {
+            const history = this.loadHistory();
+            const entry: ReviewEntry = {
+                file: filePath,
+                timestamp: new Date().toISOString(),
+                code: code.length > 500 ? code.substring(0, 500) + '...' : code,
+                review,
+                score: review.score || 0
+            };
+
+            history.push(entry);
+            // Limit to last 50 reviews to keep file size small
+            const limitedHistory = history.slice(-50);
+            fs.writeFileSync(this.historyFile, JSON.stringify(limitedHistory, null, 2));
+        } catch (error) {
+            console.error(chalk.red('Failed to save review context locally.'));
+        }
+    }
+
+    /**
+     * Conversation History for the AI Chat mode
+     */
+    async getChatHistory(): Promise<ChatMessage[]> {
+        if (!fs.existsSync(this.chatFile)) return [];
+        try {
+            const data = fs.readFileSync(this.chatFile, 'utf-8');
+            return JSON.parse(data);
+        } catch {
+            return [];
+        }
+    }
+
+    async saveChatHistory(messages: ChatMessage[]): Promise<void> {
+        try {
+            fs.writeFileSync(this.chatFile, JSON.stringify(messages, null, 2));
+        } catch (error) {
+            console.error(chalk.red('Failed to save chat history.'));
+        }
+    }
+
+    async getHistory(limit: number = 10): Promise<ReviewEntry[]> {
+        const history = this.loadHistory();
+        return history.slice(-limit).reverse();
+    }
+
+    async clearHistory(): Promise<void> {
+        fs.writeFileSync(this.historyFile, JSON.stringify([], null, 2));
+        fs.writeFileSync(this.chatFile, JSON.stringify([], null, 2));
+    }
+
+    private loadHistory(): ReviewEntry[] {
+        try {
+            if (!fs.existsSync(this.historyFile)) return [];
+            const data = fs.readFileSync(this.historyFile, 'utf-8');
+            return JSON.parse(data);
+        } catch (error) {
+            return [];
+        }
+    }
+}