PaperClipAI/ui/src/context/LiveUpdatesProvider.tsx

import { getPageVisibility, usePageVisibility } from "../lib/page-visibility";
import {
  createContext,
  useCallback,
  useContext,
  useEffect,
  useMemo,
  useRef,
  type ReactNode,
} from "react";
import {
  useQuery,
  useQueryClient,
  type InfiniteData,
  type QueryClient,
} from "@tanstack/react-query";
import {
  createCoalescingQueryClient,
  createInvalidationBatcher,
} from "../lib/query-invalidation-batcher";
import {
  patchRunStatusInList,
  removeRunFromList,
} from "../lib/live-runs-cache";
import type {
  Agent,
  HeartbeatRun,
  Issue,
  IssueComment,
  LiveEvent,
} from "@paperclipai/shared";
import type { RunForIssue } from "../api/activity";
import type { ActiveRunForIssue, LiveRunForIssue } from "../api/heartbeats";
import type { CompanyUserDirectoryResponse } from "../api/access";
import { issuesApi } from "../api/issues";
import { authApi } from "../api/auth";
import type { CompanyListResult } from "../api/companies-query";
import { healthApi } from "../api/health";
import { useCompany } from "./CompanyContext";
import type { ToastInput } from "./ToastContext";
import { useToastActions } from "./ToastContext";
import { upsertIssueCommentInPages } from "../lib/optimistic-issue-comments";
import {
  clearIssueExecutionRun,
  removeLiveRunById,
} from "../lib/optimistic-issue-runs";
import { queryKeys } from "../lib/queryKeys";
import { extractCompanyPrefixFromPath, toCompanyRelativePath } from "../lib/company-routes";
import { useLocation } from "../lib/router";
import { agentRouteRef } from "../lib/utils";
import { buildSameOriginWebSocketUrl } from "../lib/websocket-url";

const TOAST_COOLDOWN_WINDOW_MS = 10_000;
const TOAST_COOLDOWN_MAX = 3;
const RECONNECT_SUPPRESS_MS = 2000;
const SOCKET_CONNECTING = 0;
const SOCKET_OPEN = 1;
const TERMINAL_RUN_STATUSES = new Set([
  "succeeded",
  "interrupted",
  "failed",
  "cancelled",
  "timed_out",
]);

type LiveUpdatesSocketLike = {
  readyState: number;
  onopen: ((this: WebSocket, ev: Event) => unknown) | null;
  onmessage: ((this: WebSocket, ev: MessageEvent) => unknown) | null;
  onerror: ((this: WebSocket, ev: Event) => unknown) | null;
  onclose: ((this: WebSocket, ev: CloseEvent) => unknown) | null;
  close: (code?: number, reason?: string) => void;
};

export type CompanyLiveEventHandler = (event: LiveEvent) => void;

interface LiveEventSubscription {
  subscribe: (handler: CompanyLiveEventHandler) => () => void;
}

const LiveEventSubscriptionContext =
  createContext<LiveEventSubscription | null>(null);

function dispatchLiveEventToSubscribers(
  subscribers: Set<CompanyLiveEventHandler>,
  expectedCompanyId: string,
  event: LiveEvent,
) {
  if (event.companyId !== expectedCompanyId) return;
  // Snapshot so a handler that (un)subscribes mid-dispatch can't mutate the set
  // we're iterating.
  for (const handler of Array.from(subscribers)) {
    try {
      handler(event);
    } catch {
      // A misbehaving subscriber must never break the shared socket pipeline
      // or the toast/invalidation handling that runs alongside it.
    }
  }
}

/**
 * Subscribe to live company events off the single shared LiveUpdates socket.
 * Components can react to `heartbeat.run.progress`, `activity.logged`, etc.
 * without opening a WebSocket per mount. Events are already filtered to the
 * active company. No-ops when rendered outside a LiveUpdatesProvider (e.g. in
 * isolated tests), so callers get graceful degradation for free.
 */
export function useCompanyLiveEvent(handler: CompanyLiveEventHandler): void {
  const subscription = useContext(LiveEventSubscriptionContext);
  const handlerRef = useRef(handler);
  useEffect(() => {
    handlerRef.current = handler;
  });
  useEffect(() => {
    if (!subscription) return;
    return subscription.subscribe((event) => {
      handlerRef.current(event);
    });
  }, [subscription]);
}

function readString(value: unknown): string | null {
  return typeof value === "string" && value.length > 0 ? value : null;
}

function readRecord(value: unknown): Record<string, unknown> | null {
  if (typeof value !== "object" || value === null || Array.isArray(value))
    return null;
  return value as Record<string, unknown>;
}

function shortId(value: string) {
  return value.slice(0, 8);
}

function resolveAgentName(
  queryClient: QueryClient,
  companyId: string,
  agentId: string,
): string | null {
  const agents = queryClient.getQueryData<Agent[]>(
    queryKeys.agents.list(companyId),
  );
  const listedAgent = agents?.find((a) => a.id === agentId);
  if (listedAgent?.name) return listedAgent.name;
  return (
    queryClient.getQueryData<Agent>(queryKeys.agents.detail(agentId))?.name ??
    null
  );
}

function resolveUserName(
  queryClient: QueryClient,
  companyId: string,
  userId: string,
): string | null {
  const directory = queryClient.getQueryData<CompanyUserDirectoryResponse>(
    queryKeys.access.companyUserDirectory(companyId),
  );
  if (!directory) return null;
  const entry = directory.users.find((u) => u.principalId === userId);
  return entry?.user?.name?.trim() || entry?.user?.email?.trim() || null;
}

function truncate(text: string, max: number): string {
  if (text.length <= max) return text;
  return text.slice(0, max - 1) + "\u2026";
}

function resolveActorLabel(
  queryClient: QueryClient,
  companyId: string,
  actorType: string | null,
  actorId: string | null,
): string {
  if (actorType === "agent" && actorId) {
    return (
      resolveAgentName(queryClient, companyId, actorId) ??
      `Agent ${shortId(actorId)}`
    );
  }
  if (actorType === "system") return "System";
  if (actorType === "user" && actorId) {
    return resolveUserName(queryClient, companyId, actorId) ?? "Board";
  }
  return "Someone";
}

interface IssueToastContext {
  ref: string;
  title: string | null;
  label: string;
  href: string;
}

interface VisibleRouteOptions {
  isForegrounded?: boolean;
}

interface VisibleIssueRouteContext {
  routeIssueRef: string;
  issueRefs: Set<string>;
  assigneeAgentId: string | null;
  runIds: Set<string>;
  subtreeIssueRefs: Set<string>;
  subtreeAgentIds: Set<string>;
  subtreeRunIds: Set<string>;
}

function resolveIssueQueryRefs(
  queryClient: QueryClient,
  companyId: string,
  issueId: string,
  details: Record<string, unknown> | null,
): string[] {
  const refs = new Set<string>([issueId]);
  const detailIssue = queryClient.getQueryData<Issue>(
    queryKeys.issues.detail(issueId),
  );
  const listIssues = queryClient.getQueryData<Issue[]>(
    queryKeys.issues.list(companyId),
  );
  const detailsIdentifier =
    readString(details?.identifier) ?? readString(details?.issueIdentifier);

  if (detailsIdentifier) refs.add(detailsIdentifier);

  if (detailIssue?.id) refs.add(detailIssue.id);
  if (detailIssue?.identifier) refs.add(detailIssue.identifier);

  const listIssue = listIssues?.find((issue) => {
    if (issue.id === issueId) return true;
    if (issue.identifier && issue.identifier === issueId) return true;
    if (detailsIdentifier && issue.identifier === detailsIdentifier)
      return true;
    return false;
  });
  if (listIssue?.id) refs.add(listIssue.id);
  if (listIssue?.identifier) refs.add(listIssue.identifier);

  return Array.from(refs);
}

function resolveIssueToastContext(
  queryClient: QueryClient,
  companyId: string,
  issueId: string,
  details: Record<string, unknown> | null,
): IssueToastContext {
  const issueRefs = resolveIssueQueryRefs(
    queryClient,
    companyId,
    issueId,
    details,
  );
  const detailIssue = issueRefs
    .map((ref) => queryClient.getQueryData<Issue>(queryKeys.issues.detail(ref)))
    .find((issue): issue is Issue => !!issue);
  const listIssue = queryClient
    .getQueryData<Issue[]>(queryKeys.issues.list(companyId))
    ?.find((issue) =>
      issueRefs.some((ref) => issue.id === ref || issue.identifier === ref),
    );
  const cachedIssue = detailIssue ?? listIssue ?? null;
  const ref =
    readString(details?.identifier) ??
    readString(details?.issueIdentifier) ??
    cachedIssue?.identifier ??
    `Task ${shortId(issueId)}`;
  const title =
    readString(details?.title) ??
    readString(details?.issueTitle) ??
    cachedIssue?.title ??
    null;
  return {
    ref,
    title,
    label: title ? `${ref} - ${truncate(title, 72)}` : ref,
    href: `/issues/${cachedIssue?.identifier ?? issueId}`,
  };
}

function isPageForegrounded(): boolean {
  if (typeof document === "undefined") return false;
  if (document.visibilityState !== "visible") return false;
  if (typeof document.hasFocus === "function" && !document.hasFocus())
    return false;
  return true;
}

function resolveVisibleIssueRouteContext(
  queryClient: QueryClient,
  pathname: string,
  options?: VisibleRouteOptions,
): VisibleIssueRouteContext | null {
  const isForegrounded = options?.isForegrounded ?? isPageForegrounded();
  if (!isForegrounded) return null;

  const relativePath = toCompanyRelativePath(pathname);
  const segments = relativePath.split("/").filter(Boolean);
  if (!["issues", "chats"].includes(segments[0]) || !segments[1]) return null;

  let issueRef = decodeURIComponent(segments[1]);
  if (segments[0] === "chats") {
    const session = queryClient.getQueryData<Awaited<ReturnType<typeof authApi.getSession>>>(queryKeys.auth.session);
    const userId = session?.user?.id ?? session?.session?.userId ?? null;
    const companyPrefix = extractCompanyPrefixFromPath(pathname);
    const company = queryClient.getQueryData<CompanyListResult>(queryKeys.companies.list(userId))
      ?.companies.find(item => item.issuePrefix.toUpperCase() === companyPrefix?.toUpperCase());
    if (!company) return null;
    const agent = queryClient.getQueryData<Agent[]>(queryKeys.agents.list(company.id))
      ?.find(item => item.id === issueRef || agentRouteRef(item) === issueRef);
    if (!agent) return null;
    const conversation = queryClient.getQueryData<Issue | null>(queryKeys.agentChats.detail(company.id, userId, agent.id));
    if (!conversation) return null;
    issueRef = conversation.id;
  }
  const issue = queryClient.getQueryData<Issue>(queryKeys.issues.detail(issueRef)) ?? null;
  const issueRefs = new Set<string>([issueRef]);
  if (issue?.id) issueRefs.add(issue.id);
  if (issue?.identifier) issueRefs.add(issue.identifier);

  function collectCachedIssueRunIds(ref: string, target: Set<string>) {
    const activeRun = queryClient.getQueryData<ActiveRunForIssue | null>(
      queryKeys.issues.activeRun(ref),
    );
    const liveRuns =
      queryClient.getQueryData<LiveRunForIssue[]>(
        queryKeys.issues.liveRuns(ref),
      ) ?? [];
    const linkedRuns =
      queryClient.getQueryData<RunForIssue[]>(queryKeys.issues.runs(ref)) ?? [];

    if (activeRun?.id) target.add(activeRun.id);
    for (const run of liveRuns) {
      if (run.id) target.add(run.id);
    }
    for (const run of linkedRuns) {
      if (run.runId) target.add(run.runId);
    }
  }

  // The route may use an identifier while the conversation loads canonical
  // UUID-keyed history. A retryable terminal event can omit issueId after its
  // first delivery has already removed the run from the company live cache.
  const runIds = new Set<string>();
  for (const ref of issueRefs) collectCachedIssueRunIds(ref, runIds);

  // Notifications about descendants belong to the subtree already on screen.
  // Keep these separate from issueRefs, which also drives cache invalidation.
  const subtreeIssueRefs = new Set(issueRefs);
  const subtreeAgentIds = new Set<string>();
  const subtreeRunIds = new Set(runIds);
  if (issue?.companyId && issue.id) {
    const descendants =
      queryClient.getQueryData<Issue[]>(
        queryKeys.issues.listByDescendantRoot(issue.companyId, issue.id),
      ) ?? [];
    for (const descendant of descendants) {
      subtreeIssueRefs.add(descendant.id);
      if (descendant.identifier) subtreeIssueRefs.add(descendant.identifier);
      if (descendant.assigneeAgentId)
        subtreeAgentIds.add(descendant.assigneeAgentId);
      if (descendant.executionRunId)
        subtreeRunIds.add(descendant.executionRunId);
      // Only current descendants contribute history. Keep their run IDs out
      // of runIds, which also drives invalidation of the root conversation.
      collectCachedIssueRunIds(descendant.id, subtreeRunIds);
      if (descendant.identifier)
        collectCachedIssueRunIds(descendant.identifier, subtreeRunIds);
    }
  }

  if (issue?.companyId) {
    const companyRuns =
      queryClient.getQueryData<LiveRunForIssue[]>(
        queryKeys.liveRuns(issue.companyId),
      ) ?? [];
    for (const run of companyRuns) {
      if (run.issueId && subtreeIssueRefs.has(run.issueId))
        subtreeRunIds.add(run.id);
    }
  }
  return {
    routeIssueRef: issueRef,
    issueRefs,
    assigneeAgentId: issue?.assigneeAgentId ?? null,
    runIds,
    subtreeIssueRefs,
    subtreeAgentIds,
    subtreeRunIds,
  };
}

function buildIssueRefsForPayload(
  entityId: string,
  details: Record<string, unknown> | null,
): Set<string> {
  const refs = new Set<string>([entityId]);
  const identifier =
    readString(details?.identifier) ?? readString(details?.issueIdentifier);
  if (identifier) refs.add(identifier);
  return refs;
}

function overlaps(a: Set<string>, b: Set<string>): boolean {
  for (const value of a) {
    if (b.has(value)) return true;
  }
  return false;
}

function shouldSuppressActivityToastForVisibleIssue(
  queryClient: QueryClient,
  pathname: string,
  payload: Record<string, unknown>,
  options?: VisibleRouteOptions,
): boolean {
  const entityType = readString(payload.entityType);
  const entityId = readString(payload.entityId);
  if (entityType !== "issue" || !entityId) return false;

  const context = resolveVisibleIssueRouteContext(
    queryClient,
    pathname,
    options,
  );
  if (!context) return false;

  return overlaps(
    context.subtreeIssueRefs,
    buildIssueRefsForPayload(entityId, readRecord(payload.details)),
  );
}

function shouldSuppressRunStatusToastForVisibleIssue(
  queryClient: QueryClient,
  pathname: string,
  payload: Record<string, unknown>,
  options?: VisibleRouteOptions,
): boolean {
  if (!(options?.isForegrounded ?? isPageForegrounded())) return false;
  const visibleRunId = toCompanyRelativePath(pathname).match(
    /^\/agents\/[^/]+\/runs\/([^/]+)/,
  )?.[1];
  if (
    visibleRunId &&
    decodeURIComponent(visibleRunId) === readString(payload.runId)
  )
    return true;
  const context = resolveVisibleIssueRouteContext(
    queryClient,
    pathname,
    options,
  );
  if (!context) return false;

  const issueId = readString(payload.issueId);
  if (issueId) return context.subtreeIssueRefs.has(issueId);
  const runId = readString(payload.runId);
  if (runId) return context.subtreeRunIds.has(runId);

  const agentId = readString(payload.agentId);
  return (
    !!agentId &&
    (agentId === context.assigneeAgentId ||
      context.subtreeAgentIds.has(agentId))
  );
}

function invalidateVisibleIssueRunQueries(
  queryClient: QueryClient,
  pathname: string,
  payload: Record<string, unknown>,
  options?: VisibleRouteOptions,
): boolean {
  const context = resolveVisibleIssueRouteContext(
    queryClient,
    pathname,
    options,
  );
  if (!context) return false;

  const runId = readString(payload.runId);
  const agentId = readString(payload.agentId);
  const matchesVisibleIssue =
    (runId !== null && context.runIds.has(runId)) ||
    (!!agentId &&
      !!context.assigneeAgentId &&
      agentId === context.assigneeAgentId);
  if (!matchesVisibleIssue) return false;

  const status = readString(payload.status);
  if (runId && status && TERMINAL_RUN_STATUSES.has(status)) {
    for (const issueRef of context.issueRefs) {
      // Finalization may commit the selected reply without comment_added.
      // Fetch the canonical comment before handing off the live transcript;
      // progress/queued events must not refetch the whole conversation.
      queryClient.invalidateQueries({
        queryKey: queryKeys.issues.comments(issueRef),
      });
      queryClient.setQueryData(
        queryKeys.issues.liveRuns(issueRef),
        (current: LiveRunForIssue[] | undefined) =>
          removeLiveRunById(current, runId),
      );
      queryClient.setQueryData(
        queryKeys.issues.activeRun(issueRef),
        (current: ActiveRunForIssue | null | undefined) =>
          current?.id === runId ? null : current,
      );
      queryClient.setQueryData(
        queryKeys.issues.detail(issueRef),
        (current: Issue | undefined) => clearIssueExecutionRun(current, runId),
      );
    }
  }

  for (const issueRef of context.issueRefs) {
    queryClient.invalidateQueries({ queryKey: queryKeys.issues.detail(issueRef) });
    queryClient.invalidateQueries({ queryKey: queryKeys.issues.activity(issueRef) });
    queryClient.invalidateQueries({ queryKey: queryKeys.issues.runs(issueRef) });
    queryClient.invalidateQueries({ queryKey: queryKeys.issues.liveRuns(issueRef) });
    queryClient.invalidateQueries({ queryKey: queryKeys.issues.activeRun(issueRef) });
    if (status && TERMINAL_RUN_STATUSES.has(status)) {
      // A final comment can race the last in-flight history fetch. Reconcile
      // persisted messages after the turn settles.
      queryClient.invalidateQueries({ queryKey: queryKeys.issues.comments(issueRef) });
      queryClient.invalidateQueries({ queryKey: queryKeys.issues.attachments(issueRef) });
      queryClient.invalidateQueries({ queryKey: queryKeys.issues.workProducts(issueRef) });
      queryClient.invalidateQueries({ queryKey: ["issues", "tree-control-state", issueRef] });
    }
  }
  return true;
}

interface RunLiveStatusPatch {
  runId: string;
  agentId: string | null;
  issueId: string | null;
  message?: string | null;
  updatedAt?: string | null;
  currentToolName?: string | null;
  lastAssistantSnippet?: string | null;
  lastEventAt?: string | null;
}

function hasPatchKey<K extends keyof RunLiveStatusPatch>(
  patch: RunLiveStatusPatch,
  key: K,
): patch is RunLiveStatusPatch & Required<Pick<RunLiveStatusPatch, K>> {
  return Object.prototype.hasOwnProperty.call(patch, key);
}

function applyRunLiveStatusPatch<T extends ActiveRunForIssue | LiveRunForIssue>(
  run: T,
  patch: RunLiveStatusPatch,
): T {
  if (run.id !== patch.runId) return run;
  return {
    ...run,
    ...(hasPatchKey(patch, "message")
      ? { currentStatusMessage: patch.message }
      : {}),
    ...(hasPatchKey(patch, "updatedAt")
      ? { currentStatusUpdatedAt: patch.updatedAt }
      : {}),
    ...(hasPatchKey(patch, "currentToolName")
      ? { currentToolName: patch.currentToolName }
      : {}),
    ...(hasPatchKey(patch, "lastAssistantSnippet")
      ? { lastAssistantSnippet: patch.lastAssistantSnippet }
      : {}),
    ...(hasPatchKey(patch, "lastEventAt")
      ? { lastEventAt: patch.lastEventAt }
      : {}),
  };
}

function applyRunLiveStatusPatchToArray<T extends LiveRunForIssue>(
  runs: T[] | undefined,
  patch: RunLiveStatusPatch,
): T[] | undefined {
  if (!runs) return runs;
  let changed = false;
  const nextRuns = runs.map((run) => {
    if (run.id !== patch.runId) return run;
    changed = true;
    return applyRunLiveStatusPatch(run, patch);
  });
  return changed ? nextRuns : runs;
}

function readRunLiveStatusPatchFromPayload(
  payload: Record<string, unknown>,
  eventCreatedAt: string,
  eventType: string,
): RunLiveStatusPatch | null {
  const runId = readString(payload.runId);
  if (!runId) return null;

  const patch: RunLiveStatusPatch = {
    runId,
    agentId: readString(payload.agentId),
    issueId: readString(payload.issueId),
  };

  if (eventType === "heartbeat.run.progress") {
    patch.message = readString(payload.message);
    patch.updatedAt = readString(payload.updatedAt) ?? eventCreatedAt;
    patch.currentToolName = readString(payload.currentToolName);
    patch.lastAssistantSnippet = readString(payload.lastAssistantSnippet);
    patch.lastEventAt = readString(payload.lastEventAt) ?? patch.updatedAt;
    return patch;
  }

  if (eventType === "heartbeat.run.log") {
    patch.lastEventAt = readString(payload.ts) ?? eventCreatedAt;
    return patch;
  }

  if (eventType === "heartbeat.run.event") {
    const message = readString(payload.message);
    if (message) patch.message = message;
    patch.updatedAt = readString(payload.updatedAt) ?? eventCreatedAt;
    const currentToolName = readString(payload.currentToolName);
    if (currentToolName) patch.currentToolName = currentToolName;
    const lastAssistantSnippet = readString(payload.lastAssistantSnippet);
    if (lastAssistantSnippet) patch.lastAssistantSnippet = lastAssistantSnippet;
    patch.lastEventAt = readString(payload.lastEventAt) ?? eventCreatedAt;
    return patch;
  }

  return null;
}

function applyRunLiveStatusPatchToCaches(
  queryClient: QueryClient,
  companyId: string,
  pathname: string,
  patch: RunLiveStatusPatch,
  options?: VisibleRouteOptions,
): boolean {
  let changed = false;
  queryClient.setQueryData(
    queryKeys.liveRuns(companyId),
    (current: LiveRunForIssue[] | undefined) => {
      const nextRuns = applyRunLiveStatusPatchToArray(current, patch);
      if (nextRuns !== current) changed = true;
      return nextRuns;
    },
  );

  const issueRefs = new Set<string>();
  if (patch.issueId) {
    for (const ref of resolveIssueQueryRefs(
      queryClient,
      companyId,
      patch.issueId,
      null,
    )) {
      issueRefs.add(ref);
    }
  }

  const context = resolveVisibleIssueRouteContext(
    queryClient,
    pathname,
    options,
  );
  if (
    context &&
    (context.runIds.has(patch.runId) ||
      (!!patch.issueId && context.issueRefs.has(patch.issueId)) ||
      (!!patch.agentId &&
        !!context.assigneeAgentId &&
        patch.agentId === context.assigneeAgentId))
  ) {
    for (const ref of context.issueRefs) issueRefs.add(ref);
  }

  for (const issueRef of issueRefs) {
    queryClient.setQueryData(
      queryKeys.issues.activeRun(issueRef),
      (current: ActiveRunForIssue | null | undefined) => {
        if (!current || current.id !== patch.runId) return current;
        changed = true;
        return applyRunLiveStatusPatch(current, patch);
      },
    );
    queryClient.setQueryData(
      queryKeys.issues.liveRuns(issueRef),
      (current: LiveRunForIssue[] | undefined) => {
        const nextRuns = applyRunLiveStatusPatchToArray(current, patch);
        if (nextRuns !== current) changed = true;
        return nextRuns;
      },
    );
  }

  return changed;
}

function shouldSuppressAgentStatusToastForVisibleIssue(
  queryClient: QueryClient,
  pathname: string,
  payload: Record<string, unknown>,
  options?: VisibleRouteOptions,
): boolean {
  const context = resolveVisibleIssueRouteContext(
    queryClient,
    pathname,
    options,
  );
  if (!context) return false;

  const agentId = readString(payload.agentId);
  return (
    !!agentId &&
    (agentId === context.assigneeAgentId ||
      context.subtreeAgentIds.has(agentId))
  );
}

function shouldDeferIssueRefetchForVisibleAgentActivity(
  queryClient: QueryClient,
  pathname: string,
  payload: Record<string, unknown>,
  options?: VisibleRouteOptions,
): boolean {
  const entityType = readString(payload.entityType);
  const entityId = readString(payload.entityId);
  const actorType = readString(payload.actorType);
  const action = readString(payload.action);
  const details = readRecord(payload.details);

  if (entityType !== "issue" || !entityId) return false;
  if (actorType !== "agent" && actorType !== "system") return false;
  if (action !== "issue.updated") return false;
  if (readString(details?.source) === "comment") return false;

  const context = resolveVisibleIssueRouteContext(
    queryClient,
    pathname,
    options,
  );
  if (!context) return false;

  return overlaps(
    context.issueRefs,
    buildIssueRefsForPayload(entityId, details),
  );
}

function shouldDeferVisibleIssueCommentActivity(
  queryClient: QueryClient,
  pathname: string,
  payload: Record<string, unknown>,
  options?: VisibleRouteOptions,
): boolean {
  const entityType = readString(payload.entityType);
  const entityId = readString(payload.entityId);
  const action = readString(payload.action);
  const details = readRecord(payload.details);

  if (entityType !== "issue" || !entityId) return false;
  if (action !== "issue.comment_added") return false;

  const context = resolveVisibleIssueRouteContext(
    queryClient,
    pathname,
    options,
  );
  if (!context) return false;

  return overlaps(
    context.issueRefs,
    buildIssueRefsForPayload(entityId, details),
  );
}

async function hydrateVisibleIssueComment(
  queryClient: QueryClient,
  pathname: string,
  payload: Record<string, unknown>,
  options?: VisibleRouteOptions,
) {
  const entityType = readString(payload.entityType);
  const action = readString(payload.action);
  const details = readRecord(payload.details);
  const commentId = readString(details?.commentId);

  if (entityType !== "issue" || action !== "issue.comment_added" || !commentId)
    return false;

  const context = resolveVisibleIssueRouteContext(
    queryClient,
    pathname,
    options,
  );
  if (!context) return false;

  const entityId = readString(payload.entityId);
  if (
    !entityId ||
    !overlaps(context.issueRefs, buildIssueRefsForPayload(entityId, details))
  ) {
    return false;
  }

  try {
    const comment = await issuesApi.getComment(
      context.routeIssueRef,
      commentId,
    );
    queryClient.setQueryData<
      InfiniteData<IssueComment[], string | null> | undefined
    >(queryKeys.issues.comments(context.routeIssueRef), (current) => {
      if (!current) {
        return {
          pages: [[comment]],
          pageParams: [null],
        };
      }

      return {
        ...current,
        pages: upsertIssueCommentInPages(current.pages, comment),
      };
    });
    return true;
  } catch {
    queryClient.invalidateQueries({
      queryKey: queryKeys.issues.comments(context.routeIssueRef),
    });
    return false;
  }
}

const ISSUE_TOAST_ACTIONS = new Set([
  "issue.created",
  "issue.updated",
  "issue.comment_added",
]);
const ISSUE_DOCUMENT_ACTIVITY_ACTIONS = new Set([
  "issue.document_created",
  "issue.document_updated",
  "issue.document_restored",
  "issue.document_deleted",
]);
const ISSUE_DOCUMENT_ANNOTATION_ACTIVITY_ACTIONS = new Set([
  "issue.document_annotation_thread_created",
  "issue.document_annotation_comment_added",
  "issue.document_annotation_thread_resolved",
  "issue.document_annotation_thread_reopened",
  "issue.document_annotation_remapped",
]);
const ROUTINE_DOCUMENT_ANNOTATION_ACTIVITY_ACTIONS = new Set([
  "routine.document_annotation_thread_created",
  "routine.document_annotation_comment_added",
  "routine.document_annotation_thread_resolved",
  "routine.document_annotation_thread_reopened",
  "routine.document_annotation_remapped",
]);
const CASE_DOCUMENT_ANNOTATION_ACTIVITY_ACTIONS = new Set([
  "case.document_annotation_thread_created",
  "case.document_annotation_comment_added",
  "case.document_annotation_thread_resolved",
  "case.document_annotation_thread_reopened",
  "case.document_annotation_remapped",
]);
const AGENT_TOAST_STATUSES = new Set(["error"]);
const RUN_TOAST_STATUSES = new Set(["failed", "timed_out", "cancelled"]);

function describeIssueUpdate(
  details: Record<string, unknown> | null,
): string | null {
  if (!details) return null;
  const changes: string[] = [];
  if (typeof details.status === "string")
    changes.push(`status -> ${details.status.replace(/_/g, " ")}`);
  if (typeof details.priority === "string")
    changes.push(`priority -> ${details.priority}`);
  if (
    typeof details.assigneeAgentId === "string" ||
    typeof details.assigneeUserId === "string"
  ) {
    changes.push("reassigned");
  } else if (
    details.assigneeAgentId === null ||
    details.assigneeUserId === null
  ) {
    changes.push("unassigned");
  }
  if (details.reopened === true) {
    const from = readString(details.reopenedFrom);
    changes.push(
      from ? `reopened from ${from.replace(/_/g, " ")}` : "reopened",
    );
  }
  if (typeof details.title === "string") changes.push("title changed");
  if (typeof details.description === "string")
    changes.push("description changed");
  if (changes.length > 0) return changes.join(", ");
  return null;
}

function buildActivityToast(
  queryClient: QueryClient,
  companyId: string,
  payload: Record<string, unknown>,
  currentActor: { userId: string | null; agentId: string | null },
): ToastInput | null {
  const entityType = readString(payload.entityType);
  const entityId = readString(payload.entityId);
  const action = readString(payload.action);
  const details = readRecord(payload.details);
  const actorId = readString(payload.actorId);
  const actorType = readString(payload.actorType);

  if (
    entityType !== "issue" ||
    !entityId ||
    !action ||
    !ISSUE_TOAST_ACTIONS.has(action)
  ) {
    return null;
  }

  const issue = resolveIssueToastContext(
    queryClient,
    companyId,
    entityId,
    details,
  );
  const actor = resolveActorLabel(queryClient, companyId, actorType, actorId);
  const isSelfActivity =
    (actorType === "user" &&
      !!currentActor.userId &&
      actorId === currentActor.userId) ||
    (actorType === "agent" &&
      !!currentActor.agentId &&
      actorId === currentActor.agentId);
  if (isSelfActivity) return null;

  if (action === "issue.created") {
    return {
      title: `${actor} created ${issue.ref}`,
      body: issue.title ? truncate(issue.title, 96) : undefined,
      tone: "success",
      action: { label: `View ${issue.ref}`, href: issue.href },
      dedupeKey: `activity:${action}:${entityId}`,
    };
  }

  if (action === "issue.updated") {
    if (readString(details?.source) === "comment") {
      // Comment-driven updates emit a paired comment event; show one combined toast on the comment event.
      return null;
    }
    const changeDesc = describeIssueUpdate(details);
    const body = changeDesc
      ? issue.title
        ? `${truncate(issue.title, 64)} - ${changeDesc}`
        : changeDesc
      : issue.title
        ? truncate(issue.title, 96)
        : issue.label;
    return {
      title: `${actor} updated ${issue.ref}`,
      body: truncate(body, 100),
      tone: "info",
      action: { label: `View ${issue.ref}`, href: issue.href },
      dedupeKey: `activity:${action}:${entityId}`,
    };
  }

  const commentId = readString(details?.commentId);
  const bodySnippet = readString(details?.bodySnippet);
  const reopened = details?.reopened === true;
  const updated = details?.updated === true;
  const reopenedFrom = readString(details?.reopenedFrom);
  const reopenedLabel = reopened
    ? reopenedFrom
      ? `reopened from ${reopenedFrom.replace(/_/g, " ")}`
      : "reopened"
    : null;
  const title = reopened
    ? `${actor} reopened and commented on ${issue.ref}`
    : updated
      ? `${actor} commented and updated ${issue.ref}`
      : `${actor} commented on ${issue.ref}`;
  const body = bodySnippet
    ? reopenedLabel
      ? `${reopenedLabel} - ${bodySnippet.replace(/^#+\s*/m, "").replace(/\n/g, " ")}`
      : bodySnippet.replace(/^#+\s*/m, "").replace(/\n/g, " ")
    : reopenedLabel
      ? issue.title
        ? `${reopenedLabel} - ${issue.title}`
        : reopenedLabel
      : (issue.title ?? undefined);
  return {
    title,
    body: body ? truncate(body, 96) : undefined,
    tone: "info",
    action: { label: `View ${issue.ref}`, href: issue.href },
    dedupeKey: `activity:${action}:${entityId}:${commentId ?? "na"}`,
  };
}

function buildJoinRequestToast(
  payload: Record<string, unknown>,
): ToastInput | null {
  const entityType = readString(payload.entityType);
  const action = readString(payload.action);
  const entityId = readString(payload.entityId);
  const details = readRecord(payload.details);

  if (entityType !== "join_request" || !action || !entityId) return null;
  if (action !== "join.requested" && action !== "join.request_replayed")
    return null;

  const requestType = readString(details?.requestType);
  const label = requestType === "agent" ? "Agent" : "Someone";

  return {
    title: `${label} wants to join`,
    body: "A new join request is waiting for approval.",
    tone: "info",
    action: { label: "View inbox", href: "/inbox/mine" },
    dedupeKey: `join-request:${entityId}`,
  };
}

function buildAgentStatusToast(
  payload: Record<string, unknown>,
  nameOf: (id: string) => string | null,
  queryClient: QueryClient,
  companyId: string,
): ToastInput | null {
  const agentId = readString(payload.agentId);
  const status = readString(payload.status);
  if (!agentId || !status || !AGENT_TOAST_STATUSES.has(status)) return null;

  const tone = status === "error" ? "error" : "info";
  const name = nameOf(agentId) ?? `Agent ${shortId(agentId)}`;
  const title = status === "running" ? `${name} started` : `${name} errored`;

  const agents = queryClient.getQueryData<Agent[]>(
    queryKeys.agents.list(companyId),
  );
  const agent = agents?.find((a) => a.id === agentId);
  const body = agent?.title ?? undefined;

  return {
    title,
    body,
    tone,
    action: { label: "View agent", href: `/agents/${agentId}` },
    dedupeKey: `agent-status:${agentId}:${status}`,
  };
}

function buildRunStatusToast(
  payload: Record<string, unknown>,
  nameOf: (id: string) => string | null,
): ToastInput | null {
  const runId = readString(payload.runId);
  const agentId = readString(payload.agentId);
  const status = readString(payload.status);
  if (!runId || !agentId || !status || !RUN_TOAST_STATUSES.has(status))
    return null;

  const error = readString(payload.error);
  const errorCode = readString(payload.errorCode);
  // Interrupt is an intentional conversation control. Its caller gives
  // feedback; the terminal event must not announce a cancelled/failed run.
  if (errorCode === "operator_interrupted") return null;
  // Workspace contention is ordinary scheduling, not a failed user action.
  if (errorCode === "workspace_busy") return null;
  const contextSource = readString(payload.contextSource);
  const triggerDetail = readString(payload.triggerDetail);
  const name = nameOf(agentId) ?? "Agent";
  if (
    status === "failed" &&
    errorCode === "low_trust_isolation_unavailable" &&
    contextSource?.startsWith("chat:")
  ) {
    return {
      title: `${name} couldn't start this chat`,
      body: "This external chat identity isn't linked, and isolated guest workspaces are disabled. Link the identity in Connectors or enable isolated workspaces, then start a new task.",
      tone: "warn",
      ttlMs: 10_000,
      action: { label: "Open chat connections", href: "/apps" },
      dedupeKey: `run-status:${runId}:${status}`,
    };
  }
  const tone =
    status === "succeeded"
      ? "success"
      : status === "cancelled"
        ? "info"
        : "error";
  const statusLabel =
    status === "succeeded"
      ? "succeeded"
      : status === "failed"
        ? "failed"
        : status === "timed_out"
          ? "timed out"
          : "cancelled";
  const title = `${name} run ${statusLabel}`;

  let body: string | undefined;
  if (error) {
    body = truncate(error, 100);
  } else if (triggerDetail) {
    body = `Trigger: ${triggerDetail}`;
  }

  return {
    title,
    body,
    tone,
    ttlMs: status === "succeeded" ? 5000 : 7000,
    action: { label: "View run", href: `/agents/${agentId}/runs/${runId}` },
    dedupeKey: `run-status:${runId}:${status}`,
  };
}

/**
 * Event-source the company live-runs list from run-lifecycle events instead of
 * invalidating + refetching it. Returns true when the cache was fully patched;
 * false means a genuinely new run appeared that can't be reconstructed from the
 * event, so the caller should refetch once to pick it up.
 */
function applyRunLifecycleToCompanyLiveRuns(
  queryClient: QueryClient,
  companyId: string,
  payload: Record<string, unknown>,
): boolean {
  const runId = readString(payload.runId);
  const status = readString(payload.status);
  if (!runId || !status) return false;

  queryClient.setQueryData(
    queryKeys.runDetail(runId),
    (current: HeartbeatRun | undefined) => {
      if (!current) return current;
      const has = (key: string) =>
        Object.prototype.hasOwnProperty.call(payload, key);
      return {
        ...current,
        status: status as HeartbeatRun["status"],
        ...(has("invocationSource")
          ? {
              invocationSource:
                readString(payload.invocationSource) ??
                current.invocationSource,
            }
          : {}),
        ...(has("triggerDetail")
          ? { triggerDetail: readString(payload.triggerDetail) }
          : {}),
        ...(has("error") ? { error: readString(payload.error) } : {}),
        ...(has("errorCode")
          ? { errorCode: readString(payload.errorCode) }
          : {}),
        ...(has("startedAt")
          ? { startedAt: readString(payload.startedAt) }
          : {}),
        ...(has("finishedAt")
          ? { finishedAt: readString(payload.finishedAt) }
          : {}),
      };
    },
  );

  if (TERMINAL_RUN_STATUSES.has(status)) {
    queryClient.setQueryData(
      queryKeys.liveRuns(companyId),
      (current: LiveRunForIssue[] | undefined) =>
        removeRunFromList(current, runId),
    );
    // Always "handled": a terminal run must never be in the live list, so if it
    // wasn't present there is deliberately nothing to refetch (removeRunFromList
    // was a no-op and we must not re-add it).
    return true;
  }

  let present = false;
  queryClient.setQueryData(
    queryKeys.liveRuns(companyId),
    (current: LiveRunForIssue[] | undefined) => {
      const result = patchRunStatusInList(current, runId, status);
      present = result.present;
      return result.next;
    },
  );
  return present;
}

function invalidateHeartbeatQueries(
  queryClient: ReturnType<typeof useQueryClient>,
  companyId: string,
  payload: Record<string, unknown>,
) {
  // Note: liveRuns(companyId) is intentionally NOT invalidated here — it is
  // event-sourced via applyRunLifecycleToCompanyLiveRuns in the caller.
  queryClient.invalidateQueries({ queryKey: queryKeys.heartbeats(companyId) });
  queryClient.invalidateQueries({ queryKey: queryKeys.agents.list(companyId) });
  queryClient.invalidateQueries({ queryKey: queryKeys.dashboard(companyId) });
  queryClient.invalidateQueries({ queryKey: queryKeys.costs(companyId) });
  queryClient.invalidateQueries({
    queryKey: queryKeys.sidebarBadges(companyId),
  });

  const agentId = readString(payload.agentId);
  if (agentId) {
    queryClient.invalidateQueries({
      queryKey: queryKeys.agents.detail(agentId),
    });
    queryClient.invalidateQueries({
      queryKey: queryKeys.heartbeats(companyId, agentId),
    });
  }
  const runId = readString(payload.runId);
  if (runId) {
    queryClient.invalidateQueries({ queryKey: queryKeys.runDetail(runId) });
  }
}

function invalidateHeartbeatProgressQueries(
  queryClient: ReturnType<typeof useQueryClient>,
  _companyId: string,
  payload: Record<string, unknown>,
) {
  const agentId = readString(payload.agentId);
  if (agentId) {
    queryClient.invalidateQueries({
      queryKey: queryKeys.agents.detail(agentId),
    });
  }
}

function invalidateActivityQueries(
  queryClient: ReturnType<typeof useQueryClient>,
  companyId: string,
  payload: Record<string, unknown>,
  currentActor: { userId: string | null; agentId: string | null },
  options?: { pathname?: string; isForegrounded?: boolean },
) {
  queryClient.invalidateQueries({ queryKey: queryKeys.activity(companyId) });
  queryClient.invalidateQueries({ queryKey: queryKeys.dashboard(companyId) });
  queryClient.invalidateQueries({
    queryKey: queryKeys.sidebarBadges(companyId),
  });

  const entityType = readString(payload.entityType);
  const entityId = readString(payload.entityId);
  const action = readString(payload.action);
  const actorType = readString(payload.actorType);
  const actorId = readString(payload.actorId);
  const details = readRecord(payload.details);
  const ownActorActivity =
    (actorType === "user" &&
      !!currentActor.userId &&
      actorId === currentActor.userId) ||
    (actorType === "agent" &&
      !!currentActor.agentId &&
      actorId === currentActor.agentId);

  if (action?.startsWith("ai_connection.") || action?.startsWith("connection_grant.")) {
    queryClient.invalidateQueries({ queryKey: ["ai-connections", companyId] });
  }

  if (action?.startsWith("resource_membership.")) {
    const targetUserId = readString(details?.userId);
    if (!targetUserId || targetUserId === currentActor.userId) {
      queryClient.invalidateQueries({
        queryKey: queryKeys.resourceMemberships.mine(companyId),
      });
    }
  }

  if (entityType === "issue") {
    if (action === "issue.tree_hold_created" || action === "issue.tree_hold_released" || action === "issue.updated") {
      // An ancestor hold or reparenting changes descendants' effective pause.
      queryClient.invalidateQueries({ queryKey: ["issues", "tree-control-state"] });
    }
    queryClient.invalidateQueries({
      queryKey: queryKeys.issues.list(companyId),
    });
    queryClient.invalidateQueries({
      queryKey: queryKeys.issues.listMineByMe(companyId),
    });
    queryClient.invalidateQueries({
      queryKey: queryKeys.issues.listTouchedByMe(companyId),
    });
    queryClient.invalidateQueries({
      queryKey: queryKeys.issues.listUnreadTouchedByMe(companyId),
    });
    if (entityId) {
      const selfCommentActivity =
        (action === "issue.comment_added" ||
          (action === "issue.updated" &&
            readString(details?.source) === "comment")) &&
        ((actorType === "user" &&
          !!currentActor.userId &&
          actorId === currentActor.userId) ||
          (actorType === "agent" &&
            !!currentActor.agentId &&
            actorId === currentActor.agentId));
      const visibleIssueAgentActivity =
        !!options?.pathname &&
        shouldDeferIssueRefetchForVisibleAgentActivity(
          queryClient,
          options.pathname,
          payload,
          { isForegrounded: options.isForegrounded },
        );
      const visibleIssueCommentActivity =
        !!options?.pathname &&
        shouldDeferVisibleIssueCommentActivity(
          queryClient,
          options.pathname,
          payload,
          { isForegrounded: options.isForegrounded },
        );
      const issueRefs = resolveIssueQueryRefs(
        queryClient,
        companyId,
        entityId,
        details,
      );
      for (const ref of issueRefs) {
        const invalidationOptions =
          selfCommentActivity ||
          visibleIssueAgentActivity ||
          visibleIssueCommentActivity
            ? { refetchType: "inactive" as const }
            : undefined;
        queryClient.invalidateQueries({ queryKey: queryKeys.issues.detail(ref), ...invalidationOptions });
        queryClient.invalidateQueries({ queryKey: queryKeys.issues.activity(ref), ...invalidationOptions });
        if (action === "issue.comment_added" || action === "issue.conversation_session_started") {
          queryClient.invalidateQueries({ queryKey: queryKeys.issues.comments(ref), ...invalidationOptions });
        }
        if (action?.startsWith("issue.attachment_") || action?.startsWith("issue.work_product_")) {
          // These cards are durable API objects, not streamed text. Refresh the
          // visible task too, including attachments bound to an existing comment.
          queryClient.invalidateQueries({ queryKey: queryKeys.issues.attachments(ref) });
          queryClient.invalidateQueries({ queryKey: queryKeys.issues.workProducts(ref) });
        }
        if (action === "issue.conversation_session_started") {
          queryClient.invalidateQueries({ queryKey: ["issues", "tree-control-state", ref] });
          queryClient.invalidateQueries({ queryKey: queryKeys.issues.interactions(ref) });
        }
        if (action && ISSUE_DOCUMENT_ACTIVITY_ACTIONS.has(action)) {
          const documentKey = readString(details?.key);
          queryClient.invalidateQueries({
            queryKey: queryKeys.issues.documents(ref),
            ...invalidationOptions,
          });
          if (documentKey) {
            queryClient.invalidateQueries({
              queryKey: queryKeys.issues.document(ref, documentKey),
              ...invalidationOptions,
            });
            queryClient.invalidateQueries({
              queryKey: queryKeys.issues.documentRevisions(ref, documentKey),
              ...invalidationOptions,
            });
          } else {
            queryClient.invalidateQueries({
              queryKey: ["issues", "document", ref],
              ...invalidationOptions,
            });
            queryClient.invalidateQueries({
              queryKey: ["issues", "document-revisions", ref],
              ...invalidationOptions,
            });
          }
        }
        if (
          action &&
          (ISSUE_DOCUMENT_ACTIVITY_ACTIONS.has(action) ||
            ISSUE_DOCUMENT_ANNOTATION_ACTIVITY_ACTIONS.has(action))
        ) {
          const documentKey =
            readString(details?.key) ?? readString(details?.documentKey);
          queryClient.invalidateQueries({
            queryKey: documentKey
              ? ["issues", "document-annotations", ref, documentKey]
              : ["issues", "document-annotations", ref],
            ...invalidationOptions,
          });
        }
        if (
          action?.startsWith("issue.thread_interaction_") ||
          ((action === "issue.updated" ||
            action === "issue.status_decision_recorded") &&
            readString(details?.source) === "native_status_decision")
        ) {
          // Native status decisions may materialize a review interaction in the
          // same transaction. Unlike run/activity rows, that card is not present
          // in the streamed PRP projection, so inactive-only invalidation leaves
          // the visible task stale until a full reload.
          queryClient.invalidateQueries({
            queryKey: queryKeys.issues.interactions(ref),
          });
          queryClient.invalidateQueries({
            queryKey: queryKeys.tools.actionRequests(companyId, "pending"),
          });
          queryClient.invalidateQueries({
            queryKey: queryKeys.apps.attention(companyId),
          });
          queryClient.invalidateQueries({
            queryKey: queryKeys.tools.trustRules(companyId),
          });
        }
      }
    }
    return;
  }

  if (entityType === "agent") {
    queryClient.invalidateQueries({
      queryKey: queryKeys.agents.list(companyId),
    });
    queryClient.invalidateQueries({ queryKey: queryKeys.org(companyId) });
    if (entityId) {
      queryClient.invalidateQueries({
        queryKey: queryKeys.agents.detail(entityId),
      });
      queryClient.invalidateQueries({
        queryKey: queryKeys.heartbeats(companyId, entityId),
      });
    }
    return;
  }

  if (entityType === "project") {
    const sourceIssueId = readString((payload.details as Record<string, unknown> | undefined)?.sourceIssueId);
    if (sourceIssueId) queryClient.invalidateQueries({ queryKey: queryKeys.issues.activity(sourceIssueId) });
    queryClient.invalidateQueries({ queryKey: queryKeys.projects.all(companyId) });
    if (entityId) queryClient.invalidateQueries({ queryKey: queryKeys.projects.detail(entityId) });
    return;
  }

  if (entityType === "goal") {
    queryClient.invalidateQueries({
      queryKey: queryKeys.goals.list(companyId),
    });
    if (entityId)
      queryClient.invalidateQueries({
        queryKey: queryKeys.goals.detail(entityId),
      });
    return;
  }

  if (entityType === "approval") {
    queryClient.invalidateQueries({
      queryKey: queryKeys.approvals.list(companyId),
    });
    return;
  }

  if (entityType === "join_request") {
    queryClient.invalidateQueries({
      queryKey: queryKeys.access.joinRequests(companyId),
    });
    return;
  }

  if (entityType === "cost_event") {
    queryClient.invalidateQueries({ queryKey: queryKeys.costs(companyId) });
    queryClient.invalidateQueries({
      queryKey: queryKeys.usageByProvider(companyId),
    });
    queryClient.invalidateQueries({
      queryKey: queryKeys.usageWindowSpend(companyId),
    });
    // usageQuotaWindows is intentionally excluded: quota windows come from external provider
    // apis on a 5-minute poll and do not change in response to cost events logged by agents
    return;
  }

  if (
    entityType === "routine" ||
    entityType === "routine_trigger" ||
    entityType === "routine_run"
  ) {
    queryClient.invalidateQueries({ queryKey: ["routines"] });
    if (
      entityType === "routine" &&
      action &&
      ROUTINE_DOCUMENT_ANNOTATION_ACTIVITY_ACTIONS.has(action) &&
      entityId
    ) {
      const documentKey =
        readString(details?.key) ??
        readString(details?.documentKey) ??
        "description";
      const routineInvalidationOptions = ownActorActivity
        ? { refetchType: "inactive" as const }
        : undefined;
      queryClient.invalidateQueries({
        queryKey: ["routines", "document-annotations", entityId, documentKey],
        ...routineInvalidationOptions,
      });
    }
    return;
  }

  if (entityType === "case") {
    queryClient.invalidateQueries({
      queryKey: queryKeys.cases.list(companyId),
    });
    if (entityId) {
      queryClient.invalidateQueries({
        queryKey: queryKeys.cases.detail(entityId),
      });
      queryClient.invalidateQueries({
        queryKey: queryKeys.cases.events(entityId),
      });
      if (action && CASE_DOCUMENT_ANNOTATION_ACTIVITY_ACTIONS.has(action)) {
        const documentKey =
          readString(details?.key) ?? readString(details?.documentKey);
        const caseInvalidationOptions = ownActorActivity
          ? { refetchType: "inactive" as const }
          : undefined;
        queryClient.invalidateQueries({
          queryKey: documentKey
            ? ["cases", "document-annotations", entityId, documentKey]
            : ["cases", "document-annotations", entityId],
          ...caseInvalidationOptions,
        });
      }
    }
    return;
  }

  if (entityType === "summary_slot") {
    // The Summarizer's authoritative PUT logs `summary_slot.write`; refresh the
    // affected slot + its revisions so the finished summary lands instantly
    // instead of waiting for the card's 3s generating poll tick.
    const scopeKind = readString(details?.scopeKind);
    const slotKey = readString(details?.slotKey);
    const scopeId = readString(details?.scopeId);
    if (scopeKind && slotKey) {
      queryClient.invalidateQueries({
        queryKey: queryKeys.summarySlots.detail(
          companyId,
          scopeKind,
          slotKey,
          scopeId,
        ),
      });
      queryClient.invalidateQueries({
        queryKey: queryKeys.summarySlots.revisions(
          companyId,
          scopeKind,
          slotKey,
          scopeId,
        ),
      });
    }
    return;
  }

  if (entityType === "company") {
    queryClient.invalidateQueries({ queryKey: queryKeys.companies.all });
  }
}

interface ToastGate {
  cooldownHits: Map<string, number[]>;
  suppressUntil: number;
}

function shouldSuppressToast(gate: ToastGate, category: string): boolean {
  const now = Date.now();
  if (now < gate.suppressUntil) return true;

  const hits = gate.cooldownHits.get(category);
  if (!hits) return false;

  const recent = hits.filter((t) => now - t < TOAST_COOLDOWN_WINDOW_MS);
  gate.cooldownHits.set(category, recent);
  return recent.length >= TOAST_COOLDOWN_MAX;
}

function recordToastHit(gate: ToastGate, category: string) {
  const now = Date.now();
  const hits = gate.cooldownHits.get(category) ?? [];
  hits.push(now);
  gate.cooldownHits.set(category, hits);
}

function gatedPushToast(
  gate: ToastGate,
  pushToast: (toast: ToastInput) => string | null,
  category: string,
  toast: ToastInput,
) {
  if (shouldSuppressToast(gate, category)) return;
  const id = pushToast(toast);
  if (id !== null) recordToastHit(gate, category);
}

function handleLiveEvent(
  queryClient: QueryClient,
  expectedCompanyId: string,
  pathname: string,
  event: LiveEvent,
  pushToast: (toast: ToastInput) => string | null,
  gate: ToastGate,
  currentActor: { userId: string | null; agentId: string | null },
) {
  if (event.companyId !== expectedCompanyId) return;

  const nameOf = (id: string) =>
    resolveAgentName(queryClient, expectedCompanyId, id);
  const payload = event.payload ?? {};
  // Resolve membership before terminal lifecycle patches remove live-run rows.
  const suppressRunToast =
    event.type === "heartbeat.run.status" &&
    shouldSuppressRunStatusToastForVisibleIssue(queryClient, pathname, payload);
  const liveStatusPatch = readRunLiveStatusPatchFromPayload(
    payload,
    event.createdAt,
    event.type,
  );
  if (liveStatusPatch) {
    applyRunLiveStatusPatchToCaches(
      queryClient,
      expectedCompanyId,
      pathname,
      liveStatusPatch,
    );
  }
  if (event.type === "heartbeat.run.log") {
    return;
  }

  if (
    event.type === "heartbeat.run.queued" ||
    event.type === "heartbeat.run.status"
  ) {
    const liveRunsPatched = applyRunLifecycleToCompanyLiveRuns(
      queryClient,
      expectedCompanyId,
      payload,
    );
    invalidateHeartbeatQueries(queryClient, expectedCompanyId, payload);
    if (!liveRunsPatched) {
      // A new run we couldn't reconstruct from the event — refetch once to add it.
      queryClient.invalidateQueries({
        queryKey: queryKeys.liveRuns(expectedCompanyId),
      });
    }
    invalidateVisibleIssueRunQueries(queryClient, pathname, payload);
    if (event.type === "heartbeat.run.status") {
      const toast = buildRunStatusToast(payload, nameOf);
      if (toast && !suppressRunToast) {
        gatedPushToast(gate, pushToast, "run-status", toast);
      }
    }
    return;
  }

  if (event.type === "heartbeat.run.progress") {
    invalidateHeartbeatProgressQueries(queryClient, expectedCompanyId, payload);
    invalidateVisibleIssueRunQueries(queryClient, pathname, payload);
    return;
  }

  if (event.type === "heartbeat.run.event") {
    return;
  }

  if (event.type === "agent.status") {
    queryClient.invalidateQueries({
      queryKey: queryKeys.agents.list(expectedCompanyId),
    });
    queryClient.invalidateQueries({
      queryKey: queryKeys.dashboard(expectedCompanyId),
    });
    queryClient.invalidateQueries({
      queryKey: queryKeys.org(expectedCompanyId),
    });
    const agentId = readString(payload.agentId);
    if (agentId)
      queryClient.invalidateQueries({
        queryKey: queryKeys.agents.detail(agentId),
      });
    const toast = buildAgentStatusToast(
      payload,
      nameOf,
      queryClient,
      expectedCompanyId,
    );
    if (
      toast &&
      !shouldSuppressAgentStatusToastForVisibleIssue(
        queryClient,
        pathname,
        payload,
      )
    ) {
      gatedPushToast(gate, pushToast, "agent-status", toast);
    }
    return;
  }

  if (event.type === "activity.logged") {
    invalidateActivityQueries(
      queryClient,
      expectedCompanyId,
      payload,
      currentActor,
      { pathname },
    );
    if (
      shouldDeferVisibleIssueCommentActivity(queryClient, pathname, payload)
    ) {
      void hydrateVisibleIssueComment(queryClient, pathname, payload);
    }
    const action = readString(payload.action);
    const toast =
      buildActivityToast(
        queryClient,
        expectedCompanyId,
        payload,
        currentActor,
      ) ?? buildJoinRequestToast(payload);
    if (
      toast &&
      !shouldSuppressActivityToastForVisibleIssue(
        queryClient,
        pathname,
        payload,
      )
    ) {
      gatedPushToast(gate, pushToast, `activity:${action ?? "unknown"}`, toast);
    }
  }
}

function resolveLiveCompanyId(
  selectedCompanyId: string | null,
  selectedCompanyLiveId: string | null,
): string | null {
  return selectedCompanyId && selectedCompanyId === selectedCompanyLiveId
    ? selectedCompanyId
    : null;
}

function resetSocketHandlers(target: LiveUpdatesSocketLike) {
  target.onopen = null;
  target.onmessage = null;
  target.onerror = null;
  target.onclose = null;
}

function closeSocketQuietly(
  target: LiveUpdatesSocketLike | null,
  reason: string,
) {
  if (!target) return;

  if (target.readyState === SOCKET_CONNECTING) {
    // Let the handshake complete and then close. Calling close() while the
    // socket is still CONNECTING is what triggers the noisy browser error.
    target.onopen = () => {
      resetSocketHandlers(target);
      target.close(1000, reason);
    };
    target.onmessage = null;
    target.onerror = () => undefined;
    target.onclose = null;
    return;
  }

  resetSocketHandlers(target);

  if (target.readyState === SOCKET_OPEN) {
    target.close(1000, reason);
  }
}

export const __liveUpdatesTestUtils = {
  applyRunLifecycleToCompanyLiveRuns,
  buildAgentStatusToast,
  buildRunStatusToast,
  closeSocketQuietly,
  dispatchLiveEventToSubscribers,
  LiveEventSubscriptionContext,
  applyRunLiveStatusPatchToCaches,
  hydrateVisibleIssueComment,
  invalidateActivityQueries,
  invalidateHeartbeatQueries,
  invalidateHeartbeatProgressQueries,
  invalidateVisibleIssueRunQueries,
  readRunLiveStatusPatchFromPayload,
  resolveLiveCompanyId,
  canUseLiveSession,
  shouldDeferIssueRefetchForVisibleAgentActivity,
  shouldDeferVisibleIssueCommentActivity,
  shouldSuppressActivityToastForVisibleIssue,
  shouldSuppressRunStatusToastForVisibleIssue,
  shouldSuppressAgentStatusToastForVisibleIssue,
};

function canUseLiveSession(sessionStatus: string, hasSession: boolean, deploymentMode?: string) {
  return sessionStatus === "success" && (hasSession || deploymentMode === "local_trusted");
}

export function LiveUpdatesProvider({ children }: { children: ReactNode }) {
  const { visible } = usePageVisibility();
  const wasHidden = useRef(!visible);
  const { selectedCompanyId, selectedCompany } = useCompany();
  const queryClient = useQueryClient();
  const { pushToast } = useToastActions();
  const location = useLocation();
  const gateRef = useRef<ToastGate>({
    cooldownHits: new Map(),
    suppressUntil: 0,
  });
  const pathnameRef = useRef(location.pathname);
  const { data: session, status: sessionStatus } = useQuery({
    queryKey: queryKeys.auth.session,
    queryFn: () => authApi.getSession(),
    retry: false,
  });
  const { data: health } = useQuery({ queryKey: queryKeys.health, queryFn: healthApi.get });
  const currentUserId = session?.user?.id ?? session?.session?.userId ?? null;
  const socketAuthKey = session?.session?.id ?? currentUserId ?? "signed_out";
  const liveCompanyId = resolveLiveCompanyId(selectedCompanyId, selectedCompany?.id ?? null);
  const canConnectSocket = canUseLiveSession(sessionStatus, session != null, health?.deploymentMode) && liveCompanyId !== null;
  const currentActorRef = useRef<{ userId: string | null; agentId: string | null }>({
    userId: currentUserId,
    agentId: null,
  });
  const subscribersRef = useRef<Set<CompanyLiveEventHandler>>(new Set());
  const subscribe = useCallback((handler: CompanyLiveEventHandler) => {
    subscribersRef.current.add(handler);
    return () => {
      subscribersRef.current.delete(handler);
    };
  }, []);
  const subscriptionValue = useMemo<LiveEventSubscription>(
    () => ({ subscribe }),
    [subscribe],
  );

  // Coalesce the per-event invalidation storm. Optimistic setQueryData writes
  // still pass straight through (immediate); only invalidateQueries is batched
  // and flushed at most a few times per second.
  const invalidationBatcher = useMemo(
    () => createInvalidationBatcher(queryClient),
    [queryClient],
  );
  const coalescingClient = useMemo(
    () => createCoalescingQueryClient(queryClient, invalidationBatcher),
    [queryClient, invalidationBatcher],
  );
  useEffect(() => () => invalidationBatcher.dispose(), [invalidationBatcher]);

  useEffect(() => {
    pathnameRef.current = location.pathname;
  }, [location.pathname]);

  useEffect(() => {
    currentActorRef.current = {
      userId: currentUserId,
      agentId: null,
    };
  }, [currentUserId]);

  useEffect(() => {
    if (!visible) {
      wasHidden.current = true;
      invalidationBatcher.dispose();
      return;
    }
    if (!canConnectSocket || !liveCompanyId) return;
    if (wasHidden.current) {
      wasHidden.current = false;
      // Reconcile events missed while hidden, including completed runs/issues.
      void queryClient.invalidateQueries({ type: "active" }, { cancelRefetch: false });
    }

    let closed = false;
    let reconnectAttempt = 0;
    let reconnectTimer: number | null = null;
    let socket: WebSocket | null = null;

    const clearReconnect = () => {
      if (reconnectTimer !== null) {
        window.clearTimeout(reconnectTimer);
        reconnectTimer = null;
      }
    };

    const scheduleReconnect = () => {
      if (closed) return;
      reconnectAttempt += 1;
      const delayMs = Math.min(
        15000,
        1000 * 2 ** Math.min(reconnectAttempt - 1, 4),
      );
      reconnectTimer = window.setTimeout(() => {
        reconnectTimer = null;
        connect();
      }, delayMs);
    };

    const connect = () => {
      if (closed) return;
      const url = buildSameOriginWebSocketUrl(
        `/api/companies/${encodeURIComponent(liveCompanyId)}/events/ws`,
      );
      const nextSocket = new WebSocket(url);
      socket = nextSocket;

      nextSocket.onopen = () => {
        if (closed || socket !== nextSocket) {
          closeSocketQuietly(nextSocket, "stale_connection");
          return;
        }
        if (reconnectAttempt > 0) {
          gateRef.current.suppressUntil = Date.now() + RECONNECT_SUPPRESS_MS;
          // Reconcile after a gap: events missed while disconnected can't be
          // replayed yet, so refetch the event-sourced live-runs list once.
          queryClient.invalidateQueries({
            queryKey: queryKeys.liveRuns(liveCompanyId),
          });
        }
        reconnectAttempt = 0;
      };

      nextSocket.onmessage = (message) => {
        if (!getPageVisibility().visible) return;
        const raw = typeof message.data === "string" ? message.data : "";
        if (!raw) return;

        try {
          const parsed = JSON.parse(raw) as LiveEvent;
          handleLiveEvent(
            coalescingClient,
            liveCompanyId,
            pathnameRef.current,
            parsed,
            pushToast,
            gateRef.current,
            {
              userId: currentActorRef.current.userId,
              agentId: currentActorRef.current.agentId,
            },
          );
          // Fan the raw event out to component subscribers after cache
          // handling so any reader sees fresh query data.
          dispatchLiveEventToSubscribers(
            subscribersRef.current,
            liveCompanyId,
            parsed,
          );
        } catch {
          // Ignore non-JSON payloads.
        }
      };

      nextSocket.onerror = () => {
        // Wait for onclose to drive the reconnect. Self-closing here is what
        // produces the "closed before connection established" browser noise.
      };

      nextSocket.onclose = () => {
        if (socket !== nextSocket) return;
        socket = null;
        if (closed) return;
        scheduleReconnect();
      };
    };

    // Delay initial connect slightly so React StrictMode's double-invoke
    // cleanup fires before the WebSocket is created, avoiding the
    // "WebSocket closed before connection established" dev-mode error.
    const connectTimer = window.setTimeout(connect, 0);

    return () => {
      closed = true;
      window.clearTimeout(connectTimer);
      clearReconnect();
      const activeSocket = socket;
      socket = null;
      closeSocketQuietly(activeSocket, "provider_unmount");
    };
  }, [
    visible,
    invalidationBatcher,
    queryClient,
    coalescingClient,
    liveCompanyId,
    pushToast,
    canConnectSocket,
    socketAuthKey,
  ]);

  return (
    <LiveEventSubscriptionContext.Provider value={subscriptionValue}>
      {children}
    </LiveEventSubscriptionContext.Provider>
  );
}