From dd24711f21ef6097c99edade1a40202b23802374 Mon Sep 17 00:00:00 2001
From: lawvs <18554747+lawvs@users.noreply.github.com>
Date: Wed, 24 Aug 2022 01:25:25 +0800
Subject: [PATCH] refactor: update editor

---
 libs/components/affine-editor/src/Editor.tsx  |  5 +-
 libs/components/editor-core/src/Contexts.tsx  | 28 +++++--
 .../components/editor-core/src/RenderRoot.tsx | 82 +++++++++----------
 libs/components/editor-core/src/index.ts      | 27 +++---
 .../editor-core/src/kanban/Context.tsx        | 10 ++-
 5 files changed, 81 insertions(+), 71 deletions(-)

diff --git a/libs/components/affine-editor/src/Editor.tsx b/libs/components/affine-editor/src/Editor.tsx
index c280bea984..f1edb2ce74 100644
--- a/libs/components/affine-editor/src/Editor.tsx
+++ b/libs/components/affine-editor/src/Editor.tsx
@@ -1,7 +1,6 @@
 import { forwardRef, useEffect, useImperativeHandle, useRef } from 'react';
 
 import {
-    RenderBlock,
     RenderRoot,
     type BlockEditor,
 } from '@toeverything/components/editor-core';
@@ -88,9 +87,7 @@ export const AffineEditor = forwardRef<BlockEditor, AffineEditorProps>(
                 editor={editor}
                 editorElement={AffineEditor as any}
                 scrollBlank={scrollBlank}
-            >
-                <RenderBlock blockId={editor.getRootBlockId()} />
-            </RenderRoot>
+            />
         );
     }
 );
diff --git a/libs/components/editor-core/src/Contexts.tsx b/libs/components/editor-core/src/Contexts.tsx
index 418fe0e8b4..0b7c7d2270 100644
--- a/libs/components/editor-core/src/Contexts.tsx
+++ b/libs/components/editor-core/src/Contexts.tsx
@@ -1,22 +1,34 @@
-import { createContext, useContext } from 'react';
-import type { BlockEditor, AsyncBlock } from './editor';
 import { genErrorObj } from '@toeverything/utils';
+import { createContext, PropsWithChildren, useContext } from 'react';
+import type { AsyncBlock, BlockEditor } from './editor';
 
-const RootContext = createContext<{
+type EditorProps = {
     editor: BlockEditor;
     // TODO: Temporary fix, dependencies in the new architecture are bottom-up, editors do not need to be passed down from the top
     editorElement: () => JSX.Element;
-}>(
+};
+
+const EditorContext = createContext<EditorProps>(
     genErrorObj(
-        'Failed to get context! The context only can use under the "render-root"'
+        'Failed to get EditorContext! The context only can use under the "render-root"'
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
     ) as any
 );
 
-export const EditorProvider = RootContext.Provider;
-
 export const useEditor = () => {
-    return useContext(RootContext);
+    return useContext(EditorContext);
+};
+
+export const EditorProvider = ({
+    editor,
+    editorElement,
+    children,
+}: PropsWithChildren<EditorProps>) => {
+    return (
+        <EditorContext.Provider value={{ editor, editorElement }}>
+            {children}
+        </EditorContext.Provider>
+    );
 };
 
 /**
diff --git a/libs/components/editor-core/src/RenderRoot.tsx b/libs/components/editor-core/src/RenderRoot.tsx
index cbafd3ea75..c1f1dd0d49 100644
--- a/libs/components/editor-core/src/RenderRoot.tsx
+++ b/libs/components/editor-core/src/RenderRoot.tsx
@@ -4,12 +4,12 @@ import {
     services,
     type ReturnUnobserve,
 } from '@toeverything/datasource/db-service';
-import type { PropsWithChildren } from 'react';
 import React, { useCallback, useEffect, useRef, useState } from 'react';
 import { EditorProvider } from './Contexts';
 import type { BlockEditor } from './editor';
 import { useIsOnDrag } from './hooks';
 import { addNewGroup, appendNewGroup } from './recast-block';
+import { BlockRenderProvider, RenderBlock } from './render-block';
 import { SelectionRect, SelectionRef } from './Selection';
 
 interface RenderRootProps {
@@ -24,11 +24,7 @@ interface RenderRootProps {
 const MAX_PAGE_WIDTH = 5000;
 export const MIN_PAGE_WIDTH = 1480;
 
-export const RenderRoot = ({
-    editor,
-    editorElement,
-    children,
-}: PropsWithChildren<RenderRootProps>) => {
+export const RenderRoot = ({ editor, editorElement }: RenderRootProps) => {
     const selectionRef = useRef<SelectionRef>(null);
     const triggeredBySelect = useRef(false);
     const [pageWidth, setPageWidth] = useState<number>(MIN_PAGE_WIDTH);
@@ -158,39 +154,43 @@ export const RenderRoot = ({
     };
 
     return (
-        <EditorProvider value={{ editor, editorElement }}>
-            <Container
-                isEdgeless={editor.isEdgeless}
-                ref={ref => {
-                    if (ref != null && ref !== editor.container) {
-                        editor.container = ref;
-                        editor.getHooks().render();
-                    }
-                }}
-                onMouseMove={onMouseMove}
-                onMouseDown={onMouseDown}
-                onMouseUp={onMouseUp}
-                onMouseLeave={onMouseLeave}
-                onMouseOut={onMouseOut}
-                onContextMenu={onContextmenu}
-                onKeyDown={onKeyDown}
-                onKeyDownCapture={onKeyDownCapture}
-                onKeyUp={onKeyUp}
-                onDragOver={onDragOver}
-                onDragLeave={onDragLeave}
-                onDragOverCapture={onDragOverCapture}
-                onDragEnd={onDragEnd}
-                onDrop={onDrop}
-                isOnDrag={isOnDrag}
-            >
-                <Content style={{ maxWidth: pageWidth + 'px' }}>
-                    {children}
-                </Content>
-                {/** TODO: remove selectionManager insert */}
-                {editor && <SelectionRect ref={selectionRef} editor={editor} />}
-                {editor.isEdgeless ? null : <ScrollBlank editor={editor} />}
-                {patchedNodes}
-            </Container>
+        <EditorProvider editor={editor} editorElement={editorElement}>
+            <BlockRenderProvider blockRender={RenderBlock}>
+                <Container
+                    isEdgeless={editor.isEdgeless}
+                    ref={ref => {
+                        if (ref != null && ref !== editor.container) {
+                            editor.container = ref;
+                            editor.getHooks().render();
+                        }
+                    }}
+                    onMouseMove={onMouseMove}
+                    onMouseDown={onMouseDown}
+                    onMouseUp={onMouseUp}
+                    onMouseLeave={onMouseLeave}
+                    onMouseOut={onMouseOut}
+                    onContextMenu={onContextmenu}
+                    onKeyDown={onKeyDown}
+                    onKeyDownCapture={onKeyDownCapture}
+                    onKeyUp={onKeyUp}
+                    onDragOver={onDragOver}
+                    onDragLeave={onDragLeave}
+                    onDragOverCapture={onDragOverCapture}
+                    onDragEnd={onDragEnd}
+                    onDrop={onDrop}
+                    isOnDrag={isOnDrag}
+                >
+                    <Content style={{ maxWidth: pageWidth + 'px' }}>
+                        <RenderBlock blockId={editor.getRootBlockId()} />
+                    </Content>
+                    {/** TODO: remove selectionManager insert */}
+                    {editor && (
+                        <SelectionRect ref={selectionRef} editor={editor} />
+                    )}
+                    {editor.isEdgeless ? null : <ScrollBlank editor={editor} />}
+                    {patchedNodes}
+                </Container>
+            </BlockRenderProvider>
         </EditorProvider>
     );
 };
@@ -251,7 +251,7 @@ function ScrollBlank({ editor }: { editor: BlockEditor }) {
     );
 
     return (
-        <ScrollBlankContainter
+        <ScrollBlankContainer
             onMouseDown={onMouseDown}
             onMouseMove={onMouseMove}
             onClick={onClick}
@@ -283,7 +283,7 @@ const Content = styled('div')({
     transitionTimingFunction: 'ease-in',
 });
 
-const ScrollBlankContainter = styled('div')({
+const ScrollBlankContainer = styled('div')({
     paddingBottom: '30vh',
     margin: `0 -${PADDING_X}px`,
 });
diff --git a/libs/components/editor-core/src/index.ts b/libs/components/editor-core/src/index.ts
index d28d8e96b3..6059350bc9 100644
--- a/libs/components/editor-core/src/index.ts
+++ b/libs/components/editor-core/src/index.ts
@@ -1,19 +1,16 @@
-export { RenderRoot, MIN_PAGE_WIDTH } from './RenderRoot';
-export * from './render-block';
-export * from './hooks';
-
-export { RenderBlock } from './render-block';
-
-export * from './recast-block';
-export * from './recast-block/types';
-
 export * from './block-pendant';
-
+export { useEditor } from './Contexts';
+export * from './editor';
+export * from './hooks';
 export * from './kanban';
 export * from './kanban/types';
-
+export * from './recast-block';
+export * from './recast-block/types';
+export {
+    BlockRenderProvider,
+    RenderBlockChildren,
+    useBlockRender,
+    withTreeViewChildren,
+} from './render-block';
+export { MIN_PAGE_WIDTH, RenderRoot } from './RenderRoot';
 export * from './utils';
-
-export * from './editor';
-
-export { useEditor } from './Contexts';
diff --git a/libs/components/editor-core/src/kanban/Context.tsx b/libs/components/editor-core/src/kanban/Context.tsx
index 12cc0db2f1..d0fc58ed0a 100644
--- a/libs/components/editor-core/src/kanban/Context.tsx
+++ b/libs/components/editor-core/src/kanban/Context.tsx
@@ -10,6 +10,8 @@ import {
     RecastMetaProperty,
     RecastPropertyId,
 } from '../recast-block/types';
+import { BlockRenderProvider } from '../render-block';
+import { KanbanBlockRender } from '../render-block/RenderKanbanBlock';
 import { useInitKanbanEffect, useRecastKanban } from './kanban';
 import { KanbanGroup } from './types';
 
@@ -54,9 +56,11 @@ export const KanbanProvider = ({
     };
 
     return (
-        <KanbanContext.Provider value={value}>
-            {children}
-        </KanbanContext.Provider>
+        <BlockRenderProvider blockRender={KanbanBlockRender}>
+            <KanbanContext.Provider value={value}>
+                {children}
+            </KanbanContext.Provider>
+        </BlockRenderProvider>
     );
 };