cinny/src/app/hooks/usePresenceUpdater.ts

import { useEffect, useRef } from 'react';
import { SetPresence } from 'matrix-js-sdk';
import { PresenceDeduper, PresenceWrite, writePresence } from '../utils/presenceWrite';
import { useMatrixClient } from './useMatrixClient';
import { getLastLocalStatusWrite, pickHeartbeatStatus } from '../utils/ownStatusSync';
import { useSetting } from '../state/hooks/settings';
import { settingsAtom } from '../state/settings';
import {
  BUSY_PRESENCE,
  BUSY_PRESENCE_FEATURE,
  isBusyPresenceSupported,
  setBusyPresenceSupported,
} from '../utils/busyPresence';

const IDLE_TIMEOUT_MS = 10 * 60 * 1000;
const ACTIVITY_THROTTLE_MS = 1000;

export type PresenceSetting = 'auto' | 'online' | 'idle' | 'dnd' | 'invisible';
export type PresenceState = 'online' | 'unavailable' | 'offline' | typeof BUSY_PRESENCE;

/**
 * Single source of truth for mapping the user's presence preference to the
 * Matrix presence value: auto/online → 'online', idle → 'unavailable', dnd →
 * MSC3026 busy (or 'unavailable' where the server lacks it),
 * invisible (or the hidePresence override) → 'offline'. Shared with the Profile
 * status writer so setting/clearing a status message never overrides the user's
 * chosen presence (e.g. outing an Invisible user as online).
 */
export function presenceStateFromSetting(
  presenceStatus: PresenceSetting,
  hidePresence: boolean,
): PresenceState {
  if (hidePresence || presenceStatus === 'invisible') return 'offline';
  // [Gitea #283] DND is MSC3026 busy where the server supports it, so status
  // writes (save / clear / auto-clear) don't downgrade it to "unavailable".
  if (presenceStatus === 'dnd') return isBusyPresenceSupported() ? BUSY_PRESENCE : 'unavailable';
  if (presenceStatus === 'idle') return 'unavailable';
  return 'online';
}

export function usePresenceUpdater() {
  const mx = useMatrixClient();
  const [hidePresence] = useSetting(settingsAtom, 'hidePresence');
  const [presenceStatus] = useSetting(settingsAtom, 'presenceStatus');

  const idleTimerRef = useRef<ReturnType<typeof setTimeout> | undefined>(undefined);
  const isIdleRef = useRef(false);
  const lastActivityRef = useRef(0);
  const deduperRef = useRef(new PresenceDeduper());

  useEffect(() => {
    const userId = mx.getUserId();
    const deduper = deduperRef.current;
    // A manual status save changes what the next heartbeat should carry.
    deduper.reset();

    // Read status from localStorage at call time so manual updates from the
    // Profile settings are never overwritten by a stale closure value.
    const readStatus = () =>
      userId ? (localStorage.getItem(`lotus-status-msg-${userId}`) ?? '') : '';

    // Log presence failures without leaking PII (user id, token, status message).
    const warnPresenceFailure = (presence: string, err: unknown) => {
      const reason =
        err instanceof Error ? err.message : typeof err === 'string' ? err : 'unknown error';
      console.warn(`Failed to set presence to "${presence}":`, reason);
    };

    // Synapse allows ONE presence write per 10 s per user (rc_presence, shared
    // by all devices) and /sync already keeps us online — only send when the
    // state or status actually changes, so a manual status save isn't 429'd
    // by a heartbeat that changed nothing.
    const send = (write: PresenceWrite, label: string) => {
      if (!deduper.shouldSend(write)) return Promise.resolve();
      return writePresence(mx, write)
        .then(() => deduper.sent(write))
        .catch((err) => warnPresenceFailure(label, err));
    };
    // #187 DP3: a device never sees its own presence changes made on other
    // devices, so its saved status could be one another device already
    // cleared. Ask the server what the status is now before re-sending it, and
    // bring the local copy in line (falls back to the local copy on failure).
    const resolveStatus = async (): Promise<string> => {
      const local = readStatus();
      if (!userId) return local;
      const server = await mx.getPresence(userId).catch(() => undefined);
      const status = pickHeartbeatStatus({
        local,
        server,
        lastLocalWriteAt: getLastLocalStatusWrite(),
        now: Date.now(),
      });
      if (status !== local) {
        if (status) localStorage.setItem(`lotus-status-msg-${userId}`, status);
        else {
          localStorage.removeItem(`lotus-status-msg-${userId}`);
          localStorage.removeItem(`lotus-status-expiry-${userId}`);
        }
      }
      return status;
    };

    // [Gitea #282] A one-off PUT /presence doesn't stick: the SDK long-polls
    // /sync with set_presence=online by default and Synapse treats every sync
    // as "online", so Invisible / Idle / DND flipped back within seconds. Tell
    // the sync loop which presence to report too (as Element does).
    const syncPresence = (presence: SetPresence) => {
      mx.setSyncPresence(presence).catch(() => undefined);
    };

    const setOnline = async () => {
      syncPresence(SetPresence.Online);
      const status = await resolveStatus();
      return send({ presence: 'online', ...(status ? { status_msg: status } : {}) }, 'online');
    };
    const setUnavailable = async (statusMsg?: string) => {
      syncPresence(SetPresence.Unavailable);
      const status = await resolveStatus();
      return send(
        {
          presence: 'unavailable',
          ...(statusMsg ? { status_msg: statusMsg } : status ? { status_msg: status } : {}),
        },
        'unavailable',
      );
    };
    const setOffline = () => {
      syncPresence(SetPresence.Offline);
      return send({ presence: 'offline', status_msg: '' }, 'offline');
    };

    // Manual presence overrides — no activity tracking needed.
    if (hidePresence || presenceStatus === 'invisible') {
      setOffline();
      return undefined;
    }
    if (presenceStatus === 'online') {
      setOnline();
      return undefined;
    }
    if (presenceStatus === 'idle') {
      setUnavailable();
      return undefined;
    }
    if (presenceStatus === 'dnd') {
      // [Gitea #283] Real "busy" presence where supported: the status text
      // stays yours. /sync reporting "online" doesn't override busy on
      // Synapse, so the sync loop can stay online. Older servers keep the
      // legacy unavailable + "dnd" text encoding.
      mx.doesServerSupportUnstableFeature(BUSY_PRESENCE_FEATURE)
        .catch(() => false)
        .then(async (busy) => {
          setBusyPresenceSupported(busy);
          if (!busy) {
            setUnavailable('dnd');
            return;
          }
          syncPresence(SetPresence.Online);
          const status = await resolveStatus();
          send({ presence: BUSY_PRESENCE, ...(status ? { status_msg: status } : {}) }, 'busy');
        });
      return undefined;
    }

    // presenceStatus === 'auto' — original activity-tracking behavior.
    const startIdleTimer = () => {
      clearTimeout(idleTimerRef.current);
      idleTimerRef.current = setTimeout(() => {
        isIdleRef.current = true;
        setUnavailable();
      }, IDLE_TIMEOUT_MS);
    };

    const handleActivity = () => {
      const now = Date.now();
      if (now - lastActivityRef.current < ACTIVITY_THROTTLE_MS) return;
      lastActivityRef.current = now;

      if (isIdleRef.current && !document.hidden) {
        isIdleRef.current = false;
        setOnline();
      }
      startIdleTimer();
    };

    const handleVisibilityChange = () => {
      if (document.hidden) {
        clearTimeout(idleTimerRef.current);
        setUnavailable();
      } else {
        isIdleRef.current = false;
        lastActivityRef.current = Date.now();
        setOnline();
        startIdleTimer();
      }
    };

    const handlePageHide = () => {
      const token = mx.getAccessToken();
      const baseUrl = mx.getHomeserverUrl();
      if (!userId || !token || !baseUrl) return;

      // Reliable delivery during page teardown: navigator.sendBeacon cannot set the
      // Authorization header required by the authenticated Matrix presence endpoint, so
      // it isn't usable here. fetch(..., { keepalive: true }) lets the request outlive the
      // page and is the correct mechanism for an authed endpoint. (keepalive bodies are
      // capped at 64KB, which this tiny payload is well under.)
      // [Gitea #339] Synapse treats a missing status_msg as "clear it", so a bare
      // { presence: 'offline' } wiped the user's custom status for everyone whenever
      // any tab closed or reloaded. Send the current status along with it.
      const status = readStatus();
      fetch(`${baseUrl}/_matrix/client/v3/presence/${encodeURIComponent(userId)}/status`, {
        method: 'PUT',
        headers: {
          Authorization: `Bearer ${token}`,
          'Content-Type': 'application/json',
        },
        body: JSON.stringify({ presence: 'offline', ...(status ? { status_msg: status } : {}) }),
        keepalive: true,
      }).catch((err) => warnPresenceFailure('offline (pagehide)', err));
    };

    setOnline();
    startIdleTimer();

    const activityEvents = ['mousemove', 'keydown', 'touchstart', 'click', 'scroll'] as const;
    activityEvents.forEach((e) => window.addEventListener(e, handleActivity, { passive: true }));
    document.addEventListener('visibilitychange', handleVisibilityChange);
    window.addEventListener('pagehide', handlePageHide);

    return () => {
      clearTimeout(idleTimerRef.current);
      activityEvents.forEach((e) => window.removeEventListener(e, handleActivity));
      document.removeEventListener('visibilitychange', handleVisibilityChange);
      window.removeEventListener('pagehide', handlePageHide);
    };
  }, [mx, hidePresence, presenceStatus]);
}