grungnet/src/hooks/useMapNotes.tsx

import Konva from "konva";
import { KonvaEventObject } from "konva/lib/Node";
import { useState } from "react";
import { v4 as uuid } from "uuid";

import Note from "../components/konva/Note";
import NoteDragOverlay from "../components/note/NoteDragOverlay";
import NoteMenu from "../components/note/NoteMenu";
import NoteTool from "../components/tools/NoteTool";
import { useBlur, useKeyboard } from "../contexts/KeyboardContext";
import { useUserId } from "../contexts/UserIdContext";
import shortcuts from "../shortcuts";
import {
  NoteChangeEventHandler,
  NoteCreateEventHander,
  NoteRemoveEventHander,
} from "../types/Events";
import { Map, MapToolId } from "../types/Map";
import { MapState } from "../types/MapState";
import {
  Note as NoteType,
  NoteDraggingOptions,
  NoteMenuOptions,
} from "../types/Note";

function useMapNotes(
  map: Map | null,
  mapState: MapState | null,
  onNoteCreate: NoteCreateEventHander,
  onNoteChange: NoteChangeEventHandler,
  onNoteRemove: NoteRemoveEventHander,
  selectedToolId: MapToolId
) {
  const userId = useUserId();
  const allowNoteEditing = !!(
    map?.owner === userId || mapState?.editFlags.includes("notes")
  );

  const [isNoteMenuOpen, setIsNoteMenuOpen] = useState<boolean>(false);
  const [noteMenuOptions, setNoteMenuOptions] = useState<NoteMenuOptions>();
  const [noteDraggingOptions, setNoteDraggingOptions] =
    useState<NoteDraggingOptions>();

  function handleNoteMenuOpen(noteId: string, noteNode: Konva.Node) {
    setNoteMenuOptions({ noteId, noteNode });
    setIsNoteMenuOpen(true);
  }

  function handleNoteMenuClose() {
    setIsNoteMenuOpen(false);
  }

  function handleNoteDragStart(_: KonvaEventObject<DragEvent>, noteId: string) {
    if (duplicateNote) {
      const note = mapState?.notes[noteId];
      if (note) {
        onNoteCreate([{ ...note, id: uuid() }]);
      }
    }
    setNoteDraggingOptions({ dragging: true, noteId });
  }

  function handleNoteDragEnd() {
    noteDraggingOptions &&
      setNoteDraggingOptions({ ...noteDraggingOptions, dragging: false });
  }

  function handleNoteRemove(noteIds: string[]) {
    onNoteRemove(noteIds);
    setNoteDraggingOptions(undefined);
  }

  const [duplicateNote, setDuplicateNote] = useState(false);
  function handleKeyDown(event: KeyboardEvent) {
    if (shortcuts.duplicate(event)) {
      setDuplicateNote(true);
    }
  }

  function handleKeyUp(event: KeyboardEvent) {
    if (shortcuts.duplicate(event)) {
      setDuplicateNote(false);
    }
  }

  function handleBlur() {
    setDuplicateNote(false);
  }

  useKeyboard(handleKeyDown, handleKeyUp);
  useBlur(handleBlur);

  const notes = (
    <NoteTool
      map={map}
      active={selectedToolId === "note"}
      onNoteCreate={onNoteCreate}
      onNoteMenuOpen={handleNoteMenuOpen}
    >
      {(mapState
        ? Object.values(mapState.notes).sort((a, b) =>
            sortNotes(a, b, noteDraggingOptions)
          )
        : []
      ).map((note) => (
        <Note
          note={note}
          map={map}
          key={note.id}
          onNoteMenuOpen={handleNoteMenuOpen}
          onNoteMenuClose={handleNoteMenuClose}
          draggable={
            allowNoteEditing &&
            (selectedToolId === "note" || selectedToolId === "move") &&
            !note.locked
          }
          onNoteChange={onNoteChange}
          onNoteDragStart={handleNoteDragStart}
          onNoteDragEnd={handleNoteDragEnd}
          fadeOnHover={selectedToolId === "drawing"}
          selected={
            !!noteMenuOptions &&
            isNoteMenuOpen &&
            noteMenuOptions.noteId === note.id
          }
        />
      ))}
    </NoteTool>
  );

  const noteMenu = (
    <NoteMenu
      isOpen={isNoteMenuOpen}
      onRequestClose={handleNoteMenuClose}
      onNoteChange={onNoteChange}
      note={noteMenuOptions && mapState?.notes[noteMenuOptions.noteId]}
      noteNode={noteMenuOptions?.noteNode}
      map={map}
    />
  );

  const noteDragOverlay = noteDraggingOptions ? (
    <NoteDragOverlay
      draggingOptions={noteDraggingOptions}
      onNoteRemove={handleNoteRemove}
    />
  ) : null;

  return { notes, noteMenu, noteDragOverlay };
}

export default useMapNotes;

function sortNotes(
  a: NoteType,
  b: NoteType,
  noteDraggingOptions?: NoteDraggingOptions
) {
  if (
    noteDraggingOptions &&
    noteDraggingOptions.dragging &&
    noteDraggingOptions.noteId === a.id
  ) {
    // If dragging token `a` move above
    return 1;
  } else if (
    noteDraggingOptions &&
    noteDraggingOptions.dragging &&
    noteDraggingOptions.noteId === b.id
  ) {
    // If dragging token `b` move above
    return -1;
  } else {
    // Else sort so last modified is on top
    return a.lastModified - b.lastModified;
  }
}