Babylon-AST初探-代碼更新&刪除(Update & Remove)

levius 發(fā)布于2019-08-26 11:56 / 3158人閱讀

摘要：操作通常配合來完成。因為是個數(shù)組，因此，我們可以直接使用數(shù)組操作自我毀滅方法極為簡單，找到要刪除的，執(zhí)行就結(jié)束了。如上述代碼，我們要刪除屬性，代碼如下到目前為止，的我們都介紹完了，下面一篇文章以轉(zhuǎn)小程序為例，我們來實戰(zhàn)一波。

??通過前兩篇文章的介紹，大家已經(jīng)了解了Create和Retrieve，我們接著介紹Update和 Remove操作。Update操作通常配合Create來完成。我們這篇文章主要介紹幾個常用的NodePath`API：replace、insert、remove`。具體也可以看babel-handbook中的Manipulation章節(jié)。

replaceWith 使用新的節(jié)點進(jìn)行替換 將加法運(yùn)算替換成乘法

const code = `const c = a + b`
const ast = babylon.parse(code)

traverse(ast, {
  BinaryExpression(path) {
    // 注意這里要有判斷，否則會無限進(jìn)入`BinaryExpression`
    // https://stackoverflow.com/questions/37539432/babel-maximum-call-stack-size-exceeded-while-using-path-replacewith
    if (path.node.operator === "+") {
      path.replaceWith(t.binaryExpression("*", path.node.left, path.node.right))
    }
  }
})

console.log(generate(ast, {}, code).code) // const c = a * b;

將this.count替換為this.data.count

??轉(zhuǎn)換前后的AST展示如下圖：

??我們需要做的是，找到符合this.count的ThisExpression，然后把它替換為this.data

const code = `this.count`
const ast = babylon.parse(code)

traverse(ast, {
  MemberExpression(path) {
    if (
      t.isThisExpression(path.node.object) &&
      t.isIdentifier(path.node.property, {
        name: "count"
      })
    ) {
      path
        .get("object")    // 獲取`ThisExpresssion`
        .replaceWith(
          t.memberExpression(t.thisExpression(), t.identifier("data"))
        )
    }
  }
})
console.log(generate(ast, {}, code).code) // this.data.count;

replaceWithSourceString 直接使用代碼替換

??上個例子中將this.count替換為this.data.count的部分，通過t.memberExpression可以構(gòu)造node。更簡單的操作可以直接使用replaceWithSourceString，個人覺得這個API很好用。

path.get("object").replaceWithSourceString("this.data")

插入操作

??插入是樹操作的一種常見操作。子節(jié)點是個Array，前、中、后各種位置都可以插入新節(jié)點。下面來介紹下pushContainer、unshiftContainer、insertBefore、insertAfter操作。

??這里以給obj對象新增一個屬性myprop: "hello my property"為例：

const code = `
const obj = {
  count: 0,
  message: "hello world"
}
`
const ast = babylon.parse(code)

const property = t.objectProperty(
  t.identifier("myprop"),
  t.stringLiteral("hello my property")
)

pushContainer 父節(jié)點的操作

??父節(jié)點為子節(jié)點Array插入一個node

traverse(ast, {
  ObjectExpression(path) {
    path.pushContainer("properties", property)
  }
})

insertAfter 兄弟節(jié)點的操作

??insertAfter也可以完成上述操作，需要找到message屬性，然后在后面插入node就搞定啦

traverse(ast, {
  ObjectProperty(path) {
    if (
      t.isIdentifier(path.node.key, {
        name: "message"
      })
    ) {
      path.insertAfter(property)
    }
  }
})

??unshiftContainer和insertBefore與上面兩個相對應(yīng)，這里不再舉例了，大家可以自己試一試。

??因為properties是個數(shù)組，因此，我們可以直接使用數(shù)組操作

traverse(ast, {
  ObjectExpression(path) {
    // path.pushContainer("properties", property)
    path.node.properties.push(property)
  }
})

Remove 自我毀滅

??Remove方法極為簡單，找到要刪除的NodePath，執(zhí)行Remove就結(jié)束了。如上述代碼，我們要刪除message屬性，代碼如下：

traverse(ast, {
  ObjectProperty(path) {
    if (
      t.isIdentifier(path.node.key, {
        name: "message"
      })
    ) {
      path.remove()
    }
  }
})

到目前為止，AST的CURD我們都介紹完了，下面一篇文章以vue轉(zhuǎn)小程序為例，我們來實戰(zhàn)一波。