snapshot of https://github.com/toeverything/AFFiNE.git (canary, 2026-09-16; blobs > 131072B omitted)

blocksuite/affine/blocks/root/src/edgeless/utils/query.ts

@@ -0,0 +1,104 @@
+import type { CanvasElementWithText } from '@blocksuite/affine-block-surface';
+import type { PanTool } from '@blocksuite/affine-gfx-pointer';
+import {
+  type AttachmentBlockModel,
+  type Connectable,
+  type EdgelessTextBlockModel,
+  type EmbedSyncedDocModel,
+  type ImageBlockModel,
+  ShapeElementModel,
+  TextElementModel,
+} from '@blocksuite/affine-model';
+import { isTopLevelBlock } from '@blocksuite/affine-shared/utils';
+import type {
+  GfxModel,
+  GfxPrimitiveElementModel,
+  ToolOptionWithType,
+} from '@blocksuite/std/gfx';
+import type { BlockModel } from '@blocksuite/store';
+
+import { drawingCursor } from './cursors';
+
+export function isEdgelessTextBlock(
+  element: BlockModel | GfxModel | null
+): element is EdgelessTextBlockModel {
+  return (
+    !!element &&
+    'flavour' in element &&
+    element.flavour === 'affine:edgeless-text'
+  );
+}
+
+export function isImageBlock(
+  element: BlockModel | GfxModel | null
+): element is ImageBlockModel {
+  return (
+    !!element && 'flavour' in element && element.flavour === 'affine:image'
+  );
+}
+
+export function isAttachmentBlock(
+  element: BlockModel | GfxModel | null
+): element is AttachmentBlockModel {
+  return (
+    !!element && 'flavour' in element && element.flavour === 'affine:attachment'
+  );
+}
+
+export function isEmbedSyncedDocBlock(
+  element: BlockModel | GfxModel | null
+): element is EmbedSyncedDocModel {
+  return (
+    !!element &&
+    'flavour' in element &&
+    element.flavour === 'affine:embed-synced-doc'
+  );
+}
+
+export function isCanvasElement(
+  selectable: GfxModel | BlockModel | null
+): selectable is GfxPrimitiveElementModel {
+  return !isTopLevelBlock(selectable);
+}
+
+export function isCanvasElementWithText(
+  element: GfxModel
+): element is CanvasElementWithText {
+  return (
+    element instanceof TextElementModel || element instanceof ShapeElementModel
+  );
+}
+
+export function isConnectable(
+  element: GfxModel | null
+): element is Connectable {
+  return !!element && element.connectable;
+}
+
+// https://developer.mozilla.org/en-US/docs/Web/CSS/cursor
+export function getCursorMode(edgelessTool: ToolOptionWithType) {
+  if (!edgelessTool) {
+    return 'default';
+  }
+  switch (edgelessTool.toolType?.toolName) {
+    case 'default':
+      return 'default';
+    case 'pan':
+      return (edgelessTool as ToolOptionWithType<PanTool>).options?.panning
+        ? 'grabbing'
+        : 'grab';
+    case 'brush':
+    case 'highlighter':
+      return drawingCursor;
+    case 'eraser':
+    case 'shape':
+    case 'connector':
+    case 'frame':
+    case 'affine:note':
+      return 'crosshair';
+    case 'text':
+      return 'text';
+    default:
+      return 'default';
+  }
+}