RocketChat/Rocket.Chat

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();