@yashwant.dharmdas/elementor-mcp 3.2.2 → 3.2.3

package/dist/index.js

@@ -993,8 +993,9 @@ 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)"),
@@ -1003,16 +1004,47 @@ function createMcpServer(sites) {
         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, wait, auto_scroll, 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",
@@ -1104,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.2",
+  "version": "3.2.3",
   "description": "MCP server for controlling Elementor via Claude — supports multiple WordPress sites",
   "main": "dist/index.js",
   "type": "module",