UNPKG

ntts

Version:

A CLI tool for refactoring an existing NodeJs application to a fully functional TypeScript application.

130 lines (129 loc) 7.54 kB
"use strict"; var __assign = (this && this.__assign) || function () { __assign = Object.assign || function(t) { for (var s, i = 1, n = arguments.length; i < n; i++) { s = arguments[i]; for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p)) t[p] = s[p]; } return t; }; return __assign.apply(this, arguments); }; var __importDefault = (this && this.__importDefault) || function (mod) { return (mod && mod.__esModule) ? mod : { "default": mod }; }; Object.defineProperty(exports, "__esModule", { value: true }); var ts_morph_1 = require("ts-morph"); var variable_name_generator_1 = __importDefault(require("../../helpers/variable-name-generator/variable-name-generator")); var variable_creator_1 = __importDefault(require("../helpers/variable-creator")); var export_parser_1 = __importDefault(require("../helpers/export-parser")); var write_access_checker_1 = __importDefault(require("../../helpers/write-access-checker/write-access-checker")); var TopLevelRefactor = /** @class */ (function () { function TopLevelRefactor() { } var _a; _a = TopLevelRefactor; TopLevelRefactor.refactorTopLevelExport = function (exportName, binary, expression, accessExpression, exportedVariables, usedNames, defaultExport, sourceFile) { var exported = export_parser_1.default.exportVariableExists(exportName, exportedVariables, defaultExport); var exportedNames = exportedVariables.map(function (e) { return e.name; }); if (exported && exported.directExport) { return _a.refactorExistingDirectExport(exported, usedNames.concat(exportedNames), binary, accessExpression, exportedVariables, sourceFile); } if (exported) { _a.refactorExistingExport(exported, accessExpression, sourceFile); return exportedVariables; } if (ts_morph_1.Node.isIdentifier(binary.getRight())) { var newExport_1 = _a.refactorIdentifierAssignment(exportName, binary, expression, usedNames.concat(exportedNames), defaultExport, sourceFile); return exportedVariables.concat(newExport_1); } if (ts_morph_1.Node.isClassExpression(binary.getRight())) { var newExport_2 = _a.refactorClassAssignment(exportName, binary, expression, usedNames.concat(exportedNames), defaultExport, sourceFile); return exportedVariables.concat(newExport_2); } var newExport = _a.refactorNewExport(exportName, binary, expression, usedNames.concat(exportedNames), defaultExport, sourceFile); return exportedVariables.concat(newExport); }; TopLevelRefactor.refactorExistingDirectExport = function (exported, usedNames, binary, accessExpression, exportedVariables, sourceFile) { var usableName = variable_name_generator_1.default.getUsableVariableName(exported.name, usedNames); var index = export_parser_1.default.getSourceFileIndex(binary); variable_creator_1.default.createVariable(usableName, index, exported.name, ts_morph_1.VariableDeclarationKind.Let, sourceFile); accessExpression.replaceWithText(usableName); return exportedVariables.map(function (variable) { if (variable.name === exported.name && variable.alias === exported.alias) { return { name: usableName, alias: exported.alias || exported.name, defaultExport: exported.defaultExport }; } return variable; }); }; TopLevelRefactor.refactorExistingExport = function (exported, accessExpression, sourceFile) { sourceFile.getVariableStatementOrThrow(exported.name).setDeclarationKind(ts_morph_1.VariableDeclarationKind.Let); accessExpression.replaceWithText(exported.name); }; TopLevelRefactor.refactorClassAssignment = function (exportName, binary, expression, usedNames, defaultExport, sourceFile) { var assignedClass = binary.getRight().asKindOrThrow(ts_morph_1.SyntaxKind.ClassExpression); var className = _a.refactorClassExpressionExport(assignedClass, exportName, usedNames); sourceFile.insertStatements(export_parser_1.default.getSourceFileIndex(binary), assignedClass.getText()); expression.remove(); var exportModel = { name: className, directExport: true, defaultExport: defaultExport, }; return exportName === className ? exportModel : __assign(__assign({}, exportModel), { alias: exportName }); }; TopLevelRefactor.refactorIdentifierAssignment = function (exportName, binary, expression, usedNames, defaultExport, sourceFile) { var assignedIdentifier = binary.getRight().asKindOrThrow(ts_morph_1.SyntaxKind.Identifier); if (_a.canExportDirectly(assignedIdentifier, sourceFile)) { var assignedName = assignedIdentifier.getText(); expression.remove(); var exportModel = { name: assignedName, directExport: true, defaultExport: defaultExport, }; return exportName === assignedName ? exportModel : __assign(__assign({}, exportModel), { alias: exportName }); } var usableName = variable_name_generator_1.default.getUsableVariableName(exportName, usedNames); variable_creator_1.default.createVariable(usableName, export_parser_1.default.getSourceFileIndex(binary), binary.getRight().getText(), ts_morph_1.VariableDeclarationKind.Const, sourceFile); expression.remove(); return { name: usableName, alias: exportName }; }; TopLevelRefactor.refactorNewExport = function (exportName, binary, expression, usedNames, defaultExport, sourceFile) { var usableName = variable_name_generator_1.default.getUsableVariableName(exportName, usedNames); variable_creator_1.default.createVariable(usableName, export_parser_1.default.getSourceFileIndex(binary), binary.getRight().getText(), ts_morph_1.VariableDeclarationKind.Const, sourceFile); expression.remove(); return { name: usableName, alias: exportName !== usableName ? exportName : undefined, defaultExport: defaultExport, }; }; TopLevelRefactor.refactorClassExpressionExport = function (classExpression, exportName, usedNames) { var className = classExpression.getName(); if (className && !usedNames.includes(className)) { return className; } if (!usedNames.includes(exportName)) { classExpression.rename(exportName); return exportName; } var usableClassName = variable_name_generator_1.default.getUsableVariableName("".concat(exportName, "Type"), usedNames); classExpression.rename(usableClassName); return usableClassName; }; TopLevelRefactor.canExportDirectly = function (identifier, sourceFile) { var nodeName = identifier.getText(); var implementation = sourceFile.getVariableDeclaration(nodeName) || sourceFile.getFunction(nodeName) || sourceFile.getClass(nodeName); if (ts_morph_1.Node.isFunctionDeclaration(implementation) || ts_morph_1.Node.isClassDeclaration(implementation)) { return true; } if (ts_morph_1.Node.isVariableDeclaration(implementation)) { return !write_access_checker_1.default.hasValueChanged(implementation); } return false; }; return TopLevelRefactor; }()); exports.default = TopLevelRefactor;