usetraceforge-cli 0.1.4 → 0.1.5

package/dist/installers/nextjs.js

@@ -2,9 +2,10 @@ 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("Configuring Next.js...");
+    s.start("Agent: Scanning and configuring Next.js...");
     try {
         // 1. Add API key and endpoint to .env.local
         const envPath = path.resolve(process.cwd(), ".env.local");
@@ -18,62 +19,10 @@ 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;
-
-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;
-  }, []);
-
-  return <TraceForgeErrorBoundary>{children}</TraceForgeErrorBoundary>;
-}
-`;
-        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;
-        }
-        // 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 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"),
         ];
@@ -85,25 +34,78 @@ export function TraceForgeProvider({ children }: { children: React.ReactNode })
             }
         }
         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);`);
+            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);
                 }
-                fs.writeFileSync(targetConfig, configCode);
+                sourceFile.saveSync();
             }
         }
-        s.stop(chalk.green("Next.js configuration complete!"));
+        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."));
+                }
+            }
+        }
+        else {
+            console.log(chalk.yellow("\nAgent: Could not find layout.tsx. Please add <TraceForgeProvider> manually."));
+        }
+        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 while parsing files. Please configure manually."));
         console.error(error);
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "usetraceforge-cli",
-  "version": "0.1.4",
+  "version": "0.1.5",
   "description": "TraceForge CLI Wizard for 2-click installations",
   "bin": {
     "traceforge": "./dist/index.js"
@@ -18,11 +18,12 @@
   "dependencies": {
     "@clack/prompts": "^0.7.0",
     "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"
   }
 }