initial commit for show badge issue

src/controllers/notes.js

@@ -3,6 +3,7 @@ let {ipcMain} = require('electron');
 
 const Note = require('../models/note');
 const NotesList = require('../models/notesList');
+const Visits = require('../models/visits');
 
 /**
  * Time helper
@@ -22,6 +23,8 @@ class NotesController {
    * Setup event handlers
    */
   constructor() {
+    this.visits = new Visits();
+
     ipcMain.on('note - save', (event, {note}) => {
       this.saveNote(note, event);
     });
@@ -32,6 +35,9 @@ class NotesController {
 
     ipcMain.on('note - get', (event, {id}) => {
       this.getNote(id, event);
+
+      // set note as visited
+      this.visits.touch(id);
     });
 
     ipcMain.on('note - delete', (event, {id}) => {
@@ -96,6 +102,18 @@ class NotesController {
       let list = new NotesList(folderId);
       let notesInFolder = await list.get();
 
+      await this.visits.syncVisits();
+
+      /**
+       * set up note visits
+       */
+      notesInFolder.forEach( async (note) => {
+          if ( this.visits.visitedNotes[note._id] >= note.dtModify ) {
+            note.visited = true;
+          }
+      });
+
+      console.log(notesInFolder);
       let returnValue = {
         notes: notesInFolder,
         isRootFolder: !folderId

src/models/note.js

@@ -53,6 +53,7 @@ class Note {
     this.folderId = null;
     this.isRemoved = false;
     this.editorVersion = null;
+    this.visited = false;
 
     this.data = noteData;
   }

src/models/visits.js

@@ -0,0 +1,50 @@
+/**
+ * Created by khaydarovm on 27.02.2018.
+ */
+'use strict';
+const db = require('../utils/database'),
+    Time = require('../utils/time');
+
+class Visits {
+
+    constructor() {
+        this.visitedNotes = {};
+    }
+
+    /**
+     * sync visits
+     * @return {Promise.<*>}
+     */
+    async syncVisits() {
+        let allNotes = await db.find(db.VISITS, {});
+        allNotes.forEach( (note) => {
+            this.visitedNotes[note.noteId] = note.lastSeen;
+        });
+    }
+
+    async getNoteVisit(noteId) {
+        return await db.findOne(db.VISITS, {
+            noteId : noteId
+        });
+    }
+
+    async touch(noteId) {
+
+        let query = {
+            noteId : noteId
+        };
+
+        let data = {
+            noteId : noteId,
+            lastSeen : Time.now
+        };
+
+        await db.update(db.VISITS, query, {
+            $set : data
+        },{
+            upsert : true
+        });
+    }
+}
+
+module.exports = Visits;

src/utils/database.js

@@ -17,6 +17,7 @@ class Database {
     this.FOLDERS = new Datastore({ filename: path.join(this.appFolder, 'folders.db'), autoload: true});
     this.NOTES = new Datastore({ filename: path.join(this.appFolder, 'notes.db'), autoload: true});
     this.COLLABORATORS = new Datastore({ filename: path.join(this.appFolder, 'collaborators.db'), autoload: true });
+    this.VISITS = new Datastore({ filename: path.join(this.appFolder, 'visits.db'), autoload: true });
   }
 
   /**
@@ -96,13 +97,13 @@ class Database {
    * @param {Boolean} force - force drop.
    */
   drop(force = false) {
-    if (process.env.DEBUG !== 'true' || !force) {
+    if ( !force || ( process.env.DEBUG !== 'true' && !force )) {
       throw Error('Datastore dropping is not allowed for current environment');
     }
 
     let sequence = [];
 
-    [this.USER, this.FOLDERS, this.NOTES, this.COLLABORATORS].forEach( collection => {
+    [this.USER, this.FOLDERS, this.NOTES, this.COLLABORATORS, this.VISITS].forEach( collection => {
       console.log('\n\n Drop ', collection.filename, '\n\n');
       sequence.push(this.remove(collection, {}, {multi: true}, (removedRows) => {
         console.log(collection.filename, ': ', removedRows, ' docs removed');
@@ -113,8 +114,12 @@ class Database {
     this.FOLDERS = null;
     this.NOTES = null;
     this.COLLABORATORS = null;
+    this.VISITS = null;
 
-    return Promise.all(sequence);
+    return Promise.all(sequence)
+        .catch((err) => {
+            throw Error('Couldn\'t drop the database', err);
+        });
   }
 
   /**