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';
|
|
|
|
import { putKeyAttributes } 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';
|
2020-10-01 01:29:24 +00:00
|
|
|
import * as Comlink from "comlink";
|
|
|
|
|
2020-10-03 14:21:56 +00:00
|
|
|
const CryptoWorker: any = typeof window !== 'undefined'
|
2020-10-01 01:29:24 +00:00
|
|
|
&& 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) {
|
|
|
|
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);
|
|
|
|
}
|
|
|
|
}, []);
|
|
|
|
|
|
|
|
const onSubmit = async (values: formValues, { setFieldError }: FormikHelpers<formValues>) => {
|
|
|
|
setLoading(true);
|
|
|
|
try {
|
|
|
|
const { passphrase, confirm } = values;
|
|
|
|
if (passphrase === confirm) {
|
2020-10-01 01:29:24 +00:00
|
|
|
const cryptoWorker = await new CryptoWorker();
|
2020-10-03 14:21:56 +00:00
|
|
|
const key = await cryptoWorker.generateMasterKey();
|
|
|
|
const kekSalt = await cryptoWorker.generateSaltToDeriveKey();
|
|
|
|
const kek = await cryptoWorker.deriveKey(
|
|
|
|
await cryptoWorker.fromString(passphrase), kekSalt);
|
2020-10-01 01:29:24 +00:00
|
|
|
const kekHash = await cryptoWorker.hash(kek);
|
|
|
|
const encryptedKeyAttributes = await cryptoWorker.encrypt(key, kek);
|
2020-11-07 10:27:59 +00:00
|
|
|
const keyPair = await cryptoWorker.generateKeyPair();
|
|
|
|
const encryptedKeyPairAttributes = await cryptoWorker.encrypt(keyPair.privateKey, kek);
|
2020-09-13 06:30:07 +00:00
|
|
|
const keyAttributes = {
|
2020-10-03 14:21:56 +00:00
|
|
|
kekSalt: await cryptoWorker.toB64(kekSalt),
|
2020-11-07 10:27:59 +00:00
|
|
|
kekHash: kekHash,
|
2020-10-03 14:21:56 +00:00
|
|
|
encryptedKey: await cryptoWorker.toB64(encryptedKeyAttributes.encryptedData),
|
|
|
|
keyDecryptionNonce: await cryptoWorker.toB64(encryptedKeyAttributes.nonce),
|
2020-11-07 10:27:59 +00:00
|
|
|
publicKey: await cryptoWorker.toB64(keyPair.publicKey),
|
|
|
|
encryptedSecretKey: await cryptoWorker.toB64(encryptedKeyPairAttributes.encryptedData),
|
|
|
|
secretKeyDecryptionNonce: await cryptoWorker.toB64(encryptedKeyPairAttributes.nonce)
|
2020-09-13 06:30:07 +00:00
|
|
|
};
|
|
|
|
await putKeyAttributes(token, keyAttributes);
|
|
|
|
setData(LS_KEYS.KEY_ATTRIBUTES, keyAttributes);
|
2020-10-01 01:29:24 +00:00
|
|
|
|
|
|
|
const sessionKeyAttributes = await cryptoWorker.encrypt(key);
|
2020-10-03 14:21:56 +00:00
|
|
|
const sessionKey = await cryptoWorker.toB64(sessionKeyAttributes.key);
|
|
|
|
const sessionNonce = await cryptoWorker.toB64(sessionKeyAttributes.nonce);
|
|
|
|
const encryptionKey = await cryptoWorker.toB64(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) {
|
|
|
|
setFieldError('passphrase', `${constants.UNKNOWN_ERROR} ${e.message}`);
|
|
|
|
}
|
|
|
|
setLoading(false);
|
|
|
|
}
|
2020-10-01 01:29:24 +00:00
|
|
|
|
2020-09-12 21:53:41 +00:00
|
|
|
return (<Container>
|
2020-09-13 17:15:48 +00:00
|
|
|
<Image alt='vault' src='/vault.svg' />
|
2020-09-12 21:53:41 +00:00
|
|
|
<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>)
|
|
|
|
}
|