@yashwant.dharmdas/elementor-mcp 3.2.2 → 3.2.4

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",
@@ -1095,43 +1127,65 @@ function createMcpServer(sites) {
                 // 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")
-                    screenshotOptions.quality = quality ?? 82;
+                const baseOptions = { type: fmt, fullPage: !max_height && (full_page ?? true) };
                 if (max_height)
-                    screenshotOptions.clip = { x: 0, y: 0, width: vpWidth, height: vpHeight };
-                const screenshotBuffer = await page.screenshot(screenshotOptions);
-                const buf = Buffer.isBuffer(screenshotBuffer) ? screenshotBuffer : Buffer.from(screenshotBuffer);
+                    baseOptions.clip = { x: 0, y: 0, width: vpWidth, height: vpHeight };
+                // ── Adaptive quality: always fit under the MCP 1MB tool-result limit ──
+                // (base64 overhead ~33%, so raw buffer must be ≤ ~780KB)
+                const PREVIEW_LIMIT = 750 * 1024;
+                let buf;
+                let usedQuality = undefined;
+                if (fmt === "jpeg") {
+                    // Try user-requested quality first; fall back progressively lower until it fits.
+                    const q0 = quality ?? 82;
+                    const qualitySteps = Array.from(new Set([q0, 70, 55, 40, 25, 15])).filter(q => q <= q0);
+                    for (const q of qualitySteps) {
+                        const raw = await page.screenshot({ ...baseOptions, quality: q });
+                        buf = Buffer.isBuffer(raw) ? raw : Buffer.from(raw);
+                        usedQuality = q;
+                        if (buf.length <= PREVIEW_LIMIT)
+                            break;
+                    }
+                }
+                else {
+                    const raw = await page.screenshot(baseOptions);
+                    buf = Buffer.isBuffer(raw) ? raw : Buffer.from(raw);
+                }
+                buf = buf;
                 // ── 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);
                 const mimeType = `image/${fmt}`;
-                // ── 6. Return image preview if small enough, always return path ─
-                const PREVIEW_LIMIT = 750 * 1024; // 750 KB raw → ~1 MB base64
-                const content = [
-                    {
-                        type: "text",
-                        text: [
-                            `✅ Screenshot saved successfully.`,
-                            `📁 file_path: ${filePath}`,
-                            `📐 Size: ${sizeKB} KB  |  Format: ${ext.toUpperCase()}  |  URL: ${pageUrl}`,
-                            ``,
-                            `👉 To upload to Basecamp, call upload_attachment with:`,
-                            `   file_path = "${filePath}"`,
-                            `   project_id = <your Basecamp project ID>`,
-                        ].join("\n"),
-                    },
-                ];
+                const qualityNote = usedQuality !== undefined && usedQuality !== (quality ?? 82)
+                    ? `  |  Auto-downscaled to quality ${usedQuality} to fit MCP limit`
+                    : "";
+                // ── 6. Always return the image so Claude can analyse it ──────
+                const content = [];
                 if (buf.length <= PREVIEW_LIMIT) {
-                    content.unshift({
+                    content.push({
                         type: "image",
                         data: buf.toString("base64"),
                         mimeType,
                     });
                 }
+                content.push({
+                    type: "text",
+                    text: [
+                        `✅ Screenshot captured and saved.`,
+                        `📁 file_path: ${filePath}`,
+                        `📐 Size: ${sizeKB} KB  |  Format: ${ext.toUpperCase()}${qualityNote}`,
+                        `🌐 URL: ${pageUrl}`,
+                        buf.length > PREVIEW_LIMIT
+                            ? `\n⚠️  Image still exceeds the 1MB MCP return limit even at lowest quality. The file is saved to disk and can be used via upload_attachment(file_path).`
+                            : ``,
+                        ``,
+                        `👉 To upload to Basecamp: upload_attachment with file_path = "${filePath}"`,
+                    ].filter(Boolean).join("\n"),
+                });
                 return { content };
             }
             finally {

package/package.json

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