@yashwant.dharmdas/elementor-mcp 3.2.1 → 3.2.3

package/dist/index.js

@@ -993,24 +993,58 @@ function createMcpServer(sites) {
         }
     });
     // ── Group 9: Screenshot ──────────────────────────────────────────────────────
-    server.tool("screenshot-page", "Take a screenshot of a published WordPress page using system Chrome. Saves the file to disk and returns both a preview image and the saved file path. Pass the file_path to Basecamp MCP's upload_attachment tool to attach it — never relay the raw image data yourself.", {
-        page_id: z.string().describe("WordPress Page ID"),
+    server.tool("screenshot-page", "Take a screenshot of a published WordPress page using system Chrome. Works with any post type (page, post, custom post types like astra-portfolio). Pass either page_id OR a direct url. Saves the file to disk and returns both a preview image and the saved file path. Pass the file_path to Basecamp MCP's upload_attachment tool to attach it.", {
+        page_id: z.string().optional().describe("WordPress post/page ID (works for any post type). Either page_id OR url must be provided."),
+        url: z.string().optional().describe("Direct URL to screenshot (e.g. https://site.com/services/xyz/). Use this for custom post types or external URLs. Takes precedence over page_id."),
         full_page: z.boolean().optional().describe("Capture full scrollable page (default: true)"),
         width: z.number().optional().describe("Viewport width in pixels (default: 1440)"),
         format: z.enum(["jpeg", "png"]).optional().describe("Image format — jpeg is much smaller and recommended (default: jpeg)"),
         quality: z.number().min(1).max(100).optional().describe("JPEG quality 1–100 (default: 82). Lower = smaller file size."),
         max_height: z.number().optional().describe("Cap the captured height in pixels (e.g. 4000). Useful for very long pages to stay under size limits."),
+        wait: z.number().optional().describe("Extra milliseconds to wait after page load + auto-scroll before capturing (default: 2000). Increase for heavy animations/videos."),
+        auto_scroll: z.boolean().optional().describe("Scroll through the page to trigger lazy-loaded images and content (default: true). Disable only for pages with infinite scroll."),
         site: siteParam,
-    }, async ({ page_id, full_page, width, format, quality, max_height, site }) => {
+    }, async ({ page_id, url, full_page, width, format, quality, max_height, wait, auto_scroll, site }) => {
         try {
+            if (!page_id && !url) {
+                throw new Error("Provide either page_id or url.");
+            }
             const { wpUrl, authHeader } = resolveSite(sites, site);
-            // ── 1. Resolve the page's public URL ─────────────────────────────
-            const pageRes = await axios.get(`${wpUrl}/wp-json/wp/v2/pages/${page_id}`, {
-                headers: { Authorization: authHeader },
-            });
-            const pageUrl = pageRes.data.link;
+            // ── 1. Resolve the target URL ────────────────────────────────────
+            let pageUrl;
+            if (url) {
+                pageUrl = url;
+            }
+            else {
+                // Try /pages/ first (most common), then fall back to universal search
+                // (search endpoint works for ANY post type: page, post, astra-portfolio, etc.)
+                try {
+                    const r = await axios.get(`${wpUrl}/wp-json/wp/v2/pages/${page_id}`, {
+                        headers: { Authorization: authHeader },
+                    });
+                    pageUrl = r.data.link;
+                }
+                catch (e) {
+                    if (e.response?.status === 404) {
+                        // Fallback: universal search endpoint — supports all post types
+                        const s = await axios.get(`${wpUrl}/wp-json/wp/v2/search`, {
+                            params: { include: page_id },
+                            headers: { Authorization: authHeader },
+                        });
+                        if (!Array.isArray(s.data) || s.data.length === 0 || !s.data[0].url) {
+                            throw new Error(`Could not resolve URL for post ID ${page_id}. ` +
+                                `Tried /wp/v2/pages/${page_id} (404) and /wp/v2/search?include=${page_id} (no results). ` +
+                                `Pass the 'url' parameter directly instead.`);
+                        }
+                        pageUrl = s.data[0].url;
+                    }
+                    else {
+                        throw e;
+                    }
+                }
+            }
             if (!pageUrl)
-                throw new Error("Could not resolve public URL for this page.");
+                throw new Error("Could not resolve URL.");
             // ── 2. Detect system Chrome ───────────────────────────────────────
             const chromePaths = [
                 "C:\\Program Files\\Google\\Chrome\\Application\\chrome.exe",
@@ -1054,8 +1088,44 @@ function createMcpServer(sites) {
                 const vpWidth = width ?? 1440;
                 const vpHeight = max_height ?? 900;
                 await page.setViewport({ width: vpWidth, height: vpHeight, deviceScaleFactor: 1 });
-                await page.goto(pageUrl, { waitUntil: "networkidle2", timeout: 30_000 });
-                await new Promise(resolve => setTimeout(resolve, 1000));
+                // Navigate and wait for network to go idle
+                await page.goto(pageUrl, { waitUntil: "networkidle2", timeout: 45_000 });
+                // ── Auto-scroll: trigger lazy-loaded images/content ─────────────
+                if (auto_scroll !== false) {
+                    await page.evaluate(async () => {
+                        await new Promise((resolve) => {
+                            let totalHeight = 0;
+                            const distance = 200;
+                            const timer = setInterval(() => {
+                                const scrollHeight = document.body.scrollHeight;
+                                window.scrollBy(0, distance);
+                                totalHeight += distance;
+                                if (totalHeight >= scrollHeight) {
+                                    clearInterval(timer);
+                                    resolve();
+                                }
+                            }, 80);
+                        });
+                    });
+                    // Scroll back to top for the capture
+                    await page.evaluate(() => window.scrollTo(0, 0));
+                }
+                // ── Wait for all <img> tags to finish loading ──────────────────
+                await page.evaluate(async () => {
+                    const images = Array.from(document.querySelectorAll("img"));
+                    await Promise.all(images.map((img) => {
+                        if (img.complete && img.naturalWidth > 0)
+                            return Promise.resolve();
+                        return new Promise((resolve) => {
+                            img.addEventListener("load", () => resolve(), { once: true });
+                            img.addEventListener("error", () => resolve(), { once: true });
+                            // Safety timeout per image
+                            setTimeout(() => resolve(), 5000);
+                        });
+                    }));
+                });
+                // Extra settle time for fonts, animations, videos
+                await new Promise(resolve => setTimeout(resolve, wait ?? 2000));
                 const fmt = format ?? "jpeg";
                 const screenshotOptions = { type: fmt, fullPage: !max_height && (full_page ?? true) };
                 if (fmt === "jpeg")
@@ -1066,7 +1136,8 @@ function createMcpServer(sites) {
                 const buf = Buffer.isBuffer(screenshotBuffer) ? screenshotBuffer : Buffer.from(screenshotBuffer);
                 // ── 5. Save to disk ─────────────────────────────────────────────
                 const ext = fmt === "jpeg" ? "jpg" : "png";
-                const filename = `page_${page_id}_${Date.now()}.${ext}`;
+                const idPart = page_id ?? (pageUrl.replace(/https?:\/\//, "").replace(/[^a-z0-9]/gi, "_").slice(0, 40));
+                const filename = `page_${idPart}_${Date.now()}.${ext}`;
                 const filePath = path.join(screenshotsDir, filename);
                 fs.writeFileSync(filePath, buf);
                 const sizeKB = Math.round(buf.length / 1024);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@yashwant.dharmdas/elementor-mcp",
-  "version": "3.2.1",
+  "version": "3.2.3",
   "description": "MCP server for controlling Elementor via Claude — supports multiple WordPress sites",
   "main": "dist/index.js",
   "type": "module",