Merge pull request #1 from CodePapi/code-reviewer

ft: added reviewvand bug fixing features

Author: CodePapi

backend/src/converter/converter.controller.ts

@@ -6,9 +6,17 @@ export class ConverterController {
   constructor(private readonly converterService: ConverterService) {}
 
   @Post('translate')
-  async translate(
-    @Body() data: { code: string; from: string; to: string }
-  ) {
+  async translate(@Body() data: { code: string; from: string; to: string }) {
     return this.converterService.convertCode(data.code, data.from, data.to);
   }
+
+  @Post('review')
+  async review(@Body() data: { code: string; lang: string }) {
+    return this.converterService.reviewCode(data.code, data.lang);
+  }
+
+  @Post('fix')
+  async fix(@Body() data: { code: string; lang: string }) {
+    return this.converterService.fixBugs(data.code, data.lang);
+  }
 }

backend/src/converter/converter.service.ts

@@ -8,33 +8,76 @@ export class ConverterService {
   constructor() {
     this.model = new ChatOllama({
       baseUrl: process.env.OLLAMA_URL || 'http://localhost:11434',
-      model: 'qwen2.5-coder:1.5b', // Lightweight and efficient for code translation
+      model: 'qwen2.5-coder:1.5b',
     });
   }
 
-  async convertCode(code: string, from: string, to: string) {
+  // --- 1. CODE TRANSLATION ---
+async convertCode(code: string, from: string, to: string) {
+  // Check if "to" is a migration preset
+  const isMigration = to.includes('-') || from.includes('-'); 
+  
+  const prompt = `
+    You are an expert software architect specializing in ${isMigration ? 'code migration' : 'code translation'}.
+    Task: Convert the input from ${from} to ${to}.
+    
+    ${isMigration ? `
+    SPECIFIC INSTRUCTIONS FOR MIGRATION:
+    - If moving from Class to Functional components, use React Hooks (useState, useEffect).
+    - If moving to TypeScript, add proper interfaces and types.
+    - If moving between frameworks (e.g., React to Vue), map lifecycle methods and state management accurately.
+    ` : ''}
+
+    RULES:
+    - Return ONLY raw code.
+    - No markdown blocks.
+    - Maintain variable names and logic flow.
+
+    SOURCE CODE:
+    ${code}
+  `;
+    return this.executeInvoke(prompt, 'translatedCode');
+  }
+
+  // --- 2. CODE REVIEWER ---
+  async reviewCode(code: string, lang: string) {
     const prompt = `
-      You are a high-performance code translation engine.
-      Translate the following ${from} code into ${to}.
-      
+      You are a Senior Software Engineer. Review the following ${lang} code.
+      Check for: Performance, Security vulnerabilities, and Best Practices.
       RULES:
-      - Return ONLY the raw code. 
-      - Do not include markdown code blocks (\`\`\`), code fences, backticks that are not part of the original code.
-      - Do not include explanations or comments like "Here is your code" unless it was in the original code.
-      - Maintain logic and variable names exactly.
-      
-      SOURCE CODE:
+      - Provide a concise list of improvements.
+      - Use professional, constructive language.
+      - Use Markdown for formatting.
+      CODE:
       ${code}
     `;
+    return this.executeInvoke(prompt, 'reviewContent');
+  }
+
+  // --- 3. BUG FIXER ---
+  async fixBugs(code: string, lang: string) {
+    const prompt = `
+      You are a debugging expert. Analyze and fix the bugs in this ${lang} code.
+      RULES:
+      - Return the fixed code first.
+      - Follow the code with a brief "Explanation of Fixes" section.
+      - Do not use markdown fences for the code part.
+      CODE:
+      ${code}
+    `;
+    return this.executeInvoke(prompt, 'fixedCode');
+  }
 
+  // Private helper to avoid repeating try/catch blocks
+  private async executeInvoke(prompt: string, key: string) {
     try {
       const response = await this.model.invoke(prompt);
       return { 
         success: true, 
-        translatedCode: response.content 
+        [key]: response.content 
       };
     } catch (error) {
-        console.error('Error during code conversion:', error);
+      console.error(`AI Service Error:`, error);
       throw new InternalServerErrorException('Failed to connect to AI engine');
     }
   }