@lingui/babel-plugin-transform-react/index.js.flow

Transform React components to ICU message format

import Transformer from "./transformer"

// Plugin function
export default function(babel) {
  const { types: t } = babel

  const transformer = new Transformer(babel)

  return {
    visitor: {
      Program(path, state) {
        // Reset import declaration for each file.
        // Regression introduced in https://github.com/lingui/js-lingui/issues/62

        const importedNames = state.opts.importedNames
        if (importedNames) {
          const importDeclarations = {}

          importedNames.forEach(name => {
            if (typeof name === "string") {
              name = [name, name]
            }

            const [imported, local] = name
            importDeclarations[imported] = local
          })

          transformer.setImportDeclarations(importDeclarations)
        } else {
          transformer.setImportDeclarations({})
        }
      },
      JSXElement: transformer.transform,
      ImportDeclaration({ node }) {
        const importDeclarations = transformer.getImportDeclarations()

        const moduleName = node.source.value
        if (moduleName !== "@lingui/react") return

        node.specifiers.forEach(specifier => {
          importDeclarations[specifier.imported.name] = specifier.local.name
        })

        // Choices components are converted to Trans,
        // so imports can be safely removed
        const choicesComponents = ["Plural", "Select", "SelectOrdinal"]
        const isChoiceComponent = specifier =>
          choicesComponents.includes(specifier.imported.name)

        const hasChoices = node.specifiers.filter(isChoiceComponent).length

        if (hasChoices) {
          node.specifiers = [
            // Import for `Trans` component should be there always
            t.importSpecifier(
              t.identifier(transformer.getLocalImportName("Trans")),
              t.identifier("Trans")
            ),

            // Copy all original imports except choices components
            // and duplicate Trans components
            ...node.specifiers.filter(
              specifier =>
                !isChoiceComponent(specifier) &&
                specifier.imported.name !== "Trans"
            )
          ]
        }

        transformer.setImportDeclarations(importDeclarations)
      }
    } // visitor
  }
}