fix(routing): Add logic to add missing import statements to the routing file

Author: ajivanyandev

packages/devextreme-schematics/src/add-layout/index.ts

@@ -67,6 +67,25 @@ import { Change } from '@schematics/angular/utility/change';
 
 import { PatchNodePackageInstallTask } from '../utility/patch';
 
+const routes = [
+  {
+    name: 'LoginFormComponent',
+    path: 'login-form'
+  },
+  {
+    name: 'ResetPasswordFormComponent',
+    path: 'reset-password'
+  },
+  {
+    name: 'CreateAccountFormComponent',
+    path: 'create-account'
+  },
+  {
+    name: 'ChangePasswordFormComponent',
+    path: 'change-password'
+  }
+];
+
 const projectFilesSource = './files/src';
 const workspaceFilesSource = './files';
 
@@ -291,15 +310,23 @@ function updateDevextremeConfig(sourcePath: string = '') {
 const modifyRouting = (host: Tree, routingFilePath: string) => {
   // TODO: Try to use the isolated host to generate the result string
   let source = getSourceFile(host, routingFilePath)!;
-  const importChange = insertImport(source, routingFilePath, 'LoginFormComponent', './shared/components');
-  applyChanges(host, [ importChange ], routingFilePath);
-
-  source = getSourceFile(host, routingFilePath)!;
-  const routes = findRoutesInSource(source)!;
-  if (!hasComponentInRoutes(routes, 'login-form')) {
-    const loginFormRoute = getRoute('login-form');
-    insertItemToArray(host, routingFilePath, routes, loginFormRoute);
+  const importChanges = [];
+  importChanges.push(insertImport(source, routingFilePath, 'AuthGuardService', './shared/services'));
+
+  for (const route of routes) {
+    importChanges.push(insertImport(source, routingFilePath, route.name, './shared/components'));
   }
+
+  applyChanges(host, importChanges, routingFilePath);
+  for (const route of routes) {
+    source = getSourceFile(host, routingFilePath)!;
+    const routeInSource = findRoutesInSource(source)!;
+    if (!hasComponentInRoutes(routeInSource, route.path)) {
+      const routeToAdd = getRoute(route.path, route.name);
+      insertItemToArray(host, routingFilePath, routeInSource, routeToAdd);
+    }
+  }
+
 };
 
 export default function(options: any): Rule {

packages/devextreme-schematics/src/utility/routing.ts

@@ -20,10 +20,10 @@ export function hasComponentInRoutes(routes: Node, name: string) {
   return routesText.indexOf(componentName) !== -1;
 }
 
-export function getRoute(name: string) {
+export function getRoute(name: string, componentName?: string) {
   return `  {
     path: '${strings.dasherize(name)}',
-    component: ${getRouteComponentName(name)},
+    component: ${componentName || getRouteComponentName(name)},
     canActivate: [ AuthGuardService ]
   }`;
 }