ai-chatbot/packages/nodes/src/wall/move-tool.tsx

'use client'

import {
  type AnyNodeId,
  type AutoCeilingSyncPlan,
  type AutoSlabSyncPlan,
  type CeilingNode,
  detectSpacesForLevel,
  emitter,
  type GridEvent,
  getPerpendicularWallMoveAxis,
  getPlannedLinkedWallUpdates,
  pauseSceneHistory,
  planAutoCeilingsForLevel,
  planAutoSlabsForLevel,
  planWallMoveJunctions,
  projectAutoSlabsForPlan,
  resumeSceneHistory,
  type SlabNode,
  useLiveNodeOverrides,
  useScene,
  type WallMoveAxis,
  type WallMoveJunctionPlan,
  type WallNode,
} from '@pascal-app/core'
import {
  CursorSphere,
  EDITOR_LAYER,
  getSegmentGridStep,
  isSegmentLongEnough,
  markToolCancelConsumed,
  snapBuildingLocalToWorldGrid,
  snapScalarToGrid,
  triggerSFX,
  useEditor,
} from '@pascal-app/editor'
import { useViewer } from '@pascal-app/viewer'
import { useCallback, useEffect, useMemo, useRef, useState } from 'react'
import { BufferGeometry, DoubleSide, Float32BufferAttribute } from 'three'
import {
  buildBridgeWallCreates,
  buildBridgeWallPreviews,
  type GhostWallPreview,
  getLinkedWallSnapshots,
  getWallsAfterUpdates,
  type LinkedWallSnapshot,
  stripWallIsNewMetadata,
} from './move-shared'

/**
 * Phase 5 Stage D — wall whole-move tool (kind-owned).
 *
 * 1:1 port of the legacy `MoveWallTool` (804 LoC, the most complex
 * single tool in the editor). Preserves every behavior:
 *
 *  - **Center-drag with axis lock** — wall stays perpendicular to its
 *    move axis unless rotated via R/T (45° steps).
 *  - **Linked-wall corner cascade** — neighbours sharing endpoints
 *    move with the dragged wall via `planWallMoveJunctions`.
 *  - **Bridge wall ghost previews** — when a corner separates, a
 *    translucent ghost shows the new wall that would be inserted.
 *  - **Auto-slab live preview** — `planAutoSlabsForLevel` runs every
 *    tick so room slabs adapt to the new wall layout in real time.
 *  - **Single-undo dance** — paused history during drag, restore +
 *    resume + reapply on commit so one Ctrl-Z rolls back the whole
 *    operation.
 *  - **`isNew` metadata strip** — first commit after a fresh wall
 *    placement clears the placement marker.
 *  - **Activation grace** (150ms) + Shift to bypass grid snap.
 *
 * Mounted via `def.affordanceTools.move` from `wall/definition.ts`.
 */
function rotateVector([x, z]: [number, number], angle: number): [number, number] {
  const cos = Math.cos(angle)
  const sin = Math.sin(angle)
  return [x * cos - z * sin, x * sin + z * cos]
}

function getLevelSlabs(levelId: string, nodes: ReturnType<typeof useScene.getState>['nodes']) {
  return Object.values(nodes).filter(
    (entry): entry is SlabNode => entry?.type === 'slab' && (entry.parentId ?? null) === levelId,
  )
}

function getLevelCeilings(levelId: string, nodes: ReturnType<typeof useScene.getState>['nodes']) {
  return Object.values(nodes).filter(
    (entry): entry is CeilingNode =>
      entry?.type === 'ceiling' && (entry.parentId ?? null) === levelId,
  )
}

function setPreviewGeometryAttributes(
  geometry: BufferGeometry,
  positions: number[],
  normals: number[],
  uvs: number[],
) {
  geometry.setAttribute('position', new Float32BufferAttribute(positions, 3))
  geometry.setAttribute('normal', new Float32BufferAttribute(normals, 3))
  geometry.setAttribute('uv', new Float32BufferAttribute(uvs, 2))
  geometry.setAttribute('uv2', new Float32BufferAttribute([...uvs], 2))
}

function createWallPreviewGeometry(length: number, height: number) {
  const geometry = new BufferGeometry()
  setPreviewGeometryAttributes(
    geometry,
    [0, 0, 0, length, 0, 0, length, height, 0, 0, height, 0],
    [0, 0, 1, 0, 0, 1, 0, 0, 1, 0, 0, 1],
    [0, 0, 1, 0, 1, 1, 0, 1],
  )
  geometry.setIndex([0, 1, 2, 0, 2, 3])
  geometry.computeBoundingSphere()
  return geometry
}

function GhostWallPreviewMesh({ preview }: { preview: GhostWallPreview }) {
  const dx = preview.end[0] - preview.start[0]
  const dz = preview.end[1] - preview.start[1]
  const length = Math.hypot(dx, dz)
  const angle = -Math.atan2(dz, dx)
  const geometry = useMemo(() => {
    return length < 0.01 ? null : createWallPreviewGeometry(length, preview.height)
  }, [length, preview.height])

  useEffect(() => () => geometry?.dispose(), [geometry])

  if (!geometry) {
    return null
  }

  return (
    <group position={[preview.start[0], 0.02, preview.start[1]]} rotation={[0, angle, 0]}>
      <mesh
        // Pass geometry as a prop so the mesh never renders with R3F's
        // default empty `BufferGeometry`. With `frustumCulled={false}`,
        // the `<primitive attach="geometry">` path emits one frame of
        // `Draw(0, 1, 0, 0)` against an empty buffer and WebGPU flags it
        // (see wall-move-side-handles.tsx).
        frustumCulled={false}
        geometry={geometry}
        layers={EDITOR_LAYER}
        renderOrder={2}
      >
        <meshBasicMaterial
          color={preview.color}
          depthTest={false}
          depthWrite={false}
          opacity={0.32}
          side={DoubleSide}
          transparent
        />
      </mesh>
    </group>
  )
}

export const MoveWallTool: React.FC<{ node: WallNode }> = ({ node }) => {
  const meta =
    typeof node.metadata === 'object' && node.metadata !== null && !Array.isArray(node.metadata)
      ? (node.metadata as Record<string, unknown>)
      : {}
  const isNew = !!meta.isNew
  const hasDraggedRef = useRef(false)
  const previousGridPosRef = useRef<[number, number] | null>(null)
  const originalStartRef = useRef<[number, number]>([...node.start] as [number, number])
  const originalEndRef = useRef<[number, number]>([...node.end] as [number, number])
  const originalCenterRef = useRef<[number, number]>([
    (node.start[0] + node.end[0]) / 2,
    (node.start[1] + node.end[1]) / 2,
  ])
  const originalHalfVectorRef = useRef<[number, number]>([
    (node.end[0] - node.start[0]) / 2,
    (node.end[1] - node.start[1]) / 2,
  ])
  const moveAxisRef = useRef<WallMoveAxis | null>(
    getPerpendicularWallMoveAxis(node.start, node.end),
  )
  const linkedOriginalsRef = useRef<LinkedWallSnapshot[]>(
    isNew
      ? []
      : getLinkedWallSnapshots({
          wallId: node.id,
          wallParentId: node.parentId ?? null,
          originalStart: node.start,
          originalEnd: node.end,
        }),
  )
  const dragAnchorRef = useRef<[number, number] | null>(null)
  const nodeIdRef = useRef(node.id)
  const previewRef = useRef<{ start: [number, number]; end: [number, number] } | null>(null)
  const pendingRotationRef = useRef(0)
  const shiftPressedRef = useRef(false)

  const [cursorLocalPos, setCursorLocalPos] = useState<[number, number, number]>(() => {
    const centerX = (node.start[0] + node.end[0]) / 2
    const centerZ = (node.start[1] + node.end[1]) / 2
    return [centerX, 0, centerZ]
  })
  const [ghostWallPreviews, setGhostWallPreviews] = useState<GhostWallPreview[]>([])

  const exitMoveMode = useCallback(() => {
    useEditor.getState().setMovingNode(null)
  }, [])

  useEffect(() => {
    const nodeId = nodeIdRef.current
    const originalStart = originalStartRef.current
    const originalEnd = originalEndRef.current
    const originalCenter = originalCenterRef.current
    const originalHalfVector = originalHalfVectorRef.current
    const levelId = node.parentId ?? null
    pauseSceneHistory(useScene)
    let shouldRestoreOnCleanup = true

    const applyNodePreview = (
      updates: Array<{ id: WallNode['id']; start: [number, number]; end: [number, number] }>,
    ) => {
      useScene.getState().updateNodes(
        updates.map((entry) => ({
          id: entry.id as AnyNodeId,
          data: { start: entry.start, end: entry.end },
        })),
      )
      for (const entry of updates) {
        useScene.getState().markDirty(entry.id as AnyNodeId)
      }
    }

    // Live auto-slab / auto-ceiling preview. Calculates the new
    // surfaces from the moving wall configuration each tick, then
    // sends polygon overrides to `useLiveNodeOverrides` (and marks
    // the slab / ceiling dirty) so `GeometrySystem` / `CeilingSystem`
    // rebuild the mesh through the existing `getEffectiveNode` merge
    // path. *Nothing* is written to the scene store during the drag —
    // the store stays at pre-drag values; commit writes the final
    // plan in one atomic `applyNodeChanges`. Mirrors the wall's own
    // 2D drag pattern (`useLiveNodeOverrides` → mesh, store on
    // commit).
    //
    // Creates / deletes from the plan are *deferred* to commit. We
    // can't represent a "node that doesn't exist yet" through
    // overrides, so newly-detected rooms only get their slab/ceiling
    // materialised on release; rooms whose closing wall pulls away
    // keep their slab/ceiling visible (at the original polygon) until
    // commit clears them. UX-wise this trades the previous "ghost
    // appears/disappears mid-drag" feedback for atomic, undoable
    // commits — which is what the user asked for.
    //
    // `latestSurfacePlans` holds the most recent plan from the last
    // live tick. `commitSurfacesToStore` flushes it (creates,
    // updates, deletes) into the scene as part of the commit-time
    // atomic write, then `clearSurfaceOverrides` drops the override
    // map so the system reads from the now-committed store.
    let latestSurfacePlans: { slabs: AutoSlabSyncPlan; ceilings: AutoCeilingSyncPlan } | null = null
    const touchedSlabIds = new Set<AnyNodeId>()
    const touchedCeilingIds = new Set<AnyNodeId>()

    const publishLiveSurfaceOverrides = (walls: WallNode[]) => {
      if (!levelId) return

      const levelWalls = walls.filter((wall) => (wall.parentId ?? null) === levelId)
      const sceneState = useScene.getState()
      const { roomPolygons } = detectSpacesForLevel(levelId, levelWalls)

      // Plan against the pre-drag scene — `getLevelSlabs/Ceilings`
      // reads from the unchanged store, so the matcher's "remap
      // existing room → existing slab" logic sees stable IDs across
      // ticks. Without this anchor, IDs would drift as overrides
      // re-flowed through the planner.
      const existingSlabs = getLevelSlabs(levelId, sceneState.nodes)
      const slabPlan = planAutoSlabsForLevel(roomPolygons, existingSlabs)
      const ceilingPlan = planAutoCeilingsForLevel(
        roomPolygons,
        getLevelCeilings(levelId, sceneState.nodes),
        {
          walls: levelWalls,
          slabs: projectAutoSlabsForPlan(existingSlabs, slabPlan),
        },
      )

      latestSurfacePlans = { slabs: slabPlan, ceilings: ceilingPlan }

      const overrideEntries: Array<[string, Record<string, unknown>]> = []
      for (const update of slabPlan.update) {
        if (update.data.polygon === undefined) continue
        overrideEntries.push([update.id, { polygon: update.data.polygon }])
        touchedSlabIds.add(update.id as AnyNodeId)
      }
      for (const update of ceilingPlan.update) {
        if (update.data.polygon === undefined && update.data.height === undefined) continue
        overrideEntries.push([update.id, update.data as Record<string, unknown>])
        touchedCeilingIds.add(update.id as AnyNodeId)
      }

      if (overrideEntries.length > 0) {
        useLiveNodeOverrides.getState().setMany(overrideEntries)
        for (const [id] of overrideEntries) {
          sceneState.markDirty(id as AnyNodeId)
        }
      }
    }

    // Commit-time: flush the last live plan into the scene store as
    // one atomic write, then drop overrides so the renderer reads
    // from the now-current store state. Called from `commitPreview`
    // *while history is resumed*, so the entire surface delta is one
    // undoable step alongside the wall update.
    const commitSurfacesToStore = () => {
      if (!levelId || !latestSurfacePlans) return

      const { slabs, ceilings } = latestSurfacePlans
      const update = [
        ...slabs.update.map((entry) => ({
          id: entry.id as AnyNodeId,
          data: entry.data,
        })),
        ...ceilings.update.map((entry) => ({
          id: entry.id as AnyNodeId,
          data: entry.data,
        })),
      ]
      const create = [
        ...slabs.create.map((slab) => ({
          node: slab,
          parentId: levelId as AnyNodeId,
        })),
        ...ceilings.create.map((ceiling) => ({
          node: ceiling,
          parentId: levelId as AnyNodeId,
        })),
      ]
      const deleteIds = [
        ...slabs.delete.map((id) => id as AnyNodeId),
        ...ceilings.delete.map((id) => id as AnyNodeId),
      ]

      if (update.length === 0 && create.length === 0 && deleteIds.length === 0) return

      useScene.getState().applyNodeChanges({
        update,
        create,
        delete: deleteIds,
      })
    }

    const clearSurfaceOverrides = () => {
      const overrides = useLiveNodeOverrides.getState()
      for (const id of touchedSlabIds) overrides.clear(id)
      for (const id of touchedCeilingIds) overrides.clear(id)
      touchedSlabIds.clear()
      touchedCeilingIds.clear()
      latestSurfacePlans = null
    }

    const buildWallFromCenter = (center: [number, number]) => {
      const rotatedHalf = rotateVector(originalHalfVector, pendingRotationRef.current)
      const nextStart: [number, number] = [center[0] - rotatedHalf[0], center[1] - rotatedHalf[1]]
      const nextEnd: [number, number] = [center[0] + rotatedHalf[0], center[1] + rotatedHalf[1]]
      return { start: nextStart, end: nextEnd }
    }

    const getMovePlan = (nextStart: [number, number], nextEnd: [number, number]) =>
      planWallMoveJunctions(
        linkedOriginalsRef.current,
        originalStart,
        originalEnd,
        nextStart,
        nextEnd,
      )

    const getLinkedPreviewUpdates = (
      plan: WallMoveJunctionPlan<LinkedWallSnapshot>,
      nextStart: [number, number],
      nextEnd: [number, number],
    ) => {
      const movedUpdates = getPlannedLinkedWallUpdates(
        plan,
        originalStart,
        originalEnd,
        nextStart,
        nextEnd,
      )
      const movedById = new Map(movedUpdates.map((entry) => [entry.id, entry]))

      return linkedOriginalsRef.current.map(
        (wall) => movedById.get(wall.id) ?? { id: wall.id, start: wall.start, end: wall.end },
      )
    }

    const applyPreview = (nextStart: [number, number], nextEnd: [number, number]) => {
      previewRef.current = { start: nextStart, end: nextEnd }
      const centerX = (nextStart[0] + nextEnd[0]) / 2
      const centerZ = (nextStart[1] + nextEnd[1]) / 2
      setCursorLocalPos([centerX, 0, centerZ])
      const previewPlan = getMovePlan(nextStart, nextEnd)
      const previewUpdates = [
        { id: nodeId, start: nextStart, end: nextEnd },
        ...getLinkedPreviewUpdates(previewPlan, nextStart, nextEnd),
      ]
      const previewCollapsedWallIds = new Set([
        ...previewUpdates
          .filter((entry) => entry.id !== nodeId && !isSegmentLongEnough(entry.start, entry.end))
          .map((entry) => entry.id as AnyNodeId),
        ...previewPlan.wallsToDelete.map((wall) => wall.id as AnyNodeId),
      ])
      const previewSceneWalls = getWallsAfterUpdates(
        useScene.getState().nodes,
        previewUpdates.map((entry) => ({
          id: entry.id as AnyNodeId,
          data: { start: entry.start, end: entry.end },
        })),
      ).filter((wall) => !previewCollapsedWallIds.has(wall.id as AnyNodeId))
      const bridgePreviews = buildBridgeWallPreviews({
        bridgePlans: previewPlan.bridgePlans,
        nextStart,
        nextEnd,
        existingWalls: previewSceneWalls,
      })
      const nextGhostWalls = bridgePreviews.map((preview) => preview.ghost)
      const virtualBridgeWalls = bridgePreviews.map((preview) => preview.wall)
      setGhostWallPreviews(nextGhostWalls)
      applyNodePreview(previewUpdates)
      publishLiveSurfaceOverrides([...previewSceneWalls, ...virtualBridgeWalls])
    }

    const restoreOriginal = () => {
      setGhostWallPreviews([])
      applyNodePreview([
        { id: nodeId, start: originalStart, end: originalEnd },
        ...linkedOriginalsRef.current,
      ])
      // No scene rollback for surfaces — nothing was written. Just
      // clear the live overrides so the renderer falls back to the
      // (pre-drag, unchanged) store state.
      clearSurfaceOverrides()
    }

    const onGridMove = (event: GridEvent) => {
      const rawX = event.localPosition[0]
      const rawZ = event.localPosition[2]
      const snapStep = getSegmentGridStep()

      // Anchor at the raw cursor so the displacement is measured in
      // continuous space.
      const anchor = dragAnchorRef.current ?? [rawX, rawZ]
      dragAnchorRef.current = anchor

      const rawDeltaX = rawX - anchor[0]
      const rawDeltaZ = rawZ - anchor[1]

      // When the move is axis-locked (side-handle drag), snap the wall
      // center's absolute perpendicular offset to a multiple of
      // `snapStep`. For axis-aligned walls this puts the wall on grid
      // lines along its normal regardless of the wall's starting
      // position; for diagonal walls the wall steps in exact
      // `snapStep` increments along its normal.
      //
      // Why absolute (perp · centre) rather than `delta` snap: snapping
      // the displacement alone preserves any pre-existing off-grid
      // offset, so the wall could only ever sit at `originalCentre ±
      // k·snapStep` — never on actual grid lines. Snapping the wall's
      // own perpendicular coordinate fixes that.
      const axis = moveAxisRef.current
      let deltaX: number
      let deltaZ: number
      if (axis) {
        const originalProj = originalCenter[0] * axis[0] + originalCenter[1] * axis[1]
        const rawProj = originalProj + rawDeltaX * axis[0] + rawDeltaZ * axis[1]
        const snappedProj = shiftPressedRef.current ? rawProj : snapScalarToGrid(rawProj, snapStep)
        const perpDelta = snappedProj - originalProj
        deltaX = axis[0] * perpDelta
        deltaZ = axis[1] * perpDelta
      } else if (shiftPressedRef.current) {
        deltaX = rawDeltaX
        deltaZ = rawDeltaZ
      } else {
        // Snap the resulting wall center to the WORLD XZ grid (projected
        // back into building-local), then express the result as a delta
        // from the original centre. Without this, a rotated building
        // dragged the wall along its local axes instead of world ones.
        const targetLocal: [number, number] = [
          originalCenter[0] + rawDeltaX,
          originalCenter[1] + rawDeltaZ,
        ]
        const snappedLocal = snapBuildingLocalToWorldGrid(targetLocal, snapStep)
        deltaX = snappedLocal[0] - originalCenter[0]
        deltaZ = snappedLocal[1] - originalCenter[1]
      }

      const constrainedGridPos: [number, number] = [anchor[0] + deltaX, anchor[1] + deltaZ]

      if (
        previousGridPosRef.current &&
        (constrainedGridPos[0] !== previousGridPosRef.current[0] ||
          constrainedGridPos[1] !== previousGridPosRef.current[1])
      ) {
        triggerSFX('sfx:grid-snap')
      }
      previousGridPosRef.current = constrainedGridPos

      const nextCenter: [number, number] = [originalCenter[0] + deltaX, originalCenter[1] + deltaZ]
      const nextWall = buildWallFromCenter(nextCenter)
      hasDraggedRef.current = true
      applyPreview(nextWall.start, nextWall.end)
    }

    const commitPreview = () => {
      const preview = previewRef.current ?? { start: originalStart, end: originalEnd }

      shouldRestoreOnCleanup = false

      // Restore original baseline while paused so the next resume+update
      // registers as a single tracked change (undo reverts to original).
      // Surfaces stayed in the store the whole drag (override-driven
      // mesh preview), so there's nothing to restore for them.
      setGhostWallPreviews([])
      applyNodePreview([
        { id: nodeId, start: originalStart, end: originalEnd },
        ...linkedOriginalsRef.current,
      ])

      resumeSceneHistory(useScene)
      const commitPlan = getMovePlan(preview.start, preview.end)
      const linkedWallUpdates = getPlannedLinkedWallUpdates(
        commitPlan,
        originalStart,
        originalEnd,
        preview.start,
        preview.end,
      )
      const collapsedLinkedWallIds = new Set([
        ...linkedWallUpdates
          .filter((entry) => !isSegmentLongEnough(entry.start, entry.end))
          .map((entry) => entry.id as AnyNodeId),
        ...commitPlan.wallsToDelete.map((wall) => wall.id as AnyNodeId),
      ])

      const commitUpdates = [
        {
          id: nodeId as AnyNodeId,
          data: isNew
            ? {
                start: preview.start,
                end: preview.end,
                metadata: stripWallIsNewMetadata(node.metadata),
              }
            : { start: preview.start, end: preview.end },
        },
        ...linkedWallUpdates
          .filter((entry) => !collapsedLinkedWallIds.has(entry.id as AnyNodeId))
          .map((entry) => ({
            id: entry.id as AnyNodeId,
            data: { start: entry.start, end: entry.end },
          })),
      ]
      const sceneState = useScene.getState()
      const existingWalls = getWallsAfterUpdates(sceneState.nodes, commitUpdates).filter(
        (wall) => !collapsedLinkedWallIds.has(wall.id as AnyNodeId),
      )
      const bridgeCreates = buildBridgeWallCreates({
        bridgePlans: commitPlan.bridgePlans,
        nextStart: preview.start,
        nextEnd: preview.end,
        existingWalls,
        wallCount: Object.values(sceneState.nodes).filter((entry) => entry?.type === 'wall').length,
      })
      sceneState.applyNodeChanges({
        update: commitUpdates,
        create: bridgeCreates,
        delete: Array.from(collapsedLinkedWallIds),
      })

      // Flush the last live surface plan (slab + ceiling creates,
      // updates, deletes) into the store while history is still
      // resumed, so the surface delta joins the wall change as one
      // undoable step. Then drop the live overrides — the renderer
      // now reads the committed polygons directly.
      commitSurfacesToStore()
      clearSurfaceOverrides()

      pauseSceneHistory(useScene)

      // Claim teardown ownership so the 2D overlay's cleanup skips its
      // own revert when split-view has both mounted — see
      // `movingNodeOrigin` in `use-editor.tsx`.
      useEditor.getState().setMovingNodeOrigin('3d')

      triggerSFX('sfx:item-place')
      useViewer.getState().setSelection({ selectedIds: [nodeId] })
      exitMoveMode()
    }

    const onPointerUp = () => {
      // Press-release without drag: dismiss the tool without committing.
      // This is the same UX as MoveWallEndpointTool / WallHeightArrowHandle
      // — pointer-down on the affordance starts the move, drag updates the
      // preview, release commits if the cursor actually moved.
      if (!hasDraggedRef.current) {
        useViewer.getState().setSelection({ selectedIds: [nodeId] })
        exitMoveMode()
        return
      }
      commitPreview()
    }

    const onKeyDown = (event: KeyboardEvent) => {
      if (event.target instanceof HTMLInputElement || event.target instanceof HTMLTextAreaElement) {
        return
      }

      if (event.key === 'Shift') {
        shiftPressedRef.current = true
        return
      }

      const ROTATION_STEP = Math.PI / 4
      let rotationDelta = 0
      if (event.key === 'r' || event.key === 'R') rotationDelta = ROTATION_STEP
      else if (event.key === 't' || event.key === 'T') rotationDelta = -ROTATION_STEP

      if (rotationDelta === 0) {
        return
      }

      event.preventDefault()
      pendingRotationRef.current += rotationDelta
      triggerSFX('sfx:item-rotate')

      const preview = previewRef.current ?? { start: originalStart, end: originalEnd }
      const currentCenter: [number, number] = [
        (preview.start[0] + preview.end[0]) / 2,
        (preview.start[1] + preview.end[1]) / 2,
      ]
      const nextWall = buildWallFromCenter(currentCenter)
      moveAxisRef.current = getPerpendicularWallMoveAxis(nextWall.start, nextWall.end)
      applyPreview(nextWall.start, nextWall.end)
    }

    const onKeyUp = (event: KeyboardEvent) => {
      if (event.key === 'Shift') {
        shiftPressedRef.current = false
      }
    }

    const onCancel = () => {
      shouldRestoreOnCleanup = false
      restoreOriginal()
      useViewer.getState().setSelection({ selectedIds: [nodeId] })
      resumeSceneHistory(useScene)
      markToolCancelConsumed()
      // Claim teardown ownership so the 2D overlay doesn't redundantly
      // revert the same baseline on its own cleanup.
      useEditor.getState().setMovingNodeOrigin('3d')
      exitMoveMode()
    }

    emitter.on('grid:move', onGridMove)
    emitter.on('tool:cancel', onCancel)
    window.addEventListener('pointerup', onPointerUp)
    window.addEventListener('keydown', onKeyDown)
    window.addEventListener('keyup', onKeyUp)

    return () => {
      if (shouldRestoreOnCleanup) {
        // `shouldRestoreOnCleanup` is only true if neither `onPointerUp`
        // (commit branch) nor `onCancel` ran — i.e., the unmount came
        // from outside (typically the 2D overlay finalising in split
        // view). The origin flag tells us whether the 2D side committed
        // (skip restore — its write is the live state) or the unmount
        // has no claimed owner (restore to baseline).
        const finalisedBy2D = useEditor.getState().movingNodeOrigin === '2d'
        if (!finalisedBy2D) {
          restoreOriginal()
        }
      }
      shiftPressedRef.current = false
      resumeSceneHistory(useScene)
      emitter.off('grid:move', onGridMove)
      emitter.off('tool:cancel', onCancel)
      window.removeEventListener('pointerup', onPointerUp)
      window.removeEventListener('keydown', onKeyDown)
      window.removeEventListener('keyup', onKeyUp)
    }
  }, [exitMoveMode, isNew, node.metadata, node.parentId])

  return (
    <group>
      <CursorSphere position={cursorLocalPos} showTooltip={false} />
      {ghostWallPreviews.map((preview) => (
        <GhostWallPreviewMesh key={preview.id} preview={preview} />
      ))}
    </group>
  )
}

export default MoveWallTool