cleanupRegistry.ts

/**
 * Global registry for cleanup functions that should run during graceful shutdown.
 * This module is separate from gracefulShutdown.ts to avoid circular dependencies.
 */

// Global registry for cleanup functions
const cleanupFunctions = new Set<() => Promise<void>>()

/**
 * Register a cleanup function to run during graceful shutdown.
 * @param cleanupFn - Function to run during cleanup (can be sync or async)
 * @returns Unregister function that removes the cleanup handler
 */
export function registerCleanup(cleanupFn: () => Promise<void>): () => void {
  cleanupFunctions.add(cleanupFn)
  return () => cleanupFunctions.delete(cleanupFn) // Return unregister function
}

/**
 * Run all registered cleanup functions.
 * Used internally by gracefulShutdown.
 */
export async function runCleanupFunctions(): Promise<void> {
  await Promise.all(Array.from(cleanupFunctions).map(fn => fn()))
}

// --- Cleanup-safe timer and AbortController helpers ---

export function setCleanupTimeout(fn: (...args: any[]) => void, ms: number): { id: ReturnType<typeof setTimeout>; unregister: () => void } {
  const id = setTimeout(fn, ms)
  const unregister = registerCleanup(() => clearTimeout(id))
  return { id, unregister }
}

export function setCleanupInterval(fn: (...args: any[]) => void, ms: number): ReturnType<typeof setInterval> {
  const id = setInterval(fn, ms)
  registerCleanup(() => clearInterval(id))
  return id
}

export function createCleanupAbortController(): AbortController {
  const controller = new AbortController()
  registerCleanup(() => {
    try { controller.abort() } catch {}
  })
  return controller
}