update styletron example to support multiple stylesheets (#763)

vercel · Jan 13, 2017 · 5ef34c6 · 5ef34c6
1 parent 23cddda
commit 5ef34c6
Showing 1 changed file with 10 additions and 3 deletions.
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 />