| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613 |
- "use strict";
- Object.defineProperty(exports, "__esModule", { value: true });
- /**
- * @license
- * Copyright Google Inc. All Rights Reserved.
- *
- * Use of this source code is governed by an MIT-style license that can be
- * found in the LICENSE file at https://angular.io/license
- */
- const ts = require("../third_party/github.com/Microsoft/TypeScript/lib/typescript");
- const change_1 = require("./change");
- /**
- * Add Import `import { symbolName } from fileName` if the import doesn't exit
- * already. Assumes fileToEdit can be resolved and accessed.
- * @param fileToEdit (file we want to add import to)
- * @param symbolName (item to import)
- * @param fileName (path to the file)
- * @param isDefault (if true, import follows style for importing default exports)
- * @return Change
- */
- function insertImport(source, fileToEdit, symbolName, fileName, isDefault = false) {
- const rootNode = source;
- const allImports = findNodes(rootNode, ts.SyntaxKind.ImportDeclaration);
- // get nodes that map to import statements from the file fileName
- const relevantImports = allImports.filter(node => {
- // StringLiteral of the ImportDeclaration is the import file (fileName in this case).
- const importFiles = node.getChildren()
- .filter(child => child.kind === ts.SyntaxKind.StringLiteral)
- .map(n => n.text);
- return importFiles.filter(file => file === fileName).length === 1;
- });
- if (relevantImports.length > 0) {
- let importsAsterisk = false;
- // imports from import file
- const imports = [];
- relevantImports.forEach(n => {
- Array.prototype.push.apply(imports, findNodes(n, ts.SyntaxKind.Identifier));
- if (findNodes(n, ts.SyntaxKind.AsteriskToken).length > 0) {
- importsAsterisk = true;
- }
- });
- // if imports * from fileName, don't add symbolName
- if (importsAsterisk) {
- return new change_1.NoopChange();
- }
- const importTextNodes = imports.filter(n => n.text === symbolName);
- // insert import if it's not there
- if (importTextNodes.length === 0) {
- const fallbackPos = findNodes(relevantImports[0], ts.SyntaxKind.CloseBraceToken)[0].getStart() ||
- findNodes(relevantImports[0], ts.SyntaxKind.FromKeyword)[0].getStart();
- return insertAfterLastOccurrence(imports, `, ${symbolName}`, fileToEdit, fallbackPos);
- }
- return new change_1.NoopChange();
- }
- // no such import declaration exists
- const useStrict = findNodes(rootNode, ts.SyntaxKind.StringLiteral)
- .filter((n) => n.text === 'use strict');
- let fallbackPos = 0;
- if (useStrict.length > 0) {
- fallbackPos = useStrict[0].end;
- }
- const open = isDefault ? '' : '{ ';
- const close = isDefault ? '' : ' }';
- // if there are no imports or 'use strict' statement, insert import at beginning of file
- const insertAtBeginning = allImports.length === 0 && useStrict.length === 0;
- const separator = insertAtBeginning ? '' : ';\n';
- const toInsert = `${separator}import ${open}${symbolName}${close}` +
- ` from '${fileName}'${insertAtBeginning ? ';\n' : ''}`;
- return insertAfterLastOccurrence(allImports, toInsert, fileToEdit, fallbackPos, ts.SyntaxKind.StringLiteral);
- }
- exports.insertImport = insertImport;
- /**
- * Find all nodes from the AST in the subtree of node of SyntaxKind kind.
- * @param node
- * @param kind
- * @param max The maximum number of items to return.
- * @param recursive Continue looking for nodes of kind recursive until end
- * the last child even when node of kind has been found.
- * @return all nodes of kind, or [] if none is found
- */
- function findNodes(node, kind, max = Infinity, recursive = false) {
- if (!node || max == 0) {
- return [];
- }
- const arr = [];
- if (node.kind === kind) {
- arr.push(node);
- max--;
- }
- if (max > 0 && (recursive || node.kind !== kind)) {
- for (const child of node.getChildren()) {
- findNodes(child, kind, max).forEach(node => {
- if (max > 0) {
- arr.push(node);
- }
- max--;
- });
- if (max <= 0) {
- break;
- }
- }
- }
- return arr;
- }
- exports.findNodes = findNodes;
- /**
- * Get all the nodes from a source.
- * @param sourceFile The source file object.
- * @returns {Observable<ts.Node>} An observable of all the nodes in the source.
- */
- function 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;
- }
- exports.getSourceNodes = getSourceNodes;
- function findNode(node, kind, text) {
- if (node.kind === kind && node.getText() === text) {
- // throw new Error(node.getText());
- return node;
- }
- let foundNode = null;
- ts.forEachChild(node, childNode => {
- foundNode = foundNode || findNode(childNode, kind, text);
- });
- return foundNode;
- }
- exports.findNode = findNode;
- /**
- * Helper for sorting nodes.
- * @return function to sort nodes in increasing order of position in sourceFile
- */
- function nodesByPosition(first, second) {
- return first.getStart() - second.getStart();
- }
- /**
- * Insert `toInsert` after the last occurence of `ts.SyntaxKind[nodes[i].kind]`
- * or after the last of occurence of `syntaxKind` if the last occurence is a sub child
- * of ts.SyntaxKind[nodes[i].kind] and save the changes in file.
- *
- * @param nodes insert after the last occurence of nodes
- * @param toInsert string to insert
- * @param file file to insert changes into
- * @param fallbackPos position to insert if toInsert happens to be the first occurence
- * @param syntaxKind the ts.SyntaxKind of the subchildren to insert after
- * @return Change instance
- * @throw Error if toInsert is first occurence but fall back is not set
- */
- function insertAfterLastOccurrence(nodes, toInsert, file, fallbackPos, syntaxKind) {
- let lastItem;
- for (const node of nodes) {
- if (!lastItem || lastItem.getStart() < node.getStart()) {
- lastItem = node;
- }
- }
- if (syntaxKind && lastItem) {
- lastItem = findNodes(lastItem, syntaxKind).sort(nodesByPosition).pop();
- }
- if (!lastItem && fallbackPos == undefined) {
- throw new Error(`tried to insert ${toInsert} as first occurence with no fallback position`);
- }
- const lastItemPosition = lastItem ? lastItem.getEnd() : fallbackPos;
- return new change_1.InsertChange(file, lastItemPosition, toInsert);
- }
- exports.insertAfterLastOccurrence = insertAfterLastOccurrence;
- function getContentOfKeyLiteral(_source, node) {
- if (node.kind == ts.SyntaxKind.Identifier) {
- return node.text;
- }
- else if (node.kind == ts.SyntaxKind.StringLiteral) {
- return node.text;
- }
- else {
- return null;
- }
- }
- exports.getContentOfKeyLiteral = getContentOfKeyLiteral;
- function _angularImportsFromNode(node, _sourceFile) {
- const ms = node.moduleSpecifier;
- let modulePath;
- switch (ms.kind) {
- case ts.SyntaxKind.StringLiteral:
- modulePath = ms.text;
- break;
- default:
- return {};
- }
- if (!modulePath.startsWith('@angular/')) {
- return {};
- }
- if (node.importClause) {
- if (node.importClause.name) {
- // This is of the form `import Name from 'path'`. Ignore.
- return {};
- }
- else if (node.importClause.namedBindings) {
- const nb = node.importClause.namedBindings;
- if (nb.kind == ts.SyntaxKind.NamespaceImport) {
- // This is of the form `import * as name from 'path'`. Return `name.`.
- return {
- [nb.name.text + '.']: modulePath,
- };
- }
- else {
- // This is of the form `import {a,b,c} from 'path'`
- const namedImports = nb;
- return namedImports.elements
- .map((is) => is.propertyName ? is.propertyName.text : is.name.text)
- .reduce((acc, curr) => {
- acc[curr] = modulePath;
- return acc;
- }, {});
- }
- }
- return {};
- }
- else {
- // This is of the form `import 'path';`. Nothing to do.
- return {};
- }
- }
- function getDecoratorMetadata(source, identifier, module) {
- const angularImports = findNodes(source, ts.SyntaxKind.ImportDeclaration)
- .map((node) => _angularImportsFromNode(node, source))
- .reduce((acc, current) => {
- for (const key of Object.keys(current)) {
- acc[key] = current[key];
- }
- return acc;
- }, {});
- return getSourceNodes(source)
- .filter(node => {
- return node.kind == ts.SyntaxKind.Decorator
- && node.expression.kind == ts.SyntaxKind.CallExpression;
- })
- .map(node => node.expression)
- .filter(expr => {
- if (expr.expression.kind == ts.SyntaxKind.Identifier) {
- const id = expr.expression;
- return id.text == identifier && angularImports[id.text] === module;
- }
- else if (expr.expression.kind == ts.SyntaxKind.PropertyAccessExpression) {
- // This covers foo.NgModule when importing * as foo.
- const paExpr = expr.expression;
- // If the left expression is not an identifier, just give up at that point.
- if (paExpr.expression.kind !== ts.SyntaxKind.Identifier) {
- return false;
- }
- const id = paExpr.name.text;
- const moduleId = paExpr.expression.text;
- return id === identifier && (angularImports[moduleId + '.'] === module);
- }
- return false;
- })
- .filter(expr => expr.arguments[0]
- && expr.arguments[0].kind == ts.SyntaxKind.ObjectLiteralExpression)
- .map(expr => expr.arguments[0]);
- }
- exports.getDecoratorMetadata = getDecoratorMetadata;
- function findClassDeclarationParent(node) {
- if (ts.isClassDeclaration(node)) {
- return node;
- }
- return node.parent && findClassDeclarationParent(node.parent);
- }
- /**
- * Given a source file with @NgModule class(es), find the name of the first @NgModule class.
- *
- * @param source source file containing one or more @NgModule
- * @returns the name of the first @NgModule, or `undefined` if none is found
- */
- function getFirstNgModuleName(source) {
- // First, find the @NgModule decorators.
- const ngModulesMetadata = getDecoratorMetadata(source, 'NgModule', '@angular/core');
- if (ngModulesMetadata.length === 0) {
- return undefined;
- }
- // Then walk parent pointers up the AST, looking for the ClassDeclaration parent of the NgModule
- // metadata.
- const moduleClass = findClassDeclarationParent(ngModulesMetadata[0]);
- if (!moduleClass || !moduleClass.name) {
- return undefined;
- }
- // Get the class name of the module ClassDeclaration.
- return moduleClass.name.text;
- }
- exports.getFirstNgModuleName = getFirstNgModuleName;
- function getMetadataField(node, metadataField) {
- return node.properties
- .filter(prop => ts.isPropertyAssignment(prop))
- // Filter out every fields that's not "metadataField". Also handles string literals
- // (but not expressions).
- .filter(({ name }) => {
- return (ts.isIdentifier(name) || ts.isStringLiteral(name))
- && name.getText() === metadataField;
- });
- }
- exports.getMetadataField = getMetadataField;
- function addSymbolToNgModuleMetadata(source, ngModulePath, metadataField, symbolName, importPath = null) {
- const nodes = getDecoratorMetadata(source, 'NgModule', '@angular/core');
- let node = nodes[0]; // tslint:disable-line:no-any
- // Find the decorator declaration.
- if (!node) {
- return [];
- }
- // Get all the children property assignment of object literals.
- const matchingProperties = getMetadataField(node, metadataField);
- // Get the last node of the array literal.
- if (!matchingProperties) {
- return [];
- }
- if (matchingProperties.length == 0) {
- // We haven't found the field in the metadata declaration. Insert a new field.
- const expr = node;
- let position;
- let toInsert;
- if (expr.properties.length == 0) {
- position = expr.getEnd() - 1;
- toInsert = ` ${metadataField}: [${symbolName}]\n`;
- }
- else {
- node = expr.properties[expr.properties.length - 1];
- position = node.getEnd();
- // Get the indentation of the last element, if any.
- const text = node.getFullText(source);
- const matches = text.match(/^\r?\n\s*/);
- if (matches && matches.length > 0) {
- toInsert = `,${matches[0]}${metadataField}: [${symbolName}]`;
- }
- else {
- toInsert = `, ${metadataField}: [${symbolName}]`;
- }
- }
- if (importPath !== null) {
- return [
- new change_1.InsertChange(ngModulePath, position, toInsert),
- insertImport(source, ngModulePath, symbolName.replace(/\..*$/, ''), importPath),
- ];
- }
- else {
- return [new change_1.InsertChange(ngModulePath, position, toInsert)];
- }
- }
- const assignment = matchingProperties[0];
- // If it's not an array, nothing we can do really.
- if (assignment.initializer.kind !== ts.SyntaxKind.ArrayLiteralExpression) {
- return [];
- }
- const arrLiteral = assignment.initializer;
- if (arrLiteral.elements.length == 0) {
- // Forward the property.
- node = arrLiteral;
- }
- else {
- node = arrLiteral.elements;
- }
- if (!node) {
- // tslint:disable-next-line: no-console
- console.error('No app module found. Please add your new class to your component.');
- return [];
- }
- if (Array.isArray(node)) {
- const nodeArray = node;
- const symbolsArray = nodeArray.map(node => node.getText());
- if (symbolsArray.includes(symbolName)) {
- return [];
- }
- node = node[node.length - 1];
- }
- let toInsert;
- let position = node.getEnd();
- if (node.kind == ts.SyntaxKind.ObjectLiteralExpression) {
- // We haven't found the field in the metadata declaration. Insert a new
- // field.
- const expr = node;
- if (expr.properties.length == 0) {
- position = expr.getEnd() - 1;
- toInsert = ` ${symbolName}\n`;
- }
- else {
- // Get the indentation of the last element, if any.
- const text = node.getFullText(source);
- if (text.match(/^\r?\r?\n/)) {
- toInsert = `,${text.match(/^\r?\n\s*/)[0]}${symbolName}`;
- }
- else {
- toInsert = `, ${symbolName}`;
- }
- }
- }
- else if (node.kind == ts.SyntaxKind.ArrayLiteralExpression) {
- // We found the field but it's empty. Insert it just before the `]`.
- position--;
- toInsert = `${symbolName}`;
- }
- else {
- // Get the indentation of the last element, if any.
- const text = node.getFullText(source);
- if (text.match(/^\r?\n/)) {
- toInsert = `,${text.match(/^\r?\n(\r?)\s*/)[0]}${symbolName}`;
- }
- else {
- toInsert = `, ${symbolName}`;
- }
- }
- if (importPath !== null) {
- return [
- new change_1.InsertChange(ngModulePath, position, toInsert),
- insertImport(source, ngModulePath, symbolName.replace(/\..*$/, ''), importPath),
- ];
- }
- return [new change_1.InsertChange(ngModulePath, position, toInsert)];
- }
- exports.addSymbolToNgModuleMetadata = addSymbolToNgModuleMetadata;
- /**
- * Custom function to insert a declaration (component, pipe, directive)
- * into NgModule declarations. It also imports the component.
- */
- function addDeclarationToModule(source, modulePath, classifiedName, importPath) {
- return addSymbolToNgModuleMetadata(source, modulePath, 'declarations', classifiedName, importPath);
- }
- exports.addDeclarationToModule = addDeclarationToModule;
- /**
- * Custom function to insert an NgModule into NgModule imports. It also imports the module.
- */
- function addImportToModule(source, modulePath, classifiedName, importPath) {
- return addSymbolToNgModuleMetadata(source, modulePath, 'imports', classifiedName, importPath);
- }
- exports.addImportToModule = addImportToModule;
- /**
- * Custom function to insert a provider into NgModule. It also imports it.
- */
- function addProviderToModule(source, modulePath, classifiedName, importPath) {
- return addSymbolToNgModuleMetadata(source, modulePath, 'providers', classifiedName, importPath);
- }
- exports.addProviderToModule = addProviderToModule;
- /**
- * Custom function to insert an export into NgModule. It also imports it.
- */
- function addExportToModule(source, modulePath, classifiedName, importPath) {
- return addSymbolToNgModuleMetadata(source, modulePath, 'exports', classifiedName, importPath);
- }
- exports.addExportToModule = addExportToModule;
- /**
- * Custom function to insert an export into NgModule. It also imports it.
- */
- function addBootstrapToModule(source, modulePath, classifiedName, importPath) {
- return addSymbolToNgModuleMetadata(source, modulePath, 'bootstrap', classifiedName, importPath);
- }
- exports.addBootstrapToModule = addBootstrapToModule;
- /**
- * Custom function to insert an entryComponent into NgModule. It also imports it.
- * @deprecated - Since version 9.0.0 with Ivy, entryComponents is no longer necessary.
- */
- function addEntryComponentToModule(source, modulePath, classifiedName, importPath) {
- return addSymbolToNgModuleMetadata(source, modulePath, 'entryComponents', classifiedName, importPath);
- }
- exports.addEntryComponentToModule = addEntryComponentToModule;
- /**
- * Determine if an import already exists.
- */
- function isImported(source, classifiedName, importPath) {
- const allNodes = getSourceNodes(source);
- const matchingNodes = allNodes
- .filter(node => node.kind === ts.SyntaxKind.ImportDeclaration)
- .filter((imp) => imp.moduleSpecifier.kind === ts.SyntaxKind.StringLiteral)
- .filter((imp) => {
- return imp.moduleSpecifier.text === importPath;
- })
- .filter((imp) => {
- if (!imp.importClause) {
- return false;
- }
- const nodes = findNodes(imp.importClause, ts.SyntaxKind.ImportSpecifier)
- .filter(n => n.getText() === classifiedName);
- return nodes.length > 0;
- });
- return matchingNodes.length > 0;
- }
- exports.isImported = isImported;
- /**
- * This function returns the name of the environment export
- * whether this export is aliased or not. If the environment file
- * is not imported, then it will return `null`.
- */
- function getEnvironmentExportName(source) {
- // Initial value is `null` as we don't know yet if the user
- // has imported `environment` into the root module or not.
- let environmentExportName = null;
- const allNodes = getSourceNodes(source);
- allNodes
- .filter(node => node.kind === ts.SyntaxKind.ImportDeclaration)
- .filter((declaration) => declaration.moduleSpecifier.kind === ts.SyntaxKind.StringLiteral &&
- declaration.importClause !== undefined)
- .map((declaration) =>
- // If `importClause` property is defined then the first
- // child will be `NamedImports` object (or `namedBindings`).
- declaration.importClause.getChildAt(0))
- // Find those `NamedImports` object that contains `environment` keyword
- // in its text. E.g. `{ environment as env }`.
- .filter((namedImports) => namedImports.getText().includes('environment'))
- .forEach((namedImports) => {
- for (const specifier of namedImports.elements) {
- // `propertyName` is defined if the specifier
- // has an aliased import.
- const name = specifier.propertyName || specifier.name;
- // Find specifier that contains `environment` keyword in its text.
- // Whether it's `environment` or `environment as env`.
- if (name.text.includes('environment')) {
- environmentExportName = specifier.name.text;
- }
- }
- });
- return environmentExportName;
- }
- exports.getEnvironmentExportName = getEnvironmentExportName;
- /**
- * Returns the RouterModule declaration from NgModule metadata, if any.
- */
- function getRouterModuleDeclaration(source) {
- const result = getDecoratorMetadata(source, 'NgModule', '@angular/core');
- const node = result[0];
- const matchingProperties = getMetadataField(node, 'imports');
- if (!matchingProperties) {
- return;
- }
- const assignment = matchingProperties[0];
- if (assignment.initializer.kind !== ts.SyntaxKind.ArrayLiteralExpression) {
- return;
- }
- const arrLiteral = assignment.initializer;
- return arrLiteral.elements
- .filter(el => el.kind === ts.SyntaxKind.CallExpression)
- .find(el => el.getText().startsWith('RouterModule'));
- }
- exports.getRouterModuleDeclaration = getRouterModuleDeclaration;
- /**
- * Adds a new route declaration to a router module (i.e. has a RouterModule declaration)
- */
- function addRouteDeclarationToModule(source, fileToAdd, routeLiteral) {
- const routerModuleExpr = getRouterModuleDeclaration(source);
- if (!routerModuleExpr) {
- throw new Error(`Couldn't find a route declaration in ${fileToAdd}.`);
- }
- const scopeConfigMethodArgs = routerModuleExpr.arguments;
- if (!scopeConfigMethodArgs.length) {
- const { line } = source.getLineAndCharacterOfPosition(routerModuleExpr.getStart());
- throw new Error(`The router module method doesn't have arguments ` +
- `at line ${line} in ${fileToAdd}`);
- }
- let routesArr;
- const routesArg = scopeConfigMethodArgs[0];
- // Check if the route declarations array is
- // an inlined argument of RouterModule or a standalone variable
- if (ts.isArrayLiteralExpression(routesArg)) {
- routesArr = routesArg;
- }
- else {
- const routesVarName = routesArg.getText();
- let routesVar;
- if (routesArg.kind === ts.SyntaxKind.Identifier) {
- routesVar = source.statements
- .filter((s) => s.kind === ts.SyntaxKind.VariableStatement)
- .find((v) => {
- return v.declarationList.declarations[0].name.getText() === routesVarName;
- });
- }
- if (!routesVar) {
- const { line } = source.getLineAndCharacterOfPosition(routesArg.getStart());
- throw new Error(`No route declaration array was found that corresponds ` +
- `to router module at line ${line} in ${fileToAdd}`);
- }
- routesArr = findNodes(routesVar, ts.SyntaxKind.ArrayLiteralExpression, 1)[0];
- }
- const occurrencesCount = routesArr.elements.length;
- const text = routesArr.getFullText(source);
- let route = routeLiteral;
- let insertPos = routesArr.elements.pos;
- if (occurrencesCount > 0) {
- const lastRouteLiteral = [...routesArr.elements].pop();
- const lastRouteIsWildcard = ts.isObjectLiteralExpression(lastRouteLiteral)
- && lastRouteLiteral
- .properties
- .some(n => (ts.isPropertyAssignment(n)
- && ts.isIdentifier(n.name)
- && n.name.text === 'path'
- && ts.isStringLiteral(n.initializer)
- && n.initializer.text === '**'));
- const indentation = text.match(/\r?\n(\r?)\s*/) || [];
- const routeText = `${indentation[0] || ' '}${routeLiteral}`;
- // Add the new route before the wildcard route
- // otherwise we'll always redirect to the wildcard route
- if (lastRouteIsWildcard) {
- insertPos = lastRouteLiteral.pos;
- route = `${routeText},`;
- }
- else {
- insertPos = lastRouteLiteral.end;
- route = `,${routeText}`;
- }
- }
- return new change_1.InsertChange(fileToAdd, insertPos, route);
- }
- exports.addRouteDeclarationToModule = addRouteDeclarationToModule;
|