fix(schematics): exclude environment imports for libraries (#1213)

Closes #1205, #1197

Author: brandonroberts

modules/effects/schematics-core/index.ts

@@ -48,7 +48,7 @@ export {
   omit,
 } from './utility/ngrx-utils';
 
-export { getProjectPath } from './utility/project';
+export { getProjectPath, getProject, isLib } from './utility/project';
 export { insertImport } from './utility/route-utils';
 
 export const stringUtils = {

modules/effects/schematics-core/utility/ast-utils.ts

@@ -352,6 +352,8 @@ function _addSymbolToNgModuleMetadata(
       return [];
     }
 
+    node = node[node.length - 1];
+
     const effectsModule = nodeArray.find(
       node =>
         (node.getText().includes('EffectsModule.forRoot') &&
@@ -398,8 +400,6 @@ function _addSymbolToNgModuleMetadata(
     }
   }
 
-  node = node[node.length - 1];
-
   let toInsert: string;
   let position = node.getEnd();
   if (node.kind == ts.SyntaxKind.ObjectLiteralExpression) {

modules/effects/schematics-core/utility/project.ts

@@ -1,17 +1,29 @@
 import { getWorkspace } from './config';
 import { Tree } from '@angular-devkit/schematics';
 
-export function getProjectPath(
+export interface WorkspaceProject {
+  root: string;
+  projectType: string;
+}
+
+export function getProject(
   host: Tree,
   options: { project?: string | undefined; path?: string | undefined }
-) {
+): WorkspaceProject {
   const workspace = getWorkspace(host);
 
   if (!options.project) {
     options.project = Object.keys(workspace.projects)[0];
   }
 
-  const project = workspace.projects[options.project];
+  return workspace.projects[options.project];
+}
+
+export function getProjectPath(
+  host: Tree,
+  options: { project?: string | undefined; path?: string | undefined }
+) {
+  const project = getProject(host, options);
 
   if (project.root.substr(-1) === '/') {
     project.root = project.root.substr(0, project.root.length - 1);
@@ -26,3 +38,12 @@ export function getProjectPath(
 
   return options.path;
 }
+
+export function isLib(
+  host: Tree,
+  options: { project?: string | undefined; path?: string | undefined }
+) {
+  const project = getProject(host, options);
+
+  return project.projectType === 'library';
+}

modules/entity/schematics-core/index.ts

@@ -48,7 +48,7 @@ export {
   omit,
 } from './utility/ngrx-utils';
 
-export { getProjectPath } from './utility/project';
+export { getProjectPath, getProject, isLib } from './utility/project';
 export { insertImport } from './utility/route-utils';
 
 export const stringUtils = {

modules/entity/schematics-core/utility/ast-utils.ts

@@ -352,6 +352,8 @@ function _addSymbolToNgModuleMetadata(
       return [];
     }
 
+    node = node[node.length - 1];
+
     const effectsModule = nodeArray.find(
       node =>
         (node.getText().includes('EffectsModule.forRoot') &&
@@ -398,8 +400,6 @@ function _addSymbolToNgModuleMetadata(
     }
   }
 
-  node = node[node.length - 1];
-
   let toInsert: string;
   let position = node.getEnd();
   if (node.kind == ts.SyntaxKind.ObjectLiteralExpression) {

modules/entity/schematics-core/utility/project.ts

@@ -1,17 +1,29 @@
 import { getWorkspace } from './config';
 import { Tree } from '@angular-devkit/schematics';
 
-export function getProjectPath(
+export interface WorkspaceProject {
+  root: string;
+  projectType: string;
+}
+
+export function getProject(
   host: Tree,
   options: { project?: string | undefined; path?: string | undefined }
-) {
+): WorkspaceProject {
   const workspace = getWorkspace(host);
 
   if (!options.project) {
     options.project = Object.keys(workspace.projects)[0];
   }
 
-  const project = workspace.projects[options.project];
+  return workspace.projects[options.project];
+}
+
+export function getProjectPath(
+  host: Tree,
+  options: { project?: string | undefined; path?: string | undefined }
+) {
+  const project = getProject(host, options);
 
   if (project.root.substr(-1) === '/') {
     project.root = project.root.substr(0, project.root.length - 1);
@@ -26,3 +38,12 @@ export function getProjectPath(
 
   return options.path;
 }
+
+export function isLib(
+  host: Tree,
+  options: { project?: string | undefined; path?: string | undefined }
+) {
+  const project = getProject(host, options);
+
+  return project.projectType === 'library';
+}

modules/router-store/schematics-core/index.ts

@@ -48,7 +48,7 @@ export {
   omit,
 } from './utility/ngrx-utils';
 
-export { getProjectPath } from './utility/project';
+export { getProjectPath, getProject, isLib } from './utility/project';
 export { insertImport } from './utility/route-utils';
 
 export const stringUtils = {

modules/router-store/schematics-core/utility/ast-utils.ts

@@ -352,6 +352,8 @@ function _addSymbolToNgModuleMetadata(
       return [];
     }
 
+    node = node[node.length - 1];
+
     const effectsModule = nodeArray.find(
       node =>
         (node.getText().includes('EffectsModule.forRoot') &&
@@ -398,8 +400,6 @@ function _addSymbolToNgModuleMetadata(
     }
   }
 
-  node = node[node.length - 1];
-
   let toInsert: string;
   let position = node.getEnd();
   if (node.kind == ts.SyntaxKind.ObjectLiteralExpression) {

modules/router-store/schematics-core/utility/project.ts

@@ -1,17 +1,29 @@
 import { getWorkspace } from './config';
 import { Tree } from '@angular-devkit/schematics';
 
-export function getProjectPath(
+export interface WorkspaceProject {
+  root: string;
+  projectType: string;
+}
+
+export function getProject(
   host: Tree,
   options: { project?: string | undefined; path?: string | undefined }
-) {
+): WorkspaceProject {
   const workspace = getWorkspace(host);
 
   if (!options.project) {
     options.project = Object.keys(workspace.projects)[0];
   }
 
-  const project = workspace.projects[options.project];
+  return workspace.projects[options.project];
+}
+
+export function getProjectPath(
+  host: Tree,
+  options: { project?: string | undefined; path?: string | undefined }
+) {
+  const project = getProject(host, options);
 
   if (project.root.substr(-1) === '/') {
     project.root = project.root.substr(0, project.root.length - 1);
@@ -26,3 +38,12 @@ export function getProjectPath(
 
   return options.path;
 }
+
+export function isLib(
+  host: Tree,
+  options: { project?: string | undefined; path?: string | undefined }
+) {
+  const project = getProject(host, options);
+
+  return project.projectType === 'library';
+}

modules/schematics-core/index.ts

@@ -48,7 +48,7 @@ export {
   omit,
 } from './utility/ngrx-utils';
 
-export { getProjectPath } from './utility/project';
+export { getProjectPath, getProject, isLib } from './utility/project';
 export { insertImport } from './utility/route-utils';
 
 export const stringUtils = {