Pass context to render function

README.md

@@ -76,13 +76,14 @@ A class that represents an application. An application is responsible for render
 ```js
 const app: App = new App(
   (el: ReactElement),
-  (render: ?(el: ReactElement) => any)
+  (render: ?(el: ReactElement, ctx: Context) => any)
 );
 ```
 
 - `el: ReactElement` - a template root. In a React application, this would be a React element created via `React.createElement` or a JSX expression.
 - `render: ?Plugin<Render>|Render` - Optional. Defines how rendering should occur. A Plugin should provide a value of type `Render`
-  - `type Render = (el:any) => any`
+  - `type Render = (el:ReactComponent, ctx: Context) => any`
+
 
 **app.register**
 

src/__tests__/render.js

@@ -13,7 +13,9 @@ import App from '../index';
 
 test('custom render function', async t => {
   let didRender = false;
-  const app = new App(React.createElement('span', null, 'hello'), () => {
+  const app = new App(React.createElement('span', null, 'hello'), (el, ctx) => {
+    t.ok(el);
+    t.ok(ctx);
     didRender = true;
     return 10;
   });

src/index.js

@@ -9,7 +9,11 @@
 /* eslint-env browser */
 import * as React from 'react';
 
-import FusionApp, {createPlugin, CriticalChunkIdsToken} from 'fusion-core';
+import FusionApp, {
+  createPlugin,
+  CriticalChunkIdsToken,
+  type Context,
+} from 'fusion-core';
 import {prepare} from './async/index.js';
 import PrepareProvider from './async/prepare-provider';
 import {registerInjector, withServices} from './injector.js';
@@ -21,10 +25,12 @@ import ProviderPlugin from './plugin';
 import ProvidedHOC from './hoc';
 import Provider from './provider';
 
+export type Render = (el: React.Element<*>, context: Context) => any;
+
 declare var __NODE__: Boolean;
 
 export default class App extends FusionApp {
-  constructor(root: React.Element<*>, render: ?(React.Element<*>) => any) {
+  constructor(root: React.Element<*>, render: ?Render) {
     if (!React.isValidElement(root))
       throw new Error(
         'Invalid React element. Ensure your root element is a React.Element (e.g. <Foo />) and not a React.Component (e.g. Foo)'
@@ -34,10 +40,10 @@ export default class App extends FusionApp {
         criticalChunkIds: CriticalChunkIdsToken.optional,
       },
       provides() {
-        return (el: React.Element<*>) => {
+        return (el: React.Element<*>, ctx) => {
           return prepare(el).then(() => {
             if (render) {
-              return render(el);
+              return render(el, ctx);
             }
             if (__NODE__) {
               return serverRender(el);