mirror of
https://github.com/immich-app/immich.git
synced 2026-01-25 19:04:42 -08:00
99 lines
3.5 KiB
TypeScript
99 lines
3.5 KiB
TypeScript
import { page } from '$app/state';
|
|
import { authManager } from '$lib/managers/auth-manager.svelte';
|
|
import { eventManager } from '$lib/managers/event-manager.svelte';
|
|
import { Route } from '$lib/route';
|
|
import { notificationManager } from '$lib/stores/notification-manager.svelte';
|
|
import type { ReleaseEvent } from '$lib/types';
|
|
import { createEventEmitter } from '$lib/utils/eventemitter';
|
|
import { type AssetResponseDto, type NotificationDto, type ServerVersionResponseDto } from '@immich/sdk';
|
|
import { io, type Socket } from 'socket.io-client';
|
|
import { get, writable } from 'svelte/store';
|
|
import { user } from './user.store';
|
|
|
|
interface AppRestartEvent {
|
|
isMaintenanceMode: boolean;
|
|
}
|
|
|
|
export interface Events {
|
|
on_upload_success: (asset: AssetResponseDto) => void;
|
|
on_user_delete: (id: string) => void;
|
|
on_asset_delete: (assetId: string) => void;
|
|
on_asset_trash: (assetIds: string[]) => void;
|
|
on_asset_update: (asset: AssetResponseDto) => void;
|
|
on_asset_hidden: (assetId: string) => void;
|
|
on_asset_restore: (assetIds: string[]) => void;
|
|
on_asset_stack_update: (assetIds: string[]) => void;
|
|
on_person_thumbnail: (personId: string) => void;
|
|
on_server_version: (serverVersion: ServerVersionResponseDto) => void;
|
|
on_config_update: () => void;
|
|
on_new_release: (event: ReleaseEvent) => void;
|
|
on_session_delete: (sessionId: string) => void;
|
|
on_notification: (notification: NotificationDto) => void;
|
|
|
|
AppRestartV1: (event: AppRestartEvent) => void;
|
|
AssetEditReadyV1: (data: { assetId: string }) => void;
|
|
}
|
|
|
|
const websocket: Socket<Events> = io({
|
|
path: '/api/socket.io',
|
|
transports: ['websocket'],
|
|
reconnection: true,
|
|
forceNew: true,
|
|
autoConnect: false,
|
|
});
|
|
|
|
export const websocketStore = {
|
|
connected: writable<boolean>(false),
|
|
serverVersion: writable<ServerVersionResponseDto>(),
|
|
serverRestarting: writable<undefined | AppRestartEvent>(),
|
|
};
|
|
|
|
export const websocketEvents = createEventEmitter(websocket);
|
|
|
|
websocket
|
|
.on('connect', () => websocketStore.connected.set(true))
|
|
.on('disconnect', () => websocketStore.connected.set(false))
|
|
.on('on_server_version', (serverVersion) => websocketStore.serverVersion.set(serverVersion))
|
|
.on('AppRestartV1', (mode) => websocketStore.serverRestarting.set(mode))
|
|
.on('on_new_release', (event) => eventManager.emit('ReleaseEvent', event))
|
|
.on('on_session_delete', () => authManager.logout())
|
|
.on('on_user_delete', (id) => eventManager.emit('UserAdminDeleted', { id }))
|
|
.on('on_notification', () => notificationManager.refresh())
|
|
.on('connect_error', (e) => console.log('Websocket Connect Error', e));
|
|
|
|
export const openWebsocketConnection = () => {
|
|
try {
|
|
if (get(user) || page.url.pathname.startsWith(Route.maintenanceMode())) {
|
|
websocket.connect();
|
|
}
|
|
} catch (error) {
|
|
console.log('Cannot connect to websocket', error);
|
|
}
|
|
};
|
|
|
|
export const closeWebsocketConnection = () => {
|
|
websocket.disconnect();
|
|
};
|
|
|
|
export const waitForWebsocketEvent = <T extends keyof Events>(
|
|
event: T,
|
|
predicate?: (...args: Parameters<Events[T]>) => boolean,
|
|
timeout: number = 10_000,
|
|
): Promise<Parameters<Events[T]>> => {
|
|
return new Promise((resolve, reject) => {
|
|
// @ts-expect-error: The typings are weird on this?
|
|
const cleanup = websocketEvents.on(event, (...args: Parameters<Events[T]>) => {
|
|
if (!predicate || predicate(...args)) {
|
|
cleanup();
|
|
clearTimeout(timer);
|
|
resolve(args);
|
|
}
|
|
});
|
|
|
|
const timer = setTimeout(() => {
|
|
cleanup();
|
|
reject(new Error(`Timeout waiting for event: ${String(event)}`));
|
|
}, timeout);
|
|
});
|
|
};
|