fix(core): use patched preview spec builder in ai chat (#10090)

[BS-2526](https://linear.app/affine-design/issue/BS-2526/chat-panel-里的-footnote-popup-需要支持交互)
2026-08-01 09:30:01 +08:00 · 2025-02-11 15:11:54 +00:00
parent 54d194afe7
commit 19f0eb1931
11 changed files with 227 additions and 27 deletions
@@ -1,7 +1,27 @@
 import { AIChatBlockSpec } from '@affine/core/blocksuite/presets/blocks/ai-chat-block';
-import { SpecProvider } from '@blocksuite/affine/blocks';
+import { PeekViewService } from '@affine/core/modules/peek-view';
+import { AppThemeService } from '@affine/core/modules/theme';
+import {
+  type BlockStdScope,
+  LifeCycleWatcher,
+  StdIdentifier,
+} from '@blocksuite/affine/block-std';
+import {
+  ColorScheme,
+  createSignalFromObservable,
+  type Signal,
+  type SpecBuilder,
+  SpecProvider,
+  type ThemeExtension,
+  ThemeExtensionIdentifier,
+} from '@blocksuite/affine/blocks';
+import type { Container } from '@blocksuite/affine/global/di';
 import type { ExtensionType } from '@blocksuite/affine/store';
+import type { FrameworkProvider } from '@toeverything/infra';
+import type { Observable } from 'rxjs';

+import { buildDocDisplayMetaExtension } from './custom/root-block';
+import { patchPeekViewService } from './custom/spec-patchers';
 import { getFontConfigExtension } from './font-extension';

 const CustomSpecs: ExtensionType[] = [
@@ -18,3 +38,73 @@ export function effects() {
  // Patch edgeless preview spec for blocksuite surface-ref and embed-synced-doc
  patchPreviewSpec('edgeless:preview', CustomSpecs);
 }
+
+export function getPagePreviewThemeExtension(framework: FrameworkProvider) {
+  class AffinePagePreviewThemeExtension
+    extends LifeCycleWatcher
+    implements ThemeExtension
+  {
+    static override readonly key = 'affine-page-preview-theme';
+
+    readonly theme: Signal<ColorScheme>;
+
+    readonly disposables: (() => void)[] = [];
+
+    static override setup(di: Container) {
+      super.setup(di);
+      di.override(ThemeExtensionIdentifier, AffinePagePreviewThemeExtension, [
+        StdIdentifier,
+      ]);
+    }
+
+    constructor(std: BlockStdScope) {
+      super(std);
+      const theme$: Observable<ColorScheme> = framework
+        .get(AppThemeService)
+        .appTheme.theme$.map(theme => {
+          return theme === ColorScheme.Dark
+            ? ColorScheme.Dark
+            : ColorScheme.Light;
+        });
+      const { signal, cleanup } = createSignalFromObservable<ColorScheme>(
+        theme$,
+        ColorScheme.Light
+      );
+      this.theme = signal;
+      this.disposables.push(cleanup);
+    }
+
+    getAppTheme() {
+      return this.theme;
+    }
+
+    getEdgelessTheme() {
+      return this.theme;
+    }
+
+    override unmounted() {
+      this.dispose();
+    }
+
+    dispose() {
+      this.disposables.forEach(dispose => dispose());
+    }
+  }
+
+  return AffinePagePreviewThemeExtension;
+}
+
+export function createPageModePreviewSpecs(
+  framework: FrameworkProvider
+): SpecBuilder {
+  const specProvider = SpecProvider.getInstance();
+  const pagePreviewSpec = specProvider.getSpec('page:preview');
+  // Enable theme extension, doc display meta extension and peek view service
+  const peekViewService = framework.get(PeekViewService);
+  pagePreviewSpec.extend([
+    getPagePreviewThemeExtension(framework),
+    buildDocDisplayMetaExtension(framework),
+    patchPeekViewService(peekViewService),
+  ]);
+  return pagePreviewSpec;
+}