Blame view

天文台pc/tianwentai-ui/node_modules/@emotion/babel-plugin/src/utils/transformer-macro.js 1.65 KB
bc518174   王天杨   提交两个项目文件
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
  import { createMacro } from 'babel-plugin-macros'
  
  /*
  type Transformer = Function
  */
  
  export function createTransformerMacro(
    transformers /*: { [key: string]: Transformer | [Transformer, Object] } */,
    { importSource } /*: { importSource: string } */
  ) {
    let macro = createMacro(
      ({ path, source, references, state, babel, isEmotionCall }) => {
        if (!path) {
          path = state.file.scope.path
            .get('body')
            .find(p => p.isImportDeclaration() && p.node.source.value === source)
        }
  
        if (/\/macro$/.test(source)) {
          path
            .get('source')
            .replaceWith(
              babel.types.stringLiteral(source.replace(/\/macro$/, ''))
            )
        }
  
        if (!isEmotionCall) {
          state.emotionSourceMap = true
        }
        Object.keys(references).forEach(importSpecifierName => {
          if (transformers[importSpecifierName]) {
            references[importSpecifierName].reverse().forEach(reference => {
              let options
              let transformer
              if (Array.isArray(transformers[importSpecifierName])) {
                transformer = transformers[importSpecifierName][0]
                options = transformers[importSpecifierName][1]
              } else {
                transformer = transformers[importSpecifierName]
                options = {}
              }
              transformer({
                state,
                babel,
                path,
                importSource,
                importSpecifierName,
                options,
                reference
              })
            })
          }
        })
        return { keepImports: true }
      }
    )
    macro.transformers = transformers
    return macro
  }