mirror of https://github.com/Troplo/Colubrina.git
46 lines
1.0 KiB
JavaScript
46 lines
1.0 KiB
JavaScript
const { User, Theme, Session } = require("../models")
|
|
|
|
const parseCookie = (str) =>
|
|
str
|
|
.split(";")
|
|
.map((v) => v.split("="))
|
|
.reduce((acc, v) => {
|
|
acc[decodeURIComponent(v[0].trim())] = decodeURIComponent(v[1].trim())
|
|
return acc
|
|
}, {})
|
|
module.exports = async function (socket, next) {
|
|
try {
|
|
const cookies = parseCookie(socket.handshake.headers.cookie)
|
|
if (cookies["session"]) {
|
|
const session = await Session.findOne({
|
|
where: { session: cookies["session"] }
|
|
})
|
|
if (session) {
|
|
const user = await User.findOne({
|
|
where: { id: session.userId },
|
|
attributes: {
|
|
exclude: ["totp", "password"]
|
|
},
|
|
include: [
|
|
{
|
|
model: Theme,
|
|
as: "themeObject"
|
|
}
|
|
],
|
|
raw: true
|
|
})
|
|
if (user) {
|
|
socket.user = user
|
|
next()
|
|
}
|
|
} else {
|
|
//
|
|
}
|
|
} else {
|
|
//
|
|
}
|
|
} catch (error) {
|
|
//
|
|
}
|
|
}
|