Add onRemove function

mchapman · mchapman · commit c636a64ef9d7 · 2017-11-29T03:11:04.000Z
diff --git a/server/data_form.js b/server/data_form.js
@@ -1063,17 +1063,31 @@ DataForm.prototype.entityPut = function () {
 };
 DataForm.prototype.entityDelete = function () {
     return _.bind(function (req, res, next) {
-        this.processEntity(req, res, function () {
-            if (!req.resource) {
-                next();
-                return;
-            }
-            req.doc.remove(function (err) {
+        function internalRemove(doc) {
+            doc.remove(function (err) {
                 if (err) {
                     return res.send({ success: false });
                 }
                 return res.send({ success: true });
             });
+        }
+        this.processEntity(req, res, function () {
+            if (!req.resource) {
+                next();
+                return;
+            }
+            var doc = req.doc;
+            if (typeof req.resource.options.onRemove === 'function') {
+                req.resource.options.onRemove(doc, req, function (err) {
+                    if (err) {
+                        throw err;
+                    }
+                    internalRemove(doc);
+                });
+            }
+            else {
+                internalRemove(doc);
+            }
         });
     }, this);
 };
diff --git a/server/data_form.ts b/server/data_form.ts
@@ -1180,18 +1180,34 @@ DataForm.prototype.entityPut = function () {
 DataForm.prototype.entityDelete = function () {
   return _.bind(function (req, res, next) {
 
+    function internalRemove(doc) {
+      doc.remove(function (err) {
+        if (err) {
+          return res.send({success: false});
+        }
+        return res.send({success: true});
+      });
+    }
+
     this.processEntity(req, res, function () {
       if (!req.resource) {
         next();
         return;
       }
 
-      req.doc.remove(function (err) {
-        if (err) {
-          return res.send({success: false});
-        }
-        return res.send({success: true});
-      });
+      let doc = req.doc;
+      if (typeof req.resource.options.onRemove === 'function') {
+
+        req.resource.options.onRemove(doc, req, function (err) {
+          if (err) {
+            throw err;
+          }
+          internalRemove(doc);
+        });
+      } else {
+        internalRemove(doc);
+      }
+
     });
   }, this);
 };
