Jayden/AFFiNE-Mirror
0
0
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2026-07-22 12:36:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(core): use new backlink indexer (#7296)

Browse Source
This commit is contained in:
EYHN
2024-07-02 09:18:01 +00:00
parent 40e381e272
commit 27d0fc5108
33 changed files with 826 additions and 357 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/frontend/core/src/hooks/use-block-suite-page-backlinks.ts
-49
View File
@@ -1,49 +0,0 @@
import type { Doc, DocCollection } from '@blocksuite/store';
import type { Atom } from 'jotai';
import { atom, useAtomValue } from 'jotai';
import { useDocCollectionPage } from './use-block-suite-workspace-page';
const weakMap = new WeakMap<Doc, Atom<string[]>>();
function getPageBacklinks(page: Doc): string[] {
return (
page.collection.indexer.backlink
?.getBacklink(page.id)
.map(linkNode => linkNode.pageId)
.filter(id => id !== page.id) ?? []
);
}
const getPageBacklinksAtom = (page: Doc | null) => {
if (!page) {
return atom([]);
}
if (!weakMap.has(page)) {
const baseAtom = atom<string[]>([]);
baseAtom.onMount = set => {
const disposables = [
page.slots.ready.on(() => {
set(getPageBacklinks(page));
}),
page.collection.indexer.backlink?.slots.indexUpdated.on(() => {
set(getPageBacklinks(page));
}),
];
set(getPageBacklinks(page));
return () => {
disposables.forEach(disposable => disposable?.dispose());
};
};
weakMap.set(page, baseAtom);
}
return weakMap.get(page) as Atom<string[]>;
};
export function useBlockSuitePageBacklinks(
docCollection: DocCollection,
docId: string
): string[] {
const doc = useDocCollectionPage(docCollection, docId);
return useAtomValue(getPageBacklinksAtom(doc));
}
packages/frontend/core/src/hooks/use-block-suite-page-references.ts
-46
View File
@@ -1,46 +0,0 @@
import type { Doc, DocCollection } from '@blocksuite/store';
import type { Atom } from 'jotai';
import { atom, useAtomValue } from 'jotai';
import { useDocCollectionPage } from './use-block-suite-workspace-page';
const weakMap = new WeakMap<Doc, Atom<string[]>>();
function getPageReferences(page: Doc): string[] {
return Object.values(
page.collection.indexer.backlink?.linkIndexMap[page.id] ?? {}
).flatMap(linkNodes => linkNodes.map(linkNode => linkNode.pageId));
}
const getPageReferencesAtom = (page: Doc | null) => {
if (!page) {
return atom([]);
}
if (!weakMap.has(page)) {
const baseAtom = atom<string[]>([]);
baseAtom.onMount = set => {
const disposables = [
page.slots.ready.on(() => {
set(getPageReferences(page));
}),
page.collection.indexer.backlink?.slots.indexUpdated.on(() => {
set(getPageReferences(page));
}),
];
set(getPageReferences(page));
return () => {
disposables.forEach(disposable => disposable?.dispose());
};
};
weakMap.set(page, baseAtom);
}
return weakMap.get(page) as Atom<string[]>;
};
export function useBlockSuitePageReferences(
docCollection: DocCollection,
pageId: string
): string[] {
const page = useDocCollectionPage(docCollection, pageId);
return useAtomValue(getPageReferencesAtom(page));
}