Rocket.Chat/client/hooks/useRoomIcon.tsx

import React, { ReactNode } from 'react';

import { IRoom, isOmnichannelRoom, isDirectMessageRoom } from '../../definition/IRoom';
import { ReactiveUserStatus } from '../components/UserStatus';

export const colors = {
	busy: 'danger-500',
	away: 'warning-600',
	online: 'success-500',
	offline: 'neutral-600',
};

export const useRoomIcon = (room: IRoom): ReactNode | { name: string; color?: string } | null => {
	if (room.prid) {
		return { name: 'baloons' };
	}

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

	if (isOmnichannelRoom(room)) {
		const icon =
			{
				widget: 'livechat',
				email: 'mail',
				sms: 'sms',
				app: 'headset', // TODO: use app icon
				api: 'headset', // TODO: use api icon
				other: 'headset',
			}[room.source?.type || 'other'] || 'headset';

		return {
			name: icon,
			color: colors[room.v.status || 'offline'],
		};
	}
	if (isDirectMessageRoom(room)) {
		if (room.uids && room.uids.length > 2) {
			return { name: 'balloon' };
		}
		if (room.uids && room.uids.length > 0) {
			return <ReactiveUserStatus uid={room.uids.find((uid) => uid !== room.u._id) || room.u._id} />;
		}
		return { name: 'at' };
	}

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