feat(docz): add rawCode for Playground

doczjs · Jul 22, 2018 · 70d4735 · 70d4735
1 parent af88d0e
commit 70d4735
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 5 deletions.
diff --git a/packages/docz/src/components/DocPreview.tsx b/packages/docz/src/components/DocPreview.tsx
@@ -14,10 +14,12 @@ const Identity: SFC<any> = ({ children }) => <Fragment>{children}</Fragment>
 const DefaultLoading: SFC = () => <Fragment>Loading</Fragment>
 
 export type RenderComponent = ComponentType<{
-  component: JSX.Element
-  code: any
   className?: string
   style?: any
+  components: ComponentsMap
+  component: JSX.Element
+  code: (components: ComponentsMap) => any
+  rawCode: string
 }>
 
 export const DefaultRender: RenderComponent = ({ component, code }) => (

diff --git a/packages/docz/src/components/Playground.tsx b/packages/docz/src/components/Playground.tsx
@@ -6,11 +6,12 @@ import { isFn } from './Docs'
 import { ComponentsMap } from './DocPreview'
 
 export interface PlaygroundProps {
-  components: ComponentsMap
   className?: string
   style?: any
-  children: any
+  components: ComponentsMap
   __code: (components: ComponentsMap) => any
+  __rawCode: string
+  children: any
 }
 
 const BasePlayground: SFC<PlaygroundProps> = ({
@@ -19,13 +20,16 @@ const BasePlayground: SFC<PlaygroundProps> = ({
   style,
   children,
   __code,
+  __rawCode,
 }) => {
   return components && components.render ? (
     <components.render
       className={className}
       style={style}
+      components={components}
       component={isFn(children) ? children() : children}
       code={__code(components)}
+      rawCode={__rawCode}
     />
   ) : null
 }

diff --git a/packages/rehype-docz/src/index.ts b/packages/rehype-docz/src/index.ts
@@ -42,7 +42,7 @@ const addCodeProp = async (node: any) => {
 
     node.value = node.value.replace(
       tagOpen,
-      `<${name} __code={${codeComponent}}`
+      `<${name} __code={${codeComponent}} __rawCode={\`${child}\`}`
     )
   }
 }