面試官: 你了解過Babel嗎？寫過Babel插件嗎? 答: 沒有。卒
面試大廠，其中有那么一個(gè)問題:
1.你了解過Babel嗎？
了解過抽象語法樹，又稱AST，有學(xué)習(xí)過，也寫過一個(gè)基于AST的乞丐版模板引擎，先是詞法解析token，然后生產(chǎn)抽象語法樹，然后更改抽象語法樹，當(dāng)然這是插件做的事情，最后根據(jù)新的AST生成代碼。
2.寫過Babel插件嗎
沒有，只是看過相關(guān)文檔
3.如果讓你寫一個(gè)插件，你能寫的出來嗎?
啊，這，我試試！
遂卒....
開玩笑的，既然提到了，又沒回答上來什么，哎喲我這暴脾氣，一想到今晚就睡不著，連夜把它擼了。
那么我們來從零寫個(gè)插件吧。
寫一個(gè)預(yù)計(jì)算簡單表達(dá)式的插件
預(yù)覽
Before:
const result = 1 + 2 + 3 + 4 + 5;
After:
const result = 15;
以上的例子可能大家不會(huì)經(jīng)常遇到，因?yàn)樯祒才會(huì)這么寫，但是有可能你會(huì)這么寫
setTimeout(function(){
// do something
}, 1000 * 2) // 插件要做的事，就是把 1000 * 2 替換成 2000
開工
在寫代碼之前，你需要明白Babel它的原理，簡單點(diǎn)說：Babel解析成AST，然后插件更改AST，最后由Babel輸出代碼。
那么Babel的插件模塊需要你暴露一個(gè)function，function內(nèi)返回visitor。
module.export = function(babel){
return {
visitor:{
}
}
}
visitor是對(duì)各類型的AST節(jié)點(diǎn)做處理的地方，那么我們?cè)趺粗繠abel生成了的AST有哪些節(jié)點(diǎn)呢？
很簡單，你可以把Babel轉(zhuǎn)換的結(jié)果打印出來，或者這里有傳送門: AST explorer：https://astexplorer.net/
這里我們看到 const result = 1 + 2中的1 + 1是一個(gè)BinaryExpression節(jié)點(diǎn)，那么在visitor中，我們就處理這個(gè)節(jié)點(diǎn)
var babel = require('babel-core');
var t = require('babel-types');
const visitor = {
BinaryExpression(path) {
const node = path.node;
let result;
// 判斷表達(dá)式兩邊，是否都是數(shù)字
if (t.isNumericLiteral(node.left) && t.isNumericLiteral(node.right)) {
// 根據(jù)不同的操作符作運(yùn)算
switch (node.operator) {
case "+":
result = node.left.value + node.right.value;
break
case "-":
result = node.left.value - node.right.value;
break;
case "*":
result = node.left.value * node.right.value;
break;
case "/":
result = node.left.value / node.right.value;
break;
case "**":
let i = node.right.value;
while (--i) {
result = result || node.left.value;
result = result * node.left.value;
}
break;
default:
}
}
// 如果上面的運(yùn)算有結(jié)果的話
if (result !== undefined) {
// 把表達(dá)式節(jié)點(diǎn)替換成number字面量
path.replaceWith(t.numericLiteral(result));
}
}
};
module.exports = function (babel) {
return {
visitor
};
}
插件寫好了，我們運(yùn)行下插件試試
const babel = require("babel-core");
const result = babel.transform("const result = 1 + 2;",{
plugins:[
require("./index")
]
});
console.log(result.code); // const result = 3;
與預(yù)期一致，那么轉(zhuǎn)換 const result = 1 + 2 + 3 + 4 + 5;呢?
結(jié)果是: const result = 3 + 3 + 4 + 5;
這就奇怪了，為什么只計(jì)算了1 + 2之后，就沒有繼續(xù)往下運(yùn)算了?
我們看一下這個(gè)表達(dá)式的AST樹
你會(huì)發(fā)現(xiàn)Babel解析成表達(dá)式里面再嵌套表達(dá)式。
表達(dá)式( 表達(dá)式( 表達(dá)式( 表達(dá)式(1 + 2) + 3) + 4) + 5)
而我們的判斷條件并不符合所有的，只符合1 + 2
// 判斷表達(dá)式兩邊，是否都是數(shù)字
if (t.isNumericLiteral(node.left) && t.isNumericLiteral(node.right)) {}
那么我們得改一改
第一次計(jì)算1 + 2之后，我們會(huì)得到這樣的表達(dá)式
表達(dá)式( 表達(dá)式( 表達(dá)式(3 + 3) + 4) + 5)
其中 3 + 3又符合了我們的條件， 我們通過向上遞歸的方式遍歷父級(jí)節(jié)點(diǎn) 又轉(zhuǎn)換成這樣:
表達(dá)式( 表達(dá)式(6 + 4) + 5)
表達(dá)式(10 + 5)
15
// 如果上面的運(yùn)算有結(jié)果的話
if (result !== undefined) {
// 把表達(dá)式節(jié)點(diǎn)替換成number字面量
path.replaceWith(t.numericLiteral(result));
let parentPath = path.parentPath;
// 向上遍歷父級(jí)節(jié)點(diǎn)
parentPath && visitor.BinaryExpression.call(this, parentPath);
}
到這里，我們就得出了結(jié)果 const result = 15;
那么其他運(yùn)算呢:
const result = 100 + 10 - 50 >>> const result = 60;
const result = (100 / 2) + 50 >>> const result = 100;
const result = (((100 / 2) + 50 * 2) / 50) ** 2 >>> const result = 9;
完結(jié)
到這里，已經(jīng)向你大概的講解了，如何編寫一個(gè)Babel插件，再也不怕面試官問我答不出什么了哈...
你以為這就完了嗎?
并沒有
如果轉(zhuǎn)換這樣呢: const result = 0.1 + 0.2;
預(yù)期肯定是0.3, 但是實(shí)際上，Javascript有浮點(diǎn)計(jì)算誤差，得出的結(jié)果是0.30000000000000004
那是不是這個(gè)插件就沒卵用？
這就需要你去矯正浮點(diǎn)運(yùn)算誤差了，可以使用Big.js; 比如: result = node.left.value + node.right.value; 改成 result = +new Big(node.left.value).plus(node.right.value);你以為完了嗎? 這個(gè)插件還可以做很多
比如: Math.PI * 2 >>> 6.283185307179586
比如: Math.pow(2, 2) >>> 4
優(yōu)化
有旁友指出：
parentPath那一塊可以換個(gè)實(shí)現(xiàn)方法。第一個(gè)binaryExpression真正被計(jì)算之后是會(huì)被替換成numericLiteral的。由于每個(gè)node都會(huì)被visit兩次，所以在exit訪問時(shí)，對(duì)于父節(jié)點(diǎn)而言兩個(gè)子節(jié)點(diǎn)又同時(shí)是numericLiteral了，再次執(zhí)行即可。
也就是說，只要在enter和exit節(jié)點(diǎn)的時(shí)候，都執(zhí)行一下上面替換節(jié)點(diǎn)的代碼就行，無須手動(dòng)遍歷父節(jié)點(diǎn)去計(jì)算并替換~
BinaryExpression: {
exit: path => {
const node = path.node;
const result = node.left.value + node.right.value
path.replaceWith(t.numericLiteral(result));
}
}
真是個(gè)不錯(cuò)的方法呢！
