From 4edf0b1d857efaca59e3560dd599dc550691729d Mon Sep 17 00:00:00 2001
From: akumatus <12724894+akumatus@users.noreply.github.com>
Date: Tue, 18 Jun 2024 10:16:44 +0000
Subject: [PATCH] fix: add patch doc mode service (#7249)

---
 .../blocksuite/block-suite-editor/lit-adaper.tsx   | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/packages/frontend/core/src/components/blocksuite/block-suite-editor/lit-adaper.tsx b/packages/frontend/core/src/components/blocksuite/block-suite-editor/lit-adaper.tsx
index f1d40544b..cfe718e54 100644
--- a/packages/frontend/core/src/components/blocksuite/block-suite-editor/lit-adaper.tsx
+++ b/packages/frontend/core/src/components/blocksuite/block-suite-editor/lit-adaper.tsx
@@ -15,7 +15,13 @@ import {
   PageEditor,
 } from '@blocksuite/presets';
 import type { Doc } from '@blocksuite/store';
-import { type DocMode, useLiveData, useService } from '@toeverything/infra';
+import {
+  type DocMode,
+  DocService,
+  DocsService,
+  useLiveData,
+  useService,
+} from '@toeverything/infra';
 import React, {
   forwardRef,
   Fragment,
@@ -30,6 +36,7 @@ import { PagePropertiesTable } from '../../affine/page-properties';
 import { AffinePageReference } from '../../affine/reference-link';
 import { BlocksuiteEditorJournalDocTitle } from './journal-doc-title';
 import {
+  patchDocModeService,
   patchForSharedPage,
   patchNotificationService,
   patchPeekViewService,
@@ -73,6 +80,8 @@ const usePatchSpecs = (page: Doc, shared: boolean, mode: DocMode) => {
   const [reactToLit, portals] = useLitPortalFactory();
   const peekViewService = useService(PeekViewService);
   const quickSearchService = useService(QuickSearchService);
+  const docService = useService(DocService);
+  const docsService = useService(DocsService);
   const referenceRenderer: ReferenceReactRenderer = useMemo(() => {
     return function customReference(reference) {
       const pageId = reference.delta.attributes?.reference?.pageId;
@@ -101,9 +110,12 @@ const usePatchSpecs = (page: Doc, shared: boolean, mode: DocMode) => {
     if (shared) {
       patched = patchForSharedPage(patched);
     }
+    patched = patchDocModeService(patched, docService, docsService);
     return patched;
   }, [
     confirmModal,
+    docService,
+    docsService,
     page.readonly,
     peekViewService,
     quickSearchService,