immich/web/src/lib/utils/maintenance.ts

import { AppRoute } from '$lib/constants';
import { maintenanceStore } from '$lib/stores/maintenance.store';
import { websocketStore } from '$lib/stores/websocket';
import { MaintenanceAction, maintenanceLogin, maintenanceStatus } from '@immich/sdk';

export function maintenanceCreateUrl(url: URL) {
  const target = new URL(AppRoute.MAINTENANCE, url.origin);
  target.searchParams.set('continue', url.pathname + url.search);
  return target.href;
}

export function maintenanceReturnUrl(searchParams: URLSearchParams) {
  return searchParams.get('continue') ?? '/';
}

export function maintenanceShouldRedirect(maintenanceMode: boolean, currentUrl: URL | Location) {
  return maintenanceMode !== currentUrl.pathname.startsWith(AppRoute.MAINTENANCE);
}

export const loadMaintenanceAuth = async () => {
  const query = new URLSearchParams(location.search);

  try {
    const auth = await maintenanceLogin({
      maintenanceLoginDto: {
        token: query.get('token') ?? undefined,
      },
    });

    maintenanceStore.auth.set(auth);
  } catch {
    // silently fail
  }
};

export const loadMaintenanceStatus = async () => {
  try {
    const status = await maintenanceStatus();
    maintenanceStore.status.set(status);

    if (status.action === MaintenanceAction.End) {
      websocketStore.serverRestarting.set({
        isMaintenanceMode: false,
      });
    }
  } catch (error) {
    const status = (error as { status: number })?.status;
    if (status && status >= 500 && status < 600) {
      await new Promise((r) => setTimeout(r, 1e3));
      await loadMaintenanceStatus();
    }
  }
};