diff --git a/index.js b/index.js index 382a14b..640413e 100644 --- a/index.js +++ b/index.js @@ -1,4 +1,5 @@ const puppeteer = require('puppeteer'); +const containerized = require('containerized'); const MY_PACKAGE_SELECTOR_ID = 'table[style="margin-top:-10px;"] tr:first-child+tr'; @@ -36,8 +37,14 @@ async function getUsage() { } function chromeLaunchConfig() { + let defaultArgs = []; + if (containerized()) { + defaultArgs = ['--no-sandbox', '--disable-setuid-sandbox']; + } var options = { - args: ['--no-sandbox', '--disable-setuid-sandbox'].concat( + // These are set for Docker usage + // https://github.com/alekzonder/docker-puppeteer#before-usage + args: defaultArgs.concat( process.env.hasOwnProperty('PROXY_SERVER') ? [`--proxy-server=${process.env['PROXY_SERVER']}`] : [] diff --git a/package-lock.json b/package-lock.json index b54df5b..f89c883 100644 --- a/package-lock.json +++ b/package-lock.json @@ -57,6 +57,11 @@ "typedarray": "^0.0.6" } }, + "containerized": { + "version": "1.0.2", + "resolved": "https://registry.npmjs.org/containerized/-/containerized-1.0.2.tgz", + "integrity": "sha1-3+0xSeq1BC7wsG6FHH2PTcZ1DYg=" + }, "core-util-is": { "version": "1.0.2", "resolved": "https://registry.npmjs.org/core-util-is/-/core-util-is-1.0.2.tgz", diff --git a/package.json b/package.json index 870530f..47950be 100644 --- a/package.json +++ b/package.json @@ -6,10 +6,15 @@ "scripts": { "test": "echo \"Error: no test specified\" && exit 1" }, - "keywords": ["actcorp", "balance", "flexibyte"], + "keywords": [ + "actcorp", + "balance", + "flexibyte" + ], "author": "Nemo ", "license": "MIT", "dependencies": { + "containerized": "^1.0.2", "prom-client": "^11.0.0", "puppeteer": "^1.4.0" }