ai-codegen-cli-vrk 1.0.0

package/bin/index.js

@@ -0,0 +1,2 @@
+#!/usr/bin/env node
+import "../src/runner.js";

package/package.json

@@ -0,0 +1,18 @@
+{
+  "name": "ai-codegen-cli-vrk",
+  "version": "1.0.0",
+  "description": "Terminal-based AI code generator",
+  "type": "module",
+  "bin": {
+    "ai-codegen": "bin/index.js"
+  },
+  "files": [
+    "bin",
+    "src"
+  ],
+  "dependencies": {
+    "fs-extra": "^11.2.0",
+    "groq-sdk": "^0.7.0",
+    "readline-sync": "^1.4.10"
+  }
+}

package/src/aiClient.js

@@ -0,0 +1,132 @@
+import Groq from "groq-sdk";
+
+let groqClient = null;
+
+/**
+ * Called once per run from runner.js
+ */
+export function setApiKey(apiKey) {
+  groqClient = new Groq({ apiKey });
+}
+
+/**
+ * Internal helper to call Groq
+ */
+async function call(messages) {
+  if (!groqClient) {
+    throw new Error("API key not initialized");
+  }
+
+  return groqClient.chat.completions.create({
+    model: "llama-3.1-8b-instant",
+    temperature: 0,
+    response_format: { type: "json_object" },
+    messages
+  });
+}
+
+/* ================= PHASE 1: FILE PLAN ================= */
+export async function generateFilePlan(task, tests) {
+  const res = await call([
+    {
+      role: "system",
+      content: `
+Return ONLY JSON.
+
+Generate ONLY file paths required to pass the tests.
+
+FORMAT:
+{
+  "files": [
+    { "path": "models/AnyModel.js" },
+    { "path": "controllers/anyController.js" },
+    { "path": "routes/anyRoutes.js" },
+    { "path": "index.js" }
+  ]
+}
+
+No explanations.
+`
+    },
+    {
+      role: "user",
+      content: `TASK:\n${task}\n\nTEST CASES:\n${tests}`
+    }
+  ]);
+
+  const parsed = JSON.parse(res.choices[0].message.content);
+
+  if (!parsed.files || !Array.isArray(parsed.files)) {
+    throw new Error("Invalid file plan");
+  }
+
+  return parsed.files.map(f => f.path);
+}
+
+/* ================= PHASE 2: FILE CONTENT ================= */
+export async function generateFileContent(filePath, task, tests) {
+  let file = null;
+
+  try {
+    const res = await call([
+      {
+        role: "system",
+        content: `
+You are generating ONLY this file:
+
+${filePath}
+
+RULES:
+- Tests are the ONLY specification
+- Minimal code only
+- No comments
+- No console logs
+- Use CommonJS (require/module.exports)
+
+FORMAT:
+{
+  "path": "${filePath}",
+  "content": "exact source code"
+}
+`
+      },
+      {
+        role: "user",
+        content: `TASK:\n${task}\n\nTEST CASES:\n${tests}`
+      }
+    ]);
+
+    file = JSON.parse(res.choices[0].message.content);
+  } catch {
+    file = null;
+  }
+
+  /* ===== MINIMUM GENERIC FALLBACK (MODEL ONLY) ===== */
+  if (
+    !file ||
+    typeof file.path !== "string" ||
+    typeof file.content !== "string" ||
+    file.content.trim().length === 0
+  ) {
+    if (filePath.startsWith("models/")) {
+      const modelName = filePath
+        .replace("models/", "")
+        .replace(".js", "");
+
+      return {
+        path: filePath,
+        content:
+`const mongoose = require('mongoose');
+
+const schema = new mongoose.Schema({}, { strict: false });
+
+module.exports = mongoose.model('${modelName}', schema);
+`
+      };
+    }
+
+    throw new Error(`Invalid or empty content for ${filePath}`);
+  }
+
+  return file;
+}

package/src/fileWriter.js

@@ -0,0 +1,10 @@
+import fs from "fs-extra";
+import path from "path";
+
+export async function writeFiles(baseDir, files) {
+  for (const file of files) {
+    const fullPath = path.join(baseDir, file.path);
+    await fs.ensureDir(path.dirname(fullPath));
+    await fs.writeFile(fullPath, file.content, "utf8");
+  }
+}

package/src/promptTemplate.js

@@ -0,0 +1,43 @@
+export const systemPrompt = `
+You are solving an automated coding exam.
+
+ABSOLUTE PRIORITY:
+The given test cases are the ONLY specification.
+Ignore best practices unless tests require them.
+
+RULES (STRICT):
+- Generate the MINIMUM code needed to pass tests
+- Do NOT add extra logic
+- Do NOT add extra validation
+- Do NOT add console logs
+- Do NOT add comments unless required
+- Do NOT refactor
+- Do NOT optimize
+- Do NOT generalize
+
+ARCHITECTURE RULE:
+- Create ONLY the files required by tests
+- File names, exports, function names MUST match tests EXACTLY
+- Error messages MUST match character-by-character
+- Status codes MUST match exactly
+
+OUTPUT RULES:
+- Output ONLY valid JSON
+- No markdown
+- No explanations
+- No extra text
+
+JSON FORMAT:
+{
+  "files": [
+    {
+      "path": "relative/path/file.js",
+      "content": "exact code"
+    }
+  ]
+}
+
+IMPORTANT:
+Passing tests is MORE IMPORTANT than clean code.
+If a simpler solution passes tests, choose it.
+`;

package/src/runner.js

@@ -0,0 +1,56 @@
+import readlineSync from "readline-sync";
+import fs from "fs";
+import path from "path";
+
+import { setApiKey, generateFilePlan, generateFileContent } from "./aiClient.js";
+import { writeFiles } from "./fileWriter.js";
+
+/* ----------- API KEY (---) ----------- */
+const apiKey = readlineSync.question("--- ", {
+  hideEchoBack: true
+});
+
+if (!apiKey || apiKey.trim().length === 0) {
+  console.log("API key is required");
+  process.exit(1);
+}
+
+setApiKey(apiKey);
+
+/* ----------- TASK (-) ----------- */
+const task = readlineSync.question("- ");
+
+/* ----------- TESTS (-- text OR file path) ----------- */
+let testsInput = readlineSync.question("-- ");
+let tests;
+
+try {
+  if (
+    (testsInput.endsWith(".js") || testsInput.endsWith(".test.js")) &&
+    fs.existsSync(testsInput)
+  ) {
+    tests = fs.readFileSync(testsInput, "utf8");
+  } else {
+    tests = testsInput;
+  }
+} catch {
+  tests = testsInput;
+}
+
+(async () => {
+  try {
+    const paths = await generateFilePlan(task, tests);
+
+    const files = [];
+    for (const p of paths) {
+      const file = await generateFileContent(p, task, tests);
+      files.push(file);
+    }
+
+    await writeFiles(process.cwd(), files);
+    fs.writeFileSync(".ai_codegen_success", "");
+  } catch (e) {
+    fs.writeFileSync(".ai_codegen_error", e.message || "error");
+    process.exit(1);
+  }
+})();