update styletron example to support multiple stylesheets (#763)

skellyb · rauchg · commit 5ef34c6dd8d4 · 2017-01-13T15:49:35.000-08:00
diff --git a/examples/with-styletron/pages/_document.js b/examples/with-styletron/pages/_document.js
@@ -5,16 +5,23 @@ export default class MyDocument extends Document {
   static getInitialProps ({ renderPage }) {
     const page = renderPage()
     const styletron = flush()
-    const css = styletron ? styletron.getCss() : null
-    return { ...page, css }
+    const stylesheets = styletron ? styletron.getStylesheets() : []
+    return { ...page, stylesheets }
   }
 
   render () {
     return (
       <html>
         <Head>
           <title>My page</title>
-          <style className='_styletron_hydrate_' dangerouslySetInnerHTML={{ __html: this.props.css }} />
+          {this.props.stylesheets.map((sheet, i) => (
+            <style
+              className="_styletron_hydrate_"
+              dangerouslySetInnerHTML={{ __html: sheet.css }}
+              media={sheet.media || ''}
+              key={i}
+            />
+          ))}
         </Head>
         <body>
           <Main />
