From facbe77fedc210c27bc01efb9c19fd14aee4f1f6 Mon Sep 17 00:00:00 2001
From: Chris Ball <chris@echobind.com>
Date: Mon, 31 Aug 2020 22:26:32 -0400
Subject: [PATCH] adds ability to override components from consuming app

---
 src/block.tsx            | 14 ++++++++++++--
 src/components/asset.tsx |  4 +++-
 src/renderer.tsx         |  8 +++++++-
 src/types.ts             | 20 +++++++++++++++++++-
 4 files changed, 41 insertions(+), 5 deletions(-)

diff --git a/src/block.tsx b/src/block.tsx
index 672b34f..5733db1 100644
--- a/src/block.tsx
+++ b/src/block.tsx
@@ -5,7 +5,9 @@ import {
   ContentValueType,
   BlockMapType,
   MapPageUrl,
-  MapImageUrl
+  MapImageUrl,
+  BlockValueTypeKeys,
+  CustomComponent
 } from "./types";
 import Asset from "./components/asset";
 import Code from "./components/code";
@@ -62,6 +64,7 @@ interface Block {
 
   fullPage?: boolean;
   hideHeader?: boolean;
+  customComponents?: Record<BlockValueTypeKeys, CustomComponent>;
 }
 
 export const Block: React.FC<Block> = props => {
@@ -73,10 +76,17 @@ export const Block: React.FC<Block> = props => {
     hideHeader,
     blockMap,
     mapPageUrl,
-    mapImageUrl
+    mapImageUrl,
+    customComponents
   } = props;
   const blockValue = block?.value;
 
+  // render a custom component first if passed.
+  if (customComponents && customComponents[blockValue?.type]) {
+    const CustomComponent = customComponents[blockValue?.type] as any;
+    return <CustomComponent blockValue={blockValue} />;
+  }
+
   switch (blockValue?.type) {
     case "page":
       if (level === 0) {
diff --git a/src/components/asset.tsx b/src/components/asset.tsx
index a6026f1..86de042 100644
--- a/src/components/asset.tsx
+++ b/src/components/asset.tsx
@@ -34,7 +34,9 @@ const Asset: React.FC<{
       >
         <iframe
           className="notion-image-inset"
-          src={type === "figma" ? value.properties.source[0][0] : display_source}
+          src={
+            type === "figma" ? value.properties.source[0][0] : display_source
+          }
         />
       </div>
     );
diff --git a/src/renderer.tsx b/src/renderer.tsx
index bbda15b..feed608 100644
--- a/src/renderer.tsx
+++ b/src/renderer.tsx
@@ -1,5 +1,10 @@
 import React from "react";
-import { BlockMapType, MapPageUrl, MapImageUrl } from "./types";
+import {
+  BlockMapType,
+  MapPageUrl,
+  MapImageUrl,
+  CustomComponent
+} from "./types";
 import { Block } from "./block";
 import { defaultMapImageUrl, defaultMapPageUrl } from "./utils";
 
@@ -12,6 +17,7 @@ export interface NotionRendererProps {
 
   currentId?: string;
   level?: number;
+  customComponents?: Record<string, CustomComponent>;
 }
 
 export const NotionRenderer: React.FC<NotionRendererProps> = ({
diff --git a/src/types.ts b/src/types.ts
index 41583af..698a414 100644
--- a/src/types.ts
+++ b/src/types.ts
@@ -8,6 +8,8 @@
  *
  */
 
+import { FC } from "react";
+
 /**
  * Base properties that all blocks share.
  */
@@ -256,6 +258,13 @@ interface TableCollectionType extends BaseValueType {
   };
 }
 
+export interface TweetType extends BaseValueType {
+  type: "tweet";
+  properties: {
+    source: [string[]];
+  };
+}
+
 export type CollectionViewType = TableGalleryType | TableCollectionType;
 
 /**
@@ -282,7 +291,10 @@ export type BlockValueType =
   | CalloutValueType
   | BookmarkValueType
   | ToggleValueType
-  | CollectionValueType;
+  | CollectionValueType
+  | TweetType;
+
+export type BlockValueTypeKeys = Pick<BlockValueType, "type">["type"];
 
 export interface BlockType {
   role: string;
@@ -329,3 +341,9 @@ export interface LoadPageChunkData {
 
 export type MapPageUrl = (pageId: string) => string;
 export type MapImageUrl = (image: string) => string;
+
+export interface CustomComponentProps {
+  blockValue: BlockValueType;
+}
+
+export type CustomComponent = FC<CustomComponentProps>;