upstream-rocketchat/apps/meteor/client/views/room/Header/RoomHeaderE2EESetup.tsx

import { lazy } from 'react';

import RoomHeader from './RoomHeader';
import type { RoomHeaderProps } from './RoomHeader';
import { useE2EERoomState } from '../hooks/useE2EERoomState';
import { useE2EEState } from '../hooks/useE2EEState';

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

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

	if (e2eeState === 'SAVE_PASSWORD' || e2eeState === 'ENTER_PASSWORD' || e2eRoomState === 'WAITING_KEYS') {
		return (
			<RoomHeader
				room={room}
				slots={{
					toolbox: {
						content: <RoomToolboxE2EESetup />,
					},
				}}
			/>
		);
	}

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

export default RoomHeaderE2EESetup;