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.
540 lines
12 KiB
540 lines
12 KiB
"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 { |
|
isProgram, |
|
isFile, |
|
isEmptyStatement |
|
} = _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 { |
|
needsParens, |
|
needsWhitespaceAfter, |
|
needsWhitespaceBefore |
|
} = n; |
|
|
|
class Printer { |
|
constructor(format, map) { |
|
this.inForStatementInitCounter = 0; |
|
this._printStack = []; |
|
this._indent = 0; |
|
this._insideAux = false; |
|
this._parenPushNewlineState = null; |
|
this._noLineTerminator = false; |
|
this._printAuxAfterOnNextUserNode = false; |
|
this._printedComments = new WeakSet(); |
|
this._endsWithInteger = false; |
|
this._endsWithWord = false; |
|
this.format = format; |
|
this._buf = new _buffer.default(map); |
|
} |
|
|
|
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(); |
|
|
|
this._append(";", !force); |
|
} |
|
|
|
rightBrace() { |
|
if (this.format.minified) { |
|
this._buf.removeLastSemicolon(); |
|
} |
|
|
|
this.token("}"); |
|
} |
|
|
|
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) { |
|
if (this._endsWithWord || this.endsWith(47) && str.charCodeAt(0) === 47) { |
|
this._space(); |
|
} |
|
|
|
this._maybeAddAuxComment(); |
|
|
|
this._append(str); |
|
|
|
this._endsWithWord = true; |
|
} |
|
|
|
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) { |
|
const lastChar = this.getLastChar(); |
|
const strFirst = str.charCodeAt(0); |
|
|
|
if (str === "--" && lastChar === 33 || strFirst === 43 && lastChar === 43 || strFirst === 45 && lastChar === 45 || strFirst === 46 && this._endsWithInteger) { |
|
this._space(); |
|
} |
|
|
|
this._maybeAddAuxComment(); |
|
|
|
this._append(str); |
|
} |
|
|
|
newline(i = 1) { |
|
if (this.format.retainLines || this.format.compact) return; |
|
|
|
if (this.format.concise) { |
|
this.space(); |
|
return; |
|
} |
|
|
|
const charBeforeNewline = this.endsWithCharAndNewline(); |
|
if (charBeforeNewline === 10) return; |
|
|
|
if (charBeforeNewline === 123 || charBeforeNewline === 58) { |
|
i--; |
|
} |
|
|
|
if (i <= 0) return; |
|
|
|
for (let j = 0; j < i; j++) { |
|
this._newline(); |
|
} |
|
} |
|
|
|
endsWith(char) { |
|
return this.getLastChar() === char; |
|
} |
|
|
|
getLastChar() { |
|
return this._buf.getLastChar(); |
|
} |
|
|
|
endsWithCharAndNewline() { |
|
return this._buf.endsWithCharAndNewline(); |
|
} |
|
|
|
removeTrailingNewline() { |
|
this._buf.removeTrailingNewline(); |
|
} |
|
|
|
exactSource(loc, cb) { |
|
this._catchUp("start", loc); |
|
|
|
this._buf.exactSource(loc, cb); |
|
} |
|
|
|
source(prop, loc) { |
|
this._catchUp(prop, loc); |
|
|
|
this._buf.source(prop, loc); |
|
} |
|
|
|
withSource(prop, loc, cb) { |
|
this._catchUp(prop, loc); |
|
|
|
this._buf.withSource(prop, loc, cb); |
|
} |
|
|
|
_space() { |
|
this._append(" ", true); |
|
} |
|
|
|
_newline() { |
|
this._append("\n", true); |
|
} |
|
|
|
_append(str, queue = false) { |
|
this._maybeAddParen(str); |
|
|
|
this._maybeIndent(str); |
|
|
|
if (queue) this._buf.queue(str);else this._buf.append(str); |
|
this._endsWithWord = false; |
|
this._endsWithInteger = false; |
|
} |
|
|
|
_maybeIndent(str) { |
|
if (this._indent && this.endsWith(10) && str.charCodeAt(0) !== 10) { |
|
this._buf.queue(this._getIndent()); |
|
} |
|
} |
|
|
|
_maybeAddParen(str) { |
|
const parenPushNewlineState = this._parenPushNewlineState; |
|
if (!parenPushNewlineState) return; |
|
let i; |
|
|
|
for (i = 0; i < str.length && str[i] === " "; i++) continue; |
|
|
|
if (i === str.length) { |
|
return; |
|
} |
|
|
|
const cha = str[i]; |
|
|
|
if (cha !== "\n") { |
|
if (cha !== "/" || i + 1 === str.length) { |
|
this._parenPushNewlineState = null; |
|
return; |
|
} |
|
|
|
const chaPost = str[i + 1]; |
|
|
|
if (chaPost === "*") { |
|
if (PURE_ANNOTATION_RE.test(str.slice(i + 2, str.length - 2))) { |
|
return; |
|
} |
|
} else if (chaPost !== "/") { |
|
this._parenPushNewlineState = null; |
|
return; |
|
} |
|
} |
|
|
|
this.token("("); |
|
this.indent(); |
|
parenPushNewlineState.printed = true; |
|
} |
|
|
|
_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.format.indent.style.repeat(this._indent); |
|
} |
|
|
|
startTerminatorless(isLabel = false) { |
|
if (isLabel) { |
|
this._noLineTerminator = true; |
|
return null; |
|
} else { |
|
return this._parenPushNewlineState = { |
|
printed: false |
|
}; |
|
} |
|
} |
|
|
|
endTerminatorless(state) { |
|
this._noLineTerminator = false; |
|
|
|
if (state != null && state.printed) { |
|
this.dedent(); |
|
this.newline(); |
|
this.token(")"); |
|
} |
|
} |
|
|
|
print(node, parent) { |
|
if (!node) return; |
|
const oldConcise = this.format.concise; |
|
|
|
if (node._compact) { |
|
this.format.concise = true; |
|
} |
|
|
|
const printMethod = this[node.type]; |
|
|
|
if (!printMethod) { |
|
throw new ReferenceError(`unknown node of type ${JSON.stringify(node.type)} with constructor ${JSON.stringify(node == null ? void 0 : node.constructor.name)}`); |
|
} |
|
|
|
this._printStack.push(node); |
|
|
|
const oldInAux = this._insideAux; |
|
this._insideAux = !node.loc; |
|
|
|
this._maybeAddAuxComment(this._insideAux && !oldInAux); |
|
|
|
let shouldPrintParens = needsParens(node, parent, this._printStack); |
|
|
|
if (this.format.retainFunctionParens && node.type === "FunctionExpression" && node.extra && node.extra.parenthesized) { |
|
shouldPrintParens = true; |
|
} |
|
|
|
if (shouldPrintParens) this.token("("); |
|
|
|
this._printLeadingComments(node); |
|
|
|
const loc = isProgram(node) || isFile(node) ? null : node.loc; |
|
this.withSource("start", loc, () => { |
|
printMethod.call(this, node, parent); |
|
}); |
|
|
|
this._printTrailingComments(node); |
|
|
|
if (shouldPrintParens) this.token(")"); |
|
|
|
this._printStack.pop(); |
|
|
|
this.format.concise = oldConcise; |
|
this._insideAux = oldInAux; |
|
} |
|
|
|
_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 |
|
}); |
|
} |
|
} |
|
|
|
_printAuxAfterComment() { |
|
if (!this._printAuxAfterOnNextUserNode) return; |
|
this._printAuxAfterOnNextUserNode = false; |
|
const comment = this.format.auxiliaryCommentAfter; |
|
|
|
if (comment) { |
|
this._printComment({ |
|
type: "CommentBlock", |
|
value: comment |
|
}); |
|
} |
|
} |
|
|
|
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 |
|
}; |
|
|
|
for (let i = 0; i < nodes.length; i++) { |
|
const node = nodes[i]; |
|
if (!node) continue; |
|
if (opts.statement) this._printNewline(true, node, parent, newlineOpts); |
|
this.print(node, parent); |
|
|
|
if (opts.iterator) { |
|
opts.iterator(node, i); |
|
} |
|
|
|
if (opts.separator && i < nodes.length - 1) { |
|
opts.separator.call(this); |
|
} |
|
|
|
if (opts.statement) this._printNewline(false, node, parent, 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 (!isEmptyStatement(node)) { |
|
this.space(); |
|
} |
|
|
|
this.print(node, parent); |
|
} |
|
|
|
_printTrailingComments(node) { |
|
this._printComments(this._getComments(false, node)); |
|
} |
|
|
|
_printLeadingComments(node) { |
|
this._printComments(this._getComments(true, node), true); |
|
} |
|
|
|
printInnerComments(node, indent = true) { |
|
var _node$innerComments; |
|
|
|
if (!((_node$innerComments = node.innerComments) != null && _node$innerComments.length)) return; |
|
if (indent) this.indent(); |
|
|
|
this._printComments(node.innerComments); |
|
|
|
if (indent) this.dedent(); |
|
} |
|
|
|
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(leading, node, parent, opts) { |
|
if (this.format.retainLines || this.format.compact) return; |
|
|
|
if (this.format.concise) { |
|
this.space(); |
|
return; |
|
} |
|
|
|
let lines = 0; |
|
|
|
if (this._buf.hasContent()) { |
|
if (!leading) lines++; |
|
if (opts.addNewlines) lines += opts.addNewlines(leading, node) || 0; |
|
const needs = leading ? needsWhitespaceBefore : needsWhitespaceAfter; |
|
if (needs(node, parent)) lines++; |
|
} |
|
|
|
this.newline(Math.min(2, lines)); |
|
} |
|
|
|
_getComments(leading, node) { |
|
return node && (leading ? node.leadingComments : node.trailingComments) || []; |
|
} |
|
|
|
_printComment(comment, skipNewLines) { |
|
if (!this.format.shouldPrintComment(comment.value)) return; |
|
if (comment.ignore) return; |
|
if (this._printedComments.has(comment)) return; |
|
|
|
this._printedComments.add(comment); |
|
|
|
const isBlockComment = comment.type === "CommentBlock"; |
|
const printNewLines = isBlockComment && !skipNewLines && !this._noLineTerminator; |
|
if (printNewLines && this._buf.hasContent()) this.newline(1); |
|
const lastCharCode = this.getLastChar(); |
|
|
|
if (lastCharCode !== 91 && lastCharCode !== 123) { |
|
this.space(); |
|
} |
|
|
|
let val = !isBlockComment && !this._noLineTerminator ? `//${comment.value}\n` : `/*${comment.value}*/`; |
|
|
|
if (isBlockComment && 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"); |
|
} |
|
|
|
const indentSize = Math.max(this._getIndent().length, this.format.retainLines ? 0 : this._buf.getCurrentColumn()); |
|
val = val.replace(/\n(?!$)/g, `\n${" ".repeat(indentSize)}`); |
|
} |
|
|
|
if (this.endsWith(47)) this._space(); |
|
this.withSource("start", comment.loc, () => { |
|
this._append(val); |
|
}); |
|
if (printNewLines) this.newline(1); |
|
} |
|
|
|
_printComments(comments, inlinePureAnnotation) { |
|
if (!(comments != null && comments.length)) return; |
|
|
|
if (inlinePureAnnotation && comments.length === 1 && PURE_ANNOTATION_RE.test(comments[0].value)) { |
|
this._printComment(comments[0], this._buf.hasContent() && !this.endsWith(10)); |
|
} else { |
|
for (const comment of comments) { |
|
this._printComment(comment); |
|
} |
|
} |
|
} |
|
|
|
printAssertions(node) { |
|
var _node$assertions; |
|
|
|
if ((_node$assertions = node.assertions) != null && _node$assertions.length) { |
|
this.space(); |
|
this.word("assert"); |
|
this.space(); |
|
this.token("{"); |
|
this.space(); |
|
this.printList(node.assertions, node); |
|
this.space(); |
|
this.token("}"); |
|
} |
|
} |
|
|
|
} |
|
|
|
Object.assign(Printer.prototype, generatorFunctions); |
|
{ |
|
Printer.prototype.Noop = function Noop() {}; |
|
} |
|
var _default = Printer; |
|
exports.default = _default; |
|
|
|
function commaSeparator() { |
|
this.token(","); |
|
this.space(); |
|
} |