From 021b4cd6107266c345670b4884729c684235ee26 Mon Sep 17 00:00:00 2001
From: Dave C-J <dave@conway-jones.co.uk>
Date: Tue, 24 Sep 2013 20:38:37 +0100
Subject: [PATCH] Added delete to MongoDB node

---
 nodes/storage/66-mongodb.html | 12 +++++++++++-
 nodes/storage/66-mongodb.js   | 29 ++++++++++++++++++++---------
 2 files changed, 31 insertions(+), 10 deletions(-)

diff --git a/nodes/storage/66-mongodb.html b/nodes/storage/66-mongodb.html
index 4737d2e7e..f4704e537 100644
--- a/nodes/storage/66-mongodb.html
+++ b/nodes/storage/66-mongodb.html
@@ -105,6 +105,13 @@
         <label for="node-input-collection"><i class="icon-briefcase"></i> Collection</label>
         <input type="text" id="node-input-collection" placeholder="collection">
     </div>
+    <div class="form-row">
+        <label for="node-input-operation"><i class="icon-wrench"></i> Operation</label>
+        <select type="text" id="node-input-operation" style="display: inline-block; width: auto; vertical-align: top;">
+            <option value=query>Query</option>
+            <option value=delete>Delete</option>
+        </select>
+    </div>
     <div class="form-row">
         <label for="node-input-name"><i class="icon-tag"></i> Name</label>
         <input type="text" id="node-input-name" placeholder="Name">
@@ -114,7 +121,9 @@
 <script type="text/x-red" data-help-name="mongodb in">
 	<p>Queries a MongoDB collection by using the <b>msg.payload</b> to be a MongoDB query statement as per the .find() function.</p>
 	<p>You may also (via a function) set a <b>msg.projection</b> object to constrain the returned fields, a <b>msg.sort</b> object and a <b>msg.limit</b> object.</p>
-	<p>All are optional - see the <a href="http://docs.mongodb.org/manual/reference/method/db.collection.find/" target="new"><i>MongoDB find docs</i></a> for examples.
+	<p>All are optional - see the <a href="http://docs.mongodb.org/manual/reference/method/db.collection.find/" target="new"><i>MongoDB find docs</i></a> for examples.</p>
+	<p>You can also choose to <b>remove</b> items. To do so the <b>msg.payload</b> <i>MUST</i> contain an object that will select the items(s) to remove.
+	A blank object will delete <i>all of the objects</i> in the collection. You have been warned...</p>
 </script>
 
 <script type="text/javascript">
@@ -125,6 +134,7 @@
             mongodb: { type:"mongodb",required:true},
             name: {value:""},
             collection: {value:"",required:true},
+            operation: {value:"query"}
         },
         inputs:1,
         outputs:1,
diff --git a/nodes/storage/66-mongodb.js b/nodes/storage/66-mongodb.js
index 93dc615f4..2173bf156 100644
--- a/nodes/storage/66-mongodb.js
+++ b/nodes/storage/66-mongodb.js
@@ -67,6 +67,7 @@ MongoOutNode.prototype.close = function() {
 function MongoInNode(n) {
     RED.nodes.createNode(this,n);
     this.collection = n.collection;
+    this.operation = n.operation;
     this.mongodb = n.mongodb;
     this.mongoConfig = RED.nodes.getNode(this.mongodb);
 
@@ -80,15 +81,25 @@ function MongoInNode(n) {
                     if (err) { node.error(err); }
                     else {
                         node.on("input",function(msg) {
-                            msg.projection = msg.projection || {};
-                            coll.find(msg.payload,msg.projection).sort(msg.sort).limit(msg.limit).toArray(function(err, items) {
-                                if (err) { node.error(err); }
-                                msg.payload = items;
-                                delete msg.projection;
-                                delete msg.sort;
-                                delete msg.limit;
-                                node.send(msg);
-                            });
+                            if (node.operation == "query") {
+                                msg.projection = msg.projection || {};
+                                coll.find(msg.payload,msg.projection).sort(msg.sort).limit(msg.limit).toArray(function(err, items) {
+                                    if (err) { node.error(err); }
+                                    msg.payload = items;
+                                    delete msg.projection;
+                                    delete msg.sort;
+                                    delete msg.limit;
+                                    node.send(msg);
+                                });
+                            }
+                            if (node.operation == "delete") {
+                                console.log(msg.payload);
+                                coll.remove(msg.payload, {w:1}, function(err, items) {
+                                    if (err) node.error(err);
+                                    msg.payload = items;
+                                    node.send(msg);
+                                });
+                            }
                         });
                     }
                 });