buble/register.js

var fs = require('fs');
var path = require('path');
var crypto = require('crypto');
var homedir = require('os').homedir;
var buble = require('./');

var original = require.extensions[ '.js' ];
var nodeModulesPattern = path.sep === '/' ? /\/node_modules\// : /\\node_modules\\/;

var nodeVersion = /(?:0\.)?\d+/.exec(process.version)[0];
var versions = [ '0.10', '0.12', '4', '5', '6' ];

if (!~versions.indexOf(nodeVersion)) {
	if (+nodeVersion > 6) {
		nodeVersion = '6';
	} else {
		throw new Error('Unsupported version (' + nodeVersion + '). Please raise an issue at https://github.com/bublejs/buble/issues');
	}
}

var options = {
	target: {
		node: nodeVersion
	}
};

function mkdirp (dir) {
	var parent = path.dirname(dir);
	if (dir === parent) return;
	mkdirp(parent);

	try {
		fs.mkdirSync(dir);
	} catch (err) {
		if (err.code !== 'EEXIST') throw err;
	}
}

var home = homedir();
if (home) {
	var cachedir = path.join(home, '.buble-cache', String(nodeVersion));
	mkdirp(cachedir);
	fs.writeFileSync(path.join(home, '.buble-cache/README.txt'), 'These files enable a faster startup when using buble/register. You can safely delete this folder at any time. See https://buble.surge.sh/guide/ for more information.');
}

var optionsStringified = JSON.stringify(options);

require.extensions[ '.js' ] = function (m, filename) {
	if (nodeModulesPattern.test(filename)) return original(m, filename);

	var source = fs.readFileSync(filename, 'utf-8');
	var hash = crypto.createHash('sha256');
	hash.update(buble.VERSION);
	hash.update(optionsStringified);
	hash.update(source);
	var key = hash.digest('hex') + '.json';
	var cachepath = path.join(cachedir, key);

	var compiled;

	if (cachedir) {
		try {
			compiled = JSON.parse(fs.readFileSync(cachepath, 'utf-8'));
		} catch (err) {
			// noop
		}
	}

	if (!compiled) {
		try {
			compiled = buble.transform(source, options);

			if (cachedir) {
				fs.writeFileSync(cachepath, JSON.stringify(compiled));
			}
		} catch (err) {
			if (err.snippet) {
				console.log('Error compiling ' + filename + ':\n---');
				console.log(err.snippet);
				console.log(err.message);
				console.log('');
				process.exit(1);
			}

			throw err;
		}
	}

	m._compile('"use strict";\n' + compiled.code, filename);
};