123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242 |
- 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;
- }
- };
- }
|