use different browser based on env

Author: Sherlockboy

package-lock.json

@@ -12,8 +12,11 @@
     "@google/generative-ai": "^0.2.1",
     "axios": "^1.6.7",
     "cheerio": "^1.0.0-rc.12",
+    "chrome-aws-lambda": "^10.1.0",
     "dotenv": "^16.4.5",
-    "express": "^4.18.3",
+    "express": "^4.18.3"
+  },
+  "devDependencies": {
     "puppeteer": "^22.3.0"
   }
-}
+}

package.json

@@ -0,0 +1,12 @@
+function isEnvLocal() {
+  return process.env.APP_ENV === "local";
+}
+
+function isEnvProd() {
+  return process.env.APP_ENV === "production";
+}
+
+module.exports = {
+  isEnvLocal,
+  isEnvProd,
+};

src/helpers.js

@@ -1,9 +1,16 @@
 const axios = require("axios").default;
 const cheerio = require("cheerio");
-const puppeteer = require("puppeteer");
+const helpers = require("./helpers");
 
 async function scrapeSpaContent(url, element) {
-  const browser = await puppeteer.launch();
+  let browser;
+
+  if (helpers.isEnvProd()) {
+    browser = await require("chrome-aws-lambda").puppeteer.launch();
+  } else {
+    browser = await require("puppeteer").launch();
+  }
+
   const page = await browser.newPage();
 
   await page.goto(url, { waitUntil: "networkidle2" });