@esportsplus/typescript 0.21.0 → 0.22.0

package/build/compiler/imports.d.ts

@@ -0,0 +1,18 @@
+import { ts } from '../index.js';
+type ImportInfo = {
+    end: number;
+    specifiers: Map<string, string>;
+    start: number;
+};
+type ModifyOptions = {
+    add?: Iterable<string>;
+    remove?: Iterable<string>;
+};
+declare const _default: {
+    find: (sourceFile: ts.SourceFile, packageName: string) => ImportInfo[];
+    isFromPackage: (node: ts.Identifier, packageName: string, checker?: ts.TypeChecker) => boolean;
+    modify: (sourceCode: string, sourceFile: ts.SourceFile, packageName: string, options: ModifyOptions) => string;
+    trace: (node: ts.Identifier, checker: ts.TypeChecker) => string | null;
+};
+export default _default;
+export type { ImportInfo, ModifyOptions };

package/build/compiler/imports.js

@@ -0,0 +1,82 @@
+import { ts } from '../index.js';
+import code from './code.js';
+const find = (sourceFile, packageName) => {
+    let imports = [];
+    for (let i = 0, n = sourceFile.statements.length; i < n; i++) {
+        let stmt = sourceFile.statements[i];
+        if (!ts.isImportDeclaration(stmt)) {
+            continue;
+        }
+        let moduleSpecifier = stmt.moduleSpecifier;
+        if (!ts.isStringLiteral(moduleSpecifier) || moduleSpecifier.text !== packageName) {
+            continue;
+        }
+        let bindings = stmt.importClause?.namedBindings, specifiers = new Map();
+        if (bindings && ts.isNamedImports(bindings)) {
+            for (let j = 0, m = bindings.elements.length; j < m; j++) {
+                let element = bindings.elements[j], name = element.name.text, propertyName = element.propertyName?.text || name;
+                specifiers.set(propertyName, name);
+            }
+        }
+        imports.push({ end: stmt.end, specifiers, start: stmt.getStart() });
+    }
+    return imports;
+};
+const isFromPackage = (node, packageName, checker) => {
+    if (!checker) {
+        return true;
+    }
+    let origin = trace(node, checker);
+    return origin !== null && origin.includes(packageName);
+};
+const modify = (sourceCode, sourceFile, packageName, options) => {
+    if (!options.add && !options.remove) {
+        return sourceCode;
+    }
+    let add = options.add ? new Set(options.add) : null, imports = find(sourceFile, packageName), remove = options.remove ? new Set(options.remove) : null;
+    if (imports.length === 0) {
+        if (!add || add.size === 0) {
+            return sourceCode;
+        }
+        return `import { ${[...add].sort().join(', ')} } from '${packageName}';\n` + sourceCode;
+    }
+    let specifiers = new Set();
+    for (let i = 0, n = imports.length; i < n; i++) {
+        for (let [name, alias] of imports[i].specifiers) {
+            if (!remove || (!remove.has(name) && !remove.has(alias))) {
+                specifiers.add(name === alias ? name : `${name} as ${alias}`);
+            }
+        }
+    }
+    if (add) {
+        for (let name of add) {
+            specifiers.add(name);
+        }
+    }
+    let replacements = [], sorted = [...specifiers].sort().join(', ');
+    for (let i = 0, n = imports.length; i < n; i++) {
+        replacements.push({
+            end: imports[i].end,
+            newText: i === 0 && specifiers.size > 0
+                ? `import { ${sorted} } from '${packageName}';`
+                : '',
+            start: imports[i].start
+        });
+    }
+    return code.replaceReverse(sourceCode, replacements);
+};
+const trace = (node, checker) => {
+    let symbol = checker.getSymbolAtLocation(node);
+    if (!symbol) {
+        return null;
+    }
+    if (symbol.flags & ts.SymbolFlags.Alias) {
+        symbol = checker.getAliasedSymbol(symbol);
+    }
+    let declarations = symbol.getDeclarations();
+    if (!declarations || declarations.length === 0) {
+        return null;
+    }
+    return declarations[0].getSourceFile().fileName;
+};
+export default { find, isFromPackage, modify, trace };

package/build/compiler/index.d.ts

@@ -1,5 +1,6 @@
 export * as ast from './ast/index.js';
 export { default as code } from './code.js';
+export { default as imports } from './imports.js';
 export { default as plugin } from './plugins/index.js';
 export { default as program } from './program.js';
 export { default as uid } from './uid.js';

package/build/compiler/index.js

@@ -1,5 +1,6 @@
 export * as ast from './ast/index.js';
 export { default as code } from './code.js';
+export { default as imports } from './imports.js';
 export { default as plugin } from './plugins/index.js';
 export { default as program } from './program.js';
 export { default as uid } from './uid.js';

package/package.json

@@ -37,7 +37,7 @@
   },
   "type": "module",
   "types": "build/index.d.ts",
-  "version": "0.21.0",
+  "version": "0.22.0",
   "scripts": {
     "build": "tsc && tsc-alias",
     "-": "-"

package/src/compiler/imports.ts

@@ -0,0 +1,147 @@
+import type { Replacement } from './types.js';
+import { ts } from '~/index.js';
+import code from './code.js';
+
+
+type ImportInfo = {
+    end: number;
+    specifiers: Map<string, string>;
+    start: number;
+};
+
+type ModifyOptions = {
+    add?: Iterable<string>;
+    remove?: Iterable<string>;
+};
+
+
+// Find all named imports from a specific package
+const find = (sourceFile: ts.SourceFile, packageName: string): ImportInfo[] => {
+    let imports: ImportInfo[] = [];
+
+    for (let i = 0, n = sourceFile.statements.length; i < n; i++) {
+        let stmt = sourceFile.statements[i];
+
+        if (!ts.isImportDeclaration(stmt)) {
+            continue;
+        }
+
+        let moduleSpecifier = stmt.moduleSpecifier;
+
+        if (!ts.isStringLiteral(moduleSpecifier) || moduleSpecifier.text !== packageName) {
+            continue;
+        }
+
+        let bindings = stmt.importClause?.namedBindings,
+            specifiers = new Map<string, string>();
+
+        if (bindings && ts.isNamedImports(bindings)) {
+            for (let j = 0, m = bindings.elements.length; j < m; j++) {
+                let element = bindings.elements[j],
+                    name = element.name.text,
+                    propertyName = element.propertyName?.text || name;
+
+                specifiers.set(propertyName, name);
+            }
+        }
+
+        imports.push({ end: stmt.end, specifiers, start: stmt.getStart() });
+    }
+
+    return imports;
+};
+
+// Check if identifier traces back to a specific package
+const isFromPackage = (node: ts.Identifier, packageName: string, checker?: ts.TypeChecker): boolean => {
+    if (!checker) {
+        return true;
+    }
+
+    let origin = trace(node, checker);
+
+    return origin !== null && origin.includes(packageName);
+};
+
+
+// Modify imports: remove specified, add needed, delete if empty
+const modify = (
+    sourceCode: string,
+    sourceFile: ts.SourceFile,
+    packageName: string,
+    options: ModifyOptions
+): string => {
+    // Fast path: nothing to change
+    if (!options.add && !options.remove) {
+        return sourceCode;
+    }
+
+    let add = options.add ? new Set(options.add) : null,
+        imports = find(sourceFile, packageName),
+        remove = options.remove ? new Set(options.remove) : null;
+
+    if (imports.length === 0) {
+        if (!add || add.size === 0) {
+            return sourceCode;
+        }
+
+        return `import { ${[...add].sort().join(', ')} } from '${packageName}';\n` + sourceCode;
+    }
+
+    // Collect all non-removed specifiers from existing imports
+    let specifiers = new Set<string>();
+
+    for (let i = 0, n = imports.length; i < n; i++) {
+        for (let [name, alias] of imports[i].specifiers) {
+            if (!remove || (!remove.has(name) && !remove.has(alias))) {
+                specifiers.add(name === alias ? name : `${name} as ${alias}`);
+            }
+        }
+    }
+
+    if (add) {
+        for (let name of add) {
+            specifiers.add(name);
+        }
+    }
+
+    // Build replacements - replace first import, remove others
+    let replacements: Replacement[] = [],
+        sorted = [...specifiers].sort().join(', ');
+
+    for (let i = 0, n = imports.length; i < n; i++) {
+        replacements.push({
+            end: imports[i].end,
+            newText: i === 0 && specifiers.size > 0
+                ? `import { ${sorted} } from '${packageName}';`
+                : '',
+            start: imports[i].start
+        });
+    }
+
+    return code.replaceReverse(sourceCode, replacements);
+};
+
+// Trace symbol through re-exports to find original declaration source file
+const trace = (node: ts.Identifier, checker: ts.TypeChecker): string | null => {
+    let symbol = checker.getSymbolAtLocation(node);
+
+    if (!symbol) {
+        return null;
+    }
+
+    if (symbol.flags & ts.SymbolFlags.Alias) {
+        symbol = checker.getAliasedSymbol(symbol);
+    }
+
+    let declarations = symbol.getDeclarations();
+
+    if (!declarations || declarations.length === 0) {
+        return null;
+    }
+
+    return declarations[0].getSourceFile().fileName;
+};
+
+
+export default { find, isFromPackage, modify, trace };
+export type { ImportInfo, ModifyOptions };

package/src/compiler/index.ts

@@ -1,5 +1,6 @@
 export * as ast from './ast';
 export { default as code } from './code';
+export { default as imports } from './imports';
 export { default as plugin } from './plugins';
 export { default as program } from './program';
 export { default as uid } from './uid';