refactor(editor): remove selection global types (#9532)

Closes: [BS-2217](https://linear.app/affine-design/issue/BS-2217/remove-global-types-in-selection)

blocksuite/affine/block-attachment/src/attachment-block.ts

@@ -13,6 +13,11 @@ import {
 } from '@blocksuite/affine-model';
 import { ThemeProvider } from '@blocksuite/affine-shared/services';
 import { humanFileSize } from '@blocksuite/affine-shared/utils';
+import {
+  BlockSelection,
+  SurfaceSelection,
+  TextSelection,
+} from '@blocksuite/block-std';
 import { Slice } from '@blocksuite/store';
 import { flip, offset } from '@floating-ui/dom';
 import { html, nothing } from 'lit';
@@ -44,7 +49,7 @@ export class AttachmentBlockComponent extends CaptionedBlockComponent<
     this,
     ({ abortController }) => {
       const selection = this.host.selection;
-      const textSelection = selection.find('text');
+      const textSelection = selection.find(TextSelection);
       if (
         !!textSelection &&
         (!!textSelection.to || !!textSelection.from.length)
@@ -52,7 +57,7 @@ export class AttachmentBlockComponent extends CaptionedBlockComponent<
         return null;
       }
 
-      const blockSelections = selection.filter('block');
+      const blockSelections = selection.filter(BlockSelection);
       if (
         blockSelections.length > 1 ||
         (blockSelections.length === 1 &&
@@ -126,7 +131,7 @@ export class AttachmentBlockComponent extends CaptionedBlockComponent<
 
   private _selectBlock() {
     const selectionManager = this.host.selection;
-    const blockSelection = selectionManager.create('block', {
+    const blockSelection = selectionManager.create(BlockSelection, {
       blockId: this.blockId,
     });
     selectionManager.setGroup('note', [blockSelection]);
@@ -167,7 +172,8 @@ export class AttachmentBlockComponent extends CaptionedBlockComponent<
     this.disposables.add(
       this.std.selection.slots.changed.on(() => {
         this._isSelected =
-          !!this.selected?.is('block') || !!this.selected?.is('surface');
+          !!this.selected?.is(BlockSelection) ||
+          !!this.selected?.is(SurfaceSelection);
 
         this._showOverlay =
           this._isResizing || this._isDragging || !this._isSelected;

blocksuite/affine/block-bookmark/src/bookmark-block.ts

@@ -4,6 +4,7 @@ import {
 } from '@blocksuite/affine-components/caption';
 import type { BookmarkBlockModel } from '@blocksuite/affine-model';
 import { DocModeProvider } from '@blocksuite/affine-shared/services';
+import { BlockSelection } from '@blocksuite/block-std';
 import { html } from 'lit';
 import { property, query } from 'lit/decorators.js';
 import { classMap } from 'lit/directives/class-map.js';
@@ -73,7 +74,7 @@ export class BookmarkBlockComponent extends CaptionedBlockComponent<
   }
 
   override renderBlock() {
-    const selected = !!this.selected?.is('block');
+    const selected = !!this.selected?.is(BlockSelection);
     return html`
       <div
         draggable="${this.blockDraggable ? 'true' : 'false'}"

blocksuite/affine/block-bookmark/src/components/bookmark-card.ts

@@ -2,7 +2,11 @@ import { getEmbedCardIcons } from '@blocksuite/affine-block-embed';
 import { WebIcon16 } from '@blocksuite/affine-components/icons';
 import { ThemeProvider } from '@blocksuite/affine-shared/services';
 import { getHostName } from '@blocksuite/affine-shared/utils';
-import { ShadowlessElement } from '@blocksuite/block-std';
+import {
+  BlockSelection,
+  ShadowlessElement,
+  SurfaceSelection,
+} from '@blocksuite/block-std';
 import { WithDisposable } from '@blocksuite/global/utils';
 import { OpenInNewIcon } from '@blocksuite/icons/lit';
 import { html } from 'lit';
@@ -31,7 +35,7 @@ export class BookmarkCard extends WithDisposable(ShadowlessElement) {
 
   private _selectBlock() {
     const selectionManager = this.bookmark.host.selection;
-    const blockSelection = selectionManager.create('block', {
+    const blockSelection = selectionManager.create(BlockSelection, {
       blockId: this.bookmark.blockId,
     });
     selectionManager.setGroup('note', [blockSelection]);
@@ -55,8 +59,8 @@ export class BookmarkCard extends WithDisposable(ShadowlessElement) {
     this.disposables.add(
       this.bookmark.selection.slots.changed.on(() => {
         this._isSelected =
-          !!this.bookmark.selected?.is('block') ||
+          !!this.bookmark.selected?.is(BlockSelection) ||
-          !!this.bookmark.selected?.is('surface');
+          !!this.bookmark.selected?.is(SurfaceSelection);
       })
     );
   }

blocksuite/affine/block-code/src/code-block.ts

@@ -11,7 +11,11 @@ import {
 } from '@blocksuite/affine-shared/services';
 import { getViewportElement } from '@blocksuite/affine-shared/utils';
 import type { BlockComponent } from '@blocksuite/block-std';
-import { getInlineRangeProvider } from '@blocksuite/block-std';
+import {
+  BlockSelection,
+  getInlineRangeProvider,
+  TextSelection,
+} from '@blocksuite/block-std';
 import { IS_MAC } from '@blocksuite/global/env';
 import { noop } from '@blocksuite/global/utils';
 import {
@@ -178,7 +182,7 @@ export class CodeBlockComponent extends CaptionedBlockComponent<
     this.bindHotKey({
       Backspace: ctx => {
         const state = ctx.get('keyboardState');
-        const textSelection = selectionManager.find('text');
+        const textSelection = selectionManager.find(TextSelection);
         if (!textSelection) {
           state.raw.preventDefault();
           return;
@@ -189,7 +193,7 @@ export class CodeBlockComponent extends CaptionedBlockComponent<
         if (from.index === 0 && from.length === 0) {
           state.raw.preventDefault();
           selectionManager.setGroup('note', [
-            selectionManager.create('block', { blockId: this.blockId }),
+            selectionManager.create(BlockSelection, { blockId: this.blockId }),
           ]);
           return true;
         }

blocksuite/affine/block-code/src/code-toolbar/config.ts

@@ -8,6 +8,7 @@ import {
 } from '@blocksuite/affine-components/icons';
 import type { MenuItemGroup } from '@blocksuite/affine-components/toolbar';
 import { isInsidePageEditor } from '@blocksuite/affine-shared/utils';
+import { BlockSelection } from '@blocksuite/block-std';
 import { noop, sleep } from '@blocksuite/global/utils';
 import { html } from 'lit';
 import { ifDefined } from 'lit/directives/if-defined.js';
@@ -134,7 +135,7 @@ export const clipboardGroup: MenuItemGroup<CodeBlockToolbarContext> = {
         host.updateComplete
           .then(() => {
             host.selection.setGroup('note', [
-              host.selection.create('block', {
+              host.selection.create(BlockSelection, {
                 blockId: codeId,
               }),
             ]);

blocksuite/affine/block-code/src/code-toolbar/index.ts

@@ -9,7 +9,11 @@ import {
 } from '@blocksuite/affine-components/toolbar';
 import type { CodeBlockModel } from '@blocksuite/affine-model';
 import { PAGE_HEADER_HEIGHT } from '@blocksuite/affine-shared/consts';
-import { WidgetComponent } from '@blocksuite/block-std';
+import {
+  BlockSelection,
+  TextSelection,
+  WidgetComponent,
+} from '@blocksuite/block-std';
 import { limitShift, shift } from '@floating-ui/dom';
 import { html } from 'lit';
 
@@ -34,7 +38,7 @@ export class AffineCodeToolbarWidget extends WidgetComponent<
         const codeBlock = this.block;
         const selection = this.host.selection;
 
-        const textSelection = selection.find('text');
+        const textSelection = selection.find(TextSelection);
         if (
           !!textSelection &&
           (!!textSelection.to || !!textSelection.from.length)
@@ -42,7 +46,7 @@ export class AffineCodeToolbarWidget extends WidgetComponent<
           return null;
         }
 
-        const blockSelections = selection.filter('block');
+        const blockSelections = selection.filter(BlockSelection);
         if (
           blockSelections.length > 1 ||
           (blockSelections.length === 1 &&

blocksuite/affine/block-divider/src/divider-block.ts

@@ -1,6 +1,7 @@
 import { CaptionedBlockComponent } from '@blocksuite/affine-components/caption';
 import type { DividerBlockModel } from '@blocksuite/affine-model';
 import { BLOCK_CHILDREN_CONTAINER_PADDING_LEFT } from '@blocksuite/affine-shared/consts';
+import { BlockSelection } from '@blocksuite/block-std';
 import { html } from 'lit';
 
 import { dividerBlockStyles } from './styles.js';
@@ -15,7 +16,7 @@ export class DividerBlockComponent extends CaptionedBlockComponent<DividerBlockM
 
     this.handleEvent('click', () => {
       this.host.selection.setGroup('note', [
-        this.host.selection.create('block', {
+        this.host.selection.create(BlockSelection, {
           blockId: this.blockId,
         }),
       ]);

blocksuite/affine/block-edgeless-text/src/edgeless-text-block.ts

@@ -3,7 +3,11 @@ import type { EdgelessTextBlockModel } from '@blocksuite/affine-model';
 import { ThemeProvider } from '@blocksuite/affine-shared/services';
 import { matchFlavours } from '@blocksuite/affine-shared/utils';
 import type { BlockComponent } from '@blocksuite/block-std';
-import { GfxBlockComponent } from '@blocksuite/block-std';
+import {
+  BlockSelection,
+  GfxBlockComponent,
+  TextSelection,
+} from '@blocksuite/block-std';
 import { Bound } from '@blocksuite/global/utils';
 import { css, html } from 'lit';
 import { query, state } from 'lit/decorators.js';
@@ -88,7 +92,7 @@ export class EdgelessTextBlockComponent extends GfxBlockComponent<EdgelessTextBl
           .then(() => {
             const command = this.std.command;
             const blockSelections = this.model.children.map(child =>
-              this.std.selection.create('block', {
+              this.std.selection.create(BlockSelection, {
                 blockId: child.id,
               })
             );
@@ -178,7 +182,7 @@ export class EdgelessTextBlockComponent extends GfxBlockComponent<EdgelessTextBl
 
       if (newParagraphId) {
         std.selection.setGroup('note', [
-          std.selection.create('text', {
+          std.selection.create(TextSelection, {
             from: {
               blockId: newParagraphId,
               index: 0,
@@ -310,7 +314,7 @@ export class EdgelessTextBlockComponent extends GfxBlockComponent<EdgelessTextBl
     const last = paragraphOrLists.at(-1);
     if (last) {
       this.host.selection.setGroup('note', [
-        this.host.selection.create('text', {
+        this.host.selection.create(TextSelection, {
           from: {
             blockId: last.blockId,
             index: last.model.text?.length ?? 0,

blocksuite/affine/block-embed/src/common/embed-block-element.ts

@@ -9,7 +9,7 @@ import {
   EMBED_CARD_WIDTH,
 } from '@blocksuite/affine-shared/consts';
 import { DocModeProvider } from '@blocksuite/affine-shared/services';
-import type { BlockService } from '@blocksuite/block-std';
+import { BlockSelection, type BlockService } from '@blocksuite/block-std';
 import type { GfxCompatibleProps } from '@blocksuite/block-std/gfx';
 import type { BlockModel } from '@blocksuite/store';
 import type { TemplateResult } from 'lit';
@@ -56,7 +56,7 @@ export class EmbedBlockComponent<
       }
     }
 
-    const selected = !!this.selected?.is('block');
+    const selected = !!this.selected?.is(BlockSelection);
     return html`
       <div
         draggable="${this.blockDraggable ? 'true' : 'false'}"

blocksuite/affine/block-embed/src/common/insert-embed-card.ts

@@ -7,7 +7,12 @@ import {
   EMBED_CARD_HEIGHT,
   EMBED_CARD_WIDTH,
 } from '@blocksuite/affine-shared/consts';
-import type { BlockStdScope } from '@blocksuite/block-std';
+import {
+  BlockSelection,
+  type BlockStdScope,
+  SurfaceSelection,
+  TextSelection,
+} from '@blocksuite/block-std';
 import { GfxControllerIdentifier } from '@blocksuite/block-std/gfx';
 import { Bound, Vec } from '@blocksuite/global/utils';
 
@@ -26,9 +31,9 @@ export function insertEmbedCard(
   const selectionManager = host.selection;
 
   let blockId: string | undefined;
-  const textSelection = selectionManager.find('text');
+  const textSelection = selectionManager.find(TextSelection);
-  const blockSelection = selectionManager.find('block');
+  const blockSelection = selectionManager.find(BlockSelection);
-  const surfaceSelection = selectionManager.find('surface');
+  const surfaceSelection = selectionManager.find(SurfaceSelection);
   if (textSelection) {
     blockId = textSelection.blockId;
   } else if (blockSelection) {

blocksuite/affine/block-embed/src/embed-figma-block/embed-figma-block.ts

@@ -3,6 +3,7 @@ import type {
   EmbedFigmaModel,
   EmbedFigmaStyles,
 } from '@blocksuite/affine-model';
+import { BlockSelection, SurfaceSelection } from '@blocksuite/block-std';
 import { html } from 'lit';
 import { state } from 'lit/decorators.js';
 import { classMap } from 'lit/directives/class-map.js';
@@ -41,7 +42,7 @@ export class EmbedFigmaBlockComponent extends EmbedBlockComponent<
 
   private _selectBlock() {
     const selectionManager = this.host.selection;
-    const blockSelection = selectionManager.create('block', {
+    const blockSelection = selectionManager.create(BlockSelection, {
       blockId: this.blockId,
     });
     selectionManager.setGroup('note', [blockSelection]);
@@ -77,7 +78,8 @@ export class EmbedFigmaBlockComponent extends EmbedBlockComponent<
     this.disposables.add(
       this.std.selection.slots.changed.on(() => {
         this._isSelected =
-          !!this.selected?.is('block') || !!this.selected?.is('surface');
+          !!this.selected?.is(BlockSelection) ||
+          !!this.selected?.is(SurfaceSelection);
 
         this._showOverlay =
           this._isResizing || this._isDragging || !this._isSelected;

blocksuite/affine/block-embed/src/embed-github-block/embed-github-block.ts

@@ -4,6 +4,7 @@ import type {
   EmbedGithubStyles,
 } from '@blocksuite/affine-model';
 import { ThemeProvider } from '@blocksuite/affine-shared/services';
+import { BlockSelection, SurfaceSelection } from '@blocksuite/block-std';
 import { html, nothing } from 'lit';
 import { property, state } from 'lit/decorators.js';
 import { classMap } from 'lit/directives/class-map.js';
@@ -61,7 +62,7 @@ export class EmbedGithubBlockComponent extends EmbedBlockComponent<
 
   private _selectBlock() {
     const selectionManager = this.host.selection;
-    const blockSelection = selectionManager.create('block', {
+    const blockSelection = selectionManager.create(BlockSelection, {
       blockId: this.blockId,
     });
     selectionManager.setGroup('note', [blockSelection]);
@@ -111,7 +112,8 @@ export class EmbedGithubBlockComponent extends EmbedBlockComponent<
     this.disposables.add(
       this.selection.slots.changed.on(() => {
         this._isSelected =
-          !!this.selected?.is('block') || !!this.selected?.is('surface');
+          !!this.selected?.is(BlockSelection) ||
+          !!this.selected?.is(SurfaceSelection);
       })
     );
   }

blocksuite/affine/block-embed/src/embed-html-block/embed-html-block.ts

@@ -1,4 +1,5 @@
 import type { EmbedHtmlModel, EmbedHtmlStyles } from '@blocksuite/affine-model';
+import { BlockSelection, SurfaceSelection } from '@blocksuite/block-std';
 import { html } from 'lit';
 import { query, state } from 'lit/decorators.js';
 import { classMap } from 'lit/directives/class-map.js';
@@ -35,7 +36,7 @@ export class EmbedHtmlBlockComponent extends EmbedBlockComponent<EmbedHtmlModel>
 
   private _selectBlock() {
     const selectionManager = this.host.selection;
-    const blockSelection = selectionManager.create('block', {
+    const blockSelection = selectionManager.create(BlockSelection, {
       blockId: this.blockId,
     });
     selectionManager.setGroup('note', [blockSelection]);
@@ -54,7 +55,8 @@ export class EmbedHtmlBlockComponent extends EmbedBlockComponent<EmbedHtmlModel>
     this.disposables.add(
       this.std.selection.slots.changed.on(() => {
         this._isSelected =
-          !!this.selected?.is('block') || !!this.selected?.is('surface');
+          !!this.selected?.is(BlockSelection) ||
+          !!this.selected?.is(SurfaceSelection);
 
         this._showOverlay =
           this._isResizing || this._isDragging || !this._isSelected;

blocksuite/affine/block-embed/src/embed-linked-doc-block/embed-linked-doc-block.ts

@@ -21,6 +21,7 @@ import {
   matchFlavours,
   referenceToNode,
 } from '@blocksuite/affine-shared/utils';
+import { BlockSelection } from '@blocksuite/block-std';
 import { Bound, throttle } from '@blocksuite/global/utils';
 import { Text } from '@blocksuite/store';
 import { computed } from '@preact/signals-core';
@@ -107,7 +108,7 @@ export class EmbedLinkedDocBlockComponent extends EmbedBlockComponent<EmbedLinke
 
   private readonly _selectBlock = () => {
     const selectionManager = this.host.selection;
-    const blockSelection = selectionManager.create('block', {
+    const blockSelection = selectionManager.create(BlockSelection, {
       blockId: this.blockId,
     });
     selectionManager.setGroup('note', [blockSelection]);

blocksuite/affine/block-embed/src/embed-loom-block/embed-loom-block.ts

@@ -1,6 +1,7 @@
 import { OpenIcon } from '@blocksuite/affine-components/icons';
 import type { EmbedLoomModel, EmbedLoomStyles } from '@blocksuite/affine-model';
 import { ThemeProvider } from '@blocksuite/affine-shared/services';
+import { BlockSelection, SurfaceSelection } from '@blocksuite/block-std';
 import { html } from 'lit';
 import { property, state } from 'lit/decorators.js';
 import { classMap } from 'lit/directives/class-map.js';
@@ -46,7 +47,7 @@ export class EmbedLoomBlockComponent extends EmbedBlockComponent<
 
   private _selectBlock() {
     const selectionManager = this.host.selection;
-    const blockSelection = selectionManager.create('block', {
+    const blockSelection = selectionManager.create(BlockSelection, {
       blockId: this.blockId,
     });
     selectionManager.setGroup('note', [blockSelection]);
@@ -93,7 +94,8 @@ export class EmbedLoomBlockComponent extends EmbedBlockComponent<
     this.disposables.add(
       this.std.selection.slots.changed.on(() => {
         this._isSelected =
-          !!this.selected?.is('block') || !!this.selected?.is('surface');
+          !!this.selected?.is(BlockSelection) ||
+          !!this.selected?.is(SurfaceSelection);
 
         this._showOverlay =
           this._isResizing || this._isDragging || !this._isSelected;

blocksuite/affine/block-embed/src/embed-synced-doc-block/components/embed-synced-doc-card.ts

@@ -1,5 +1,9 @@
 import { ThemeProvider } from '@blocksuite/affine-shared/services';
-import { isGfxBlockComponent, ShadowlessElement } from '@blocksuite/block-std';
+import {
+  BlockSelection,
+  isGfxBlockComponent,
+  ShadowlessElement,
+} from '@blocksuite/block-std';
 import { throttle, WithDisposable } from '@blocksuite/global/utils';
 import { html, nothing } from 'lit';
 import { property, queryAsync } from 'lit/decorators.js';
@@ -64,7 +68,7 @@ export class EmbedSyncedDocCard extends WithDisposable(ShadowlessElement) {
 
   private _selectBlock() {
     const selectionManager = this.host.selection;
-    const blockSelection = selectionManager.create('block', {
+    const blockSelection = selectionManager.create(BlockSelection, {
       blockId: this.block.blockId,
     });
     selectionManager.setGroup('note', [blockSelection]);

blocksuite/affine/block-embed/src/embed-synced-doc-block/embed-edgeless-synced-doc-block.ts

@@ -8,7 +8,7 @@ import {
   ThemeExtensionIdentifier,
   ThemeProvider,
 } from '@blocksuite/affine-shared/services';
-import { BlockStdScope } from '@blocksuite/block-std';
+import { BlockSelection, BlockStdScope } from '@blocksuite/block-std';
 import { assertExists, Bound } from '@blocksuite/global/utils';
 import { html } from 'lit';
 import { choose } from 'lit/directives/choose.js';
@@ -52,7 +52,7 @@ export class EmbedEdgelessSyncedDocBlockComponent extends toEdgelessEmbedBlock(
     }
     const theme = this.isPageMode ? appTheme : edgelessTheme;
 
-    const isSelected = !!this.selected?.is('block');
+    const isSelected = !!this.selected?.is(BlockSelection);
     const scale = this.model.scale ?? 1;
 
     this.dataset.nestedEditor = '';

blocksuite/affine/block-embed/src/embed-synced-doc-block/embed-synced-doc-block.ts

@@ -19,6 +19,7 @@ import {
   SpecProvider,
 } from '@blocksuite/affine-shared/utils';
 import {
+  BlockSelection,
   BlockServiceWatcher,
   BlockStdScope,
   type EditorHost,
@@ -164,7 +165,7 @@ export class EmbedSyncedDocBlockComponent extends EmbedBlockComponent<EmbedSynce
       edgelessTheme = themeExtension.getEdgelessTheme(this.syncedDoc.id).value;
     }
     const theme = isPageMode ? appTheme : edgelessTheme;
-    const isSelected = !!this.selected?.is('block');
+    const isSelected = !!this.selected?.is(BlockSelection);
 
     this.dataset.nestedEditor = '';
 
@@ -428,7 +429,7 @@ export class EmbedSyncedDocBlockComponent extends EmbedBlockComponent<EmbedSynce
 
   private _selectBlock() {
     const selectionManager = this.host.selection;
-    const blockSelection = selectionManager.create('block', {
+    const blockSelection = selectionManager.create(BlockSelection, {
       blockId: this.blockId,
     });
     selectionManager.setGroup('note', [blockSelection]);

blocksuite/affine/block-embed/src/embed-youtube-block/embed-youtube-block.ts

@@ -4,6 +4,7 @@ import type {
   EmbedYoutubeStyles,
 } from '@blocksuite/affine-model';
 import { ThemeProvider } from '@blocksuite/affine-shared/services';
+import { BlockSelection, SurfaceSelection } from '@blocksuite/block-std';
 import { html, nothing } from 'lit';
 import { property, state } from 'lit/decorators.js';
 import { classMap } from 'lit/directives/class-map.js';
@@ -49,7 +50,7 @@ export class EmbedYoutubeBlockComponent extends EmbedBlockComponent<
 
   private _selectBlock() {
     const selectionManager = this.host.selection;
-    const blockSelection = selectionManager.create('block', {
+    const blockSelection = selectionManager.create(BlockSelection, {
       blockId: this.blockId,
     });
     selectionManager.setGroup('note', [blockSelection]);
@@ -96,7 +97,8 @@ export class EmbedYoutubeBlockComponent extends EmbedBlockComponent<
     this.disposables.add(
       this.std.selection.slots.changed.on(() => {
         this._isSelected =
-          !!this.selected?.is('block') || !!this.selected?.is('surface');
+          !!this.selected?.is(BlockSelection) ||
+          !!this.selected?.is(SurfaceSelection);
 
         this._showOverlay =
           this._isResizing || this._isDragging || !this._isSelected;

blocksuite/affine/block-image/src/components/page-image-block.ts

@@ -1,5 +1,10 @@
+import { ImageSelection } from '@blocksuite/affine-shared/selection';
 import type { BaseSelection, UIEventStateContext } from '@blocksuite/block-std';
-import { ShadowlessElement } from '@blocksuite/block-std';
+import {
+  BlockSelection,
+  ShadowlessElement,
+  TextSelection,
+} from '@blocksuite/block-std';
 import { WithDisposable } from '@blocksuite/global/utils';
 import { css, html, type PropertyValues } from 'lit';
 import { property, query, state } from 'lit/decorators.js';
@@ -66,9 +71,9 @@ export class ImageBlockPageComponent extends WithDisposable(ShadowlessElement) {
 
       selection.update(selList =>
         selList
-          .filter<BaseSelection>(sel => !sel.is('image'))
+          .filter<BaseSelection>(sel => !sel.is(ImageSelection))
           .concat(
-            selection.create('text', {
+            selection.create(TextSelection, {
               from: {
                 blockId,
                 index: 0,
@@ -86,9 +91,11 @@ export class ImageBlockPageComponent extends WithDisposable(ShadowlessElement) {
         selection.update(selList => {
           return selList.map(sel => {
             const current =
-              sel.is('image') && sel.blockId === this.block.blockId;
+              sel.is(ImageSelection) && sel.blockId === this.block.blockId;
             if (current) {
-              return selection.create('block', { blockId: this.block.blockId });
+              return selection.create(BlockSelection, {
+                blockId: this.block.blockId,
+              });
             }
             return sel;
           });
@@ -119,7 +126,7 @@ export class ImageBlockPageComponent extends WithDisposable(ShadowlessElement) {
         const std = this._host.std;
 
         // If the selection is not image selection, we should not handle it.
-        if (!std.selection.find('image')) {
+        if (!std.selection.find(ImageSelection)) {
           return false;
         }
 
@@ -145,7 +152,7 @@ export class ImageBlockPageComponent extends WithDisposable(ShadowlessElement) {
 
         // If the selection is not image selection, we should not handle it.
 
-        if (!std.selection.find('image')) {
+        if (!std.selection.find(ImageSelection)) {
           return false;
         }
 
@@ -178,7 +185,7 @@ export class ImageBlockPageComponent extends WithDisposable(ShadowlessElement) {
     this._disposables.add(
       selection.slots.changed.on(selList => {
         this._isSelected = selList.some(
-          sel => sel.blockId === this.block.blockId && sel.is('image')
+          sel => sel.blockId === this.block.blockId && sel.is(ImageSelection)
         );
       })
     );
@@ -200,7 +207,9 @@ export class ImageBlockPageComponent extends WithDisposable(ShadowlessElement) {
         selection.update(selList => {
           return selList
             .filter(sel => !['block', 'image', 'text'].includes(sel.type))
-            .concat(selection.create('image', { blockId: this.block.blockId }));
+            .concat(
+              selection.create(ImageSelection, { blockId: this.block.blockId })
+            );
         });
         return true;
       }
@@ -213,7 +222,8 @@ export class ImageBlockPageComponent extends WithDisposable(ShadowlessElement) {
 
         selection.update(selList =>
           selList.filter(
-            sel => !(sel.is('image') && sel.blockId === this.block.blockId)
+            sel =>
+              !(sel.is(ImageSelection) && sel.blockId === this.block.blockId)
           )
         );
       },

blocksuite/affine/block-image/src/image-block.ts

@@ -1,6 +1,7 @@
 import { CaptionedBlockComponent } from '@blocksuite/affine-components/caption';
 import { Peekable } from '@blocksuite/affine-components/peek';
 import type { ImageBlockModel } from '@blocksuite/affine-model';
+import { BlockSelection } from '@blocksuite/block-std';
 import { IS_MOBILE } from '@blocksuite/global/env';
 import { html } from 'lit';
 import { property, query, state } from 'lit/decorators.js';
@@ -51,7 +52,7 @@ export class ImageBlockComponent extends CaptionedBlockComponent<
 
     event.stopPropagation();
     const selectionManager = this.host.selection;
-    const blockSelection = selectionManager.create('block', {
+    const blockSelection = selectionManager.create(BlockSelection, {
       blockId: this.blockId,
     });
     selectionManager.setGroup('note', [blockSelection]);

blocksuite/affine/block-latex/src/latex-block.ts

@@ -1,6 +1,7 @@
 import { CaptionedBlockComponent } from '@blocksuite/affine-components/caption';
 import { createLitPortal } from '@blocksuite/affine-components/portal';
 import type { LatexBlockModel } from '@blocksuite/affine-model';
+import { BlockSelection } from '@blocksuite/block-std';
 import type { Placement } from '@floating-ui/dom';
 import { effect } from '@preact/signals-core';
 import katex from 'katex';
@@ -19,7 +20,7 @@ export class LatexBlockComponent extends CaptionedBlockComponent<LatexBlockModel
   }
 
   get isBlockSelected() {
-    const blockSelection = this.selection.filter('block');
+    const blockSelection = this.selection.filter(BlockSelection);
     return blockSelection.some(
       selection => selection.blockId === this.model.id
     );

blocksuite/affine/block-list/src/commands/dedent-list.ts

@@ -1,6 +1,6 @@
 import type { IndentContext } from '@blocksuite/affine-shared/types';
 import { matchFlavours } from '@blocksuite/affine-shared/utils';
-import type { Command } from '@blocksuite/block-std';
+import { type Command, TextSelection } from '@blocksuite/block-std';
 
 import { correctNumberedListsOrderToPrev } from './utils.js';
 
@@ -13,7 +13,7 @@ export const canDedentListCommand: Command<
   const { std } = ctx;
   const { selection, doc } = std;
   if (!blockId) {
-    const text = selection.find('text');
+    const text = selection.find(TextSelection);
     /**
      * Do nothing if the selection:
      * - is not a text selection
@@ -148,7 +148,7 @@ export const dedentListCommand: Command<'indentContext'> = (ctx, next) => {
   doc.moveBlocks([model], grandParent, parent, false);
   correctNumberedListsOrderToPrev(doc, model);
 
-  const textSelection = selection.find('text');
+  const textSelection = selection.find(TextSelection);
   if (textSelection) {
     host.updateComplete
       .then(() => {

blocksuite/affine/block-list/src/commands/indent-list.ts

@@ -3,7 +3,7 @@ import {
   getNearestHeadingBefore,
   matchFlavours,
 } from '@blocksuite/affine-shared/utils';
-import type { Command } from '@blocksuite/block-std';
+import { type Command, TextSelection } from '@blocksuite/block-std';
 
 import { correctNumberedListsOrderToPrev } from './utils.js';
 
@@ -16,7 +16,7 @@ export const canIndentListCommand: Command<
   const { std } = ctx;
   const { selection, doc } = std;
   if (!blockId) {
-    const text = selection.find('text');
+    const text = selection.find(TextSelection);
     /**
      * Do nothing if the selection:
      * - is not a text selection
@@ -133,7 +133,7 @@ export const indentListCommand: Command<'indentContext', never> = (
     });
   }
 
-  const textSelection = selection.find('text');
+  const textSelection = selection.find(TextSelection);
   if (textSelection) {
     host.updateComplete
       .then(() => {

blocksuite/affine/block-list/src/list-block.ts

@@ -15,7 +15,11 @@ import {
 import { DocModeProvider } from '@blocksuite/affine-shared/services';
 import { getViewportElement } from '@blocksuite/affine-shared/utils';
 import type { BaseSelection, BlockComponent } from '@blocksuite/block-std';
-import { getInlineRangeProvider } from '@blocksuite/block-std';
+import {
+  BlockSelection,
+  getInlineRangeProvider,
+  TextSelection,
+} from '@blocksuite/block-std';
 import type { InlineRangeProvider } from '@blocksuite/inline';
 import { effect } from '@preact/signals-core';
 import { html, nothing, type TemplateResult } from 'lit';
@@ -95,8 +99,10 @@ export class ListBlockComponent extends CaptionedBlockComponent<ListBlockModel>
     const selection = this.host.selection;
     selection.update(selList => {
       return selList
-        .filter<BaseSelection>(sel => !sel.is('text') && !sel.is('block'))
+        .filter<BaseSelection>(
-        .concat(selection.create('block', { blockId: this.blockId }));
+          sel => !sel.is(TextSelection) && !sel.is(BlockSelection)
+        )
+        .concat(selection.create(BlockSelection, { blockId: this.blockId }));
     });
   }
 

blocksuite/affine/block-list/src/list-keymap.ts

@@ -3,7 +3,7 @@ import {
   textKeymap,
 } from '@blocksuite/affine-components/rich-text';
 import { ListBlockSchema } from '@blocksuite/affine-model';
-import { KeymapExtension } from '@blocksuite/block-std';
+import { KeymapExtension, TextSelection } from '@blocksuite/block-std';
 import { IS_MAC } from '@blocksuite/global/env';
 
 import { forwardDelete } from './utils/forward-delete.js';
@@ -12,7 +12,7 @@ export const ListKeymapExtension = KeymapExtension(
   std => {
     return {
       Enter: ctx => {
-        const text = std.selection.find('text');
+        const text = std.selection.find(TextSelection);
         if (!text) return false;
 
         ctx.get('keyboardState').raw.preventDefault();
@@ -23,7 +23,7 @@ export const ListKeymapExtension = KeymapExtension(
         return true;
       },
       'Mod-Enter': ctx => {
-        const text = std.selection.find('text');
+        const text = std.selection.find(TextSelection);
         if (!text) return false;
 
         ctx.get('keyboardState').raw.preventDefault();
@@ -40,7 +40,7 @@ export const ListKeymapExtension = KeymapExtension(
         if (selectedModels?.length !== 1) {
           return false;
         }
-        const text = std.selection.find('text');
+        const text = std.selection.find(TextSelection);
         if (!text) return false;
 
         ctx.get('keyboardState').raw.preventDefault();
@@ -61,7 +61,7 @@ export const ListKeymapExtension = KeymapExtension(
         if (selectedModels?.length !== 1) {
           return;
         }
-        const text = std.selection.find('text');
+        const text = std.selection.find(TextSelection);
         if (!text) return false;
 
         ctx.get('keyboardState').raw.preventDefault();
@@ -76,7 +76,7 @@ export const ListKeymapExtension = KeymapExtension(
         return true;
       },
       Backspace: ctx => {
-        const text = std.selection.find('text');
+        const text = std.selection.find(TextSelection);
         if (!text) return false;
         const isCollapsed = text.isCollapsed();
         const isStart = isCollapsed && text.from.index === 0;

blocksuite/affine/block-list/src/utils/forward-delete.ts

@@ -2,7 +2,7 @@ import {
   getNextContentBlock,
   matchFlavours,
 } from '@blocksuite/affine-shared/utils';
-import type { BlockStdScope } from '@blocksuite/block-std';
+import { type BlockStdScope, TextSelection } from '@blocksuite/block-std';
 import type { Text } from '@blocksuite/store';
 
 // When deleting at line end of a list block,
@@ -20,7 +20,7 @@ import type { Text } from '@blocksuite/store';
  - Line9
  */
 export function forwardDelete(std: BlockStdScope): true | undefined {
-  const text = std.selection.find('text');
+  const text = std.selection.find(TextSelection);
   if (!text) return;
   const isCollapsed = text.isCollapsed();
   const doc = std.doc;

blocksuite/affine/block-note/src/commands/block-type.ts

@@ -8,7 +8,11 @@ import {
   mergeToCodeModel,
   transformModel,
 } from '@blocksuite/affine-shared/utils';
-import type { Command } from '@blocksuite/block-std';
+import {
+  BlockSelection,
+  type Command,
+  TextSelection,
+} from '@blocksuite/block-std';
 import type { BlockModel } from '@blocksuite/store';
 
 type UpdateBlockConfig = {
@@ -108,11 +112,11 @@ export const updateBlockType: Command<
       onModelTextUpdated(host, model)
     );
     const selectionManager = host.selection;
-    const textSelection = selectionManager.find('text');
+    const textSelection = selectionManager.find(TextSelection);
     if (!textSelection) {
       return false;
     }
-    const newTextSelection = selectionManager.create('text', {
+    const newTextSelection = selectionManager.create(TextSelection, {
       from: {
         blockId: firstNewModel.id,
         index: textSelection.from.index,
@@ -143,13 +147,13 @@ export const updateBlockType: Command<
 
     const selectionManager = host.selection;
 
-    const blockSelections = selectionManager.filter('block');
+    const blockSelections = selectionManager.filter(BlockSelection);
     if (blockSelections.length === 0) {
       return false;
     }
     requestAnimationFrame(() => {
       const selections = updatedBlocks.map(model => {
-        return selectionManager.create('block', {
+        return selectionManager.create(BlockSelection, {
           blockId: model.id,
         });
       });

blocksuite/affine/block-note/src/commands/dedent-blocks-to-root.ts

@@ -1,5 +1,5 @@
 import { matchFlavours } from '@blocksuite/affine-shared/utils';
-import type { Command } from '@blocksuite/block-std';
+import { type Command, TextSelection } from '@blocksuite/block-std';
 
 export const dedentBlocksToRoot: Command<
   never,
@@ -13,7 +13,7 @@ export const dedentBlocksToRoot: Command<
   const { std, stopCapture = true } = ctx;
   const { doc } = std;
   if (!blockIds || !blockIds.length) {
-    const text = std.selection.find('text');
+    const text = std.selection.find(TextSelection);
     if (text) {
       // If the text selection is not at the beginning of the block, use default behavior
       if (text.from.index !== 0) return;

blocksuite/affine/block-note/src/commands/dedent-blocks.ts

@@ -2,7 +2,7 @@ import {
   calculateCollapsedSiblings,
   matchFlavours,
 } from '@blocksuite/affine-shared/utils';
-import type { Command } from '@blocksuite/block-std';
+import { type Command, TextSelection } from '@blocksuite/block-std';
 
 export const dedentBlocks: Command<
   never,
@@ -75,7 +75,7 @@ export const dedentBlocks: Command<
     std.command.exec('dedentBlock', { blockId: id, stopCapture: false });
   });
 
-  const textSelection = selection.find('text');
+  const textSelection = selection.find(TextSelection);
   if (textSelection) {
     host.updateComplete
       .then(() => {

blocksuite/affine/block-note/src/commands/focus-block-end.ts

@@ -1,4 +1,4 @@
-import type { Command } from '@blocksuite/block-std';
+import { type Command, TextSelection } from '@blocksuite/block-std';
 
 export const focusBlockEnd: Command<'focusBlock'> = (ctx, next) => {
   const { focusBlock, std } = ctx;
@@ -7,7 +7,7 @@ export const focusBlockEnd: Command<'focusBlock'> = (ctx, next) => {
   const { selection } = std;
 
   selection.setGroup('note', [
-    selection.create('text', {
+    selection.create(TextSelection, {
       from: {
         blockId: focusBlock.blockId,
         index: focusBlock.model.text.length,

blocksuite/affine/block-note/src/commands/focus-block-start.ts

@@ -1,4 +1,4 @@
-import type { Command } from '@blocksuite/block-std';
+import { type Command, TextSelection } from '@blocksuite/block-std';
 
 export const focusBlockStart: Command<'focusBlock'> = (ctx, next) => {
   const { focusBlock, std } = ctx;
@@ -7,7 +7,7 @@ export const focusBlockStart: Command<'focusBlock'> = (ctx, next) => {
   const { selection } = std;
 
   selection.setGroup('note', [
-    selection.create('text', {
+    selection.create(TextSelection, {
       from: { blockId: focusBlock.blockId, index: 0, length: 0 },
       to: null,
     }),

blocksuite/affine/block-note/src/commands/indent-blocks.ts

@@ -3,7 +3,7 @@ import {
   getNearestHeadingBefore,
   matchFlavours,
 } from '@blocksuite/affine-shared/utils';
-import type { Command } from '@blocksuite/block-std';
+import { type Command, TextSelection } from '@blocksuite/block-std';
 
 export const indentBlocks: Command<
   never,
@@ -117,7 +117,7 @@ export const indentBlocks: Command<
     }
   }
 
-  const textSelection = selection.find('text');
+  const textSelection = selection.find(TextSelection);
   if (textSelection) {
     host.updateComplete
       .then(() => {

blocksuite/affine/block-note/src/commands/select-block.ts

@@ -1,4 +1,4 @@
-import type { Command } from '@blocksuite/block-std';
+import { BlockSelection, type Command } from '@blocksuite/block-std';
 
 export const selectBlock: Command<'focusBlock'> = (ctx, next) => {
   const { focusBlock, std } = ctx;
@@ -9,7 +9,7 @@ export const selectBlock: Command<'focusBlock'> = (ctx, next) => {
   const { selection } = std;
 
   selection.setGroup('note', [
-    selection.create('block', { blockId: focusBlock.blockId }),
+    selection.create(BlockSelection, { blockId: focusBlock.blockId }),
   ]);
 
   return next();

blocksuite/affine/block-note/src/commands/select-blocks-between.ts

@@ -1,4 +1,4 @@
-import type { Command } from '@blocksuite/block-std';
+import { BlockSelection, type Command } from '@blocksuite/block-std';
 
 export const selectBlocksBetween: Command<
   'focusBlock' | 'anchorBlock',
@@ -14,7 +14,7 @@ export const selectBlocksBetween: Command<
   // In same block
   if (anchorBlock.blockId === focusBlock.blockId) {
     const blockId = focusBlock.blockId;
-    selection.setGroup('note', [selection.create('block', { blockId })]);
+    selection.setGroup('note', [selection.create(BlockSelection, { blockId })]);
     return next();
   }
 
@@ -23,11 +23,11 @@ export const selectBlocksBetween: Command<
   if (selections.every(sel => sel.blockId !== focusBlock.blockId)) {
     if (tail) {
       selections.push(
-        selection.create('block', { blockId: focusBlock.blockId })
+        selection.create(BlockSelection, { blockId: focusBlock.blockId })
       );
     } else {
       selections.unshift(
-        selection.create('block', { blockId: focusBlock.blockId })
+        selection.create(BlockSelection, { blockId: focusBlock.blockId })
       );
     }
   }

blocksuite/affine/block-note/src/move-block.ts

@@ -1,17 +1,21 @@
-import type { BlockSelection, BlockStdScope } from '@blocksuite/block-std';
+import {
+  BlockSelection,
+  type BlockStdScope,
+  TextSelection,
+} from '@blocksuite/block-std';
 
 const getSelection = (std: BlockStdScope) => std.selection;
 
 function getBlockSelectionBySide(std: BlockStdScope, tail: boolean) {
   const selection = getSelection(std);
-  const selections = selection.filter('block');
+  const selections = selection.filter(BlockSelection);
   const sel = selections.at(tail ? -1 : 0) as BlockSelection | undefined;
   return sel ?? null;
 }
 
 function getTextSelection(std: BlockStdScope) {
   const selection = getSelection(std);
-  return selection.find('text');
+  return selection.find(TextSelection);
 }
 
 const pathToBlock = (std: BlockStdScope, blockId: string) =>

blocksuite/affine/block-note/src/note-edgeless-block.ts

@@ -15,7 +15,11 @@ import {
   stopPropagation,
 } from '@blocksuite/affine-shared/utils';
 import type { BlockComponent, EditorHost } from '@blocksuite/block-std';
-import { ShadowlessElement, toGfxBlockComponent } from '@blocksuite/block-std';
+import {
+  ShadowlessElement,
+  TextSelection,
+  toGfxBlockComponent,
+} from '@blocksuite/block-std';
 import {
   almostEqual,
   Bound,
@@ -305,7 +309,7 @@ export class EdgelessNoteBlockComponent extends toGfxBlockComponent(
       this.updateComplete
         .then(() => {
           this.std.selection.setGroup('note', [
-            this.std.selection.create('text', {
+            this.std.selection.create(TextSelection, {
               from: {
                 blockId: pId,
                 index: 0,

blocksuite/affine/block-note/src/note-service.ts

@@ -4,9 +4,10 @@ import { matchFlavours } from '@blocksuite/affine-shared/utils';
 import {
   type BaseSelection,
   type BlockComponent,
-  type BlockSelection,
+  BlockSelection,
   BlockService,
   type BlockStdScope,
+  TextSelection,
   type UIEventHandler,
   type UIEventStateContext,
 } from '@blocksuite/block-std';
@@ -95,7 +96,7 @@ export class NoteBlockService extends BlockService {
                       const [codeModel] = newModels;
                       onModelElementUpdated(ctx.std, codeModel, codeElement => {
                         this._std.selection.setGroup('note', [
-                          this._std.selection.create('text', {
+                          this._std.selection.create(TextSelection, {
                             from: {
                               blockId: codeElement.blockId,
                               index: 0,
@@ -439,7 +440,7 @@ export class NoteBlockService extends BlockService {
 
         const blockId = doc.addBlock('affine:paragraph', {}, parent, index + 1);
 
-        const sel = selection.create('text', {
+        const sel = selection.create(TextSelection, {
           from: {
             blockId,
             index: 0,
@@ -469,7 +470,7 @@ export class NoteBlockService extends BlockService {
         }
 
         ctx.std.selection.update(selList => {
-          return selList.filter(sel => !sel.is('block'));
+          return selList.filter(sel => !sel.is(BlockSelection));
         });
 
         return next();
@@ -481,7 +482,7 @@ export class NoteBlockService extends BlockService {
 
   private readonly _onSelectAll: UIEventHandler = ctx => {
     const selection = this._std.selection;
-    const block = selection.find('block');
+    const block = selection.find(BlockSelection);
     if (!block) {
       return;
     }
@@ -492,13 +493,13 @@ export class NoteBlockService extends BlockService {
     ctx.get('defaultState').event.preventDefault();
     const children = note.children;
     const blocks: BlockSelection[] = children.map(child => {
-      return selection.create('block', {
+      return selection.create(BlockSelection, {
         blockId: child.id,
       });
     });
     selection.update(selList => {
       return selList
-        .filter<BaseSelection>(sel => !sel.is('block'))
+        .filter<BaseSelection>(sel => !sel.is(BlockSelection))
         .concat(blocks);
     });
   };

blocksuite/affine/block-paragraph/src/commands/add-paragraph.ts

@@ -1,5 +1,5 @@
 import { focusTextModel } from '@blocksuite/affine-components/rich-text';
-import type { Command } from '@blocksuite/block-std';
+import { type Command, TextSelection } from '@blocksuite/block-std';
 
 /**
  * Add a paragraph next to the current block.
@@ -17,7 +17,7 @@ export const addParagraphCommand: Command<
 
   let blockId = ctx.blockId;
   if (!blockId) {
-    const text = selection.find('text');
+    const text = selection.find(TextSelection);
     blockId = text?.blockId;
   }
   if (!blockId) return;

blocksuite/affine/block-paragraph/src/commands/dedent-paragraph.ts

@@ -3,7 +3,7 @@ import {
   calculateCollapsedSiblings,
   matchFlavours,
 } from '@blocksuite/affine-shared/utils';
-import type { Command } from '@blocksuite/block-std';
+import { type Command, TextSelection } from '@blocksuite/block-std';
 
 export const canDedentParagraphCommand: Command<
   never,
@@ -13,7 +13,7 @@ export const canDedentParagraphCommand: Command<
   let { blockId, inlineIndex } = ctx;
   const { std } = ctx;
   const { selection, doc } = std;
-  const text = selection.find('text');
+  const text = selection.find(TextSelection);
 
   if (!blockId) {
     /**
@@ -97,7 +97,7 @@ export const dedentParagraphCommand: Command<'indentContext'> = (ctx, next) => {
     doc.moveBlocks([model], grandParent, parent, false);
   }
 
-  const textSelection = selection.find('text');
+  const textSelection = selection.find(TextSelection);
   if (textSelection) {
     host.updateComplete
       .then(() => {

blocksuite/affine/block-paragraph/src/commands/indent-paragraph.ts

@@ -5,7 +5,7 @@ import {
   getNearestHeadingBefore,
   matchFlavours,
 } from '@blocksuite/affine-shared/utils';
-import type { Command } from '@blocksuite/block-std';
+import { type Command, TextSelection } from '@blocksuite/block-std';
 
 export const canIndentParagraphCommand: Command<
   never,
@@ -18,7 +18,7 @@ export const canIndentParagraphCommand: Command<
   const { schema } = doc;
 
   if (!blockId) {
-    const text = selection.find('text');
+    const text = selection.find(TextSelection);
     /**
      * Do nothing if the selection:
      * - is not a text selection
@@ -140,7 +140,7 @@ export const indentParagraphCommand: Command<'indentContext'> = (ctx, next) => {
     } as Partial<ListBlockModel>);
   }
 
-  const textSelection = selection.find('text');
+  const textSelection = selection.find(TextSelection);
   if (textSelection) {
     host.updateComplete
       .then(() => {

blocksuite/affine/block-paragraph/src/commands/split-paragraph.ts

@@ -3,7 +3,7 @@ import {
   getInlineEditorByModel,
 } from '@blocksuite/affine-components/rich-text';
 import { matchFlavours } from '@blocksuite/affine-shared/utils';
-import type { Command } from '@blocksuite/block-std';
+import { type Command, TextSelection } from '@blocksuite/block-std';
 
 export const splitParagraphCommand: Command<
   never,
@@ -16,7 +16,7 @@ export const splitParagraphCommand: Command<
   const { doc, host, selection } = std;
   let blockId = ctx.blockId;
   if (!blockId) {
-    const text = selection.find('text');
+    const text = selection.find(TextSelection);
     blockId = text?.blockId;
   }
   if (!blockId) return;

blocksuite/affine/block-paragraph/src/paragraph-block.ts

@@ -16,7 +16,7 @@ import {
   getViewportElement,
 } from '@blocksuite/affine-shared/utils';
 import type { BlockComponent } from '@blocksuite/block-std';
-import { getInlineRangeProvider } from '@blocksuite/block-std';
+import { getInlineRangeProvider, TextSelection } from '@blocksuite/block-std';
 import type { InlineRangeProvider } from '@blocksuite/inline';
 import { effect, signal } from '@preact/signals-core';
 import { html, nothing, type TemplateResult } from 'lit';
@@ -119,7 +119,7 @@ export class ParagraphBlockComponent extends CaptionedBlockComponent<
           this._displayPlaceholder.value = false;
           return;
         }
-        const textSelection = this.host.selection.find('text');
+        const textSelection = this.host.selection.find(TextSelection);
         const isCollapsed = textSelection?.isCollapsed() ?? false;
         if (!this.selected || !isCollapsed) {
           this._displayPlaceholder.value = false;
@@ -159,7 +159,7 @@ export class ParagraphBlockComponent extends CaptionedBlockComponent<
         // reset text selection when selected block is collapsed
         if (this.model.type.startsWith('h') && collapsed) {
           const collapsedSiblings = this.collapsedSiblings;
-          const textSelection = this.host.selection.find('text');
+          const textSelection = this.host.selection.find(TextSelection);
 
           if (
             textSelection &&

blocksuite/affine/block-paragraph/src/paragraph-keymap.ts

@@ -9,7 +9,7 @@ import {
   calculateCollapsedSiblings,
   matchFlavours,
 } from '@blocksuite/affine-shared/utils';
-import { KeymapExtension } from '@blocksuite/block-std';
+import { KeymapExtension, TextSelection } from '@blocksuite/block-std';
 import { IS_MAC } from '@blocksuite/global/env';
 
 import { forwardDelete } from './utils/forward-delete.js';
@@ -19,7 +19,7 @@ export const ParagraphKeymapExtension = KeymapExtension(
   std => {
     return {
       Backspace: ctx => {
-        const text = std.selection.find('text');
+        const text = std.selection.find(TextSelection);
         if (!text) return;
         const isCollapsed = text.isCollapsed();
         const isStart = isCollapsed && text.from.index === 0;
@@ -52,7 +52,7 @@ export const ParagraphKeymapExtension = KeymapExtension(
       },
       'Mod-Enter': ctx => {
         const { doc } = std;
-        const text = std.selection.find('text');
+        const text = std.selection.find(TextSelection);
         if (!text) return;
         const model = doc.getBlock(text.from.blockId)?.model;
         if (!model || !matchFlavours(model, ['affine:paragraph'])) return;
@@ -79,7 +79,7 @@ export const ParagraphKeymapExtension = KeymapExtension(
       },
       Enter: ctx => {
         const { doc } = std;
-        const text = std.selection.find('text');
+        const text = std.selection.find(TextSelection);
         if (!text) return;
         const model = doc.getBlock(text.from.blockId)?.model;
         if (!model || !matchFlavours(model, ['affine:paragraph'])) return;

blocksuite/affine/block-paragraph/src/utils/forward-delete.ts

@@ -3,11 +3,15 @@ import {
   getNextContentBlock,
   matchFlavours,
 } from '@blocksuite/affine-shared/utils';
-import type { BlockStdScope } from '@blocksuite/block-std';
+import {
+  BlockSelection,
+  type BlockStdScope,
+  TextSelection,
+} from '@blocksuite/block-std';
 
 export function forwardDelete(std: BlockStdScope) {
   const { doc, host } = std;
-  const text = std.selection.find('text');
+  const text = std.selection.find(TextSelection);
   if (!text) return;
   const isCollapsed = text.isCollapsed();
   const model = doc.getBlock(text.from.blockId)?.model;
@@ -30,7 +34,7 @@ export function forwardDelete(std: BlockStdScope) {
 
   if (matchFlavours(nextSibling, ignoreForwardDeleteFlavourList)) {
     std.selection.setGroup('note', [
-      std.selection.create('block', { blockId: nextSibling.id }),
+      std.selection.create(BlockSelection, { blockId: nextSibling.id }),
     ]);
     return true;
   }
@@ -61,7 +65,7 @@ export function forwardDelete(std: BlockStdScope) {
 
   if (nextBlock) {
     std.selection.setGroup('note', [
-      std.selection.create('block', { blockId: nextBlock.id }),
+      std.selection.create(BlockSelection, { blockId: nextBlock.id }),
     ]);
   }
   return true;

blocksuite/affine/block-paragraph/src/utils/merge-with-prev.ts

@@ -11,7 +11,7 @@ import {
   getPrevContentBlock,
   matchFlavours,
 } from '@blocksuite/affine-shared/utils';
-import type { EditorHost } from '@blocksuite/block-std';
+import { BlockSelection, type EditorHost } from '@blocksuite/block-std';
 import type { BlockModel, Text } from '@blocksuite/store';
 
 /**
@@ -72,7 +72,7 @@ export function mergeWithPrev(editorHost: EditorHost, model: BlockModel) {
       ...EMBED_BLOCK_FLAVOUR_LIST,
     ])
   ) {
-    const selection = editorHost.selection.create('block', {
+    const selection = editorHost.selection.create(BlockSelection, {
       blockId: prevBlock.id,
     });
     editorHost.selection.setGroup('note', [selection]);

blocksuite/affine/block-surface-ref/src/surface-ref-block.ts

@@ -28,10 +28,12 @@ import {
 import {
   type BaseSelection,
   BlockComponent,
+  BlockSelection,
   BlockServiceWatcher,
   BlockStdScope,
   type EditorHost,
   LifeCycleWatcher,
+  TextSelection,
 } from '@blocksuite/block-std';
 import {
   GfxBlockElementModel,
@@ -267,7 +269,7 @@ export class SurfaceRefBlockComponent extends BlockComponent<SurfaceRefBlockMode
 
   private _focusBlock() {
     this.selection.update(() => {
-      return [this.selection.create('block', { blockId: this.blockId })];
+      return [this.selection.create(BlockSelection, { blockId: this.blockId })];
     });
   }
 
@@ -287,9 +289,9 @@ export class SurfaceRefBlockComponent extends BlockComponent<SurfaceRefBlockMode
       requestConnectedFrame(() => {
         selection.update(selList => {
           return selList
-            .filter<BaseSelection>(sel => !sel.is('block'))
+            .filter<BaseSelection>(sel => !sel.is(BlockSelection))
             .concat(
-              selection.create('text', {
+              selection.create(TextSelection, {
                 from: {
                   blockId: model.id,
                   index: 0,
@@ -415,7 +417,7 @@ export class SurfaceRefBlockComponent extends BlockComponent<SurfaceRefBlockMode
     this._disposables.add(
       selection.slots.changed.on(selList => {
         this._focused = selList.some(
-          sel => sel.blockId === this.blockId && sel.is('block')
+          sel => sel.blockId === this.blockId && sel.is(BlockSelection)
         );
       })
     );

blocksuite/affine/components/src/block-selection/index.ts

@@ -1,4 +1,7 @@
-import type { BlockComponent } from '@blocksuite/block-std';
+import {
+  type BlockComponent,
+  BlockSelection as StdBlockSelection,
+} from '@blocksuite/block-std';
 import { SignalWatcher } from '@blocksuite/global/utils';
 import { css, LitElement, type PropertyValues } from 'lit';
 import { property } from 'lit/decorators.js';
@@ -55,7 +58,9 @@ export class BlockSelection extends SignalWatcher(LitElement) {
   protected override updated(_changedProperties: PropertyValues): void {
     super.updated(_changedProperties);
     if (this.block) {
-      this.style.display = this.block.selected?.is('block') ? 'block' : 'none';
+      this.style.display = this.block.selected?.is(StdBlockSelection)
+        ? 'block'
+        : 'none';
     }
   }
 

blocksuite/affine/components/src/rich-text/dom.ts

@@ -3,7 +3,11 @@ import {
   getCurrentNativeRange,
   matchFlavours,
 } from '@blocksuite/affine-shared/utils';
-import type { BlockStdScope, EditorHost } from '@blocksuite/block-std';
+import {
+  type BlockStdScope,
+  type EditorHost,
+  TextSelection,
+} from '@blocksuite/block-std';
 import type { InlineEditor, InlineRange } from '@blocksuite/inline';
 import { BlockModel } from '@blocksuite/store';
 
@@ -81,7 +85,7 @@ export function selectTextModel(
 ) {
   const { selection } = std;
   selection.setGroup('note', [
-    selection.create('text', {
+    selection.create(TextSelection, {
       from: { blockId: id, index, length },
       to: null,
     }),

blocksuite/affine/components/src/rich-text/format/delete-text.ts

@@ -1,5 +1,5 @@
 import { matchFlavours } from '@blocksuite/affine-shared/utils';
-import type { Command, TextSelection } from '@blocksuite/block-std';
+import { type Command, TextSelection } from '@blocksuite/block-std';
 import type { Text } from '@blocksuite/store';
 
 export const deleteTextCommand: Command<
@@ -36,7 +36,7 @@ export const deleteTextCommand: Command<
   if (!to) {
     fromText.delete(from.index, from.length);
     ctx.std.selection.setGroup('note', [
-      ctx.std.selection.create('text', {
+      ctx.std.selection.create(TextSelection, {
         from: {
           blockId: from.blockId,
           index: from.index,
@@ -69,7 +69,7 @@ export const deleteTextCommand: Command<
     });
 
   ctx.std.selection.setGroup('note', [
-    ctx.std.selection.create('text', {
+    ctx.std.selection.create(TextSelection, {
       from: {
         blockId: from.blockId,
         index: from.index,

blocksuite/affine/components/src/rich-text/inline/presets/nodes/link-node/affine-link.ts

@@ -2,7 +2,12 @@ import type { ReferenceInfo } from '@blocksuite/affine-model';
 import { ParseDocUrlProvider } from '@blocksuite/affine-shared/services';
 import type { AffineTextAttributes } from '@blocksuite/affine-shared/types';
 import type { BlockComponent } from '@blocksuite/block-std';
-import { BLOCK_ID_ATTR, ShadowlessElement } from '@blocksuite/block-std';
+import {
+  BLOCK_ID_ATTR,
+  BlockSelection,
+  ShadowlessElement,
+  TextSelection,
+} from '@blocksuite/block-std';
 import {
   type DeltaInsert,
   INLINE_ROOT_ATTR,
@@ -69,12 +74,12 @@ export class AffineLink extends ShadowlessElement {
       }
 
       const selection = this.std?.selection;
-      const textSelection = selection?.find('text');
+      const textSelection = selection?.find(TextSelection);
       if (!!textSelection && !textSelection.isCollapsed()) {
         return null;
       }
 
-      const blockSelections = selection?.filter('block');
+      const blockSelections = selection?.filter(BlockSelection);
       if (blockSelections?.length) {
         return null;
       }

blocksuite/affine/components/src/rich-text/inline/presets/nodes/link-node/link-popup/link-popup.ts

@@ -15,6 +15,7 @@ import {
   BLOCK_ID_ATTR,
   type BlockComponent,
   type BlockStdScope,
+  TextSelection,
 } from '@blocksuite/block-std';
 import { WithDisposable } from '@blocksuite/global/utils';
 import type { InlineRange } from '@blocksuite/inline/types';
@@ -438,7 +439,7 @@ export class LinkPopup extends WithDisposable(LitElement) {
         reference: null,
       });
       this.inlineEditor.setInlineRange(this.targetInlineRange);
-      const textSelection = this.host?.selection.find('text');
+      const textSelection = this.host?.selection.find(TextSelection);
       if (!textSelection) return;
 
       this.std?.range.syncTextSelectionToRange(textSelection);
@@ -452,7 +453,7 @@ export class LinkPopup extends WithDisposable(LitElement) {
         index: this.targetInlineRange.index,
         length: text.length,
       });
-      const textSelection = this.host?.selection.find('text');
+      const textSelection = this.host?.selection.find(TextSelection);
       if (!textSelection) return;
 
       this.std?.range.syncTextSelectionToRange(textSelection);

blocksuite/affine/components/src/rich-text/inline/presets/nodes/reference-node/reference-node.ts

@@ -12,7 +12,9 @@ import {
 import {
   BLOCK_ID_ATTR,
   type BlockComponent,
+  BlockSelection,
   ShadowlessElement,
+  TextSelection,
 } from '@blocksuite/block-std';
 import { BlockSuiteError, ErrorCode } from '@blocksuite/global/exceptions';
 import { WithDisposable } from '@blocksuite/global/utils';
@@ -108,12 +110,12 @@ export class AffineReference extends WithDisposable(ShadowlessElement) {
       if (!selection) {
         return null;
       }
-      const textSelection = selection.find('text');
+      const textSelection = selection.find(TextSelection);
       if (!!textSelection && !textSelection.isCollapsed()) {
         return null;
       }
 
-      const blockSelections = selection.filter('block');
+      const blockSelections = selection.filter(BlockSelection);
       if (blockSelections.length) {
         return null;
       }

blocksuite/affine/components/src/rich-text/keymap/basic.ts

@@ -1,4 +1,9 @@
-import type { BlockStdScope, UIEventHandler } from '@blocksuite/block-std';
+import {
+  BlockSelection,
+  type BlockStdScope,
+  TextSelection,
+  type UIEventHandler,
+} from '@blocksuite/block-std';
 
 import {
   focusTextModel,
@@ -11,21 +16,21 @@ export const textCommonKeymap = (
 ): Record<string, UIEventHandler> => {
   return {
     ArrowUp: () => {
-      const text = std.selection.find('text');
+      const text = std.selection.find(TextSelection);
       if (!text) return;
       const inline = getInlineEditorByModel(std.host, text.from.blockId);
       if (!inline) return;
       return !inline.isFirstLine(inline.getInlineRange());
     },
     ArrowDown: () => {
-      const text = std.selection.find('text');
+      const text = std.selection.find(TextSelection);
       if (!text) return;
       const inline = getInlineEditorByModel(std.host, text.from.blockId);
       if (!inline) return;
       return !inline.isLastLine(inline.getInlineRange());
     },
     Escape: ctx => {
-      const text = std.selection.find('text');
+      const text = std.selection.find(TextSelection);
       if (!text) return;
 
       selectBlock(std, text.from.blockId);
@@ -33,7 +38,7 @@ export const textCommonKeymap = (
       return true;
     },
     'Mod-a': ctx => {
-      const text = std.selection.find('text');
+      const text = std.selection.find(TextSelection);
       if (!text) return;
 
       const model = std.doc.getBlock(text.from.blockId)?.model;
@@ -53,7 +58,7 @@ export const textCommonKeymap = (
       return true;
     },
     Enter: ctx => {
-      const blocks = std.selection.filter('block');
+      const blocks = std.selection.filter(BlockSelection);
       const blockId = blocks.at(-1)?.blockId;
 
       if (!blockId) return;
@@ -68,5 +73,7 @@ export const textCommonKeymap = (
 };
 
 function selectBlock(std: BlockStdScope, blockId: string) {
-  std.selection.setGroup('note', [std.selection.create('block', { blockId })]);
+  std.selection.setGroup('note', [
+    std.selection.create(BlockSelection, { blockId }),
+  ]);
 }

blocksuite/affine/components/src/rich-text/keymap/bracket.ts

@@ -3,7 +3,11 @@ import {
   createDefaultDoc,
   matchFlavours,
 } from '@blocksuite/affine-shared/utils';
-import type { BlockStdScope, UIEventHandler } from '@blocksuite/block-std';
+import {
+  type BlockStdScope,
+  TextSelection,
+  type UIEventHandler,
+} from '@blocksuite/block-std';
 import type { InlineEditor } from '@blocksuite/inline';
 
 import { getInlineEditorByModel } from '../dom.js';
@@ -20,7 +24,7 @@ export const bracketKeymap = (
           const { doc, selection } = std;
           if (doc.readonly) return;
 
-          const textSelection = selection.find('text');
+          const textSelection = selection.find(TextSelection);
           if (!textSelection) return;
           const model = doc.getBlock(textSelection.from.blockId)?.model;
           if (!model) return;
@@ -46,7 +50,7 @@ export const bracketKeymap = (
           const { doc, selection } = std;
           if (doc.readonly) return;
 
-          const textSelection = selection.find('text');
+          const textSelection = selection.find(TextSelection);
           if (!textSelection) return;
           const model = doc.getBlock(textSelection.from.blockId)?.model;
           if (!model) return;
@@ -97,7 +101,7 @@ export const bracketKeymap = (
       const { doc, selection } = std;
       if (doc.readonly) return;
 
-      const textSelection = selection.find('text');
+      const textSelection = selection.find(TextSelection);
       if (!textSelection || textSelection.isCollapsed()) return;
       if (!textSelection.isInSameBlock()) return;
       const model = doc.getBlock(textSelection.from.blockId)?.model;

blocksuite/affine/components/src/rich-text/keymap/format.ts

@@ -1,4 +1,8 @@
-import type { BlockStdScope, UIEventHandler } from '@blocksuite/block-std';
+import {
+  type BlockStdScope,
+  TextSelection,
+  type UIEventHandler,
+} from '@blocksuite/block-std';
 
 import { textFormatConfigs } from '../format/index.js';
 
@@ -13,7 +17,7 @@ export const textFormatKeymap = (std: BlockStdScope) =>
             const { doc, selection } = std;
             if (doc.readonly) return;
 
-            const textSelection = selection.find('text');
+            const textSelection = selection.find(TextSelection);
             if (!textSelection) return;
 
             config.action(std.host);

blocksuite/affine/components/src/rich-text/markdown/markdown-input.ts

@@ -2,7 +2,7 @@ import {
   isMarkdownPrefix,
   matchFlavours,
 } from '@blocksuite/affine-shared/utils';
-import type { BlockStdScope } from '@blocksuite/block-std';
+import { type BlockStdScope, TextSelection } from '@blocksuite/block-std';
 
 import { getInlineEditorByModel } from '../dom.js';
 import { toDivider } from './divider.js';
@@ -17,7 +17,7 @@ export function markdownInput(
 ): string | undefined {
   if (!id) {
     const selection = std.selection;
-    const text = selection.find('text');
+    const text = selection.find(TextSelection);
     id = text?.from.blockId;
   }
   if (!id) return;

blocksuite/affine/shared/src/adapters/middlewares/copy.ts

@@ -1,7 +1,8 @@
-import type {
+import {
-  BlockStdScope,
+  type BlockStdScope,
-  EditorHost,
+  type EditorHost,
-  TextRangePoint,
+  type TextRangePoint,
+  TextSelection,
 } from '@blocksuite/block-std';
 import type {
   BlockSnapshot,
@@ -38,7 +39,7 @@ const sliceText = (slots: JobSlots, std: EditorHost['std']) => {
       const snapshot = payload.snapshot;
 
       const model = payload.model;
-      const text = std.selection.find('text');
+      const text = std.selection.find(TextSelection);
       if (text && text.from.blockId === model.id) {
         handlePoint(text.from, snapshot, model);
         return;

blocksuite/affine/shared/src/adapters/middlewares/paste.ts

@@ -7,9 +7,10 @@ import {
 import {
   BLOCK_ID_ATTR,
   type BlockComponent,
+  BlockSelection,
   type EditorHost,
   type TextRangePoint,
-  type TextSelection,
+  TextSelection,
 } from '@blocksuite/block-std';
 import { BlockSuiteError, ErrorCode } from '@blocksuite/global/exceptions';
 import { assertExists } from '@blocksuite/global/utils';
@@ -25,6 +26,7 @@ import {
 import * as Y from 'yjs';
 
 import { REFERENCE_NODE } from '../../consts';
+import { ImageSelection } from '../../selection';
 import {
   ParseDocUrlProvider,
   type ParseDocUrlService,
@@ -290,19 +292,19 @@ class PasteTr {
         }
         if (!cursorModel.text) {
           if (matchFlavours(cursorModel, ['affine:image'])) {
-            const selection = this.std.selection.create('image', {
+            const selection = this.std.selection.create(ImageSelection, {
               blockId: target.blockId,
             });
             this.std.selection.setGroup('note', [selection]);
             return;
           }
-          const selection = this.std.selection.create('block', {
+          const selection = this.std.selection.create(BlockSelection, {
             blockId: target.blockId,
           });
           this.std.selection.setGroup('note', [selection]);
           return;
         }
-        const selection = this.std.selection.create('text', {
+        const selection = this.std.selection.create(TextSelection, {
           from: {
             blockId: target.blockId,
             index: cursorModel.text ? this.lastIndex : 0,
@@ -511,7 +513,7 @@ export const pasteMiddleware = (std: EditorHost['std']): JobMiddleware => {
         const { snapshot } = payload;
         flatNote(snapshot);
 
-        const text = std.selection.find('text');
+        const text = std.selection.find(TextSelection);
         if (!text) {
           return;
         }

blocksuite/affine/shared/src/commands/block-crud/get-selected-blocks.ts

@@ -16,7 +16,7 @@ export const getSelectedBlocksCommand: Command<
     blockSelections?: BlockSelection[];
     imageSelections?: ImageSelection[];
     filter?: (el: BlockComponent) => boolean;
-    types?: Extract<BlockSuite.SelectionType, 'block' | 'text' | 'image'>[];
+    types?: Array<'image' | 'text' | 'block'>;
     roles?: RoleType[];
     mode?: 'all' | 'flat' | 'highest';
   }

blocksuite/affine/shared/src/commands/model-crud/clear-and-select-first-model.ts

@@ -1,4 +1,4 @@
-import type { Command } from '@blocksuite/block-std';
+import { type Command, TextSelection } from '@blocksuite/block-std';
 
 export const clearAndSelectFirstModelCommand: Command<'selectedModels'> = (
   ctx,
@@ -17,7 +17,7 @@ export const clearAndSelectFirstModelCommand: Command<'selectedModels'> = (
     const firstModel = models[0];
     if (firstModel.text) {
       firstModel.text.clear();
-      const selection = ctx.std.selection.create('text', {
+      const selection = ctx.std.selection.create(TextSelection, {
         from: {
           blockId: firstModel.id,
           index: 0,

blocksuite/affine/shared/src/commands/model-crud/get-selected-models.ts

@@ -32,7 +32,7 @@ export const getSelectedModelsCommand: Command<
   never,
   'selectedModels',
   {
-    types?: Extract<BlockSuite.SelectionType, 'block' | 'text' | 'image'>[];
+    types?: Array<'image' | 'text' | 'block'>;
     mode?: 'all' | 'flat' | 'highest';
   }
 > = (ctx, next) => {

blocksuite/affine/shared/src/commands/selection/get-block-selections.ts

@@ -1,10 +1,10 @@
-import type { BlockSelection, Command } from '@blocksuite/block-std';
+import { BlockSelection, type Command } from '@blocksuite/block-std';
 
 export const getBlockSelectionsCommand: Command<
   never,
   'currentBlockSelections'
 > = (ctx, next) => {
-  const currentBlockSelections = ctx.std.selection.filter('block');
+  const currentBlockSelections = ctx.std.selection.filter(BlockSelection);
   if (currentBlockSelections.length === 0) return;
 
   next({ currentBlockSelections });

blocksuite/affine/shared/src/commands/selection/get-image-selections.ts

@@ -1,12 +1,12 @@
 import type { Command } from '@blocksuite/block-std';
 
-import type { ImageSelection } from '../../selection/index.js';
+import { ImageSelection } from '../../selection/index.js';
 
 export const getImageSelectionsCommand: Command<
   never,
   'currentImageSelections'
 > = (ctx, next) => {
-  const currentImageSelections = ctx.std.selection.filter('image');
+  const currentImageSelections = ctx.std.selection.filter(ImageSelection);
   if (currentImageSelections.length === 0) return;
 
   next({ currentImageSelections });

blocksuite/affine/shared/src/commands/selection/get-text-selection.ts

@@ -1,10 +1,10 @@
-import type { Command, TextSelection } from '@blocksuite/block-std';
+import { type Command, TextSelection } from '@blocksuite/block-std';
 
 export const getTextSelectionCommand: Command<never, 'currentTextSelection'> = (
   ctx,
   next
 ) => {
-  const currentTextSelection = ctx.std.selection.find('text');
+  const currentTextSelection = ctx.std.selection.find(TextSelection);
   if (!currentTextSelection) return;
 
   next({ currentTextSelection });

blocksuite/affine/widget-drag-handle/src/helpers/selection-helper.ts

@@ -1,5 +1,10 @@
 import { findNoteBlockModel } from '@blocksuite/affine-shared/utils';
-import type { BlockComponent } from '@blocksuite/block-std';
+import {
+  type BlockComponent,
+  BlockSelection,
+  SurfaceSelection,
+  TextSelection,
+} from '@blocksuite/block-std';
 
 import type { AffineDragHandleWidget } from '../drag-handle.js';
 
@@ -15,7 +20,7 @@ export class SelectionHelper {
   setSelectedBlocks = (blocks: BlockComponent[], noteId?: string) => {
     const { selection } = this;
     const selections = blocks.map(block =>
-      selection.create('block', {
+      selection.create(BlockSelection, {
         blockId: block.blockId,
       })
     );
@@ -28,7 +33,7 @@ export class SelectionHelper {
         : findNoteBlockModel(blocks[0].model)?.id;
       if (!surfaceElementId) return;
       const surfaceSelection = selection.create(
-        'surface',
+        SurfaceSelection,
         blocks[0]!.blockId,
         [surfaceElementId],
         true
@@ -49,9 +54,9 @@ export class SelectionHelper {
   get selectedBlocks() {
     const selection = this.selection;
 
-    return selection.find('text')
+    return selection.find(TextSelection)
-      ? selection.filter('text')
+      ? selection.filter(TextSelection)
-      : selection.filter('block');
+      : selection.filter(BlockSelection);
   }
 
   get selection() {

blocksuite/affine/widget-drag-handle/src/watchers/drag-event-watcher.ts

@@ -23,6 +23,7 @@ import {
 } from '@blocksuite/affine-shared/utils';
 import {
   type BlockComponent,
+  BlockSelection,
   type DndEventState,
   isGfxBlockComponent,
   type UIEventHandler,
@@ -158,7 +159,7 @@ export class DragEventWatcher {
 
     const selectBlockAndStartDragging = () => {
       this._std.selection.setGroup('note', [
-        this._std.selection.create('block', {
+        this._std.selection.create(BlockSelection, {
           blockId: hoverBlock.blockId,
         }),
       ]);

blocksuite/affine/widget-remote-selection/src/doc/doc-remote-selection.ts

@@ -101,7 +101,7 @@ export class AffineDocRemoteSelectionWidget extends WidgetComponent {
 
     if (textSelection) {
       const range = this.std.range.textSelectionToRange(
-        this._selectionManager.create('text', {
+        this._selectionManager.create(TextSelection, {
           from: {
             blockId: textSelection.to
               ? textSelection.to.blockId

blocksuite/affine/widget-scroll-anchoring/src/scroll-anchoring.ts

@@ -1,6 +1,5 @@
-import '@blocksuite/affine-shared/selection';
-
 import type { DocMode } from '@blocksuite/affine-model';
+import { HighlightSelection } from '@blocksuite/affine-shared/selection';
 import { WidgetComponent } from '@blocksuite/block-std';
 import {
   GfxControllerIdentifier,
@@ -65,7 +64,7 @@ export class AffineScrollAnchoringWidget extends WidgetComponent {
   anchorBounds$ = signal<Bound | null>(null);
 
   highlighted$ = computed(() =>
-    this.service.selectionManager.find('highlight')
+    this.service.selectionManager.find(HighlightSelection)
   );
 
   #getBoundsInEdgeless() {

blocksuite/blocks/src/index.ts

@@ -114,6 +114,7 @@ export {
   PlainTextAdapterFactoryExtension,
   PlainTextAdapterFactoryIdentifier,
 } from '@blocksuite/affine-shared/adapters';
+export { HighlightSelection } from '@blocksuite/affine-shared/selection';
 export * from '@blocksuite/affine-shared/services';
 export { scrollbarStyle } from '@blocksuite/affine-shared/styles';
 export {

blocksuite/blocks/src/root-block/edgeless/edgeless-keyboard.ts

@@ -16,6 +16,7 @@ import {
 } from '@blocksuite/affine-shared/services';
 import { LassoMode } from '@blocksuite/affine-shared/types';
 import { matchFlavours } from '@blocksuite/affine-shared/utils';
+import { SurfaceSelection, TextSelection } from '@blocksuite/block-std';
 import {
   GfxBlockElementModel,
   type GfxToolsMap,
@@ -167,8 +168,8 @@ export class EdgelessPageKeyboardManager extends PageKeyboardManager {
           const std = this.rootComponent.std;
           if (
             std.selection.getGroup('note').length > 0 ||
-            std.selection.find('text') ||
+            std.selection.find(TextSelection) ||
-            Boolean(std.selection.find('surface')?.editing)
+            Boolean(std.selection.find(SurfaceSelection)?.editing)
           ) {
             return;
           }

blocksuite/blocks/src/root-block/edgeless/edgeless-root-block.ts

@@ -22,12 +22,12 @@ import {
   requestConnectedFrame,
   requestThrottledConnectedFrame,
 } from '@blocksuite/affine-shared/utils';
-import type {
+import {
-  GfxBlockComponent,
+  BlockComponent,
+  type GfxBlockComponent,
   SurfaceSelection,
-  UIEventHandler,
+  type UIEventHandler,
 } from '@blocksuite/block-std';
-import { BlockComponent } from '@blocksuite/block-std';
 import {
   GfxControllerIdentifier,
   type GfxViewportElement,
@@ -454,7 +454,7 @@ export class EdgelessRootBlockComponent extends BlockComponent<
 
     this.handleEvent('selectionChange', () => {
       const surface = this.host.selection.value.find(
-        (sel): sel is SurfaceSelection => sel.is('surface')
+        (sel): sel is SurfaceSelection => sel.is(SurfaceSelection)
       );
       if (!surface) return;
 

blocksuite/blocks/src/root-block/edgeless/edgeless-root-preview-block.ts

@@ -9,11 +9,11 @@ import {
   ThemeProvider,
 } from '@blocksuite/affine-shared/services';
 import { requestThrottledConnectedFrame } from '@blocksuite/affine-shared/utils';
-import type {
+import {
-  GfxBlockComponent,
+  BlockComponent,
+  type GfxBlockComponent,
   SurfaceSelection,
 } from '@blocksuite/block-std';
-import { BlockComponent } from '@blocksuite/block-std';
 import type { GfxViewportElement } from '@blocksuite/block-std/gfx';
 import { assertExists } from '@blocksuite/global/utils';
 import { css, html } from 'lit';
@@ -181,7 +181,7 @@ export class EdgelessRootPreviewBlockComponent
 
     this.handleEvent('selectionChange', () => {
       const surface = this.host.selection.value.find(
-        (sel): sel is SurfaceSelection => sel.is('surface')
+        (sel): sel is SurfaceSelection => sel.is(SurfaceSelection)
       );
       if (!surface) return;
 

blocksuite/blocks/src/root-block/keyboard/keyboard-manager.ts

@@ -6,14 +6,18 @@ import {
 } from '@blocksuite/affine-block-embed';
 import { ParagraphBlockComponent } from '@blocksuite/affine-block-paragraph';
 import { matchFlavours } from '@blocksuite/affine-shared/utils';
-import type { BlockComponent, UIEventHandler } from '@blocksuite/block-std';
+import {
+  type BlockComponent,
+  BlockSelection,
+  type UIEventHandler,
+} from '@blocksuite/block-std';
 import { IS_MAC, IS_WINDOWS } from '@blocksuite/global/env';
 
 export class PageKeyboardManager {
   private readonly _handleDelete: UIEventHandler = ctx => {
     const event = ctx.get('keyboardState').raw;
     const blockSelections = this._currentSelection.filter(sel =>
-      sel.is('block')
+      sel.is(BlockSelection)
     );
     if (blockSelections.length === 0) {
       return;

blocksuite/blocks/src/root-block/page/page-root-block.ts

@@ -10,7 +10,11 @@ import {
   matchFlavours,
 } from '@blocksuite/affine-shared/utils';
 import type { PointerEventState } from '@blocksuite/block-std';
-import { BlockComponent } from '@blocksuite/block-std';
+import {
+  BlockComponent,
+  BlockSelection,
+  TextSelection,
+} from '@blocksuite/block-std';
 import type { BlockModel, Text } from '@blocksuite/store';
 import { css, html } from 'lit';
 import { query } from 'lit/decorators.js';
@@ -236,7 +240,7 @@ export class PageRootBlockComponent extends BlockComponent<
           })
           .flatMap(model => {
             return model.children.map(child => {
-              return this.std.selection.create('block', {
+              return this.std.selection.create(BlockSelection, {
                 blockId: child.id,
               });
             });
@@ -247,7 +251,7 @@ export class PageRootBlockComponent extends BlockComponent<
       ArrowUp: () => {
         const selection = this.host.selection;
         const sel = selection.value.find(
-          sel => sel.is('text') || sel.is('block')
+          sel => sel.is(TextSelection) || sel.is(BlockSelection)
         );
         if (!sel) return;
         let model: BlockModel | null = null;
@@ -262,8 +266,8 @@ export class PageRootBlockComponent extends BlockComponent<
         if (!model) return;
         const prevNote = this.doc.getPrev(model);
         if (!prevNote || prevNote.flavour !== 'affine:note') {
-          const isFirstText = sel.is('text') && sel.start.index === 0;
+          const isFirstText = sel.is(TextSelection) && sel.start.index === 0;
-          const isBlock = sel.is('block');
+          const isBlock = sel.is(BlockSelection);
           if (isBlock || isFirstText) {
             focusTitle(this.host);
           }
@@ -361,7 +365,7 @@ export class PageRootBlockComponent extends BlockComponent<
         .then(() => {
           if (!newTextSelectionId) return;
           this.host.selection.setGroup('note', [
-            this.host.selection.create('text', {
+            this.host.selection.create(TextSelection, {
               from: {
                 blockId: newTextSelectionId,
                 index: 0,

blocksuite/blocks/src/root-block/widgets/embed-card-toolbar/embed-card-toolbar.ts

@@ -52,7 +52,12 @@ import {
   ThemeProvider,
 } from '@blocksuite/affine-shared/services';
 import { getHostName, referenceToNode } from '@blocksuite/affine-shared/utils';
-import { type BlockStdScope, WidgetComponent } from '@blocksuite/block-std';
+import {
+  BlockSelection,
+  type BlockStdScope,
+  TextSelection,
+  WidgetComponent,
+} from '@blocksuite/block-std';
 import { type BlockModel, Text } from '@blocksuite/store';
 import { autoUpdate, computePosition, flip, offset } from '@floating-ui/dom';
 import { html, nothing, type TemplateResult } from 'lit';
@@ -709,13 +714,13 @@ export class EmbedCardToolbar extends WidgetComponent<
 
     this.disposables.add(
       this._selection.slots.changed.on(() => {
-        const hasTextSelection = this._selection.find('text');
+        const hasTextSelection = this._selection.find(TextSelection);
         if (hasTextSelection) {
           this._hide();
           return;
         }
 
-        const blockSelections = this._selection.filter('block');
+        const blockSelections = this._selection.filter(BlockSelection);
         if (!blockSelections || blockSelections.length !== 1) {
           this._hide();
           return;

blocksuite/blocks/src/root-block/widgets/format-bar/format-bar.ts

@@ -8,12 +8,15 @@ import {
 } from '@blocksuite/affine-components/toolbar';
 import type { AffineTextAttributes } from '@blocksuite/affine-shared/types';
 import { matchFlavours } from '@blocksuite/affine-shared/utils';
-import type {
+import {
-  BaseSelection,
+  type BaseSelection,
-  BlockComponent,
+  type BlockComponent,
+  BlockSelection,
   CursorSelection,
+  TextSelection,
+  WidgetComponent,
 } from '@blocksuite/block-std';
-import { WidgetComponent } from '@blocksuite/block-std';
+import { DatabaseSelection } from '@blocksuite/data-view';
 import {
   assertExists,
   DisposableGroup,
@@ -129,11 +132,12 @@ export class AffineFormatBarWidget extends WidgetComponent {
     this.disposables.add(
       this._selectionManager.slots.changed.on(() => {
         const update = async () => {
-          const textSelection = rootComponent.selection.find('text');
+          const textSelection = rootComponent.selection.find(TextSelection);
-          const blockSelections = rootComponent.selection.filter('block');
+          const blockSelections =
+            rootComponent.selection.filter(BlockSelection);
 
           // Should not re-render format bar when only cursor selection changed in edgeless
-          const cursorSelection = rootComponent.selection.find('cursor');
+          const cursorSelection = rootComponent.selection.find(CursorSelection);
           if (cursorSelection) {
             if (!this._lastCursor) {
               this._lastCursor = cursorSelection;
@@ -202,7 +206,7 @@ export class AffineFormatBarWidget extends WidgetComponent {
     this.disposables.addFromEvent(document, 'selectionchange', () => {
       if (!this.host.event.active) return;
 
-      const databaseSelection = this.host.selection.find('database');
+      const databaseSelection = this.host.selection.find(DatabaseSelection);
       if (!databaseSelection) {
         return;
       }

blocksuite/blocks/src/root-block/widgets/image-toolbar/index.ts

@@ -10,7 +10,11 @@ import {
 } from '@blocksuite/affine-components/toolbar';
 import type { ImageBlockModel } from '@blocksuite/affine-model';
 import { PAGE_HEADER_HEIGHT } from '@blocksuite/affine-shared/consts';
-import { WidgetComponent } from '@blocksuite/block-std';
+import {
+  BlockSelection,
+  TextSelection,
+  WidgetComponent,
+} from '@blocksuite/block-std';
 import { limitShift, shift } from '@floating-ui/dom';
 import { html } from 'lit';
 
@@ -35,7 +39,7 @@ export class AffineImageToolbarWidget extends WidgetComponent<
         const imageBlock = this.block;
         const selection = this.host.selection;
 
-        const textSelection = selection.find('text');
+        const textSelection = selection.find(TextSelection);
         if (
           !!textSelection &&
           (!!textSelection.to || !!textSelection.from.length)
@@ -43,7 +47,7 @@ export class AffineImageToolbarWidget extends WidgetComponent<
           return null;
         }
 
-        const blockSelections = selection.filter('block');
+        const blockSelections = selection.filter(BlockSelection);
         if (
           blockSelections.length > 1 ||
           (blockSelections.length === 1 &&

blocksuite/blocks/src/root-block/widgets/image-toolbar/utils.ts

@@ -3,6 +3,7 @@ import {
   getBlockProps,
   isInsidePageEditor,
 } from '@blocksuite/affine-shared/utils';
+import { BlockSelection } from '@blocksuite/block-std';
 import { assertExists } from '@blocksuite/global/utils';
 
 export function duplicate(
@@ -38,7 +39,7 @@ export function duplicate(
     .then(() => {
       const { selection } = editorHost;
       selection.setGroup('note', [
-        selection.create('block', {
+        selection.create(BlockSelection, {
           blockId: duplicateId,
         }),
       ]);

blocksuite/blocks/src/root-block/widgets/page-dragging-area/page-dragging-area.ts

@@ -7,6 +7,7 @@ import {
 import {
   BLOCK_ID_ATTR,
   BlockComponent,
+  BlockSelection,
   type PointerEventState,
   WidgetComponent,
 } from '@blocksuite/block-std';
@@ -188,7 +189,7 @@ export class AffinePageDraggingAreaWidget extends WidgetComponent<
       this._allBlocksWithRect,
       userRect
     ).map(blockPath => {
-      return this.host.selection.create('block', {
+      return this.host.selection.create(BlockSelection, {
         blockId: blockPath,
       });
     });

blocksuite/blocks/src/root-block/widgets/slash-menu/index.ts

@@ -7,7 +7,7 @@ import {
   matchFlavours,
 } from '@blocksuite/affine-shared/utils';
 import type { UIEventStateContext } from '@blocksuite/block-std';
-import { WidgetComponent } from '@blocksuite/block-std';
+import { TextSelection, WidgetComponent } from '@blocksuite/block-std';
 import {
   assertExists,
   assertType,
@@ -122,7 +122,7 @@ export class AffineSlashMenuWidget extends WidgetComponent {
       }
     }
 
-    const textSelection = this.host.selection.find('text');
+    const textSelection = this.host.selection.find(TextSelection);
     if (!textSelection) return;
 
     const model = this.host.doc.getBlock(textSelection.blockId)?.model;
@@ -149,7 +149,7 @@ export class AffineSlashMenuWidget extends WidgetComponent {
     assertType<RootBlockComponent>(rootComponent);
 
     inlineRangeApplyCallback(() => {
-      const textSelection = this.host.selection.find('text');
+      const textSelection = this.host.selection.find(TextSelection);
       if (!textSelection) return;
 
       const model = this.host.doc.getBlock(textSelection.blockId)?.model;

blocksuite/blocks/src/root-block/widgets/slash-menu/utils.ts

@@ -3,6 +3,7 @@ import type {
   TextFormatConfig,
 } from '@blocksuite/affine-components/rich-text';
 import { isInsideBlockByFlavour } from '@blocksuite/affine-shared/utils';
+import { BlockSelection } from '@blocksuite/block-std';
 import { assertType } from '@blocksuite/global/utils';
 import type { BlockModel } from '@blocksuite/store';
 
@@ -128,7 +129,7 @@ export function createTextFormatItem(
           .chain()
           .formatBlock({
             blockSelections: [
-              std.selection.create('block', {
+              std.selection.create(BlockSelection, {
                 blockId: model.id,
               }),
             ],

blocksuite/blocks/src/root-block/widgets/surface-ref-toolbar/surface-ref-toolbar.ts

@@ -19,7 +19,11 @@ import {
 } from '@blocksuite/affine-components/toolbar';
 import type { SurfaceRefBlockModel } from '@blocksuite/affine-model';
 import { PAGE_HEADER_HEIGHT } from '@blocksuite/affine-shared/consts';
-import { WidgetComponent } from '@blocksuite/block-std';
+import {
+  BlockSelection,
+  TextSelection,
+  WidgetComponent,
+} from '@blocksuite/block-std';
 import { offset, shift } from '@floating-ui/dom';
 import { html, nothing } from 'lit';
 import { ifDefined } from 'lit/directives/if-defined.js';
@@ -48,7 +52,7 @@ export class AffineSurfaceRefToolbar extends WidgetComponent<
       const surfaceRefBlock = this.block;
       const selection = this.host.selection;
 
-      const textSelection = selection.find('text');
+      const textSelection = selection.find(TextSelection);
       if (
         !!textSelection &&
         (!!textSelection.to || !!textSelection.from.length)
@@ -56,7 +60,7 @@ export class AffineSurfaceRefToolbar extends WidgetComponent<
         return null;
       }
 
-      const blockSelections = selection.filter('block');
+      const blockSelections = selection.filter(BlockSelection);
       if (
         blockSelections.length > 1 ||
         (blockSelections.length === 1 &&

blocksuite/framework/block-std/src/gfx/selection.ts

@@ -7,7 +7,12 @@ import {
   Slot,
 } from '@blocksuite/global/utils';
 
-import type { CursorSelection, SurfaceSelection } from '../selection/index.js';
+import {
+  BlockSelection,
+  CursorSelection,
+  SurfaceSelection,
+  TextSelection,
+} from '../selection/index.js';
 import type { GfxController } from './controller.js';
 import { GfxExtension, GfxExtensionIdentifier } from './extension.js';
 import type { GfxModel } from './model/model.js';
@@ -215,9 +220,9 @@ export class GfxSelectionManager extends GfxExtension {
     this.disposable.add(
       this.stdSelection.slots.changed.on(selections => {
         const { cursor = [], surface = [] } = groupBy(selections, sel => {
-          if (sel.is('surface')) {
+          if (sel.is(SurfaceSelection)) {
             return 'surface';
-          } else if (sel.is('cursor')) {
+          } else if (sel.is(CursorSelection)) {
             return 'cursor';
           }
 
@@ -261,15 +266,15 @@ export class GfxSelectionManager extends GfxExtension {
           let hasBlockSelection = false;
 
           selections.forEach(selection => {
-            if (selection.is('text')) {
+            if (selection.is(TextSelection)) {
               hasTextSelection = true;
             }
 
-            if (selection.is('block')) {
+            if (selection.is(BlockSelection)) {
               hasBlockSelection = true;
             }
 
-            if (selection.is('surface')) {
+            if (selection.is(SurfaceSelection)) {
               const surfaceSelections = surfaceMap.get(id) ?? [];
               surfaceSelections.push(selection);
               surfaceMap.set(id, surfaceSelections);
@@ -277,7 +282,7 @@ export class GfxSelectionManager extends GfxExtension {
               selection.elements.forEach(id => selectedSet.add(id));
             }
 
-            if (selection.is('cursor')) {
+            if (selection.is(CursorSelection)) {
               cursorMap.set(id, selection);
             }
           });
@@ -318,7 +323,7 @@ export class GfxSelectionManager extends GfxExtension {
     if (elements.length > 0 && this.surfaceModel) {
       instances.push(
         this.stdSelection.create(
-          'surface',
+          SurfaceSelection,
           this.surfaceModel.id,
           elements,
           selection.editing ?? false,
@@ -331,7 +336,7 @@ export class GfxSelectionManager extends GfxExtension {
       instances = instances.concat(
         blocks.map(blockId =>
           this.stdSelection.create(
-            'surface',
+            SurfaceSelection,
             blockId,
             [blockId],
             selection.editing ?? false,
@@ -368,7 +373,11 @@ export class GfxSelectionManager extends GfxExtension {
   }
 
   setCursor(cursor: CursorSelection | IPoint) {
-    const instance = this.stdSelection.create('cursor', cursor.x, cursor.y);
+    const instance = this.stdSelection.create(
+      CursorSelection,
+      cursor.x,
+      cursor.y
+    );
 
     this.stdSelection.setGroup('gfx', [...this.surfaceSelections, instance]);
   }

blocksuite/framework/block-std/src/range/inline-range-provider.ts

@@ -1,7 +1,7 @@
 import type { InlineRange, InlineRangeProvider } from '@blocksuite/inline';
 import { signal } from '@preact/signals-core';
 
-import type { TextSelection } from '../selection/index.js';
+import { TextSelection } from '../selection/index.js';
 import type { BlockComponent } from '../view/element/block-component.js';
 
 export const getInlineRangeProvider: (
@@ -40,7 +40,7 @@ export const getInlineRangeProvider: (
     }
 
     const elementRange = rangeManager.textSelectionToRange(
-      selectionManager.create('text', {
+      selectionManager.create(TextSelection, {
         from: {
           index: 0,
           blockId: element.blockId,
@@ -72,7 +72,7 @@ export const getInlineRangeProvider: (
     if (!inlineRange) {
       selectionManager.clear(['text']);
     } else {
-      const textSelection = selectionManager.create('text', {
+      const textSelection = selectionManager.create(TextSelection, {
         from: {
           blockId: element.blockId,
           index: inlineRange.index,

blocksuite/framework/block-std/src/range/range-binding.ts

@@ -1,7 +1,7 @@
 import { throttle } from '@blocksuite/global/utils';
 import type { BlockModel } from '@blocksuite/store';
 
-import type { BaseSelection, TextSelection } from '../selection/index.js';
+import { type BaseSelection, TextSelection } from '../selection/index.js';
 import type { BlockComponent } from '../view/element/block-component.js';
 import { BLOCK_ID_ATTR } from '../view/index.js';
 import { RANGE_SYNC_EXCLUDE_ATTR } from './consts.js';
@@ -30,7 +30,7 @@ export class RangeBinding {
   };
 
   private readonly _onBeforeInput = (event: InputEvent) => {
-    const selection = this.selectionManager.find('text');
+    const selection = this.selectionManager.find(TextSelection);
     if (!selection) return;
 
     if (event.isComposing) return;
@@ -74,7 +74,7 @@ export class RangeBinding {
         });
     });
 
-    const newSelection = this.selectionManager.create('text', {
+    const newSelection = this.selectionManager.create(TextSelection, {
       from: {
         blockId: from.blockId,
         index: from.index + (event.data?.length ?? 0),
@@ -95,7 +95,7 @@ export class RangeBinding {
   };
 
   private readonly _onCompositionStart = () => {
-    const selection = this.selectionManager.find('text');
+    const selection = this.selectionManager.find(TextSelection);
     if (!selection) return;
 
     const { from, to } = selection;
@@ -153,7 +153,7 @@ export class RangeBinding {
 
       await this.host.updateComplete;
 
-      const selection = this.selectionManager.create('text', {
+      const selection = this.selectionManager.create(TextSelection, {
         from: {
           blockId: from.blockId,
           index: from.index + (event.data?.length ?? 0),
@@ -249,7 +249,7 @@ export class RangeBinding {
   private readonly _onStdSelectionChanged = (selections: BaseSelection[]) => {
     const text =
       selections.find((selection): selection is TextSelection =>
-        selection.is('text')
+        selection.is(TextSelection)
       ) ?? null;
 
     if (text === this._prevTextSelection) {

blocksuite/framework/block-std/src/range/range-manager.ts

@@ -1,7 +1,7 @@
 import { INLINE_ROOT_ATTR, type InlineRootElement } from '@blocksuite/inline';
 
 import { LifeCycleWatcher } from '../extension/index.js';
-import type { TextSelection } from '../selection/index.js';
+import { TextSelection } from '../selection/index.js';
 import type { BlockComponent } from '../view/element/block-component.js';
 import { BLOCK_ID_ATTR } from '../view/index.js';
 import { RANGE_QUERY_EXCLUDE_ATTR, RANGE_SYNC_EXCLUDE_ATTR } from './consts.js';
@@ -163,7 +163,7 @@ export class RangeManager extends LifeCycleWatcher {
       return null;
     }
 
-    return this.std.host.selection.create('text', {
+    return this.std.host.selection.create(TextSelection, {
       from: {
         blockId: startBlock.blockId,
         index: startInlineRange.index,

blocksuite/framework/block-std/src/selection/base.ts

@@ -1,10 +1,6 @@
 import { BlockSuiteError, ErrorCode } from '@blocksuite/global/exceptions';
 
-type SelectionConstructor<T = unknown> = {
+import type { SelectionConstructor } from './manager';
-  type: string;
-  group: string;
-  new (...args: unknown[]): T;
-};
 
 export type BaseSelectionOptions = {
   blockId: string;
@@ -21,9 +17,8 @@ export abstract class BaseSelection {
     return (this.constructor as SelectionConstructor).group;
   }
 
-  get type(): BlockSuite.SelectionType {
+  get type(): string {
-    return (this.constructor as SelectionConstructor)
+    return (this.constructor as SelectionConstructor).type as string;
-      .type as BlockSuite.SelectionType;
   }
 
   constructor({ blockId }: BaseSelectionOptions) {
@@ -39,10 +34,10 @@ export abstract class BaseSelection {
 
   abstract equals(other: BaseSelection): boolean;
 
-  is<T extends BlockSuite.SelectionType>(
+  is<T extends SelectionConstructor>(
     type: T
-  ): this is BlockSuite.SelectionInstance[T] {
+  ): this is T extends SelectionConstructor<infer U> ? U : never {
-    return this.type === type;
+    return this.type === type.type;
   }
 
   abstract toJSON(): Record<string, unknown>;

blocksuite/framework/block-std/src/selection/index.ts

@@ -1,27 +1,3 @@
-import type {
-  BlockSelection,
-  CursorSelection,
-  SurfaceSelection,
-  TextSelection,
-} from './variants/index.js';
-
 export * from './base.js';
 export * from './manager.js';
 export * from './variants/index.js';
-
-declare global {
-  namespace BlockSuite {
-    interface Selection {
-      block: typeof BlockSelection;
-      cursor: typeof CursorSelection;
-      surface: typeof SurfaceSelection;
-      text: typeof TextSelection;
-    }
-
-    type SelectionType = keyof Selection;
-
-    type SelectionInstance = {
-      [P in SelectionType]: InstanceType<Selection[P]>;
-    };
-  }
-}

blocksuite/framework/block-std/src/selection/manager.ts

@@ -8,11 +8,12 @@ import { SelectionIdentifier } from '../identifier.js';
 import type { BlockStdScope } from '../scope/index.js';
 import type { BaseSelection } from './base.js';
 
-export interface SelectionConstructor {
+export interface SelectionConstructor<T extends BaseSelection = BaseSelection> {
   type: string;
+  group: string;
 
-  new (...args: any[]): BaseSelection;
+  new (...args: any[]): T;
-  fromJSON(json: Record<string, unknown>): BaseSelection;
+  fromJSON(json: Record<string, unknown>): T;
 }
 
 export class SelectionManager extends LifeCycleWatcher {
@@ -143,18 +144,11 @@ export class SelectionManager extends LifeCycleWatcher {
     }
   }
 
-  create<T extends BlockSuite.SelectionType>(
+  create<T extends SelectionConstructor>(
-    type: T,
+    Type: T,
-    ...args: ConstructorParameters<BlockSuite.Selection[T]>
+    ...args: ConstructorParameters<T>
-  ): BlockSuite.SelectionInstance[T] {
+  ): InstanceType<T> {
-    const ctor = this._selectionConstructors[type];
+    return new Type(...args) as InstanceType<T>;
-    if (!ctor) {
-      throw new BlockSuiteError(
-        ErrorCode.SelectionError,
-        `Unknown selection type: ${type}`
-      );
-    }
-    return new ctor(...args) as BlockSuite.SelectionInstance[T];
   }
 
   dispose() {
@@ -162,27 +156,23 @@ export class SelectionManager extends LifeCycleWatcher {
     this.disposables.dispose();
   }
 
-  filter<T extends BlockSuite.SelectionType>(type: T) {
+  filter<T extends SelectionConstructor>(type: T) {
     return this.filter$(type).value;
   }
 
-  filter$<T extends BlockSuite.SelectionType>(type: T) {
+  filter$<T extends SelectionConstructor>(type: T) {
     return computed(() =>
-      this.value.filter((sel): sel is BlockSuite.SelectionInstance[T] =>
+      this.value.filter((sel): sel is InstanceType<T> => sel.is(type))
-        sel.is(type)
-      )
     );
   }
 
-  find<T extends BlockSuite.SelectionType>(type: T) {
+  find<T extends SelectionConstructor>(type: T) {
     return this.find$(type).value;
   }
 
-  find$<T extends BlockSuite.SelectionType>(type: T) {
+  find$<T extends SelectionConstructor>(type: T) {
     return computed(() =>
-      this.value.find((sel): sel is BlockSuite.SelectionInstance[T] =>
+      this.value.find((sel): sel is InstanceType<T> => sel.is(type))
-        sel.is(type)
-      )
     );
   }
 

blocksuite/presets/src/fragments/comment/comment-input.ts

@@ -1,5 +1,4 @@
-import type { TextSelection } from '@blocksuite/block-std';
+import { ShadowlessElement, TextSelection } from '@blocksuite/block-std';
-import { ShadowlessElement } from '@blocksuite/block-std';
 import type { RichText } from '@blocksuite/blocks';
 import { WithDisposable } from '@blocksuite/global/utils';
 import { css, html, nothing } from 'lit';
@@ -69,7 +68,7 @@ export class CommentInput extends WithDisposable(ShadowlessElement) {
   }
 
   override render() {
-    const textSelection = this.host.selection.find('text');
+    const textSelection = this.host.selection.find(TextSelection);
     if (!textSelection) {
       this.remove();
       return nothing;

blocksuite/presets/src/fragments/comment/comment-panel.ts

@@ -1,4 +1,4 @@
-import { ShadowlessElement } from '@blocksuite/block-std';
+import { ShadowlessElement, TextSelection } from '@blocksuite/block-std';
 import { WithDisposable } from '@blocksuite/global/utils';
 import { css, html } from 'lit';
 import { property, query } from 'lit/decorators.js';
@@ -60,7 +60,7 @@ export class CommentPanel extends WithDisposable(ShadowlessElement) {
   commentManager: CommentManager | null = null;
 
   private _addComment() {
-    const textSelection = this.editor.host?.selection.find('text');
+    const textSelection = this.editor.host?.selection.find(TextSelection);
     if (!textSelection) return;
 
     const commentInput = new CommentInput();

blocksuite/tests-legacy/utils/asserts.ts

@@ -11,7 +11,11 @@ import {
   DEFAULT_NOTE_WIDTH,
   DefaultTheme,
 } from '@blocksuite/affine-model';
-import type { BlockComponent, EditorHost } from '@blocksuite/block-std';
+import type {
+  BlockComponent,
+  EditorHost,
+  TextSelection,
+} from '@blocksuite/block-std';
 import { BLOCK_ID_ATTR } from '@blocksuite/block-std';
 import { assertExists } from '@blocksuite/global/utils';
 import type { InlineRootElement } from '@inline/inline-editor.js';
@@ -1180,7 +1184,7 @@ export async function assertBlockSelections(page: Page, paths: string[]) {
     if (!host) {
       throw new Error('editor-host host not found');
     }
-    return host.selection.filter('block');
+    return host.selection.value.filter(b => b.type === 'block');
   });
   const actualPaths = selections.map(selection => selection.blockId);
   expect(actualPaths).toEqual(paths);
@@ -1199,13 +1203,13 @@ export async function assertTextSelection(
     length: number;
   }
 ) {
-  const selection = await page.evaluate(() => {
+  const selection = (await page.evaluate(() => {
     const host = document.querySelector<EditorHost>('editor-host');
     if (!host) {
       throw new Error('editor-host host not found');
     }
-    return host.selection.find('text');
+    return host.selection.value.find(b => b.type === 'text');
-  });
+  })) as TextSelection | undefined;
 
   if (!from && !to) {
     expect(selection).toBeUndefined();

packages/frontend/core/src/blocksuite/presets/_common/utils/markdown-utils.ts

@@ -1,7 +1,7 @@
 import { WorkspaceImpl } from '@affine/core/modules/workspace/impls/workspace';
-import type {
+import {
-  EditorHost,
+  type EditorHost,
-  TextRangePoint,
+  type TextRangePoint,
   TextSelection,
 } from '@blocksuite/affine/block-std';
 import {
@@ -68,7 +68,7 @@ function processSnapshot(
  */
 function processTextInSnapshot(snapshot: SliceSnapshot, host: EditorHost) {
   const { content } = snapshot;
-  const text = host.selection.find('text');
+  const text = host.selection.find(TextSelection);
   if (!content.length || !text) return;
 
   content.forEach(snapshot => processSnapshot(snapshot, text, host));

packages/frontend/core/src/blocksuite/presets/ai/_common/chat-actions-handle.ts

@@ -1,7 +1,7 @@
 import { ChatHistoryOrder } from '@affine/graphql';
-import type {
+import {
   BlockSelection,
-  EditorHost,
+  type EditorHost,
   TextSelection,
 } from '@blocksuite/affine/block-std';
 import type {
@@ -198,8 +198,8 @@ const REPLACE_SELECTION = {
   icon: ReplaceIcon,
   title: 'Replace selection',
   showWhen: (host: EditorHost) => {
-    const textSelection = host.selection.find('text');
+    const textSelection = host.selection.find(TextSelection);
-    const blockSelections = host.selection.filter('block');
+    const blockSelections = host.selection.filter(BlockSelection);
     if (
       (!textSelection || textSelection.from.length === 0) &&
       blockSelections?.length === 0

packages/frontend/core/src/blocksuite/presets/ai/_common/components/ask-ai-toolbar.ts

@@ -1,4 +1,8 @@
-import type { EditorHost } from '@blocksuite/affine/block-std';
+import {
+  BlockSelection,
+  type EditorHost,
+  TextSelection,
+} from '@blocksuite/affine/block-std';
 import {
   type AffineAIPanelWidgetConfig,
   type AIItemGroupConfig,
@@ -92,8 +96,8 @@ export class AskAIToolbarButton extends WithDisposable(LitElement) {
       this._panelRoot.style.visibility = text ? 'hidden' : 'visible';
     };
 
-    const textSelection = this.host.selection.find('text');
+    const textSelection = this.host.selection.find(TextSelection);
-    const blockSelections = this.host.selection.filter('block');
+    const blockSelections = this.host.selection.filter(BlockSelection);
     let lastBlockId: string | undefined;
     if (textSelection) {
       lastBlockId = textSelection.to?.blockId ?? textSelection.blockId;

packages/frontend/core/src/blocksuite/presets/ai/actions/doc-handler.ts

@@ -1,4 +1,4 @@
-import type { EditorHost } from '@blocksuite/affine/block-std';
+import { type EditorHost, TextSelection } from '@blocksuite/affine/block-std';
 import type {
   AffineAIPanelWidget,
   AffineAIPanelWidgetConfig,
@@ -209,7 +209,7 @@ export function actionToHandler<T extends keyof BlockSuitePresets.AIActions>(
 
 export function handleInlineAskAIAction(host: EditorHost) {
   const panel = getAIPanelWidget(host);
-  const selection = host.selection.find('text');
+  const selection = host.selection.find(TextSelection);
   const lastBlockPath = selection
     ? (selection.to?.blockId ?? selection.blockId)
     : null;

packages/frontend/core/src/blocksuite/presets/ai/actions/page-response.ts

@@ -1,4 +1,4 @@
-import type { EditorHost } from '@blocksuite/affine/block-std';
+import { type EditorHost, TextSelection } from '@blocksuite/affine/block-std';
 import {
   GfxBlockElementModel,
   type GfxModel,
@@ -214,7 +214,7 @@ async function insertMarkdownAbove(host: EditorHost) {
 }
 
 function getSelection(host: EditorHost) {
-  const textSelection = host.selection.find('text');
+  const textSelection = host.selection.find(TextSelection);
   const mode = textSelection ? 'flat' : 'highest';
   const { selectedBlocks } = getSelections(host, mode);
   if (!selectedBlocks) return;

packages/frontend/core/src/blocksuite/presets/ai/entries/code-toolbar/setup-code-toolbar.ts

@@ -9,6 +9,8 @@ const buttonOptions: AskAIButtonOptions = {
   panelWidth: 240,
 };
 
+import { BlockSelection } from '@blocksuite/affine/block-std';
+
 import type { AskAIButtonOptions } from '../../_common/components/ask-ai-button';
 import { buildAICodeItemGroups } from '../../_common/config';
 
@@ -22,7 +24,9 @@ export function setupCodeToolbarAIEntry(codeToolbar: AffineCodeToolbarWidget) {
           action: () => {
             const { selection } = host;
             selection.setGroup('note', [
-              selection.create('block', { blockId: blockComponent.blockId }),
+              selection.create(BlockSelection, {
+                blockId: blockComponent.blockId,
+              }),
             ]);
           },
           render: item =>

packages/frontend/core/src/blocksuite/presets/ai/entries/image-toolbar/setup-image-toolbar.ts

@@ -1,6 +1,9 @@
 import '../../_common/components/ask-ai-button';
 
-import type { AffineImageToolbarWidget } from '@blocksuite/affine/blocks';
+import {
+  type AffineImageToolbarWidget,
+  ImageSelection,
+} from '@blocksuite/affine/blocks';
 import { html } from 'lit';
 
 import type { AskAIButtonOptions } from '../../_common/components/ask-ai-button';
@@ -26,7 +29,9 @@ export function setupImageToolbarAIEntry(
             action: () => {
               const { selection } = host;
               selection.setGroup('note', [
-                selection.create('image', { blockId: blockComponent.blockId }),
+                selection.create(ImageSelection, {
+                  blockId: blockComponent.blockId,
+                }),
               ]);
             },
             render: item =>