Merge pull request #38 from StructZ/develop

feat/fix: new feature and fixes added and updated from head of develop branch

Author: dineshsutihar

.github/ISSUE_TEMPLATE/bug_report.md

@@ -0,0 +1,38 @@
+---
+name: Bug report
+about: Create a report to help us improve
+title: "[BUG]"
+labels: bug, review
+assignees: ''
+
+---
+
+**Describe the bug**
+A clear and concise description of what the bug is.
+
+**To Reproduce**
+Steps to reproduce the behavior:
+1. Go to '...'
+2. Click on '....'
+3. Scroll down to '....'
+4. See error
+
+**Expected behavior**
+A clear and concise description of what you expected to happen.
+
+**Screenshots**
+If applicable, add screenshots to help explain your problem.
+
+**Desktop (please complete the following information):**
+ - OS: [e.g. iOS]
+ - Browser [e.g. chrome, safari]
+ - Version [e.g. 22]
+
+**Smartphone (please complete the following information):**
+ - Device: [e.g. iPhone6]
+ - OS: [e.g. iOS8.1]
+ - Browser [e.g. stock browser, safari]
+ - Version [e.g. 22]
+
+**Additional context**
+Add any other context about the problem here.

.github/ISSUE_TEMPLATE/feature_request.md

@@ -0,0 +1,20 @@
+---
+name: Feature request
+about: Suggest an idea for this project
+title: "[ISSUE]"
+labels: feature, review
+assignees: ''
+
+---
+
+**Is your feature request related to a problem? Please describe.**
+A clear and concise description of what the problem is. Ex. I'm always frustrated when [...]
+
+**Describe the solution you'd like**
+A clear and concise description of what you want to happen.
+
+**Describe alternatives you've considered**
+A clear and concise description of any alternative solutions or features you've considered.
+
+**Additional context**
+Add any other context or screenshots about the feature request here.

.github/workflows/welcome.yml

@@ -0,0 +1,26 @@
+name: Welcome First-Time Contributors
+
+on:
+  pull_request_target:
+    types: [opened]
+  issues:
+    types: [opened]
+
+jobs:
+  greet:
+    runs-on: ubuntu-latest
+    permissions:
+      pull-requests: write
+      issues: write
+    steps:
+      - uses: actions/first-interaction@v1
+        with:
+          repo-token: ${{ secrets.GITHUB_TOKEN }}
+          issue-message: |
+            Hello @${{ github.actor }}! 👋 
+            Thank you so much for taking the time to open your first issue with CodeTranslateAI. We appreciate you helping us make the project better! 
+            We'll take a look and get back to you soon.
+          pr-message: |
+            Hi @${{ github.actor }}! 🎉
+            Welcome, and thank you for submitting your first pull request to CodeTranslateAI! We're thrilled to have your contribution.
+            Please make sure you've read our CONTRIBUTING.md guide. We'll review your changes shortly.

backend/src/index.ts

@@ -1,54 +1,137 @@
 import { GoogleGenerativeAI } from '@google/generative-ai';
 
 export interface Env {
+	RATE_LIMIT: KVNamespace;
 	GEMINI_API_KEY: string;
 }
 
-export default {
-	async fetch(request: Request, env: Env, ctx: ExecutionContext): Promise<Response> {
-		const corsHeaders = {
-			'Access-Control-Allow-Origin': '*',
-			'Access-Control-Allow-Methods': 'POST, OPTIONS',
-			'Access-Control-Allow-Headers': 'Content-Type',
-		};
+const corsHeaders = {
+	'Access-Control-Allow-Origin': '*',
+	'Access-Control-Allow-Methods': 'POST, OPTIONS',
+	'Access-Control-Allow-Headers': 'Content-Type',
+};
 
-		if (request.method === 'OPTIONS') {
-			return new Response(null, { headers: corsHeaders });
-		}
+const MAX_REQUESTS_ALLOWED = 10;
+const DURATION = 60_000;
+
+async function checkRateLimit(ip: string, env: Env) {
+	const key = `ip_key:${ip}`;
+	const now = Date.now();
+	let value = await env.RATE_LIMIT.get(key);
+	let data = { count: 0, time: now };
 
+	if (value) {
 		try {
-			const { code, targetLanguage } = await request.json<{ code: string; targetLanguage: string }>();
+			data = JSON.parse(value);
+		} catch {
+			data = { count: 0, time: now };
+		}
+	}
 
-			if (!code || !targetLanguage) {
-				return new Response(JSON.stringify({ error: "Missing 'code' or 'targetLanguage' in request body." }), {
-					status: 400,
-					headers: { ...corsHeaders, 'Content-Type': 'application/json' },
-				});
-			}
+	if (now - data.time > DURATION) {
+		data.count = 0;
+		data.time = now;
+	}
 
-			const genAI = new GoogleGenerativeAI(env.GEMINI_API_KEY);
+	data.count += 1;
+	await env.RATE_LIMIT.put(key, JSON.stringify(data), { expirationTtl: 65 });
 
-			const model = genAI.getGenerativeModel({ model: 'gemini-2.0-flash' });
+	return data.count <= MAX_REQUESTS_ALLOWED;
+}
+async function handleTranslate(request: Request, model: ReturnType<GoogleGenerativeAI['getGenerativeModel']>) {
+	const { code, targetLanguage } = await request.json<{ code: string; targetLanguage: string }>();
+
+	if (!code || !targetLanguage) {
+		return new Response(JSON.stringify({ error: "Missing 'code' or 'targetLanguage' in request body." }), {
+			status: 400,
+			headers: { ...corsHeaders, 'Content-Type': 'application/json' },
+		});
+	}
 
-			const prompt = `Translate the following code snippet to ${targetLanguage}.
+	const prompt = `Translate the following code snippet to ${targetLanguage}.
 Do not add any explanation, commentary, or markdown formatting like \`\`\` around the code.
-**IMPORTANT: Preserve all original comments and their exact placement in the translated code. do not add extra spaces in between.**
+**IMPORTANT: Preserve all original comments and their exact placement in the translated code. Do not add extra spaces in between.**
 Only provide the raw, translated code itself.
 
 Original Code:
 ${code}`;
 
-			const result = await model.generateContent(prompt);
-			const geminiResponse = result.response;
-			const translatedCode = geminiResponse.text();
+	const result = await model.generateContent(prompt);
+	const translatedCode = result.response.text();
+
+	return new Response(JSON.stringify({ translation: translatedCode }), {
+		status: 200,
+		headers: { ...corsHeaders, 'Content-Type': 'application/json' },
+	});
+}
+
+async function handleExplain(request: Request, model: ReturnType<GoogleGenerativeAI['getGenerativeModel']>) {
+	const { code } = await request.json<{ code: string }>();
+
+	if (!code) {
+		return new Response(JSON.stringify({ error: "Missing 'code' in request body." }), {
+			status: 400,
+			headers: { ...corsHeaders, 'Content-Type': 'application/json' },
+		});
+	}
+
+	const prompt = `Explain the following code snippet in detail:
+1. Provide a clear breakdown of what each part (functions, variables, logic blocks) does.
+2. If applicable, describe the overall purpose or intent of the code.
+3. Offer a step-by-step explanation of how the code executes.
+4. If the code is executable, show a sample input and the corresponding output.
+5. Keep the explanation beginner-friendly but technically accurate.
+
+Code:
+${code}`;
+
+	const result = await model.generateContent(prompt);
+	const explanation = result.response.text();
+
+	return new Response(JSON.stringify({ explanation }), {
+		status: 200,
+		headers: { ...corsHeaders, 'Content-Type': 'application/json' },
+	});
+}
+
+export default {
+	async fetch(request: Request, env: Env, ctx: ExecutionContext): Promise<Response> {
+		if (request.method === 'OPTIONS') {
+			return new Response(null, { headers: corsHeaders });
+		}
+
+		try {
+			const ip = request.headers.get('CF-Connecting-IP') || 'unknown';
+      		const allowed = await checkRateLimit(ip, env);
+      		if (!allowed) {
+        		return new Response(JSON.stringify({ error: "Too many requests. Try again later." }), {
+          		status: 429,
+          		headers: { ...corsHeaders, 'Content-Type': 'application/json' },
+        	});
+      		}
+			const url = new URL(request.url);
+			const path = url.pathname;
+			const genAI = new GoogleGenerativeAI(env.GEMINI_API_KEY);
+			const model = genAI.getGenerativeModel({ model: 'gemini-2.0-flash' });
+
+			if(path==="/test-rate-limit"){
+				return new Response(JSON.stringify("Proceed !"))
+			}
+			if (path === '/' || path === '/v1/translate') {
+				return await handleTranslate(request, model);
+			}
+
+			if (path === '/v1/explain') {
+				return await handleExplain(request, model);
+			}
 
-			return new Response(JSON.stringify({ translation: translatedCode }), {
-				status: 200,
+			return new Response(JSON.stringify({ error: 'Route not found.' }), {
+				status: 404,
 				headers: { ...corsHeaders, 'Content-Type': 'application/json' },
 			});
 		} catch (error) {
-			console.error('Error during translation:', error);
-			return new Response(JSON.stringify({ error: 'An error occurred while translating the code.' }), {
+			console.error('Error during request:', error);
+			return new Response(JSON.stringify({ error: 'An internal error occurred.' }), {
 				status: 500,
 				headers: { ...corsHeaders, 'Content-Type': 'application/json' },
 			});

backend/worker-configuration.d.ts

@@ -3,6 +3,7 @@
 // Runtime types generated with workerd@1.20250712.0 2025-07-15 
 declare namespace Cloudflare {
 	interface Env {
+		RATE_LIMIT: KVNamespace;
 	}
 }
 interface Env extends Cloudflare.Env {}

backend/wrangler.jsonc

@@ -9,7 +9,14 @@
 	"compatibility_date": "2025-07-15",
 	"observability": {
 		"enabled": true
-	}
+	},
+	"kv_namespaces": [
+		{
+			"binding": "RATE_LIMIT",
+			"id": "<your_kv_id>"
+		}
+	]
+
 	/**
 	 * Smart Placement
 	 * Docs: https://developers.cloudflare.com/workers/configuration/smart-placement/#smart-placement

frontend/.husky/pre-commit

@@ -0,0 +1,4 @@
+#!/usr/bin/env sh
+. "$(dirname -- "$0")/_/husky.sh"
+
+npx lint-staged

frontend/.prettierignore

@@ -0,0 +1,4 @@
+dist
+node_modules
+packages
+package-lock.json

frontend/.prettierrc.json

@@ -0,0 +1,5 @@
+{
+  "semi": true,
+  "singleQuote": false,
+  "trailingComma": "es5"
+}

frontend/background.js

@@ -1,41 +1,52 @@
-
 chrome.runtime.onMessage.addListener((request, _, sendResponse) => {
-    if (request.type === "TRANSLATE_CODE") {
-        const BACKEND_URL = process.env.BACKEND_URL;
-
-        chrome.storage.sync.get(['targetLanguage'], (result) => {
-            const targetLanguage = result.targetLanguage || 'Java';
-            fetch(BACKEND_URL, {
-                method: 'POST',
-                headers: {
-                    'Content-Type': 'application/json',
-                },
-                body: JSON.stringify({
-                    code: request.code,
-                    targetLanguage: targetLanguage,
-                }),
-            })
-                .then(response => {
-                    if (!response.ok) {
-                        throw new Error(`Network response was not ok: ${response.statusText}`);
-                    }
-                    return response.json();
-                })
-                .then(data => {
-
-                    if (data.error) {
-                        sendResponse({ error: data.error });
-                    } else {
-                        sendResponse({ translation: data.translation });
-                    }
-                })
-                .catch(error => {
+  if (request.type === "TRANSLATE_CODE") {
+    const BACKEND_URL = process.env.BACKEND_URL;
 
-                    console.error("Error calling backend:", error);
-                    sendResponse({ error: `Failed to connect to the translation service: ${error.message}` });
-                });
+    chrome.storage.sync.get(["targetLanguage"], (result) => {
+      const targetLanguage = result.targetLanguage || "Java";
+      fetch(BACKEND_URL, {
+        method: "POST",
+        headers: {
+          "Content-Type": "application/json",
+        },
+        body: JSON.stringify({
+          code: request.code,
+          targetLanguage: targetLanguage,
+        }),
+      })
+        .then((response) => {
+          if (!response.ok) {
+            throw new Error(
+              `Network response was not ok: ${response.statusText}`
+            );
+          }
+          return response.json();
+        })
+        .then((data) => {
+          if (data.error) {
+            sendResponse({ error: data.error });
+          } else {
+            sendResponse({ translation: data.translation });
+          }
+        })
+        .catch((error) => {
+          console.error("Error calling backend:", error);
+          sendResponse({
+            error: `Failed to connect to the translation service: ${error.message}`,
+          });
         });
+    });
 
-        return true;
-    }
-});
+    return true;
+  }
+});
+//Default commmand = Alt+T , Mac  = Option+T
+chrome.commands.onCommand.addListener((command) => {
+  if (command === "translate") {
+    chrome.tabs.query({ active: true, currentWindow: true }, (tabs) => {
+      if (tabs.length > 0) {
+        chrome.tabs.sendMessage(tabs[0].id, { type: "ENABLE_PICKER" });
+      }
+    });
+  }
+});