feat(onEnter/Exit/Retain): Use onExit/onEnter/onRetain from 56955stat…

…e(), not state.self
ui-router · Feb 26, 2017 · bc1f554 · bc1f554
1 parent 6aefc06
commit bc1f554
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/src/hooks/onEnterExitRetain.ts b/src/hooks/onEnterExitRetain.ts
@@ -1,8 +1,10 @@
-/** @module hooks */ /** for typedoc */
-import {TransitionStateHookFn} from "../transition/interface";
-import {Transition} from "../transition/transition";
-import {TransitionService} from "../transition/transitionService";
+/** @module hooks */
+/** for typedoc */
+import { TransitionStateHookFn } from '../transition/interface';
+import { Transition } from '../transition/transition';
+import { TransitionService } from '../transition/transitionService';
 import { StateDeclaration } from '../state/interface';
+import { StateObject } from '../state/stateObject';
 
 /**
  * A factory which creates an onEnter, onExit or onRetain transition hook function
@@ -14,7 +16,8 @@ import { StateDeclaration } from '../state/interface';
  */
 function makeEnterExitRetainHook(hookName: string): TransitionStateHookFn {
   return (transition: Transition, state: StateDeclaration) => {
-    let hookFn: TransitionStateHookFn = state[hookName];
+    let _state: StateObject = state.$$state();
+    let hookFn: TransitionStateHookFn = _state[hookName];
     return hookFn(transition, state);
   }
 }