feat: add hooks for body also (#16)

* feat: add hooks for body also * chore: move the renderTags function outside
flybondi · Feb 4, 2019 · abcd587 · abcd587
1 parent d7f828e
commit abcd587
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 13 deletions.
diff --git a/flow-typed/documentComponent_vx.x.x.js b/flow-typed/documentComponent_vx.x.x.js
@@ -53,6 +53,15 @@ declare module 'Document.component' {
     error: Error
   };
 
+  declare type ExtraTag = {
+    name: string,
+    tag: string,
+    content: string,
+    attribs?: {
+      [key: string]: string
+    }
+  };
+
   declare type DocumentInitialProps = {
     assets: Assets,
     criticalCSS: boolean | string,
@@ -66,14 +75,8 @@ declare module 'Document.component' {
     errorComponent: React$ComponentType<ErrorProps>,
     filterServerData: (data: DataType) => DataType,
     html: string,
-    extraHeadTags: Array<{
-      name: string,
-      tag: string,
-      content: string,
-      attribs?: {
-        [key: string]: string
-      }
-    }>,
+    extraHeadTags: Array<ExtraTag>,
+    extraBodyTags: Array<ExtraTag>,
     [key: string]: any
   };
 

diff --git a/src/Document.component.jsx b/src/Document.component.jsx
@@ -1,5 +1,11 @@
 // @flow strict
-import type { DocumentInitialProps, Context, Extractor, DataType } from 'Document.component';
+import type {
+  ExtraTag,
+  DocumentInitialProps,
+  Context,
+  Extractor,
+  DataType
+} from 'Document.component';
 import React, { PureComponent } from 'react';
 import { F, identity, path } from 'ramda';
 import Error from './Error.component';
@@ -11,6 +17,10 @@ const getHeaderTags = (extractor: Extractor) => [
   ...extractor.getLinkElements()
 ];
 
+const renderTags = ({ tag: Tag, content, name, attribs }: ExtraTag) => (
+  <Tag key={name} {...attribs} dangerouslySetInnerHTML={{ __html: content }} />
+);
+
 export class DocumentComponent extends PureComponent<DocumentInitialProps> {
   static async getInitialProps({
     assets,
@@ -35,7 +45,8 @@ export class DocumentComponent extends PureComponent<DocumentInitialProps> {
       errorComponent: ErrorComponent,
       filterServerData = identity,
       extractor,
-      extraHeadTags = []
+      extraHeadTags = [],
+      extraBodyTags = []
     } = this.props;
     // get attributes from React Helmet
     const htmlAttrs = helmet.htmlAttributes.toComponent();
@@ -57,9 +68,7 @@ export class DocumentComponent extends PureComponent<DocumentInitialProps> {
           {extractor && getHeaderTags(extractor)}
           {criticalCSS !== false && criticalCSS}
           {clientCss && <link rel="stylesheet" href={clientCss} />}
-          {extraHeadTags.map(({ tag: Tag, content, name, attribs }) => (
-            <Tag key={name} {...attribs} dangerouslySetInnerHTML={{ __html: content }} />
-          ))}
+          {extraHeadTags.map(renderTags)}
         </head>
         <body {...bodyAttrs}>
           <Root />
@@ -71,6 +80,7 @@ export class DocumentComponent extends PureComponent<DocumentInitialProps> {
               <Error message={error.message} stack={error.stack} />
             )
           ) : null}
+          {extraBodyTags.map(renderTags)}
         </body>
       </html>
     );