This commit is contained in:
mrfry 2021-05-29 21:44:48 +02:00
parent b0fad8d4a2
commit aea8ad22d1
6 changed files with 115 additions and 69 deletions

View file

@ -8,7 +8,7 @@ interface Options {
exceptions: Array<string> exceptions: Array<string>
} }
const testUser = { export const testUser = {
id: 19, id: 19,
avaiblePWRequests: 645, avaiblePWRequests: 645,
pwRequestCount: 19, pwRequestCount: 19,

View file

@ -2,34 +2,44 @@ import logger from '../utils/logger'
import dbtools from '../utils/dbtools' import dbtools from '../utils/dbtools'
import cookie from 'cookie' import cookie from 'cookie'
import { testUser } from './auth.middleware'
interface Options { interface Options {
userDB: any userDB: any
} }
export default function (options: Options): any { export default function SocketAuth(options: Options): any {
const { userDB } = options const { userDB } = options
return function (socket, next) { return (socket, next) => {
const cookies = cookie.parse(socket.handshake.headers.cookie) try {
const cookies = cookie.parse(socket.handshake.headers.cookie || '')
const sessionID = cookies.sessionID const sessionID = cookies.sessionID
if (process.env.NS_NOUSER) { if (process.env.NS_NOUSER) {
socket.user = testUser
next() next()
return return
} }
if (!sessionID) { if (!sessionID) {
next(new Error('Authentication error')) next(new Error('Not authenticated, please log in'))
return return
} }
const user = GetUserBySessionID(userDB, sessionID) const user = GetUserBySessionID(userDB, sessionID)
if (!user) { if (!user) {
next(new Error('Authentication error')) next(new Error('Not authenticated, please log in'))
return return
} }
socket.user = user
next() next()
} catch (e) {
next(new Error('Authentication server error'))
console.error('Authentication server error')
console.error(e)
}
} }
} }

View file

@ -1,6 +1,11 @@
{ {
"msgs": { "msgs": {
"tableStruct": { "tableStruct": {
"id": {
"type": "integer",
"primary": true,
"autoIncrement": true
},
"sender": { "sender": {
"type": "integer", "type": "integer",
"notNull": true "notNull": true
@ -12,6 +17,9 @@
"msg": { "msg": {
"type": "text" "type": "text"
}, },
"type": {
"type": "text"
},
"date": { "date": {
"type": "integer" "type": "integer"
}, },

View file

@ -1,16 +1,24 @@
import { Server as socket } from 'socket.io' import { Server as socket, Socket } from 'socket.io'
import utils from '../../../utils/utils'
import dbtools from '../../../utils/dbtools' import dbtools from '../../../utils/dbtools'
import logger from '../../../utils/logger' import logger from '../../../utils/logger'
import { Request, SubmoduleData } from '../../../types/basicTypes' import { Request, SubmoduleData, User } from '../../../types/basicTypes'
import socketAuth from '../../../middlewares/socketAuth.middleware' import socketAuth from '../../../middlewares/socketAuth.middleware'
const msgDbPath = './data/dbs/msgs.db' const msgDbPath = './data/dbs/msgs.db'
const msgPaginationLimit = 15
interface ExtendedSocket extends Socket {
user: User
}
function setup(data: SubmoduleData): void { function setup(data: SubmoduleData): void {
const { app, httpServer, httpsServer, userDB } = data const { app, httpServer, httpsServer, userDB, publicdirs } = data
const msgDB = dbtools.GetDB(msgDbPath) const msgDB = dbtools.GetDB(msgDbPath)
const publicDir = publicdirs[0]
const uloadFiles = publicDir + 'chatFiles'
logger.Log(`Starting Socket.io Server on ${httpsServer ? 'https' : 'http'}`) logger.Log(`Starting Socket.io Server on ${httpsServer ? 'https' : 'http'}`)
// https://socket.io/docs/v4/handling-cors/#Configuration // https://socket.io/docs/v4/handling-cors/#Configuration
const io = new socket(httpsServer || httpServer, { const io = new socket(httpsServer || httpServer, {
@ -29,29 +37,28 @@ function setup(data: SubmoduleData): void {
where sender = ${sender} and reciever = ${reciever}`, where sender = ${sender} and reciever = ${reciever}`,
'run' 'run'
) )
io.sockets.in(sender).emit('chat message read', { io.sockets.in(sender.toString()).emit('chat message read', {
userReadMsg: reciever, userReadMsg: reciever,
}) })
} }
io.use(socketAuth({ userDB: userDB })) io.use(socketAuth({ userDB: userDB }))
io.on('connection', (socket) => { io.on('connection', (socket: ExtendedSocket) => {
// TODO // TODO: UNCOMMENT
// https://stackoverflow.com/questions/19106861/authorizing-and-handshaking-with-socket-io#19106961 // -----------------------------------------------------------------
// socket.handshake.headers // const userid = socket.user.id
// if (data.id !== user.id) { // logger.Log(`Chat connect: ${userid}`, logger.GetColor('green'))
// socket.disconnect() // -----------------------------------------------------------------
// socket.emit('connectData', {
// success: false,
// msg: `You are only authorized to connect as ${user.id}!`
// })
// return
// }
socket.on('join', function (data) { socket.on('join', function (data) {
socket.join(data.id) // TODO: REMOVE
const userid = data.id // -----------------------------------------------------------------
const userid = parseInt(data.id)
logger.Log(`Chat connect: ${userid}`, logger.GetColor('green'))
// -----------------------------------------------------------------
socket.join(userid.toString())
let currUser = dbtools.Select(msgDB, 'users', { let currUser = dbtools.Select(msgDB, 'users', {
id: userid, id: userid,
})[0] })[0]
@ -96,21 +103,24 @@ function setup(data: SubmoduleData): void {
}), }),
}) })
socket.on('chat message open', (data) => { socket.on('get chat messages', (data) => {
const { chatPartner } = data const { chatPartner, from } = data
// TODO: pagination const msgs = dbtools.runStatement(
socket.emit(
'chat message open',
dbtools.runStatement(
msgDB, msgDB,
`select * from msgs `select * from msgs
where sender = ${userid} and reciever = ${chatPartner} or where (sender = ${userid} and reciever = ${chatPartner} or
sender = ${chatPartner} and reciever = ${userid} sender = ${chatPartner} and reciever = ${userid})
order by date asc` ${from ? `and date < ${from}` : ''}
) order by date desc
limit ${msgPaginationLimit}`
) )
socket.emit('get chat messages', {
requestsdMsgs: msgs,
hasMore: msgs.length === msgPaginationLimit,
})
// Read update // Read update
chatMessageRead({ sender: chatPartner, reciever: userid }) chatMessageRead({ sender: chatPartner, reciever: userid })
}) })
@ -121,7 +131,7 @@ function setup(data: SubmoduleData): void {
}) })
socket.on('chat message', (message) => { socket.on('chat message', (message) => {
const { reciever, msg } = message const { reciever, msg, type } = message
const recieverUser = dbtools.Select(msgDB, 'users', { const recieverUser = dbtools.Select(msgDB, 'users', {
id: reciever, id: reciever,
})[0] })[0]
@ -131,26 +141,45 @@ function setup(data: SubmoduleData): void {
date: new Date().getTime(), date: new Date().getTime(),
sender: reciever, sender: reciever,
reciever: userid, reciever: userid,
type: 'text',
msg: `A #${reciever} számú felhasználó nem létezik`, msg: `A #${reciever} számú felhasználó nem létezik`,
}) })
return return
} }
const msgObj = { const msgObj = {
sender: parseInt(userid), sender: userid,
reciever: parseInt(reciever), reciever: parseInt(reciever),
msg: msg, msg: msg,
type: type || 'text',
date: new Date().getTime(), date: new Date().getTime(),
unread: 1, unread: 1,
} }
dbtools.Insert(msgDB, 'msgs', msgObj) dbtools.Insert(msgDB, 'msgs', msgObj)
if (userid !== reciever) { if (userid !== reciever) {
io.sockets.in(reciever).emit('chat message', msgObj) io.sockets.in(reciever.toString()).emit('chat message', msgObj)
} }
}) })
}) })
// socket.on('disconnect', () => {}) // socket.on('disconnect', () => {})
// socket.on('close', () => {})
})
app.post('/postchatfile', function (req: Request, res: any) {
logger.LogReq(req)
utils
.uploadFile(req, uloadFiles)
.then((result) => {
res.json({
success: true,
path: result.filePath.replace(publicDir, ''),
})
})
.catch(() => {
res.json({ success: false, msg: 'error during uploading' })
return
})
}) })
app.get('/hasNewMsg', (req: Request, res) => { app.get('/hasNewMsg', (req: Request, res) => {

View file

@ -206,7 +206,6 @@ function uploadFile(req: Request, path: string): Promise<any> {
const temp = file.name.split('.') const temp = file.name.split('.')
const extension = temp.pop() const extension = temp.pop()
fileName = temp.join('.') + '_' + id + '.' + extension fileName = temp.join('.') + '_' + id + '.' + extension
console.log(fileName)
fileDestination = path + '/' + fileName fileDestination = path + '/' + fileName
} }

@ -1 +1 @@
Subproject commit c60ace4f9b9b7a0f45963c226b809d0b8b6bab29 Subproject commit 602e16046e6600a405ae68814c11349fde745244