123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665 |
- "use strict";
- Object.defineProperty(exports, "__esModule", {
- value: true
- });
- exports.default = void 0;
- var _buffer = require("./buffer");
- var n = require("./node");
- var _t = require("@babel/types");
- var generatorFunctions = require("./generators");
- const {
- isFunction,
- isStatement,
- isClassBody,
- isTSInterfaceBody,
- isTSEnumDeclaration
- } = _t;
- const SCIENTIFIC_NOTATION = /e/i;
- const ZERO_DECIMAL_INTEGER = /\.0+$/;
- const NON_DECIMAL_LITERAL = /^0[box]/;
- const PURE_ANNOTATION_RE = /^\s*[@#]__PURE__\s*$/;
- const HAS_NEWLINE = /[\n\r\u2028\u2029]/;
- const HAS_BlOCK_COMMENT_END = /\*\//;
- const {
- needsParens
- } = n;
- class Printer {
- constructor(format, map) {
- this.inForStatementInitCounter = 0;
- this._printStack = [];
- this._indent = 0;
- this._indentChar = 0;
- this._indentRepeat = 0;
- this._insideAux = false;
- this._parenPushNewlineState = null;
- this._noLineTerminator = false;
- this._printAuxAfterOnNextUserNode = false;
- this._printedComments = new Set();
- this._endsWithInteger = false;
- this._endsWithWord = false;
- this._lastCommentLine = 0;
- this._endsWithInnerRaw = false;
- this._indentInnerComments = true;
- this.format = format;
- this._buf = new _buffer.default(map);
- this._indentChar = format.indent.style.charCodeAt(0);
- this._indentRepeat = format.indent.style.length;
- }
- generate(ast) {
- this.print(ast);
- this._maybeAddAuxComment();
- return this._buf.get();
- }
- indent() {
- if (this.format.compact || this.format.concise) return;
- this._indent++;
- }
- dedent() {
- if (this.format.compact || this.format.concise) return;
- this._indent--;
- }
- semicolon(force = false) {
- this._maybeAddAuxComment();
- if (force) {
- this._appendChar(59);
- } else {
- this._queue(59);
- }
- this._noLineTerminator = false;
- }
- rightBrace() {
- if (this.format.minified) {
- this._buf.removeLastSemicolon();
- }
- this.tokenChar(125);
- }
- space(force = false) {
- if (this.format.compact) return;
- if (force) {
- this._space();
- } else if (this._buf.hasContent()) {
- const lastCp = this.getLastChar();
- if (lastCp !== 32 && lastCp !== 10) {
- this._space();
- }
- }
- }
- word(str, noLineTerminatorAfter = false) {
- this._maybePrintInnerComments();
- if (this._endsWithWord || str.charCodeAt(0) === 47 && this.endsWith(47)) {
- this._space();
- }
- this._maybeAddAuxComment();
- this._append(str, false);
- this._endsWithWord = true;
- this._noLineTerminator = noLineTerminatorAfter;
- }
- number(str) {
- this.word(str);
- this._endsWithInteger = Number.isInteger(+str) && !NON_DECIMAL_LITERAL.test(str) && !SCIENTIFIC_NOTATION.test(str) && !ZERO_DECIMAL_INTEGER.test(str) && str.charCodeAt(str.length - 1) !== 46;
- }
- token(str, maybeNewline = false) {
- this._maybePrintInnerComments();
- const lastChar = this.getLastChar();
- const strFirst = str.charCodeAt(0);
- if (lastChar === 33 && str === "--" ||
- strFirst === 43 && lastChar === 43 || strFirst === 45 && lastChar === 45 ||
- strFirst === 46 && this._endsWithInteger) {
- this._space();
- }
- this._maybeAddAuxComment();
- this._append(str, maybeNewline);
- this._noLineTerminator = false;
- }
- tokenChar(char) {
- this._maybePrintInnerComments();
- const lastChar = this.getLastChar();
- if (
- char === 43 && lastChar === 43 || char === 45 && lastChar === 45 ||
- char === 46 && this._endsWithInteger) {
- this._space();
- }
- this._maybeAddAuxComment();
- this._appendChar(char);
- this._noLineTerminator = false;
- }
- newline(i = 1, force) {
- if (i <= 0) return;
- if (!force) {
- if (this.format.retainLines || this.format.compact) return;
- if (this.format.concise) {
- this.space();
- return;
- }
- }
- if (i > 2) i = 2;
- i -= this._buf.getNewlineCount();
- for (let j = 0; j < i; j++) {
- this._newline();
- }
- return;
- }
- endsWith(char) {
- return this.getLastChar() === char;
- }
- getLastChar() {
- return this._buf.getLastChar();
- }
- endsWithCharAndNewline() {
- return this._buf.endsWithCharAndNewline();
- }
- removeTrailingNewline() {
- this._buf.removeTrailingNewline();
- }
- exactSource(loc, cb) {
- if (!loc) return cb();
- this._catchUp("start", loc);
- this._buf.exactSource(loc, cb);
- }
- source(prop, loc) {
- if (!loc) return;
- this._catchUp(prop, loc);
- this._buf.source(prop, loc);
- }
- sourceWithOffset(prop, loc, lineOffset, columnOffset) {
- if (!loc) return;
- this._catchUp(prop, loc);
- this._buf.sourceWithOffset(prop, loc, lineOffset, columnOffset);
- }
- withSource(prop, loc, cb) {
- if (!loc) return cb();
- this._catchUp(prop, loc);
- this._buf.withSource(prop, loc, cb);
- }
- _space() {
- this._queue(32);
- }
- _newline() {
- this._queue(10);
- }
- _append(str, maybeNewline) {
- this._maybeAddParen(str);
- this._maybeIndent(str.charCodeAt(0));
- this._buf.append(str, maybeNewline);
- this._endsWithWord = false;
- this._endsWithInteger = false;
- }
- _appendChar(char) {
- this._maybeAddParenChar(char);
- this._maybeIndent(char);
- this._buf.appendChar(char);
- this._endsWithWord = false;
- this._endsWithInteger = false;
- }
- _queue(char) {
- this._maybeAddParenChar(char);
- this._maybeIndent(char);
- this._buf.queue(char);
- this._endsWithWord = false;
- this._endsWithInteger = false;
- }
- _maybeIndent(firstChar) {
- if (this._indent && firstChar !== 10 && this.endsWith(10)) {
- this._buf.queueIndentation(this._indentChar, this._getIndent());
- }
- }
- _shouldIndent(firstChar) {
- if (this._indent && firstChar !== 10 && this.endsWith(10)) {
- return true;
- }
- }
- _maybeAddParenChar(char) {
- const parenPushNewlineState = this._parenPushNewlineState;
- if (!parenPushNewlineState) return;
- if (char === 32) {
- return;
- }
- if (char !== 10) {
- this._parenPushNewlineState = null;
- return;
- }
- this.tokenChar(40);
- this.indent();
- parenPushNewlineState.printed = true;
- }
- _maybeAddParen(str) {
- const parenPushNewlineState = this._parenPushNewlineState;
- if (!parenPushNewlineState) return;
- const len = str.length;
- let i;
- for (i = 0; i < len && str.charCodeAt(i) === 32; i++) continue;
- if (i === len) {
- return;
- }
- const cha = str.charCodeAt(i);
- if (cha !== 10) {
- if (
- cha !== 47 ||
- i + 1 === len) {
- this._parenPushNewlineState = null;
- return;
- }
- const chaPost = str.charCodeAt(i + 1);
- if (chaPost === 42) {
- if (PURE_ANNOTATION_RE.test(str.slice(i + 2, len - 2))) {
- return;
- }
- } else if (chaPost !== 47) {
- this._parenPushNewlineState = null;
- return;
- }
- }
- this.tokenChar(40);
- this.indent();
- parenPushNewlineState.printed = true;
- }
- catchUp(line) {
- if (!this.format.retainLines) return;
- const count = line - this._buf.getCurrentLine();
- for (let i = 0; i < count; i++) {
- this._newline();
- }
- }
- _catchUp(prop, loc) {
- if (!this.format.retainLines) return;
- const pos = loc ? loc[prop] : null;
- if ((pos == null ? void 0 : pos.line) != null) {
- const count = pos.line - this._buf.getCurrentLine();
- for (let i = 0; i < count; i++) {
- this._newline();
- }
- }
- }
- _getIndent() {
- return this._indentRepeat * this._indent;
- }
- printTerminatorless(node, parent, isLabel) {
- if (isLabel) {
- this._noLineTerminator = true;
- this.print(node, parent);
- } else {
- const terminatorState = {
- printed: false
- };
- this._parenPushNewlineState = terminatorState;
- this.print(node, parent);
- if (terminatorState.printed) {
- this.dedent();
- this.newline();
- this.tokenChar(41);
- }
- }
- }
- print(node, parent, noLineTerminatorAfter,
- trailingCommentsLineOffset, forceParens) {
- if (!node) return;
- this._endsWithInnerRaw = false;
- const nodeType = node.type;
- const format = this.format;
- const oldConcise = format.concise;
- if (
- node._compact) {
- format.concise = true;
- }
- const printMethod = this[nodeType];
- if (printMethod === undefined) {
- throw new ReferenceError(`unknown node of type ${JSON.stringify(nodeType)} with constructor ${JSON.stringify(node.constructor.name)}`);
- }
- this._printStack.push(node);
- const oldInAux = this._insideAux;
- this._insideAux = node.loc == undefined;
- this._maybeAddAuxComment(this._insideAux && !oldInAux);
- let shouldPrintParens = false;
- if (forceParens) {
- shouldPrintParens = true;
- } else if (format.retainFunctionParens && nodeType === "FunctionExpression" && node.extra && node.extra.parenthesized) {
- shouldPrintParens = true;
- } else {
- shouldPrintParens = needsParens(node, parent, this._printStack);
- }
- if (shouldPrintParens) {
- this.tokenChar(40);
- this._endsWithInnerRaw = false;
- }
- this._lastCommentLine = 0;
- this._printLeadingComments(node, parent);
- const loc = nodeType === "Program" || nodeType === "File" ? null : node.loc;
- this.exactSource(loc, printMethod.bind(this, node, parent));
- if (shouldPrintParens) {
- this._printTrailingComments(node, parent);
- this.tokenChar(41);
- this._noLineTerminator = noLineTerminatorAfter;
- } else if (noLineTerminatorAfter && !this._noLineTerminator) {
- this._noLineTerminator = true;
- this._printTrailingComments(node, parent);
- } else {
- this._printTrailingComments(node, parent, trailingCommentsLineOffset);
- }
- this._printStack.pop();
- format.concise = oldConcise;
- this._insideAux = oldInAux;
- this._endsWithInnerRaw = false;
- }
- _maybeAddAuxComment(enteredPositionlessNode) {
- if (enteredPositionlessNode) this._printAuxBeforeComment();
- if (!this._insideAux) this._printAuxAfterComment();
- }
- _printAuxBeforeComment() {
- if (this._printAuxAfterOnNextUserNode) return;
- this._printAuxAfterOnNextUserNode = true;
- const comment = this.format.auxiliaryCommentBefore;
- if (comment) {
- this._printComment({
- type: "CommentBlock",
- value: comment
- }, 0);
- }
- }
- _printAuxAfterComment() {
- if (!this._printAuxAfterOnNextUserNode) return;
- this._printAuxAfterOnNextUserNode = false;
- const comment = this.format.auxiliaryCommentAfter;
- if (comment) {
- this._printComment({
- type: "CommentBlock",
- value: comment
- }, 0);
- }
- }
- getPossibleRaw(node) {
- const extra = node.extra;
- if (extra && extra.raw != null && extra.rawValue != null && node.value === extra.rawValue) {
- return extra.raw;
- }
- }
- printJoin(nodes, parent, opts = {}) {
- if (!(nodes != null && nodes.length)) return;
- if (opts.indent) this.indent();
- const newlineOpts = {
- addNewlines: opts.addNewlines,
- nextNodeStartLine: 0
- };
- const separator = opts.separator ? opts.separator.bind(this) : null;
- const len = nodes.length;
- for (let i = 0; i < len; i++) {
- const node = nodes[i];
- if (!node) continue;
- if (opts.statement) this._printNewline(i === 0, newlineOpts);
- this.print(node, parent, undefined, opts.trailingCommentsLineOffset || 0);
- opts.iterator == null ? void 0 : opts.iterator(node, i);
- if (i < len - 1) separator == null ? void 0 : separator();
- if (opts.statement) {
- if (i + 1 === len) {
- this.newline(1);
- } else {
- var _nextNode$loc;
- const nextNode = nodes[i + 1];
- newlineOpts.nextNodeStartLine = ((_nextNode$loc = nextNode.loc) == null ? void 0 : _nextNode$loc.start.line) || 0;
- this._printNewline(true, newlineOpts);
- }
- }
- }
- if (opts.indent) this.dedent();
- }
- printAndIndentOnComments(node, parent) {
- const indent = node.leadingComments && node.leadingComments.length > 0;
- if (indent) this.indent();
- this.print(node, parent);
- if (indent) this.dedent();
- }
- printBlock(parent) {
- const node = parent.body;
- if (node.type !== "EmptyStatement") {
- this.space();
- }
- this.print(node, parent);
- }
- _printTrailingComments(node, parent, lineOffset) {
- const {
- innerComments,
- trailingComments
- } = node;
- if (innerComments != null && innerComments.length) {
- this._printComments(2, innerComments, node, parent, lineOffset);
- }
- if (trailingComments != null && trailingComments.length) {
- this._printComments(2, trailingComments, node, parent, lineOffset);
- }
- }
- _printLeadingComments(node, parent) {
- const comments = node.leadingComments;
- if (!(comments != null && comments.length)) return;
- this._printComments(0, comments, node, parent);
- }
- _maybePrintInnerComments() {
- if (this._endsWithInnerRaw) this.printInnerComments();
- this._endsWithInnerRaw = true;
- this._indentInnerComments = true;
- }
- printInnerComments() {
- const node = this._printStack[this._printStack.length - 1];
- const comments = node.innerComments;
- if (!(comments != null && comments.length)) return;
- const hasSpace = this.endsWith(32);
- const indent = this._indentInnerComments;
- const printedCommentsCount = this._printedComments.size;
- if (indent) this.indent();
- this._printComments(1, comments, node);
- if (hasSpace && printedCommentsCount !== this._printedComments.size) {
- this.space();
- }
- if (indent) this.dedent();
- }
- noIndentInnerCommentsHere() {
- this._indentInnerComments = false;
- }
- printSequence(nodes, parent, opts = {}) {
- opts.statement = true;
- return this.printJoin(nodes, parent, opts);
- }
- printList(items, parent, opts = {}) {
- if (opts.separator == null) {
- opts.separator = commaSeparator;
- }
- return this.printJoin(items, parent, opts);
- }
- _printNewline(newLine, opts) {
- if (this.format.retainLines || this.format.compact) return;
- if (this.format.concise) {
- this.space();
- return;
- }
- if (!newLine) {
- return;
- }
- const startLine = opts.nextNodeStartLine;
- const lastCommentLine = this._lastCommentLine;
- if (startLine > 0 && lastCommentLine > 0) {
- const offset = startLine - lastCommentLine;
- if (offset >= 0) {
- this.newline(offset || 1);
- return;
- }
- }
- if (this._buf.hasContent()) {
- this.newline(1);
- }
- }
- _shouldPrintComment(comment) {
- if (comment.ignore) return 0;
- if (this._printedComments.has(comment)) return 0;
- if (this._noLineTerminator && (HAS_NEWLINE.test(comment.value) || HAS_BlOCK_COMMENT_END.test(comment.value))) {
- return 2;
- }
- this._printedComments.add(comment);
- if (!this.format.shouldPrintComment(comment.value)) {
- return 0;
- }
- return 1;
- }
- _printComment(comment, skipNewLines) {
- const noLineTerminator = this._noLineTerminator;
- const isBlockComment = comment.type === "CommentBlock";
- const printNewLines = isBlockComment && skipNewLines !== 1 && !this._noLineTerminator;
- if (printNewLines && this._buf.hasContent() && skipNewLines !== 2) {
- this.newline(1);
- }
- const lastCharCode = this.getLastChar();
- if (lastCharCode !== 91 && lastCharCode !== 123) {
- this.space();
- }
- let val;
- if (isBlockComment) {
- val = `/*${comment.value}*/`;
- if (this.format.indent.adjustMultilineComment) {
- var _comment$loc;
- const offset = (_comment$loc = comment.loc) == null ? void 0 : _comment$loc.start.column;
- if (offset) {
- const newlineRegex = new RegExp("\\n\\s{1," + offset + "}", "g");
- val = val.replace(newlineRegex, "\n");
- }
- let indentSize = this.format.retainLines ? 0 : this._buf.getCurrentColumn();
- if (this._shouldIndent(47) || this.format.retainLines) {
- indentSize += this._getIndent();
- }
- val = val.replace(/\n(?!$)/g, `\n${" ".repeat(indentSize)}`);
- }
- } else if (!noLineTerminator) {
- val = `//${comment.value}`;
- } else {
- val = `/*${comment.value}*/`;
- }
- if (this.endsWith(47)) this._space();
- this.source("start", comment.loc);
- this._append(val, isBlockComment);
- if (!isBlockComment && !noLineTerminator) {
- this.newline(1, true);
- }
- if (printNewLines && skipNewLines !== 3) {
- this.newline(1);
- }
- }
- _printComments(type, comments, node, parent, lineOffset = 0) {
- const nodeLoc = node.loc;
- const len = comments.length;
- let hasLoc = !!nodeLoc;
- const nodeStartLine = hasLoc ? nodeLoc.start.line : 0;
- const nodeEndLine = hasLoc ? nodeLoc.end.line : 0;
- let lastLine = 0;
- let leadingCommentNewline = 0;
- const maybeNewline = this._noLineTerminator ? function () {} : this.newline.bind(this);
- for (let i = 0; i < len; i++) {
- const comment = comments[i];
- const shouldPrint = this._shouldPrintComment(comment);
- if (shouldPrint === 2) {
- hasLoc = false;
- break;
- }
- if (hasLoc && comment.loc && shouldPrint === 1) {
- const commentStartLine = comment.loc.start.line;
- const commentEndLine = comment.loc.end.line;
- if (type === 0) {
- let offset = 0;
- if (i === 0) {
- if (this._buf.hasContent() && (comment.type === "CommentLine" || commentStartLine != commentEndLine)) {
- offset = leadingCommentNewline = 1;
- }
- } else {
- offset = commentStartLine - lastLine;
- }
- lastLine = commentEndLine;
- maybeNewline(offset);
- this._printComment(comment, 1);
- if (i + 1 === len) {
- maybeNewline(Math.max(nodeStartLine - lastLine, leadingCommentNewline));
- lastLine = nodeStartLine;
- }
- } else if (type === 1) {
- const offset = commentStartLine - (i === 0 ? nodeStartLine : lastLine);
- lastLine = commentEndLine;
- maybeNewline(offset);
- this._printComment(comment, 1);
- if (i + 1 === len) {
- maybeNewline(Math.min(1, nodeEndLine - lastLine));
- lastLine = nodeEndLine;
- }
- } else {
- const offset = commentStartLine - (i === 0 ? nodeEndLine - lineOffset : lastLine);
- lastLine = commentEndLine;
- maybeNewline(offset);
- this._printComment(comment, 1);
- }
- } else {
- hasLoc = false;
- if (shouldPrint !== 1) {
- continue;
- }
- if (len === 1) {
- const singleLine = comment.loc ? comment.loc.start.line === comment.loc.end.line : !HAS_NEWLINE.test(comment.value);
- const shouldSkipNewline = singleLine && !isStatement(node) && !isClassBody(parent) && !isTSInterfaceBody(parent) && !isTSEnumDeclaration(parent);
- if (type === 0) {
- this._printComment(comment, shouldSkipNewline && node.type !== "ObjectExpression" || singleLine && isFunction(parent, {
- body: node
- }) ? 1 : 0);
- } else if (shouldSkipNewline && type === 2) {
- this._printComment(comment, 1);
- } else {
- this._printComment(comment, 0);
- }
- } else if (type === 1 && !(node.type === "ObjectExpression" && node.properties.length > 1) && node.type !== "ClassBody" && node.type !== "TSInterfaceBody") {
- this._printComment(comment, i === 0 ? 2 : i === len - 1 ? 3 : 0);
- } else {
- this._printComment(comment, 0);
- }
- }
- }
- if (type === 2 && hasLoc && lastLine) {
- this._lastCommentLine = lastLine;
- }
- }
- }
- Object.assign(Printer.prototype, generatorFunctions);
- {
- Printer.prototype.Noop = function Noop() {};
- }
- var _default = Printer;
- exports.default = _default;
- function commaSeparator() {
- this.tokenChar(44);
- this.space();
- }
- //# sourceMappingURL=printer.js.map
|