jsxstyle-loader/utils/ast/getSourceModule.js

'use strict';

const t = require('babel-types');

function getSourceModule(itemName, itemBinding) {
  // TODO: deal with reassignment
  if (!itemBinding.constant) {
    return null;
  }

  let sourceModule;
  let imported;
  let local;
  let destructured;
  let usesImportSyntax = false;

  if (
    // import x from 'y';
    t.isImportDefaultSpecifier(itemBinding.path.node) ||
    // import {x} from 'y';
    t.isImportSpecifier(itemBinding.path.node)
  ) {
    if (
      t.isImportDeclaration(itemBinding.path.parent) &&
      t.isStringLiteral(itemBinding.path.parent.source)
    ) {
      sourceModule = itemBinding.path.parent.source.value;
      local = itemBinding.path.node.local.name;
      usesImportSyntax = true;
      if (itemBinding.path.node.imported) {
        imported = itemBinding.path.node.imported.name;
        destructured = true;
      } else {
        imported = itemBinding.path.node.local.name;
        destructured = false;
      }
    }
  } else if (
    t.isVariableDeclarator(itemBinding.path.node) &&
    t.isCallExpression(itemBinding.path.node.init) &&
    t.isIdentifier(itemBinding.path.node.init.callee) &&
    itemBinding.path.node.init.callee.name === 'require' &&
    itemBinding.path.node.init.arguments.length === 1 &&
    t.isStringLiteral(itemBinding.path.node.init.arguments[0])
  ) {
    sourceModule = itemBinding.path.node.init.arguments[0].value;

    if (t.isIdentifier(itemBinding.path.node.id)) {
      local = itemBinding.path.node.id.name;
      imported = itemBinding.path.node.id.name;
      destructured = false;
    } else if (t.isObjectPattern(itemBinding.path.node.id)) {
      // TODO: better way to get ObjectProperty
      const objProp = itemBinding.path.node.id.properties.find(
        p => t.isIdentifier(p.value) && p.value.name === itemName
      );

      if (!objProp) {
        console.error('could not find prop with value `%s`', itemName);
        return null;
      }

      local = objProp.value.name;
      imported = objProp.key.name;
      destructured = true;
    } else {
      console.error('Unhandled id type: %s', itemBinding.path.node.id.type);
      return null;
    }
  } else {
    return null;
  }

  return {
    sourceModule,
    imported,
    local,
    destructured,
    usesImportSyntax,
  };
}

module.exports = getSourceModule;