@litsx/babel-preset-litsx 0.8.1 → 0.9.0

package/src/internal/transform-litsx-element-candidates.js

@@ -6,6 +6,10 @@ import fs from "node:fs";
 import path from "node:path";
 import { normalizeFilePath } from "@litsx/typescript-session";
 import { ensureTypescriptModule } from "./transform-litsx-properties.js";
+import {
+  ensureStaticIr,
+  setStaticIrBabelTypes,
+} from "./transform-litsx-static-ir.js";
 
 const { declare } = helperPluginUtils;
 const traverse = babelTraverse.default || babelTraverse;
@@ -32,6 +36,7 @@ let t;
 
 export function setElementCandidatesBabelTypes(nextTypes) {
   t = nextTypes;
+  setStaticIrBabelTypes(nextTypes);
 }
 
 function isInsideFunctionOrClass(path) {
@@ -59,6 +64,13 @@ function createEmptyCandidateResult() {
   };
 }
 
+function annotateElementCandidates(node, result) {
+  if (!node) return;
+  const staticIr = ensureStaticIr(node);
+  staticIr.elements.localCandidates = [...result.localCandidates];
+  staticIr.elements.importedCandidates = [...result.importedCandidates.values()];
+}
+
 function cloneCandidateResult(result) {
   return {
     localCandidates: new Set(result?.localCandidates || []),
@@ -536,7 +548,10 @@ function getOrCreateModuleAnalysis(filename, context) {
 
   let programPath = null;
   try {
-    const ast = parser.parse(source, { sourceType: "module" });
+    const ast = parser.parse(source, {
+      sourceType: "module",
+      plugins: getParserPluginsForModule(normalizedFilename, source),
+    });
     traverse(ast, {
       Program(path) {
         if (!programPath) {
@@ -665,6 +680,73 @@ function resolveExportedHelper(moduleAnalysis, exportedName, context, seen = new
   return null;
 }
 
+function getParserPluginsForModule(filename, source) {
+  if (/\.(?:[cm]?ts|tsx|litsx)$/i.test(filename)) {
+    return ["typescript"];
+  }
+
+  if (/\b(?:as|satisfies)\s+[^;,)]+/.test(source)) {
+    return ["typescript"];
+  }
+
+  return [];
+}
+
+function unwrapNamespaceAliasExpression(node) {
+  let current = node;
+  while (
+    t.isTSAsExpression(current) ||
+    t.isTSTypeAssertion(current) ||
+    t.isTSNonNullExpression(current) ||
+    t.isTSSatisfiesExpression?.(current)
+  ) {
+    current = current.expression;
+  }
+  return current;
+}
+
+function getNamespaceMemberAliasInfo(candidateName, moduleAnalysis) {
+  const binding = moduleAnalysis.programPath.scope.getBinding(candidateName);
+  if (!binding || !isProgramLevelBinding(binding)) {
+    return null;
+  }
+
+  const declaratorPath = binding.path.isVariableDeclarator?.()
+    ? binding.path
+    : binding.path.parentPath;
+  if (!declaratorPath?.isVariableDeclarator?.()) {
+    return null;
+  }
+
+  const init = unwrapNamespaceAliasExpression(declaratorPath.node.init);
+  if (
+    !t.isMemberExpression(init) ||
+    init.computed ||
+    !t.isIdentifier(unwrapNamespaceAliasExpression(init.object)) ||
+    !t.isIdentifier(init.property)
+  ) {
+    return null;
+  }
+
+  const namespaceObject = unwrapNamespaceAliasExpression(init.object);
+  const namespaceImport = moduleAnalysis.importBindings.get(namespaceObject.name);
+  if (
+    !namespaceImport ||
+    namespaceImport.importedName !== "*" ||
+    !namespaceImport.resolvedSource
+  ) {
+    return null;
+  }
+
+  return {
+    localName: candidateName,
+    namespaceName: namespaceObject.name,
+    importedName: init.property.name,
+    sourceValue: namespaceImport.sourceValue,
+    resolvedSource: namespaceImport.resolvedSource,
+  };
+}
+
 function resolveImportedElementRequirement(candidateName, moduleAnalysis, context, rootFilename) {
   const binding = moduleAnalysis.programPath.scope.getBinding(candidateName);
   if (!binding || !isProgramLevelBinding(binding)) {
@@ -692,6 +774,20 @@ function resolveImportedElementRequirement(candidateName, moduleAnalysis, contex
     };
   }
 
+  const namespaceAliasInfo = getNamespaceMemberAliasInfo(candidateName, moduleAnalysis);
+  if (namespaceAliasInfo) {
+    return {
+      sourceFile: namespaceAliasInfo.resolvedSource,
+      sourceSpecifier: isRelativeSpecifier(namespaceAliasInfo.sourceValue)
+        ? null
+        : namespaceAliasInfo.sourceValue,
+      importedName: namespaceAliasInfo.importedName,
+      originalName: candidateName,
+      tagName: candidateName.replace(/([a-z])([A-Z])/g, "$1-$2").toLowerCase(),
+      rootFilename,
+    };
+  }
+
   const exportInfo =
     moduleAnalysis.exportBindings.get(candidateName) ||
     moduleAnalysis.exportBindings.get("default");
@@ -904,16 +1000,18 @@ function collectCandidateResult(functionPath, programPath, options = {}) {
 }
 
 export function getAnnotatedElementCandidates(path, programPath, options = {}) {
-  if (path?.node?._litsxElementCandidates instanceof Set) {
-    return new Set(path.node._litsxElementCandidates);
+  const localCandidates = path?.node?._litsxStaticIr?.elements?.localCandidates;
+  if (Array.isArray(localCandidates)) {
+    return new Set(localCandidates);
   }
 
   return collectCandidateResult(path, programPath, options).localCandidates;
 }
 
 export function getAnnotatedImportedElementCandidates(path, programPath, options = {}) {
-  if (Array.isArray(path?.node?._litsxImportedElementCandidates)) {
-    return [...path.node._litsxImportedElementCandidates];
+  const importedCandidates = path?.node?._litsxStaticIr?.elements?.importedCandidates;
+  if (Array.isArray(importedCandidates)) {
+    return [...importedCandidates];
   }
 
   return [...collectCandidateResult(path, programPath, options).importedCandidates.values()];
@@ -1126,8 +1224,7 @@ export default declare((api) => {
               filename: state.file?.opts?.filename || "",
             }
           );
-          path.node._litsxElementCandidates = result.localCandidates;
-          path.node._litsxImportedElementCandidates = [...result.importedCandidates.values()];
+          annotateElementCandidates(path.node, result);
         },
       },
       ArrowFunctionExpression: {
@@ -1145,8 +1242,7 @@ export default declare((api) => {
               filename: state.file?.opts?.filename || "",
             }
           );
-          path.node._litsxElementCandidates = result.localCandidates;
-          path.node._litsxImportedElementCandidates = [...result.importedCandidates.values()];
+          annotateElementCandidates(path.node, result);
         },
       },
       FunctionExpression: {
@@ -1164,8 +1260,7 @@ export default declare((api) => {
               filename: state.file?.opts?.filename || "",
             }
           );
-          path.node._litsxElementCandidates = result.localCandidates;
-          path.node._litsxImportedElementCandidates = [...result.importedCandidates.values()];
+          annotateElementCandidates(path.node, result);
         },
       },
     },