feat(editor): note extension (#11856)

Closes: BS-3200
Closes: BS-3201

blocksuite/affine/blocks/note/package.json

@@ -13,6 +13,7 @@
     "@blocksuite/affine-block-embed": "workspace:*",
     "@blocksuite/affine-block-surface": "workspace:*",
     "@blocksuite/affine-components": "workspace:*",
+    "@blocksuite/affine-ext-loader": "workspace:*",
     "@blocksuite/affine-fragment-doc-title": "workspace:*",
     "@blocksuite/affine-gfx-turbo-renderer": "workspace:*",
     "@blocksuite/affine-inline-preset": "workspace:*",
@@ -39,7 +40,9 @@
   "exports": {
     ".": "./src/index.ts",
     "./effects": "./src/effects.ts",
-    "./turbo-painter": "./src/turbo/note-painter.worker.ts"
+    "./turbo-painter": "./src/turbo/note-painter.worker.ts",
+    "./store": "./src/store.ts",
+    "./view": "./src/view.ts"
   },
   "files": [
     "src",

blocksuite/affine/blocks/note/src/store.ts

@@ -0,0 +1,36 @@
+import {
+  type StoreExtensionContext,
+  StoreExtensionProvider,
+} from '@blocksuite/affine-ext-loader';
+import { NoteBlockSchemaExtension } from '@blocksuite/affine-model';
+import { z } from 'zod';
+
+import {
+  DocNoteBlockAdapterExtensions,
+  EdgelessNoteBlockAdapterExtensions,
+} from './adapters';
+
+const optionsSchema = z.object({
+  mode: z.enum(['doc', 'edgeless']).optional(),
+});
+
+export class NoteStoreExtension extends StoreExtensionProvider<
+  z.infer<typeof optionsSchema>
+> {
+  override name = 'affine-note-block';
+
+  override schema = optionsSchema;
+
+  override setup(
+    context: StoreExtensionContext,
+    options?: z.infer<typeof optionsSchema>
+  ) {
+    super.setup(context);
+    context.register(NoteBlockSchemaExtension);
+    if (options?.mode === 'edgeless') {
+      context.register(EdgelessNoteBlockAdapterExtensions);
+    } else {
+      context.register(DocNoteBlockAdapterExtensions);
+    }
+  }
+}

blocksuite/affine/blocks/note/src/view.ts

@@ -0,0 +1,47 @@
+import {
+  type ViewExtensionContext,
+  ViewExtensionProvider,
+} from '@blocksuite/affine-ext-loader';
+import { NoteBlockSchema } from '@blocksuite/affine-model';
+import { BlockViewExtension, FlavourExtension } from '@blocksuite/std';
+import { literal } from 'lit/static-html.js';
+
+import { NoteSlashMenuConfigExtension } from './configs/slash-menu';
+import { createBuiltinToolbarConfigExtension } from './configs/toolbar';
+import { effects } from './effects';
+import { NoteKeymapExtension } from './note-keymap';
+
+const flavour = NoteBlockSchema.model.flavour;
+
+export class NoteViewExtension extends ViewExtensionProvider {
+  override name = 'affine-note-block';
+
+  override effect() {
+    super.effect();
+    effects();
+  }
+
+  override setup(context: ViewExtensionContext) {
+    super.setup(context);
+    context.register([
+      FlavourExtension(flavour),
+      NoteSlashMenuConfigExtension,
+      NoteKeymapExtension,
+    ]);
+
+    const scope = context.scope;
+    const isEdgeless =
+      scope === 'edgeless' ||
+      scope === 'preview-edgeless' ||
+      scope === 'mobile-edgeless';
+
+    if (isEdgeless) {
+      context.register(
+        BlockViewExtension(flavour, literal`affine-edgeless-note`)
+      );
+      context.register(createBuiltinToolbarConfigExtension(flavour));
+    } else {
+      context.register(BlockViewExtension(flavour, literal`affine-note`));
+    }
+  }
+}

blocksuite/affine/blocks/note/tsconfig.json

@@ -10,6 +10,7 @@
     { "path": "../embed" },
     { "path": "../surface" },
     { "path": "../../components" },
+    { "path": "../../ext-loader" },
     { "path": "../../fragments/doc-title" },
     { "path": "../../gfx/turbo-renderer" },
     { "path": "../../inlines/preset" },