b51eb6471f
Co-authored-by: Cursor <cursoragent@cursor.com>
725 lines
24 KiB
TypeScript
725 lines
24 KiB
TypeScript
import { ApolloError, useApolloClient } from '@apollo/client';
|
|
import { useCallback } from 'react';
|
|
import {
|
|
snapshot_UNSTABLE,
|
|
useGotoRecoilSnapshot,
|
|
useRecoilCallback,
|
|
} from 'recoil';
|
|
import { AppPath } from 'twenty-shared/types';
|
|
|
|
import { REACT_APP_SERVER_BASE_URL } from '~/config';
|
|
import {
|
|
useCheckUserExistsLazyQuery,
|
|
useGetAuthTokensFromLoginTokenMutation,
|
|
useGetAuthTokensFromOtpMutation,
|
|
useGetLoginTokenFromCredentialsMutation,
|
|
useSignInMutation,
|
|
useSignUpInWorkspaceMutation,
|
|
useSignUpMutation,
|
|
useVerifyEmailAndGetLoginTokenMutation,
|
|
useVerifyEmailAndGetWorkspaceAgnosticTokenMutation,
|
|
type AuthToken,
|
|
type AuthTokenPair,
|
|
} from '~/generated-metadata/graphql';
|
|
|
|
import { tokenPairState } from '@/auth/states/tokenPairState';
|
|
import { useRecoilValueV2 } from '@/ui/utilities/state/jotai/hooks/useRecoilValueV2';
|
|
import { useSetRecoilStateV2 } from '@/ui/utilities/state/jotai/hooks/useSetRecoilStateV2';
|
|
import { jotaiStore } from '@/ui/utilities/state/jotai/jotaiStore';
|
|
|
|
import { isAppEffectRedirectEnabledState } from '@/app/states/isAppEffectRedirectEnabledState';
|
|
import { availableWorkspacesState } from '@/auth/states/availableWorkspacesState';
|
|
import { currentUserState } from '@/auth/states/currentUserState';
|
|
import { currentUserWorkspaceState } from '@/auth/states/currentUserWorkspaceState';
|
|
import { currentWorkspaceMemberState } from '@/auth/states/currentWorkspaceMemberState';
|
|
import { currentWorkspaceMembersState } from '@/auth/states/currentWorkspaceMembersState';
|
|
import { currentWorkspaceState } from '@/auth/states/currentWorkspaceState';
|
|
import { useSignUpInNewWorkspace } from '@/auth/sign-in-up/hooks/useSignUpInNewWorkspace';
|
|
import { metadataStoreState } from '@/app/states/metadataStoreState';
|
|
import { lastAuthenticatedMethodState } from '@/auth/states/lastAuthenticatedMethodState';
|
|
import { loginTokenState } from '@/auth/states/loginTokenState';
|
|
import {
|
|
SignInUpStep,
|
|
signInUpStepState,
|
|
} from '@/auth/states/signInUpStepState';
|
|
import { workspacePublicDataState } from '@/auth/states/workspacePublicDataState';
|
|
import { coreViewsState } from '@/views/states/coreViewState';
|
|
import { type BillingCheckoutSession } from '@/auth/types/billingCheckoutSession.type';
|
|
import {
|
|
countAvailableWorkspaces,
|
|
getFirstAvailableWorkspaces,
|
|
} from '@/auth/utils/availableWorkspacesUtils';
|
|
import { useRequestFreshCaptchaToken } from '@/captcha/hooks/useRequestFreshCaptchaToken';
|
|
import { isCaptchaScriptLoadedState } from '@/captcha/states/isCaptchaScriptLoadedState';
|
|
import { isEmailVerificationRequiredState } from '@/client-config/states/isEmailVerificationRequiredState';
|
|
import { isMultiWorkspaceEnabledState } from '@/client-config/states/isMultiWorkspaceEnabledState';
|
|
import { useLastAuthenticatedWorkspaceDomain } from '@/domain-manager/hooks/useLastAuthenticatedWorkspaceDomain';
|
|
import { useOrigin } from '@/domain-manager/hooks/useOrigin';
|
|
import { useRedirect } from '@/domain-manager/hooks/useRedirect';
|
|
import { useRedirectToWorkspaceDomain } from '@/domain-manager/hooks/useRedirectToWorkspaceDomain';
|
|
import { domainConfigurationState } from '@/domain-manager/states/domainConfigurationState';
|
|
import { useLoadMockedObjectMetadataItems } from '@/object-metadata/hooks/useLoadMockedObjectMetadataItems';
|
|
import { useRefreshObjectMetadataItems } from '@/object-metadata/hooks/useRefreshObjectMetadataItems';
|
|
import { sseClientState } from '@/sse-db-event/states/sseClientState';
|
|
import { getSnapshotValue } from '@/ui/utilities/state/utils/getSnapshotValue';
|
|
import { useLoadCurrentUser } from '@/users/hooks/useLoadCurrentUser';
|
|
import { workspaceAuthProvidersState } from '@/workspace/states/workspaceAuthProvidersState';
|
|
import { i18n } from '@lingui/core';
|
|
import { useNavigate, useSearchParams } from 'react-router-dom';
|
|
import { SOURCE_LOCALE } from 'twenty-shared/translations';
|
|
import { isDefined } from 'twenty-shared/utils';
|
|
import { cookieStorage } from '~/utils/cookie-storage';
|
|
import { getWorkspaceUrl } from '~/utils/getWorkspaceUrl';
|
|
|
|
export const useAuth = () => {
|
|
const setTokenPair = useSetRecoilStateV2(tokenPairState);
|
|
const setLoginToken = useSetRecoilStateV2(loginTokenState);
|
|
const setIsAppEffectRedirectEnabled = useSetRecoilStateV2(
|
|
isAppEffectRedirectEnabledState,
|
|
);
|
|
|
|
const { origin } = useOrigin();
|
|
const { requestFreshCaptchaToken } = useRequestFreshCaptchaToken();
|
|
const isCaptchaScriptLoaded = useRecoilValueV2(isCaptchaScriptLoadedState);
|
|
const isMultiWorkspaceEnabled = useRecoilValueV2(
|
|
isMultiWorkspaceEnabledState,
|
|
);
|
|
const isEmailVerificationRequired = useRecoilValueV2(
|
|
isEmailVerificationRequiredState,
|
|
);
|
|
const { loadCurrentUser } = useLoadCurrentUser();
|
|
|
|
const { refreshObjectMetadataItems } = useRefreshObjectMetadataItems();
|
|
const { createWorkspace } = useSignUpInNewWorkspace();
|
|
|
|
const setSignInUpStep = useSetRecoilStateV2(signInUpStepState);
|
|
const { redirect } = useRedirect();
|
|
const { redirectToWorkspaceDomain } = useRedirectToWorkspaceDomain();
|
|
|
|
const [getLoginTokenFromCredentials] =
|
|
useGetLoginTokenFromCredentialsMutation();
|
|
const [signIn] = useSignInMutation();
|
|
const [signUp] = useSignUpMutation();
|
|
const [signUpInWorkspace] = useSignUpInWorkspaceMutation();
|
|
const [getAuthTokensFromLoginToken] =
|
|
useGetAuthTokensFromLoginTokenMutation();
|
|
const [verifyEmailAndGetLoginToken] =
|
|
useVerifyEmailAndGetLoginTokenMutation();
|
|
const [verifyEmailAndGetWorkspaceAgnosticToken] =
|
|
useVerifyEmailAndGetWorkspaceAgnosticTokenMutation();
|
|
const [getAuthTokensFromOtp] = useGetAuthTokensFromOtpMutation();
|
|
|
|
const workspacePublicData = useRecoilValueV2(workspacePublicDataState);
|
|
|
|
const { setLastAuthenticateWorkspaceDomain } =
|
|
useLastAuthenticatedWorkspaceDomain();
|
|
const [checkUserExistsQuery, { data: checkUserExistsData }] =
|
|
useCheckUserExistsLazyQuery();
|
|
|
|
const client = useApolloClient();
|
|
|
|
const goToRecoilSnapshot = useGotoRecoilSnapshot();
|
|
|
|
const [, setSearchParams] = useSearchParams();
|
|
|
|
const navigate = useNavigate();
|
|
const { loadMockedObjectMetadataItems } = useLoadMockedObjectMetadataItems();
|
|
|
|
const clearSession = useRecoilCallback(
|
|
({ snapshot }) =>
|
|
async () => {
|
|
const sseClient = getSnapshotValue(snapshot, sseClientState);
|
|
|
|
sseClient?.dispose();
|
|
|
|
const emptySnapshot = snapshot_UNSTABLE();
|
|
|
|
const authProvidersValue = jotaiStore.get(
|
|
workspaceAuthProvidersState.atom,
|
|
);
|
|
const domainConfigurationValue = jotaiStore.get(
|
|
domainConfigurationState.atom,
|
|
);
|
|
const workspacePublicDataValue = jotaiStore.get(
|
|
workspacePublicDataState.atom,
|
|
);
|
|
const lastAuthenticatedMethod = jotaiStore.get(
|
|
lastAuthenticatedMethodState.atom,
|
|
);
|
|
const isCaptchaScriptLoadedValue = jotaiStore.get(
|
|
isCaptchaScriptLoadedState.atom,
|
|
);
|
|
|
|
const initialSnapshot = emptySnapshot.map(() => {
|
|
return undefined;
|
|
});
|
|
|
|
sessionStorage.clear();
|
|
localStorage.clear();
|
|
|
|
goToRecoilSnapshot(initialSnapshot);
|
|
|
|
jotaiStore.set(workspaceAuthProvidersState.atom, authProvidersValue);
|
|
jotaiStore.set(workspacePublicDataState.atom, workspacePublicDataValue);
|
|
jotaiStore.set(metadataStoreState.atomFamily('objects'), {
|
|
current: [],
|
|
draft: [],
|
|
status: 'empty',
|
|
});
|
|
jotaiStore.set(metadataStoreState.atomFamily('views'), {
|
|
current: [],
|
|
draft: [],
|
|
status: 'empty',
|
|
});
|
|
jotaiStore.set(metadataStoreState.atomFamily('pageLayouts'), {
|
|
current: [],
|
|
draft: [],
|
|
status: 'empty',
|
|
});
|
|
jotaiStore.set(metadataStoreState.atomFamily('logicFunctions'), {
|
|
current: [],
|
|
draft: [],
|
|
status: 'empty',
|
|
});
|
|
jotaiStore.set(domainConfigurationState.atom, domainConfigurationValue);
|
|
jotaiStore.set(
|
|
isCaptchaScriptLoadedState.atom,
|
|
isCaptchaScriptLoadedValue,
|
|
);
|
|
jotaiStore.set(
|
|
lastAuthenticatedMethodState.atom,
|
|
lastAuthenticatedMethod,
|
|
);
|
|
|
|
// Reset user-data Jotai states that were migrated from Recoil
|
|
// (Recoil snapshot reset no longer handles these since they are Jotai V2)
|
|
jotaiStore.set(tokenPairState.atom, null);
|
|
jotaiStore.set(currentUserState.atom, null);
|
|
jotaiStore.set(currentWorkspaceState.atom, null);
|
|
jotaiStore.set(currentUserWorkspaceState.atom, null);
|
|
jotaiStore.set(currentWorkspaceMemberState.atom, null);
|
|
jotaiStore.set(currentWorkspaceMembersState.atom, []);
|
|
jotaiStore.set(availableWorkspacesState.atom, {
|
|
availableWorkspacesForSignIn: [],
|
|
availableWorkspacesForSignUp: [],
|
|
});
|
|
jotaiStore.set(loginTokenState.atom, null);
|
|
jotaiStore.set(signInUpStepState.atom, SignInUpStep.Init);
|
|
jotaiStore.set(coreViewsState.atom, []);
|
|
|
|
await client.clearStore();
|
|
setLastAuthenticateWorkspaceDomain(null);
|
|
await loadMockedObjectMetadataItems();
|
|
navigate(AppPath.SignInUp);
|
|
},
|
|
[
|
|
goToRecoilSnapshot,
|
|
client,
|
|
setLastAuthenticateWorkspaceDomain,
|
|
loadMockedObjectMetadataItems,
|
|
navigate,
|
|
],
|
|
);
|
|
|
|
const handleSetAuthTokens = useCallback(
|
|
(tokens: AuthTokenPair) => {
|
|
setTokenPair(tokens);
|
|
cookieStorage.setItem('tokenPair', JSON.stringify(tokens));
|
|
},
|
|
[setTokenPair],
|
|
);
|
|
|
|
const handleGetLoginTokenFromCredentials = useCallback(
|
|
async (email: string, password: string, captchaToken?: string) => {
|
|
try {
|
|
const getLoginTokenResult = await getLoginTokenFromCredentials({
|
|
variables: {
|
|
email,
|
|
password,
|
|
captchaToken,
|
|
origin,
|
|
},
|
|
});
|
|
if (isDefined(getLoginTokenResult.errors)) {
|
|
throw getLoginTokenResult.errors;
|
|
}
|
|
|
|
if (!getLoginTokenResult.data?.getLoginTokenFromCredentials) {
|
|
throw new Error('No login token');
|
|
}
|
|
|
|
return getLoginTokenResult.data.getLoginTokenFromCredentials;
|
|
} catch (error) {
|
|
// TODO: Get intellisense for graphql error extensions code (codegen?)
|
|
if (
|
|
error instanceof ApolloError &&
|
|
error.graphQLErrors[0]?.extensions?.subCode === 'EMAIL_NOT_VERIFIED'
|
|
) {
|
|
setSearchParams({ email });
|
|
setSignInUpStep(SignInUpStep.EmailVerification);
|
|
throw error;
|
|
}
|
|
throw error;
|
|
}
|
|
},
|
|
[getLoginTokenFromCredentials, setSearchParams, setSignInUpStep, origin],
|
|
);
|
|
|
|
const handleverifyEmailAndGetLoginToken = useCallback(
|
|
async (
|
|
emailVerificationToken: string,
|
|
email: string,
|
|
captchaToken?: string,
|
|
) => {
|
|
const loginTokenResult = await verifyEmailAndGetLoginToken({
|
|
variables: {
|
|
email,
|
|
emailVerificationToken,
|
|
captchaToken,
|
|
origin,
|
|
},
|
|
});
|
|
|
|
if (isDefined(loginTokenResult.errors)) {
|
|
throw loginTokenResult.errors;
|
|
}
|
|
|
|
if (!loginTokenResult.data?.verifyEmailAndGetLoginToken) {
|
|
throw new Error('No login token');
|
|
}
|
|
|
|
return loginTokenResult.data.verifyEmailAndGetLoginToken;
|
|
},
|
|
[verifyEmailAndGetLoginToken, origin],
|
|
);
|
|
|
|
const handleverifyEmailAndGetWorkspaceAgnosticToken = useCallback(
|
|
async (
|
|
emailVerificationToken: string,
|
|
email: string,
|
|
captchaToken?: string,
|
|
) => {
|
|
const { data, errors } = await verifyEmailAndGetWorkspaceAgnosticToken({
|
|
variables: {
|
|
email,
|
|
emailVerificationToken,
|
|
captchaToken,
|
|
},
|
|
});
|
|
|
|
if (isDefined(errors)) {
|
|
throw errors;
|
|
}
|
|
|
|
if (!data?.verifyEmailAndGetWorkspaceAgnosticToken) {
|
|
throw new Error('No workspace agnostic token in result');
|
|
}
|
|
|
|
handleSetAuthTokens(data.verifyEmailAndGetWorkspaceAgnosticToken.tokens);
|
|
|
|
const { user } = await loadCurrentUser();
|
|
|
|
if (countAvailableWorkspaces(user.availableWorkspaces) === 0) {
|
|
return await createWorkspace({ newTab: false });
|
|
}
|
|
|
|
setSignInUpStep(SignInUpStep.WorkspaceSelection);
|
|
},
|
|
[
|
|
createWorkspace,
|
|
verifyEmailAndGetWorkspaceAgnosticToken,
|
|
handleSetAuthTokens,
|
|
loadCurrentUser,
|
|
setSignInUpStep,
|
|
],
|
|
);
|
|
|
|
const handleSetLoginToken = useCallback(
|
|
(token: AuthToken['token']) => {
|
|
setLoginToken(token);
|
|
},
|
|
[setLoginToken],
|
|
);
|
|
|
|
const handleLoadWorkspaceAfterAuthentication = useCallback(
|
|
async (authTokens: AuthTokenPair) => {
|
|
handleSetAuthTokens(authTokens);
|
|
|
|
setIsAppEffectRedirectEnabled(false);
|
|
|
|
// TODO: We can't parallelize this yet because when loadCurrentUSer is loaded
|
|
// then UserProvider updates its children and PrefetchDataProvider is then triggered
|
|
// which requires the correct metadata to be loaded (not the mocks)
|
|
await loadCurrentUser();
|
|
await refreshObjectMetadataItems();
|
|
},
|
|
[
|
|
loadCurrentUser,
|
|
handleSetAuthTokens,
|
|
refreshObjectMetadataItems,
|
|
setIsAppEffectRedirectEnabled,
|
|
],
|
|
);
|
|
|
|
const handleGetAuthTokensFromLoginToken = useCallback(
|
|
async (loginToken: string) => {
|
|
try {
|
|
const getAuthTokensResult = await getAuthTokensFromLoginToken({
|
|
variables: {
|
|
loginToken: loginToken,
|
|
origin,
|
|
},
|
|
});
|
|
|
|
if (isDefined(getAuthTokensResult.errors)) {
|
|
throw getAuthTokensResult.errors;
|
|
}
|
|
|
|
if (!getAuthTokensResult.data?.getAuthTokensFromLoginToken) {
|
|
throw new Error('No getAuthTokensFromLoginToken result');
|
|
}
|
|
|
|
await handleLoadWorkspaceAfterAuthentication(
|
|
getAuthTokensResult.data.getAuthTokensFromLoginToken.tokens,
|
|
);
|
|
} catch (error) {
|
|
if (
|
|
error instanceof ApolloError &&
|
|
error.graphQLErrors[0]?.extensions?.subCode ===
|
|
'TWO_FACTOR_AUTHENTICATION_PROVISION_REQUIRED'
|
|
) {
|
|
handleSetLoginToken(loginToken);
|
|
navigate(AppPath.SignInUp);
|
|
setSignInUpStep(SignInUpStep.TwoFactorAuthenticationProvision);
|
|
}
|
|
|
|
if (
|
|
error instanceof ApolloError &&
|
|
error.graphQLErrors[0]?.extensions?.subCode ===
|
|
'TWO_FACTOR_AUTHENTICATION_VERIFICATION_REQUIRED'
|
|
) {
|
|
handleSetLoginToken(loginToken);
|
|
navigate(AppPath.SignInUp);
|
|
setSignInUpStep(SignInUpStep.TwoFactorAuthenticationVerification);
|
|
}
|
|
}
|
|
},
|
|
[
|
|
handleSetLoginToken,
|
|
getAuthTokensFromLoginToken,
|
|
origin,
|
|
handleLoadWorkspaceAfterAuthentication,
|
|
setSignInUpStep,
|
|
navigate,
|
|
],
|
|
);
|
|
|
|
const handleCredentialsSignIn = useCallback(
|
|
async (email: string, password: string, captchaToken?: string) => {
|
|
signIn({
|
|
variables: { email, password, captchaToken },
|
|
onCompleted: async (data) => {
|
|
handleSetAuthTokens(data.signIn.tokens);
|
|
const { user } = await loadCurrentUser();
|
|
|
|
const availableWorkspacesCount = countAvailableWorkspaces(
|
|
user.availableWorkspaces,
|
|
);
|
|
|
|
if (availableWorkspacesCount === 0) {
|
|
return createWorkspace();
|
|
}
|
|
|
|
if (availableWorkspacesCount === 1) {
|
|
const targetWorkspace = getFirstAvailableWorkspaces(
|
|
user.availableWorkspaces,
|
|
);
|
|
return await redirectToWorkspaceDomain(
|
|
getWorkspaceUrl(targetWorkspace.workspaceUrls),
|
|
targetWorkspace.loginToken ? AppPath.Verify : AppPath.SignInUp,
|
|
{
|
|
...(targetWorkspace.loginToken && {
|
|
loginToken: targetWorkspace.loginToken,
|
|
}),
|
|
email: user.email,
|
|
},
|
|
);
|
|
}
|
|
|
|
setSignInUpStep(SignInUpStep.WorkspaceSelection);
|
|
},
|
|
onError: (error) => {
|
|
if (
|
|
error instanceof ApolloError &&
|
|
error.graphQLErrors[0]?.extensions?.subCode === 'EMAIL_NOT_VERIFIED'
|
|
) {
|
|
setSearchParams({ email });
|
|
setSignInUpStep(SignInUpStep.EmailVerification);
|
|
throw error;
|
|
}
|
|
throw error;
|
|
},
|
|
});
|
|
},
|
|
[
|
|
handleSetAuthTokens,
|
|
redirectToWorkspaceDomain,
|
|
signIn,
|
|
loadCurrentUser,
|
|
setSearchParams,
|
|
setSignInUpStep,
|
|
createWorkspace,
|
|
],
|
|
);
|
|
|
|
const handleCredentialsSignUp = useCallback(
|
|
async (email: string, password: string, captchaToken?: string) => {
|
|
const signUpResult = await signUp({
|
|
variables: {
|
|
email,
|
|
password,
|
|
captchaToken,
|
|
locale: i18n.locale ?? SOURCE_LOCALE,
|
|
},
|
|
});
|
|
|
|
if (isDefined(signUpResult.errors)) {
|
|
throw signUpResult.errors;
|
|
}
|
|
|
|
if (isEmailVerificationRequired) {
|
|
setSearchParams({ email });
|
|
setSignInUpStep(SignInUpStep.EmailVerification);
|
|
return null;
|
|
}
|
|
|
|
if (!signUpResult.data?.signUp) {
|
|
throw new Error('No signUp result');
|
|
}
|
|
|
|
handleSetAuthTokens(signUpResult.data.signUp.tokens);
|
|
|
|
const { user } = await loadCurrentUser();
|
|
|
|
if (countAvailableWorkspaces(user.availableWorkspaces) === 0) {
|
|
return await createWorkspace({ newTab: false });
|
|
}
|
|
|
|
setSignInUpStep(SignInUpStep.WorkspaceSelection);
|
|
},
|
|
[
|
|
isEmailVerificationRequired,
|
|
setSearchParams,
|
|
handleSetAuthTokens,
|
|
signUp,
|
|
loadCurrentUser,
|
|
setSignInUpStep,
|
|
createWorkspace,
|
|
],
|
|
);
|
|
|
|
const handleCredentialsSignInInWorkspace = useCallback(
|
|
async (email: string, password: string, captchaToken?: string) => {
|
|
const { loginToken } = await handleGetLoginTokenFromCredentials(
|
|
email,
|
|
password,
|
|
captchaToken,
|
|
);
|
|
await handleGetAuthTokensFromLoginToken(loginToken.token);
|
|
},
|
|
[handleGetLoginTokenFromCredentials, handleGetAuthTokensFromLoginToken],
|
|
);
|
|
|
|
const handleSignOut = useCallback(async () => {
|
|
await clearSession();
|
|
if (isCaptchaScriptLoaded) await requestFreshCaptchaToken();
|
|
}, [clearSession, isCaptchaScriptLoaded, requestFreshCaptchaToken]);
|
|
|
|
const handleCredentialsSignUpInWorkspace = useCallback(
|
|
async ({
|
|
email,
|
|
password,
|
|
workspaceInviteHash,
|
|
workspacePersonalInviteToken,
|
|
captchaToken,
|
|
verifyEmailRedirectPath,
|
|
}: {
|
|
email: string;
|
|
password: string;
|
|
workspaceInviteHash?: string;
|
|
workspacePersonalInviteToken?: string;
|
|
captchaToken?: string;
|
|
verifyEmailRedirectPath?: string;
|
|
}) => {
|
|
const signUpInWorkspaceResult = await signUpInWorkspace({
|
|
variables: {
|
|
email,
|
|
password,
|
|
workspaceInviteHash,
|
|
workspacePersonalInviteToken,
|
|
captchaToken,
|
|
locale: i18n.locale ?? SOURCE_LOCALE,
|
|
...(workspacePublicData?.id
|
|
? { workspaceId: workspacePublicData.id }
|
|
: {}),
|
|
verifyEmailRedirectPath,
|
|
},
|
|
});
|
|
|
|
if (isDefined(signUpInWorkspaceResult.errors)) {
|
|
throw signUpInWorkspaceResult.errors;
|
|
}
|
|
|
|
if (!signUpInWorkspaceResult.data?.signUpInWorkspace) {
|
|
throw new Error('No login token');
|
|
}
|
|
|
|
if (isEmailVerificationRequired) {
|
|
setSearchParams({ email });
|
|
setSignInUpStep(SignInUpStep.EmailVerification);
|
|
return null;
|
|
}
|
|
|
|
if (isMultiWorkspaceEnabled) {
|
|
return await redirectToWorkspaceDomain(
|
|
getWorkspaceUrl(
|
|
signUpInWorkspaceResult.data.signUpInWorkspace.workspace
|
|
.workspaceUrls,
|
|
),
|
|
isEmailVerificationRequired ? AppPath.SignInUp : AppPath.Verify,
|
|
{
|
|
...(!isEmailVerificationRequired && {
|
|
loginToken:
|
|
signUpInWorkspaceResult.data.signUpInWorkspace.loginToken.token,
|
|
}),
|
|
email,
|
|
},
|
|
);
|
|
}
|
|
|
|
await handleGetAuthTokensFromLoginToken(
|
|
signUpInWorkspaceResult.data?.signUpInWorkspace.loginToken.token,
|
|
);
|
|
},
|
|
[
|
|
signUpInWorkspace,
|
|
workspacePublicData,
|
|
isMultiWorkspaceEnabled,
|
|
handleGetAuthTokensFromLoginToken,
|
|
setSignInUpStep,
|
|
setSearchParams,
|
|
isEmailVerificationRequired,
|
|
redirectToWorkspaceDomain,
|
|
],
|
|
);
|
|
|
|
const buildRedirectUrl = useCallback(
|
|
(
|
|
path: string,
|
|
params: {
|
|
workspacePersonalInviteToken?: string;
|
|
workspaceInviteHash?: string;
|
|
billingCheckoutSession?: BillingCheckoutSession;
|
|
action?: string;
|
|
},
|
|
) => {
|
|
const url = new URL(`${REACT_APP_SERVER_BASE_URL}${path}`);
|
|
if (isDefined(params.workspaceInviteHash)) {
|
|
url.searchParams.set('workspaceInviteHash', params.workspaceInviteHash);
|
|
}
|
|
if (isDefined(params.workspacePersonalInviteToken)) {
|
|
url.searchParams.set(
|
|
'inviteToken',
|
|
params.workspacePersonalInviteToken,
|
|
);
|
|
}
|
|
if (isDefined(params.billingCheckoutSession)) {
|
|
url.searchParams.set(
|
|
'billingCheckoutSessionState',
|
|
JSON.stringify(params.billingCheckoutSession),
|
|
);
|
|
}
|
|
|
|
if (isDefined(params.action)) {
|
|
url.searchParams.set('action', params.action);
|
|
}
|
|
|
|
if (isDefined(workspacePublicData)) {
|
|
url.searchParams.set('workspaceId', workspacePublicData.id);
|
|
}
|
|
|
|
return url.toString();
|
|
},
|
|
[workspacePublicData],
|
|
);
|
|
|
|
const handleGoogleLogin = useCallback(
|
|
(params: {
|
|
workspacePersonalInviteToken?: string;
|
|
workspaceInviteHash?: string;
|
|
billingCheckoutSession?: BillingCheckoutSession;
|
|
action: string;
|
|
}) => {
|
|
redirect(buildRedirectUrl('/auth/google', params));
|
|
},
|
|
[buildRedirectUrl, redirect],
|
|
);
|
|
|
|
const handleMicrosoftLogin = useCallback(
|
|
(params: {
|
|
workspacePersonalInviteToken?: string;
|
|
workspaceInviteHash?: string;
|
|
billingCheckoutSession?: BillingCheckoutSession;
|
|
action: string;
|
|
}) => {
|
|
redirect(buildRedirectUrl('/auth/microsoft', params));
|
|
},
|
|
[buildRedirectUrl, redirect],
|
|
);
|
|
|
|
const handleGetAuthTokensFromOTP = useCallback(
|
|
async (otp: string, loginToken: string, captchaToken?: string) => {
|
|
const getAuthTokensFromOtpResult = await getAuthTokensFromOtp({
|
|
variables: {
|
|
captchaToken,
|
|
origin,
|
|
otp,
|
|
loginToken,
|
|
},
|
|
});
|
|
|
|
if (isDefined(getAuthTokensFromOtpResult.errors)) {
|
|
throw getAuthTokensFromOtpResult.errors;
|
|
}
|
|
|
|
if (!getAuthTokensFromOtpResult.data?.getAuthTokensFromOTP) {
|
|
throw new Error('No getAuthTokensFromOTP result');
|
|
}
|
|
|
|
await handleLoadWorkspaceAfterAuthentication(
|
|
getAuthTokensFromOtpResult.data.getAuthTokensFromOTP.tokens,
|
|
);
|
|
},
|
|
[getAuthTokensFromOtp, origin, handleLoadWorkspaceAfterAuthentication],
|
|
);
|
|
|
|
return {
|
|
getLoginTokenFromCredentials: handleGetLoginTokenFromCredentials,
|
|
verifyEmailAndGetWorkspaceAgnosticToken:
|
|
handleverifyEmailAndGetWorkspaceAgnosticToken,
|
|
verifyEmailAndGetLoginToken: handleverifyEmailAndGetLoginToken,
|
|
getAuthTokensFromLoginToken: handleGetAuthTokensFromLoginToken,
|
|
checkUserExists: { checkUserExistsData, checkUserExistsQuery },
|
|
clearSession,
|
|
signOut: handleSignOut,
|
|
signUpWithCredentials: handleCredentialsSignUp,
|
|
signUpWithCredentialsInWorkspace: handleCredentialsSignUpInWorkspace,
|
|
signInWithCredentialsInWorkspace: handleCredentialsSignInInWorkspace,
|
|
signInWithCredentials: handleCredentialsSignIn,
|
|
signInWithGoogle: handleGoogleLogin,
|
|
signInWithMicrosoft: handleMicrosoftLogin,
|
|
setAuthTokens: handleSetAuthTokens,
|
|
getAuthTokensFromOTP: handleGetAuthTokensFromOTP,
|
|
};
|
|
};
|