89 lines
2.3 KiB
JavaScript
89 lines
2.3 KiB
JavaScript
|
import {ContextualKeyword} from "../parser/tokenizer/keywords";
|
||
|
|
import {TokenType as tt} from "../parser/tokenizer/types";
|
||
|
|
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Special case code to scan for imported names in ESM TypeScript. We need to do this so we can
|
||
|
|
* properly get globals so we can compute shadowed globals.
|
||
|
|
*
|
||
|
|
* This is similar to logic in CJSImportProcessor, but trimmed down to avoid logic with CJS
|
||
|
|
* replacement and flow type imports.
|
||
|
|
*/
|
||
|
|
export default function getTSImportedNames(tokens) {
|
||
|
|
const importedNames = new Set();
|
||
|
|
for (let i = 0; i < tokens.tokens.length; i++) {
|
||
|
|
if (
|
||
|
|
tokens.matches1AtIndex(i, tt._import) &&
|
||
|
|
!tokens.matches3AtIndex(i, tt._import, tt.name, tt.eq)
|
||
|
|
) {
|
||
|
|
collectNamesForImport(tokens, i, importedNames);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return importedNames;
|
||
|
|
}
|
||
|
|
|
||
|
|
function collectNamesForImport(
|
||
|
|
tokens,
|
||
|
|
index,
|
||
|
|
importedNames,
|
||
|
|
) {
|
||
|
|
index++;
|
||
|
|
|
||
|
|
if (tokens.matches1AtIndex(index, tt.parenL)) {
|
||
|
|
// Dynamic import, so nothing to do
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (tokens.matches1AtIndex(index, tt.name)) {
|
||
|
|
importedNames.add(tokens.identifierNameAtIndex(index));
|
||
|
|
index++;
|
||
|
|
if (tokens.matches1AtIndex(index, tt.comma)) {
|
||
|
|
index++;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (tokens.matches1AtIndex(index, tt.star)) {
|
||
|
|
// * as
|
||
|
|
index += 2;
|
||
|
|
importedNames.add(tokens.identifierNameAtIndex(index));
|
||
|
|
index++;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (tokens.matches1AtIndex(index, tt.braceL)) {
|
||
|
|
index++;
|
||
|
|
collectNamesForNamedImport(tokens, index, importedNames);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
function collectNamesForNamedImport(
|
||
|
|
tokens,
|
||
|
|
index,
|
||
|
|
importedNames,
|
||
|
|
) {
|
||
|
|
while (true) {
|
||
|
|
if (tokens.matches1AtIndex(index, tt.braceR)) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
// We care about the local name, which might be the first token, or if there's an "as", is the
|
||
|
|
// one after that.
|
||
|
|
let name = tokens.identifierNameAtIndex(index);
|
||
|
|
index++;
|
||
|
|
if (tokens.matchesContextualAtIndex(index, ContextualKeyword._as)) {
|
||
|
|
index++;
|
||
|
|
name = tokens.identifierNameAtIndex(index);
|
||
|
|
index++;
|
||
|
|
}
|
||
|
|
importedNames.add(name);
|
||
|
|
if (tokens.matches2AtIndex(index, tt.comma, tt.braceR)) {
|
||
|
|
return;
|
||
|
|
} else if (tokens.matches1AtIndex(index, tt.braceR)) {
|
||
|
|
return;
|
||
|
|
} else if (tokens.matches1AtIndex(index, tt.comma)) {
|
||
|
|
index++;
|
||
|
|
} else {
|
||
|
|
throw new Error(`Unexpected token: ${JSON.stringify(tokens.tokens[index])}`);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|