Migrate realtime.js to TypeScript

Signed-off-by: David Mehren <dmehren1@gmail.com>
This commit is contained in:
David Mehren 2020-04-11 11:27:54 +02:00
parent 0ad30a0029
commit 77e84a18ee
No known key found for this signature in database
GPG key ID: 6017AF117F9756CB

View file

@ -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) {
var distinctaddresses = []
var regaddresses = []
var distinctregaddresses = []
Note.count().then(function (notecount) {
var distinctaddresses: string[] = []
var regaddresses: string[] = []
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]
var userId = null
const note = notes[noteId]
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) {
var noteId = note.alias ? note.alias : models.Note.encodeNoteId(note.id)
function updateHistory(userId, note, time?) {
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