@open-wc/dev-server-hmr/src/babel/functions.js

Plugin for HMR with web components

/** @typedef {import('@babel/types').VariableDeclaration} VariableDeclaration */
/** @typedef {import('@babel/types').CallExpression} CallExpression */
/** @typedef {import('@babel/types').Identifier} Identifier */
/** @typedef {import('./babelPluginWcHmr').Decorator} Decorator */
/** @template T @typedef {import('@babel/core').NodePath<T>} NodePath<T> */

const { singlePath } = require('./utils.js');

/**
 * @param {NodePath<CallExpression>} callExpr
 * @param {Set<string>} functionNames
 */
function isFunctionComponent(callExpr, functionNames) {
  const callee = callExpr.get('callee');
  if (!singlePath(callee) || !callee.isIdentifier()) {
    return false;
  }
  return functionNames.has(callee.node.name);
}

module.exports = { isFunctionComponent };