Function Program

packages/compiler/src/utils/index.ts:152–185

(path: NodePath<t.Program>)

): void {
  traverse(ast, {
    Program(path: NodePath<t.Program>) {
      // Create the import specifier
      const importSpecifier = t.importSpecifier(
        t.identifier(localName),
        t.identifier(exportedName),
      );

      // Check if we already have a non-type import from this module
      const existingImport = path
        .get("body")
        .find(
          (nodePath: NodePath) =>
            t.isImportDeclaration(nodePath.node) &&
            moduleName.includes(nodePath.node.source.value) &&
            nodePath.node.importKind !== "type",
        );

      if (existingImport && t.isImportDeclaration(existingImport.node)) {
        // Add to existing import declaration
        existingImport.node.specifiers.push(importSpecifier);
      } else {
        // Create a new import declaration
        const importDeclaration = t.importDeclaration(
          [importSpecifier],
          t.stringLiteral(moduleName[0]),
        );

        // Add it at the top of the file, after any existing imports
        const lastImportIndex = findLastImportIndex(path);
        path.node.body.splice(lastImportIndex + 1, 0, importDeclaration);
      }

      path.stop();
    },
  });
}