Jayden/AFFiNE-Mirror
1
0
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2026-02-13 21:05:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(editor): linked doc slash menu config extension (#10682)

Browse Source
This commit is contained in:
L-Sun
2025-03-07 14:42:56 +00:00
parent 9f3580263a
commit 63bf2b80c1
18 changed files with 220 additions and 178 deletions
Download Patch File Download Diff File Expand all files Collapse all files

109
packages/frontend/core/src/blocksuite/extensions/quick-search-service.ts
View File

@@ -10,14 +10,19 @@ import {
import { ExternalLinksQuickSearchSession } from '@affine/core/modules/quicksearch/impls/external-links';
import { JournalsQuickSearchSession } from '@affine/core/modules/quicksearch/impls/journals';
import { track } from '@affine/track';
import { LifeCycleWatcher } from '@blocksuite/affine/block-std';
import type { QuickSearchResult } from '@blocksuite/affine/blocks';
import type {
QuickSearchResult,
SlashMenuConfig,
SlashMenuItem,
} from '@blocksuite/affine/blocks';
import {
AffineSlashMenuWidget,
BookmarkSlashMenuConfigIdentifier,
insertLinkByQuickSearchCommand,
LinkedDocSlashMenuConfigIdentifier,
QuickSearchExtension,
} from '@blocksuite/affine/blocks';
import { Text } from '@blocksuite/affine/store';
import type { ServiceIdentifier } from '@blocksuite/affine/global/di';
import { type ExtensionType, Text } from '@blocksuite/affine/store';
import type { FrameworkProvider } from '@toeverything/infra';
import { pick } from 'lodash-es';
@@ -120,56 +125,66 @@ export function patchQuickSearchService(framework: FrameworkProvider) {
},
});
class SlashMenuQuickSearchExtension extends LifeCycleWatcher {
static override key = 'slash-menu-quick-search-extension';
const SlashMenuQuickSearchExtension: ExtensionType = {
setup: di => {
const overrideFn = (identifier: ServiceIdentifier<SlashMenuConfig>) => {
const prev = di.getFactory(identifier);
if (!prev) return;
override mounted() {
super.mounted();
const { view } = this.std;
view.viewUpdated.on(payload => {
if (payload.type !== 'widget' || payload.method !== 'add') {
return;
}
const component = payload.view;
if (component instanceof AffineSlashMenuWidget) {
component.configItemTransform = item => {
if (
'action' in item &&
(item.name === 'Linked Doc' || item.name === 'Link')
) {
item.action = ({ std }) => {
const [success, { insertedLinkType }] = std.command.exec(
insertLinkByQuickSearchCommand
);
di.override(identifier, provider => {
const prevConfig: SlashMenuConfig = prev(provider);
if (!success) return;
if (typeof prevConfig.items === 'function') {
const prevConfigItemGenerator = prevConfig.items;
prevConfig.items = ctx =>
prevConfigItemGenerator(ctx).map(modifyFn);
} else {
prevConfig.items = prevConfig.items.map(modifyFn);
}
insertedLinkType
?.then(type => {
const flavour = type?.flavour;
if (!flavour) return;
return prevConfig;
});
};
if (flavour === 'affine:bookmark') {
track.doc.editor.slashMenu.bookmark();
return;
}
overrideFn(LinkedDocSlashMenuConfigIdentifier);
overrideFn(BookmarkSlashMenuConfigIdentifier);
if (flavour === 'affine:embed-linked-doc') {
track.doc.editor.slashMenu.linkDoc({
control: 'linkDoc',
});
return;
}
})
.catch(console.error);
};
}
return item;
const modifyFn = (item: SlashMenuItem): SlashMenuItem => {
if (
'action' in item &&
(item.name === 'Linked Doc' || item.name === 'Link')
) {
item.action = ({ std }) => {
const [success, { insertedLinkType }] = std.command.exec(
insertLinkByQuickSearchCommand
);
if (!success) return;
insertedLinkType
?.then(type => {
const flavour = type?.flavour;
if (!flavour) return;
if (flavour === 'affine:bookmark') {
track.doc.editor.slashMenu.bookmark();
return;
}
if (flavour === 'affine:embed-linked-doc') {
track.doc.editor.slashMenu.linkDoc({
control: 'linkDoc',
});
return;
}
})
.catch(console.error);
};
}
});
}
}
return item;
};
},
};
return [QuickSearch, SlashMenuQuickSearchExtension];
}