AFFiNE-Mirror/libs/components/board-sessions/src/brush-session.ts

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],
                    },
                },
            },
        };
    };
}