From 49387ec78890091cde2cf77c1a98b7abaf3704e9 Mon Sep 17 00:00:00 2001
From: Rachel Fenichel <fenichel@google.com>
Date: Mon, 7 Apr 2025 12:44:40 -0700
Subject: [PATCH] feat: add shouldHealStack method (#8872)

* feat: add shouldHealStack method

* chore: format
---
 core/dragging/block_drag_strategy.ts | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/core/dragging/block_drag_strategy.ts b/core/dragging/block_drag_strategy.ts
index 9a2cd747c..d1e5cf8d3 100644
--- a/core/dragging/block_drag_strategy.ts
+++ b/core/dragging/block_drag_strategy.ts
@@ -113,7 +113,7 @@ export class BlockDragStrategy implements IDragStrategy {
     this.workspace.setResizesEnabled(false);
     blockAnimation.disconnectUiStop();
 
-    const healStack = !!e && (e.altKey || e.ctrlKey || e.metaKey);
+    const healStack = this.shouldHealStack(e);
 
     if (this.shouldDisconnect(healStack)) {
       this.disconnectBlock(healStack);
@@ -122,6 +122,17 @@ export class BlockDragStrategy implements IDragStrategy {
     this.workspace.getLayerManager()?.moveToDragLayer(this.block);
   }
 
+  /**
+   * Get whether the drag should act on a single block or a block stack.
+   *
+   * @param e The instigating pointer event, if any.
+   * @returns True if just the initial block should be dragged out, false
+   *     if all following blocks should also be dragged.
+   */
+  protected shouldHealStack(e: PointerEvent | undefined) {
+    return !!e && (e.altKey || e.ctrlKey || e.metaKey);
+  }
+
   /** Starts a drag on a shadow, recording the drag offset. */
   private startDraggingShadow(e?: PointerEvent) {
     const parent = this.block.getParent();