Jayden/AFFiNE-Mirror
0
0
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2026-09-22 03:33:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(editor): ai slash menu config extension (#10680)

Browse Source
This commit is contained in:
L-Sun
2025-03-07 09:05:03 +00:00
parent 264f0dd2be
commit d37868d97a
9 changed files with 26 additions and 165 deletions
Download Patch File Download Diff File Expand all files Collapse all files
blocksuite/tests-legacy/e2e/slash-menu.spec.ts
-124
View File
@@ -1,4 +1,3 @@
import type { SlashMenuActionItem } from '@blocksuite/blocks';
import { expect } from '@playwright/test';
import { addNote, switchEditorMode } from './utils/actions/edgeless.js';
@@ -762,129 +761,6 @@ test('should insert database', async ({ page }) => {
expect(await defaultRows.count()).toBe(3);
});
// TODO(@L-Sun): Refactor this test after refactoring the slash menu
test.describe('slash menu with customize menu', () => {
test('can remove specified menus', async ({ page }) => {
await enterPlaygroundRoom(page);
await initEmptyParagraphState(page);
await page.evaluate(async () => {
// https://github.com/lit/lit/blob/84df6ef8c73fffec92384891b4b031d7efc01a64/packages/lit-html/src/static.ts#L93
const fakeLiteral = (strings: TemplateStringsArray) =>
({
['_$litStatic$']: strings[0],
r: Symbol.for(''),
}) as const;
const editor = document.querySelector('affine-editor-container');
if (!editor) throw new Error("Can't find affine-editor-container");
const SlashMenuWidget = window.$blocksuite.blocks.AffineSlashMenuWidget;
class CustomSlashMenu extends SlashMenuWidget {
override get config() {
return {
items: super.config.items
.filter(item => 'action' in item)
.slice(0, 5)
.map<SlashMenuActionItem>((item, index) => ({
...item,
group: `0_custom-group@${index++}`,
})),
};
}
}
// Fix `Illegal constructor` error
// see https://stackoverflow.com/questions/41521812/illegal-constructor-with-ecmascript-6
customElements.define('affine-custom-slash-menu', CustomSlashMenu);
const pageSpecs = window.$blocksuite.blocks.PageEditorBlockSpecs;
editor.pageSpecs = [
...pageSpecs,
{
setup: di => {
di.override(
window.$blocksuite.blockStd.WidgetViewIdentifier(
'affine:page|affine-slash-menu-widget'
),
// @ts-ignore
fakeLiteral`affine-custom-slash-menu`
);
},
},
];
await editor.updateComplete;
});
await focusRichText(page);
const slashMenu = page.locator(`.slash-menu`);
const slashItems = slashMenu.locator('icon-button');
await type(page, '/');
await expect(slashMenu).toBeVisible();
await expect(slashItems).toHaveCount(5);
});
test('can add some menus', async ({ page }) => {
await enterPlaygroundRoom(page);
await initEmptyParagraphState(page);
await page.evaluate(async () => {
// https://github.com/lit/lit/blob/84df6ef8c73fffec92384891b4b031d7efc01a64/packages/lit-html/src/static.ts#L93
// eslint-disable-next-line sonarjs/no-identical-functions
const fakeLiteral = (strings: TemplateStringsArray) =>
({
['_$litStatic$']: strings[0],
r: Symbol.for(''),
}) as const;
const editor = document.querySelector('affine-editor-container');
if (!editor) throw new Error("Can't find affine-editor-container");
const SlashMenuWidget = window.$blocksuite.blocks.AffineSlashMenuWidget;
class CustomSlashMenu extends SlashMenuWidget {
override get config() {
return {
items: [
{
name: 'Custom Menu Item',
group: '0_custom-group@0',
action: () => {},
} satisfies SlashMenuActionItem,
],
};
}
}
// Fix `Illegal constructor` error
// see https://stackoverflow.com/questions/41521812/illegal-constructor-with-ecmascript-6
customElements.define('affine-custom-slash-menu', CustomSlashMenu);
const pageSpecs = window.$blocksuite.blocks.PageEditorBlockSpecs;
editor.pageSpecs = [
...pageSpecs,
{
setup: di =>
di.override(
window.$blocksuite.blockStd.WidgetViewIdentifier(
'affine:page|affine-slash-menu-widget'
),
// @ts-ignore
fakeLiteral`affine-custom-slash-menu`
),
},
];
await editor.updateComplete;
});
await focusRichText(page);
const slashMenu = page.locator(`.slash-menu`);
const slashItems = slashMenu.locator('icon-button');
await type(page, '/');
await expect(slashMenu).toBeVisible();
await expect(slashItems).toHaveCount(1);
});
});
test('move block up and down by slash menu', async ({ page }) => {
await enterPlaygroundRoom(page);
await initEmptyParagraphState(page);