Reprint pp.js
parent
0d4d21cc68
commit
b5ae965274
154
src/pp.js
154
src/pp.js
|
@ -2,9 +2,7 @@ const assert = require("assert");
|
|||
|
||||
function assertDoc(val) {
|
||||
assert(
|
||||
typeof val === "string" ||
|
||||
(val != null &&
|
||||
typeof val.type === "string"),
|
||||
typeof val === "string" || val != null && typeof val.type === "string",
|
||||
"Value is a valid document"
|
||||
);
|
||||
}
|
||||
|
@ -15,19 +13,23 @@ function fromString(text) {
|
|||
|
||||
function concat(parts) {
|
||||
parts.forEach(assertDoc);
|
||||
return { type: 'concat', parts };
|
||||
|
||||
return {type: "concat", parts};
|
||||
}
|
||||
|
||||
function indent(n, contents) {
|
||||
assertDoc(contents);
|
||||
return { type: 'indent', contents, n };
|
||||
|
||||
return {type: "indent", contents, n};
|
||||
}
|
||||
|
||||
function group(contents, opts) {
|
||||
opts = opts || {};
|
||||
|
||||
assertDoc(contents);
|
||||
|
||||
return {
|
||||
type: 'group',
|
||||
type: "group",
|
||||
contents: contents,
|
||||
break: !!opts.shouldBreak,
|
||||
expandedStates: opts.expandedStates
|
||||
|
@ -41,30 +43,28 @@ function multilineGroup(contents, opts) {
|
|||
}
|
||||
|
||||
function conditionalGroup(states, opts) {
|
||||
return group(states[0], Object.assign(opts || {}, {
|
||||
expandedStates: states
|
||||
}));
|
||||
return group(states[0], Object.assign(opts || {}, {expandedStates: states}));
|
||||
}
|
||||
|
||||
function ifBreak(contents) {
|
||||
return { type: 'if-break', contents };
|
||||
return {type: "if-break", contents};
|
||||
}
|
||||
|
||||
function iterDoc(topDoc, func) {
|
||||
const docs = [ topDoc ];
|
||||
|
||||
while (docs.length !== 0) {
|
||||
const doc = docs.pop();
|
||||
let res = undefined;
|
||||
|
||||
if (typeof doc === "string") {
|
||||
const res = func("string", doc);
|
||||
|
||||
if (res) {
|
||||
return res;
|
||||
}
|
||||
}
|
||||
else {
|
||||
} else {
|
||||
const res = func(doc.type, doc);
|
||||
|
||||
if (res) {
|
||||
return res;
|
||||
}
|
||||
|
@ -73,18 +73,17 @@ function iterDoc(topDoc, func) {
|
|||
for (var i = doc.parts.length - 1; i >= 0; i--) {
|
||||
docs.push(doc.parts[i]);
|
||||
}
|
||||
}
|
||||
else if(doc.type !== "line") {
|
||||
} else if (doc.type !== "line") {
|
||||
docs.push(doc.contents);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
const line = { type: 'line' };
|
||||
const softline = { type: 'line', soft: true };
|
||||
const hardline = { type: 'line', hard: true };
|
||||
const literalline = { type: 'line', hard: true, literal: true };
|
||||
const line = {type: "line"};
|
||||
const softline = {type: "line", soft: true};
|
||||
const hardline = {type: "line", hard: true};
|
||||
const literalline = {type: "line", hard: true, literal: true};
|
||||
|
||||
function isEmpty(n) {
|
||||
return typeof n === "string" && n.length === 0;
|
||||
|
@ -92,12 +91,15 @@ function isEmpty(n) {
|
|||
|
||||
function join(sep, arr) {
|
||||
var res = [];
|
||||
|
||||
for (var i = 0; i < arr.length; i++) {
|
||||
if (i !== 0) {
|
||||
res.push(sep);
|
||||
}
|
||||
|
||||
res.push(arr[i]);
|
||||
}
|
||||
|
||||
return concat(res);
|
||||
}
|
||||
|
||||
|
@ -115,9 +117,11 @@ function hasHardLine(doc) {
|
|||
return !!iterDoc(doc, (type, doc) => {
|
||||
switch (type) {
|
||||
case "line":
|
||||
|
||||
if (doc.hard) {
|
||||
return true;
|
||||
}
|
||||
|
||||
break;
|
||||
}
|
||||
});
|
||||
|
@ -125,9 +129,11 @@ function hasHardLine(doc) {
|
|||
|
||||
function _makeIndent(n) {
|
||||
var s = "";
|
||||
|
||||
for (var i = 0; i < n; i++) {
|
||||
s += " ";
|
||||
}
|
||||
|
||||
return s;
|
||||
}
|
||||
|
||||
|
@ -137,51 +143,62 @@ const MODE_FLAT = 2;
|
|||
function fits(next, restCommands, width) {
|
||||
let restIdx = restCommands.length;
|
||||
const cmds = [ next ];
|
||||
|
||||
while (width >= 0) {
|
||||
if (cmds.length === 0) {
|
||||
if (restIdx === 0) {
|
||||
return true;
|
||||
}
|
||||
else {
|
||||
} else {
|
||||
cmds.push(restCommands[restIdx - 1]);
|
||||
|
||||
restIdx--;
|
||||
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
const [ ind, mode, doc ] = cmds.pop();
|
||||
|
||||
if (typeof doc === "string") {
|
||||
width -= doc.length;
|
||||
}
|
||||
else {
|
||||
} else {
|
||||
switch (doc.type) {
|
||||
case "concat":
|
||||
|
||||
for (var i = doc.parts.length - 1; i >= 0; i--) {
|
||||
cmds.push([ ind, mode, doc.parts[i] ]);
|
||||
}
|
||||
|
||||
break;
|
||||
case "indent":
|
||||
|
||||
cmds.push([ ind + doc.n, mode, doc.contents ]);
|
||||
|
||||
break;
|
||||
case "group":
|
||||
cmds.push([ind, doc.break ? MODE_BREAK : mode, doc.contents]);
|
||||
|
||||
cmds.push([ ind, (doc.break ? MODE_BREAK : mode), doc.contents ]);
|
||||
|
||||
break;
|
||||
case "if-break":
|
||||
|
||||
if (mode === MODE_BREAK) {
|
||||
cmds.push([ ind, mode, doc.contents ]);
|
||||
}
|
||||
|
||||
break;
|
||||
case "line":
|
||||
switch (mode) {
|
||||
// fallthrough
|
||||
case MODE_FLAT:
|
||||
|
||||
if (!doc.hard) {
|
||||
if (!doc.soft) {
|
||||
width -= 1;
|
||||
}
|
||||
|
||||
break;
|
||||
}
|
||||
// fallthrough
|
||||
|
||||
case MODE_BREAK:
|
||||
return true;
|
||||
}
|
||||
|
@ -189,7 +206,6 @@ function fits(next, restCommands, width) {
|
|||
}
|
||||
}
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
|
@ -201,41 +217,52 @@ function print(w, doc) {
|
|||
let cmds = [ [ 0, MODE_BREAK, doc ] ];
|
||||
let out = [];
|
||||
let shouldRemeasure = false;
|
||||
|
||||
while (cmds.length !== 0) {
|
||||
const [ ind, mode, doc ] = cmds.pop();
|
||||
|
||||
if (typeof doc === "string") {
|
||||
out.push(doc);
|
||||
|
||||
pos += doc.length;
|
||||
}
|
||||
else {
|
||||
} else {
|
||||
switch (doc.type) {
|
||||
case "concat":
|
||||
|
||||
for (var i = doc.parts.length - 1; i >= 0; i--) {
|
||||
cmds.push([ ind, mode, doc.parts[i] ]);
|
||||
}
|
||||
|
||||
break;
|
||||
case "indent":
|
||||
|
||||
cmds.push([ ind + doc.n, mode, doc.contents ]);
|
||||
|
||||
break;
|
||||
case "group":
|
||||
switch (mode) {
|
||||
// fallthrough
|
||||
case MODE_FLAT:
|
||||
|
||||
if (!shouldRemeasure) {
|
||||
cmds.push([ind, doc.break ? MODE_BREAK : MODE_FLAT, doc.contents]);
|
||||
cmds.push([
|
||||
ind,
|
||||
(doc.break ? MODE_BREAK : MODE_FLAT),
|
||||
doc.contents
|
||||
]);
|
||||
|
||||
break;
|
||||
}
|
||||
// fallthrough
|
||||
|
||||
case MODE_BREAK:
|
||||
|
||||
shouldRemeasure = false;
|
||||
|
||||
const next = [ ind, MODE_FLAT, doc.contents ];
|
||||
let rem = w - pos;
|
||||
|
||||
if (!doc.break && fits(next, cmds, rem)) {
|
||||
cmds.push(next);
|
||||
}
|
||||
else {
|
||||
} else {
|
||||
// Expanded states are a rare case where a document
|
||||
// can manually provide multiple representations of
|
||||
// itself. It provides an array of documents
|
||||
|
@ -244,51 +271,60 @@ function print(w, doc) {
|
|||
// group has these, we need to manually go through
|
||||
// these states and find the first one that fits.
|
||||
if (doc.expandedStates) {
|
||||
const mostExpanded = doc.expandedStates[doc.expandedStates.length - 1];
|
||||
const mostExpanded = doc.expandedStates[doc.expandedStates.length -
|
||||
1];
|
||||
|
||||
if (doc.break) {
|
||||
cmds.push([ ind, MODE_BREAK, mostExpanded ]);
|
||||
|
||||
break;
|
||||
}
|
||||
else {
|
||||
} else {
|
||||
for (var i = 1; i < doc.expandedStates.length + 1; i++) {
|
||||
if (i >= doc.expandedStates.length) {
|
||||
cmds.push([ ind, MODE_BREAK, mostExpanded ]);
|
||||
|
||||
break;
|
||||
}
|
||||
else {
|
||||
} else {
|
||||
const state = doc.expandedStates[i];
|
||||
const cmd = [ ind, MODE_FLAT, state ];
|
||||
|
||||
if (fits(cmd, cmds, rem)) {
|
||||
cmds.push(cmd);
|
||||
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
else {
|
||||
} else {
|
||||
cmds.push([ ind, MODE_BREAK, doc.contents ]);
|
||||
}
|
||||
}
|
||||
|
||||
break;
|
||||
}
|
||||
break;
|
||||
case "if-break":
|
||||
|
||||
if (mode === MODE_BREAK) {
|
||||
cmds.push([ ind, MODE_BREAK, doc.contents ]);
|
||||
}
|
||||
|
||||
break;
|
||||
case "line":
|
||||
switch (mode) {
|
||||
// fallthrough
|
||||
case MODE_FLAT:
|
||||
|
||||
if (!doc.hard) {
|
||||
if (!doc.soft) {
|
||||
out.push(" ");
|
||||
|
||||
pos += 1;
|
||||
}
|
||||
|
||||
break;
|
||||
}
|
||||
else {
|
||||
} else {
|
||||
// This line was forced into the output even if we
|
||||
// were in flattened mode, so we need to tell the next
|
||||
// group that no matter what, it needs to remeasure
|
||||
|
@ -297,10 +333,12 @@ function print(w, doc) {
|
|||
// for nested groups)
|
||||
shouldRemeasure = true;
|
||||
}
|
||||
// fallthrough
|
||||
|
||||
case MODE_BREAK:
|
||||
|
||||
if (out.length > 0) {
|
||||
const lastString = out[out.length - 1];
|
||||
|
||||
if (lastString.match(/^\s*\n\s*$/)) {
|
||||
out[out.length - 1] = "\n";
|
||||
}
|
||||
|
@ -308,25 +346,39 @@ function print(w, doc) {
|
|||
|
||||
if (doc.literal) {
|
||||
out.push("\n");
|
||||
|
||||
pos = 0;
|
||||
}
|
||||
else {
|
||||
} else {
|
||||
out.push("\n" + _makeIndent(ind));
|
||||
|
||||
pos = ind;
|
||||
}
|
||||
|
||||
break;
|
||||
}
|
||||
break;
|
||||
default:
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
}
|
||||
return out.join("");
|
||||
}
|
||||
|
||||
module.exports = {
|
||||
fromString, concat, isEmpty, join,
|
||||
line, softline, hardline, literalline, group, multilineGroup,
|
||||
conditionalGroup, ifBreak, hasHardLine, indent, print, getFirstString
|
||||
fromString,
|
||||
concat,
|
||||
isEmpty,
|
||||
join,
|
||||
line,
|
||||
softline,
|
||||
hardline,
|
||||
literalline,
|
||||
group,
|
||||
multilineGroup,
|
||||
conditionalGroup,
|
||||
ifBreak,
|
||||
hasHardLine,
|
||||
indent,
|
||||
print,
|
||||
getFirstString
|
||||
};
|
||||
|
|
Loading…
Reference in New Issue