72 lines
1.6 KiB
JavaScript
72 lines
1.6 KiB
JavaScript
"use strict";
|
|
const babylon = require("babylon");
|
|
const Printer = require("./src/printer").Printer;
|
|
const flowParser = require("flow-parser");
|
|
const comments = require("./src/comments");
|
|
|
|
var babylonOptions = {
|
|
sourceType: "module",
|
|
allowImportExportEverywhere: false,
|
|
allowReturnOutsideFunction: false,
|
|
plugins: [
|
|
"jsx",
|
|
"flow",
|
|
"doExpressions",
|
|
"objectRestSpread",
|
|
"decorators",
|
|
"classProperties",
|
|
"exportExtensions",
|
|
"asyncGenerators",
|
|
"functionBind",
|
|
"functionSent",
|
|
"dynamicImport"
|
|
]
|
|
};
|
|
|
|
function format(text, opts) {
|
|
opts = opts || {};
|
|
let ast;
|
|
|
|
if (opts.useFlowParser) {
|
|
ast = flowParser.parse(text);
|
|
if (ast.errors.length > 0) {
|
|
let msg = ast.errors[(0)].message + " on line " +
|
|
ast.errors[(0)].loc.start.line;
|
|
if (opts.filename) {
|
|
msg += " in file " + opts.filename;
|
|
}
|
|
throw new Error(msg);
|
|
}
|
|
} else {
|
|
ast = babylon.parse(text, babylonOptions);
|
|
}
|
|
|
|
// Interleave comment nodes
|
|
if (ast.comments) {
|
|
comments.attach(ast.comments, ast, text);
|
|
ast.comments = [];
|
|
}
|
|
ast.tokens = [];
|
|
opts.originalText = text;
|
|
|
|
const printer = new Printer(opts);
|
|
return printer.printGenerically(ast).code;
|
|
}
|
|
|
|
function formatWithShebang(text, opts) {
|
|
if (!text.startsWith("#!")) {
|
|
return format(text, opts);
|
|
}
|
|
|
|
const index = text.indexOf("\n");
|
|
const shebang = text.slice(0, index + 1);
|
|
const programText = text.slice(index + 1);
|
|
return shebang + format(programText, opts);
|
|
}
|
|
|
|
module.exports = {
|
|
format: function(text, opts) {
|
|
return formatWithShebang(text, opts);
|
|
}
|
|
};
|