init: the first public commit for AFFiNE

2026-08-02 01:49:51 +08:00 · 2022-07-22 15:49:21 +08:00
commit e3e3741393
1451 changed files with 108124 additions and 0 deletions
@@ -0,0 +1,149 @@
+import { TLBounds, Utils } from '@tldraw/core';
+import {
+    SessionType,
+    TldrawPatch,
+    TDStatus,
+    TldrawCommand,
+} from '@toeverything/components/board-types';
+import type { TldrawApp } from '@toeverything/components/board-state';
+import { BaseSession } from './base-session';
+
+export class BrushSession extends BaseSession {
+    type = SessionType.Brush;
+    performanceMode: undefined;
+    status = TDStatus.Brushing;
+    initialSelectedIds: Set<string>;
+    shapesToTest: {
+        id: string;
+        bounds: TLBounds;
+        selectId: string;
+    }[];
+
+    constructor(app: TldrawApp) {
+        super(app);
+        const { currentPageId } = app;
+        this.initialSelectedIds = new Set(this.app.selectedIds);
+        this.shapesToTest = this.app.shapes
+            .filter(
+                shape =>
+                    !(
+                        shape.isLocked ||
+                        shape.isHidden ||
+                        shape.parentId !== currentPageId ||
+                        this.initialSelectedIds.has(shape.id) ||
+                        this.initialSelectedIds.has(shape.parentId)
+                    )
+            )
+            .map(shape => ({
+                id: shape.id,
+                bounds: this.app.getShapeUtil(shape).getBounds(shape),
+                selectId: shape.id, //TLDR.getTopParentId(data, shape.id, currentPageId),
+            }));
+
+        this.update();
+    }
+
+    start = (): TldrawPatch | undefined => void null;
+
+    update = (): TldrawPatch | undefined => {
+        const {
+            initialSelectedIds,
+            shapesToTest,
+            app: { metaKey, settings, originPoint, currentPoint },
+        } = this;
+
+        // Create a bounding box between the origin and the new point
+        const brush = Utils.getBoundsFromPoints([originPoint, currentPoint]);
+
+        // Decide weather to select by intersecting or by overlapping
+        // Using a xor to revers the behaviour if the ctrl key is pressed
+        // Do it only if the user choose to enable cad like selection
+        const selectByContain = settings.isCadSelectMode
+            ? !metaKey && originPoint[0] < currentPoint[0]
+            : metaKey;
+
+        // Find ids of brushed shapes
+        const hits = new Set<string>();
+
+        const selectedIds = new Set(initialSelectedIds);
+
+        shapesToTest.forEach(({ id, selectId }) => {
+            const shape = this.app.getShape(id);
+            if (!hits.has(selectId)) {
+                const util = this.app.getShapeUtil(shape);
+                if (
+                    selectByContain
+                        ? Utils.boundsContain(brush, util.getBounds(shape))
+                        : util.hitTestBounds(shape, brush)
+                ) {
+                    hits.add(selectId);
+
+                    // When brushing a shape, select its top group parent.
+                    if (!selectedIds.has(selectId)) {
+                        selectedIds.add(selectId);
+                    }
+                } else if (selectedIds.has(selectId)) {
+                    selectedIds.delete(selectId);
+                }
+            }
+        });
+
+        const currentSelectedIds = this.app.selectedIds;
+
+        const didChange =
+            selectedIds.size !== currentSelectedIds.length ||
+            currentSelectedIds.some(id => !selectedIds.has(id));
+
+        const afterSelectedIds = didChange
+            ? Array.from(selectedIds.values())
+            : currentSelectedIds;
+
+        return {
+            appState: {
+                selectByContain,
+            },
+            document: {
+                pageStates: {
+                    [this.app.currentPageId]: {
+                        brush,
+                        selectedIds: afterSelectedIds,
+                    },
+                },
+            },
+        };
+    };
+
+    cancel = (): TldrawPatch | undefined => {
+        return {
+            appState: {
+                selectByContain: false,
+            },
+            document: {
+                pageStates: {
+                    [this.app.currentPageId]: {
+                        brush: null,
+                        selectedIds: Array.from(
+                            this.initialSelectedIds.values()
+                        ),
+                    },
+                },
+            },
+        };
+    };
+
+    complete = (): TldrawPatch | TldrawCommand | undefined => {
+        return {
+            appState: {
+                selectByContain: false,
+            },
+            document: {
+                pageStates: {
+                    [this.app.currentPageId]: {
+                        brush: null,
+                        selectedIds: [...this.app.selectedIds],
+                    },
+                },
+            },
+        };
+    };
+}