@yoamigo.com/core 0.3.8 → 0.3.9

package/dist/index.js

@@ -181,9 +181,13 @@ var BuilderSelectionManager = class {
       page: window.location.pathname
     });
   }
-  sendToParent(message) {
+  sendToParent(message, transfer) {
     try {
-      window.parent.postMessage(message, "*");
+      if (transfer && transfer.length > 0) {
+        window.parent.postMessage(message, "*", transfer);
+      } else {
+        window.parent.postMessage(message, "*");
+      }
     } catch (error) {
       console.error("[BuilderSelection] Failed to send message to parent:", error);
     }
@@ -210,6 +214,25 @@ var BuilderSelectionManager = class {
             this.captureRegionScreenshot(data.region);
           }
           break;
+        case "NAVIGATE":
+          if (data.path) {
+            this.handleNavigate(data.path);
+          }
+          break;
+        case "CAPTURE_ALL_PAGES":
+          if (data.checkpointId && data.viewport) {
+            this.captureAllPagesScreenshots(data.checkpointId, data.viewport);
+          }
+          break;
+        case "GET_SCROLL_POSITION":
+          this.sendToParent({ type: "SCROLL_POSITION", scrollY: window.scrollY });
+          break;
+        case "GET_ALL_ROUTES":
+          this.sendToParent({ type: "ALL_ROUTES", routes: this.getAllRoutes() });
+          break;
+        case "GET_CURRENT_PAGE":
+          this.sendToParent({ type: "CURRENT_PAGE", pagePath: window.location.pathname });
+          break;
       }
     });
   }
@@ -285,6 +308,25 @@ var BuilderSelectionManager = class {
   handleKeyDown = (e) => {
     const isMac = navigator.platform.toUpperCase().indexOf("MAC") >= 0;
     const modifier = isMac ? e.metaKey : e.ctrlKey;
+    const activeEl = document.activeElement;
+    const isEditing = activeEl?.tagName === "INPUT" || activeEl?.tagName === "TEXTAREA" || activeEl?.getAttribute("contenteditable") === "true" || activeEl?.closest(".ya-text-editing") || activeEl?.closest(".ya-link-editing");
+    if (!isEditing && modifier) {
+      if (e.key === "z" && !e.shiftKey) {
+        e.preventDefault();
+        this.sendToParent({ type: "IFRAME_UNDO" });
+        return;
+      }
+      if (e.key === "z" && e.shiftKey) {
+        e.preventDefault();
+        this.sendToParent({ type: "IFRAME_REDO" });
+        return;
+      }
+    }
+    if (!isEditing && e.ctrlKey && e.key === "y") {
+      e.preventDefault();
+      this.sendToParent({ type: "IFRAME_REDO" });
+      return;
+    }
     if (modifier) {
       if (e.key === "=" || e.key === "+" || e.key === "-" || e.key === "0" || e.key === "1") {
         e.preventDefault();
@@ -293,12 +335,8 @@ var BuilderSelectionManager = class {
         return;
       }
     }
-    if (e.key === "Shift") {
-      const activeElement = document.activeElement;
-      const isEditing = activeElement?.closest(".ya-text-editing") || activeElement?.closest(".ya-link-editing");
-      if (!isEditing) {
-        this.sendToParent({ type: "SHIFT_KEY_PRESSED" });
-      }
+    if (e.key === "Shift" && !isEditing) {
+      this.sendToParent({ type: "SHIFT_KEY_PRESSED" });
     }
     if (e.key === "Escape") {
       this.sendToParent({ type: "ESCAPE_KEY_PRESSED" });
@@ -517,6 +555,35 @@ var BuilderSelectionManager = class {
       this.notifyPageReady();
     };
   }
+  /**
+   * Navigate to a new route with retry logic for HMR timing
+   * Used by AI agent to navigate after creating new pages
+   * Retries navigation if route doesn't exist yet (new page being created)
+   */
+  handleNavigate(path) {
+    console.log("[BuilderSelection] Navigate request:", path);
+    const MAX_RETRIES = 10;
+    const RETRY_DELAY_MS = 300;
+    let retries = 0;
+    const attemptNavigation = () => {
+      history.pushState({}, "", path);
+      window.dispatchEvent(new PopStateEvent("popstate"));
+      setTimeout(() => {
+        if (window.location.pathname === path) {
+          console.log("[BuilderSelection] Navigation successful:", path);
+          this.notifyPageReady();
+        } else if (retries < MAX_RETRIES) {
+          retries++;
+          console.log(`[BuilderSelection] Navigation retry ${retries}/${MAX_RETRIES} for path: ${path}`);
+          setTimeout(attemptNavigation, RETRY_DELAY_MS);
+        } else {
+          console.warn("[BuilderSelection] Navigation failed after max retries:", path);
+          this.notifyPageReady();
+        }
+      }, 100);
+    };
+    attemptNavigation();
+  }
   showHoverOverlay(element) {
     if (!this.hoverOverlay) return;
     const rect = element.getBoundingClientRect();
@@ -757,6 +824,154 @@ var BuilderSelectionManager = class {
     document.body.appendChild(container);
     this.selections.set(selectionId, { element, container, badge, border });
   }
+  /**
+   * Get all available routes from the page
+   * Looks for routes registered via wouter or standard link patterns
+   */
+  getAllRoutes() {
+    const wouterRoutes = window.__wouterRoutes;
+    if (wouterRoutes && Array.isArray(wouterRoutes)) {
+      return wouterRoutes;
+    }
+    const links = document.querySelectorAll('a[href^="/"]');
+    const routes = /* @__PURE__ */ new Set(["/"]);
+    links.forEach((link) => {
+      const href = link.getAttribute("href");
+      if (href && href.startsWith("/") && !href.includes("http")) {
+        const path = href.split("?")[0].replace(/\/$/, "") || "/";
+        routes.add(path);
+      }
+    });
+    const navLinks = document.querySelectorAll("nav a[href]");
+    navLinks.forEach((link) => {
+      const href = link.getAttribute("href");
+      if (href && href.startsWith("/")) {
+        const path = href.split("?")[0].replace(/\/$/, "") || "/";
+        routes.add(path);
+      }
+    });
+    return Array.from(routes).sort();
+  }
+  /**
+   * Capture full-page screenshots for all pages
+   * Non-blocking: uses requestIdleCallback between pages
+   * Transfers ArrayBuffer directly (zero-copy) for performance
+   */
+  async captureAllPagesScreenshots(checkpointId, viewport) {
+    console.log("[BuilderSelection] Starting multi-page capture for checkpoint:", checkpointId);
+    const routes = this.getAllRoutes();
+    const originalPath = window.location.pathname;
+    const total = routes.length;
+    this.sendToParent({
+      type: "CAPTURE_PROGRESS",
+      checkpointId,
+      completed: 0,
+      total
+    });
+    for (let i = 0; i < routes.length; i++) {
+      const pagePath = routes[i];
+      try {
+        await new Promise((resolve) => {
+          if ("requestIdleCallback" in window) {
+            requestIdleCallback(() => resolve(), { timeout: 1e3 });
+          } else {
+            setTimeout(resolve, 50);
+          }
+        });
+        if (window.location.pathname !== pagePath) {
+          await this.navigateAndWait(pagePath);
+        }
+        await new Promise((resolve) => setTimeout(resolve, 100));
+        const buffer = await this.capturePageAsArrayBuffer();
+        if (buffer) {
+          this.sendToParent(
+            {
+              type: "PAGE_SCREENSHOT_READY",
+              checkpointId,
+              pagePath,
+              buffer
+            },
+            [buffer]
+            // Transfer list - moves ownership, no copy
+          );
+        }
+        this.sendToParent({
+          type: "CAPTURE_PROGRESS",
+          checkpointId,
+          completed: i + 1,
+          total
+        });
+      } catch (error) {
+        console.error(`[BuilderSelection] Failed to capture page ${pagePath}:`, error);
+      }
+    }
+    if (window.location.pathname !== originalPath) {
+      await this.navigateAndWait(originalPath);
+    }
+    this.sendToParent({
+      type: "ALL_SCREENSHOTS_COMPLETE",
+      checkpointId
+    });
+    console.log("[BuilderSelection] Multi-page capture complete:", checkpointId);
+  }
+  /**
+   * Navigate to a path and wait for page ready
+   */
+  async navigateAndWait(path) {
+    return new Promise((resolve) => {
+      history.pushState({}, "", path);
+      window.dispatchEvent(new PopStateEvent("popstate"));
+      requestAnimationFrame(() => {
+        requestAnimationFrame(() => {
+          resolve();
+        });
+      });
+    });
+  }
+  /**
+   * Capture current page as ArrayBuffer (for zero-copy transfer)
+   */
+  async capturePageAsArrayBuffer() {
+    try {
+      const html2canvas = (await import("html2canvas-pro")).default;
+      const overlays = document.querySelectorAll(".builder-selection-container, #builder-hover-overlay");
+      overlays.forEach((el) => {
+        ;
+        el.style.display = "none";
+      });
+      const originalScrollY = window.scrollY;
+      window.scrollTo(0, 0);
+      await new Promise((resolve) => requestAnimationFrame(resolve));
+      const canvas = await html2canvas(document.body, {
+        scale: 0.5,
+        // Lower resolution for reasonable file size
+        logging: false,
+        useCORS: true,
+        allowTaint: true,
+        backgroundColor: "#ffffff",
+        windowHeight: document.body.scrollHeight,
+        // Full page height
+        height: document.body.scrollHeight
+      });
+      window.scrollTo(0, originalScrollY);
+      overlays.forEach((el) => {
+        ;
+        el.style.display = "";
+      });
+      const blob = await new Promise((resolve) => {
+        canvas.toBlob((b) => resolve(b), "image/jpeg", 0.6);
+      });
+      if (!blob) {
+        console.error("[BuilderSelection] Failed to create blob from canvas");
+        return null;
+      }
+      const buffer = await blob.arrayBuffer();
+      return buffer;
+    } catch (error) {
+      console.error("[BuilderSelection] Page capture failed:", error);
+      return null;
+    }
+  }
 };
 var instance = null;
 function initBuilderSelection() {

package/dist/lib.js

@@ -140,9 +140,13 @@ var BuilderSelectionManager = class {
       page: window.location.pathname
     });
   }
-  sendToParent(message) {
+  sendToParent(message, transfer) {
     try {
-      window.parent.postMessage(message, "*");
+      if (transfer && transfer.length > 0) {
+        window.parent.postMessage(message, "*", transfer);
+      } else {
+        window.parent.postMessage(message, "*");
+      }
     } catch (error) {
       console.error("[BuilderSelection] Failed to send message to parent:", error);
     }
@@ -169,6 +173,25 @@ var BuilderSelectionManager = class {
             this.captureRegionScreenshot(data.region);
           }
           break;
+        case "NAVIGATE":
+          if (data.path) {
+            this.handleNavigate(data.path);
+          }
+          break;
+        case "CAPTURE_ALL_PAGES":
+          if (data.checkpointId && data.viewport) {
+            this.captureAllPagesScreenshots(data.checkpointId, data.viewport);
+          }
+          break;
+        case "GET_SCROLL_POSITION":
+          this.sendToParent({ type: "SCROLL_POSITION", scrollY: window.scrollY });
+          break;
+        case "GET_ALL_ROUTES":
+          this.sendToParent({ type: "ALL_ROUTES", routes: this.getAllRoutes() });
+          break;
+        case "GET_CURRENT_PAGE":
+          this.sendToParent({ type: "CURRENT_PAGE", pagePath: window.location.pathname });
+          break;
       }
     });
   }
@@ -244,6 +267,25 @@ var BuilderSelectionManager = class {
   handleKeyDown = (e) => {
     const isMac = navigator.platform.toUpperCase().indexOf("MAC") >= 0;
     const modifier = isMac ? e.metaKey : e.ctrlKey;
+    const activeEl = document.activeElement;
+    const isEditing = activeEl?.tagName === "INPUT" || activeEl?.tagName === "TEXTAREA" || activeEl?.getAttribute("contenteditable") === "true" || activeEl?.closest(".ya-text-editing") || activeEl?.closest(".ya-link-editing");
+    if (!isEditing && modifier) {
+      if (e.key === "z" && !e.shiftKey) {
+        e.preventDefault();
+        this.sendToParent({ type: "IFRAME_UNDO" });
+        return;
+      }
+      if (e.key === "z" && e.shiftKey) {
+        e.preventDefault();
+        this.sendToParent({ type: "IFRAME_REDO" });
+        return;
+      }
+    }
+    if (!isEditing && e.ctrlKey && e.key === "y") {
+      e.preventDefault();
+      this.sendToParent({ type: "IFRAME_REDO" });
+      return;
+    }
     if (modifier) {
       if (e.key === "=" || e.key === "+" || e.key === "-" || e.key === "0" || e.key === "1") {
         e.preventDefault();
@@ -252,12 +294,8 @@ var BuilderSelectionManager = class {
         return;
       }
     }
-    if (e.key === "Shift") {
-      const activeElement = document.activeElement;
-      const isEditing = activeElement?.closest(".ya-text-editing") || activeElement?.closest(".ya-link-editing");
-      if (!isEditing) {
-        this.sendToParent({ type: "SHIFT_KEY_PRESSED" });
-      }
+    if (e.key === "Shift" && !isEditing) {
+      this.sendToParent({ type: "SHIFT_KEY_PRESSED" });
     }
     if (e.key === "Escape") {
       this.sendToParent({ type: "ESCAPE_KEY_PRESSED" });
@@ -476,6 +514,35 @@ var BuilderSelectionManager = class {
       this.notifyPageReady();
     };
   }
+  /**
+   * Navigate to a new route with retry logic for HMR timing
+   * Used by AI agent to navigate after creating new pages
+   * Retries navigation if route doesn't exist yet (new page being created)
+   */
+  handleNavigate(path) {
+    console.log("[BuilderSelection] Navigate request:", path);
+    const MAX_RETRIES = 10;
+    const RETRY_DELAY_MS = 300;
+    let retries = 0;
+    const attemptNavigation = () => {
+      history.pushState({}, "", path);
+      window.dispatchEvent(new PopStateEvent("popstate"));
+      setTimeout(() => {
+        if (window.location.pathname === path) {
+          console.log("[BuilderSelection] Navigation successful:", path);
+          this.notifyPageReady();
+        } else if (retries < MAX_RETRIES) {
+          retries++;
+          console.log(`[BuilderSelection] Navigation retry ${retries}/${MAX_RETRIES} for path: ${path}`);
+          setTimeout(attemptNavigation, RETRY_DELAY_MS);
+        } else {
+          console.warn("[BuilderSelection] Navigation failed after max retries:", path);
+          this.notifyPageReady();
+        }
+      }, 100);
+    };
+    attemptNavigation();
+  }
   showHoverOverlay(element) {
     if (!this.hoverOverlay) return;
     const rect = element.getBoundingClientRect();
@@ -716,6 +783,154 @@ var BuilderSelectionManager = class {
     document.body.appendChild(container);
     this.selections.set(selectionId, { element, container, badge, border });
   }
+  /**
+   * Get all available routes from the page
+   * Looks for routes registered via wouter or standard link patterns
+   */
+  getAllRoutes() {
+    const wouterRoutes = window.__wouterRoutes;
+    if (wouterRoutes && Array.isArray(wouterRoutes)) {
+      return wouterRoutes;
+    }
+    const links = document.querySelectorAll('a[href^="/"]');
+    const routes = /* @__PURE__ */ new Set(["/"]);
+    links.forEach((link) => {
+      const href = link.getAttribute("href");
+      if (href && href.startsWith("/") && !href.includes("http")) {
+        const path = href.split("?")[0].replace(/\/$/, "") || "/";
+        routes.add(path);
+      }
+    });
+    const navLinks = document.querySelectorAll("nav a[href]");
+    navLinks.forEach((link) => {
+      const href = link.getAttribute("href");
+      if (href && href.startsWith("/")) {
+        const path = href.split("?")[0].replace(/\/$/, "") || "/";
+        routes.add(path);
+      }
+    });
+    return Array.from(routes).sort();
+  }
+  /**
+   * Capture full-page screenshots for all pages
+   * Non-blocking: uses requestIdleCallback between pages
+   * Transfers ArrayBuffer directly (zero-copy) for performance
+   */
+  async captureAllPagesScreenshots(checkpointId, viewport) {
+    console.log("[BuilderSelection] Starting multi-page capture for checkpoint:", checkpointId);
+    const routes = this.getAllRoutes();
+    const originalPath = window.location.pathname;
+    const total = routes.length;
+    this.sendToParent({
+      type: "CAPTURE_PROGRESS",
+      checkpointId,
+      completed: 0,
+      total
+    });
+    for (let i = 0; i < routes.length; i++) {
+      const pagePath = routes[i];
+      try {
+        await new Promise((resolve) => {
+          if ("requestIdleCallback" in window) {
+            requestIdleCallback(() => resolve(), { timeout: 1e3 });
+          } else {
+            setTimeout(resolve, 50);
+          }
+        });
+        if (window.location.pathname !== pagePath) {
+          await this.navigateAndWait(pagePath);
+        }
+        await new Promise((resolve) => setTimeout(resolve, 100));
+        const buffer = await this.capturePageAsArrayBuffer();
+        if (buffer) {
+          this.sendToParent(
+            {
+              type: "PAGE_SCREENSHOT_READY",
+              checkpointId,
+              pagePath,
+              buffer
+            },
+            [buffer]
+            // Transfer list - moves ownership, no copy
+          );
+        }
+        this.sendToParent({
+          type: "CAPTURE_PROGRESS",
+          checkpointId,
+          completed: i + 1,
+          total
+        });
+      } catch (error) {
+        console.error(`[BuilderSelection] Failed to capture page ${pagePath}:`, error);
+      }
+    }
+    if (window.location.pathname !== originalPath) {
+      await this.navigateAndWait(originalPath);
+    }
+    this.sendToParent({
+      type: "ALL_SCREENSHOTS_COMPLETE",
+      checkpointId
+    });
+    console.log("[BuilderSelection] Multi-page capture complete:", checkpointId);
+  }
+  /**
+   * Navigate to a path and wait for page ready
+   */
+  async navigateAndWait(path) {
+    return new Promise((resolve) => {
+      history.pushState({}, "", path);
+      window.dispatchEvent(new PopStateEvent("popstate"));
+      requestAnimationFrame(() => {
+        requestAnimationFrame(() => {
+          resolve();
+        });
+      });
+    });
+  }
+  /**
+   * Capture current page as ArrayBuffer (for zero-copy transfer)
+   */
+  async capturePageAsArrayBuffer() {
+    try {
+      const html2canvas = (await import("html2canvas-pro")).default;
+      const overlays = document.querySelectorAll(".builder-selection-container, #builder-hover-overlay");
+      overlays.forEach((el) => {
+        ;
+        el.style.display = "none";
+      });
+      const originalScrollY = window.scrollY;
+      window.scrollTo(0, 0);
+      await new Promise((resolve) => requestAnimationFrame(resolve));
+      const canvas = await html2canvas(document.body, {
+        scale: 0.5,
+        // Lower resolution for reasonable file size
+        logging: false,
+        useCORS: true,
+        allowTaint: true,
+        backgroundColor: "#ffffff",
+        windowHeight: document.body.scrollHeight,
+        // Full page height
+        height: document.body.scrollHeight
+      });
+      window.scrollTo(0, originalScrollY);
+      overlays.forEach((el) => {
+        ;
+        el.style.display = "";
+      });
+      const blob = await new Promise((resolve) => {
+        canvas.toBlob((b) => resolve(b), "image/jpeg", 0.6);
+      });
+      if (!blob) {
+        console.error("[BuilderSelection] Failed to create blob from canvas");
+        return null;
+      }
+      const buffer = await blob.arrayBuffer();
+      return buffer;
+    } catch (error) {
+      console.error("[BuilderSelection] Page capture failed:", error);
+      return null;
+    }
+  }
 };
 var instance = null;
 function initBuilderSelection() {

package/dist/plugin.js

@@ -10,6 +10,7 @@ function yoamigoPlugin(options = {}) {
       name: "yoamigo:content-hmr",
       handleHotUpdate({ file, server }) {
         if (!file.endsWith("/content.ts") && !file.endsWith("\\content.ts")) return;
+        console.log("[yoamigo:content-hmr] content.ts changed, sending HMR event to clients");
         server.ws.send({
           type: "custom",
           event: "yoamigo:content-update"
@@ -21,12 +22,15 @@ function yoamigoPlugin(options = {}) {
         const hmrCode = `
 // YoAmigo Content HMR - injected by yoamigo:content-hmr plugin
 if (import.meta.hot) {
+  console.log('[yoamigo:content-hmr] HMR listener registered')
   import.meta.hot.on('yoamigo:content-update', async () => {
+    console.log('[yoamigo:content-hmr] Received content update event')
     try {
       const { default: newContent } = await import(/* @vite-ignore */ './content?t=' + Date.now())
       const { registerContent } = await import('@yoamigo.com/core')
       registerContent(newContent)
       window.dispatchEvent(new CustomEvent('content-updated'))
+      console.log('[yoamigo:content-hmr] Content updated successfully')
     } catch (err) {
       console.error('[yoamigo:content-hmr] Failed to hot reload content:', err)
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@yoamigo.com/core",
-  "version": "0.3.8",
+  "version": "0.3.9",
   "description": "Core components, router, and utilities for YoAmigo templates",
   "type": "module",
   "license": "SEE LICENSE IN LICENSE",