✨ add ability to add needed ngModule imports

src/boostrap.ts

@@ -12,7 +12,7 @@ export function bootstrap(sandBox: UIGuideSandbox): Promise<NgModuleRef<UIGuideM
   /** grab all the ui guides */
   const uiGuides = sandBox.loadUIGuides()
   /** resolve all the ui guide components and ng module */
-  const resolved = getModuleForUIGuides(sandBox.ngModule, uiGuides)
+  const resolved = getModuleForUIGuides(sandBox.ngModule, uiGuides, sandBox.ngModuleImports)
   /** this is the markdown used in your index page */
   const entryMarkdown = sandBox.entryMarkdown
 

src/generate-hosts.ts

@@ -3,8 +3,9 @@ import { UIGuide, ResolvedUIGuideSandbox, UIGuideExample } from './interfaces'
 const flatten = require('lodash.flatten')
 
 export function getModuleForUIGuides(
-  givenModule: ModuleWithProviders | Type<any>,
-  uiGuides: UIGuide[]
+  givenModule: Type<any>,
+  uiGuides: UIGuide[],
+  ngModuleImports: Type<any>[] = []
 ): ResolvedUIGuideSandbox {
 
   const componentsWithIds: {id: string, component: Type<any>}[] = flatten(uiGuides.map((uiGuide => uiGuide.examples.map((ex => {
@@ -22,7 +23,8 @@ export function getModuleForUIGuides(
     })
   }, {})
 
-  const ngModule = generateNgModule(givenModule, componentsWithIds.map(e => e.component))
+  ngModuleImports.push(givenModule)
+  const ngModule = generateNgModule(ngModuleImports, componentsWithIds.map(e => e.component))
   return {ngModule, components}
 }
 
@@ -32,7 +34,7 @@ export function generateComponent(
   @Component({
     template: example.template,
     styles: example.styles,
-    providers: [...example.providers]
+    providers: example.providers || []
   })
   class UIGuideExampleComponent {
     constructor() {
@@ -44,13 +46,11 @@ export function generateComponent(
 }
 
 export function generateNgModule(
-  givenModule: ModuleWithProviders | Type<any>,
+  givenModules: Type<any>[],
   components: Type<any>[]
 ): Type<any> {
   @NgModule({
-    imports: [
-      givenModule
-    ],
+    imports: givenModules,
     declarations: [
       ...components
     ],

src/interfaces.ts

@@ -37,9 +37,9 @@ export interface CompiledUIGuide {
 }
 
 export interface UIGuideSandbox {
-  module?: NodeModule
   entryMarkdown: string
-  ngModule: Type<any> | ModuleWithProviders
+  ngModule: Type<any>
+  ngModuleImports?: Type<any>[]
   loadUIGuides(): UIGuide[]
 };