feat(reducer): allow routes after action creators

Author: colinmeinke

src/helpers/getAction.js

@@ -3,39 +3,60 @@ import getRoute from '../helpers/getRoute';
 import updateUrl from '../actions/updateUrl';
 import { CHANGE_PAGE_TO } from '../constants';
 
-const getActionList = ( url, actionCreators, data ) => {
+const getActions = ( url, actionCreators, data ) => {
   const actions = [
     updateUrl( url ),
   ];
 
   const keys = Object.keys( actionCreators );
 
   keys.forEach( k => {
-    if ( Array.isArray( actionCreators[ k ])) {
-      actionCreators[ k ].forEach( actionCreator => {
+    if ( k !== 'after' ) {
+      if ( Array.isArray( actionCreators[ k ])) {
+        actionCreators[ k ].forEach( actionCreator => {
+          actions.push( actionCreator( data[ k ]));
+        });
+      } else {
+        const actionCreator = actionCreators[ k ];
         actions.push( actionCreator( data[ k ]));
-      });
-    } else {
-      const actionCreator = actionCreators[ k ];
-      actions.push( actionCreator( data[ k ]));
+      }
     }
   });
 
   return actions;
 };
 
+const getAfter = actionCreators => {
+  const after = [];
+
+  if ( actionCreators.after ) {
+    if ( Array.isArray( actionCreators.after )) {
+      actionCreators.after.forEach( actionCreator => {
+        after.push( actionCreator );
+      });
+    } else {
+      after.push( actionCreators.after );
+    }
+  }
+
+  return after;
+};
+
 const getAction = ( to, routes ) => {
   const { query, url } = getLocation( to );
   const { actionCreators, params } = getRoute( url, routes );
 
-  const actions = getActionList(
+  const actions = getActions(
     url,
     actionCreators,
     { ...query, ...params }
   );
 
+  const after = getAfter( actionCreators );
+
   return {
     actions,
+    after,
     type: CHANGE_PAGE_TO,
     url,
   };

src/reducers/router.js

@@ -6,10 +6,21 @@ import { CHANGE_PAGE_TO } from '../constants';
 const router = reducers => {
   const rootReducer = combineReducers({ ...reducers, url });
 
+  const actionCreatorReducer = ( state, actionCreator ) => {
+    const action = actionCreator( state );
+    return rootReducer( state, action );
+  };
+
   const batchReducer = ( state, action ) => {
     switch ( action.type ) {
       case CHANGE_PAGE_TO:
-        return action.actions.reduce( rootReducer, state );
+        let newState = action.actions.reduce( rootReducer, state );
+
+        if ( action.after.length ) {
+          newState = action.after.reduce( actionCreatorReducer, newState );
+        }
+
+        return newState;
       default:
         return rootReducer( state, action );
     }

test/helpers/getAction.js

@@ -10,17 +10,24 @@ describe( 'helper', () => {
     it( 'should get to url and associated route actions', () => {
       const actionFoo = () => ({ type: 'FOO' });
       const actionBar = () => ({ type: 'BAR' });
+      const actionBaz = () => ({ type: 'BAZ' });
 
       const to = [ 'hello', 'world' ];
 
       const routes = [
-        [ 'hello', 'world', { foo: actionFoo, bar: actionBar }, <p>Hello world</p> ],
+        [
+          'hello',
+          'world',
+          { foo: actionFoo, bar: actionBar, after: [ actionBaz ]},
+          <p>Hello world</p>,
+        ],
       ];
 
       const url = '/hello/world';
 
       const expectedAction = {
         actions: [ updateUrl( url ), actionFoo(), actionBar() ],
+        after: [ actionBaz ],
         type: CHANGE_PAGE_TO,
         url,
       };

test/reducers/router.js

@@ -39,6 +39,7 @@ describe( 'reducer', () => {
 
       expect( reducer({}, {
         actions: [{ type: UPDATE_URL, url: '/hello/world' }, { page: undefined }],
+        after: [],
         to: [ 'hello', 'world' ],
         type: CHANGE_PAGE_TO,
       })).toEqual( expectedState );