Refactor pushStatusHandler to use Parse instead of direct access (#4173)

* Refactors pushStatusHandler to use HTTP interface so we can bind CloudCode hooks

* Handle correctly nested dot atomic operations

* Better handling of restricted class names, add support for afterSave _PushStatus

* Adds simple testing for afterSave(PushStatus)

* Reverts jobStatusHandler

* Addresses fixes

* adds delays to all methods

src/Controllers/DatabaseController.js

@@ -308,6 +308,19 @@ DatabaseController.prototype.update = function(className, query, update, {
     });
 };
 
+function expandResultOnKeyPath(object, key, value) {
+  if (key.indexOf('.') < 0) {
+    object[key] = value[key];
+    return object;
+  }
+  const path = key.split('.');
+  const firstKey = path[0];
+  const nextPath = path.slice(1).join('.');
+  object[firstKey] = expandResultOnKeyPath(object[firstKey] || {}, nextPath, value[firstKey]);
+  delete object[key];
+  return object;
+}
+
 function sanitizeDatabaseResult(originalObject, result) {
   const response = {};
   if (!result) {
@@ -319,7 +332,8 @@ function sanitizeDatabaseResult(originalObject, result) {
     if (keyUpdate && typeof keyUpdate === 'object' && keyUpdate.__op
       && ['Add', 'AddUnique', 'Remove', 'Increment'].indexOf(keyUpdate.__op) > -1) {
       // only valid ops that produce an actionable result
-      response[key] = result[key];
+      // the op may have happend on a keypath
+      expandResultOnKeyPath(response, key, result);
     }
   });
   return Promise.resolve(response);

src/Push/PushWorker.js

@@ -52,6 +52,7 @@ export class PushWorker {
     const auth = master(config);
     const where = utils.applyDeviceTokenExists(query.where);
     delete query.where;
+    pushStatus = pushStatusHandler(config, pushStatus.objectId);
     return rest.find(config, auth, '_Installation', where, query).then(({results}) => {
       if (results.length == 0) {
         return;
@@ -63,7 +64,6 @@ export class PushWorker {
   }
 
   sendToAdapter(body: any, installations: any[], pushStatus: any, config: Config, UTCOffset: ?any): Promise<*> {
-    pushStatus = pushStatusHandler(config, pushStatus.objectId);
     // Check if we have locales in the push body
     const locales = utils.getLocalesFromPush(body);
     if (locales.length > 0) {

src/StatusHandler.js

@@ -1,5 +1,6 @@
 import { md5Hash, newObjectId } from './cryptoUtils';
 import { logger }               from './logger';
+import Parse                    from 'parse/node';
 
 const PUSH_STATUS_COLLECTION = '_PushStatus';
 const JOB_STATUS_COLLECTION = '_JobStatus';
@@ -50,6 +51,47 @@ function statusHandler(className, database) {
   })
 }
 
+function restStatusHandler(className) {
+  let lastPromise = Promise.resolve();
+
+  function create(object) {
+    lastPromise = lastPromise.then(() => {
+      return Parse._request(
+        'POST',
+        `classes/${className}`,
+        object,
+        { useMasterKey: true }
+      ).then((result) => {
+        // merge the objects
+        const response = Object.assign({}, object, result);
+        return Promise.resolve(response);
+      });
+    });
+    return lastPromise;
+  }
+
+  function update(where, object) {
+    // TODO: when we have updateWhere, use that for proper interfacing
+    lastPromise = lastPromise.then(() => {
+      return Parse._request(
+        'PUT',
+        `classes/${className}/${where.objectId}`,
+        object,
+        { useMasterKey: true }).then((result) => {
+        // merge the objects
+        const response = Object.assign({}, object, result);
+        return Promise.resolve(response);
+      });
+    });
+    return lastPromise;
+  }
+
+  return Object.freeze({
+    create,
+    update
+  })
+}
+
 export function jobStatusHandler(config) {
   let jobStatus;
   const objectId = newObjectId(config.objectIdSize);
@@ -103,11 +145,12 @@ export function jobStatusHandler(config) {
   });
 }
 
-export function pushStatusHandler(config, objectId = newObjectId(config.objectIdSize)) {
+export function pushStatusHandler(config, existingObjectId) {
 
   let pushStatus;
   const database = config.database;
-  const handler = statusHandler(PUSH_STATUS_COLLECTION, database);
+  const handler = restStatusHandler(PUSH_STATUS_COLLECTION);
+  let objectId = existingObjectId;
   const setInitial = function(body = {}, where, options = {source: 'rest'}) {
     const now = new Date();
     let pushTime = now.toISOString();
@@ -133,8 +176,6 @@ export function pushStatusHandler(config, objectId = newObjectId(config.objectId
       pushHash = 'd41d8cd98f00b204e9800998ecf8427e';
     }
     const object = {
-      objectId,
-      createdAt: now,
       pushTime,
       query: JSON.stringify(where),
       payload: payloadString,
@@ -148,7 +189,8 @@ export function pushStatusHandler(config, objectId = newObjectId(config.objectId
       ACL: {}
     }
 
-    return handler.create(object).then(() => {
+    return handler.create(object).then((result) => {
+      objectId = result.objectId;
       pushStatus = {
         objectId
       };
@@ -159,12 +201,11 @@ export function pushStatusHandler(config, objectId = newObjectId(config.objectId
   const setRunning = function(count) {
     logger.verbose(`_PushStatus ${objectId}: sending push to %d installations`, count);
     return handler.update({status:"pending", objectId: objectId},
-      {status: "running", updatedAt: new Date(), count });
+      {status: "running", count });
   }
 
   const trackSent = function(results, UTCOffset, cleanupInstallations = process.env.PARSE_SERVER_CLEANUP_INVALID_INSTALLATIONS) {
     const update = {
-      updatedAt: new Date(),
       numSent: 0,
       numFailed: 0
     };
@@ -236,26 +277,33 @@ export function pushStatusHandler(config, objectId = newObjectId(config.objectId
   const complete = function() {
     return handler.update({ objectId }, {
       status: 'succeeded',
-      count: {__op: 'Delete'},
-      updatedAt: new Date()
+      count: {__op: 'Delete'}
     });
   }
 
   const fail = function(err) {
+    if (typeof err === 'string') {
+      err = { message: err };
+    }
     const update = {
-      errorMessage: JSON.stringify(err),
-      status: 'failed',
-      updatedAt: new Date()
+      errorMessage: err,
+      status: 'failed'
     }
     return handler.update({ objectId }, update);
   }
 
-  return Object.freeze({
-    objectId,
+  const rval = {
     setInitial,
     setRunning,
     trackSent,
     complete,
     fail
-  })
+  };
+
+  // define objectId to be dynamic
+  Object.defineProperty(rval, "objectId", {
+    get: () => objectId
+  });
+
+  return Object.freeze(rval);
 }

src/cloud-code/Parse.Cloud.js

@@ -1,19 +1,11 @@
 import { Parse }     from 'parse/node';
 import * as triggers from '../triggers';
 
-function validateClassNameForTriggers(className) {
-  const restrictedClassNames = [ '_Session' ];
-  if (restrictedClassNames.indexOf(className) != -1) {
-    throw `Triggers are not supported for ${className} class.`;
-  }
-  return className;
-}
-
 function getClassName(parseClass) {
   if (parseClass && parseClass.className) {
-    return validateClassNameForTriggers(parseClass.className);
+    return parseClass.className;
   }
-  return validateClassNameForTriggers(parseClass);
+  return parseClass;
 }
 
 var ParseCloud = {};

src/triggers.js

@@ -30,6 +30,20 @@ const baseStore = function() {
   });
 };
 
+function validateClassNameForTriggers(className, type) {
+  const restrictedClassNames = [ '_Session' ];
+  if (restrictedClassNames.indexOf(className) != -1) {
+    throw `Triggers are not supported for ${className} class.`;
+  }
+  if (type == Types.beforeSave && className === '_PushStatus') {
+    // _PushStatus uses undocumented nested key increment ops
+    // allowing beforeSave would mess up the objects big time
+    // TODO: Allow proper documented way of using nested increment ops
+    throw 'Only afterSave is allowed on _PushStatus';
+  }
+  return className;
+}
+
 const _triggerStore = {};
 
 export function addFunction(functionName, handler, validationHandler, applicationId) {
@@ -46,6 +60,7 @@ export function addJob(jobName, handler, applicationId) {
 }
 
 export function addTrigger(type, className, handler, applicationId) {
+  validateClassNameForTriggers(className, type);
   applicationId = applicationId || Parse.applicationId;
   _triggerStore[applicationId] =  _triggerStore[applicationId] || baseStore();
   _triggerStore[applicationId].Triggers[type][className] = handler;