Rocket.Chat/apps/meteor/client/lib/userStatuses.ts

import { UserStatus } from '@rocket.chat/core-typings';
import type { ICustomUserStatus } from '@rocket.chat/core-typings';

import { sdk } from '../../app/utils/client/lib/SDKClient';

export type UserStatusDescriptor = {
	id: string;
	name: string;
	statusType: UserStatus;
	localizeName: boolean;
};

export class UserStatuses implements Iterable<UserStatusDescriptor> {
	public invisibleAllowed = true;

	private store: Map<UserStatusDescriptor['id'], UserStatusDescriptor> = new Map(
		[UserStatus.ONLINE, UserStatus.AWAY, UserStatus.BUSY, UserStatus.OFFLINE].map((status) => [
			status,
			{
				id: status,
				name: status,
				statusType: status,
				localizeName: true,
			},
		]),
	);

	public delete(id: string): void {
		this.store.delete(id);
	}

	public put(customUserStatus: UserStatusDescriptor): void {
		this.store.set(customUserStatus.id, customUserStatus);
	}

	public createFromCustom(customUserStatus: ICustomUserStatus): UserStatusDescriptor {
		if (!this.isValidType(customUserStatus.statusType)) {
			throw new Error('Invalid user status type');
		}

		return {
			name: customUserStatus.name,
			id: customUserStatus._id,
			statusType: customUserStatus.statusType as UserStatus,
			localizeName: false,
		};
	}

	public isValidType(type: string): type is UserStatus {
		return (Object.values(UserStatus) as string[]).includes(type);
	}

	public *[Symbol.iterator]() {
		for (const value of this.store.values()) {
			if (this.invisibleAllowed || value.statusType !== UserStatus.OFFLINE) {
				yield value;
			}
		}
	}

	public async sync() {
		const result = await sdk.call('listCustomUserStatus');
		if (!result) {
			return;
		}

		for (const customStatus of result) {
			this.put(this.createFromCustom(customStatus));
		}
	}

	public watch(cb?: () => void) {
		const updateSubscription = sdk.stream('notify-logged', ['updateCustomUserStatus'], (data) => {
			this.put(this.createFromCustom(data.userStatusData));
			cb?.();
		});

		const deleteSubscription = sdk.stream('notify-logged', ['deleteCustomUserStatus'], (data) => {
			this.delete(data.userStatusData._id);
			cb?.();
		});

		return () => {
			updateSubscription.stop();
			deleteSubscription.stop();
		};
	}
}

export const userStatuses = new UserStatuses();