refactor(core): add useAIChatConfig hook (#11424)

Close [BS-2583](https://linear.app/affine-design/issue/BS-2583).
2026-07-19 11:06:25 +08:00 · 2025-04-03 14:53:49 +00:00
parent 363476a46c
commit 6033baeb86
6 changed files with 175 additions and 101 deletions
@@ -0,0 +1,115 @@
+// packages/frontend/core/src/blocksuite/ai/hooks/useChatPanelConfig.ts
+import { AINetworkSearchService } from '@affine/core/modules/ai-button/services/network-search';
+import { CollectionService } from '@affine/core/modules/collection';
+import { DocsService } from '@affine/core/modules/doc';
+import { DocDisplayMetaService } from '@affine/core/modules/doc-display-meta';
+import { DocsSearchService } from '@affine/core/modules/docs-search';
+import {
+  type SearchCollectionMenuAction,
+  type SearchDocMenuAction,
+  SearchMenuService,
+  type SearchTagMenuAction,
+} from '@affine/core/modules/search-menu/services';
+import { TagService } from '@affine/core/modules/tag';
+import { WorkspaceService } from '@affine/core/modules/workspace';
+import { createSignalFromObservable } from '@blocksuite/affine/shared/utils';
+import { useFramework } from '@toeverything/infra';
+
+export function useAIChatConfig() {
+  const framework = useFramework();
+
+  const searchService = framework.get(AINetworkSearchService);
+  const docDisplayMetaService = framework.get(DocDisplayMetaService);
+  const workspaceService = framework.get(WorkspaceService);
+  const searchMenuService = framework.get(SearchMenuService);
+  const docsSearchService = framework.get(DocsSearchService);
+  const tagService = framework.get(TagService);
+  const collectionService = framework.get(CollectionService);
+  const docsService = framework.get(DocsService);
+
+  const networkSearchConfig = {
+    visible: searchService.visible,
+    enabled: searchService.enabled,
+    setEnabled: searchService.setEnabled,
+  };
+
+  const docDisplayConfig = {
+    getIcon: (docId: string) => {
+      return docDisplayMetaService.icon$(docId, { type: 'lit' }).value;
+    },
+    getTitle: (docId: string) => {
+      return docDisplayMetaService.title$(docId).value;
+    },
+    getTitleSignal: (docId: string) => {
+      const title$ = docDisplayMetaService.title$(docId);
+      return createSignalFromObservable(title$, '');
+    },
+    getDocMeta: (docId: string) => {
+      const docRecord = docsService.list.doc$(docId).value;
+      return docRecord?.meta$.value ?? null;
+    },
+    getDocPrimaryMode: (docId: string) => {
+      const docRecord = docsService.list.doc$(docId).value;
+      return docRecord?.primaryMode$.value ?? 'page';
+    },
+    getDoc: (docId: string) => {
+      const doc = workspaceService.workspace.docCollection.getDoc(docId);
+      return doc?.getStore() ?? null;
+    },
+    getReferenceDocs: (docIds: string[]) => {
+      const docs$ = docsSearchService.watchRefsFrom(docIds);
+      return createSignalFromObservable(docs$, []);
+    },
+    getTags: () => {
+      const tagMetas$ = tagService.tagList.tagMetas$;
+      return createSignalFromObservable(tagMetas$, []);
+    },
+    getTagTitle: (tagId: string) => {
+      const tag$ = tagService.tagList.tagByTagId$(tagId);
+      return tag$.value?.value$.value ?? '';
+    },
+    getTagPageIds: (tagId: string) => {
+      const tag$ = tagService.tagList.tagByTagId$(tagId);
+      if (!tag$) return [];
+      return tag$.value?.pageIds$.value ?? [];
+    },
+    getCollections: () => {
+      const collections$ = collectionService.collections$;
+      return createSignalFromObservable(collections$, []);
+    },
+  };
+
+  const searchMenuConfig = {
+    getDocMenuGroup: (
+      query: string,
+      action: SearchDocMenuAction,
+      abortSignal: AbortSignal
+    ) => {
+      return searchMenuService.getDocMenuGroup(query, action, abortSignal);
+    },
+    getTagMenuGroup: (
+      query: string,
+      action: SearchTagMenuAction,
+      abortSignal: AbortSignal
+    ) => {
+      return searchMenuService.getTagMenuGroup(query, action, abortSignal);
+    },
+    getCollectionMenuGroup: (
+      query: string,
+      action: SearchCollectionMenuAction,
+      abortSignal: AbortSignal
+    ) => {
+      return searchMenuService.getCollectionMenuGroup(
+        query,
+        action,
+        abortSignal
+      );
+    },
+  };
+
+  return {
+    networkSearchConfig,
+    docDisplayConfig,
+    searchMenuConfig,
+  };
+}