First version

Author: fishcharlie

.gitignore

@@ -0,0 +1 @@
+node_modules

index.js

@@ -0,0 +1,59 @@
+const fs = require('fs');
+const path = require('path');
+const util = require('util');
+const exec = util.promisify(require('child_process').exec);
+const axios = require('axios');
+const version = require('utils/version_cleaner')(process.args[process.args.length - 1]);
+const downloadURL = `https://nodejs.org/dist/v${version}/node-v${version}-linux-x64.tar.xz`;
+
+async function main() {
+	let runtimeFileNumber;
+	while (fs.existsSync(path.join(`node_runtime${runtimeFileNumber ? `_${runtimeFileNumber}` : ""}.js`))) {
+		if (runtimeFileNumber) {
+			runtimeFileNumber++;
+		} else {
+			runtimeFileNumber = 0;
+		}
+	}
+	const nodeRunnerFile = `node_runtime${runtimeFileNumber ? `_${runtimeFileNumber}` : ""}.js`;
+	fs.copyFileSync(path.join(__dirname, 'templates', 'node_runtime.js'), path.join(nodeRunnerFile));
+	fs.copyFileSync(path.join(__dirname, 'templates', 'bootstrap'), path.join(`bootstrap`));
+
+	let bootstrap = fs.readFileSync(path.join(`bootstrap`), 'utf8').replace(/{{NODE_VERSION}}/g, version).replace(/{{NODE_VERSION}}/g, version).replace(/{{NODE_RUNNER_FILE}}/g, nodeRunnerFile);
+	fs.writeFileSync(path.join(`bootstrap`), bootstrap);
+
+	// Download Node.js
+	try {
+		await downloadFile(downloadURL, path.join(`node-v${version}-linux-x64.tar.xz`));
+		await unzip(path.join(`node-v${version}-linux-x64.tar.xz`));
+		fs.unlinkSync(path.join(`node-v${version}-linux-x64.tar.xz`));
+	} catch (e) {
+		console.error(e);
+	}
+}
+main();
+
+
+async function downloadFile(url, path) {
+	const response = await axios({
+		method: 'GET',
+		url,
+		responseType: 'stream'
+	});
+
+	response.data.pipe(fs.createWriteStream(path));
+
+	return new Promise((resolve, reject) => {
+		response.data.on('end', () => {
+			resolve();
+		});
+
+		response.data.on('error', () => {
+			reject();
+		});
+	});
+}
+
+async function unzip(file) {
+	const { stdout, stderr } = await exec(`tar -xJf ${file}`);
+}

package-lock.json

@@ -1,12 +1,15 @@
 {
   "name": "aws-lambda-custom-node-runtime",
-  "version": "1.0.0",
+  "version": "0.0.1",
   "description": "",
   "main": "index.js",
   "scripts": {
     "test": "echo \"Error: no test specified\" && exit 1"
   },
   "keywords": [],
   "author": "rrainn, Inc.",
-  "license": "MIT"
+  "license": "MIT",
+  "dependencies": {
+    "axios": "^0.18.0"
+  }
 }

package.json

@@ -0,0 +1,3 @@
+#!/bin/sh
+cd $LAMBDA_TASK_ROOT
+./node-v{{NODE_VERSION}}-linux-x64/bin/node {{NODE_RUNNER_FILE}}

templates/bootstrap

@@ -0,0 +1,83 @@
+var http = require('http');
+var Buffer = require('buffer').Buffer;
+
+function run() {
+	request({
+		url: process.env.AWS_LAMBDA_RUNTIME_API + '/2018-06-01/runtime/invocation/next',
+	}, function(err, invoke_result) {
+		if (err) {
+			return request({
+				url: process.env.AWS_LAMBDA_RUNTIME_API + '/2018-06-01/runtime/init/error',
+				method: 'POST',
+				data: err
+			}, run);
+		}
+		var event_data = invoke_result.data;
+		var request_id = invoke_result.resp.headers['lambda-runtime-aws-request-id'];
+
+		var response = require(process.env.LAMBDA_TASK_ROOT + '/' + process.env._HANDLER.split('.')[0] + '.js')[process.env._HANDLER.split('.')[1]](JSON.parse(event_data), {}, function (err, result) {
+			if (err) {
+				failure(err);
+			} else {
+				success(result);
+			}
+		});
+		if (response && response.then && typeof response.then === 'function') {
+			response.then(success);
+		}
+		if (response && response.catch && typeof response.catch === 'function') {
+			response.catch(failure);
+		}
+
+		function success(result) {
+			request({
+				url: process.env.AWS_LAMBDA_RUNTIME_API + '/2018-06-01/runtime/invocation/' + request_id + '/response',
+				method: 'POST',
+				data: result
+			}, run);
+		}
+		function failure(err) {
+			request({
+				url: process.env.AWS_LAMBDA_RUNTIME_API + '/2018-06-01/runtime/invocation/' + request_id + '/error',
+				method: 'POST',
+				data: err
+			}, run);
+		}
+	});
+}
+run();
+
+function request(options, cb) {
+	if (!cb) {
+		cb = function(){};
+	}
+	if (options.data && typeof options.data === 'object') {
+		options.data = JSON.stringify(options.data);
+	}
+	if (options.data && !options.headers) {
+		options.headers = {};
+	}
+	if (options.data && !options.headers['Content-Length']) {
+		options.headers['Content-Length'] = Buffer.byteLength(options.data);
+	}
+	if (options.data && !options.headers['Content-Type']) {
+		options.headers['Content-Type'] = 'application/x-www-form-urlencoded';
+	}
+	options.timeout = 1000;
+	var req = http.request('http://' + options.url, options, function(resp) {
+		var data = '';
+		resp.on('data', function(chunk) {
+			data += chunk;
+		});
+		resp.on('end', function() {
+			cb(null, {data: data, resp: resp});
+		});
+	}).on('error', function(err) {
+		cb(err);
+	});
+
+	if (options.data) {
+		req.write(options.data);
+	}
+	req.end();
+}

templates/node_runtime.js

@@ -0,0 +1,8 @@
+module.exports = version => {
+	version = version.replace(/^v/g, "");
+	let versionArray = version.split(".");
+	while (versionArray.length < 3) {
+		versionArray.push("0");
+	}
+	return versionArray.join(".");
+};