@nstudio/angular 15.0.4-rc.1 → 16.5.0

package/src/utils/ast.js

@@ -1,24 +1,25 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getDecoratorPropertyValueNode = exports.readBootstrapInfo = exports.addEntryComponents = exports.addDeclarationToModule = exports.addProviderToModule = exports.addRoute = exports.getBootstrapComponent = exports.addImportToTestBed = exports.addImportToModule = exports.removeFromNgModule = exports.addSymbolToNgModuleMetadata = exports.addToCollection = void 0;
-const ts = require("typescript");
+exports.getTsSourceFile = exports.getDecoratorPropertyValueNode = exports.readBootstrapInfo = exports.addEntryComponents = exports.addDeclarationToModule = exports.addProviderToComponent = exports.addProviderToModule = exports.addProviderToBootstrapApplication = exports.addRouteToNgModule = exports.getBootstrapComponent = exports.replaceIntoToTestBed = exports.addDeclarationsToTestBed = exports.addImportToTestBed = exports.addImportToModule = exports.addImportToPipe = exports.addImportToDirective = exports.addImportToComponent = exports.removeFromNgModule = exports._addSymbolToNgModuleMetadata = exports.getDecoratorMetadata = exports.isStandalone = exports.addToCollection = void 0;
+const ensure_typescript_1 = require("@nx/js/src/utils/typescript/ensure-typescript");
 const typescript_1 = require("nx/src/utils/typescript");
-const path = require("path");
-const xplat_utils_1 = require("@nstudio/xplat-utils");
-const xplat_1 = require("@nstudio/xplat");
-function addToCollection(source, barrelIndexPath, symbolName, insertSpaces = '') {
+const ts = require("typescript");
+const js_1 = require("@nx/js");
+const path_1 = require("path");
+const devkit_1 = require("@nx/devkit");
+let tsModule;
+function addToCollection(tree, source, barrelIndexPath, symbolName, insertSpaces = '') {
     const collection = getCollection(source);
     if (!collection)
-        return [];
+        return source;
     // if (!collection) return [new NoopChange()];
     // return [new NoopChange()];
+    console.log('collection.hasTrailingComma:', collection.hasTrailingComma);
     if (collection.hasTrailingComma || collection.length === 0) {
-        return [new xplat_1.InsertChange(barrelIndexPath, collection.end, symbolName)];
+        return (0, js_1.insertChange)(tree, source, barrelIndexPath, collection.end, symbolName);
     }
     else {
-        return [
-            new xplat_1.InsertChange(barrelIndexPath, collection.end, `,\n${insertSpaces}${symbolName}`),
-        ];
+        return (0, js_1.insertChange)(tree, source, barrelIndexPath, collection.end, `,\n${insertSpaces}${symbolName}`);
     }
 }
 exports.addToCollection = addToCollection;
@@ -36,10 +37,13 @@ function getCollection(source) {
     return null;
 }
 function _angularImportsFromNode(node, _sourceFile) {
+    if (!tsModule) {
+        tsModule = (0, ensure_typescript_1.ensureTypescript)();
+    }
     const ms = node.moduleSpecifier;
     let modulePath;
     switch (ms.kind) {
-        case ts.SyntaxKind.StringLiteral:
+        case tsModule.SyntaxKind.StringLiteral:
             modulePath = ms.text;
             break;
         default:
@@ -55,10 +59,10 @@ function _angularImportsFromNode(node, _sourceFile) {
         }
         else if (node.importClause.namedBindings) {
             const nb = node.importClause.namedBindings;
-            if (nb.kind == ts.SyntaxKind.NamespaceImport) {
+            if (nb.kind == tsModule.SyntaxKind.NamespaceImport) {
                 // This is of the form `import * as name from 'path'`. Return `name.`.
                 return {
-                    [nb.name.text + '.']: modulePath,
+                    [`${nb.name.text}.`]: modulePath,
                 };
             }
             else {
@@ -79,8 +83,21 @@ function _angularImportsFromNode(node, _sourceFile) {
         return {};
     }
 }
+/**
+ * Check if the Component, Directive or Pipe is standalone
+ * @param sourceFile TS Source File containing the token to check
+ * @param decoratorName The type of decorator to check (Component, Directive, Pipe)
+ */
+function isStandalone(sourceFile, decoratorName) {
+    const decoratorMetadata = getDecoratorMetadata(sourceFile, decoratorName, '@angular/core');
+    return decoratorMetadata.some((node) => node.getText().includes('standalone: true'));
+}
+exports.isStandalone = isStandalone;
 function getDecoratorMetadata(source, identifier, module) {
-    const angularImports = (0, typescript_1.findNodes)(source, ts.SyntaxKind.ImportDeclaration)
+    if (!tsModule) {
+        tsModule = (0, ensure_typescript_1.ensureTypescript)();
+    }
+    const angularImports = (0, typescript_1.findNodes)(source, tsModule.SyntaxKind.ImportDeclaration)
         .map((node) => _angularImportsFromNode(node, source))
         .reduce((acc, current) => {
         for (const key of Object.keys(current)) {
@@ -88,60 +105,65 @@ function getDecoratorMetadata(source, identifier, module) {
         }
         return acc;
     }, {});
-    return (0, xplat_1.getSourceNodes)(source)
+    return (0, js_1.getSourceNodes)(source)
         .filter((node) => {
-        return (node.kind == ts.SyntaxKind.Decorator &&
-            node.expression.kind == ts.SyntaxKind.CallExpression);
+        return (node.kind == tsModule.SyntaxKind.Decorator &&
+            node.expression.kind ==
+                tsModule.SyntaxKind.CallExpression);
     })
         .map((node) => node.expression)
         .filter((expr) => {
-        if (expr.expression.kind == ts.SyntaxKind.Identifier) {
+        if (expr.expression.kind == tsModule.SyntaxKind.Identifier) {
             const id = expr.expression;
             return (id.getFullText(source) == identifier &&
                 angularImports[id.getFullText(source)] === module);
         }
-        else if (expr.expression.kind == ts.SyntaxKind.PropertyAccessExpression) {
+        else if (expr.expression.kind == tsModule.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) {
+            if (paExpr.expression.kind !== tsModule.SyntaxKind.Identifier) {
                 return false;
             }
             const id = paExpr.name.text;
             const moduleId = paExpr.expression.getText(source);
-            return id === identifier && angularImports[moduleId + '.'] === module;
+            return id === identifier && angularImports[`${moduleId}.`] === module;
         }
         return false;
     })
         .filter((expr) => expr.arguments[0] &&
-        expr.arguments[0].kind == ts.SyntaxKind.ObjectLiteralExpression)
+        expr.arguments[0].kind == tsModule.SyntaxKind.ObjectLiteralExpression)
         .map((expr) => expr.arguments[0]);
 }
-function addSymbolToNgModuleMetadata(source, ngModulePath, metadataField, expression) {
-    const nodes = getDecoratorMetadata(source, 'NgModule', '@angular/core');
+exports.getDecoratorMetadata = getDecoratorMetadata;
+function _addSymbolToDecoratorMetadata(host, source, filePath, metadataField, expression, decoratorName) {
+    const nodes = getDecoratorMetadata(source, decoratorName, '@angular/core');
     let node = nodes[0]; // tslint:disable-line:no-any
     // Find the decorator declaration.
     if (!node) {
-        return [];
+        return source;
+    }
+    if (!tsModule) {
+        tsModule = (0, ensure_typescript_1.ensureTypescript)();
     }
     // Get all the children property assignment of object literals.
     const matchingProperties = node.properties
-        .filter((prop) => prop.kind == ts.SyntaxKind.PropertyAssignment)
+        .filter((prop) => prop.kind == tsModule.SyntaxKind.PropertyAssignment)
         // Filter out every fields that's not "metadataField". Also handles string literals
         // (but not expressions).
         .filter((prop) => {
         const name = prop.name;
         switch (name.kind) {
-            case ts.SyntaxKind.Identifier:
+            case tsModule.SyntaxKind.Identifier:
                 return name.getText(source) == metadataField;
-            case ts.SyntaxKind.StringLiteral:
+            case tsModule.SyntaxKind.StringLiteral:
                 return name.text == metadataField;
         }
         return false;
     });
     // Get the last node of the array literal.
     if (!matchingProperties) {
-        return [];
+        return source;
     }
     if (matchingProperties.length == 0) {
         // We haven't found the field in the metadata declaration. Insert a new field.
@@ -164,13 +186,12 @@ function addSymbolToNgModuleMetadata(source, ngModulePath, metadataField, expres
                 toInsert = `, ${metadataField}: [${expression}]`;
             }
         }
-        const newMetadataProperty = new xplat_1.InsertChange(ngModulePath, position, toInsert);
-        return [newMetadataProperty];
+        return (0, js_1.insertChange)(host, source, filePath, 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 [];
+    if (assignment.initializer.kind !== tsModule.SyntaxKind.ArrayLiteralExpression) {
+        return source;
     }
     const arrLiteral = assignment.initializer;
     if (arrLiteral.elements.length == 0) {
@@ -182,20 +203,20 @@ function addSymbolToNgModuleMetadata(source, ngModulePath, metadataField, expres
     }
     if (!node) {
         console.log('No app module found. Please add your new class to your component.');
-        return [];
+        return source;
     }
     const isArray = Array.isArray(node);
     if (isArray) {
         const nodeArray = node;
         const symbolsArray = nodeArray.map((node) => node.getText());
         if (symbolsArray.includes(expression)) {
-            return [];
+            return source;
         }
         node = node[node.length - 1];
     }
     let toInsert;
     let position = node.getEnd();
-    if (!isArray && node.kind == ts.SyntaxKind.ObjectLiteralExpression) {
+    if (!isArray && node.kind == tsModule.SyntaxKind.ObjectLiteralExpression) {
         // We haven't found the field in the metadata declaration. Insert a new
         // field.
         const expr = node;
@@ -216,7 +237,8 @@ function addSymbolToNgModuleMetadata(source, ngModulePath, metadataField, expres
             }
         }
     }
-    else if (!isArray && node.kind == ts.SyntaxKind.ArrayLiteralExpression) {
+    else if (!isArray &&
+        node.kind == tsModule.SyntaxKind.ArrayLiteralExpression) {
         // We found the field but it's empty. Insert it just before the `]`.
         position--;
         toInsert = `${expression}`;
@@ -231,54 +253,130 @@ function addSymbolToNgModuleMetadata(source, ngModulePath, metadataField, expres
             toInsert = `, ${expression}`;
         }
     }
-    const insert = new xplat_1.InsertChange(ngModulePath, position, toInsert);
-    return [insert];
+    return (0, js_1.insertChange)(host, source, filePath, position, toInsert);
 }
-exports.addSymbolToNgModuleMetadata = addSymbolToNgModuleMetadata;
-function removeFromNgModule(source, modulePath, property) {
+function _addSymbolToNgModuleMetadata(host, source, ngModulePath, metadataField, expression) {
+    return _addSymbolToDecoratorMetadata(host, source, ngModulePath, metadataField, expression, 'NgModule');
+}
+exports._addSymbolToNgModuleMetadata = _addSymbolToNgModuleMetadata;
+function removeFromNgModule(host, source, modulePath, property) {
     const nodes = getDecoratorMetadata(source, 'NgModule', '@angular/core');
     let node = nodes[0]; // tslint:disable-line:no-any
     // Find the decorator declaration.
     if (!node) {
-        return [];
+        return source;
     }
     // Get all the children property assignment of object literals.
     const matchingProperty = getMatchingProperty(source, property, 'NgModule', '@angular/core');
     if (matchingProperty) {
-        return [
-            new xplat_1.RemoveChange(modulePath, matchingProperty.getStart(source), matchingProperty.getFullText(source)),
-        ];
-    }
-    else {
-        return [];
+        return (0, js_1.removeChange)(host, source, modulePath, matchingProperty.getStart(source), matchingProperty.getFullText(source));
     }
 }
 exports.removeFromNgModule = removeFromNgModule;
-function addImportToModule(source, modulePath, symbolName) {
-    return addSymbolToNgModuleMetadata(source, modulePath, 'imports', symbolName);
+/**
+ * Add an import to a Standalone Component
+ * @param host Virtual Tree
+ * @param source TS Source File containing the Component
+ * @param componentPath The path to the Component
+ * @param symbolName The import to add to the Component
+ */
+function addImportToComponent(host, source, componentPath, symbolName) {
+    return _addSymbolToDecoratorMetadata(host, source, componentPath, 'imports', symbolName, 'Component');
+}
+exports.addImportToComponent = addImportToComponent;
+/**
+ * Add an import to a Standalone Directive
+ * @param host Virtual Tree
+ * @param source TS Source File containing the Directive
+ * @param directivePath The path to the Directive
+ * @param symbolName The import to add to the Directive
+ */
+function addImportToDirective(host, source, directivePath, symbolName) {
+    return _addSymbolToDecoratorMetadata(host, source, directivePath, 'imports', symbolName, 'Directive');
+}
+exports.addImportToDirective = addImportToDirective;
+/**
+ * Add an import to a Standalone Pipe
+ * @param host Virtual Tree
+ * @param source TS Source File containing the Pipe
+ * @param pipePath The path to the Pipe
+ * @param symbolName The import to add to the Pipe
+ */
+function addImportToPipe(host, source, pipePath, symbolName) {
+    return _addSymbolToDecoratorMetadata(host, source, pipePath, 'imports', symbolName, 'Pipe');
+}
+exports.addImportToPipe = addImportToPipe;
+/**
+ * Add an import to an NgModule
+ * @param host Virtual Tree
+ * @param source TS Source File containing the NgModule
+ * @param modulePath The path to the NgModule
+ * @param symbolName The import to add to the NgModule
+ */
+function addImportToModule(host, source, modulePath, symbolName) {
+    return _addSymbolToNgModuleMetadata(host, source, modulePath, 'imports', symbolName);
 }
 exports.addImportToModule = addImportToModule;
-function addImportToTestBed(source, specPath, symbolName) {
-    const allCalls = ((0, typescript_1.findNodes)(source, ts.SyntaxKind.CallExpression));
+function addImportToTestBed(host, source, specPath, symbolName) {
+    if (!tsModule) {
+        tsModule = (0, ensure_typescript_1.ensureTypescript)();
+    }
+    const allCalls = ((0, typescript_1.findNodes)(source, tsModule.SyntaxKind.CallExpression));
     const configureTestingModuleObjectLiterals = allCalls
-        .filter((c) => c.expression.kind === ts.SyntaxKind.PropertyAccessExpression)
+        .filter((c) => c.expression.kind === tsModule.SyntaxKind.PropertyAccessExpression)
         .filter((c) => c.expression.name.getText(source) === 'configureTestingModule')
-        .map((c) => c.arguments[0].kind === ts.SyntaxKind.ObjectLiteralExpression
+        .map((c) => c.arguments[0].kind === tsModule.SyntaxKind.ObjectLiteralExpression
         ? c.arguments[0]
         : null);
     if (configureTestingModuleObjectLiterals.length > 0) {
         const startPosition = configureTestingModuleObjectLiterals[0]
             .getFirstToken(source)
             .getEnd();
-        return [
-            new xplat_1.InsertChange(specPath, startPosition, `imports: [${symbolName}], `),
-        ];
-    }
-    else {
-        return [];
+        return (0, js_1.insertChange)(host, source, specPath, startPosition, `imports: [${symbolName}], `);
     }
+    return source;
 }
 exports.addImportToTestBed = addImportToTestBed;
+function addDeclarationsToTestBed(host, source, specPath, symbolName) {
+    if (!tsModule) {
+        tsModule = (0, ensure_typescript_1.ensureTypescript)();
+    }
+    const allCalls = ((0, typescript_1.findNodes)(source, tsModule.SyntaxKind.CallExpression));
+    const configureTestingModuleObjectLiterals = allCalls
+        .filter((c) => c.expression.kind === tsModule.SyntaxKind.PropertyAccessExpression)
+        .filter((c) => c.expression.name.getText(source) === 'configureTestingModule')
+        .map((c) => c.arguments[0].kind === tsModule.SyntaxKind.ObjectLiteralExpression
+        ? c.arguments[0]
+        : null);
+    if (configureTestingModuleObjectLiterals.length > 0) {
+        const startPosition = configureTestingModuleObjectLiterals[0]
+            .getFirstToken(source)
+            .getEnd();
+        return (0, js_1.insertChange)(host, source, specPath, startPosition, `declarations: [${symbolName.join(',')}], `);
+    }
+    return source;
+}
+exports.addDeclarationsToTestBed = addDeclarationsToTestBed;
+function replaceIntoToTestBed(host, source, specPath, newSymbol, previousSymbol) {
+    if (!tsModule) {
+        tsModule = (0, ensure_typescript_1.ensureTypescript)();
+    }
+    const allCalls = ((0, typescript_1.findNodes)(source, tsModule.SyntaxKind.CallExpression));
+    const configureTestingModuleObjectLiterals = allCalls
+        .filter((c) => c.expression.kind === tsModule.SyntaxKind.PropertyAccessExpression)
+        .filter((c) => c.expression.name.getText(source) === 'configureTestingModule')
+        .map((c) => c.arguments[0].kind === tsModule.SyntaxKind.ObjectLiteralExpression
+        ? c.arguments[0]
+        : null);
+    if (configureTestingModuleObjectLiterals.length > 0) {
+        const startPosition = configureTestingModuleObjectLiterals[0]
+            .getFirstToken(source)
+            .getEnd();
+        return (0, js_1.replaceChange)(host, source, specPath, startPosition, newSymbol, previousSymbol);
+    }
+    return source;
+}
+exports.replaceIntoToTestBed = replaceIntoToTestBed;
 function getBootstrapComponent(source, moduleClassName) {
     const bootstrap = getMatchingProperty(source, 'bootstrap', 'NgModule', '@angular/core');
     if (!bootstrap) {
@@ -301,56 +399,113 @@ function getMatchingProperty(source, property, identifier, module) {
     // Get all the children property assignment of object literals.
     return getMatchingObjectLiteralElement(node, source, property);
 }
-function addRoute(ngModulePath, source, route) {
+function addRouteToNgModule(host, ngModulePath, source, route) {
     const routes = getListOfRoutes(source);
     if (!routes)
-        return [];
+        return source;
     if (routes.hasTrailingComma || routes.length === 0) {
-        return [new xplat_1.InsertChange(ngModulePath, routes.end, route)];
+        return (0, js_1.insertChange)(host, source, ngModulePath, routes.end, route);
     }
     else {
-        return [new xplat_1.InsertChange(ngModulePath, routes.end, `, ${route}`)];
+        return (0, js_1.insertChange)(host, source, ngModulePath, routes.end, `, ${route}`);
     }
 }
-exports.addRoute = addRoute;
+exports.addRouteToNgModule = addRouteToNgModule;
 function getListOfRoutes(source) {
+    if (!tsModule) {
+        tsModule = (0, ensure_typescript_1.ensureTypescript)();
+    }
     const imports = getMatchingProperty(source, 'imports', 'NgModule', '@angular/core');
-    if (imports.initializer.kind === ts.SyntaxKind.ArrayLiteralExpression) {
+    if ((imports === null || imports === void 0 ? void 0 : imports.initializer.kind) === tsModule.SyntaxKind.ArrayLiteralExpression) {
         const a = imports.initializer;
-        for (let e of a.elements) {
-            if (e.kind === ts.SyntaxKind.CallExpression) {
+        for (const e of a.elements) {
+            if (e.kind === tsModule.SyntaxKind.CallExpression) {
                 const ee = e;
                 const text = ee.expression.getText(source);
                 if ((text === 'RouterModule.forRoot' ||
                     text === 'RouterModule.forChild') &&
                     ee.arguments.length > 0) {
                     const routes = ee.arguments[0];
-                    if (routes.kind === ts.SyntaxKind.ArrayLiteralExpression) {
+                    if (routes.kind === tsModule.SyntaxKind.ArrayLiteralExpression) {
                         return routes.elements;
                     }
+                    else if (routes.kind === tsModule.SyntaxKind.Identifier) {
+                        // find the array expression
+                        const variableDeclarations = (0, typescript_1.findNodes)(source, tsModule.SyntaxKind.VariableDeclaration);
+                        const routesDeclaration = variableDeclarations.find((x) => {
+                            return x.name.getText() === routes.getText();
+                        });
+                        if (routesDeclaration) {
+                            return routesDeclaration.initializer.elements;
+                        }
+                    }
                 }
             }
         }
     }
     return null;
 }
-function addProviderToModule(source, modulePath, symbolName) {
-    return addSymbolToNgModuleMetadata(source, modulePath, 'providers', symbolName);
+/**
+ * Add a provider to bootstrapApplication call for Standalone Applications
+ * @param tree Virtual Tree
+ * @param filePath Path to the file containing the bootstrapApplication call
+ * @param providerToAdd Provider to add
+ */
+function addProviderToBootstrapApplication(tree, filePath, providerToAdd) {
+    (0, ensure_typescript_1.ensureTypescript)();
+    const { tsquery } = require('@phenomnomnominal/tsquery');
+    const PROVIDERS_ARRAY_SELECTOR = 'CallExpression:has(Identifier[name=bootstrapApplication]) ObjectLiteralExpression > PropertyAssignment:has(Identifier[name=providers]) > ArrayLiteralExpression';
+    const fileContents = tree.read(filePath, 'utf-8');
+    const ast = tsquery.ast(fileContents);
+    const providersArrayNodes = tsquery(ast, PROVIDERS_ARRAY_SELECTOR, {
+        visitAllChildren: true,
+    });
+    if (providersArrayNodes.length === 0) {
+        throw new Error(`Providers does not exist in the bootstrapApplication call within ${filePath}.`);
+    }
+    const arrayNode = providersArrayNodes[0];
+    const newFileContents = `${fileContents.slice(0, arrayNode.getStart() + 1)}${providerToAdd},${fileContents.slice(arrayNode.getStart() + 1, fileContents.length)}`;
+    tree.write(filePath, newFileContents);
+}
+exports.addProviderToBootstrapApplication = addProviderToBootstrapApplication;
+/**
+ * Add a provider to an NgModule
+ * @param host Virtual Tree
+ * @param source TS Source File containing the NgModule
+ * @param modulePath Path to the NgModule
+ * @param symbolName The provider to add
+ */
+function addProviderToModule(host, source, modulePath, symbolName) {
+    return _addSymbolToNgModuleMetadata(host, source, modulePath, 'providers', symbolName);
 }
 exports.addProviderToModule = addProviderToModule;
-function addDeclarationToModule(source, modulePath, symbolName) {
-    return addSymbolToNgModuleMetadata(source, modulePath, 'declarations', symbolName);
+/**
+ * Add a provider to a Standalone Component
+ * @param host Virtual Tree
+ * @param source TS Source File containing the Component
+ * @param componentPath Path to the Component
+ * @param symbolName The provider to add
+ */
+function addProviderToComponent(host, source, componentPath, symbolName) {
+    return _addSymbolToDecoratorMetadata(host, source, componentPath, 'providers', symbolName, 'Component');
+}
+exports.addProviderToComponent = addProviderToComponent;
+function addDeclarationToModule(host, source, modulePath, symbolName) {
+    return _addSymbolToNgModuleMetadata(host, source, modulePath, 'declarations', symbolName);
 }
 exports.addDeclarationToModule = addDeclarationToModule;
-function addEntryComponents(source, modulePath, symbolName) {
-    return addSymbolToNgModuleMetadata(source, modulePath, 'entryComponents', symbolName);
+function addEntryComponents(host, source, modulePath, symbolName) {
+    return _addSymbolToNgModuleMetadata(host, source, modulePath, 'entryComponents', symbolName);
 }
 exports.addEntryComponents = addEntryComponents;
 function readBootstrapInfo(host, app) {
-    const config = (0, xplat_1.getProjectConfig)(host, app);
+    if (!tsModule) {
+        tsModule = (0, ensure_typescript_1.ensureTypescript)();
+    }
+    const config = (0, devkit_1.readProjectConfiguration)(host, app);
     let mainPath;
     try {
-        mainPath = config.architect.build.options.main;
+        mainPath = config.targets.build.options.main;
     }
     catch (e) {
         throw new Error('Main file cannot be located');
@@ -359,21 +514,21 @@ function readBootstrapInfo(host, app) {
         throw new Error('Main file cannot be located');
     }
     const mainSource = host.read(mainPath).toString('utf-8');
-    const main = ts.createSourceFile(mainPath, mainSource, ts.ScriptTarget.Latest, true);
-    const moduleImports = (0, xplat_1.getImport)(main, (s) => s.indexOf('.module') > -1);
+    const main = tsModule.createSourceFile(mainPath, mainSource, tsModule.ScriptTarget.Latest, true);
+    const moduleImports = (0, js_1.getImport)(main, (s) => s.indexOf('.module') > -1);
     if (moduleImports.length !== 1) {
         throw new Error(`main.ts can only import a single module`);
     }
     const moduleImport = moduleImports[0];
     const moduleClassName = moduleImport.bindings.filter((b) => b.endsWith('Module'))[0];
-    const modulePath = `${path.join(path.dirname(mainPath), moduleImport.moduleSpec)}.ts`;
+    const modulePath = `${(0, path_1.join)((0, path_1.dirname)(mainPath), moduleImport.moduleSpec)}.ts`;
     if (!host.exists(modulePath)) {
         throw new Error(`Cannot find '${modulePath}'`);
     }
     const moduleSourceText = host.read(modulePath).toString('utf-8');
-    const moduleSource = ts.createSourceFile(modulePath, moduleSourceText, ts.ScriptTarget.Latest, true);
+    const moduleSource = tsModule.createSourceFile(modulePath, moduleSourceText, tsModule.ScriptTarget.Latest, true);
     const bootstrapComponentClassName = getBootstrapComponent(moduleSource, moduleClassName);
-    const bootstrapComponentFileName = `./${path.join(path.dirname(moduleImport.moduleSpec), `${(0, xplat_utils_1.toFileName)(bootstrapComponentClassName.substring(0, bootstrapComponentClassName.length - 9))}.component`)}`;
+    const bootstrapComponentFileName = `./${(0, path_1.join)((0, path_1.dirname)(moduleImport.moduleSpec), `${(0, devkit_1.names)(bootstrapComponentClassName.substring(0, bootstrapComponentClassName.length - 9)).fileName}.component`)}`;
     return {
         moduleSpec: moduleImport.moduleSpec,
         mainPath,
@@ -386,25 +541,44 @@ function readBootstrapInfo(host, app) {
 }
 exports.readBootstrapInfo = readBootstrapInfo;
 function getDecoratorPropertyValueNode(host, modulePath, identifier, property, module) {
+    if (!tsModule) {
+        tsModule = (0, ensure_typescript_1.ensureTypescript)();
+    }
     const moduleSourceText = host.read(modulePath).toString('utf-8');
-    const moduleSource = ts.createSourceFile(modulePath, moduleSourceText, ts.ScriptTarget.Latest, true);
+    const moduleSource = tsModule.createSourceFile(modulePath, moduleSourceText, tsModule.ScriptTarget.Latest, true);
     const templateNode = getMatchingProperty(moduleSource, property, identifier, module);
     return templateNode.getChildAt(templateNode.getChildCount() - 1);
 }
 exports.getDecoratorPropertyValueNode = getDecoratorPropertyValueNode;
 function getMatchingObjectLiteralElement(node, source, property) {
+    if (!tsModule) {
+        tsModule = (0, ensure_typescript_1.ensureTypescript)();
+    }
     return (node.properties
-        .filter((prop) => prop.kind == ts.SyntaxKind.PropertyAssignment)
+        .filter((prop) => prop.kind == tsModule.SyntaxKind.PropertyAssignment)
         // Filter out every fields that's not "metadataField". Also handles string literals
         // (but not expressions).
         .filter((prop) => {
         const name = prop.name;
         switch (name.kind) {
-            case ts.SyntaxKind.Identifier:
+            case tsModule.SyntaxKind.Identifier:
                 return name.getText(source) === property;
-            case ts.SyntaxKind.StringLiteral:
+            case tsModule.SyntaxKind.StringLiteral:
                 return name.text === property;
         }
         return false;
     })[0]);
 }
+function getTsSourceFile(host, path) {
+    if (!tsModule) {
+        tsModule = (0, ensure_typescript_1.ensureTypescript)();
+    }
+    const buffer = host.read(path);
+    if (!buffer) {
+        throw new Error(`Could not read TS file (${path}).`);
+    }
+    const content = buffer.toString();
+    const source = tsModule.createSourceFile(path, content, tsModule.ScriptTarget.Latest, true);
+    return source;
+}
+exports.getTsSourceFile = getTsSourceFile;