applications/drive/src/app/utils/test/crypto.ts

   1 import type { PrivateKeyReference, SessionKey } from '@proton/crypto';
   2 import { CryptoProxy } from '@proton/crypto';
   3 import type { Address, Key } from '@proton/shared/lib/interfaces';
   4
   5 export async function generatePrivateKey(name = 'name', email = 'name@example.com'): Promise<PrivateKeyReference> {
   6     const { privateKeys } = await generateKeys(name, email);
   7     if (privateKeys.length !== 1) {
   8         throw new Error('Private key was not generated');
   9     }
  10     return privateKeys[0];
  11 }
  12
  13 export async function generateKeys(name = 'name', email = 'name@example.com') {
  14     const privateKey = await CryptoProxy.generateKey({
  15         userIDs: [{ name, email }],
  16     });
  17     const privateKeyArmored = await CryptoProxy.exportPrivateKey({ privateKey: privateKey, passphrase: null });
  18     const publicKey = await CryptoProxy.importPublicKey({ armoredKey: privateKeyArmored });
  19     const publicKeyArmored = await CryptoProxy.exportPublicKey({ key: publicKey });
  20
  21     return {
  22         name,
  23         email,
  24         publicKeyArmored,
  25         privateKeyArmored,
  26         publicKeys: [publicKey],
  27         privateKeys: [privateKey],
  28     };
  29 }
  30
  31 type SessionKeyAlgorithm = Parameters<typeof CryptoProxy.generateSessionKeyForAlgorithm>[0];
  32 export async function generateSessionKey(algorithm: SessionKeyAlgorithm = 'aes256'): Promise<SessionKey> {
  33     return {
  34         data: await CryptoProxy.generateSessionKeyForAlgorithm(algorithm),
  35         algorithm,
  36     };
  37 }
  38
  39 export const generateAddress = async (keys: Key[], email = 'test@pm.me'): Promise<Address> => {
  40     return {
  41         Email: email,
  42         Keys: keys,
  43     } as Address;
  44 };
  45
  46 /**
  47  * Load Crypto API outside of web workers, for testing purposes.
  48  */
  49 export async function setupCryptoProxyForTesting() {
  50     const useV6Canary = Math.random() < 0.5;
  51     // dynamic import to avoid loading the library unless required
  52     const { Api: CryptoApi } = useV6Canary
  53         ? await import('@proton/crypto/lib/worker/api_v6_canary')
  54         : await import('@proton/crypto/lib/worker/api');
  55     CryptoApi.init();
  56     CryptoProxy.setEndpoint(
  57         // @ts-ignore the v6 canary is effectively compatible
  58         new CryptoApi(),
  59         (endpoint) => endpoint.clearKeyStore()
  60     );
  61 }
  62
  63 export function releaseCryptoProxy() {
  64     return CryptoProxy.releaseEndpoint();
  65 }