geo/apps/desktop-client/src/main/lease-manager.ts

interface LeaseSnapshot {
  activeLeaseId: string | null;
  activeTaskId: string | null;
  activeTaskIds: string[];
  activeCount: number;
  attemptId: string | null;
  startedAt: number | null;
  leaseExpiresAt: number | null;
  lastExtendedAt: number | null;
  lastReleasedAt: number | null;
  lastOutcome: "leased" | "extended" | "completed" | "failed" | "cleared" | null;
}

interface ActiveLeaseState {
  taskId: string;
  attemptId: string | null;
  leaseExpiresAt: number | null;
  startedAt: number;
  lastExtendedAt: number | null;
}

const activeLeases = new Map<string, ActiveLeaseState>();
const leaseState: LeaseSnapshot = {
  activeLeaseId: null,
  activeTaskId: null,
  activeTaskIds: [],
  activeCount: 0,
  attemptId: null,
  startedAt: null,
  leaseExpiresAt: null,
  lastExtendedAt: null,
  lastReleasedAt: null,
  lastOutcome: null,
};

function normalizeLeaseExpiresAt(value: number | string | null | undefined): number | null {
  if (typeof value === "number") {
    return Number.isFinite(value) ? value : null;
  }

  if (typeof value !== "string" || !value.trim()) {
    return null;
  }

  const timestamp = Date.parse(value);
  return Number.isNaN(timestamp) ? null : timestamp;
}

export function setActiveLease(input: {
  taskId: string;
  attemptId?: string | null;
  leaseExpiresAt?: number | string | null;
} | null): void {
  if (!input) {
    activeLeases.clear();
    leaseState.activeLeaseId = null;
    leaseState.activeTaskId = null;
    leaseState.activeTaskIds = [];
    leaseState.activeCount = 0;
    leaseState.attemptId = null;
    leaseState.startedAt = null;
    leaseState.leaseExpiresAt = null;
    leaseState.lastOutcome = "cleared";
    leaseState.lastReleasedAt = Date.now();
    return;
  }

  const normalized: ActiveLeaseState = {
    taskId: input.taskId,
    attemptId: input.attemptId ?? null,
    leaseExpiresAt: normalizeLeaseExpiresAt(input.leaseExpiresAt),
    startedAt: Date.now(),
    lastExtendedAt: null,
  };

  activeLeases.set(normalized.taskId, normalized);
  syncLeaseSnapshot();
  leaseState.lastOutcome = "leased";
}

export function noteLeaseExtended(
  leaseExpiresAt?: number | string | null,
  taskId?: string,
): void {
  if (taskId) {
    const active = activeLeases.get(taskId);
    if (active) {
      active.leaseExpiresAt = normalizeLeaseExpiresAt(leaseExpiresAt) ?? active.leaseExpiresAt;
      active.lastExtendedAt = Date.now();
      activeLeases.set(taskId, active);
    }
  } else {
    const first = activeLeases.values().next().value as ActiveLeaseState | undefined;
    if (first) {
      first.leaseExpiresAt = normalizeLeaseExpiresAt(leaseExpiresAt) ?? first.leaseExpiresAt;
      first.lastExtendedAt = Date.now();
      activeLeases.set(first.taskId, first);
    }
  }

  syncLeaseSnapshot();
  leaseState.lastExtendedAt = Date.now();
  leaseState.lastOutcome = "extended";
}

export function noteLeaseReleased(
  reason: Exclude<LeaseSnapshot["lastOutcome"], "leased" | "extended" | null>,
  taskId?: string,
): void {
  if (taskId) {
    activeLeases.delete(taskId);
  } else {
    activeLeases.clear();
  }

  syncLeaseSnapshot();
  leaseState.lastReleasedAt = Date.now();
  leaseState.lastOutcome = reason;
}

export function getLeaseManagerSnapshot(): LeaseSnapshot {
  return { ...leaseState };
}

function syncLeaseSnapshot(): void {
  const first = activeLeases.values().next().value as ActiveLeaseState | undefined;

  leaseState.activeTaskIds = [...activeLeases.keys()];
  leaseState.activeCount = activeLeases.size;
  leaseState.activeLeaseId = first?.taskId ?? null;
  leaseState.activeTaskId = first?.taskId ?? null;
  leaseState.attemptId = first?.attemptId ?? null;
  leaseState.startedAt = first?.startedAt ?? null;
  leaseState.leaseExpiresAt = first?.leaseExpiresAt ?? null;
}