Differential D7872 Diff 26669 keyserver/src/push/crypto.js

Changeset View

Standalone View

keyserver/src/push/crypto.js

// @flow		// @flow

import apn from '@parse/node-apn';		import apn from '@parse/node-apn';
import _cloneDeep from 'lodash/fp/cloneDeep.js';		import _cloneDeep from 'lodash/fp/cloneDeep.js';

		import type { AndroidNotification } from './types.js';
import { encryptAndUpdateOlmSession } from '../updaters/olm-session-updater.js';		import { encryptAndUpdateOlmSession } from '../updaters/olm-session-updater.js';

async function encryptIOSNotification(		async function encryptIOSNotification(
cookieID: string,		cookieID: string,
notification: apn.Notification,		notification: apn.Notification,
): Promise<apn.Notification> {		): Promise<apn.Notification> {
const encryptedNotification = _cloneDeep(notification);		const encryptedNotification = _cloneDeep(notification);
const { id, ...payloadSansId } = notification.payload;		const { id, ...payloadSansId } = notification.payload;
▲ Show 20 Lines • Show All 42 Lines • ▼ Show 20 Lines	for (const payloadField in restPayload) {
encryptedNotification.payload[payloadField] =		encryptedNotification.payload[payloadField] =
restPayload[payloadField].body;		restPayload[payloadField].body;
}		}

encryptedNotification.payload.encrypted = 1;		encryptedNotification.payload.encrypted = 1;
return notification;		return notification;
}		}

		async function encryptAndroidNotification(
		cookieID: string,
		notification: AndroidNotification,
		): Promise<AndroidNotification> {
		const encryptedNotification = { ...notification };
		ashoatUnsubmitted Not Done Inline Actions Can we construct a new one instead of copying the input? ashoat: Can we construct a new one instead of copying the input?
		const {
		data: { id, badgeOnly, ...notificationFieldsToEncrypt },
		} = encryptedNotification;

		let encryptedFields;
		try {
		encryptedFields = await encryptAndUpdateOlmSession(
		cookieID,
		'notifications',
		notificationFieldsToEncrypt,
		);
		} catch (e) {
		console.log('Notification encryption failed: ' + e);
		encryptedNotification.data.encrypted = '0';
		return encryptedNotification;
		}

		for (const fieldName in encryptedFields) {
		encryptedNotification.data[fieldName] = encryptedFields[fieldName].body;
		}
		encryptedNotification.data.encrypted = '1';
		return encryptedNotification;
		}

function prepareEncryptedIOSNotifications(		function prepareEncryptedIOSNotifications(
cookieIDs: $ReadOnlyArray<string>,		cookieIDs: $ReadOnlyArray<string>,
notification: apn.Notification,		notification: apn.Notification,
): Promise<Array<apn.Notification>> {		): Promise<Array<apn.Notification>> {
const notificationPromises = cookieIDs.map(cookieID =>		const notificationPromises = cookieIDs.map(cookieID =>
encryptIOSNotification(cookieID, notification),		encryptIOSNotification(cookieID, notification),
);		);
return Promise.all(notificationPromises);		return Promise.all(notificationPromises);
}		}

export { prepareEncryptedIOSNotifications };		function prepareEncryptedAndroidNotifications(
		cookieIDs: $ReadOnlyArray<string>,
		notification: AndroidNotification,
		): Promise<Array<AndroidNotification>> {
		const notificationPromises = cookieIDs.map(cookieID =>
		encryptAndroidNotification(cookieID, notification),
		);
		return Promise.all(notificationPromises);
		}

		export {
		prepareEncryptedIOSNotifications,
		prepareEncryptedAndroidNotifications,
		};