usetraceforge-cli 0.1.1 → 0.1.4

package/dist/index.js

@@ -29,6 +29,11 @@ async function main() {
         outro(chalk.red("Installation cancelled."));
         process.exit(1);
     }
+    const endpoint = await text({
+        message: "What is your TraceForge Ingest Endpoint URL?",
+        placeholder: "http://localhost:80/ingest",
+        defaultValue: "http://localhost:80/ingest",
+    });
     const s = spinner();
     s.start("Installing usetraceforge SDK...");
     try {
@@ -39,7 +44,7 @@ async function main() {
         s.stop(chalk.red("Failed to install SDK. Please run 'npm install usetraceforge' manually."));
     }
     if (framework === "nextjs") {
-        await installNextJs(apiKey);
+        await installNextJs(apiKey, endpoint);
     }
     else if (framework === "express") {
         // await installExpress(apiKey);

package/dist/installers/nextjs.js

@@ -2,13 +2,13 @@ import fs from "fs";
 import path from "path";
 import { spinner } from "@clack/prompts";
 import chalk from "chalk";
-export async function installNextJs(apiKey) {
+export async function installNextJs(apiKey, endpoint) {
     const s = spinner();
     s.start("Configuring Next.js...");
     try {
-        // 1. Add API key to .env.local
+        // 1. Add API key and endpoint to .env.local
         const envPath = path.resolve(process.cwd(), ".env.local");
-        const envVar = `\nNEXT_PUBLIC_TRACEFORGE_API_KEY="${apiKey}"\n`;
+        const envVar = `\nNEXT_PUBLIC_TRACEFORGE_API_KEY="${apiKey}"\nNEXT_PUBLIC_TRACEFORGE_INGEST_URL="${endpoint}"\n`;
         if (fs.existsSync(envPath)) {
             const content = fs.readFileSync(envPath, "utf-8");
             if (!content.includes("NEXT_PUBLIC_TRACEFORGE_API_KEY")) {
@@ -18,12 +18,37 @@ export async function installNextJs(apiKey) {
         else {
             fs.writeFileSync(envPath, envVar);
         }
-        // 2. Find layout.tsx
+        // 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"),
-            path.resolve(process.cwd(), "app/layout.jsx"),
-            path.resolve(process.cwd(), "src/app/layout.jsx"),
         ];
         let targetLayout = null;
         for (const p of layoutPaths) {
@@ -36,21 +61,45 @@ export async function installNextJs(apiKey) {
             s.stop(chalk.yellow("Could not find app/layout.tsx. Please configure TraceForge manually."));
             return;
         }
-        // 3. Inject code into layout.tsx
+        // 4. Inject code into layout.tsx
         let layoutCode = fs.readFileSync(targetLayout, "utf-8");
-        // Check if already injected
-        if (layoutCode.includes("TraceForgeErrorBoundary")) {
+        if (layoutCode.includes("TraceForgeProvider")) {
             s.stop(chalk.green("TraceForge is already configured in layout.tsx!"));
             return;
         }
-        // Add import statement
-        const importStatement = `import { TraceForgeErrorBoundary } from "usetraceforge/react";\n`;
+        const importStatement = `import { TraceForgeProvider } from "@/components/TraceForgeProvider";\n`;
         layoutCode = importStatement + layoutCode;
-        // Replace <body> with <body><TraceForgeErrorBoundary>
-        layoutCode = layoutCode.replace(/(<body[^>]*>)/g, `$1\n        <TraceForgeErrorBoundary>`);
-        // Replace </body> with </TraceForgeErrorBoundary></body>
-        layoutCode = layoutCode.replace(/<\/body>/g, `        </TraceForgeErrorBoundary>\n      </body>`);
+        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);
+            }
+        }
         s.stop(chalk.green("Next.js configuration complete!"));
     }
     catch (error) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "usetraceforge-cli",
-  "version": "0.1.1",
+  "version": "0.1.4",
   "description": "TraceForge CLI Wizard for 2-click installations",
   "bin": {
     "traceforge": "./dist/index.js"