@mayson-org/inject-script 1.0.0

package/README.md

@@ -0,0 +1,71 @@
+# @mayson/inject-script
+
+A lightweight CLI tool and Node.js package designed for **Next.js App Router** applications. It dynamically locates the project's root layout file (regardless of whether it's stored in `app/layout.tsx`, `src/app/layout.tsx`, or inside route groups like `app/(main)/layout.js`) and injects a floating `<iframe>` pointing to `https://mayson.dev/`.
+
+---
+
+## Quick Usage via `npx`
+
+Run directly in any Next.js project root:
+
+```bash
+npx @mayson/inject-script
+```
+
+Or pass optional flags:
+
+```bash
+# Specify custom iframe URL
+npx @mayson/inject-script --url=https://mayson.dev/my-app
+
+# Dry-run mode (preview changes without modifying files)
+npx @mayson/inject-script --dry-run
+
+# Specify target project directory
+npx @mayson/inject-script --dir=./my-nextjs-app
+```
+
+---
+
+## Adding to Next.js `package.json` Build Scripts
+
+You can automate this script during deployment builds (e.g. on Vercel, Netlify, Docker, or GitHub Actions) by adding it to `package.json`:
+
+```json
+{
+  "scripts": {
+    "dev": "next dev",
+    "prebuild": "npx @mayson/inject-script",
+    "build": "next build"
+  }
+}
+```
+
+---
+
+## Programmatic API Usage
+
+You can also import and use `inject-script` in custom build scripts:
+
+```javascript
+import { runAutoInject, findRootLayout, injectMaysonIframe } from 'inject-script';
+
+// 1. Auto-detect and inject in one call
+const result = runAutoInject(process.cwd(), 'https://mayson.dev/');
+
+if (result.success) {
+  console.log(`Injected to: ${result.filePath}`);
+}
+
+// 2. Or manually find layout and inspect
+const rootLayoutPath = findRootLayout('./my-project');
+console.log('Root Layout found at:', rootLayoutPath);
+```
+
+---
+
+## Features
+
+- 🔍 **Dynamic Root Layout Resolution**: Scans `app/`, `src/app/`, and nested route groups for layout files and scores candidates based on HTML/Body root tags.
+- ⚡ **Idempotent**: Safe to run multiple times without creating duplicate iframe injections.
+- 🛡️ **Zero Runtime Overhead**: Inserts a self-contained React `<iframe />` directly into the JSX tree.

package/bin/cli.mjs

@@ -0,0 +1,159 @@
+#!/usr/bin/env node
+
+import fs from 'fs';
+import path from 'path';
+
+function findLayoutFiles(dir, fileList = []) {
+  if (!fs.existsSync(dir)) return fileList;
+  const entries = fs.readdirSync(dir, { withFileTypes: true });
+
+  for (const entry of entries) {
+    const fullPath = path.join(dir, entry.name);
+    if (entry.isDirectory()) {
+      if (entry.name === 'node_modules' || entry.name.startsWith('.')) continue;
+      findLayoutFiles(fullPath, fileList);
+    } else if (entry.isFile()) {
+      const ext = path.extname(entry.name);
+      const base = path.basename(entry.name, ext);
+      if (base === 'layout' && ['.tsx', '.jsx', '.ts', '.js'].includes(ext)) {
+        fileList.push(fullPath);
+      }
+    }
+  }
+  return fileList;
+}
+
+function scoreLayout(filePath, projectRoot) {
+  const content = fs.readFileSync(filePath, 'utf-8');
+  const relativePath = path.relative(projectRoot, filePath);
+  const depth = relativePath.split(path.sep).length;
+  let score = 100 - depth;
+
+  if (/<html/i.test(content)) score += 50;
+  if (/<body/i.test(content)) score += 50;
+  if (/^src[/\\]app[/\\]layout\.[jt]sx?$/.test(relativePath) || /^app[/\\]layout\.[jt]sx?$/.test(relativePath)) {
+    score += 40;
+  }
+  return score;
+}
+
+function findRootLayout(projectRoot) {
+  const possibleDirs = [path.join(projectRoot, 'src', 'app'), path.join(projectRoot, 'app')];
+  const candidates = [];
+
+  for (const dir of possibleDirs) {
+    if (fs.existsSync(dir)) findLayoutFiles(dir, candidates);
+  }
+
+  if (candidates.length === 0) return null;
+  candidates.sort((a, b) => scoreLayout(b, projectRoot) - scoreLayout(a, projectRoot));
+  return candidates[0];
+}
+
+function parseArgs() {
+  const args = process.argv.slice(2);
+  const options = {
+    projectRoot: process.cwd(),
+    url: 'https://mayson.dev/',
+    dryRun: false,
+    help: false,
+  };
+
+  for (let i = 0; i < args.length; i++) {
+    const arg = args[i];
+    if (arg === '--help' || arg === '-h') {
+      options.help = true;
+    } else if (arg === '--dry-run') {
+      options.dryRun = true;
+    } else if (arg.startsWith('--url=')) {
+      options.url = arg.split('=')[1];
+    } else if (arg === '--url' && args[i + 1]) {
+      options.url = args[++i];
+    } else if (arg.startsWith('--dir=')) {
+      options.projectRoot = path.resolve(arg.split('=')[1]);
+    } else if (arg === '--dir' && args[i + 1]) {
+      options.projectRoot = path.resolve(args[++i]);
+    }
+  }
+  return options;
+}
+
+const options = parseArgs();
+
+if (options.help) {
+  console.log(`
+Inject Script CLI - Next.js App Router Root Layout Iframe Injector
+
+Usage:
+  npx inject-script [options]
+
+Options:
+  --dir <path>       Target project root directory (default: current working directory)
+  --url <url>        Iframe URL to inject (default: https://mayson.dev/)
+  --dry-run          Simulate layout detection and code injection without modifying files
+  --help, -h         Show this help message
+`);
+  process.exit(0);
+}
+
+console.log(`🔍 Searching for Next.js root layout in: ${options.projectRoot}`);
+
+const layoutFile = findRootLayout(options.projectRoot);
+
+if (!layoutFile) {
+  console.error(`❌ Error: Could not find any App Router layout file in ${options.projectRoot}`);
+  process.exit(1);
+}
+
+console.log(`📍 Found Root Layout: ${path.relative(options.projectRoot, layoutFile)}`);
+
+const content = fs.readFileSync(layoutFile, 'utf-8');
+const INJECTION_MARKER = '/* @mayson-iframe-injected */';
+
+if (content.includes(INJECTION_MARKER) || content.includes(options.url)) {
+  console.log(`✅ Mayson iframe is already injected in ${path.basename(layoutFile)}.`);
+  process.exit(0);
+}
+
+const snippet = `
+      {${INJECTION_MARKER}}
+      <iframe
+        src="${options.url}"
+        style={{
+          position: 'fixed',
+          bottom: '20px',
+          right: '20px',
+          width: '400px',
+          height: '600px',
+          border: 'none',
+          borderRadius: '12px',
+          boxShadow: '0 8px 32px rgba(0, 0, 0, 0.25)',
+          zIndex: 999999,
+        }}
+        title="Mayson Dev Overlay"
+      />`;
+
+let updatedContent = null;
+
+if (/<\/body>/i.test(content)) {
+  updatedContent = content.replace(/<\/body>/i, `${snippet}\n      </body>`);
+} else if (/<\/html>/i.test(content)) {
+  updatedContent = content.replace(/<\/html>/i, `${snippet}\n      </html>`);
+} else {
+  const lastIndex = content.lastIndexOf('</');
+  if (lastIndex !== -1) {
+    updatedContent = content.slice(0, lastIndex) + snippet + '\n      ' + content.slice(lastIndex);
+  }
+}
+
+if (!updatedContent) {
+  console.error(`❌ Error: Could not find suitable insertion point in ${layoutFile}`);
+  process.exit(1);
+}
+
+if (options.dryRun) {
+  console.log(`🧪 [Dry Run] Would inject iframe to ${layoutFile}`);
+} else {
+  fs.writeFileSync(layoutFile, updatedContent, 'utf-8');
+  console.log(`🎉 Successfully injected Mayson iframe (${options.url}) into ${path.relative(options.projectRoot, layoutFile)}!`);
+}

package/dist/find-root-layout.d.ts

@@ -0,0 +1,5 @@
+/**
+ * Finds the most likely Next.js App Router root layout file in a project.
+ */
+export declare function findRootLayout(projectRoot?: string): string | null;
+//# sourceMappingURL=find-root-layout.d.ts.map

package/dist/find-root-layout.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"find-root-layout.d.ts","sourceRoot":"","sources":["../src/find-root-layout.ts"],"names":[],"mappings":"AAyDA;;GAEG;AACH,wBAAgB,cAAc,CAAC,WAAW,GAAE,MAAsB,GAAG,MAAM,GAAG,IAAI,CA2BjF"}

package/dist/find-root-layout.js

@@ -0,0 +1,75 @@
+import fs from 'fs';
+import path from 'path';
+const LAYOUT_EXTENSIONS = ['.tsx', '.jsx', '.ts', '.js'];
+/**
+ * Recursively search a directory for layout files.
+ */
+function findLayoutFiles(dir, fileList = []) {
+    if (!fs.existsSync(dir))
+        return fileList;
+    const entries = fs.readdirSync(dir, { withFileTypes: true });
+    for (const entry of entries) {
+        const fullPath = path.join(dir, entry.name);
+        if (entry.isDirectory()) {
+            // Don't traverse node_modules or hidden folders (.next, .git)
+            if (entry.name === 'node_modules' || entry.name.startsWith('.')) {
+                continue;
+            }
+            findLayoutFiles(fullPath, fileList);
+        }
+        else if (entry.isFile()) {
+            const ext = path.extname(entry.name);
+            const base = path.basename(entry.name, ext);
+            if (base === 'layout' && LAYOUT_EXTENSIONS.includes(ext)) {
+                fileList.push(fullPath);
+            }
+        }
+    }
+    return fileList;
+}
+/**
+ * Scores a layout file candidate to determine if it is the root layout.
+ * Higher score = higher likelihood of being the root layout.
+ */
+function scoreLayoutCandidate(filePath, projectRoot) {
+    const content = fs.readFileSync(filePath, 'utf-8');
+    const relativePath = path.relative(projectRoot, filePath);
+    const depth = relativePath.split(path.sep).length;
+    let score = 100 - depth; // Shorter path depth is preferred
+    // Check for HTML/Body root tags
+    if (/<html/i.test(content))
+        score += 50;
+    if (/<body/i.test(content))
+        score += 50;
+    // Check if located directly under app/ or src/app/
+    const isDirectAppChild = /^src[/\\]app[/\\]layout\.[jt]sx?$/.test(relativePath) ||
+        /^app[/\\]layout\.[jt]sx?$/.test(relativePath);
+    if (isDirectAppChild)
+        score += 40;
+    return score;
+}
+/**
+ * Finds the most likely Next.js App Router root layout file in a project.
+ */
+export function findRootLayout(projectRoot = process.cwd()) {
+    const possibleAppDirs = [
+        path.join(projectRoot, 'src', 'app'),
+        path.join(projectRoot, 'app'),
+    ];
+    const candidateFiles = [];
+    for (const appDir of possibleAppDirs) {
+        if (fs.existsSync(appDir)) {
+            findLayoutFiles(appDir, candidateFiles);
+        }
+    }
+    if (candidateFiles.length === 0) {
+        return null;
+    }
+    // Score each candidate to pick the true root layout
+    const scored = candidateFiles.map((filePath) => ({
+        filePath,
+        score: scoreLayoutCandidate(filePath, projectRoot),
+    }));
+    scored.sort((a, b) => b.score - a.score);
+    return scored[0]?.filePath || null;
+}

package/dist/index.d.ts

@@ -0,0 +1,9 @@
+import { findRootLayout } from './find-root-layout.js';
+import { injectMaysonIframe, InjectorOptions, InjectorResult } from './injector.js';
+export { findRootLayout, injectMaysonIframe };
+export type { InjectorOptions, InjectorResult };
+/**
+ * Convenience function to automatically locate and inject the Mayson iframe into a Next.js project.
+ */
+export declare function runAutoInject(projectRoot?: string, iframeUrl?: string, dryRun?: boolean): InjectorResult;
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,kBAAkB,EAAE,eAAe,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAEpF,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC;AAC9C,YAAY,EAAE,eAAe,EAAE,cAAc,EAAE,CAAC;AAEhD;;GAEG;AACH,wBAAgB,aAAa,CAAC,WAAW,GAAE,MAAsB,EAAE,SAAS,GAAE,MAA8B,EAAE,MAAM,GAAE,OAAe,GAAG,cAAc,CAiBrJ"}

package/dist/index.js

@@ -0,0 +1,22 @@
+import { findRootLayout } from './find-root-layout.js';
+import { injectMaysonIframe } from './injector.js';
+export { findRootLayout, injectMaysonIframe };
+/**
+ * Convenience function to automatically locate and inject the Mayson iframe into a Next.js project.
+ */
+export function runAutoInject(projectRoot = process.cwd(), iframeUrl = 'https://mayson.dev/', dryRun = false) {
+    const layoutPath = findRootLayout(projectRoot);
+    if (!layoutPath) {
+        return {
+            success: false,
+            filePath: '',
+            alreadyInjected: false,
+            error: `Could not locate Next.js App Router root layout file in ${projectRoot}`,
+        };
+    }
+    return injectMaysonIframe({
+        filePath: layoutPath,
+        iframeUrl,
+        dryRun,
+    });
+}

package/dist/injector.d.ts

@@ -0,0 +1,14 @@
+export interface InjectorOptions {
+    filePath: string;
+    iframeUrl?: string;
+    dryRun?: boolean;
+}
+export interface InjectorResult {
+    success: boolean;
+    filePath: string;
+    alreadyInjected: boolean;
+    modifiedContent?: string;
+    error?: string;
+}
+export declare function injectMaysonIframe(options: InjectorOptions): InjectorResult;
+//# sourceMappingURL=injector.d.ts.map

package/dist/injector.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"injector.d.ts","sourceRoot":"","sources":["../src/injector.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,eAAe;IAC9B,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,OAAO,CAAC;CAClB;AAED,MAAM,WAAW,cAAc;IAC7B,OAAO,EAAE,OAAO,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,eAAe,EAAE,OAAO,CAAC;IACzB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAID,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,eAAe,GAAG,cAAc,CAkF3E"}

package/dist/injector.js

@@ -0,0 +1,76 @@
+import fs from 'fs';
+const INJECTION_MARKER = '/* @mayson-iframe-injected */';
+export function injectMaysonIframe(options) {
+    const { filePath, iframeUrl = 'https://mayson.dev/', dryRun = false } = options;
+    if (!fs.existsSync(filePath)) {
+        return {
+            success: false,
+            filePath,
+            alreadyInjected: false,
+            error: `File not found: ${filePath}`,
+        };
+    }
+    const content = fs.readFileSync(filePath, 'utf-8');
+    // Idempotency check: Don't inject twice
+    if (content.includes(INJECTION_MARKER) || content.includes(iframeUrl)) {
+        return {
+            success: true,
+            filePath,
+            alreadyInjected: true,
+        };
+    }
+    const snippet = `
+      {${INJECTION_MARKER}}
+      <iframe
+        src="${iframeUrl}"
+        style={{
+          position: 'fixed',
+          bottom: '20px',
+          right: '20px',
+          width: '400px',
+          height: '600px',
+          border: 'none',
+          borderRadius: '12px',
+          boxShadow: '0 8px 32px rgba(0, 0, 0, 0.25)',
+          zIndex: 999999,
+        }}
+        title="Mayson Dev Overlay"
+      />`;
+    let updatedContent = null;
+    // Option 1: Inject before </body>
+    if (/<\/body>/i.test(content)) {
+        updatedContent = content.replace(/<\/body>/i, `${snippet}\n      </body>`);
+    }
+    // Option 2: Inject before </html>
+    else if (/<\/html>/i.test(content)) {
+        updatedContent = content.replace(/<\/html>/i, `${snippet}\n      </html>`);
+    }
+    // Option 3: Inject before the closing tag of the return block (e.g. </main> or </div>)
+    else {
+        const lastClosingTagIndex = content.lastIndexOf('</');
+        if (lastClosingTagIndex !== -1) {
+            updatedContent =
+                content.slice(0, lastClosingTagIndex) +
+                    snippet +
+                    '\n      ' +
+                    content.slice(lastClosingTagIndex);
+        }
+    }
+    if (!updatedContent) {
+        return {
+            success: false,
+            filePath,
+            alreadyInjected: false,
+            error: 'Could not find suitable JSX insertion point in layout file.',
+        };
+    }
+    if (!dryRun) {
+        fs.writeFileSync(filePath, updatedContent, 'utf-8');
+    }
+    return {
+        success: true,
+        filePath,
+        alreadyInjected: false,
+        modifiedContent: updatedContent,
+    };
+}

package/package.json

@@ -0,0 +1,44 @@
+{
+  "name": "@mayson-org/inject-script",
+  "version": "1.0.0",
+  "description": "CLI tool and utility to dynamically locate Next.js App Router root layout files and inject the Mayson iframe script/component.",
+  "main": "./dist/index.js",
+  "module": "./dist/index.mjs",
+  "types": "./dist/index.d.ts",
+  "type": "module",
+  "bin": {
+    "inject-script": "bin/cli.mjs"
+  },
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.mjs",
+      "default": "./dist/index.js"
+    }
+  },
+  "scripts": {
+    "build": "tsc",
+    "test": "node --test",
+    "prepublishOnly": "npm run build",
+    "publish:npm": "npm publish --access public"
+  },
+  "files": [
+    "bin",
+    "dist",
+    "src",
+    "README.md"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "keywords": [
+    "nextjs",
+    "app-router",
+    "layout",
+    "iframe",
+    "mayson",
+    "cli"
+  ],
+  "author": "",
+  "license": "MIT"
+}

package/src/find-root-layout.ts

@@ -0,0 +1,88 @@
+import fs from 'fs';
+import path from 'path';
+
+const LAYOUT_EXTENSIONS = ['.tsx', '.jsx', '.ts', '.js'];
+
+/**
+ * Recursively search a directory for layout files.
+ */
+function findLayoutFiles(dir: string, fileList: string[] = []): string[] {
+  if (!fs.existsSync(dir)) return fileList;
+
+  const entries = fs.readdirSync(dir, { withFileTypes: true });
+
+  for (const entry of entries) {
+    const fullPath = path.join(dir, entry.name);
+
+    if (entry.isDirectory()) {
+      // Don't traverse node_modules or hidden folders (.next, .git)
+      if (entry.name === 'node_modules' || entry.name.startsWith('.')) {
+        continue;
+      }
+      findLayoutFiles(fullPath, fileList);
+    } else if (entry.isFile()) {
+      const ext = path.extname(entry.name);
+      const base = path.basename(entry.name, ext);
+      if (base === 'layout' && LAYOUT_EXTENSIONS.includes(ext)) {
+        fileList.push(fullPath);
+      }
+    }
+  }
+
+  return fileList;
+}
+
+/**
+ * Scores a layout file candidate to determine if it is the root layout.
+ * Higher score = higher likelihood of being the root layout.
+ */
+function scoreLayoutCandidate(filePath: string, projectRoot: string): number {
+  const content = fs.readFileSync(filePath, 'utf-8');
+  const relativePath = path.relative(projectRoot, filePath);
+  const depth = relativePath.split(path.sep).length;
+
+  let score = 100 - depth; // Shorter path depth is preferred
+
+  // Check for HTML/Body root tags
+  if (/<html/i.test(content)) score += 50;
+  if (/<body/i.test(content)) score += 50;
+
+  // Check if located directly under app/ or src/app/
+  const isDirectAppChild = /^src[/\\]app[/\\]layout\.[jt]sx?$/.test(relativePath) ||
+                           /^app[/\\]layout\.[jt]sx?$/.test(relativePath);
+  if (isDirectAppChild) score += 40;
+
+  return score;
+}
+
+/**
+ * Finds the most likely Next.js App Router root layout file in a project.
+ */
+export function findRootLayout(projectRoot: string = process.cwd()): string | null {
+  const possibleAppDirs = [
+    path.join(projectRoot, 'src', 'app'),
+    path.join(projectRoot, 'app'),
+  ];
+
+  const candidateFiles: string[] = [];
+
+  for (const appDir of possibleAppDirs) {
+    if (fs.existsSync(appDir)) {
+      findLayoutFiles(appDir, candidateFiles);
+    }
+  }
+
+  if (candidateFiles.length === 0) {
+    return null;
+  }
+
+  // Score each candidate to pick the true root layout
+  const scored = candidateFiles.map((filePath) => ({
+    filePath,
+    score: scoreLayoutCandidate(filePath, projectRoot),
+  }));
+
+  scored.sort((a, b) => b.score - a.score);
+
+  return scored[0]?.filePath || null;
+}

package/src/index.ts

@@ -0,0 +1,27 @@
+import { findRootLayout } from './find-root-layout.js';
+import { injectMaysonIframe, InjectorOptions, InjectorResult } from './injector.js';
+
+export { findRootLayout, injectMaysonIframe };
+export type { InjectorOptions, InjectorResult };
+
+/**
+ * Convenience function to automatically locate and inject the Mayson iframe into a Next.js project.
+ */
+export function runAutoInject(projectRoot: string = process.cwd(), iframeUrl: string = 'https://mayson.dev/', dryRun: boolean = false): InjectorResult {
+  const layoutPath = findRootLayout(projectRoot);
+
+  if (!layoutPath) {
+    return {
+      success: false,
+      filePath: '',
+      alreadyInjected: false,
+      error: `Could not locate Next.js App Router root layout file in ${projectRoot}`,
+    };
+  }
+
+  return injectMaysonIframe({
+    filePath: layoutPath,
+    iframeUrl,
+    dryRun,
+  });
+}

package/src/injector.ts

@@ -0,0 +1,101 @@
+import fs from 'fs';
+
+export interface InjectorOptions {
+  filePath: string;
+  iframeUrl?: string;
+  dryRun?: boolean;
+}
+
+export interface InjectorResult {
+  success: boolean;
+  filePath: string;
+  alreadyInjected: boolean;
+  modifiedContent?: string;
+  error?: string;
+}
+
+const INJECTION_MARKER = '/* @mayson-iframe-injected */';
+
+export function injectMaysonIframe(options: InjectorOptions): InjectorResult {
+  const { filePath, iframeUrl = 'https://mayson.dev/', dryRun = false } = options;
+
+  if (!fs.existsSync(filePath)) {
+    return {
+      success: false,
+      filePath,
+      alreadyInjected: false,
+      error: `File not found: ${filePath}`,
+    };
+  }
+
+  const content = fs.readFileSync(filePath, 'utf-8');
+
+  // Idempotency check: Don't inject twice
+  if (content.includes(INJECTION_MARKER) || content.includes(iframeUrl)) {
+    return {
+      success: true,
+      filePath,
+      alreadyInjected: true,
+    };
+  }
+
+  const snippet = `
+      {${INJECTION_MARKER}}
+      <iframe
+        src="${iframeUrl}"
+        style={{
+          position: 'fixed',
+          bottom: '20px',
+          right: '20px',
+          width: '400px',
+          height: '600px',
+          border: 'none',
+          borderRadius: '12px',
+          boxShadow: '0 8px 32px rgba(0, 0, 0, 0.25)',
+          zIndex: 999999,
+        }}
+        title="Mayson Dev Overlay"
+      />`;
+
+  let updatedContent: string | null = null;
+
+  // Option 1: Inject before </body>
+  if (/<\/body>/i.test(content)) {
+    updatedContent = content.replace(/<\/body>/i, `${snippet}\n      </body>`);
+  }
+  // Option 2: Inject before </html>
+  else if (/<\/html>/i.test(content)) {
+    updatedContent = content.replace(/<\/html>/i, `${snippet}\n      </html>`);
+  }
+  // Option 3: Inject before the closing tag of the return block (e.g. </main> or </div>)
+  else {
+    const lastClosingTagIndex = content.lastIndexOf('</');
+    if (lastClosingTagIndex !== -1) {
+      updatedContent =
+        content.slice(0, lastClosingTagIndex) +
+        snippet +
+        '\n      ' +
+        content.slice(lastClosingTagIndex);
+    }
+  }
+
+  if (!updatedContent) {
+    return {
+      success: false,
+      filePath,
+      alreadyInjected: false,
+      error: 'Could not find suitable JSX insertion point in layout file.',
+    };
+  }
+
+  if (!dryRun) {
+    fs.writeFileSync(filePath, updatedContent, 'utf-8');
+  }
+
+  return {
+    success: true,
+    filePath,
+    alreadyInjected: false,
+    modifiedContent: updatedContent,
+  };
+}