import { NgModule, ApplicationRef } from '@angular/core';
import { BrowserModule } from '@angular/platform-browser';
import { FormsModule } from '@angular/forms';
import { removeNgStyles, createNewHosts } from '@angularclass/hmr';
import { Angular2DataTableModule } from 'angular2-data-table';

import { App } from './app.component';
import { SWUIModule } from '../../src/index';

@NgModule({
  declarations: [App],
  imports: [BrowserModule, SWUIModule, FormsModule, Angular2DataTableModule],
  bootstrap: [App]
})
export class AppModule {

  constructor(private applicationRef: ApplicationRef) {
  }

  hmrOnDestroy(store) {
    let cmpLocation = this.applicationRef.components.map(cmp => cmp.location.nativeElement);

    // recreate elements
    store.disposeOldHosts = createNewHosts(cmpLocation);

    // inject your AppStore and grab state then set it on store
    // let appState = this.AppStore.get()
    // Object.assign(store, appState)

    // remove styles
    removeNgStyles();
  }

  hmrAfterDestroy(store) {
    // display new elements
    // anything you need done the component is removed
    store.disposeOldHosts();
    delete store.disposeOldHosts;
  }
}