Jayden/AFFiNE-Mirror
1
0
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2026-02-12 04:18:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(hooks): missing page preview and references (#4863)

Browse Source
This commit is contained in:
EYHN
2023-11-09 16:59:36 +08:00
committed by GitHub
parent 248fb1fa69
commit e3e0553c56
2 changed files with 21 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
packages/frontend/hooks/src/use-block-suite-page-preview.ts
View File

@@ -25,14 +25,19 @@ export function useBlockSuitePagePreview(page: Page | null): Atom<string> {
} else if (weakMap.has(page)) {
return weakMap.get(page) as Atom<string>;
} else {
const baseAtom = atom<string>(getPagePreviewText(page));
const baseAtom = atom<string>('');
baseAtom.onMount = set => {
const disposable = page.slots.yUpdated.on(() => {
set(getPagePreviewText(page));
});
const disposables = [
page.slots.ready.on(() => {
set(getPagePreviewText(page));
}),
page.slots.yUpdated.on(() => {
set(getPagePreviewText(page));
}),
];
set(getPagePreviewText(page));
return () => {
disposable.dispose();
disposables.forEach(disposable => disposable.dispose());
};
};
weakMap.set(page, baseAtom);

16
packages/frontend/hooks/src/use-block-suite-page-references.ts
View File

@@ -19,13 +19,19 @@ const getPageReferencesAtom = (page: Page | null) => {
}
if (!weakMap.has(page)) {
const baseAtom = atom<string[]>(getPageReferences(page));
const baseAtom = atom<string[]>([]);
baseAtom.onMount = set => {
const dispose = page.slots.yUpdated.on(() => {
set(getPageReferences(page));
});
const disposables = [
page.slots.ready.on(() => {
set(getPageReferences(page));
}),
page.slots.yUpdated.on(() => {
set(getPageReferences(page));
}),
];
set(getPageReferences(page));
return () => {
dispose.dispose();
disposables.forEach(disposable => disposable.dispose());
};
};
weakMap.set(page, baseAtom);