@atlaskit/eslint-plugin-design-system/dist/es2019/rules/utils/create-no-exported-rule/is-styled-component.js

The essential plugin for use with the Atlassian Design System.

import { getFirstSupportedImport } from '../get-first-supported-import';
/**
 * Given a list of node, find and return the callee of the first Compiled or styled-components `styled` function call found in the list.
 *
 * For example, given `styled.div({ ... })`, we return the node corresponding to the
 * `styled.div` part. Alternatively, given `styled(button)(style)`, we return the `styled`
 * part.
 *
 * @param nodes
 * @returns The callee of the first `styled` function call found.
 */
const findNode = nodes => {
  const node = nodes.find(n => n.type === 'TaggedTemplateExpression' || n.type === 'CallExpression');
  if (!node) {
    return;
  }
  if (node.type === 'CallExpression') {
    // Eg. const Component = styled.button(style)
    if (node.callee.type === 'MemberExpression') {
      return node.callee;
    }

    // Eg. const Component = styled(button)(style)
    if (node.callee.type === 'CallExpression' && node.callee.callee.type === 'Identifier') {
      return node.callee.callee;
    }
  }

  // Eg. const Component = styled.div`${styles}`;
  if (node.type === 'TaggedTemplateExpression' && node.tag.type === 'MemberExpression') {
    return node.tag;
  }
  return;
};

/**
 * Given a rule, return the local name used to import the `styled` API. (for Compiled or styled-components).
 *
 * @param context Rule context.
 * @returns The local name used to import the `styled` API.
 */
const getStyledImportSpecifierName = (context, importSources) => {
  var _supportedImport$spec;
  const supportedImport = getFirstSupportedImport(context, importSources);
  return supportedImport === null || supportedImport === void 0 ? void 0 : (_supportedImport$spec = supportedImport.specifiers.find(spec => spec.type === 'ImportSpecifier' && 'name' in spec.imported && spec.imported.name === 'styled' || spec.type === 'ImportDefaultSpecifier' && spec.local.name === 'styled')) === null || _supportedImport$spec === void 0 ? void 0 : _supportedImport$spec.local.name;
};

/**
 * Returns whether the node is a usage of the `styled` API in the libraries we support.
 *
 * @param nodes Nodes to check.
 * @param context Rule context.
 * @param importSources A list of libraries we support.
 * @returns Whether the node is a usage of the `styled` API.
 */
export const isStyledComponent = (nodes, context, importSources) => {
  const node = findNode(nodes);
  if (!node) {
    return false;
  }
  const styledImportSpecifierName = getStyledImportSpecifierName(context, importSources);
  if (styledImportSpecifierName) {
    if (node.type === 'Identifier') {
      return node.name === styledImportSpecifierName;
    } else {
      return node.object.type === 'Identifier' && node.object.name === styledImportSpecifierName;
    }
  }
  return false;
};