beenull/ente
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

update two-factor setup

Browse source
This commit is contained in:
Abhinav 2022-07-01 13:45:08 +05:30
parent d53da12763
commit 5e6c82c1ca
2 changed files with 14 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
src/pages/two-factor/setup/index.tsx
View file

@ -3,7 +3,9 @@ import { enableTwoFactor, setupTwoFactor } from 'services/userService';
import constants from 'utils/strings/constants';
import VerticallyCentered from 'components/Container';
import { useRouter } from 'next/router';
import VerifyTwoFactor from 'components/TwoFactor/VerifyForm';
import VerifyTwoFactor, {
VerifyTwoFactorCallback,
} from 'components/TwoFactor/VerifyForm';
import { encryptWithRecoveryKey } from 'utils/crypto';
import { setData, LS_KEYS, getData } from 'utils/storage/localStorage';
import { AppContext, FLASH_MESSAGE_TYPE } from 'pages/_app';
@ -13,6 +15,7 @@ import Card from '@mui/material/Card';
import { Box, CardContent, Typography } from '@mui/material';
import { TwoFactorSetup } from 'components/TwoFactor/Setup';
import LinkButton from 'components/pages/gallery/LinkButton';
import { logError } from 'utils/sentry';
export enum SetupMode {
QR_CODE,
@ -34,21 +37,21 @@ export default function SetupTwoFactor() {
const twoFactorSecret = await setupTwoFactor();
setTwoFactorSecret(twoFactorSecret);
} catch (e) {
appContext.setDisappearingFlashMessage({
message: constants.TWO_FACTOR_SETUP_FAILED,
type: FLASH_MESSAGE_TYPE.DANGER,
});
router.push(PAGES.GALLERY);
logError(e, 'failed to get two factor setup code');
}
};
main();
}, []);
const onSubmit = async (otp: string) => {
const onSubmit: VerifyTwoFactorCallback = async (
otp: string,
markSuccessful
) => {
const recoveryEncryptedTwoFactorSecret = await encryptWithRecoveryKey(
twoFactorSecret.secretCode
);
await enableTwoFactor(otp, recoveryEncryptedTwoFactorSecret);
await markSuccessful();
setData(LS_KEYS.USER, {
...getData(LS_KEYS.USER),
isTwoFactorEnabled: true,

6
src/pages/two-factor/verify/index.tsx
View file

@ -1,4 +1,6 @@
import VerifyTwoFactor from 'components/TwoFactor/VerifyForm';
import VerifyTwoFactor, {
VerifyTwoFactorCallback,
} from 'components/TwoFactor/VerifyForm';
import router from 'next/router';
import React, { useEffect, useState } from 'react';
import { logoutUser, verifyTwoFactor } from 'services/userService';
@ -33,7 +35,7 @@ export default function Home() {
main();
}, []);
const onSubmit = async (otp: string) => {
const onSubmit: VerifyTwoFactorCallback = async (otp) => {
try {
const resp = await verifyTwoFactor(otp, sessionID);
const { keyAttributes, encryptedToken, token, id } = resp;