cinny/src/app/hooks/useHydrateMsgDrafts.ts

import { useEffect } from 'react';
import { useStore } from 'jotai';
import { Descendant } from 'slate';
import { roomIdToMsgDraftAtomFamily } from '../state/room/roomInputDrafts';
import { DRAFT_MSG_KEY_PREFIX, hasMsgDraft } from '../utils/draft';

/**
 * On startup, pre-fill the per-room message-draft atoms from their localStorage
 * persistence (`draft-msg-<roomId>`). RoomInput only restores a draft into its
 * atom when that room's composer mounts, so without this a background room's
 * persisted draft wouldn't reflect in the room-nav draft indicator until the
 * room is opened. Runs once; only touches the same atoms RoomInput restores from
 * (identical content), so composer restore is unaffected.
 *
 * Thread drafts (key contains `::`) are skipped — the nav indicator is room-level.
 */
export function useHydrateMsgDrafts(): void {
  const store = useStore();

  useEffect(() => {
    let keys: string[];
    try {
      keys = Object.keys(localStorage);
    } catch {
      return;
    }
    keys.forEach((key) => {
      if (!key.startsWith(DRAFT_MSG_KEY_PREFIX)) return;
      const draftKey = key.slice(DRAFT_MSG_KEY_PREFIX.length);
      // Skip thread drafts (`<roomId>::<$threadRootEventId>`) — room-level only.
      // Match `::$` specifically so an IPv6-literal server name (e.g. `[::1]`)
      // in a roomId isn't mistaken for a thread key.
      if (!draftKey || draftKey.includes('::$')) return;
      try {
        const stored = localStorage.getItem(key);
        if (!stored) return;
        const nodes = JSON.parse(stored) as Descendant[];
        if (Array.isArray(nodes) && hasMsgDraft(nodes)) {
          store.set(roomIdToMsgDraftAtomFamily(draftKey), nodes);
        }
      } catch {
        // Ignore a malformed stored draft.
      }
    });
  }, [store]);
}