2020-09-12 21:53:41 +00:00
|
|
|
import React, { useState, useEffect, useContext } from 'react';
|
|
|
|
import Container from 'components/Container';
|
|
|
|
import styled from 'styled-components';
|
|
|
|
import Card from 'react-bootstrap/Card';
|
|
|
|
import Form from 'react-bootstrap/Form';
|
|
|
|
import constants from 'utils/strings/constants';
|
|
|
|
import { Formik, FormikHelpers } from 'formik';
|
|
|
|
import * as Yup from 'yup';
|
|
|
|
import Button from 'react-bootstrap/Button';
|
2020-11-21 12:18:03 +00:00
|
|
|
import { putAttributes } from 'services/userService';
|
2020-09-13 06:30:07 +00:00
|
|
|
import { getData, LS_KEYS, setData } from 'utils/storage/localStorage';
|
2020-09-12 21:53:41 +00:00
|
|
|
import { useRouter } from 'next/router';
|
2020-09-13 06:30:07 +00:00
|
|
|
import { getKey, SESSION_KEYS, setKey } from 'utils/storage/sessionStorage';
|
2021-02-09 09:33:54 +00:00
|
|
|
import * as Comlink from 'comlink';
|
2021-02-16 11:43:21 +00:00
|
|
|
import { B64EncryptionResult } from 'services/uploadService';
|
2020-10-01 01:29:24 +00:00
|
|
|
|
2021-02-09 09:33:54 +00:00
|
|
|
const CryptoWorker: any =
|
|
|
|
typeof window !== 'undefined' &&
|
|
|
|
Comlink.wrap(new Worker('worker/crypto.worker.js', { type: 'module' }));
|
2020-09-12 21:53:41 +00:00
|
|
|
|
|
|
|
const Image = styled.img`
|
|
|
|
width: 200px;
|
|
|
|
margin-bottom: 20px;
|
|
|
|
max-width: 100%;
|
|
|
|
`;
|
|
|
|
|
|
|
|
interface formValues {
|
|
|
|
passphrase: string;
|
|
|
|
confirm: string;
|
|
|
|
}
|
|
|
|
|
|
|
|
export default function Generate() {
|
|
|
|
const [loading, setLoading] = useState(false);
|
|
|
|
const [token, setToken] = useState<string>();
|
|
|
|
const router = useRouter();
|
2020-09-13 06:30:07 +00:00
|
|
|
const key = getKey(SESSION_KEYS.ENCRYPTION_KEY);
|
2020-10-01 01:29:24 +00:00
|
|
|
|
2020-09-12 21:53:41 +00:00
|
|
|
useEffect(() => {
|
2020-09-14 09:32:01 +00:00
|
|
|
router.prefetch('/gallery');
|
2020-09-13 06:30:07 +00:00
|
|
|
const user = getData(LS_KEYS.USER);
|
2020-09-12 21:53:41 +00:00
|
|
|
if (!user?.token) {
|
2021-02-09 09:33:54 +00:00
|
|
|
router.push('/');
|
2020-09-13 06:30:07 +00:00
|
|
|
} else if (key) {
|
2020-09-12 21:53:41 +00:00
|
|
|
router.push('/gallery');
|
|
|
|
} else {
|
|
|
|
setToken(user.token);
|
|
|
|
}
|
|
|
|
}, []);
|
|
|
|
|
2021-02-09 09:33:54 +00:00
|
|
|
const onSubmit = async (
|
|
|
|
values: formValues,
|
|
|
|
{ setFieldError }: FormikHelpers<formValues>
|
|
|
|
) => {
|
2020-09-12 21:53:41 +00:00
|
|
|
setLoading(true);
|
|
|
|
try {
|
|
|
|
const { passphrase, confirm } = values;
|
|
|
|
if (passphrase === confirm) {
|
2020-10-01 01:29:24 +00:00
|
|
|
const cryptoWorker = await new CryptoWorker();
|
2021-01-18 13:08:14 +00:00
|
|
|
const key: string = await cryptoWorker.generateMasterKey();
|
|
|
|
const kekSalt: string = await cryptoWorker.generateSaltToDeriveKey();
|
2021-02-09 09:33:54 +00:00
|
|
|
const kek: string = await cryptoWorker.deriveKey(
|
|
|
|
passphrase,
|
|
|
|
kekSalt
|
|
|
|
);
|
2021-01-18 13:08:14 +00:00
|
|
|
const kekHash: string = await cryptoWorker.hash(kek);
|
2021-02-16 11:43:21 +00:00
|
|
|
const encryptedKeyAttributes: B64EncryptionResult = await cryptoWorker.encryptToB64(
|
2021-02-09 09:33:54 +00:00
|
|
|
key,
|
|
|
|
kek
|
|
|
|
);
|
2020-11-07 10:27:59 +00:00
|
|
|
const keyPair = await cryptoWorker.generateKeyPair();
|
2021-02-16 11:43:21 +00:00
|
|
|
const encryptedKeyPairAttributes: B64EncryptionResult = await cryptoWorker.encryptToB64(
|
2021-02-09 09:33:54 +00:00
|
|
|
keyPair.privateKey,
|
|
|
|
key
|
|
|
|
);
|
2021-01-18 13:08:14 +00:00
|
|
|
|
2020-09-13 06:30:07 +00:00
|
|
|
const keyAttributes = {
|
2021-01-18 13:08:14 +00:00
|
|
|
kekSalt,
|
2020-11-07 10:27:59 +00:00
|
|
|
kekHash: kekHash,
|
2021-01-18 13:08:14 +00:00
|
|
|
encryptedKey: encryptedKeyAttributes.encryptedData,
|
|
|
|
keyDecryptionNonce: encryptedKeyAttributes.nonce,
|
|
|
|
publicKey: keyPair.publicKey,
|
2021-02-09 09:33:54 +00:00
|
|
|
encryptedSecretKey:
|
|
|
|
encryptedKeyPairAttributes.encryptedData,
|
|
|
|
secretKeyDecryptionNonce: encryptedKeyPairAttributes.nonce,
|
2020-09-13 06:30:07 +00:00
|
|
|
};
|
2021-02-09 09:33:54 +00:00
|
|
|
await putAttributes(
|
|
|
|
token,
|
|
|
|
getData(LS_KEYS.USER).name,
|
|
|
|
keyAttributes
|
|
|
|
);
|
2020-09-13 06:30:07 +00:00
|
|
|
setData(LS_KEYS.KEY_ATTRIBUTES, keyAttributes);
|
2020-10-01 01:29:24 +00:00
|
|
|
|
2021-02-09 09:33:54 +00:00
|
|
|
const sessionKeyAttributes = await cryptoWorker.encryptToB64(
|
|
|
|
key
|
|
|
|
);
|
2021-01-18 13:08:14 +00:00
|
|
|
const sessionKey = sessionKeyAttributes.key;
|
|
|
|
const sessionNonce = sessionKeyAttributes.nonce;
|
|
|
|
const encryptionKey = sessionKeyAttributes.encryptedData;
|
2020-09-13 06:30:07 +00:00
|
|
|
setKey(SESSION_KEYS.ENCRYPTION_KEY, { encryptionKey });
|
2020-10-01 01:29:24 +00:00
|
|
|
setData(LS_KEYS.SESSION, { sessionKey, sessionNonce });
|
2020-09-12 21:53:41 +00:00
|
|
|
router.push('/gallery');
|
|
|
|
} else {
|
|
|
|
setFieldError('confirm', constants.PASSPHRASE_MATCH_ERROR);
|
|
|
|
}
|
|
|
|
} catch (e) {
|
2021-02-09 09:33:54 +00:00
|
|
|
setFieldError(
|
|
|
|
'passphrase',
|
|
|
|
`${constants.UNKNOWN_ERROR} ${e.message}`
|
|
|
|
);
|
2020-09-12 21:53:41 +00:00
|
|
|
}
|
|
|
|
setLoading(false);
|
2021-02-09 09:33:54 +00:00
|
|
|
};
|
2020-10-01 01:29:24 +00:00
|
|
|
|
2021-02-09 09:33:54 +00:00
|
|
|
return (
|
|
|
|
<Container>
|
|
|
|
<Image alt="vault" src="/vault.svg" />
|
|
|
|
<Card>
|
|
|
|
<Card.Body>
|
|
|
|
<div className="text-center">
|
|
|
|
<p>{constants.ENTER_ENC_PASSPHRASE}</p>
|
|
|
|
<p>{constants.PASSPHRASE_DISCLAIMER()}</p>
|
|
|
|
</div>
|
|
|
|
<Formik<formValues>
|
|
|
|
initialValues={{ passphrase: '', confirm: '' }}
|
|
|
|
validationSchema={Yup.object().shape({
|
|
|
|
passphrase: Yup.string().required(
|
|
|
|
constants.REQUIRED
|
|
|
|
),
|
|
|
|
confirm: Yup.string().required(constants.REQUIRED),
|
|
|
|
})}
|
|
|
|
onSubmit={onSubmit}
|
|
|
|
>
|
|
|
|
{({
|
|
|
|
values,
|
|
|
|
touched,
|
|
|
|
errors,
|
|
|
|
handleChange,
|
|
|
|
handleBlur,
|
|
|
|
handleSubmit,
|
|
|
|
}) => (
|
|
|
|
<Form noValidate onSubmit={handleSubmit}>
|
|
|
|
<Form.Group>
|
|
|
|
<Form.Control
|
|
|
|
type="text"
|
|
|
|
placeholder={constants.PASSPHRASE_HINT}
|
|
|
|
value={values.passphrase}
|
|
|
|
onChange={handleChange('passphrase')}
|
|
|
|
onBlur={handleBlur('passphrase')}
|
|
|
|
isInvalid={Boolean(
|
|
|
|
touched.passphrase &&
|
|
|
|
errors.passphrase
|
|
|
|
)}
|
|
|
|
disabled={loading}
|
|
|
|
/>
|
|
|
|
<Form.Control.Feedback type="invalid">
|
|
|
|
{errors.passphrase}
|
|
|
|
</Form.Control.Feedback>
|
|
|
|
</Form.Group>
|
|
|
|
<Form.Group>
|
|
|
|
<Form.Control
|
|
|
|
type="text"
|
|
|
|
placeholder={
|
|
|
|
constants.PASSPHRASE_CONFIRM
|
|
|
|
}
|
|
|
|
value={values.confirm}
|
|
|
|
onChange={handleChange('confirm')}
|
|
|
|
onBlur={handleBlur('confirm')}
|
|
|
|
isInvalid={Boolean(
|
|
|
|
touched.confirm && errors.confirm
|
|
|
|
)}
|
|
|
|
disabled={loading}
|
|
|
|
/>
|
|
|
|
<Form.Control.Feedback type="invalid">
|
|
|
|
{errors.confirm}
|
|
|
|
</Form.Control.Feedback>
|
|
|
|
</Form.Group>
|
|
|
|
<Button type="submit" block disabled={loading}>
|
|
|
|
{constants.SET_PASSPHRASE}
|
|
|
|
</Button>
|
|
|
|
</Form>
|
|
|
|
)}
|
|
|
|
</Formik>
|
|
|
|
</Card.Body>
|
|
|
|
</Card>
|
|
|
|
</Container>
|
|
|
|
);
|
2020-09-12 21:53:41 +00:00
|
|
|
}
|