upstream-rocketchat/apps/meteor/client/views/root/hooks/useIframe.ts

import { useStableCallback } from '@rocket.chat/fuselage-hooks';
import { useLoginWithIframe, useLoginWithToken, useSetting } from '@rocket.chat/ui-contexts';
import { useCallback, useEffect, useState } from 'react';

type CallbackError = Error & { error?: string | number; reason?: string; details?: unknown };

export const useIframe = () => {
	const [iframeLoginUrl, setIframeLoginUrl] = useState<string | undefined>(undefined);

	const iframeEnabled = useSetting('Accounts_iframe_enabled', false);
	const accountIframeUrl = useSetting('Accounts_iframe_url', '');
	const apiUrl = useSetting('Accounts_Iframe_api_url', '');
	const apiMethod = useSetting('Accounts_Iframe_api_method', '');

	const iframeLogin = useLoginWithIframe();
	const tokenLogin = useLoginWithToken();

	const enabled = Boolean(iframeEnabled && accountIframeUrl && apiUrl && apiMethod);

	const loginWithToken = useCallback(
		(tokenData: string | { loginToken: string } | { token: string }, callback?: (error: Error | null | undefined) => void) => {
			if (typeof tokenData === 'string') {
				tokenData = {
					token: tokenData,
				};
			}
			if ('loginToken' in tokenData) {
				tokenLogin(tokenData.loginToken, callback);
			}
			if ('token' in tokenData) {
				iframeLogin(tokenData.token, callback);
			}
		},
		[iframeLogin, tokenLogin],
	);

	const tryLogin = useStableCallback(async (callback?: (error: Error | null | undefined, result: unknown) => void) => {
		if (!enabled) {
			return;
		}

		let url = accountIframeUrl;
		let separator = '?';
		if (url.indexOf('?') > -1) {
			separator = '&';
		}

		if (navigator.userAgent.indexOf('Electron') > -1) {
			url += `${separator}client=electron`;
		}

		try {
			const result = await fetch(apiUrl, {
				method: apiMethod,
				headers: undefined,
				credentials: 'include',
			});

			if (!result.ok || result.status !== 200) {
				setIframeLoginUrl(url);
				callback?.(new Error(), null);
				return;
			}

			const body = await result.json();
			loginWithToken(body, async (error: CallbackError | Error | null | undefined) => {
				if (error) {
					setIframeLoginUrl(url);
				} else {
					setIframeLoginUrl(undefined);
				}
				callback?.(error, body);
			});
		} catch (error) {
			setIframeLoginUrl(url);
			callback?.(error instanceof Error ? error : undefined, null);
		}
	});

	useEffect(() => {
		tryLogin();
	}, [tryLogin]);

	return {
		enabled,
		tryLogin,
		loginWithToken,
		iframeLoginUrl,
	};
};