fix(StateUpdates): Generate metadata for StateUpdates class to make it easy to provide

Author: MikeRyanDev

lib/ng2.ts

@@ -8,36 +8,33 @@ import {Store, Dispatcher} from "@ngrx/store";
 export const BOOTSTRAP_EFFECTS = new String('@ngrx/effects Bootstrap Effects');
 
 export function runEffects(...effects: any[]) {
-    const individuals = flatten(effects);
+  const individuals = flatten(effects);
 
-    const allEffects = individuals.map(effectClass => createDynamicProvider(effectClass));
-    return [
-        ...individuals,
-        ...allEffects,
-        CONNECT_EFFECTS_PROVIDER,
-        STATE_UPDATES_PROVIDER
-    ];
+  const allEffects = individuals.map(effectClass => createDynamicProvider(effectClass));
+  return [
+    ...individuals,
+    ...allEffects,
+    CONNECT_EFFECTS_PROVIDER,
+    STATE_UPDATES_PROVIDER
+  ];
 }
 
 export const CONNECT_EFFECTS_PROVIDER = {
-    provide: APP_INITIALIZER,
-    multi: true,
-    deps: [Store, BOOTSTRAP_EFFECTS],
-    useFactory: connectEffectsToStore
+  provide: APP_INITIALIZER,
+  multi: true,
+  deps: [Store, BOOTSTRAP_EFFECTS],
+  useFactory: connectEffectsToStore
 };
 
 export const STATE_UPDATES_PROVIDER = {
-    provide: StateUpdates,
-    deps: [Dispatcher, Store],
-    useFactory(dispatcher: Dispatcher, store: Store<any>) {
-        return new StateUpdates(dispatcher, store);
-    }
+  provide: StateUpdates,
+  useClass: StateUpdates
 };
 
 function createDynamicProvider(effectClass: any) {
-    return {
-        provide: BOOTSTRAP_EFFECTS,
-        useExisting: effectClass,
-        multi: true
-    }
-}
+  return {
+    provide: BOOTSTRAP_EFFECTS,
+    useExisting: effectClass,
+    multi: true
+  }
+}

lib/state-updates.ts

@@ -3,12 +3,14 @@ import { filter } from 'rxjs/operator/filter';
 import { Observable } from 'rxjs/Observable';
 import { ReplaySubject } from 'rxjs/ReplaySubject';
 import { Action, Dispatcher, State } from '@ngrx/store';
+import { Injectable } from '@angular/core';
 
 export interface StateUpdate<S> {
   state: S;
   action: Action;
 }
 
+@Injectable()
 export class StateUpdates<S> extends ReplaySubject<StateUpdate<S>> {
   constructor(actions$: Dispatcher, state$: State<S>) {
     super(1);
@@ -29,4 +31,4 @@ export class StateUpdates<S> extends ReplaySubject<StateUpdate<S>> {
       ({ action }: StateUpdate<S>) => actionTypes.indexOf(action.type) > -1
     );
   }
-}
+}