From 2ba910aa9ae21478b51c1fef9cafd54d3b3ccd56 Mon Sep 17 00:00:00 2001
From: Chai Wei Jian <chaiweijian0911@gmail.com>
Date: Sat, 18 Nov 2017 11:03:35 +0800
Subject: [PATCH] fix(schematics): import StoreRouterConnectingModule

Closes #108
---
 packages/schematics/src/ngrx/index.ts | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/packages/schematics/src/ngrx/index.ts b/packages/schematics/src/ngrx/index.ts
index 19ea3fdc77..bf989240ec 100644
--- a/packages/schematics/src/ngrx/index.ts
+++ b/packages/schematics/src/ngrx/index.ts
@@ -42,9 +42,11 @@ function addImportsToModule(name: string, options: Schema): Rule {
         insertImport(source, modulePath, 'EffectsModule', '@ngrx/effects'),
         insertImport(source, modulePath, 'StoreDevtoolsModule', '@ngrx/store-devtools'),
         insertImport(source, modulePath, 'environment', '../environments/environment'),
+        insertImport(source, modulePath, 'StoreRouterConnectingModule', '@ngrx/router-store'),
         ...addImportToModule(source, modulePath, `StoreModule.forRoot({})`),
         ...addImportToModule(source, modulePath, `EffectsModule.forRoot([])`),
-        ...addImportToModule(source, modulePath, `!environment.production ? StoreDevtoolsModule.instrument() : []`)
+        ...addImportToModule(source, modulePath, `!environment.production ? StoreDevtoolsModule.instrument() : []`),
+        ...addImportToModule(source, modulePath, `StoreRouterConnectingModule`)
       ]);
       return host;
     } else {
@@ -70,6 +72,7 @@ function addImportsToModule(name: string, options: Schema): Rule {
           ...common,
           insertImport(source, modulePath, 'StoreDevtoolsModule', '@ngrx/store-devtools'),
           insertImport(source, modulePath, 'environment', '../environments/environment'),
+          insertImport(source, modulePath, 'StoreRouterConnectingModule', '@ngrx/router-store'),
           ...addImportToModule(
             source,
             modulePath,
@@ -78,7 +81,8 @@ function addImportsToModule(name: string, options: Schema): Rule {
             )}: ${initName}}})`
           ),
           ...addImportToModule(source, modulePath, `EffectsModule.forRoot([${effectsName}])`),
-          ...addImportToModule(source, modulePath, `!environment.production ? StoreDevtoolsModule.instrument() : []`)
+          ...addImportToModule(source, modulePath, `!environment.production ? StoreDevtoolsModule.instrument() : []`),
+          ...addImportToModule(source, modulePath, `StoreRouterConnectingModule`)
         ]);
       } else {
         insert(host, modulePath, [