5chan/src/lib/media-hosting/provider-availability.ts

import { MEDIA_HOSTING_PROVIDERS } from './providers';
import type { MediaHostingRuntime, ProviderId } from './types';

export type ProviderAvailabilityStatus = 'unknown' | 'checking' | 'available' | 'unavailable';
export type ProviderAvailabilitySnapshot = Partial<Record<ProviderId, ProviderAvailabilityStatus>>;

const PROBE_TIMEOUT_MS = 5_000;
const PROBE_CACHE_MS = 10 * 60_000;

type Listener = () => void;

interface ProviderAvailabilityEntry {
  status: ProviderAvailabilityStatus;
  checkedAt: number;
}

const providerById = new Map(MEDIA_HOSTING_PROVIDERS.map((provider) => [provider.id, provider]));
let entries: Record<ProviderId, ProviderAvailabilityEntry> = Object.fromEntries(
  MEDIA_HOSTING_PROVIDERS.map((provider) => [provider.id, { status: 'unknown', checkedAt: 0 }]),
) as Record<ProviderId, ProviderAvailabilityEntry>;
let snapshot: ProviderAvailabilitySnapshot = buildSnapshot();
const listeners = new Set<Listener>();
const inFlight = new Map<ProviderId, Promise<ProviderAvailabilityStatus>>();

function buildSnapshot(): ProviderAvailabilitySnapshot {
  return Object.fromEntries(MEDIA_HOSTING_PROVIDERS.map((provider) => [provider.id, entries[provider.id]?.status ?? 'unknown'])) as ProviderAvailabilitySnapshot;
}

function emit() {
  snapshot = buildSnapshot();
  for (const listener of listeners) {
    listener();
  }
}

function setProviderStatus(provider: ProviderId, status: ProviderAvailabilityStatus) {
  entries = {
    ...entries,
    [provider]: {
      status,
      checkedAt: status === 'checking' ? entries[provider].checkedAt : Date.now(),
    },
  };
  emit();
}

function getRuntimeSupportedProviders(runtime: MediaHostingRuntime): ProviderId[] {
  return MEDIA_HOSTING_PROVIDERS.filter((provider) => provider.supportedRuntimes.includes(runtime)).map((provider) => provider.id);
}

function shouldProbeProvider(provider: ProviderId): boolean {
  if (inFlight.has(provider)) return false;
  const entry = entries[provider];
  return entry.status === 'unknown' || Date.now() - entry.checkedAt > PROBE_CACHE_MS;
}

function probeImageUrl(url: string): Promise<boolean> {
  return new Promise((resolve) => {
    const image = new Image();
    let settled = false;
    const timeoutId = window.setTimeout(() => settle(false), PROBE_TIMEOUT_MS);

    const settle = (available: boolean) => {
      if (settled) return;
      settled = true;
      window.clearTimeout(timeoutId);
      image.onload = null;
      image.onerror = null;
      resolve(available);
    };

    image.onload = () => settle(true);
    image.onerror = () => settle(false);
    image.referrerPolicy = 'no-referrer';
    image.src = url;
  });
}

export function getProviderAvailabilitySnapshot(): ProviderAvailabilitySnapshot {
  return snapshot;
}

export function subscribeProviderAvailability(listener: Listener, runtime?: MediaHostingRuntime): () => void {
  listeners.add(listener);
  if (runtime) {
    startProviderAvailabilityProbes(runtime);
  }
  return () => {
    listeners.delete(listener);
  };
}

function startProviderAvailabilityProbes(runtime: MediaHostingRuntime) {
  for (const provider of getRuntimeSupportedProviders(runtime)) {
    if (shouldProbeProvider(provider)) {
      void probeProviderAvailability(provider);
    }
  }
}

export async function probeProviderAvailability(provider: ProviderId): Promise<ProviderAvailabilityStatus> {
  const running = inFlight.get(provider);
  if (running) return running;

  const definition = providerById.get(provider);
  if (!definition) return 'unavailable';
  if (!shouldProbeProvider(provider)) return entries[provider].status;

  const promise = (async () => {
    setProviderStatus(provider, 'checking');

    if (!definition.availabilityProbeUrls.length) {
      setProviderStatus(provider, 'available');
      return 'available' as const;
    }
    if (typeof window === 'undefined' || typeof Image === 'undefined') {
      setProviderStatus(provider, 'unknown');
      return 'unknown' as const;
    }

    const results = await Promise.all(definition.availabilityProbeUrls.map((url) => probeImageUrl(url)));
    const status: ProviderAvailabilityStatus = results.every(Boolean) ? 'available' : 'unavailable';
    setProviderStatus(provider, status);
    return status;
  })();

  inFlight.set(provider, promise);
  try {
    return await promise;
  } finally {
    inFlight.delete(provider);
  }
}

export async function ensureProviderAvailability(runtime: MediaHostingRuntime): Promise<ProviderAvailabilitySnapshot> {
  await Promise.all(getRuntimeSupportedProviders(runtime).map((provider) => probeProviderAvailability(provider)));
  return getProviderAvailabilitySnapshot();
}

export function resetProviderAvailabilityForTests() {
  entries = Object.fromEntries(MEDIA_HOSTING_PROVIDERS.map((provider) => [provider.id, { status: 'unknown', checkedAt: 0 }])) as Record<
    ProviderId,
    ProviderAvailabilityEntry
  >;
  inFlight.clear();
  emit();
}