Initial commit

jblotus · Oct 26, 2015 · bf35742 · bf35742
1 parent 370d5cd
commit bf35742
Show file tree

Hide file tree

Showing 312 changed files with 109,697 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -1,2 +1,17 @@
 # aws-lambda-wkhtmltopdf
 Convert HTML to PDF using Webkit (QtWebKit) on AWS Lambda
+
+## Input
+    {
+        "html" : "<!DOCTYPE html><html><body>Hello world</body></html>"
+    }
+
+## Output
+    {
+        "filename": "8rqj9td0pvjf9a4i.pdf"
+    }
+
+## Configuration
+
+1. Open `config.js` and set `dstBucket` variable to name of S3 bucket where you want function to save output PDF files.
+2. Make sure AWS Lambda function has `PutObject` access to S3 bucket
diff --git a/config.js b/config.js
@@ -0,0 +1,6 @@
+dstBucket = 'your-s3-bucket-name';
+
+process.env['PATH'] = process.env['PATH'] + ':' + process.env['LAMBDA_TASK_ROOT'];
+
+module.exports = function() {
+};
diff --git a/index.js b/index.js
@@ -0,0 +1,44 @@
+var wkhtmltopdf = require('wkhtmltopdf');
+var fs = require('fs');
+var AWS = require('aws-sdk');
+var config = require('./config.js');
+
+var s3 = new AWS.S3();
+
+exports.handler = function(event, context) {
+	return_data = {};
+	if (event.html) {
+
+		var output_filename = Math.random().toString(36).slice(2) + '.pdf';
+		var output = '/tmp/' + output_filename;
+
+		writeStream = fs.createWriteStream(output);
+
+		wkhtmltopdf(event.html, function(code, signal) {
+
+			s3.putObject({
+				Bucket : dstBucket,
+				Key : output_filename,
+				Body : fs.createReadStream(output),
+				ContentType : "application/pdf"
+			}, function(error, data) {
+
+				if (error != null) {
+					console.log("error: " + error);
+				} else {
+					console.log('upload done...');
+				}
+				return_data = {
+					filename : output_filename
+				};
+				// context.succeed("File has been uploaded");
+				context.done(null, return_data);
+			});
+
+		}).pipe(writeStream);
+	} else {
+		console.log('error');
+		context.done('unable to get the html', {});
+	}
+
+};
diff --git a/node_modules/aws-sdk/.jshintrc b/node_modules/aws-sdk/.jshintrc
diff --git a/node_modules/aws-sdk/.npmignore b/node_modules/aws-sdk/.npmignore
diff --git a/node_modules/aws-sdk/CONTRIBUTING.md b/node_modules/aws-sdk/CONTRIBUTING.md