把从@tarojs/taro引入的hooks API转变成从nervjs引入

babel-plugin-transform-taro-hooks.js

module.exports = function(babel) {
  const t = babel.types

  const hookTest = /^use[A-Z]/

  let hooksImport = new Set()

  return {
    name: 'babel-plugin-transform-taro-hooks',
    visitor: {
      ImportDeclaration(path) {
        const n = path.node
        if (t.isStringLiteral(n.source, { value: '@tarojs/taro' })) {
          path.traverse({
            ImportSpecifier(iPath) {
              const node = iPath.node
              // console.log(node.local.name)

              if (hookTest.test(node.local.name)) {
                hooksImport.add(node.local.name)
                iPath.remove()
              }
            },
          })
        }
      },
      Program: {
        enter() {
          hooksImport = new Set()
          console.log('start transform +++++++++++++++')
        },
        exit(ast) {
          const iptd = t.importDeclaration([], t.StringLiteral('nervjs'))

          for (let spec of hooksImport) {
            iptd.specifiers.push(
              t.importSpecifier(t.identifier(spec), t.identifier(spec))
            )
          }

          // console.log('----------', iptd)

          ast.node.body.unshift(iptd)
        },
      },
    },
  }
}