RocketChat/Rocket.Chat

import type { ILivechatInquiryRecord, IRoom, ISubscription } from '@rocket.chat/core-typings';
import { useDebouncedState } from '@rocket.chat/fuselage-hooks';
import { useUserPreference, useUserSubscriptions, useSetting } from '@rocket.chat/ui-contexts';
import { useEffect } from 'react';

import { useVideoConfIncomingCalls } from '../../contexts/VideoConfContext';
import { useOmnichannelEnabled } from '../../hooks/omnichannel/useOmnichannelEnabled';
import { useQueuedInquiries } from '../../hooks/omnichannel/useQueuedInquiries';
import { useQueryOptions } from './useQueryOptions';

const query = { open: { $ne: false } };

const emptyQueue: ILivechatInquiryRecord[] = [];

const order: (
    | 'Incoming_Calls'
    | 'Incoming_Livechats'
    | 'Open_Livechats'
    | 'On_Hold_Chats'
    | 'Unread'
    | 'Favorites'
    | 'Teams'
    | 'Discussions'
    | 'Channels'
    | 'Direct_Messages'
    | 'Conversations'
)[] = [
    'Incoming_Calls',
    'Incoming_Livechats',
    'Open_Livechats',
    'On_Hold_Chats',
    'Unread',
    'Favorites',
    'Teams',
    'Discussions',
    'Channels',
    'Direct_Messages',
    'Conversations',
];

export const useRoomList = (): Array<ISubscription & IRoom> => {
    const [roomList, setRoomList] = useDebouncedState<(ISubscription & IRoom)[]>([], 150);

    const showOmnichannel = useOmnichannelEnabled();
    const sidebarGroupByType = useUserPreference('sidebarGroupByType');
    const favoritesEnabled = useUserPreference('sidebarShowFavorites');
    const sidebarOrder = useUserPreference<typeof order>('sidebarSectionsOrder') ?? order;
    const isDiscussionEnabled = useSetting('Discussion_enabled');
    const sidebarShowUnread = useUserPreference('sidebarShowUnread');

    const options = useQueryOptions();

    const rooms = useUserSubscriptions(query, options);

    const inquiries = useQueuedInquiries();

    const incomingCalls = useVideoConfIncomingCalls();

    let queue = emptyQueue;
    if (inquiries.enabled) {
        queue = inquiries.queue;
    }

    useEffect(() => {
        setRoomList(() => {
            const incomingCall = new Set();
            const favorite = new Set();
            const team = new Set();
            const omnichannel = new Set();
            const unread = new Set();
            const channels = new Set();
            const direct = new Set();
            const discussion = new Set();
            const conversation = new Set();
            const onHold = new Set();

            rooms.forEach((room) => {
                if (room.archived) {
                    return;
                }

                if (incomingCalls.find((call) => call.rid === room.rid)) {
                    return incomingCall.add(room);
                }

                if (sidebarShowUnread && (room.alert || room.unread) && !room.hideUnreadStatus) {
                    return unread.add(room);
                }

                if (favoritesEnabled && room.f) {
                    return favorite.add(room);
                }

                if (sidebarGroupByType && room.teamMain) {
                    return team.add(room);
                }

                if (sidebarGroupByType && isDiscussionEnabled && room.prid) {
                    return discussion.add(room);
                }

                if (room.t === 'c' || room.t === 'p') {
                    channels.add(room);
                }

                if (room.t === 'l' && room.onHold) {
                    return showOmnichannel && onHold.add(room);
                }

                if (room.t === 'l') {
                    return showOmnichannel && omnichannel.add(room);
                }

                if (room.t === 'd') {
                    direct.add(room);
                }

                conversation.add(room);
            });

            const groups = new Map();
            incomingCall.size && groups.set('Incoming_Calls', incomingCall);
            showOmnichannel && inquiries.enabled && queue.length && groups.set('Incoming_Livechats', queue);
            showOmnichannel && omnichannel.size && groups.set('Open_Livechats', omnichannel);
            showOmnichannel && onHold.size && groups.set('On_Hold_Chats', onHold);
            sidebarShowUnread && unread.size && groups.set('Unread', unread);
            favoritesEnabled && favorite.size && groups.set('Favorites', favorite);
            sidebarGroupByType && team.size && groups.set('Teams', team);
            sidebarGroupByType && isDiscussionEnabled && discussion.size && groups.set('Discussions', discussion);
            sidebarGroupByType && channels.size && groups.set('Channels', channels);
            sidebarGroupByType && direct.size && groups.set('Direct_Messages', direct);
            !sidebarGroupByType && groups.set('Conversations', conversation);
            return sidebarOrder
                .map((key) => {
                    const group = groups.get(key);
                    if (!group) {
                        return [];
                    }

                    return [key, ...group];
                })
                .flat();
        });
    }, [
        rooms,
        showOmnichannel,
        incomingCalls,
        inquiries.enabled,
        queue,
        sidebarShowUnread,
        favoritesEnabled,
        sidebarGroupByType,
        setRoomList,
        isDiscussionEnabled,
        sidebarOrder,
    ]);

    return roomList;
};