Your IP : 216.73.216.43
"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.MetadataManager = void 0;
const typescript_1 = require("typescript");
class MetadataManager {
constructor(content) {
this.content = content;
}
insert(metadata, symbol, staticOptions) {
const source = (0, typescript_1.createSourceFile)('filename.ts', this.content, typescript_1.ScriptTarget.ES2017);
const decoratorNodes = this.getDecoratorMetadata(source, '@Module');
const node = decoratorNodes[0];
const matchingProperties = node.properties
.filter((prop) => prop.kind === typescript_1.SyntaxKind.PropertyAssignment)
.filter((prop) => {
const name = prop.name;
switch (name.kind) {
case typescript_1.SyntaxKind.Identifier:
return name.getText(source) === metadata;
case typescript_1.SyntaxKind.StringLiteral:
return name.text === metadata;
default:
return false;
}
});
symbol = this.mergeSymbolAndExpr(symbol, staticOptions);
const addBlankLinesIfDynamic = () => {
symbol = staticOptions ? this.addBlankLines(symbol) : symbol;
};
if (matchingProperties.length === 0) {
const expr = node;
if (expr.properties.length === 0) {
addBlankLinesIfDynamic();
return this.insertMetadataToEmptyModuleDecorator(expr, metadata, symbol);
}
else {
addBlankLinesIfDynamic();
return this.insertNewMetadataToDecorator(expr, source, metadata, symbol);
}
}
else {
return this.insertSymbolToMetadata(source, matchingProperties, symbol, staticOptions);
}
}
getDecoratorMetadata(source, identifier) {
return this.getSourceNodes(source)
.filter((node) => node.kind === typescript_1.SyntaxKind.Decorator &&
node.expression.kind === typescript_1.SyntaxKind.CallExpression)
.map((node) => node.expression)
.filter((expr) => expr.arguments[0] &&
expr.arguments[0].kind === typescript_1.SyntaxKind.ObjectLiteralExpression)
.map((expr) => expr.arguments[0]);
}
getSourceNodes(sourceFile) {
const nodes = [sourceFile];
const result = [];
while (nodes.length > 0) {
const node = nodes.shift();
if (node) {
result.push(node);
if (node.getChildCount(sourceFile) >= 0) {
nodes.unshift(...node.getChildren());
}
}
}
return result;
}
insertMetadataToEmptyModuleDecorator(expr, metadata, symbol) {
const position = expr.getEnd() - 1;
const toInsert = ` ${metadata}: [${symbol}]`;
return this.content.split('').reduce((content, char, index) => {
if (index === position) {
return `${content}\n${toInsert}\n${char}`;
}
else {
return `${content}${char}`;
}
}, '');
}
insertNewMetadataToDecorator(expr, source, metadata, symbol) {
const node = expr.properties[expr.properties.length - 1];
const position = node.getEnd();
const text = node.getFullText(source);
const matches = text.match(/^\r?\n\s*/);
let toInsert;
if (matches) {
toInsert = `,${matches[0]}${metadata}: [${symbol}]`;
}
else {
toInsert = `, ${metadata}: [${symbol}]`;
}
return this.content.split('').reduce((content, char, index) => {
if (index === position) {
return `${content}${toInsert}${char}`;
}
else {
return `${content}${char}`;
}
}, '');
}
insertSymbolToMetadata(source, matchingProperties, symbol, staticOptions) {
const assignment = matchingProperties[0];
let node;
const arrLiteral = assignment.initializer;
if (!arrLiteral.elements) {
return this.content;
}
if (arrLiteral.elements.length === 0) {
node = arrLiteral;
}
else {
node = arrLiteral.elements;
}
if (Array.isArray(node)) {
const nodeArray = node;
const symbolsArray = nodeArray.map((childNode) => childNode.getText(source));
if (symbolsArray.includes(symbol)) {
return this.content;
}
node = node[node.length - 1];
}
let toInsert;
let position = node.getEnd();
if (node.kind === typescript_1.SyntaxKind.ArrayLiteralExpression) {
position--;
toInsert = staticOptions ? this.addBlankLines(symbol) : `${symbol}`;
}
else {
const text = node.getFullText(source);
if (text.match(/^\r?\n/)) {
toInsert = `,${text.match(/^\r?\n(\r?)\s+/)[0]}${symbol}`;
}
else {
toInsert = `, ${symbol}`;
}
}
return this.content.split('').reduce((content, char, index) => {
if (index === position) {
return `${content}${toInsert}${char}`;
}
else {
return `${content}${char}`;
}
}, '');
}
mergeSymbolAndExpr(symbol, staticOptions) {
if (!staticOptions) {
return symbol;
}
const spacing = 6;
let options = JSON.stringify(staticOptions.value, null, spacing);
options = options.replace(/\"([^(\")"]+)\":/g, '$1:');
options = options.replace(/\"/g, `'`);
options = options.slice(0, options.length - 1) + ' }';
symbol += `.${staticOptions.name}(${options})`;
return symbol;
}
addBlankLines(expr) {
return `\n ${expr}\n `;
}
}
exports.MetadataManager = MetadataManager;