fix(schematics): fix component option

valor-software · Dec 5, 2018 · cba4b35 · cba4b35
1 parent 3a17cc3
commit cba4b35
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/schematics/src/ng-add/index.ts b/schematics/src/ng-add/index.ts
@@ -30,10 +30,14 @@ const datePickerStylePath =  `./node_modules/ngx-bootstrap/datepicker/bs-datepic
 /* tslint:disable-next-line: no-default-export */
 export default function (options: Schema): Rule {
   return chain([
-    addStyles(options),
     addPackageJsonDependencies(),
     installPackageJsonDependencies(),
-    options.component ? addModuleOfComponent(options.project, options.component) : noop()
+    options.component
+      ? addStyles(options, _insertBootstrapStyle)
+      : addStyles(options, _insertCommonStyles),
+    options.component
+      ? addModuleOfComponent(options.project, options.component)
+      : noop()
   ]);
 }
 
@@ -97,7 +101,7 @@ function addPackageJsonDependencies(): Rule {
   };
 }
 
-export function addStyles(options: Schema): (host: Tree) => Tree {
+export function addStyles(options: Schema, insertStyle: Function): (host: Tree) => Tree {
   return function (host: Tree): Tree {
     const workspace = getWorkspace(host);
     const project = getProjectFromWorkspace(workspace, options.project);
@@ -108,7 +112,12 @@ export function addStyles(options: Schema): (host: Tree) => Tree {
   };
 }
 
-function insertStyle(project: WorkspaceProject, host: Tree, workspace: WorkspaceSchema) {
+function _insertBootstrapStyle(project: WorkspaceProject, host: Tree, workspace: WorkspaceSchema) {
+  addStyleToTarget(project, 'build', host, bootstrapStylePath, workspace);
+  addStyleToTarget(project, 'test', host, bootstrapStylePath, workspace);
+}
+
+function _insertCommonStyles(project: WorkspaceProject, host: Tree, workspace: WorkspaceSchema) {
   addStyleToTarget(project, 'build', host, datePickerStylePath, workspace);
   addStyleToTarget(project, 'test', host, datePickerStylePath, workspace);
   addStyleToTarget(project, 'build', host, bootstrapStylePath, workspace);