Update to support delete note

3 files changed, 41 insertions, 0 deletions

diff --git a/lib/migrations/20161009040430-support-delete-note.js b/lib/migrations/20161009040430-support-delete-note.js
new file mode 100644
index 00000000..f478b6fe
--- /dev/null
+++ b/lib/migrations/20161009040430-support-delete-note.js
@@ -0,0 +1,11 @@
+'use strict';
+
+module.exports = {
+  up: function (queryInterface, Sequelize) {
+    queryInterface.addColumn('Notes', 'deletedAt', Sequelize.DATE);
+  },
+
+  down: function (queryInterface, Sequelize) {
+    queryInterface.removeColumn('Notes', 'deletedAt', Sequelize.DATE);
+  }
+};
diff --git a/lib/models/note.js b/lib/models/note.js
index 08ef083d..6efa5d4f 100644
--- a/lib/models/note.js
+++ b/lib/models/note.js
@@ -67,6 +67,7 @@ module.exports = function (sequelize, DataTypes) {
             type: DataTypes.DATE
         }
     }, {
+        paranoid: true,
         classMethods: {
             associate: function (models) {
                 Note.belongsTo(models.User, {
diff --git a/lib/realtime.js b/lib/realtime.js
index c6c9ffeb..0fcd2eaa 100644
--- a/lib/realtime.js
+++ b/lib/realtime.js
@@ -796,6 +796,35 @@ function connection(socket) {
         }
     });
 
+    // delete a note
+    socket.on('delete', function () {
+        //need login to do more actions
+        if (socket.request.user && socket.request.user.logged_in) {
+            var noteId = socket.noteId;
+            if (!noteId || !notes[noteId]) return;
+            var note = notes[noteId];
+            //Only owner can delete note
+            if (note.owner && note.owner == socket.request.user.id) {
+                models.Note.destroy({
+                    where: {
+                        id: noteId
+                    }
+                }).then(function (count) {
+                    if (!count) return;
+                    for (var i = 0, l = note.socks.length; i < l; i++) {
+                        var sock = note.socks[i];
+                        if (typeof sock !== 'undefined' && sock) {
+                            sock.emit('delete');
+                            return sock.disconnect(true);
+                        }
+                    }
+                }).catch(function (err) {
+                    return logger.error('delete note failed: ' + err);
+                });
+            }
+        }
+    });
+
     //reveiced when user logout or changed
     socket.on('user changed', function () {
         logger.info('user changed');