Function processJS

scripts/extractStarter.mjs:142–266

Processes the JS code for a "Reusable wrappers" example from the docs.

(sourceFilename, code, usedClasses)

/** Processes the JS code for a "Reusable wrappers" example from the docs. */
function processJS(sourceFilename, code, usedClasses) {
  if (/^<(.|\n)*>$/m.test(code)) {
    code = code.replace(/^(<(.|\n)*>)$/m, '').trim() + '\n';
  }

  let ast = recast.parse(code, {
    parser: {
      parse() {
        return parse(code, {
          sourceFilename,
          sourceType: 'module',
          plugins: ['jsx', 'typescript'],
          tokens: true,
          errorRecovery: true
        });
      }
    }
  });

  removeDuplicateImports(ast);

  // Remove unused imports, add exports.
  let imports = new Map();
  let component = basename(sourceFilename, '.mdx');
  let importsToAppend = [t.importDeclaration([], t.stringLiteral(`./${component}.css`))];
  traverse.default(ast, {
    'ImportSpecifier|ImportDefaultSpecifier'(specifier) {
      let binding = specifier.scope.getBinding(specifier.node.local.name);
      if (binding?.referencePaths.length === 0) {
        specifier.remove();
      }
    },
    ImportDeclaration: {
      exit(path) {
        if (path.node.source.value.endsWith('.css')) {
          return;
        }

        if (path.node.specifiers.length === 0) {
          path.remove();
        } else if (imports.has(path.node.source.value)) {
          // merge imports with the same source
          imports.get(path.node.source.value).pushContainer('specifiers', path.node.specifiers);
          path.remove();
          path.scope.crawl();
        } else {
          imports.set(path.node.source.value, path);
        }
      }
    },
    'FunctionDeclaration|TSInterfaceDeclaration': {
      exit(path) {
        // GridList/Table define their own checkboxes, but we can share the one from the Checkbox docs.
        if (path.node.id.name === 'MyCheckbox' && component !== 'Checkbox') {
          path.scope.getBinding('Checkbox')?.path.remove();
          path.scope.getBinding('CheckboxProps')?.path.remove();
          path.scope.rename('MyCheckbox', 'Checkbox');
          if (path.scope.getBinding('Checkbox').referencePaths.length > 0) {