mirror of https://github.com/jkjoy/sunpeiwen.git
70 lines
2.0 KiB
JavaScript
70 lines
2.0 KiB
JavaScript
|
'use strict';
|
||
|
|
|
||
|
|
var fs = require('fs');
|
||
|
|
var rimraf = require('rimraf');
|
||
|
|
var acorn = require('acorn');
|
||
|
|
var walk = require('acorn/dist/walk');
|
||
|
|
|
||
|
|
var ids = [];
|
||
|
|
var names = {};
|
||
|
|
|
||
|
|
function getIdFor(name) {
|
||
|
|
if (name in names) return names[name];
|
||
|
|
var id;
|
||
|
|
do {
|
||
|
|
id = '_' + Math.floor(Math.random() * 100);
|
||
|
|
} while (ids.indexOf(id) !== -1)
|
||
|
|
ids.push(id);
|
||
|
|
names[name] = id;
|
||
|
|
return id;
|
||
|
|
}
|
||
|
|
|
||
|
|
function fixup(src) {
|
||
|
|
var ast = acorn.parse(src);
|
||
|
|
src = src.split('');
|
||
|
|
walk.simple(ast, {
|
||
|
|
MemberExpression: function (node) {
|
||
|
|
if (node.computed) return;
|
||
|
|
if (node.property.type !== 'Identifier') return;
|
||
|
|
if (node.property.name[0] !== '_') return;
|
||
|
|
replace(node.property, getIdFor(node.property.name));
|
||
|
|
}
|
||
|
|
});
|
||
|
|
function source(node) {
|
||
|
|
return src.slice(node.start, node.end).join('');
|
||
|
|
}
|
||
|
|
function replace(node, str) {
|
||
|
|
for (var i = node.start; i < node.end; i++) {
|
||
|
|
src[i] = '';
|
||
|
|
}
|
||
|
|
src[node.start] = str;
|
||
|
|
}
|
||
|
|
return src.join('');
|
||
|
|
}
|
||
|
|
rimraf.sync(__dirname + '/lib/');
|
||
|
|
fs.mkdirSync(__dirname + '/lib/');
|
||
|
|
fs.readdirSync(__dirname + '/src').forEach(function (filename) {
|
||
|
|
var src = fs.readFileSync(__dirname + '/src/' + filename, 'utf8');
|
||
|
|
var out = fixup(src);
|
||
|
|
fs.writeFileSync(__dirname + '/lib/' + filename, out);
|
||
|
|
});
|
||
|
|
|
||
|
|
rimraf.sync(__dirname + '/domains/');
|
||
|
|
fs.mkdirSync(__dirname + '/domains/');
|
||
|
|
fs.readdirSync(__dirname + '/src').forEach(function (filename) {
|
||
|
|
var src = fs.readFileSync(__dirname + '/src/' + filename, 'utf8');
|
||
|
|
var out = fixup(src);
|
||
|
|
out = out.replace(/require\(\'asap\/raw\'\)/g, "require('asap')");
|
||
|
|
fs.writeFileSync(__dirname + '/domains/' + filename, out);
|
||
|
|
});
|
||
|
|
|
||
|
|
rimraf.sync(__dirname + '/setimmediate/');
|
||
|
|
fs.mkdirSync(__dirname + '/setimmediate/');
|
||
|
|
fs.readdirSync(__dirname + '/src').forEach(function (filename) {
|
||
|
|
var src = fs.readFileSync(__dirname + '/src/' + filename, 'utf8');
|
||
|
|
var out = fixup(src);
|
||
|
|
out = out.replace(/var asap \= require\(\'([a-z\/]+)\'\);/g, '');
|
||
|
|
out = out.replace(/asap/g, "setImmediate");
|
||
|
|
fs.writeFileSync(__dirname + '/setimmediate/' + filename, out);
|
||
|
|
});
|