111 lines
3.0 KiB
JavaScript
111 lines
3.0 KiB
JavaScript
import { md5Hash, newObjectId } from './cryptoUtils';
|
||
import { logger } from './logger';
|
||
|
||
const PUSH_STATUS_COLLECTION = '_PushStatus';
|
||
|
||
export function flatten(array) {
|
||
return array.reduce((memo, element) => {
|
||
if (Array.isArray(element)) {
|
||
memo = memo.concat(flatten(element));
|
||
} else {
|
||
memo = memo.concat(element);
|
||
}
|
||
return memo;
|
||
}, []);
|
||
}
|
||
|
||
export default function pushStatusHandler(config) {
|
||
|
||
let initialPromise;
|
||
let pushStatus;
|
||
let objectId = newObjectId();
|
||
let database = config.database.WithoutValidation();
|
||
|
||
let setInitial = function(body = {}, where, options = {source: 'rest'}) {
|
||
let now = new Date();
|
||
let data = body.data || {};
|
||
let payloadString = JSON.stringify(data);
|
||
let object = {
|
||
objectId,
|
||
createdAt: now,
|
||
pushTime: now.toISOString(),
|
||
query: JSON.stringify(where),
|
||
payload: payloadString,
|
||
source: options.source,
|
||
title: options.title,
|
||
expiry: body.expiration_time,
|
||
status: "pending",
|
||
numSent: 0,
|
||
pushHash: md5Hash(payloadString),
|
||
// lockdown!
|
||
ACL: {}
|
||
}
|
||
|
||
return database.create(PUSH_STATUS_COLLECTION, object).then(() => {
|
||
pushStatus = {
|
||
objectId
|
||
};
|
||
return Promise.resolve(pushStatus);
|
||
});
|
||
}
|
||
|
||
let setRunning = function(installations) {
|
||
logger.verbose('sending push to %d installations', installations.length);
|
||
return database.update(PUSH_STATUS_COLLECTION,
|
||
{status:"pending", objectId: objectId},
|
||
{status: "running", updatedAt: new Date() });
|
||
}
|
||
|
||
let complete = function(results) {
|
||
let update = {
|
||
status: 'succeeded',
|
||
updatedAt: new Date(),
|
||
numSent: 0,
|
||
numFailed: 0,
|
||
};
|
||
if (Array.isArray(results)) {
|
||
results = flatten(results);
|
||
results.reduce((memo, result) => {
|
||
// Cannot handle that
|
||
if (!result || !result.device || !result.device.deviceType) {
|
||
return memo;
|
||
}
|
||
let deviceType = result.device.deviceType;
|
||
if (result.transmitted)
|
||
{
|
||
memo.numSent++;
|
||
memo.sentPerType = memo.sentPerType || {};
|
||
memo.sentPerType[deviceType] = memo.sentPerType[deviceType] || 0;
|
||
memo.sentPerType[deviceType]++;
|
||
} else {
|
||
memo.numFailed++;
|
||
memo.failedPerType = memo.failedPerType || {};
|
||
memo.failedPerType[deviceType] = memo.failedPerType[deviceType] || 0;
|
||
memo.failedPerType[deviceType]++;
|
||
}
|
||
return memo;
|
||
}, update);
|
||
}
|
||
logger.verbose('sent push! %d success, %d failures', update.numSent, update.numFailed);
|
||
return database.update(PUSH_STATUS_COLLECTION, {status:"running", objectId }, update);
|
||
}
|
||
|
||
let fail = function(err) {
|
||
let update = {
|
||
errorMessage: JSON.stringify(err),
|
||
status: 'failed',
|
||
updatedAt: new Date()
|
||
}
|
||
logger.info('warning: error while sending push', err);
|
||
return database.update(PUSH_STATUS_COLLECTION, { objectId }, update);
|
||
}
|
||
|
||
return Object.freeze({
|
||
objectId,
|
||
setInitial,
|
||
setRunning,
|
||
complete,
|
||
fail
|
||
})
|
||
}
|