add docker support

iamdual · iamdual · commit d60299bb6c5e · 2023-03-30T00:01:10.000+03:00
diff --git a/.dockerignore b/.dockerignore
@@ -0,0 +1 @@
+node_modules
diff --git a/Dockerfile b/Dockerfile
@@ -0,0 +1,21 @@
+FROM alpine:3.17.1
+
+RUN apk add --no-cache \
+    chromium \
+    nss \
+    freetype \
+    harfbuzz \
+    ca-certificates \
+    ttf-freefont \
+    nodejs \
+    npm
+
+ENV PUPPETEER_SKIP_CHROMIUM_DOWNLOAD=true
+ENV PUPPETEER_EXECUTABLE_PATH=/usr/bin/chromium-browser
+ENV HTML2PDF_NO_SANDBOX=true
+
+WORKDIR /app
+COPY ["package.json", "package-lock.json", "./"]
+RUN npm install --production
+COPY . .
+CMD ["npm", "start"]
diff --git a/README.md b/README.md
@@ -1,35 +1,10 @@
 # html2pdf
-HTML to PDF converter provided by command line interface and RESTful API. Powered by `puppeteer`.
-
-## Installing
-```
-npm install -g https://github.com/iamdual/html2pdf
-```
+An HTML to PDF converter provided by RESTful API and command-line tool. Powered by `puppeteer`.
 
 ## Usage
 
-### Command-line
-```bash
-html2pdf '<!DOCTYPE html><strong>Hello world!</strong>' -o output.pdf
-
-html2pdf "https://www.google.com" --format A4 --timeout 10 -o google.pdf
-```
-
-### Library
-```js
-const html2pdf = require("@iamdual/html2pdf");
-const config = html2pdf.config({source: 'https://google.com', format: 'A4', timeout: 10});
-html2pdf.generate(config).then(pdf => {
-    console.log('PDF generated!');
-    require('fs').writeFile('google.pdf', pdf, 'binary', err => {});
-});
-```
-
 ### RESTful API
 ```bash
-git clone https://github.com/iamdual/html2pdf
-npm start
-
 curl -X POST 'http://localhost:3000/generate' \
     --header 'Content-Type: application/json' \
     --data '{
@@ -39,6 +14,13 @@ curl -X POST 'http://localhost:3000/generate' \
     }' > google.pdf
 ```
 
+### Command-line
+```bash
+npm install -g https://github.com/iamdual/html2pdf
+html2pdf '<!DOCTYPE html><strong>Hello world!</strong>' -o output.pdf
+html2pdf "https://www.google.com" --format A4 --timeout 10 -o google.pdf
+```
+
 ## Parameters
 | Name       | Type    | Description                                                                                                                         |
 |------------|---------|-------------------------------------------------------------------------------------------------------------------------------------|
diff --git a/src/generator/index.js b/src/generator/index.js
@@ -5,14 +5,18 @@
  */
 
 const puppeteer = require('puppeteer');
-const chromiumPath = process.env.CHROMIUM_PATH ? process.env.CHROMIUM_PATH : undefined;
+
+let args = [];
+if (process.env.HTML2PDF_NO_SANDBOX) {
+    args.push('--no-sandbox')
+}
 
 module.exports = async function generator(config) {
 
     // Create a browser instance
     const browser = await puppeteer.launch({
         headless: true,
-        executablePath: chromiumPath,
+        args: args
     });
 
     // Create a new page
diff --git a/src/rest-api/generate.js b/src/rest-api/generate.js
@@ -10,6 +10,11 @@ const generator = require('../generator');
 
 module.exports = async function generate(req, res) {
 
+    if (!['GET', 'POST'].includes(req.method)) {
+        res.status(405).json({ error: 'INVALID_METHOD', errorMessage: 'Method not allowed.' });
+        return;
+    }
+
     let config;
     if (req.method === 'POST') {
         config = configParser(req.body);
@@ -32,6 +37,9 @@ module.exports = async function generate(req, res) {
             res.status(408).json({ error: 'TIMEOUT', errorMessage: 'Request timeout.' });
             return;
         }
+        if (process.env.NODE_ENV === 'development') {
+            console.log(error);
+        }
         res.status(500).json({ error: 'SERVER_ERROR', errorMessage: 'We are unable to fulfill your request.' });
     }
 }
diff --git a/src/rest-api/index.js b/src/rest-api/index.js
@@ -9,13 +9,14 @@ const app = express();
 app.use(express.json());
 app.use((err, req, res, next) => {
     if (err instanceof SyntaxError && err.status === 400 && 'body' in err) {
-        return res.status(400).send({ error: 'INVALID_REQUEST', errorMessage: err.message });
+        res.status(400).send({ error: 'INVALID_REQUEST', errorMessage: err.message });
+        return;
     }
     next();
 });
 app.all('/generate', require('./generate'));
 
-const port = process.env.LISTEN_PORT || 3000;
+const port = process.env.HTML2PDF_API_PORT || 3000;
 app.listen(port, function () {
     console.log('Server has been started on http://localhost:%d', port);
 });

Original file line number	Diff line number	Diff line change
`@@ -10,6 +10,11 @@ const generator = require('../generator');`
`10`	`10`
`11`	`11`	`module.exports = async function generate(req, res) {`
`12`	`12`
	`13`	`+ if (!['GET', 'POST'].includes(req.method)) {`
	`14`	`+ res.status(405).json({ error: 'INVALID_METHOD', errorMessage: 'Method not allowed.' });`
	`15`	`+ return;`
	`16`	`+ }`
	`17`	`+`
`13`	`18`	`let config;`
`14`	`19`	`if (req.method === 'POST') {`
`15`	`20`	`config = configParser(req.body);`
`@@ -32,6 +37,9 @@ module.exports = async function generate(req, res) {`
`32`	`37`	`res.status(408).json({ error: 'TIMEOUT', errorMessage: 'Request timeout.' });`
`33`	`38`	`return;`
`34`	`39`	`}`
	`40`	`+ if (process.env.NODE_ENV === 'development') {`
	`41`	`+ console.log(error);`
	`42`	`+ }`
`35`	`43`	`res.status(500).json({ error: 'SERVER_ERROR', errorMessage: 'We are unable to fulfill your request.' });`
`36`	`44`	`}`
`37`	`45`	`}`