usetraceforge-cli 0.1.5 → 0.1.7

package/dist/index.js

@@ -3,6 +3,7 @@ import { intro, outro, text, select, spinner } from "@clack/prompts";
 import chalk from "chalk";
 import { execa } from "execa";
 import { installNextJs } from "./installers/nextjs.js";
+import { runGeminiAgent } from "./installers/gemini.js";
 async function main() {
     console.log();
     intro(chalk.bgBlue.white.bold(" TraceForge Wizard "));
@@ -21,8 +22,9 @@ async function main() {
     const framework = await select({
         message: "Which framework are you using?",
         options: [
-            { value: "nextjs", label: "Next.js (React)" },
-            { value: "express", label: "Express (Node.js)" },
+            { value: "nextjs", label: "Next.js (Prebuilt)" },
+            { value: "express", label: "Express.js (Prebuilt)" },
+            { value: "gemini", label: "🤖 Let AI scan my codebase (Gemini Agent)" },
         ],
     });
     if (typeof framework !== "string") {
@@ -47,9 +49,11 @@ async function main() {
         await installNextJs(apiKey, endpoint);
     }
     else if (framework === "express") {
-        // await installExpress(apiKey);
         console.log(chalk.yellow("Express auto-installation coming soon."));
     }
+    else if (framework === "gemini") {
+        await runGeminiAgent(apiKey, endpoint);
+    }
     outro(chalk.green("✨ You're all set! TraceForge is now protecting your application."));
 }
 main().catch((err) => {

package/dist/installers/gemini.js

@@ -0,0 +1,81 @@
+import fs from "fs";
+import path from "path";
+import { text, spinner } from "@clack/prompts";
+import chalk from "chalk";
+import { GoogleGenerativeAI } from "@google/generative-ai";
+function getFiles(dir, fileList = [], depth = 0) {
+    if (depth > 3)
+        return fileList;
+    const files = fs.readdirSync(dir);
+    for (const file of files) {
+        if (file === "node_modules" || file === ".git" || file === ".next" || file === "dist") {
+            continue;
+        }
+        const filePath = path.join(dir, file);
+        if (fs.statSync(filePath).isDirectory()) {
+            getFiles(filePath, fileList, depth + 1);
+        }
+        else {
+            fileList.push(filePath.replace(process.cwd(), ""));
+        }
+    }
+    return fileList;
+}
+export async function runGeminiAgent(apiKey, endpoint) {
+    console.log(chalk.blue("\n🤖 Initializing TraceForge AI Agent..."));
+    // Try to find Gemini Key in .env
+    let geminiKey = process.env.GEMINI_API_KEY;
+    const envPath = path.resolve(process.cwd(), ".env");
+    if (!geminiKey && fs.existsSync(envPath)) {
+        const envContent = fs.readFileSync(envPath, "utf-8");
+        const match = envContent.match(/GEMINI_API_KEY="?([^"\n]+)"?/);
+        if (match)
+            geminiKey = match[1];
+    }
+    if (!geminiKey) {
+        const keyResponse = await text({
+            message: "Please enter your Google Gemini API Key (we won't save this):",
+            placeholder: "AIzaSy...",
+        });
+        if (typeof keyResponse === "symbol" || !keyResponse) {
+            console.log(chalk.red("Agent: Cancelled."));
+            return;
+        }
+        geminiKey = keyResponse;
+    }
+    const s = spinner();
+    s.start("Agent: Scanning your codebase...");
+    const files = getFiles(process.cwd());
+    const folderStructure = files.join("\n");
+    s.message("Agent: Analyzing architecture with Gemini...");
+    try {
+        const genAI = new GoogleGenerativeAI(geminiKey);
+        // Use gemini-pro as a safe fallback since 1.5-flash threw a 404 on your key
+        const model = genAI.getGenerativeModel({ model: "gemini/gemini-3.1-flash-lite" });
+        const prompt = `
+I am trying to install an SDK called "usetraceforge".
+I have the following files in my project:
+
+${folderStructure}
+
+Here is the documentation for usetraceforge:
+- Next.js 15+ Server: Create 'instrumentation.ts' (or 'src/instrumentation.ts') exporting 'register' and 'onRequestError(err, request)' which calls 'TraceForge.captureException(err, { tags: { route: request.url } })'.
+- Next.js Client: Wrap your app with '<TraceForgeProvider>' from 'usetraceforge/react'.
+- Express: Call 'app.use(TraceForge.expressErrorHandler())' AFTER all routes but BEFORE custom error handlers.
+- Initialization: You must always call 'TraceForge.init({ apiKey, endpoint })' as early as possible.
+
+The user's API Key is: ${apiKey}
+The user's Endpoint is: ${endpoint}
+
+Based on the folder structure above, tell the developer exactly which files they need to edit and provide the exact code snippets they need to copy and paste to configure TraceForge. Be concise, accurate, and professional.
+`;
+        const result = await model.generateContent(prompt);
+        const response = result.response;
+        s.stop(chalk.green("Agent: Analysis complete!"));
+        console.log("\n" + chalk.cyan(response.text()) + "\n");
+    }
+    catch (error) {
+        s.stop(chalk.red("Agent: Failed to communicate with Gemini API."));
+        console.error(error.message || error);
+    }
+}

package/dist/installers/nextjs.js

@@ -2,10 +2,9 @@ import fs from "fs";
 import path from "path";
 import { spinner } from "@clack/prompts";
 import chalk from "chalk";
-import { Project, SyntaxKind } from "ts-morph";
 export async function installNextJs(apiKey, endpoint) {
     const s = spinner();
-    s.start("Agent: Scanning and configuring Next.js...");
+    s.start("Agent: Configuring Next.js Instrumentation...");
     try {
         // 1. Add API key and endpoint to .env.local
         const envPath = path.resolve(process.cwd(), ".env.local");
@@ -19,93 +18,35 @@ export async function installNextJs(apiKey, endpoint) {
         else {
             fs.writeFileSync(envPath, envVar);
         }
-        const project = new Project();
-        // 2. Next.js Config AST Injection
-        const configPaths = [
-            path.resolve(process.cwd(), "next.config.ts"),
-            path.resolve(process.cwd(), "next.config.mjs"),
-            path.resolve(process.cwd(), "next.config.js"),
-        ];
-        let targetConfig = null;
-        for (const p of configPaths) {
-            if (fs.existsSync(p)) {
-                targetConfig = p;
-                break;
-            }
-        }
-        if (targetConfig) {
-            const sourceFile = project.addSourceFileAtPath(targetConfig);
-            const text = sourceFile.getFullText();
-            if (!text.includes("withTraceForgeConfig")) {
-                const isEsm = targetConfig.endsWith(".ts") || targetConfig.endsWith(".mjs") || text.includes("export default");
-                if (isEsm) {
-                    sourceFile.addImportDeclaration({
-                        namedImports: ["withTraceForgeConfig"],
-                        moduleSpecifier: "usetraceforge/next-plugin"
-                    });
-                    const defaultExport = sourceFile.getExportAssignment(d => !d.isExportEquals());
-                    if (defaultExport) {
-                        const expression = defaultExport.getExpression();
-                        defaultExport.setExpression(`withTraceForgeConfig(${expression.getText()})`);
-                    }
-                    else {
-                        console.log(chalk.yellow("\nAgent: Could not parse export default in next.config. Please wrap your config manually."));
-                    }
-                }
-                else {
-                    // Fallback to simpler regex for CJS files if AST binary expression is too complex
-                    let configCode = fs.readFileSync(targetConfig, "utf-8");
-                    configCode = `const { withTraceForgeConfig } = require("usetraceforge/next-plugin");\n` + configCode;
-                    configCode = configCode.replace(/module\.exports = (.+);/g, `module.exports = withTraceForgeConfig($1);`);
-                    fs.writeFileSync(targetConfig, configCode);
-                }
-                sourceFile.saveSync();
-            }
-        }
-        else {
-            console.log(chalk.yellow("\nAgent: Could not locate next.config.ts/mjs. Please configure Webpack loader manually."));
-        }
-        // 3. Layout AST Injection
-        const layoutPaths = [
-            path.resolve(process.cwd(), "app/layout.tsx"),
-            path.resolve(process.cwd(), "src/app/layout.tsx"),
-        ];
-        let targetLayout = null;
-        for (const p of layoutPaths) {
-            if (fs.existsSync(p)) {
-                targetLayout = p;
-                break;
-            }
-        }
-        if (targetLayout) {
-            const sourceFile = project.addSourceFileAtPath(targetLayout);
-            if (!sourceFile.getFullText().includes("TraceForgeProvider")) {
-                sourceFile.addImportDeclaration({
-                    namedImports: ["TraceForgeProvider"],
-                    moduleSpecifier: "usetraceforge/react"
-                });
-                // Find body tag
-                const jsxElements = sourceFile.getDescendantsOfKind(SyntaxKind.JsxElement);
-                let bodyTag = jsxElements.find(el => el.getOpeningElement().getTagNameNode().getText() === "body");
-                if (bodyTag) {
-                    const childrenText = bodyTag.getJsxChildren().map(c => c.getText()).join("");
-                    const opening = bodyTag.getOpeningElement().getText();
-                    const closing = bodyTag.getClosingElement().getText();
-                    bodyTag.replaceWithText(`${opening}\n        <TraceForgeProvider>\n          ${childrenText}\n        </TraceForgeProvider>\n      ${closing}`);
-                    sourceFile.saveSync();
-                }
-                else {
-                    console.log(chalk.yellow("\nAgent: Could not find <body> tag in layout.tsx. Please add <TraceForgeProvider> manually."));
-                }
-            }
+        // 2. Generate instrumentation.ts
+        const isSrc = fs.existsSync(path.resolve(process.cwd(), "src"));
+        const instrumentationDir = isSrc ? path.resolve(process.cwd(), "src") : process.cwd();
+        const instrumentationPath = path.resolve(instrumentationDir, "instrumentation.ts");
+        const instrumentationCode = `import TraceForge from "usetraceforge";
+
+export function register() {
+  TraceForge.init({
+    apiKey: process.env.NEXT_PUBLIC_TRACEFORGE_API_KEY!,
+    endpoint: process.env.NEXT_PUBLIC_TRACEFORGE_INGEST_URL,
+    autoCapture: true,
+  });
+}
+
+export function onRequestError(err: any, request: any) {
+  TraceForge.captureException(err, { tags: { route: request.url } });
+}
+`;
+        if (!fs.existsSync(instrumentationPath)) {
+            fs.writeFileSync(instrumentationPath, instrumentationCode);
+            console.log(chalk.green(`\nAgent: Created ${isSrc ? "src/" : ""}instrumentation.ts!`));
         }
         else {
-            console.log(chalk.yellow("\nAgent: Could not find layout.tsx. Please add <TraceForgeProvider> manually."));
+            console.log(chalk.yellow(`\nAgent: instrumentation.ts already exists. Please manually add TraceForge.`));
         }
         s.stop(chalk.green("Agent: Next.js configuration complete!"));
     }
     catch (error) {
-        s.stop(chalk.red("Agent: An error occurred while parsing files. Please configure manually."));
+        s.stop(chalk.red("Agent: An error occurred."));
         console.error(error);
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "usetraceforge-cli",
-  "version": "0.1.5",
+  "version": "0.1.7",
   "description": "TraceForge CLI Wizard for 2-click installations",
   "bin": {
     "traceforge": "./dist/index.js"
@@ -17,6 +17,7 @@
   },
   "dependencies": {
     "@clack/prompts": "^0.7.0",
+    "@google/generative-ai": "^0.24.1",
     "chalk": "^5.3.0",
     "execa": "^9.3.0",
     "ts-morph": "^28.0.0"