image-studio/view/canvas/useCanvasInput.ts

import { useEffect, useState, type RefObject } from "react";
import type { AppStore } from "@editor/store";
import { isPanInteractionMode } from "@editor/tools";
import type { GlobalKeybindConsumer, GlobalPointerConsumer, GlobalWheelConsumer } from "@input/index";
import {
  createTransformControlsInputController,
  createViewportPanInputController,
  handleArtboardSelection,
  handleViewportWheel,
  keybindEventFromKeyboardEvent,
  pointerInputEventFromPointerEvent,
  wheelInputEventFromWheelEvent,
} from "@input/index";

export type CanvasInputOptions = {
  globalKeybindConsumer: GlobalKeybindConsumer;
  globalPointerConsumer: GlobalPointerConsumer;
  globalWheelConsumer: GlobalWheelConsumer;
};

export type CanvasInputState = {
  isPanning: boolean;
};

export function useCanvasInput(
  canvasRef: RefObject<HTMLCanvasElement | null>,
  store: AppStore,
  options: CanvasInputOptions,
): CanvasInputState {
  const [isPanning, setIsPanning] = useState(false);

  useEffect(() => {
    const canvas = canvasRef.current;
    if (!canvas) return;

    const transformHandler = createTransformControlsInputController({
      getDocument: () => store.getState().document,
      getEditor: () => store.getState().editor,
      dispatch: store.dispatch,
    });

    const panHandler = createViewportPanInputController({
      globalKeyConsumer: options.globalKeybindConsumer,
      globalPointerConsumer: options.globalPointerConsumer,
      dispatch: store.dispatch,
      getCurrentZoom: () => store.getState().editor.viewport.zoom,
      isPanMode: () => isPanInteractionMode(store.getState().editor.tools.interactionMode),
    });

    const handleKeyDown = (event: KeyboardEvent) => {
      const consumed = panHandler.keyDown(keybindEventFromKeyboardEvent(event));
      if (consumed) event.preventDefault();
    };

    const handleKeyUp = (event: KeyboardEvent) => {
      const consumed = panHandler.keyUp(keybindEventFromKeyboardEvent(event));
      if (consumed) event.preventDefault();
    };

    const handlePointerDown = (event: PointerEvent) => {
      const inputEvent = pointerInputEventFromPointerEvent(event);
      const transformed = transformHandler.pointerDown(inputEvent);
      if (transformed) {
        canvas.setPointerCapture(event.pointerId);
        event.preventDefault();
        return;
      }

      const consumed = panHandler.pointerDown(inputEvent);
      if (consumed) {
        canvas.setPointerCapture(event.pointerId);
        setIsPanning(true);
        event.preventDefault();
        return;
      }

      const state = store.getState();
      const selected = state.editor.tools.activeTool === "select" && handleArtboardSelection({
        event: inputEvent,
        document: state.document,
        viewport: state.editor.viewport,
        dispatch: store.dispatch,
      });
      if (selected) event.preventDefault();
    };

    const handlePointerMove = (event: PointerEvent) => {
      const inputEvent = pointerInputEventFromPointerEvent(event);
      const transformed = transformHandler.pointerMove(inputEvent);
      if (transformed) {
        event.preventDefault();
        return;
      }

      const consumed = panHandler.pointerMove(inputEvent);
      if (consumed) event.preventDefault();
    };

    const handlePointerUp = (event: PointerEvent) => {
      const inputEvent = pointerInputEventFromPointerEvent(event);
      const transformed = transformHandler.pointerUp(inputEvent);
      if (transformed) {
        event.preventDefault();
        return;
      }

      const consumed = panHandler.pointerUp(inputEvent);
      if (!consumed) return;

      setIsPanning(false);
      event.preventDefault();
    };

    const handleWheel = (event: WheelEvent) => {
      const consumed = handleViewportWheel({
        event: wheelInputEventFromWheelEvent(event),
        globalConsumer: options.globalWheelConsumer,
        dispatch: store.dispatch,
        currentZoom: store.getState().editor.viewport.zoom,
      });

      if (consumed) event.preventDefault();
    };

    window.addEventListener("keydown", handleKeyDown);
    window.addEventListener("keyup", handleKeyUp);
    canvas.addEventListener("pointerdown", handlePointerDown);
    canvas.addEventListener("pointermove", handlePointerMove);
    canvas.addEventListener("pointerup", handlePointerUp);
    canvas.addEventListener("pointercancel", handlePointerUp);
    canvas.addEventListener("wheel", handleWheel, { passive: false });

    return () => {
      window.removeEventListener("keydown", handleKeyDown);
      window.removeEventListener("keyup", handleKeyUp);
      canvas.removeEventListener("pointerdown", handlePointerDown);
      canvas.removeEventListener("pointermove", handlePointerMove);
      canvas.removeEventListener("pointerup", handlePointerUp);
      canvas.removeEventListener("pointercancel", handlePointerUp);
      canvas.removeEventListener("wheel", handleWheel);
    };
  }, [canvasRef, options, store]);

  return { isPanning };
}