edits

Author: artyom-activeloop

lib/plugins/tokenAuth.js

@@ -0,0 +1,13 @@
+module.exports = {
+  requestReceived: (req, res, next) => {
+    let token = req.headers['x-prerender-token'];
+    console.log('token: ', token)
+    if (
+      !token ||
+      token !== process.env.AUTH_TOKEN
+    )
+      return res.send(401);
+
+    return next();
+  },
+};

server.js

@@ -1,10 +1,14 @@
 #!/usr/bin/env node
 const prerender = require('./lib');
+const util = require('./lib/util');
+
+const browserLocation = process.env.BROWSER || '/usr/bin/chromium-browser';
+util.log(`browserLocation: ${browserLocation}`)
 
 const server = prerender({
     chromeFlags: ['--no-sandbox', '--headless', '--disable-gpu', '--remote-debugging-port=9222', '--hide-scrollbars', '--disable-dev-shm-usage'],
     forwardHeaders: true,
-    chromeLocation: '/usr/bin/chromium-browser'
+    chromeLocation: browserLocation
 });
 
 
@@ -14,6 +18,15 @@ server.use(prerender.browserForceRestart());
 server.use(prerender.addMetaTags());
 server.use(prerender.removeScriptTags());
 server.use(prerender.httpHeaders());
-server.use(prerender.s3Cache());
+
+if (process.env.S3_BUCKET_NAME) {
+  util.log("EnabledPlugin: s3Cache")
+  server.use(prerender.s3Cache());
+}
+
+if (process.env.AUTH_TOKEN) {
+  util.log("EnabledPlugin: tokenAuth")
+  server.use(prerender.tokenAuth()); 
+}
 
 server.start();