186 lines
9.4 KiB
JavaScript
186 lines
9.4 KiB
JavaScript
|
"use strict";
|
||
|
|
var __extends = (this && this.__extends) || (function () {
|
||
|
|
var extendStatics = function (d, b) {
|
||
|
|
extendStatics = Object.setPrototypeOf ||
|
||
|
|
({ __proto__: [] } instanceof Array && function (d, b) { d.__proto__ = b; }) ||
|
||
|
|
function (d, b) { for (var p in b) if (Object.prototype.hasOwnProperty.call(b, p)) d[p] = b[p]; };
|
||
|
|
return extendStatics(d, b);
|
||
|
|
};
|
||
|
|
return function (d, b) {
|
||
|
|
if (typeof b !== "function" && b !== null)
|
||
|
|
throw new TypeError("Class extends value " + String(b) + " is not a constructor or null");
|
||
|
|
extendStatics(d, b);
|
||
|
|
function __() { this.constructor = d; }
|
||
|
|
d.prototype = b === null ? Object.create(b) : (__.prototype = b.prototype, new __());
|
||
|
|
};
|
||
|
|
})();
|
||
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
||
|
|
exports.PackagesAndModulesPrettierVisitor = void 0;
|
||
|
|
var prettier_builder_1 = require("./prettier-builder");
|
||
|
|
var format_comments_1 = require("./comments/format-comments");
|
||
|
|
var printer_utils_1 = require("./printer-utils");
|
||
|
|
var doc_1 = require("prettier/doc");
|
||
|
|
var base_cst_printer_1 = require("../base-cst-printer");
|
||
|
|
var utils_1 = require("../types/utils");
|
||
|
|
var line = doc_1.builders.line, hardline = doc_1.builders.hardline, indent = doc_1.builders.indent, group = doc_1.builders.group;
|
||
|
|
var PackagesAndModulesPrettierVisitor = /** @class */ (function (_super) {
|
||
|
|
__extends(PackagesAndModulesPrettierVisitor, _super);
|
||
|
|
function PackagesAndModulesPrettierVisitor() {
|
||
|
|
return _super !== null && _super.apply(this, arguments) || this;
|
||
|
|
}
|
||
|
|
PackagesAndModulesPrettierVisitor.prototype.compilationUnit = function (ctx) {
|
||
|
|
var compilationUnit = utils_1.isOrdinaryCompilationUnitCtx(ctx)
|
||
|
|
? ctx.ordinaryCompilationUnit
|
||
|
|
: ctx.modularCompilationUnit;
|
||
|
|
return prettier_builder_1.concat([this.visit(compilationUnit[0]), line]);
|
||
|
|
};
|
||
|
|
PackagesAndModulesPrettierVisitor.prototype.ordinaryCompilationUnit = function (ctx) {
|
||
|
|
var packageDecl = this.visit(ctx.packageDeclaration);
|
||
|
|
var sortedImportsDecl = printer_utils_1.sortImports(ctx.importDeclaration);
|
||
|
|
var nonStaticImports = this.mapVisit(sortedImportsDecl.nonStaticImports);
|
||
|
|
var staticImports = this.mapVisit(sortedImportsDecl.staticImports);
|
||
|
|
var typesDecl = this.mapVisit(ctx.typeDeclaration);
|
||
|
|
// TODO: utility to add item+line (or multiple lines) but only if an item exists
|
||
|
|
return printer_utils_1.rejectAndConcat([
|
||
|
|
printer_utils_1.rejectAndJoin(prettier_builder_1.concat([hardline, hardline]), [
|
||
|
|
packageDecl,
|
||
|
|
printer_utils_1.rejectAndJoin(hardline, staticImports),
|
||
|
|
printer_utils_1.rejectAndJoin(hardline, nonStaticImports),
|
||
|
|
printer_utils_1.rejectAndJoin(prettier_builder_1.concat([hardline, hardline]), typesDecl)
|
||
|
|
])
|
||
|
|
]);
|
||
|
|
};
|
||
|
|
PackagesAndModulesPrettierVisitor.prototype.modularCompilationUnit = function (ctx) {
|
||
|
|
var sortedImportsDecl = printer_utils_1.sortImports(ctx.importDeclaration);
|
||
|
|
var nonStaticImports = this.mapVisit(sortedImportsDecl.nonStaticImports);
|
||
|
|
var staticImports = this.mapVisit(sortedImportsDecl.staticImports);
|
||
|
|
var moduleDeclaration = this.visit(ctx.moduleDeclaration);
|
||
|
|
return printer_utils_1.rejectAndConcat([
|
||
|
|
printer_utils_1.rejectAndJoin(prettier_builder_1.concat([hardline, hardline]), [
|
||
|
|
printer_utils_1.rejectAndJoin(hardline, staticImports),
|
||
|
|
printer_utils_1.rejectAndJoin(hardline, nonStaticImports),
|
||
|
|
moduleDeclaration
|
||
|
|
])
|
||
|
|
]);
|
||
|
|
};
|
||
|
|
PackagesAndModulesPrettierVisitor.prototype.packageDeclaration = function (ctx) {
|
||
|
|
var modifiers = this.mapVisit(ctx.packageModifier);
|
||
|
|
var name = printer_utils_1.buildFqn(ctx.Identifier, ctx.Dot);
|
||
|
|
return printer_utils_1.rejectAndJoin(hardline, [
|
||
|
|
printer_utils_1.rejectAndJoin(hardline, modifiers),
|
||
|
|
prettier_builder_1.concat([ctx.Package[0], " ", name, ctx.Semicolon[0]])
|
||
|
|
]);
|
||
|
|
};
|
||
|
|
PackagesAndModulesPrettierVisitor.prototype.packageModifier = function (ctx) {
|
||
|
|
return this.visitSingle(ctx);
|
||
|
|
};
|
||
|
|
PackagesAndModulesPrettierVisitor.prototype.importDeclaration = function (ctx) {
|
||
|
|
if (ctx.emptyStatement !== undefined) {
|
||
|
|
return this.visit(ctx.emptyStatement);
|
||
|
|
}
|
||
|
|
var optionalStatic = ctx.Static ? ctx.Static[0] : "";
|
||
|
|
var packageOrTypeName = this.visit(ctx.packageOrTypeName);
|
||
|
|
var optionalDotStar = ctx.Dot ? prettier_builder_1.concat([ctx.Dot[0], ctx.Star[0]]) : "";
|
||
|
|
return printer_utils_1.rejectAndJoin(" ", [
|
||
|
|
ctx.Import[0],
|
||
|
|
optionalStatic,
|
||
|
|
printer_utils_1.rejectAndConcat([packageOrTypeName, optionalDotStar, ctx.Semicolon[0]])
|
||
|
|
]);
|
||
|
|
};
|
||
|
|
PackagesAndModulesPrettierVisitor.prototype.typeDeclaration = function (ctx) {
|
||
|
|
if (ctx.Semicolon) {
|
||
|
|
return printer_utils_1.displaySemicolon(ctx.Semicolon[0]);
|
||
|
|
}
|
||
|
|
return this.visitSingle(ctx);
|
||
|
|
};
|
||
|
|
PackagesAndModulesPrettierVisitor.prototype.moduleDeclaration = function (ctx) {
|
||
|
|
var annotations = this.mapVisit(ctx.annotation);
|
||
|
|
var optionalOpen = ctx.Open ? ctx.Open[0] : "";
|
||
|
|
var name = printer_utils_1.buildFqn(ctx.Identifier, ctx.Dot);
|
||
|
|
var moduleDirectives = this.mapVisit(ctx.moduleDirective);
|
||
|
|
var content = printer_utils_1.rejectAndJoinSeps(printer_utils_1.getBlankLinesSeparator(ctx.moduleDirective), moduleDirectives);
|
||
|
|
return printer_utils_1.rejectAndJoin(" ", [
|
||
|
|
prettier_builder_1.join(" ", annotations),
|
||
|
|
optionalOpen,
|
||
|
|
ctx.Module[0],
|
||
|
|
name,
|
||
|
|
printer_utils_1.putIntoBraces(content, hardline, ctx.LCurly[0], ctx.RCurly[0])
|
||
|
|
]);
|
||
|
|
};
|
||
|
|
PackagesAndModulesPrettierVisitor.prototype.moduleDirective = function (ctx) {
|
||
|
|
return this.visitSingle(ctx);
|
||
|
|
};
|
||
|
|
PackagesAndModulesPrettierVisitor.prototype.requiresModuleDirective = function (ctx) {
|
||
|
|
var modifiers = this.mapVisit(ctx.requiresModifier);
|
||
|
|
var moduleName = this.visit(ctx.moduleName);
|
||
|
|
return printer_utils_1.rejectAndJoin(" ", [
|
||
|
|
ctx.Requires[0],
|
||
|
|
prettier_builder_1.join(" ", modifiers),
|
||
|
|
prettier_builder_1.concat([moduleName, ctx.Semicolon[0]])
|
||
|
|
]);
|
||
|
|
};
|
||
|
|
PackagesAndModulesPrettierVisitor.prototype.exportsModuleDirective = function (ctx) {
|
||
|
|
var packageName = this.visit(ctx.packageName);
|
||
|
|
var to = ctx.To ? ctx.To[0] : "";
|
||
|
|
var moduleNames = this.mapVisit(ctx.moduleName);
|
||
|
|
var commas = ctx.Comma ? ctx.Comma.map(function (elt) { return prettier_builder_1.concat([elt, line]); }) : [];
|
||
|
|
return group(printer_utils_1.rejectAndConcat([
|
||
|
|
indent(printer_utils_1.rejectAndJoin(line, [
|
||
|
|
printer_utils_1.rejectAndJoin(" ", [ctx.Exports[0], packageName]),
|
||
|
|
group(indent(printer_utils_1.rejectAndJoin(line, [
|
||
|
|
to,
|
||
|
|
printer_utils_1.rejectAndJoinSeps(commas, moduleNames)
|
||
|
|
])))
|
||
|
|
])),
|
||
|
|
ctx.Semicolon[0]
|
||
|
|
]));
|
||
|
|
};
|
||
|
|
PackagesAndModulesPrettierVisitor.prototype.opensModuleDirective = function (ctx) {
|
||
|
|
var packageName = this.visit(ctx.packageName);
|
||
|
|
var to = ctx.To ? ctx.To[0] : "";
|
||
|
|
var moduleNames = this.mapVisit(ctx.moduleName);
|
||
|
|
var commas = ctx.Comma ? ctx.Comma.map(function (elt) { return prettier_builder_1.concat([elt, line]); }) : [];
|
||
|
|
return group(printer_utils_1.rejectAndConcat([
|
||
|
|
indent(printer_utils_1.rejectAndJoin(line, [
|
||
|
|
printer_utils_1.rejectAndJoin(" ", [ctx.Opens[0], packageName]),
|
||
|
|
group(indent(printer_utils_1.rejectAndJoin(line, [
|
||
|
|
to,
|
||
|
|
printer_utils_1.rejectAndJoinSeps(commas, moduleNames)
|
||
|
|
])))
|
||
|
|
])),
|
||
|
|
ctx.Semicolon[0]
|
||
|
|
]));
|
||
|
|
};
|
||
|
|
PackagesAndModulesPrettierVisitor.prototype.usesModuleDirective = function (ctx) {
|
||
|
|
var typeName = this.visit(ctx.typeName);
|
||
|
|
return printer_utils_1.rejectAndConcat([
|
||
|
|
prettier_builder_1.concat([ctx.Uses[0], " "]),
|
||
|
|
typeName,
|
||
|
|
ctx.Semicolon[0]
|
||
|
|
]);
|
||
|
|
};
|
||
|
|
PackagesAndModulesPrettierVisitor.prototype.providesModuleDirective = function (ctx) {
|
||
|
|
var firstTypeName = this.visit(ctx.typeName[0]);
|
||
|
|
var otherTypeNames = this.mapVisit(ctx.typeName.slice(1));
|
||
|
|
var commas = ctx.Comma ? ctx.Comma.map(function (elt) { return prettier_builder_1.concat([elt, line]); }) : [];
|
||
|
|
return group(printer_utils_1.rejectAndConcat([
|
||
|
|
indent(printer_utils_1.rejectAndJoin(line, [
|
||
|
|
printer_utils_1.rejectAndJoin(" ", [ctx.Provides[0], firstTypeName]),
|
||
|
|
group(indent(printer_utils_1.rejectAndJoin(line, [
|
||
|
|
ctx.With[0],
|
||
|
|
printer_utils_1.rejectAndJoinSeps(commas, otherTypeNames)
|
||
|
|
])))
|
||
|
|
])),
|
||
|
|
ctx.Semicolon[0]
|
||
|
|
]));
|
||
|
|
};
|
||
|
|
PackagesAndModulesPrettierVisitor.prototype.requiresModifier = function (ctx) {
|
||
|
|
return format_comments_1.printTokenWithComments(this.getSingle(ctx));
|
||
|
|
};
|
||
|
|
PackagesAndModulesPrettierVisitor.prototype.isModuleCompilationUnit = function () {
|
||
|
|
return "isModuleCompilationUnit";
|
||
|
|
};
|
||
|
|
return PackagesAndModulesPrettierVisitor;
|
||
|
|
}(base_cst_printer_1.BaseCstPrettierPrinter));
|
||
|
|
exports.PackagesAndModulesPrettierVisitor = PackagesAndModulesPrettierVisitor;
|