Add resolve feature

modules/AsyncProps.js

@@ -29,12 +29,18 @@ function filterAndFlattenComponents(components) {
   return flattened
 }
 
-function loadAsyncProps(components, params, cb) {
+function defaultResolver(Component, params, cb) {
+  Component.loadProps(params, cb)
+}
+
+function loadAsyncProps(components, params, cb, resolver) {
   // flatten the multi-component routes
   let componentsArray = []
   let propsArray = []
   let needToLoadCounter = components.length
 
+  resolver = resolver || defaultResolver
+
   const maybeFinish = () => {
     if (needToLoadCounter === 0)
       cb(null, { propsArray, componentsArray })
@@ -46,7 +52,7 @@ function loadAsyncProps(components, params, cb) {
   }
 
   components.forEach((Component, index) => {
-    Component.loadProps(params, (error, props) => {
+    resolver(Component, params, (error, props) => {
       needToLoadCounter--
       propsArray[index] = props
       componentsArray[index] = Component
@@ -112,7 +118,7 @@ function createElement(Component, props) {
     return <Component {...props}/>
 }
 
-export function loadPropsOnServer({ components, params }, cb) {
+export function loadPropsOnServer({ components, params }, cb, resolver) {
   loadAsyncProps(
     filterAndFlattenComponents(components),
     params,
@@ -125,7 +131,8 @@ export function loadPropsOnServer({ components, params }, cb) {
         const scriptString = `<script>__ASYNC_PROPS__ = ${json}</script>`
         cb(null, propsAndComponents, scriptString)
       }
-    }
+    },
+    resolver
   )
 }
 
@@ -189,6 +196,7 @@ class AsyncProps extends React.Component {
     location: object.isRequired,
     onError: func.isRequired,
     renderLoading: func.isRequired,
+    resolver: func,
 
     // server rendering
     propsArray: array,
@@ -303,7 +311,8 @@ class AsyncProps extends React.Component {
             prevProps: null
           })
         }
-      })
+      }),
+      this.props.resolver
     )
   }
 

modules/__tests__/AsyncProps-test.js

@@ -395,6 +395,33 @@ describe('AsyncProps', () => {
         />
       ), div, next)
     })
+
+    it('allows to override resolver function', (done) => {
+      const appSpy = spyOn(App, 'loadProps').andCallThrough()
+
+      const next = execNext([
+        () => {},
+        () => {
+          expect(appSpy.calls.length).toEqual(1)
+          expect(appSpy.calls[0].arguments[2]).toEqual('check')
+        },
+        done
+      ])
+
+      function resolver(Component, params, cb) {
+        Component.loadProps(params, cb, 'check')
+      }
+
+      App.setAssertions(next)
+
+      render((
+          <Router
+              history={createHistory('/')}
+              routes={routes}
+              render={(props) => <AsyncProps {...props} resolver={resolver} />}
+          />
+      ), div, next)
+    })
   })
 
   describe('server rendering', () => {
@@ -420,6 +447,24 @@ describe('AsyncProps', () => {
       })
     })
 
+    it('allows to override resolver function', (done) => {
+      const appSpy = spyOn(App, 'loadProps').andCallThrough()
+      function resolver(Component, params, cb) {
+        Component.loadProps(params, cb, 'check')
+      }
+      const loadPropsRoutes = {
+        path: '/',
+        component: App
+      }
+      match({ routes: loadPropsRoutes, location: '/' }, (err, redirect, renderProps) => {
+        loadPropsOnServer(renderProps, () => {
+          expect(appSpy.calls.length).toEqual(1)
+          expect(appSpy.calls[0].arguments[2]).toEqual('check1')
+          done()
+        }, resolver)
+      })
+    })
+
     it('renders synchronously with props from hydration', () => {
       const html = renderToString(
         <Router