65 lines
2.0 KiB
TypeScript
65 lines
2.0 KiB
TypeScript
|
import { ReactNode, useCallback, useEffect, useMemo } from 'react';
|
||
|
|
import { MatrixError, createClient } from 'matrix-js-sdk';
|
||
|
|
import { AsyncStatus, useAsyncCallback } from '../hooks/useAsyncCallback';
|
||
|
|
import { useAutoDiscoveryInfo } from '../hooks/useAutoDiscoveryInfo';
|
||
|
|
import { promiseFulfilledResult, promiseRejectedResult } from '../utils/common';
|
||
|
|
import {
|
||
|
|
AuthFlows,
|
||
|
|
RegisterFlowStatus,
|
||
|
|
RegisterFlowsResponse,
|
||
|
|
parseRegisterErrResp,
|
||
|
|
} from '../hooks/useAuthFlows';
|
||
|
|
|
||
|
|
type AuthFlowsLoaderProps = {
|
||
|
|
fallback?: () => ReactNode;
|
||
|
|
error?: (err: unknown) => ReactNode;
|
||
|
|
children: (authFlows: AuthFlows) => ReactNode;
|
||
|
|
};
|
||
|
|
export function AuthFlowsLoader({ fallback, error, children }: AuthFlowsLoaderProps) {
|
||
|
|
const autoDiscoveryInfo = useAutoDiscoveryInfo();
|
||
|
|
const baseUrl = autoDiscoveryInfo['m.homeserver'].base_url;
|
||
|
|
|
||
|
|
const mx = useMemo(() => createClient({ baseUrl }), [baseUrl]);
|
||
|
|
|
||
|
|
const [state, load] = useAsyncCallback(
|
||
|
|
useCallback(async () => {
|
||
|
|
const result = await Promise.allSettled([mx.loginFlows(), mx.registerRequest({})]);
|
||
|
|
const loginFlows = promiseFulfilledResult(result[0]);
|
||
|
|
const registerResp = promiseRejectedResult(result[1]) as MatrixError | undefined;
|
||
|
|
let registerFlows: RegisterFlowsResponse = { status: RegisterFlowStatus.InvalidRequest };
|
||
|
|
|
||
|
|
if (typeof registerResp === 'object' && registerResp.httpStatus) {
|
||
|
|
registerFlows = parseRegisterErrResp(registerResp);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!loginFlows) {
|
||
|
|
throw new Error('Missing auth flow!');
|
||
|
|
}
|
||
|
|
if ('errcode' in loginFlows) {
|
||
|
|
throw new Error('Failed to load auth flow!');
|
||
|
|
}
|
||
|
|
|
||
|
|
const authFlows: AuthFlows = {
|
||
|
|
loginFlows,
|
||
|
|
registerFlows,
|
||
|
|
};
|
||
|
|
|
||
|
|
return authFlows;
|
||
|
|
}, [mx])
|
||
|
|
);
|
||
|
|
|
||
|
|
useEffect(() => {
|
||
|
|
load();
|
||
|
|
}, [load]);
|
||
|
|
|
||
|
|
if (state.status === AsyncStatus.Idle || state.status === AsyncStatus.Loading) {
|
||
|
|
return fallback?.();
|
||
|
|
}
|
||
|
|
|
||
|
|
if (state.status === AsyncStatus.Error) {
|
||
|
|
return error?.(state.error);
|
||
|
|
}
|
||
|
|
|
||
|
|
return children(state.data);
|
||
|
|
}
|