From 9321ce94a781c9a02f840e3d8499a88adac63de8 Mon Sep 17 00:00:00 2001
From: Saul-Mirone <Saul-Mirone@outlook.com>
Date: Thu, 13 Feb 2025 04:35:35 +0000
Subject: [PATCH] refactor(editor): remove global types in config (#10143)

Closes: [BS-2554](https://linear.app/affine-design/issue/BS-2554/remove-global-types-in-block-config)
---
 .../block-code/src/code-block-config.ts       |  4 +++
 .../block-code/src/code-block-service.ts      | 10 ++++++--
 .../affine/block-code/src/code-block.ts       |  4 ++-
 blocksuite/affine/block-code/src/effects.ts   |  7 ------
 .../affine/block-database/src/config.ts       |  4 +++
 .../block-database/src/database-block.ts      |  7 ++++--
 .../affine/block-database/src/effects.ts      |  9 -------
 blocksuite/affine/block-database/src/index.ts |  2 +-
 .../components/edgeless-note-background.ts    |  3 ++-
 .../components/edgeless-page-block-title.ts   | 11 +++++---
 blocksuite/affine/block-note/src/config.ts    | 10 +++++---
 blocksuite/affine/block-note/src/effects.ts   |  9 -------
 .../affine/block-surface/src/effects.ts       |  9 -------
 .../affine/components/src/toolbar/config.ts   |  7 ++++++
 .../affine/components/src/toolbar/index.ts    |  1 +
 .../affine/components/src/toolbar/utils.ts    |  7 ++----
 blocksuite/blocks/src/effects.ts              |  9 -------
 blocksuite/blocks/src/index.ts                |  1 +
 .../root-block/edgeless/edgeless-root-spec.ts |  2 +-
 .../src/root-block/page/page-root-spec.ts     |  2 +-
 .../blocks/src/root-block/root-config.ts      |  6 +++--
 blocksuite/blocks/src/root-block/types.ts     |  2 +-
 .../widgets/keyboard-toolbar/index.ts         |  4 ++-
 .../root-block/widgets/linked-doc/config.ts   |  2 ++
 .../root-block/widgets/linked-doc/effects.ts  |  3 ++-
 .../root-block/widgets/linked-doc/index.ts    | 11 +++++---
 .../block-std/src/__tests__/test-schema.ts    | 10 --------
 .../block-std/src/extension/config.ts         | 25 +++++++++++++------
 .../block-std/src/scope/block-std-scope.ts    | 22 ----------------
 .../src/__tests__/collection.unit.spec.ts     | 10 --------
 .../store/src/__tests__/schema.unit.spec.ts   | 10 --------
 .../src/__tests__/transformer.unit.spec.ts    |  8 ------
 blocksuite/tests-legacy/utils/actions/drag.ts |  2 +-
 .../specs/custom/root-block.ts                | 18 ++++++-------
 34 files changed, 98 insertions(+), 153 deletions(-)
 create mode 100644 blocksuite/affine/components/src/toolbar/config.ts

diff --git a/blocksuite/affine/block-code/src/code-block-config.ts b/blocksuite/affine/block-code/src/code-block-config.ts
index fd94d55c13..edacfba115 100644
--- a/blocksuite/affine/block-code/src/code-block-config.ts
+++ b/blocksuite/affine/block-code/src/code-block-config.ts
@@ -1,3 +1,4 @@
+import { ConfigExtensionFactory } from '@blocksuite/block-std';
 import type { BundledLanguageInfo, ThemeInput } from 'shiki';
 
 export interface CodeBlockConfig {
@@ -13,3 +14,6 @@ export interface CodeBlockConfig {
    */
   showLineNumbers?: boolean;
 }
+
+export const CodeBlockConfigExtension =
+  ConfigExtensionFactory<CodeBlockConfig>('affine:code');
diff --git a/blocksuite/affine/block-code/src/code-block-service.ts b/blocksuite/affine/block-code/src/code-block-service.ts
index 9bc2a5140f..5cf4fcd9db 100644
--- a/blocksuite/affine/block-code/src/code-block-service.ts
+++ b/blocksuite/affine/block-code/src/code-block-service.ts
@@ -12,6 +12,7 @@ import {
 } from 'shiki';
 import getWasm from 'shiki/wasm';
 
+import { CodeBlockConfigExtension } from './code-block-config.js';
 import {
   CODE_BLOCK_DEFAULT_DARK_THEME,
   CODE_BLOCK_DEFAULT_LIGHT_THEME,
@@ -27,7 +28,10 @@ export class CodeBlockService extends BlockService {
   highlighter$: Signal<HighlighterCore | null> = signal(null);
 
   get langs() {
-    return this.std.getConfig('affine:code')?.langs ?? bundledLanguagesInfo;
+    return (
+      this.std.getOptional(CodeBlockConfigExtension.identifier)?.langs ??
+      bundledLanguagesInfo
+    );
   }
 
   get themeKey() {
@@ -46,7 +50,9 @@ export class CodeBlockService extends BlockService {
       engine: createOnigurumaEngine(() => getWasm),
     })
       .then(async highlighter => {
-        const config = this.std.getConfig('affine:code');
+        const config = this.std.getOptional(
+          CodeBlockConfigExtension.identifier
+        );
         const darkTheme = config?.theme?.dark ?? CODE_BLOCK_DEFAULT_DARK_THEME;
         const lightTheme =
           config?.theme?.light ?? CODE_BLOCK_DEFAULT_LIGHT_THEME;
diff --git a/blocksuite/affine/block-code/src/code-block.ts b/blocksuite/affine/block-code/src/code-block.ts
index 9a178a698a..77c464a128 100644
--- a/blocksuite/affine/block-code/src/code-block.ts
+++ b/blocksuite/affine/block-code/src/code-block.ts
@@ -32,6 +32,7 @@ import { classMap } from 'lit/directives/class-map.js';
 import type { ThemedToken } from 'shiki';
 
 import { CodeClipboardController } from './clipboard/index.js';
+import { CodeBlockConfigExtension } from './code-block-config.js';
 import { CodeBlockInlineManagerExtension } from './code-block-inline.js';
 import type { CodeBlockService } from './code-block-service.js';
 import { codeBlockStyles } from './styles.js';
@@ -383,7 +384,8 @@ export class CodeBlockComponent extends CaptionedBlockComponent<
 
   override renderBlock(): TemplateResult<1> {
     const showLineNumbers =
-      this.std.getConfig('affine:code')?.showLineNumbers ?? true;
+      this.std.getOptional(CodeBlockConfigExtension.identifier)
+        ?.showLineNumbers ?? true;
 
     return html`
       <div
diff --git a/blocksuite/affine/block-code/src/effects.ts b/blocksuite/affine/block-code/src/effects.ts
index ac7b0ec0ef..63c59037e4 100644
--- a/blocksuite/affine/block-code/src/effects.ts
+++ b/blocksuite/affine/block-code/src/effects.ts
@@ -1,5 +1,4 @@
 import { CodeBlockComponent } from './code-block';
-import type { CodeBlockConfig } from './code-block-config';
 import {
   AFFINE_CODE_TOOLBAR_WIDGET,
   AffineCodeToolbarWidget,
@@ -17,12 +16,6 @@ export function effects() {
 }
 
 declare global {
-  namespace BlockSuite {
-    interface BlockConfigs {
-      'affine:code': CodeBlockConfig;
-    }
-  }
-
   interface HTMLElementTagNameMap {
     'language-list-button': LanguageListButton;
     'affine-code-toolbar': AffineCodeToolbar;
diff --git a/blocksuite/affine/block-database/src/config.ts b/blocksuite/affine/block-database/src/config.ts
index ca68868160..35dc7ec44d 100644
--- a/blocksuite/affine/block-database/src/config.ts
+++ b/blocksuite/affine/block-database/src/config.ts
@@ -1,6 +1,10 @@
 import type { MenuOptions } from '@blocksuite/affine-components/context-menu';
 import { type DatabaseBlockModel } from '@blocksuite/affine-model';
+import { ConfigExtensionFactory } from '@blocksuite/block-std';
 
 export interface DatabaseOptionsConfig {
   configure: (model: DatabaseBlockModel, options: MenuOptions) => MenuOptions;
 }
+
+export const DatabaseConfigExtension =
+  ConfigExtensionFactory<DatabaseOptionsConfig>('affine:database');
diff --git a/blocksuite/affine/block-database/src/database-block.ts b/blocksuite/affine/block-database/src/database-block.ts
index 757e6bd662..12c5e6b17c 100644
--- a/blocksuite/affine/block-database/src/database-block.ts
+++ b/blocksuite/affine/block-database/src/database-block.ts
@@ -48,7 +48,10 @@ import { computed, signal } from '@preact/signals-core';
 import { css, html, nothing, unsafeCSS } from 'lit';
 
 import { popSideDetail } from './components/layout.js';
-import type { DatabaseOptionsConfig } from './config.js';
+import {
+  DatabaseConfigExtension,
+  type DatabaseOptionsConfig,
+} from './config.js';
 import { HostContextKey } from './context/host-context.js';
 import { DatabaseBlockDataSource } from './data-source.js';
 import { BlockRenderer } from './detail-panel/block-renderer.js';
@@ -343,7 +346,7 @@ export class DatabaseBlockComponent extends CaptionedBlockComponent<DatabaseBloc
   get optionsConfig(): DatabaseOptionsConfig {
     return {
       configure: (_model, options) => options,
-      ...this.std.getConfig('affine:database'),
+      ...this.std.getOptional(DatabaseConfigExtension.identifier),
     };
   }
 
diff --git a/blocksuite/affine/block-database/src/effects.ts b/blocksuite/affine/block-database/src/effects.ts
index 33cb8c8fc9..63821adc8b 100644
--- a/blocksuite/affine/block-database/src/effects.ts
+++ b/blocksuite/affine/block-database/src/effects.ts
@@ -1,6 +1,5 @@
 import { CenterPeek } from './components/layout';
 import { DatabaseTitle } from './components/title';
-import type { DatabaseOptionsConfig } from './config';
 import { DatabaseBlockComponent } from './database-block';
 import { BlockRenderer } from './detail-panel/block-renderer';
 import { NoteRenderer } from './detail-panel/note-renderer';
@@ -37,11 +36,3 @@ export function effects() {
   customElements.define('affine-database-link-node', LinkNode);
   customElements.define('affine-database', DatabaseBlockComponent);
 }
-
-declare global {
-  namespace BlockSuite {
-    interface BlockConfigs {
-      'affine:database': Partial<DatabaseOptionsConfig>;
-    }
-  }
-}
diff --git a/blocksuite/affine/block-database/src/index.ts b/blocksuite/affine/block-database/src/index.ts
index 2480d51ce6..1e58ca1c87 100644
--- a/blocksuite/affine/block-database/src/index.ts
+++ b/blocksuite/affine/block-database/src/index.ts
@@ -1,6 +1,6 @@
 export * from './adapters';
 export * from './commands';
-export type { DatabaseOptionsConfig } from './config';
+export * from './config';
 export * from './data-source';
 export * from './database-block';
 export * from './database-spec';
diff --git a/blocksuite/affine/block-note/src/components/edgeless-note-background.ts b/blocksuite/affine/block-note/src/components/edgeless-note-background.ts
index bef130ace4..c91af88cc7 100644
--- a/blocksuite/affine/block-note/src/components/edgeless-note-background.ts
+++ b/blocksuite/affine/block-note/src/components/edgeless-note-background.ts
@@ -35,6 +35,7 @@ import { html, nothing } from 'lit';
 import { property } from 'lit/decorators.js';
 import { styleMap } from 'lit/directives/style-map.js';
 
+import { NoteConfigExtension } from '../config';
 import { isPageBlock } from '../utils';
 import * as styles from './edgeless-note-background.css';
 
@@ -148,7 +149,7 @@ export class EdgelessNoteBackground extends SignalWatcher(
 
   private _renderHeader() {
     const header = this.std
-      .getConfig('affine:note')
+      .getOptional(NoteConfigExtension.identifier)
       ?.edgelessNoteHeader({ note: this.note, std: this.std });
 
     return header;
diff --git a/blocksuite/affine/block-note/src/components/edgeless-page-block-title.ts b/blocksuite/affine/block-note/src/components/edgeless-page-block-title.ts
index 693454afb0..6ece191c60 100644
--- a/blocksuite/affine/block-note/src/components/edgeless-page-block-title.ts
+++ b/blocksuite/affine/block-note/src/components/edgeless-page-block-title.ts
@@ -11,6 +11,7 @@ import { consume } from '@lit/context';
 import { html } from 'lit';
 import { property } from 'lit/decorators.js';
 
+import { NoteConfigExtension } from '../config';
 import { isPageBlock } from '../utils';
 import * as styles from './edgeless-page-block-title.css';
 
@@ -23,10 +24,12 @@ export class EdgelessPageBlockTitle extends SignalWatcher(
   override render() {
     if (!isPageBlock(this.std, this.note)) return;
 
-    const title = this.std.getConfig('affine:note')?.pageBlockTitle({
-      note: this.note,
-      std: this.std,
-    });
+    const title = this.std
+      .getOptional(NoteConfigExtension.identifier)
+      ?.pageBlockTitle({
+        note: this.note,
+        std: this.std,
+      });
 
     return html`<div class=${styles.pageBlockTitle}>${title}</div>`;
   }
diff --git a/blocksuite/affine/block-note/src/config.ts b/blocksuite/affine/block-note/src/config.ts
index 751ec1b49a..69e6edcc9f 100644
--- a/blocksuite/affine/block-note/src/config.ts
+++ b/blocksuite/affine/block-note/src/config.ts
@@ -1,5 +1,8 @@
 import type { NoteBlockModel } from '@blocksuite/affine-model';
-import { type BlockStdScope, ConfigExtension } from '@blocksuite/block-std';
+import {
+  type BlockStdScope,
+  ConfigExtensionFactory,
+} from '@blocksuite/block-std';
 import type { TemplateResult } from 'lit';
 
 type NoteBlockContext = {
@@ -12,6 +15,5 @@ export type NoteConfig = {
   pageBlockTitle: (context: NoteBlockContext) => TemplateResult;
 };
 
-export function NoteConfigExtension(config: NoteConfig) {
-  return ConfigExtension('affine:note', config);
-}
+export const NoteConfigExtension =
+  ConfigExtensionFactory<NoteConfig>('affine:note');
diff --git a/blocksuite/affine/block-note/src/effects.ts b/blocksuite/affine/block-note/src/effects.ts
index 3321d00d3e..8100164d1f 100644
--- a/blocksuite/affine/block-note/src/effects.ts
+++ b/blocksuite/affine/block-note/src/effects.ts
@@ -1,7 +1,6 @@
 import { EdgelessNoteBackground } from './components/edgeless-note-background';
 import { EdgelessNoteMask } from './components/edgeless-note-mask';
 import { EdgelessPageBlockTitle } from './components/edgeless-page-block-title';
-import type { NoteConfig } from './config';
 import { NoteBlockComponent } from './note-block';
 import {
   AFFINE_EDGELESS_NOTE,
@@ -15,11 +14,3 @@ export function effects() {
   customElements.define('edgeless-note-background', EdgelessNoteBackground);
   customElements.define('edgeless-page-block-title', EdgelessPageBlockTitle);
 }
-
-declare global {
-  namespace BlockSuite {
-    interface BlockConfigs {
-      'affine:note': NoteConfig;
-    }
-  }
-}
diff --git a/blocksuite/affine/block-surface/src/effects.ts b/blocksuite/affine/block-surface/src/effects.ts
index 7bb01e35c7..37aa1f137b 100644
--- a/blocksuite/affine/block-surface/src/effects.ts
+++ b/blocksuite/affine/block-surface/src/effects.ts
@@ -1,16 +1,7 @@
 import { SurfaceBlockComponent } from './surface-block.js';
 import { SurfaceBlockVoidComponent } from './surface-block-void.js';
-import type { SurfaceBlockService } from './surface-service.js';
 
 export function effects() {
   customElements.define('affine-surface-void', SurfaceBlockVoidComponent);
   customElements.define('affine-surface', SurfaceBlockComponent);
 }
-
-declare global {
-  namespace BlockSuite {
-    interface BlockServices {
-      'affine:surface': SurfaceBlockService;
-    }
-  }
-}
diff --git a/blocksuite/affine/components/src/toolbar/config.ts b/blocksuite/affine/components/src/toolbar/config.ts
new file mode 100644
index 0000000000..3c2e0ea2be
--- /dev/null
+++ b/blocksuite/affine/components/src/toolbar/config.ts
@@ -0,0 +1,7 @@
+import { ConfigExtensionFactory } from '@blocksuite/block-std';
+
+import type { ToolbarMoreMenuConfig } from './types';
+
+export const ToolbarMoreMenuConfigExtension = ConfigExtensionFactory<
+  Partial<ToolbarMoreMenuConfig>
+>('affine-toolbar-more-menu');
diff --git a/blocksuite/affine/components/src/toolbar/index.ts b/blocksuite/affine/components/src/toolbar/index.ts
index 53dfeb6cec..e215c4ef6c 100644
--- a/blocksuite/affine/components/src/toolbar/index.ts
+++ b/blocksuite/affine/components/src/toolbar/index.ts
@@ -8,6 +8,7 @@ import { EditorToolbarSeparator } from './separator.js';
 import { EditorToolbar } from './toolbar.js';
 import { Tooltip } from './tooltip.js';
 
+export { ToolbarMoreMenuConfigExtension } from './config.js';
 export { EditorIconButton } from './icon-button.js';
 export {
   EditorMenuAction,
diff --git a/blocksuite/affine/components/src/toolbar/utils.ts b/blocksuite/affine/components/src/toolbar/utils.ts
index b3876874b6..3f0e1cb0ea 100644
--- a/blocksuite/affine/components/src/toolbar/utils.ts
+++ b/blocksuite/affine/components/src/toolbar/utils.ts
@@ -4,6 +4,7 @@ import { ifDefined } from 'lit/directives/if-defined.js';
 import { join } from 'lit/directives/join.js';
 import { repeat } from 'lit/directives/repeat.js';
 
+import { ToolbarMoreMenuConfigExtension } from './config.js';
 import type { MenuContext } from './menu-context.js';
 import type {
   FatMenuItems,
@@ -112,10 +113,6 @@ export function renderToolbarSeparator() {
 export function getMoreMenuConfig(std: BlockStdScope): ToolbarMoreMenuConfig {
   return {
     configure: <T extends MenuContext>(groups: MenuItemGroup<T>[]) => groups,
-    ...(
-      std.getConfig('affine:page' as BlockSuite.ConfigKeys) as null | {
-        toolbarMoreMenu: Partial<ToolbarMoreMenuConfig>;
-      }
-    )?.toolbarMoreMenu,
+    ...std.getOptional(ToolbarMoreMenuConfigExtension.identifier),
   };
 }
diff --git a/blocksuite/blocks/src/effects.ts b/blocksuite/blocks/src/effects.ts
index aa50bbbee2..ace51b213d 100644
--- a/blocksuite/blocks/src/effects.ts
+++ b/blocksuite/blocks/src/effects.ts
@@ -142,7 +142,6 @@ import {
   FramePreview,
   PageRootBlockComponent,
   PreviewRootBlockComponent,
-  type RootBlockConfig,
 } from './root-block/index.js';
 import { AIFinishTip } from './root-block/widgets/ai-panel/components/finish-tip.js';
 import { GeneratingPlaceholder } from './root-block/widgets/ai-panel/components/generating-placeholder.js';
@@ -393,11 +392,3 @@ export function effects() {
   customElements.define(AFFINE_SURFACE_REF_TOOLBAR, AffineSurfaceRefToolbar);
   customElements.define(AFFINE_FORMAT_BAR_WIDGET, AffineFormatBarWidget);
 }
-
-declare global {
-  namespace BlockSuite {
-    interface BlockConfigs {
-      'affine:page': RootBlockConfig;
-    }
-  }
-}
diff --git a/blocksuite/blocks/src/index.ts b/blocksuite/blocks/src/index.ts
index 608c00d99f..bb52c61b82 100644
--- a/blocksuite/blocks/src/index.ts
+++ b/blocksuite/blocks/src/index.ts
@@ -95,6 +95,7 @@ export {
   renderActions,
   renderGroups,
   renderToolbarSeparator,
+  ToolbarMoreMenuConfigExtension,
   Tooltip,
 } from '@blocksuite/affine-components/toolbar';
 export * from '@blocksuite/affine-model';
diff --git a/blocksuite/blocks/src/root-block/edgeless/edgeless-root-spec.ts b/blocksuite/blocks/src/root-block/edgeless/edgeless-root-spec.ts
index 2bdfb4ac58..93770e0e1c 100644
--- a/blocksuite/blocks/src/root-block/edgeless/edgeless-root-spec.ts
+++ b/blocksuite/blocks/src/root-block/edgeless/edgeless-root-spec.ts
@@ -31,7 +31,7 @@ import { EDGELESS_ELEMENT_TOOLBAR_WIDGET } from '../widgets/element-toolbar/inde
 import { AFFINE_EMBED_CARD_TOOLBAR_WIDGET } from '../widgets/embed-card-toolbar/embed-card-toolbar.js';
 import { AFFINE_FORMAT_BAR_WIDGET } from '../widgets/format-bar/format-bar.js';
 import { AFFINE_INNER_MODAL_WIDGET } from '../widgets/inner-modal/inner-modal.js';
-import { AFFINE_LINKED_DOC_WIDGET } from '../widgets/linked-doc/index.js';
+import { AFFINE_LINKED_DOC_WIDGET } from '../widgets/linked-doc/config.js';
 import { AFFINE_MODAL_WIDGET } from '../widgets/modal/modal.js';
 import { AFFINE_SLASH_MENU_WIDGET } from '../widgets/slash-menu/index.js';
 import { AFFINE_VIEWPORT_OVERLAY_WIDGET } from '../widgets/viewport-overlay/viewport-overlay.js';
diff --git a/blocksuite/blocks/src/root-block/page/page-root-spec.ts b/blocksuite/blocks/src/root-block/page/page-root-spec.ts
index 4c71974af6..126d0c5073 100644
--- a/blocksuite/blocks/src/root-block/page/page-root-spec.ts
+++ b/blocksuite/blocks/src/root-block/page/page-root-spec.ts
@@ -23,7 +23,7 @@ import { AFFINE_EMBED_CARD_TOOLBAR_WIDGET } from '../widgets/embed-card-toolbar/
 import { AFFINE_FORMAT_BAR_WIDGET } from '../widgets/format-bar/format-bar.js';
 import { AFFINE_INNER_MODAL_WIDGET } from '../widgets/inner-modal/inner-modal.js';
 import { AFFINE_KEYBOARD_TOOLBAR_WIDGET } from '../widgets/keyboard-toolbar/index.js';
-import { AFFINE_LINKED_DOC_WIDGET } from '../widgets/linked-doc/index.js';
+import { AFFINE_LINKED_DOC_WIDGET } from '../widgets/linked-doc/config.js';
 import { AFFINE_MODAL_WIDGET } from '../widgets/modal/modal.js';
 import { AFFINE_PAGE_DRAGGING_AREA_WIDGET } from '../widgets/page-dragging-area/page-dragging-area.js';
 import { AFFINE_SLASH_MENU_WIDGET } from '../widgets/slash-menu/index.js';
diff --git a/blocksuite/blocks/src/root-block/root-config.ts b/blocksuite/blocks/src/root-block/root-config.ts
index af83f06c31..43575c229e 100644
--- a/blocksuite/blocks/src/root-block/root-config.ts
+++ b/blocksuite/blocks/src/root-block/root-config.ts
@@ -1,10 +1,12 @@
-import type { ToolbarMoreMenuConfig } from '@blocksuite/affine-components/toolbar';
+import { ConfigExtensionFactory } from '@blocksuite/block-std';
 
 import type { KeyboardToolbarConfig } from './widgets/keyboard-toolbar/config.js';
 import type { LinkedWidgetConfig } from './widgets/linked-doc/index.js';
 
 export interface RootBlockConfig {
   linkedWidget?: Partial<LinkedWidgetConfig>;
-  toolbarMoreMenu?: Partial<ToolbarMoreMenuConfig>;
   keyboardToolbar?: Partial<KeyboardToolbarConfig>;
 }
+
+export const RootBlockConfigExtension =
+  ConfigExtensionFactory<RootBlockConfig>('affine:root-block');
diff --git a/blocksuite/blocks/src/root-block/types.ts b/blocksuite/blocks/src/root-block/types.ts
index 9667e36fb6..d7a89c9ba1 100644
--- a/blocksuite/blocks/src/root-block/types.ts
+++ b/blocksuite/blocks/src/root-block/types.ts
@@ -13,7 +13,7 @@ import type { AFFINE_EMBED_CARD_TOOLBAR_WIDGET } from './widgets/embed-card-tool
 import type { AFFINE_FORMAT_BAR_WIDGET } from './widgets/format-bar/format-bar.js';
 import type { AFFINE_KEYBOARD_TOOLBAR_WIDGET } from './widgets/index.js';
 import type { AFFINE_INNER_MODAL_WIDGET } from './widgets/inner-modal/inner-modal.js';
-import type { AFFINE_LINKED_DOC_WIDGET } from './widgets/linked-doc/index.js';
+import type { AFFINE_LINKED_DOC_WIDGET } from './widgets/linked-doc/config.js';
 import type { AFFINE_MODAL_WIDGET } from './widgets/modal/modal.js';
 import type { AFFINE_PAGE_DRAGGING_AREA_WIDGET } from './widgets/page-dragging-area/page-dragging-area.js';
 import type { AFFINE_SLASH_MENU_WIDGET } from './widgets/slash-menu/index.js';
diff --git a/blocksuite/blocks/src/root-block/widgets/keyboard-toolbar/index.ts b/blocksuite/blocks/src/root-block/widgets/keyboard-toolbar/index.ts
index c251dd2698..4747f1769c 100644
--- a/blocksuite/blocks/src/root-block/widgets/keyboard-toolbar/index.ts
+++ b/blocksuite/blocks/src/root-block/widgets/keyboard-toolbar/index.ts
@@ -7,6 +7,7 @@ import { signal } from '@preact/signals-core';
 import { html, nothing } from 'lit';
 
 import type { PageRootBlockComponent } from '../../page/page-root-block.js';
+import { RootBlockConfigExtension } from '../../root-config.js';
 import { defaultKeyboardToolbarConfig } from './config.js';
 
 export * from './config.js';
@@ -41,7 +42,8 @@ export class AffineKeyboardToolbarWidget extends WidgetComponent<
   get config() {
     return {
       ...defaultKeyboardToolbarConfig,
-      ...this.std.getConfig('affine:page')?.keyboardToolbar,
+      ...this.std.getOptional(RootBlockConfigExtension.identifier)
+        ?.keyboardToolbar,
     };
   }
 
diff --git a/blocksuite/blocks/src/root-block/widgets/linked-doc/config.ts b/blocksuite/blocks/src/root-block/widgets/linked-doc/config.ts
index 4f38cd1cfc..e74cfe73e3 100644
--- a/blocksuite/blocks/src/root-block/widgets/linked-doc/config.ts
+++ b/blocksuite/blocks/src/root-block/widgets/linked-doc/config.ts
@@ -259,3 +259,5 @@ export const LinkedWidgetUtils = {
   createNewDocMenuGroup,
   insertLinkedNode,
 };
+
+export const AFFINE_LINKED_DOC_WIDGET = 'affine-linked-doc-widget';
diff --git a/blocksuite/blocks/src/root-block/widgets/linked-doc/effects.ts b/blocksuite/blocks/src/root-block/widgets/linked-doc/effects.ts
index dda5747aba..2496ea4926 100644
--- a/blocksuite/blocks/src/root-block/widgets/linked-doc/effects.ts
+++ b/blocksuite/blocks/src/root-block/widgets/linked-doc/effects.ts
@@ -1,5 +1,6 @@
+import { AFFINE_LINKED_DOC_WIDGET } from './config.js';
 import { ImportDoc } from './import-doc/import-doc.js';
-import { AFFINE_LINKED_DOC_WIDGET, AffineLinkedDocWidget } from './index.js';
+import { AffineLinkedDocWidget } from './index.js';
 import { LinkedDocPopover } from './linked-doc-popover.js';
 import { AffineMobileLinkedDocMenu } from './mobile-linked-doc-menu.js';
 
diff --git a/blocksuite/blocks/src/root-block/widgets/linked-doc/index.ts b/blocksuite/blocks/src/root-block/widgets/linked-doc/index.ts
index c116bc2e10..ee2c759e97 100644
--- a/blocksuite/blocks/src/root-block/widgets/linked-doc/index.ts
+++ b/blocksuite/blocks/src/root-block/widgets/linked-doc/index.ts
@@ -20,8 +20,12 @@ import { choose } from 'lit/directives/choose.js';
 import { repeat } from 'lit/directives/repeat.js';
 import { styleMap } from 'lit/directives/style-map.js';
 
-import type { PageRootBlockComponent } from '../../index.js';
 import {
+  type PageRootBlockComponent,
+  RootBlockConfigExtension,
+} from '../../index.js';
+import {
+  type AFFINE_LINKED_DOC_WIDGET,
   getMenus,
   type LinkedDocContext,
   type LinkedWidgetConfig,
@@ -29,8 +33,6 @@ import {
 import { linkedDocWidgetStyles } from './styles.js';
 export { type LinkedWidgetConfig } from './config.js';
 
-export const AFFINE_LINKED_DOC_WIDGET = 'affine-linked-doc-widget';
-
 export class AffineLinkedDocWidget extends WidgetComponent<
   RootBlockModel,
   PageRootBlockComponent
@@ -218,7 +220,8 @@ export class AffineLinkedDocWidget extends WidgetComponent<
         scrollContainer: getViewportElement(this.std.host) ?? window,
         scrollTopOffset: 46,
       },
-      ...this.std.getConfig('affine:page')?.linkedWidget,
+      ...this.std.getOptional(RootBlockConfigExtension.identifier)
+        ?.linkedWidget,
     };
   }
 
diff --git a/blocksuite/framework/block-std/src/__tests__/test-schema.ts b/blocksuite/framework/block-std/src/__tests__/test-schema.ts
index e97e43e9ed..39f7446d18 100644
--- a/blocksuite/framework/block-std/src/__tests__/test-schema.ts
+++ b/blocksuite/framework/block-std/src/__tests__/test-schema.ts
@@ -50,13 +50,3 @@ export const HeadingBlockSchema = defineBlockSchema({
 export class HeadingBlockModel extends BlockModel<
   ReturnType<(typeof HeadingBlockSchema)['model']['props']>
 > {}
-
-declare global {
-  namespace BlockSuite {
-    interface BlockModels {
-      'test:page': RootBlockModel;
-      'test:note': NoteBlockModel;
-      'test:heading': HeadingBlockModel;
-    }
-  }
-}
diff --git a/blocksuite/framework/block-std/src/extension/config.ts b/blocksuite/framework/block-std/src/extension/config.ts
index de310041ea..9be9b0864f 100644
--- a/blocksuite/framework/block-std/src/extension/config.ts
+++ b/blocksuite/framework/block-std/src/extension/config.ts
@@ -1,3 +1,4 @@
+import type { ServiceIdentifier } from '@blocksuite/global/di';
 import type { ExtensionType } from '@blocksuite/store';
 
 import { ConfigIdentifier } from '../identifier.js';
@@ -15,17 +16,25 @@ import { ConfigIdentifier } from '../identifier.js';
  *
  * @example
  * ```ts
- * import { ConfigExtension } from '@blocksuite/block-std';
- * const MyConfigExtension = ConfigExtension('my-flavour', config);
+ * import { ConfigExtensionFactory } from '@blocksuite/block-std';
+ * const MyConfigExtensionFactory = ConfigExtensionFactory<ConfigType>('my-flavour');
+ * const MyConfigExtension = MyConfigExtensionFactory({
+ *   option1: 'value1',
+ *   option2: 'value2',
+ * });
  * ```
  */
-export function ConfigExtension(
-  flavor: string,
-  config: Record<string, unknown>
-): ExtensionType {
-  return {
+export function ConfigExtensionFactory<Config extends Record<string, any>>(
+  flavor: string
+): ((config: Config) => ExtensionType) & {
+  identifier: ServiceIdentifier<Config>;
+} {
+  const identifier = ConfigIdentifier(flavor) as ServiceIdentifier<Config>;
+  const extensionFactory = (config: Config): ExtensionType => ({
     setup: di => {
       di.addImpl(ConfigIdentifier(flavor), () => config);
     },
-  };
+  });
+  extensionFactory.identifier = identifier;
+  return extensionFactory;
 }
diff --git a/blocksuite/framework/block-std/src/scope/block-std-scope.ts b/blocksuite/framework/block-std/src/scope/block-std-scope.ts
index f37b9b66b1..bf228e9928 100644
--- a/blocksuite/framework/block-std/src/scope/block-std-scope.ts
+++ b/blocksuite/framework/block-std/src/scope/block-std-scope.ts
@@ -18,7 +18,6 @@ import { SurfaceMiddlewareExtension } from '../gfx/surface-middleware.js';
 import { ViewManager } from '../gfx/view/view-manager.js';
 import {
   BlockViewIdentifier,
-  ConfigIdentifier,
   LifeCycleWatcherIdentifier,
   StdIdentifier,
 } from '../identifier.js';
@@ -137,19 +136,6 @@ export class BlockStdScope {
     });
   }
 
-  getConfig<Key extends BlockSuite.ConfigKeys>(
-    flavour: Key
-  ): BlockSuite.BlockConfigs[Key] | null;
-
-  getConfig(flavour: string) {
-    const config = this.provider.getOptional(ConfigIdentifier(flavour));
-    if (!config) {
-      return null;
-    }
-
-    return config;
-  }
-
   getView(flavour: string) {
     return this.getOptional(BlockViewIdentifier(flavour));
   }
@@ -191,11 +177,3 @@ export class BlockStdScope {
     });
   }
 }
-
-declare global {
-  namespace BlockSuite {
-    interface BlockConfigs {}
-
-    type ConfigKeys = string & keyof BlockConfigs;
-  }
-}
diff --git a/blocksuite/framework/store/src/__tests__/collection.unit.spec.ts b/blocksuite/framework/store/src/__tests__/collection.unit.spec.ts
index 8341d563b2..41da956274 100644
--- a/blocksuite/framework/store/src/__tests__/collection.unit.spec.ts
+++ b/blocksuite/framework/store/src/__tests__/collection.unit.spec.ts
@@ -859,13 +859,3 @@ describe('getBlock', () => {
     assert.equal(invalid, null);
   });
 });
-
-declare global {
-  namespace BlockSuite {
-    interface BlockModels {
-      'affine:page': BlockModel;
-      'affine:paragraph': BlockModel;
-      'affine:note': BlockModel;
-    }
-  }
-}
diff --git a/blocksuite/framework/store/src/__tests__/schema.unit.spec.ts b/blocksuite/framework/store/src/__tests__/schema.unit.spec.ts
index e2bdebd243..4343f328f0 100644
--- a/blocksuite/framework/store/src/__tests__/schema.unit.spec.ts
+++ b/blocksuite/framework/store/src/__tests__/schema.unit.spec.ts
@@ -1,7 +1,6 @@
 import { literal } from 'lit/static-html.js';
 import { describe, expect, it, vi } from 'vitest';
 
-import type { BlockModel } from '../model/block/block-model.js';
 import { defineBlockSchema } from '../model/block/zod.js';
 // import some blocks
 import { SchemaValidateError } from '../schema/error.js';
@@ -124,12 +123,3 @@ describe('schema', () => {
     });
   });
 });
-
-declare global {
-  namespace BlockSuite {
-    interface BlockModels {
-      'affine:note-block-video': BlockModel;
-      'affine:note-invalid-block-video': BlockModel;
-    }
-  }
-}
diff --git a/blocksuite/framework/store/src/__tests__/transformer.unit.spec.ts b/blocksuite/framework/store/src/__tests__/transformer.unit.spec.ts
index 05cc675ab7..3e98474ecf 100644
--- a/blocksuite/framework/store/src/__tests__/transformer.unit.spec.ts
+++ b/blocksuite/framework/store/src/__tests__/transformer.unit.spec.ts
@@ -133,11 +133,3 @@ test('snapshot to model', async () => {
     expect(item.content.toString()).toBe(`item ${index + 1}`);
   });
 });
-
-declare global {
-  namespace BlockSuite {
-    interface BlockModels {
-      page: BlockModel;
-    }
-  }
-}
diff --git a/blocksuite/tests-legacy/utils/actions/drag.ts b/blocksuite/tests-legacy/utils/actions/drag.ts
index c824f4f544..0c58cc1fb5 100644
--- a/blocksuite/tests-legacy/utils/actions/drag.ts
+++ b/blocksuite/tests-legacy/utils/actions/drag.ts
@@ -1,6 +1,6 @@
 import type { Page } from '@playwright/test';
-import { assertImageOption } from 'utils/asserts.js';
 
+import { assertImageOption } from '../asserts.js';
 import { getIndexCoordinate, waitNextFrame } from './misc.js';
 
 export async function dragBetweenCoords(
diff --git a/packages/frontend/core/src/components/blocksuite/block-suite-editor/specs/custom/root-block.ts b/packages/frontend/core/src/components/blocksuite/block-suite-editor/specs/custom/root-block.ts
index b7e84d7387..ed7c53df9d 100644
--- a/packages/frontend/core/src/components/blocksuite/block-suite-editor/specs/custom/root-block.ts
+++ b/packages/frontend/core/src/components/blocksuite/block-suite-editor/specs/custom/root-block.ts
@@ -9,15 +9,10 @@ import { DocDisplayMetaService } from '@affine/core/modules/doc-display-meta';
 import { EditorSettingService } from '@affine/core/modules/editor-setting';
 import { AppThemeService } from '@affine/core/modules/theme';
 import { mixpanel } from '@affine/track';
-import {
-  ConfigExtension,
-  LifeCycleWatcher,
-  StdIdentifier,
-} from '@blocksuite/affine/block-std';
+import { LifeCycleWatcher, StdIdentifier } from '@blocksuite/affine/block-std';
 import type {
   DocDisplayMetaExtension,
   DocDisplayMetaParams,
-  RootBlockConfig,
   Signal,
   SpecBuilder,
   TelemetryEventMap,
@@ -27,14 +22,17 @@ import {
   CodeBlockSpec,
   ColorScheme,
   createSignalFromObservable,
+  DatabaseConfigExtension,
   DocDisplayMetaProvider,
   EditorSettingExtension,
   ImageBlockSpec,
   ParagraphBlockSpec,
   referenceToNode,
+  RootBlockConfigExtension,
   SpecProvider,
   TelemetryProvider,
   ThemeExtensionIdentifier,
+  ToolbarMoreMenuConfigExtension,
 } from '@blocksuite/affine/blocks';
 import type { Container } from '@blocksuite/affine/global/di';
 import type { ExtensionType } from '@blocksuite/affine/store';
@@ -231,11 +229,11 @@ function getEditorConfigExtension(
   const editorSettingService = framework.get(EditorSettingService);
   return [
     EditorSettingExtension(editorSettingService.editorSetting.settingSignal),
-    ConfigExtension('affine:database', createDatabaseOptionsConfig(framework)),
-    ConfigExtension('affine:page', {
+    DatabaseConfigExtension(createDatabaseOptionsConfig(framework)),
+    RootBlockConfigExtension({
       linkedWidget: createLinkedWidgetConfig(framework),
-      toolbarMoreMenu: createToolbarMoreMenuConfig(framework),
-    } satisfies RootBlockConfig),
+    }),
+    ToolbarMoreMenuConfigExtension(createToolbarMoreMenuConfig(framework)),
   ];
 }