redraw()

src/client/index.ts

@@ -1,3 +1,3 @@
-export { setup } from './setup'
+export { setup, redraw } from './setup'
 export { send, gameId } from './websocket'
 export { sessionId } from './session'

src/client/setup.tsx

@@ -3,16 +3,21 @@ import { initWebsocket, onUpdate } from './websocket'
 
 const root = document.getElementById('root')!
 
+export let redraw = () => { }
+
 export function setup<G>(render: typeof renderJsx, onRender: (game: G) => Child) {
   let game: G | undefined = undefined
 
+  redraw = () =>
+    render(game ? onRender(game) : <Loading />, root)
+
   onUpdate((newGame: G) => {
     game = newGame
-    render(game ? onRender(game) : <Loading />, root)
+    redraw()
   })
 
   initWebsocket()
-  render(game ? onRender(game) : <Loading />, root)
+  redraw()
 }
 
 const Loading = () => <h1>Loading...</h1>