babel是一個(gè)轉(zhuǎn)碼器，目前開發(fā)react、vue項(xiàng)目都要使用到它。它可以把es6+的語法轉(zhuǎn)換為es5，也可以轉(zhuǎn)換JSX等語法。

我們在項(xiàng)目中都是通過配置插件和預(yù)設(shè)(多個(gè)插件的集合)來轉(zhuǎn)換特定代碼，例如env、stage-0等。

實(shí)際上babel可以通過自定義插件的方式實(shí)現(xiàn)任何代碼的轉(zhuǎn)換，接下來我們通過一個(gè)“把es6的 class 轉(zhuǎn)換為es5”的例子來了解一下babel。

內(nèi)容如下：

webpack環(huán)境配置

大家應(yīng)該都配置過babel-core這個(gè)loader，它的作用是提供babel的核心Api,實(shí)際上我們的代碼轉(zhuǎn)換都是通過插件來實(shí)現(xiàn)的。

接下來我們不用第三方的插件，自己實(shí)現(xiàn)一個(gè)es6類轉(zhuǎn)換插件。先執(zhí)行以下幾步初始化一個(gè)項(xiàng)目：

1. npm install webpack webpack-cli babel-core -D
2. 新建一個(gè)webpack.config.js
3. 配置webpack.config.js

如果我們的插件名字想叫transform-class，需要在webpack配置中做如下配置：

接下來我們在node_modules中新建一個(gè)babel-plugin-transform-class的文件夾來寫插件的邏輯(如果是真實(shí)項(xiàng)目，你需要編寫這個(gè)插件并發(fā)布到npm倉庫),如下圖：

紅色區(qū)域是我新建的文件夾，它上面的是一個(gè)標(biāo)準(zhǔn)的插件的項(xiàng)目結(jié)構(gòu)，為了方便我只寫了核心的index.js文件。

如何編寫bable插件

babel插件其實(shí)是通過AST(抽象語法樹)實(shí)現(xiàn)的。

babel幫助我們把js代碼轉(zhuǎn)換為AST，然后允許我們修改，最后再把它轉(zhuǎn)換成js代碼。

那么就涉及到兩個(gè)問題：js代碼和AST之間的映射關(guān)系是什么？如何替換或者新增AST？

好，先介紹一個(gè)工具：astexplorer.net:

這個(gè)工具可以把一段代碼轉(zhuǎn)換為AST：

如圖，我們寫了一個(gè)es6的類，然后網(wǎng)頁的右邊幫我們生成了一個(gè)AST，其實(shí)就是把每一行代碼變成了一個(gè)對象，這樣我們就實(shí)現(xiàn)了一個(gè)映射。

再介紹一個(gè)文檔： babel-types :

這是創(chuàng)建AST節(jié)點(diǎn)的api文檔。

比如，我們想創(chuàng)建一個(gè)類，先到astexplorer.net中轉(zhuǎn)換，發(fā)現(xiàn)類對應(yīng)的AST類型是 ClassDeclaration 。好，我們?nèi)ノ臋n中搜索，發(fā)現(xiàn)調(diào)用下面的api就可以了：

創(chuàng)建其他語句也是一樣的道理，有了上面這兩個(gè)東西，我們可以做任何轉(zhuǎn)換了。

下面我們開始真正編寫一個(gè)插件，分為以下幾步：

1. 在index.js中export一個(gè)函數(shù)
2. 函數(shù)中返回一個(gè)對象，對象有一個(gè)visitor參數(shù)(必須叫visitor)
3. 通過astexplorer.net查詢出 class 對應(yīng)的AST節(jié)點(diǎn)為 ClassDeclaration
4. 在vistor中設(shè)置一個(gè)捕獲函數(shù) ClassDeclaration ，意思是我要捕獲js代碼中所有 ClassDeclaration 節(jié)點(diǎn)
5. 編寫邏輯代碼，完成轉(zhuǎn)換

module.exports = function ({ types: t }) {
 return {
  visitor: {
   ClassDeclaration(path) {
    //在這里完成轉(zhuǎn)換
   }
  }
 };
}

代碼中有兩個(gè)參數(shù)，第一個(gè) {types:t} 東西是從參數(shù)中解構(gòu)出變量t，它其實(shí)就是babel-types文檔中的t(下圖紅框)，它是用來創(chuàng)建節(jié)點(diǎn)的：

第二個(gè)參數(shù) path ，它是捕獲到的節(jié)點(diǎn)對應(yīng)的信息，我們可以通過 path.node 獲得這個(gè)節(jié)點(diǎn)的AST，在這個(gè)基礎(chǔ)上進(jìn)行修改就能完成了我們的目標(biāo)。

如何把es6的class轉(zhuǎn)換為es5的類

上面都是預(yù)備工作，真正的邏輯從現(xiàn)在才開始，我們先考慮兩個(gè)問題：

我們要做如下轉(zhuǎn)換，首先把es6的類，轉(zhuǎn)換為es5的類寫法(也就是普通函數(shù))，我們觀察到，很多代碼是可以復(fù)用的，包括函數(shù)名字、函數(shù)內(nèi)部的代碼塊等。

如果不定義class中的 constructor 方法，JavaScript引擎會自動(dòng)為它添加一個(gè)空的 constructor() 方法，這需要我們做兼容處理。

接下來我們開始寫代碼，思路是：

1. 拿到老的AST節(jié)點(diǎn)
2. 創(chuàng)建一個(gè)數(shù)組用來盛放新的AST節(jié)點(diǎn)(雖然原class只是一個(gè)節(jié)點(diǎn)，但是替換后它會被若干個(gè)函數(shù)節(jié)點(diǎn)取代) 初始化默認(rèn)的 constructor 節(jié)點(diǎn)(上文提到，class中有可能沒有定義constructor)
3. 循環(huán)老節(jié)點(diǎn)的AST對象(會循環(huán)出若干個(gè)函數(shù)節(jié)點(diǎn))
4. 判斷函數(shù)的類型是不是 constructor ，如果是，通過取到數(shù)據(jù)創(chuàng)建一個(gè)普通函數(shù)節(jié)點(diǎn)，并更新默認(rèn) constructor 節(jié)點(diǎn)
5. 處理其余不是 constructor 的節(jié)點(diǎn)，通過數(shù)據(jù)創(chuàng)建 prototype 類型的函數(shù)，并放到 es5Fns 中
6. 循環(huán)結(jié)束，把 constructor 節(jié)點(diǎn)也放到 es5Fns 中
7. 判斷es5Fns的長度是否大于1，如果大于1使用 replaceWithMultiple 這個(gè)API更新AST

module.exports = function ({ types: t }) {
 return {
  visitor: {
   ClassDeclaration(path) {
    //拿到老的AST節(jié)點(diǎn)
    let node = path.node
    let className = node.id.name
    let classInner = node.body.body
    //創(chuàng)建一個(gè)數(shù)組用來成盛放新生成AST
    let es5Fns = []
    //初始化默認(rèn)的constructor節(jié)點(diǎn)
    let newConstructorId = t.identifier(className)
    let constructorFn = t.functionDeclaration(newConstructorId, [t.identifier('')], t.blockStatement([]), false, false)
    //循環(huán)老節(jié)點(diǎn)的AST對象
    for (let i = 0; i < classInner.length; i++) {
     let item = classInner[i]
     //判斷函數(shù)的類型是不是constructor
     if (item.kind == 'constructor') {
      let constructorParams = item.params.length ? item.params[0].name : []
      let newConstructorParams = t.identifier(constructorParams)
      let constructorBody = classInner[i].body
      constructorFn = t.functionDeclaration(newConstructorId, [newConstructorParams], constructorBody, false, false)
     } 
     //處理其余不是constructor的節(jié)點(diǎn)
     else {
      let protoTypeObj = t.memberExpression(t.identifier(className), t.identifier('prototype'), false)
      let left = t.memberExpression(protoTypeObj, t.identifier(item.key.name), false)
      //定義等號右邊
      let prototypeParams = classInner[i].params.length ? classInner[i].params[i].name : []
      let newPrototypeParams = t.identifier(prototypeParams)
      let prototypeBody = classInner[i].body
      let right = t.functionExpression(null, [newPrototypeParams], prototypeBody, false, false)
      let protoTypeExpression = t.assignmentExpression("=", left, right)
      es5Fns.push(protoTypeExpression)
     }

    }
    //循環(huán)結(jié)束，把constructor節(jié)點(diǎn)也放到es5Fns中
    es5Fns.push(constructorFn)
    //判斷es5Fns的長度是否大于1
    if (es5Fns.length > 1) {
     path.replaceWithMultiple(es5Fns)
    } else {
     path.replaceWith(constructorFn)
    }
   }
  }
 };
}

優(yōu)化繼承

其實(shí)，類還涉及到繼承，思路也不復(fù)雜，就是判斷AST中沒有 superClass 屬性，如果有的話，我們需要多添加一行代碼 Bird.prototype = Object.create(Parent) ,當(dāng)然別忘了處理 super 關(guān)鍵字。

打包后代碼

運(yùn)行 npm start 打包后,我們看到打包后的文件里 class

語法已經(jīng)成功轉(zhuǎn)換為一個(gè)個(gè)的es5函數(shù)。

結(jié)尾

現(xiàn)在一個(gè)類轉(zhuǎn)換器就寫完了，希望能對大家了解babel有一點(diǎn)幫助。也希望大家多多支持腳本之家。

最新評論