@knighted/module 1.0.0-alpha.9 → 1.0.0-beta.1

package/dist/utils/exports.js

@@ -0,0 +1,221 @@
+import { ancestorWalk } from '#walk';
+const exportsRename = '__exports';
+const requireMainRgx = /(require\.main\s*===\s*module|module\s*===\s*require\.main)/g;
+const literalPropName = (prop, literals) => {
+  if (prop.type === 'Identifier') return literals?.get(prop.name)?.toString() ?? prop.name;
+  if (prop.type === 'Literal' && (typeof prop.value === 'string' || typeof prop.value === 'number')) {
+    return String(prop.value);
+  }
+  if (prop.type === 'TemplateLiteral' && prop.expressions.length === 0 && prop.quasis.length === 1) {
+    return prop.quasis[0].value.cooked ?? prop.quasis[0].value.raw;
+  }
+  return null;
+};
+const resolveExportTarget = (node, aliases, literals) => {
+  if (node.type === 'Identifier' && node.name === 'exports') {
+    return {
+      key: 'default',
+      via: 'exports'
+    };
+  }
+  if (node.type === 'MemberExpression' && node.object.type === 'Identifier' && node.object.name === 'module' && node.property.type === 'Identifier' && node.property.name === 'exports') {
+    return {
+      key: 'default',
+      via: 'module.exports'
+    };
+  }
+  if (node.type !== 'MemberExpression') return null;
+  const base = node.object;
+  const prop = node.property;
+  const key = literalPropName(prop, literals);
+  if (!key) return null;
+  const baseVia = resolveBase(base, aliases);
+  if (!baseVia) return null;
+  if (baseVia === 'module.exports' && key === 'exports') {
+    return {
+      key: 'default',
+      via: 'module.exports'
+    };
+  }
+  return {
+    key,
+    via: baseVia
+  };
+};
+const resolveBase = (node, aliases) => {
+  if (node.type === 'Identifier') {
+    if (node.name === 'exports') return 'exports';
+    const alias = aliases.get(node.name);
+    if (alias) return alias;
+  }
+  if (node.type === 'MemberExpression' && node.object.type === 'Identifier' && node.object.name === 'module' && node.property.type === 'Identifier' && node.property.name === 'exports') {
+    return 'module.exports';
+  }
+  return null;
+};
+const collectCjsExports = async ast => {
+  const exportsMap = new Map();
+  const localToExport = new Map();
+  const aliases = new Map();
+  const literals = new Map();
+  const addExport = (ref, node, rhs, options) => {
+    const entry = exportsMap.get(ref.key) ?? {
+      key: ref.key,
+      writes: [],
+      via: new Set(),
+      reassignments: []
+    };
+    entry.via.add(ref.via);
+    entry.writes.push(node);
+    if (options?.hasGetter) {
+      entry.hasGetter = true;
+    }
+    if (rhs) {
+      entry.fromIdentifier ??= rhs.name;
+      const set = localToExport.get(rhs.name) ?? new Set();
+      set.add(ref.key);
+      localToExport.set(rhs.name, set);
+    }
+    exportsMap.set(ref.key, entry);
+  };
+  await ancestorWalk(ast, {
+    enter(node) {
+      if (node.type === 'VariableDeclarator' && node.id.type === 'Identifier' && node.init) {
+        const via = resolveBase(node.init, aliases);
+        if (via) {
+          aliases.set(node.id.name, via);
+        }
+        if (node.init.type === 'Literal' && (typeof node.init.value === 'string' || typeof node.init.value === 'number')) {
+          literals.set(node.id.name, node.init.value);
+        }
+        if (node.init.type === 'TemplateLiteral' && node.init.expressions.length === 0 && node.init.quasis.length === 1) {
+          const cooked = node.init.quasis[0].value.cooked ?? node.init.quasis[0].value.raw;
+          literals.set(node.id.name, cooked);
+        }
+      }
+      if (node.type === 'AssignmentExpression') {
+        const target = resolveExportTarget(node.left, aliases, literals);
+        if (target) {
+          const rhsIdent = node.right.type === 'Identifier' ? node.right : undefined;
+          addExport(target, node, rhsIdent);
+          return;
+        }
+        if (node.left.type === 'Identifier') {
+          const keys = localToExport.get(node.left.name);
+          if (keys) {
+            keys.forEach(key => {
+              const entry = exportsMap.get(key);
+              if (entry) {
+                entry.reassignments.push(node);
+                exportsMap.set(key, entry);
+              }
+            });
+          }
+        }
+        if (node.left.type === 'ObjectPattern') {
+          for (const prop of node.left.properties) {
+            if (prop.type === 'Property' && prop.value.type === 'MemberExpression') {
+              const ref = resolveExportTarget(prop.value, aliases, literals);
+              if (ref) {
+                addExport(ref, node);
+              }
+            }
+          }
+        }
+      }
+      if (node.type === 'CallExpression') {
+        const callee = node.callee;
+
+        // Object.assign(exports, { foo: bar })
+        if (callee.type === 'MemberExpression' && callee.object.type === 'Identifier' && callee.object.name === 'Object' && callee.property.type === 'Identifier' && callee.property.name === 'assign' && node.arguments.length >= 2) {
+          const targetArg = node.arguments[0];
+          const ref = resolveBase(targetArg, aliases);
+          if (!ref) return;
+          for (let i = 1; i < node.arguments.length; i++) {
+            const arg = node.arguments[i];
+            if (arg.type === 'ObjectExpression') {
+              for (const prop of arg.properties) {
+                if (prop.type !== 'Property') continue;
+                const keyName = literalPropName(prop.key, literals);
+                if (!keyName) continue;
+                let rhsIdent;
+                if (prop.value.type === 'Identifier') {
+                  rhsIdent = prop.value;
+                }
+                addExport({
+                  key: keyName,
+                  via: ref
+                }, node, rhsIdent);
+              }
+            }
+          }
+        }
+
+        // Object.defineProperty(exports, 'foo', { value, get, set })
+        if (callee.type === 'MemberExpression' && callee.object.type === 'Identifier' && callee.object.name === 'Object' && callee.property.type === 'Identifier' && callee.property.name === 'defineProperty' && node.arguments.length >= 3) {
+          const target = resolveBase(node.arguments[0], aliases);
+          if (!target) return;
+          const keyName = literalPropName(node.arguments[1], literals);
+          if (!keyName) return;
+          const desc = node.arguments[2];
+          if (desc.type !== 'ObjectExpression') return;
+          let rhsIdent;
+          let hasGetter = false;
+          for (const prop of desc.properties) {
+            if (prop.type !== 'Property') continue;
+            if (prop.key.type !== 'Identifier') continue;
+            if (prop.key.name === 'value' && prop.value.type === 'Identifier') {
+              rhsIdent = prop.value;
+            }
+            if (prop.key.name === 'get' && prop.value.type === 'Identifier') {
+              hasGetter = true;
+            }
+            if (prop.key.name === 'set' && prop.value.type === 'Identifier') {
+              // Setter-only doesn’t create a readable export; ignore beyond marking write
+            }
+          }
+          addExport({
+            key: keyName,
+            via: target
+          }, node, rhsIdent, {
+            hasGetter
+          });
+        }
+
+        // Object.defineProperties(exports, { foo: { value: ... }, bar: { get: ... } })
+        if (callee.type === 'MemberExpression' && callee.object.type === 'Identifier' && callee.object.name === 'Object' && callee.property.type === 'Identifier' && callee.property.name === 'defineProperties' && node.arguments.length >= 2) {
+          const target = resolveBase(node.arguments[0], aliases);
+          if (!target) return;
+          const descMap = node.arguments[1];
+          if (descMap.type !== 'ObjectExpression') return;
+          for (const prop of descMap.properties) {
+            if (prop.type !== 'Property') continue;
+            const keyName = literalPropName(prop.key, literals);
+            if (!keyName) continue;
+            if (prop.value.type !== 'ObjectExpression') continue;
+            let rhsIdent;
+            let hasGetter = false;
+            for (const descProp of prop.value.properties) {
+              if (descProp.type !== 'Property') continue;
+              if (descProp.key.type !== 'Identifier') continue;
+              if (descProp.key.name === 'value' && descProp.value.type === 'Identifier') {
+                rhsIdent = descProp.value;
+              }
+              if (descProp.key.name === 'get' && descProp.value.type === 'Identifier') {
+                hasGetter = true;
+              }
+            }
+            addExport({
+              key: keyName,
+              via: target
+            }, node, rhsIdent, {
+              hasGetter
+            });
+          }
+        }
+      }
+    }
+  });
+  return exportsMap;
+};
+export { exportsRename, requireMainRgx, collectCjsExports };

package/dist/utils/identifiers.js

@@ -0,0 +1,183 @@
+import { ancestorWalk } from '#walk';
+import { scopes as scopeNodes } from '#helpers/scope.js';
+import { identifier } from '#helpers/identifier.js';
+const collectScopeIdentifiers = (node, scopes) => {
+  const {
+    type
+  } = node;
+  switch (type) {
+    case 'BlockStatement':
+    case 'ClassBody':
+      scopes.push({
+        node,
+        type: 'Block',
+        name: type,
+        idents: new Set()
+      });
+      break;
+    case 'FunctionDeclaration':
+    case 'FunctionExpression':
+    case 'ArrowFunctionExpression':
+      {
+        const name = node.id ? node.id.name : 'anonymous';
+        const scope = {
+          node,
+          name,
+          type: 'Function',
+          idents: new Set()
+        };
+        node.params.map(param => {
+          if (param.type === 'TSParameterProperty') {
+            return param.parameter;
+          }
+          if (param.type === 'RestElement') {
+            return param.argument;
+          }
+          if (param.type === 'AssignmentPattern') {
+            return param.left;
+          }
+          return param;
+        }).filter(identifier.isNamed).forEach(param => {
+          scope.idents.add(param.name);
+        });
+
+        /**
+         * If a FunctionExpression has an id, it is a named function expression.
+         * The function expression name shadows the module scope identifier, so
+         * we don't want to count reads of module identifers that have the same name.
+         * They also do not cause a SyntaxError if the function expression name is
+         * the same as a module scope identifier.
+         *
+         * TODO: Is this necessary for FunctionDeclaration?
+         */
+        if (node.type === 'FunctionExpression' && node.id) {
+          scope.idents.add(node.id.name);
+        }
+
+        // First add the function to any previous scopes
+        if (scopes.length > 0) {
+          scopes[scopes.length - 1].idents.add(name);
+        }
+
+        // Then add the function scope to the scopes stack
+        scopes.push(scope);
+      }
+      break;
+    case 'ClassDeclaration':
+      {
+        const className = node.id ? node.id.name : 'anonymous';
+
+        // First add the class to any previous scopes
+        if (scopes.length > 0) {
+          scopes[scopes.length - 1].idents.add(className);
+        }
+
+        // Then add the class to the scopes stack
+        scopes.push({
+          node,
+          name: className,
+          type: 'Class',
+          idents: new Set()
+        });
+      }
+      break;
+    case 'ClassExpression':
+      {}
+      break;
+    case 'VariableDeclaration':
+      if (scopes.length > 0) {
+        const scope = scopes[scopes.length - 1];
+        node.declarations.forEach(decl => {
+          if (decl.type === 'VariableDeclarator' && decl.id.type === 'Identifier') {
+            scope.idents.add(decl.id.name);
+          }
+        });
+      }
+      break;
+  }
+};
+
+/**
+ * Collects all module scope identifiers in the AST.
+ *
+ * Ignores identifiers that are in functions or classes.
+ * Ignores new scopes for StaticBlock nodes (can only reference static class members).
+ *
+ * Special case handling for these which create their own scopes,
+ * but are also valid module scope identifiers:
+ * - ClassDeclaration
+ * - FunctionDeclaration
+ *
+ * Special case handling for var inside BlockStatement
+ * which are also valid module scope identifiers.
+ */
+const collectModuleIdentifiers = async (ast, hoisting = true) => {
+  const identifiers = new Map();
+  const globalReads = new Map();
+  const scopes = [];
+  await ancestorWalk(ast, {
+    enter(node, ancestors) {
+      const {
+        type
+      } = node;
+      collectScopeIdentifiers(node, scopes);
+
+      // Add module scope identifiers to the registry map
+
+      if (type === 'Identifier') {
+        const {
+          name
+        } = node;
+        const meta = identifiers.get(name) ?? {
+          declare: [],
+          read: []
+        };
+        const isDeclaration = identifier.isDeclaration(ancestors);
+        const inScope = scopes.some(scope => scope.idents.has(name) || scope.name === name);
+        if (hoisting && !identifier.isDeclaration(ancestors) && !identifier.isFunctionExpressionId(ancestors) && !identifier.isExportSpecifierAlias(ancestors) && !identifier.isClassPropertyKey(ancestors) && !identifier.isMethodDefinitionKey(ancestors) && !identifier.isMemberKey(ancestors) && !identifier.isPropertyKey(ancestors) && !identifier.isIife(ancestors) && !inScope) {
+          if (globalReads.has(name)) {
+            globalReads.get(name)?.push(node);
+          } else {
+            globalReads.set(name, [node]);
+          }
+        }
+        if (isDeclaration) {
+          const isModuleScope = identifier.isModuleScope(ancestors);
+          const isClassOrFuncDeclaration = identifier.isClassOrFuncDeclarationId(ancestors);
+          const isVarDeclarationInGlobalScope = identifier.isVarDeclarationInGlobalScope(ancestors);
+          if (isModuleScope || isClassOrFuncDeclaration || isVarDeclarationInGlobalScope) {
+            meta.declare.push(node);
+
+            // Check for hoisted reads
+            if (hoisting && globalReads.has(name)) {
+              const reads = globalReads.get(name);
+              if (reads) {
+                reads.forEach(read => {
+                  if (!meta.read.includes(read)) {
+                    meta.read.push(read);
+                  }
+                });
+              }
+            }
+            identifiers.set(name, meta);
+          }
+        } else {
+          if (identifiers.has(name) && !inScope && !identifier.isIife(ancestors) && !identifier.isFunctionExpressionId(ancestors) && !identifier.isExportSpecifierAlias(ancestors) && !identifier.isClassPropertyKey(ancestors) && !identifier.isMethodDefinitionKey(ancestors) && !identifier.isMemberKey(ancestors) && !identifier.isPropertyKey(ancestors)) {
+            // Closure is referencing module scope identifier
+            meta.read.push(node);
+          }
+        }
+      }
+    },
+    leave(node) {
+      const {
+        type
+      } = node;
+      if (scopeNodes.includes(type)) {
+        scopes.pop();
+      }
+    }
+  });
+  return identifiers;
+};
+export { collectScopeIdentifiers, collectModuleIdentifiers };

package/dist/utils/lang.js

@@ -0,0 +1,20 @@
+import { extname } from 'node:path';
+
+// Determine language from filename extension for specifier rewrite.
+
+const getLangFromExt = filename => {
+  const ext = extname(filename);
+  if (ext.endsWith('.js')) {
+    return 'js';
+  }
+  if (ext.endsWith('.ts')) {
+    return 'ts';
+  }
+  if (ext === '.tsx') {
+    return 'tsx';
+  }
+  if (ext === '.jsx') {
+    return 'jsx';
+  }
+};
+export { getLangFromExt };

package/dist/utils/url.js

@@ -0,0 +1,10 @@
+// URL validation used when rewriting import.meta.url assignments.
+const isValidUrl = url => {
+  try {
+    new URL(url);
+    return true;
+  } catch {
+    return false;
+  }
+};
+export { isValidUrl };

package/dist/utils.d.ts

@@ -0,0 +1,26 @@
+import type { Node } from 'oxc-parser';
+import type { Specifier } from '@knighted/specifier';
+import type { IdentMeta, Scope, CjsExport } from './types.js';
+type UpdateSrcLang = Parameters<Specifier['updateSrc']>[1];
+declare const getLangFromExt: (filename: string) => UpdateSrcLang;
+declare const isValidUrl: (url: string) => boolean;
+declare const exportsRename = "__exports";
+declare const requireMainRgx: RegExp;
+declare const collectCjsExports: (ast: Node) => Promise<Map<string, CjsExport>>;
+declare const collectScopeIdentifiers: (node: Node, scopes: Scope[]) => void;
+/**
+ * Collects all module scope identifiers in the AST.
+ *
+ * Ignores identifiers that are in functions or classes.
+ * Ignores new scopes for StaticBlock nodes (can only reference static class members).
+ *
+ * Special case handling for these which create their own scopes,
+ * but are also valid module scope identifiers:
+ * - ClassDeclaration
+ * - FunctionDeclaration
+ *
+ * Special case handling for var inside BlockStatement
+ * which are also valid module scope identifiers.
+ */
+declare const collectModuleIdentifiers: (ast: Node, hoisting?: boolean) => Promise<Map<string, IdentMeta>>;
+export { getLangFromExt, isValidUrl, collectScopeIdentifiers, collectModuleIdentifiers, collectCjsExports, exportsRename, requireMainRgx, };