RocketChat/Rocket.Chat

import type { IRoom } from '@rocket.chat/core-typings';
import { isRoomFederated, isDirectMessageRoom } from '@rocket.chat/core-typings';
import type { Icon } from '@rocket.chat/fuselage';
import type { ComponentProps, ReactElement } from 'react';
import React from 'react';

import { ReactiveUserStatus } from '../components/UserStatus';

export const useRoomIcon = (
    room: Pick<IRoom, 't' | 'prid' | 'teamMain' | 'uids' | 'u'>,
): ReactElement | ComponentProps<typeof Icon> | null => {
    if (isRoomFederated(room)) {
        return { name: 'globe' };
    }

    if (room.prid) {
        return { name: 'baloons' };
    }

    if (room.teamMain) {
        return { name: room.t === 'p' ? 'team-lock' : 'team' };
    }

    if (isDirectMessageRoom(room)) {
        if (room.uids && room.uids.length > 2) {
            return { name: 'balloon' };
        }

        if (room.uids && room.uids.length > 0) {
            const uid = room.uids.find((uid) => uid !== room?.u?._id) || room?.u?._id;

            if (!uid) {
                return null;
            }

            return <ReactiveUserStatus uid={uid} />;
        }

        return { name: 'at' };
    }

    switch (room.t) {
        case 'p':
            return { name: 'hashtag-lock' };
        case 'c':
            return { name: 'hash' };
        default:
            return null;
    }
};