You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
243 lines
7.4 KiB
243 lines
7.4 KiB
2 years ago
|
import * as _acorn from "acorn";
|
||
|
|
||
|
const leftCurlyBrace = "{".charCodeAt(0);
|
||
|
const space = " ".charCodeAt(0);
|
||
|
|
||
|
const keyword = "assert";
|
||
|
const FUNC_STATEMENT = 1, FUNC_HANGING_STATEMENT = 2, FUNC_NULLABLE_ID = 4
|
||
|
|
||
|
export function importAssertions(Parser) {
|
||
|
// Use supplied version acorn version if present, to avoid
|
||
|
// reference mismatches due to different acorn versions. This
|
||
|
// allows this plugin to be used with Rollup which supplies
|
||
|
// its own internal version of acorn and thereby sidesteps
|
||
|
// the package manager.
|
||
|
const acorn = Parser.acorn || _acorn;
|
||
|
const { tokTypes: tt, TokenType } = acorn;
|
||
|
|
||
|
return class extends Parser {
|
||
|
constructor(...args) {
|
||
|
super(...args);
|
||
|
this.assertToken = new TokenType(keyword);
|
||
|
}
|
||
|
|
||
|
_codeAt(i) {
|
||
|
return this.input.charCodeAt(i);
|
||
|
}
|
||
|
|
||
|
_eat(t) {
|
||
|
if (this.type !== t) {
|
||
|
this.unexpected();
|
||
|
}
|
||
|
this.next();
|
||
|
}
|
||
|
|
||
|
readToken(code) {
|
||
|
let i = 0;
|
||
|
for (; i < keyword.length; i++) {
|
||
|
if (this._codeAt(this.pos + i) !== keyword.charCodeAt(i)) {
|
||
|
return super.readToken(code);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// ensure that the keyword is at the correct location
|
||
|
// ie `assert{...` or `assert {...`
|
||
|
for (;; i++) {
|
||
|
if (this._codeAt(this.pos + i) === leftCurlyBrace) {
|
||
|
// Found '{'
|
||
|
break;
|
||
|
} else if (this._codeAt(this.pos + i) === space) {
|
||
|
// white space is allowed between `assert` and `{`, so continue.
|
||
|
continue;
|
||
|
} else {
|
||
|
return super.readToken(code);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// If we're inside a dynamic import expression we'll parse
|
||
|
// the `assert` keyword as a standard object property name
|
||
|
// ie `import(""./foo.json", { assert: { type: "json" } })`
|
||
|
if (this.type.label === "{") {
|
||
|
return super.readToken(code);
|
||
|
}
|
||
|
|
||
|
this.pos += keyword.length;
|
||
|
return this.finishToken(this.assertToken);
|
||
|
}
|
||
|
|
||
|
parseDynamicImport(node) {
|
||
|
this.next(); // skip `(`
|
||
|
|
||
|
// Parse node.source.
|
||
|
node.source = this.parseMaybeAssign();
|
||
|
|
||
|
if (this.eat(tt.comma)) {
|
||
|
const obj = this.parseObj(false);
|
||
|
node.arguments = [obj];
|
||
|
}
|
||
|
this._eat(tt.parenR);
|
||
|
return this.finishNode(node, "ImportExpression");
|
||
|
}
|
||
|
|
||
|
// ported from acorn/src/statement.js pp.parseExport
|
||
|
parseExport(node, exports) {
|
||
|
this.next();
|
||
|
// export * from '...'
|
||
|
if (this.eat(tt.star)) {
|
||
|
if (this.options.ecmaVersion >= 11) {
|
||
|
if (this.eatContextual("as")) {
|
||
|
node.exported = this.parseIdent(true);
|
||
|
this.checkExport(exports, node.exported.name, this.lastTokStart);
|
||
|
} else {
|
||
|
node.exported = null;
|
||
|
}
|
||
|
}
|
||
|
this.expectContextual("from");
|
||
|
if (this.type !== tt.string) { this.unexpected(); }
|
||
|
node.source = this.parseExprAtom();
|
||
|
|
||
|
if (this.type === this.assertToken) {
|
||
|
this.next();
|
||
|
const assertions = this.parseImportAssertions();
|
||
|
if (assertions) {
|
||
|
node.assertions = assertions;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
this.semicolon();
|
||
|
return this.finishNode(node, "ExportAllDeclaration")
|
||
|
}
|
||
|
if (this.eat(tt._default)) { // export default ...
|
||
|
this.checkExport(exports, "default", this.lastTokStart);
|
||
|
var isAsync;
|
||
|
if (this.type === tt._function || (isAsync = this.isAsyncFunction())) {
|
||
|
var fNode = this.startNode();
|
||
|
this.next();
|
||
|
if (isAsync) { this.next(); }
|
||
|
node.declaration = this.parseFunction(fNode, FUNC_STATEMENT | FUNC_NULLABLE_ID, false, isAsync);
|
||
|
} else if (this.type === tt._class) {
|
||
|
var cNode = this.startNode();
|
||
|
node.declaration = this.parseClass(cNode, "nullableID");
|
||
|
} else {
|
||
|
node.declaration = this.parseMaybeAssign();
|
||
|
this.semicolon();
|
||
|
}
|
||
|
return this.finishNode(node, "ExportDefaultDeclaration")
|
||
|
}
|
||
|
// export var|const|let|function|class ...
|
||
|
if (this.shouldParseExportStatement()) {
|
||
|
node.declaration = this.parseStatement(null);
|
||
|
if (node.declaration.type === "VariableDeclaration")
|
||
|
{ this.checkVariableExport(exports, node.declaration.declarations); }
|
||
|
else
|
||
|
{ this.checkExport(exports, node.declaration.id.name, node.declaration.id.start); }
|
||
|
node.specifiers = [];
|
||
|
node.source = null;
|
||
|
} else { // export { x, y as z } [from '...']
|
||
|
node.declaration = null;
|
||
|
node.specifiers = this.parseExportSpecifiers(exports);
|
||
|
if (this.eatContextual("from")) {
|
||
|
if (this.type !== tt.string) { this.unexpected(); }
|
||
|
node.source = this.parseExprAtom();
|
||
|
|
||
|
if (this.type === this.assertToken) {
|
||
|
this.next();
|
||
|
const assertions = this.parseImportAssertions();
|
||
|
if (assertions) {
|
||
|
node.assertions = assertions;
|
||
|
}
|
||
|
}
|
||
|
} else {
|
||
|
for (var i = 0, list = node.specifiers; i < list.length; i += 1) {
|
||
|
// check for keywords used as local names
|
||
|
var spec = list[i];
|
||
|
|
||
|
this.checkUnreserved(spec.local);
|
||
|
// check if export is defined
|
||
|
this.checkLocalExport(spec.local);
|
||
|
}
|
||
|
|
||
|
node.source = null;
|
||
|
}
|
||
|
this.semicolon();
|
||
|
}
|
||
|
return this.finishNode(node, "ExportNamedDeclaration")
|
||
|
}
|
||
|
|
||
|
parseImport(node) {
|
||
|
this.next();
|
||
|
// import '...'
|
||
|
if (this.type === tt.string) {
|
||
|
node.specifiers = [];
|
||
|
node.source = this.parseExprAtom();
|
||
|
} else {
|
||
|
node.specifiers = this.parseImportSpecifiers();
|
||
|
this.expectContextual("from");
|
||
|
node.source =
|
||
|
this.type === tt.string ? this.parseExprAtom() : this.unexpected();
|
||
|
}
|
||
|
|
||
|
if (this.type === this.assertToken) {
|
||
|
this.next();
|
||
|
const assertions = this.parseImportAssertions();
|
||
|
if (assertions) {
|
||
|
node.assertions = assertions;
|
||
|
}
|
||
|
}
|
||
|
this.semicolon();
|
||
|
return this.finishNode(node, "ImportDeclaration");
|
||
|
}
|
||
|
|
||
|
parseImportAssertions() {
|
||
|
this._eat(tt.braceL);
|
||
|
const attrs = this.parseAssertEntries();
|
||
|
this._eat(tt.braceR);
|
||
|
return attrs;
|
||
|
}
|
||
|
|
||
|
parseAssertEntries() {
|
||
|
const attrs = [];
|
||
|
const attrNames = new Set();
|
||
|
|
||
|
do {
|
||
|
if (this.type === tt.braceR) {
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
const node = this.startNode();
|
||
|
|
||
|
// parse AssertionKey : IdentifierName, StringLiteral
|
||
|
let assertionKeyNode;
|
||
|
if (this.type === tt.string) {
|
||
|
assertionKeyNode = this.parseLiteral(this.value);
|
||
|
} else {
|
||
|
assertionKeyNode = this.parseIdent(true);
|
||
|
}
|
||
|
this.next();
|
||
|
node.key = assertionKeyNode;
|
||
|
|
||
|
// check if we already have an entry for an attribute
|
||
|
// if a duplicate entry is found, throw an error
|
||
|
// for now this logic will come into play only when someone declares `type` twice
|
||
|
if (attrNames.has(node.key.name)) {
|
||
|
this.raise(this.pos, "Duplicated key in assertions");
|
||
|
}
|
||
|
attrNames.add(node.key.name);
|
||
|
|
||
|
if (this.type !== tt.string) {
|
||
|
this.raise(
|
||
|
this.pos,
|
||
|
"Only string is supported as an assertion value"
|
||
|
);
|
||
|
}
|
||
|
|
||
|
node.value = this.parseLiteral(this.value);
|
||
|
|
||
|
attrs.push(this.finishNode(node, "ImportAttribute"));
|
||
|
} while (this.eat(tt.comma));
|
||
|
|
||
|
return attrs;
|
||
|
}
|
||
|
};
|
||
|
}
|