sample image

Author: koustov

express/server.js

@@ -10,6 +10,7 @@ const bodyParser = require("body-parser");
 const path = require("path");
 const { UpdateBadges, MetaImage } = require("../models/routes/badges.js");
 const { HealthCheck, BaseRoute } = require("../models/routes/base.js");
+const { Screenshot } = require("../services/meta/snapshot-capture.js");
 
 // Configuration
 var app = express();
@@ -33,7 +34,9 @@ router.post("/badges", function (req, res) {
 });
 
 router.get("/:userId/badges", async function (req, res) {
-  MetaImage(req, res);
+  // MetaImage(req, res);
+  const base64 = await Screenshot();
+  res.end(base64);
 });
 
 app.use(bodyParser.json());

package.json

@@ -19,11 +19,12 @@
 		"json-graphql-parser": "^0.1.9",
 		"netlify-lambda": "^2.0.15",
 		"node-html-to-image": "^3.2.4",
+		"puppeteer": "^19.3.0",
 		"puppeteer-chromium-resolver": "^19.1.0",
 		"serverless-http": "^3.1.0",
 		"webpack-node-externals": "^3.0.0"
 	},
 	"pcr": {
-        "revision": "869685"
-    }
+		"revision": "869685"
+	}
 }

services/meta/snapshot-capture.js

@@ -0,0 +1,23 @@
+const puppeteer = require("puppeteer"); // Require Puppeteer moduleconst url = "https://www.testim.io/"; // Set website you want to screenshot
+const url = "https://www.testim.io/"; // Set website you want to screenshot
+
+const Screenshot = async () => {
+  // Define Screenshot function
+  console.log("Inside Screenshot");
+  const browser = await puppeteer.launch(); // Launch a "browser"
+  const page = await browser.newPage(); // Open new page
+  await page.goto(url); // Go website
+  await page.waitForSelector("#header > div > div > a.h-logo"); // Method to ensure that the element is loaded
+  const logo = await page.$("#header > div > div > a.h-logo"); // logo is the element you want to capture
+  await logo.screenshot({
+    path: "testim.png",
+  });
+  const base64 = await page.screenshot({ encoding: "base64" });
+  console.log("Result obtained");
+  await page.close(); // Close the website
+  await browser.close(); // Close the browser
+  return base64;
+};
+Screenshot();
+
+module.exports.Screenshot = Screenshot;