RocketChat/Rocket.Chat

import React, { lazy } from 'react';

import { E2EEState } from '../../../../app/e2e/client/E2EEState';
import { E2ERoomState } from '../../../../app/e2e/client/E2ERoomState';
import { useE2EERoomState } from '../hooks/useE2EERoomState';
import { useE2EEState } from '../hooks/useE2EEState';
import RoomHeader from './RoomHeader';
import type { RoomHeaderProps } from './RoomHeader';

const RoomToolboxE2EESetup = lazy(() => import('./RoomToolbox/RoomToolboxE2EESetup'));

const RoomHeaderE2EESetup = ({ room, slots = {} }: RoomHeaderProps) => {
    const e2eeState = useE2EEState();
    const e2eRoomState = useE2EERoomState(room._id);

    if (e2eeState === E2EEState.SAVE_PASSWORD || e2eeState === E2EEState.ENTER_PASSWORD || e2eRoomState === E2ERoomState.WAITING_KEYS) {
        return <RoomHeader room={room} slots={slots} roomToolbox={<RoomToolboxE2EESetup />} />;
    }

    return <RoomHeader room={room} slots={slots} />;
};

export default RoomHeaderE2EESetup;