usetraceforge-cli 0.1.4 → 0.1.6

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,76 @@
+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);
+        const model = genAI.getGenerativeModel({ model: "gemini-1.5-flash" });
+        const prompt = `
+I am trying to install an SDK called "usetraceforge".
+I have the following files in my project:
+
+${folderStructure}
+
+Please tell me EXACTLY how to configure TraceForge for my framework.
+If this is Next.js 15, tell me to create 'instrumentation.ts'.
+If this is Express, tell me to add 'TraceForge.expressErrorHandler()'.
+Keep your answer very short, professional, and provide the exact code snippets I need to copy and paste.
+The API Key they should use is: ${apiKey}
+The Endpoint is: ${endpoint}
+`;
+        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

@@ -4,7 +4,7 @@ import { spinner } from "@clack/prompts";
 import chalk from "chalk";
 export async function installNextJs(apiKey, endpoint) {
     const s = spinner();
-    s.start("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");
@@ -18,92 +18,35 @@ export async function installNextJs(apiKey, endpoint) {
         else {
             fs.writeFileSync(envPath, envVar);
         }
-        // 2. Generate TraceForgeProvider.tsx
-        const componentsDir = path.resolve(process.cwd(), "components");
-        if (!fs.existsSync(componentsDir)) {
-            fs.mkdirSync(componentsDir);
-        }
-        const providerCode = `"use client";
-import { useEffect } from "react";
-import TraceForge from "usetraceforge";
-import { TraceForgeErrorBoundary } from "usetraceforge/react";
-
-let initialized = false;
+        // 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 TraceForgeProvider({ children }: { children: React.ReactNode }) {
-  useEffect(() => {
-    if (initialized) return;
-    TraceForge.init({
-      apiKey: process.env.NEXT_PUBLIC_TRACEFORGE_API_KEY!,
-      endpoint: process.env.NEXT_PUBLIC_TRACEFORGE_INGEST_URL,
-      autoCapture: true,
-    });
-    initialized = true;
-  }, []);
+export function register() {
+  TraceForge.init({
+    apiKey: process.env.NEXT_PUBLIC_TRACEFORGE_API_KEY!,
+    endpoint: process.env.NEXT_PUBLIC_TRACEFORGE_INGEST_URL,
+    autoCapture: true,
+  });
+}
 
-  return <TraceForgeErrorBoundary>{children}</TraceForgeErrorBoundary>;
+export function onRequestError(err: any, request: any) {
+  TraceForge.captureException(err, { tags: { route: request.url } });
 }
 `;
-        fs.writeFileSync(path.join(componentsDir, "TraceForgeProvider.tsx"), providerCode);
-        // 3. Find layout.tsx
-        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) {
-            s.stop(chalk.yellow("Could not find app/layout.tsx. Please configure TraceForge manually."));
-            return;
+        if (!fs.existsSync(instrumentationPath)) {
+            fs.writeFileSync(instrumentationPath, instrumentationCode);
+            console.log(chalk.green(`\nAgent: Created ${isSrc ? "src/" : ""}instrumentation.ts!`));
         }
-        // 4. Inject code into layout.tsx
-        let layoutCode = fs.readFileSync(targetLayout, "utf-8");
-        if (layoutCode.includes("TraceForgeProvider")) {
-            s.stop(chalk.green("TraceForge is already configured in layout.tsx!"));
-            return;
-        }
-        const importStatement = `import { TraceForgeProvider } from "@/components/TraceForgeProvider";\n`;
-        layoutCode = importStatement + layoutCode;
-        layoutCode = layoutCode.replace(/(<body[^>]*>)/g, `$1\n        <TraceForgeProvider>`);
-        layoutCode = layoutCode.replace(/<\/body>/g, `        </TraceForgeProvider>\n      </body>`);
-        fs.writeFileSync(targetLayout, layoutCode);
-        // 5. Update next.config.js / next.config.mjs
-        const configPaths = [
-            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) {
-            let configCode = fs.readFileSync(targetConfig, "utf-8");
-            if (!configCode.includes("withTraceForgeConfig")) {
-                const isMjs = targetConfig.endsWith(".mjs") || configCode.includes("export default");
-                // Add import
-                if (isMjs) {
-                    configCode = `import { withTraceForgeConfig } from "usetraceforge/next-plugin";\n` + configCode;
-                    configCode = configCode.replace(/export default (.+);/g, `export default withTraceForgeConfig($1);`);
-                }
-                else {
-                    configCode = `const { withTraceForgeConfig } = require("usetraceforge/next-plugin");\n` + configCode;
-                    configCode = configCode.replace(/module\.exports = (.+);/g, `module.exports = withTraceForgeConfig($1);`);
-                }
-                fs.writeFileSync(targetConfig, configCode);
-            }
+        else {
+            console.log(chalk.yellow(`\nAgent: instrumentation.ts already exists. Please manually add TraceForge.`));
         }
-        s.stop(chalk.green("Next.js configuration complete!"));
+        s.stop(chalk.green("Agent: Next.js configuration complete!"));
     }
     catch (error) {
-        s.stop(chalk.red("Failed to configure Next.js automatically."));
+        s.stop(chalk.red("Agent: An error occurred."));
         console.error(error);
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "usetraceforge-cli",
-  "version": "0.1.4",
+  "version": "0.1.6",
   "description": "TraceForge CLI Wizard for 2-click installations",
   "bin": {
     "traceforge": "./dist/index.js"
@@ -17,12 +17,14 @@
   },
   "dependencies": {
     "@clack/prompts": "^0.7.0",
+    "@google/generative-ai": "^0.24.1",
     "chalk": "^5.3.0",
-    "execa": "^9.3.0"
+    "execa": "^9.3.0",
+    "ts-morph": "^28.0.0"
   },
   "devDependencies": {
     "@types/node": "^20.0.0",
-    "typescript": "^5.5.4",
-    "tsx": "^4.16.2"
+    "tsx": "^4.16.2",
+    "typescript": "^5.5.4"
   }
 }