Migrate realtime.js to TypeScript

Signed-off-by: David Mehren <dmehren1@gmail.com>

lib/realtime.ts

@@ -1,6 +1,11 @@
 'use strict'
 // realtime
 // external modules
+import {User} from "./models/user";
+import {Author} from "./models/author";
+import {Note} from "./models/note";
+import {Revision} from "./models/revision";
+
 var cookie = require('cookie')
 var cookieParser = require('cookie-parser')
 var async = require('async')
@@ -13,13 +18,12 @@ var moment = require('moment')
 var config = require('./config')
 var logger = require('./logger')
 var history = require('./history')
-var models = require('./models')
 
 // ot
 var ot = require('./ot')
 
 // public
-var realtime = {
+var realtime: any = {
   io: null,
   onAuthorizeSuccess: onAuthorizeSuccess,
   onAuthorizeFail: onAuthorizeFail,
@@ -116,7 +120,7 @@ setInterval(function () {
 }, 1000)
 
 function updateNote(note, callback) {
-  models.Note.findOne({
+  Note.findOne({
     where: {
       id: note.id
     }
@@ -130,13 +134,13 @@ function updateNote (note, callback) {
     })
     if (note.lastchangeuser) {
       if (_note.lastchangeuserId !== note.lastchangeuser) {
-        models.User.findOne({
+        User.findOne({
           where: {
             id: note.lastchangeuser
           }
         }).then(function (user) {
           if (!user) return callback(null, null)
-          note.lastchangeuserprofile = models.User.getProfile(user)
+          note.lastchangeuserprofile = User.getProfile(user)
           return finishUpdateNote(note, _note, callback)
         }).catch(function (err) {
           logger.error(err)
@@ -158,7 +162,7 @@ function updateNote (note, callback) {
 function finishUpdateNote(note, _note, callback) {
   if (!note || !note.server) return callback(null, null)
   var body = note.server.document
-  var title = note.title = models.Note.parseNoteTitle(body)
+  var title = note.title = Note.parseNoteTitle(body)
   var values = {
     title: title,
     content: body,
@@ -200,7 +204,7 @@ var saverSleep = false
 // save note revision in interval
 setInterval(function () {
   if (saverSleep) return
-  models.Revision.saveAllNotesRevision(function (err, notes) {
+  Revision.saveAllNotesRevision(function (err, notes) {
     if (err) return logger.error('revision saver failed: ' + err)
     if (notes && notes.length <= 0) {
       saverSleep = true
@@ -209,10 +213,10 @@ setInterval(function () {
 }, 60000 * 5)
 
 function getStatus(callback) {
-  models.Note.count().then(function (notecount) {
+  Note.count().then(function (notecount) {
-    var distinctaddresses = []
+    var distinctaddresses: string[] = []
-    var regaddresses = []
+    var regaddresses: string[] = []
-    var distinctregaddresses = []
+    var distinctregaddresses: string[] = []
     Object.keys(users).forEach(function (key) {
       var user = users[key]
       if (!user) return
@@ -240,7 +244,7 @@ function getStatus (callback) {
         }
       }
     })
-    models.User.count().then(function (regcount) {
+    User.count().then(function (regcount) {
       // eslint-disable-next-line standard/no-callback-literal
       return callback ? callback({
         onlineNotes: Object.keys(notes).length,
@@ -286,7 +290,7 @@ function parseNoteIdFromSocket (socket, callback) {
   if (!noteId) {
     return callback(null, null)
   }
-  models.Note.parseNoteId(noteId, function (err, id) {
+  Note.parseNoteId(noteId, function (err, id) {
     if (err || !id) return callback(err, id)
     return callback(null, id)
   })
@@ -295,10 +299,12 @@ function parseNoteIdFromSocket (socket, callback) {
 function emitOnlineUsers(socket) {
   var noteId = socket.noteId
   if (!noteId || !notes[noteId]) return
-  var users = []
+  var users: any[] = []
   Object.keys(notes[noteId].users).forEach(function (key) {
     var user = notes[noteId].users[key]
-    if (user) { users.push(buildUserOutData(user)) }
+    if (user) {
+      users.push(buildUserOutData(user))
+    }
   })
   var out = {
     users: users
@@ -364,24 +370,36 @@ function connectNextSocket () {
 function interruptConnection(socket, noteId, socketId) {
   if (notes[noteId]) delete notes[noteId]
   if (users[socketId]) delete users[socketId]
-  if (socket) { clearSocketQueue(connectionSocketQueue, socket) } else { connectionSocketQueue.shift() }
+  if (socket) {
+    clearSocketQueue(connectionSocketQueue, socket)
+  } else {
+    connectionSocketQueue.shift()
+  }
   connectNextSocket()
 }
 
 function checkViewPermission(req, note) {
   if (note.permission === 'private') {
-    if (req.user && req.user.logged_in && req.user.id === note.owner) { return true } else { return false }
+    if (req.user && req.user.logged_in && req.user.id === note.owner) {
+      return true
+    } else {
+      return false
+    }
   } else if (note.permission === 'limited' || note.permission === 'protected') {
-    if (req.user && req.user.logged_in) { return true } else { return false }
+    if (req.user && req.user.logged_in) {
+      return true
+    } else {
+      return false
+    }
   } else {
     return true
   }
 }
 
 var isConnectionBusy = false
-var connectionSocketQueue = []
+var connectionSocketQueue: any = []
 var isDisconnectBusy = false
-var disconnectSocketQueue = []
+var disconnectSocketQueue: any = []
 
 function finishConnection(socket, noteId, socketId) {
   // if no valid info provided will drop the client
@@ -438,21 +456,21 @@ function startConnection (socket) {
 
   if (!notes[noteId]) {
     var include = [{
-      model: models.User,
+      model: User,
       as: 'owner'
     }, {
-      model: models.User,
+      model: User,
       as: 'lastchangeuser'
     }, {
-      model: models.Author,
+      model: Author,
       as: 'authors',
       include: [{
-        model: models.User,
+        model: User,
         as: 'user'
       }]
     }]
 
-    models.Note.findOne({
+    Note.findOne({
       where: {
         id: noteId
       },
@@ -462,10 +480,10 @@ function startConnection (socket) {
         return failConnection(404, 'note not found', socket)
       }
       var owner = note.ownerId
-      var ownerprofile = note.owner ? models.User.getProfile(note.owner) : null
+      var ownerprofile = note.owner ? User.getProfile(note.owner) : null
 
       var lastchangeuser = note.lastchangeuserId
-      var lastchangeuserprofile = note.lastchangeuser ? models.User.getProfile(note.lastchangeuser) : null
+      var lastchangeuserprofile = note.lastchangeuser ? User.getProfile(note.lastchangeuser) : null
 
       var body = note.content
       var createtime = note.createdAt
@@ -475,7 +493,7 @@ function startConnection (socket) {
       var authors = {}
       for (var i = 0; i < note.authors.length; i++) {
         var author = note.authors[i]
-        var profile = models.User.getProfile(author.user)
+        var profile = User.getProfile(author.user)
         if (profile) {
           authors[author.userId] = {
             userid: author.userId,
@@ -579,7 +597,9 @@ function disconnect (socket) {
   clearSocketQueue(disconnectSocketQueue, socket)
   // seek for next socket
   isDisconnectBusy = false
-  if (disconnectSocketQueue.length > 0) { disconnect(disconnectSocketQueue[0]) }
+  if (disconnectSocketQueue.length > 0) {
+    disconnect(disconnectSocketQueue[0])
+  }
 
   if (config.debug) {
     logger.debug(notes)
@@ -607,7 +627,7 @@ function buildUserOutData (user) {
 function updateUserData(socket, user) {
   // retrieve user data from passport
   if (socket.request.user && socket.request.user.logged_in) {
-    var profile = models.User.getProfile(socket.request.user)
+    var profile = User.getProfile(socket.request.user)
     user.photo = profile.photo
     user.name = profile.name
     user.userid = socket.request.user.id
@@ -628,13 +648,20 @@ function ifMayEdit (socket, callback) {
     case 'freely':
       // not blocking anyone
       break
-    case 'editable': case 'limited':
+    case 'editable':
+    case 'limited':
       // only login user can change
-      if (!socket.request.user || !socket.request.user.logged_in) { mayEdit = false }
+      if (!socket.request.user || !socket.request.user.logged_in) {
+        mayEdit = false
+      }
       break
-    case 'locked': case 'private': case 'protected':
+    case 'locked':
+    case 'private':
+    case 'protected':
       // only owner can change
-      if (!note.owner || note.owner !== socket.request.user.id) { mayEdit = false }
+      if (!note.owner || note.owner !== socket.request.user.id) {
+        mayEdit = false
+      }
       break
   }
   // if user may edit and this is a text operation
@@ -650,17 +677,17 @@ function ifMayEdit (socket, callback) {
 }
 
 function operationCallback(socket, operation) {
-  var noteId = socket.noteId
+  const noteId = socket.noteId
   if (!noteId || !notes[noteId]) return
-  var note = notes[noteId]
+  const note = notes[noteId]
-  var userId = null
+  let userId = null
   // save authors
   if (socket.request.user && socket.request.user.logged_in) {
     var user = users[socket.id]
     if (!user) return
     userId = socket.request.user.id
     if (!note.authors[userId]) {
-      models.Author.findOrCreate({
+      Author.findOrCreate({
         where: {
           noteId: noteId,
           userId: userId
@@ -670,7 +697,7 @@ function operationCallback (socket, operation) {
           userId: userId,
           color: user.color
         }
-      }).spread(function (author, created) {
+      }).then(function ([author, created]) {
         if (author) {
           note.authors[author.userId] = {
             userid: author.userId,
@@ -687,12 +714,12 @@ function operationCallback (socket, operation) {
   }
   // save authorship - use timer here because it's an O(n) complexity algorithm
   setImmediate(function () {
-    note.authorship = models.Note.updateAuthorshipByOperation(operation, userId, note.authorship)
+    note.authorship = Note.updateAuthorshipByOperation(operation, userId, note.authorship)
   })
 }
 
-function updateHistory (userId, note, time) {
+function updateHistory(userId, note, time?) {
-  var noteId = note.alias ? note.alias : models.Note.encodeNoteId(note.id)
+  var noteId = note.alias ? note.alias : Note.encodeNoteId(note.id)
   if (note.server) history.updateHistory(userId, noteId, note.server.document, time)
 }
 
@@ -780,7 +807,7 @@ function connection (socket) {
       if (note.owner && note.owner === socket.request.user.id) {
         if (permission === 'freely' && !config.allowAnonymous && !config.allowAnonymousEdits) return
         note.permission = permission
-        models.Note.update({
+        Note.update({
           permission: permission
         }, {
           where: {
@@ -824,7 +851,7 @@ function connection (socket) {
       var note = notes[noteId]
       // Only owner can delete note
       if (note.owner && note.owner === socket.request.user.id) {
-        models.Note.destroy({
+        Note.destroy({
           where: {
             id: noteId
           }
@@ -861,10 +888,12 @@ function connection (socket) {
   socket.on('online users', function () {
     var noteId = socket.noteId
     if (!noteId || !notes[noteId]) return
-    var users = []
+    var users: any = []
     Object.keys(notes[noteId].users).forEach(function (key) {
       var user = notes[noteId].users[key]
-      if (user) { users.push(buildUserOutData(user)) }
+      if (user) {
+        users.push(buildUserOutData(user))
+      }
     })
     var out = {
       users: users