mirror of
https://github.com/ragestudio/comty.git
synced 2025-06-09 18:44:16 +00:00
split controller init
This commit is contained in:
parent
9290fc0a5d
commit
aa9211e75c
273
packages/server/src/api.js
Normal file
273
packages/server/src/api.js
Normal file
@ -0,0 +1,273 @@
|
||||
import path from "path"
|
||||
import { Server as LinebridgeServer } from "linebridge/dist/server"
|
||||
import express from "express"
|
||||
import bcrypt from "bcrypt"
|
||||
import passport from "passport"
|
||||
|
||||
import jwt from "jsonwebtoken"
|
||||
|
||||
import { User, Session, Config } from "./models"
|
||||
import DbManager from "./classes/DbManager"
|
||||
|
||||
const ExtractJwt = require("passport-jwt").ExtractJwt
|
||||
const LocalStrategy = require("passport-local").Strategy
|
||||
|
||||
const controllers = require("./controllers")
|
||||
const middlewares = require("./middlewares")
|
||||
|
||||
export default class Server {
|
||||
env = process.env
|
||||
|
||||
DB = new DbManager()
|
||||
|
||||
httpListenPort = this.env.listenPort ?? 3000
|
||||
|
||||
controllers = [
|
||||
controllers.ConfigController,
|
||||
controllers.RolesController,
|
||||
controllers.SessionController,
|
||||
controllers.UserController,
|
||||
controllers.FilesController,
|
||||
controllers.PublicController,
|
||||
controllers.PostsController,
|
||||
controllers.StreamingController,
|
||||
]
|
||||
|
||||
middlewares = middlewares
|
||||
|
||||
server = new LinebridgeServer({
|
||||
port: this.httpListenPort,
|
||||
headers: {
|
||||
"Access-Control-Expose-Headers": "regenerated_token",
|
||||
},
|
||||
onWSClientConnection: this.onWSClientConnection,
|
||||
onWSClientDisconnection: this.onWSClientDisconnection,
|
||||
}, this.controllers, this.middlewares)
|
||||
|
||||
options = {
|
||||
jwtStrategy: {
|
||||
sessionLocationSign: this.server.id,
|
||||
jwtFromRequest: ExtractJwt.fromAuthHeaderAsBearerToken(),
|
||||
secretOrKey: this.server.oskid,
|
||||
algorithms: ["sha1", "RS256", "HS256"],
|
||||
expiresIn: this.env.signLifetime ?? "1h",
|
||||
}
|
||||
}
|
||||
|
||||
constructor() {
|
||||
this.server.engineInstance.use(express.json())
|
||||
this.server.engineInstance.use(express.urlencoded({ extended: true }))
|
||||
|
||||
this.server.engineInstance.use("/storage", express.static(path.join(__dirname, "../uploads")))
|
||||
|
||||
this.server.wsInterface["clients"] = []
|
||||
this.server.wsInterface["findUserIdFromClientID"] = (searchClientId) => {
|
||||
return this.server.wsInterface.clients.find(client => client.id === searchClientId)?.userId ?? false
|
||||
}
|
||||
this.server.wsInterface["getClientSockets"] = (userId) => {
|
||||
return this.server.wsInterface.clients.filter(client => client.userId === userId).map((client) => {
|
||||
return client?.socket
|
||||
})
|
||||
}
|
||||
this.server.wsInterface["broadcast"] = async (channel, ...args) => {
|
||||
for await (const client of this.server.wsInterface.clients) {
|
||||
client.socket.emit(channel, ...args)
|
||||
}
|
||||
}
|
||||
|
||||
global.wsInterface = this.server.wsInterface
|
||||
global.httpListenPort = this.listenPort
|
||||
|
||||
global.publicHostname = this.env.publicHostname
|
||||
global.publicProtocol = this.env.publicProtocol
|
||||
global.globalPublicUri = `${this.env.publicProtocol}://${this.env.publicHost}`
|
||||
|
||||
global.uploadPath = this.env.uploadPath ?? path.resolve(process.cwd(), "uploads")
|
||||
global.uploadCachePath = this.env.uploadCachePath ?? path.resolve(process.cwd(), "cache")
|
||||
|
||||
global.jwtStrategy = this.options.jwtStrategy
|
||||
global.signLocation = this.env.signLocation
|
||||
}
|
||||
|
||||
async initialize() {
|
||||
await this.DB.connect()
|
||||
await this.initializeConfigDB()
|
||||
|
||||
await this.checkSetup()
|
||||
await this.initPassport()
|
||||
await this.initWebsockets()
|
||||
|
||||
await this.server.initialize()
|
||||
}
|
||||
|
||||
initializeConfigDB = async () => {
|
||||
let serverConfig = await Config.findOne({ key: "server" }).catch(() => {
|
||||
return false
|
||||
})
|
||||
|
||||
if (!serverConfig) {
|
||||
serverConfig = new Config({
|
||||
key: "server",
|
||||
value: {
|
||||
setup: false,
|
||||
},
|
||||
})
|
||||
|
||||
|
||||
await serverConfig.save()
|
||||
}
|
||||
}
|
||||
|
||||
checkSetup = async () => {
|
||||
return new Promise(async (resolve, reject) => {
|
||||
let setupOk = (await Config.findOne({ key: "server" })).value?.setup ?? false
|
||||
|
||||
if (!setupOk) {
|
||||
console.log("⚠️ Server setup is not complete, running setup proccess.")
|
||||
let setupScript = await import("./mainAPI/setup")
|
||||
|
||||
setupScript = setupScript.default ?? setupScript
|
||||
|
||||
try {
|
||||
for await (let script of setupScript) {
|
||||
await script()
|
||||
}
|
||||
|
||||
console.log("✅ Server setup complete.")
|
||||
|
||||
await Config.updateOne({ key: "server" }, { value: { setup: true } })
|
||||
|
||||
return resolve()
|
||||
} catch (error) {
|
||||
console.log("❌ Server setup failed.")
|
||||
console.error(error)
|
||||
process.exit(1)
|
||||
}
|
||||
}
|
||||
|
||||
return resolve()
|
||||
})
|
||||
}
|
||||
|
||||
initPassport() {
|
||||
this.server.middlewares["useJwtStrategy"] = (req, res, next) => {
|
||||
req.jwtStrategy = this.options.jwtStrategy
|
||||
next()
|
||||
}
|
||||
|
||||
passport.use(new LocalStrategy({
|
||||
usernameField: "username",
|
||||
passwordField: "password",
|
||||
session: false
|
||||
}, (username, password, done) => {
|
||||
User.findOne({ username }).select("+password")
|
||||
.then((data) => {
|
||||
if (data === null) {
|
||||
return done(null, false, this.options.jwtStrategy)
|
||||
} else if (!bcrypt.compareSync(password, data.password)) {
|
||||
return done(null, false, this.options.jwtStrategy)
|
||||
}
|
||||
|
||||
// create a token
|
||||
return done(null, data, this.options.jwtStrategy, { username, password })
|
||||
})
|
||||
.catch(err => done(err, null, this.options.jwtStrategy))
|
||||
}))
|
||||
|
||||
this.server.engineInstance.use(passport.initialize())
|
||||
}
|
||||
|
||||
initWebsockets() {
|
||||
this.server.middlewares["useWS"] = (req, res, next) => {
|
||||
req.ws = global.wsInterface
|
||||
next()
|
||||
}
|
||||
|
||||
const onAuthenticated = (socket, user_id) => {
|
||||
this.attachClientSocket(socket, user_id)
|
||||
socket.emit("authenticated")
|
||||
}
|
||||
|
||||
const onAuthenticatedFailed = (socket, error) => {
|
||||
this.detachClientSocket(socket)
|
||||
socket.emit("authenticateFailed", {
|
||||
error,
|
||||
})
|
||||
}
|
||||
|
||||
this.server.wsInterface.eventsChannels.push(["/main", "authenticate", async (socket, token) => {
|
||||
const session = await Session.findOne({ token }).catch(err => {
|
||||
return false
|
||||
})
|
||||
|
||||
if (!session) {
|
||||
return onAuthenticatedFailed(socket, "Session not found")
|
||||
}
|
||||
|
||||
this.verifyJwt(token, async (err, decoded) => {
|
||||
if (err) {
|
||||
return onAuthenticatedFailed(socket, err)
|
||||
} else {
|
||||
const user = await User.findById(decoded.user_id).catch(err => {
|
||||
return false
|
||||
})
|
||||
|
||||
if (!user) {
|
||||
return onAuthenticatedFailed(socket, "User not found")
|
||||
}
|
||||
|
||||
return onAuthenticated(socket, user)
|
||||
}
|
||||
})
|
||||
}])
|
||||
}
|
||||
|
||||
onWSClientConnection = async (socket) => {
|
||||
console.log(`🌐 Client connected: ${socket.id}`)
|
||||
}
|
||||
|
||||
onWSClientDisconnection = async (socket) => {
|
||||
console.log(`🌐 Client disconnected: ${socket.id}`)
|
||||
this.detachClientSocket(socket)
|
||||
}
|
||||
|
||||
attachClientSocket = async (client, userData) => {
|
||||
const socket = this.server.wsInterface.clients.find(c => c.id === client.id)
|
||||
|
||||
if (socket) {
|
||||
socket.socket.disconnect()
|
||||
}
|
||||
|
||||
const clientObj = {
|
||||
id: client.id,
|
||||
socket: client,
|
||||
userId: userData._id.toString(),
|
||||
user: userData,
|
||||
}
|
||||
|
||||
this.server.wsInterface.clients.push(clientObj)
|
||||
|
||||
this.server.wsInterface.io.emit("userConnected", userData)
|
||||
}
|
||||
|
||||
detachClientSocket = async (client) => {
|
||||
const socket = this.server.wsInterface.clients.find(c => c.id === client.id)
|
||||
|
||||
if (socket) {
|
||||
socket.socket.disconnect()
|
||||
this.server.wsInterface.clients = this.server.wsInterface.clients.filter(c => c.id !== client.id)
|
||||
}
|
||||
|
||||
this.server.wsInterface.io.emit("userDisconnect", client.id)
|
||||
}
|
||||
|
||||
verifyJwt = (token, callback) => {
|
||||
jwt.verify(token, this.options.jwtStrategy.secretOrKey, async (err, decoded) => {
|
||||
if (err) {
|
||||
return callback(err)
|
||||
}
|
||||
|
||||
return callback(null, decoded)
|
||||
})
|
||||
}
|
||||
}
|
||||
@ -1,19 +1,8 @@
|
||||
import { default as ConfigController } from "./ConfigController"
|
||||
import { default as RolesController } from "./RolesController"
|
||||
import { default as SessionController } from "./SessionController"
|
||||
import { default as UserController } from "./UserController"
|
||||
import { default as FilesController } from "./FilesController"
|
||||
import { default as PublicController } from "./PublicController"
|
||||
import { default as PostsController } from "./PostsController"
|
||||
import { default as StreamingController } from "./StreamingController"
|
||||
|
||||
export default [
|
||||
PostsController,
|
||||
ConfigController,
|
||||
PublicController,
|
||||
RolesController,
|
||||
SessionController,
|
||||
UserController,
|
||||
FilesController,
|
||||
StreamingController,
|
||||
]
|
||||
export { default as ConfigController } from "./ConfigController"
|
||||
export { default as RolesController } from "./RolesController"
|
||||
export { default as SessionController } from "./SessionController"
|
||||
export { default as UserController } from "./UserController"
|
||||
export { default as FilesController } from "./FilesController"
|
||||
export { default as PublicController } from "./PublicController"
|
||||
export { default as PostsController } from "./PostsController"
|
||||
export { default as StreamingController } from "./StreamingController"
|
||||
@ -20,267 +20,15 @@ Array.prototype.updateFromObjectKeys = function (obj) {
|
||||
return this
|
||||
}
|
||||
|
||||
import path from "path"
|
||||
import { Server as LinebridgeServer } from "linebridge/dist/server"
|
||||
import express from "express"
|
||||
import bcrypt from "bcrypt"
|
||||
import passport from "passport"
|
||||
import API from "./api"
|
||||
|
||||
import jwt from "jsonwebtoken"
|
||||
async function main() {
|
||||
const mainAPI = new API()
|
||||
|
||||
import { User, Session, Config } from "./models"
|
||||
import DbManager from "./classes/DbManager"
|
||||
|
||||
const ExtractJwt = require("passport-jwt").ExtractJwt
|
||||
const LocalStrategy = require("passport-local").Strategy
|
||||
|
||||
class Server {
|
||||
env = process.env
|
||||
|
||||
DB = new DbManager()
|
||||
|
||||
httpListenPort = this.env.listenPort ?? 3000
|
||||
|
||||
controllers = require("./controllers").default
|
||||
middlewares = require("./middlewares")
|
||||
|
||||
server = new LinebridgeServer({
|
||||
port: this.httpListenPort,
|
||||
headers: {
|
||||
"Access-Control-Expose-Headers": "regenerated_token",
|
||||
},
|
||||
onWSClientConnection: this.onWSClientConnection,
|
||||
onWSClientDisconnection: this.onWSClientDisconnection,
|
||||
}, this.controllers, this.middlewares)
|
||||
|
||||
options = {
|
||||
jwtStrategy: {
|
||||
sessionLocationSign: this.server.id,
|
||||
jwtFromRequest: ExtractJwt.fromAuthHeaderAsBearerToken(),
|
||||
secretOrKey: this.server.oskid,
|
||||
algorithms: ["sha1", "RS256", "HS256"],
|
||||
expiresIn: this.env.signLifetime ?? "1h",
|
||||
}
|
||||
}
|
||||
|
||||
constructor() {
|
||||
this.server.engineInstance.use(express.json())
|
||||
this.server.engineInstance.use(express.urlencoded({ extended: true }))
|
||||
|
||||
this.server.engineInstance.use("/storage", express.static(path.join(__dirname, "../uploads")))
|
||||
|
||||
this.server.wsInterface["clients"] = []
|
||||
this.server.wsInterface["findUserIdFromClientID"] = (searchClientId) => {
|
||||
return this.server.wsInterface.clients.find(client => client.id === searchClientId)?.userId ?? false
|
||||
}
|
||||
this.server.wsInterface["getClientSockets"] = (userId) => {
|
||||
return this.server.wsInterface.clients.filter(client => client.userId === userId).map((client) => {
|
||||
return client?.socket
|
||||
})
|
||||
}
|
||||
this.server.wsInterface["broadcast"] = async (channel, ...args) => {
|
||||
for await (const client of this.server.wsInterface.clients) {
|
||||
client.socket.emit(channel, ...args)
|
||||
}
|
||||
}
|
||||
|
||||
global.wsInterface = this.server.wsInterface
|
||||
global.httpListenPort = this.listenPort
|
||||
|
||||
global.publicHostname = this.env.publicHostname
|
||||
global.publicProtocol = this.env.publicProtocol
|
||||
global.globalPublicUri = `${this.env.publicProtocol}://${this.env.publicHost}`
|
||||
|
||||
global.uploadPath = this.env.uploadPath ?? path.resolve(process.cwd(), "uploads")
|
||||
global.uploadCachePath = this.env.uploadCachePath ?? path.resolve(process.cwd(), "cache")
|
||||
|
||||
global.jwtStrategy = this.options.jwtStrategy
|
||||
global.signLocation = this.env.signLocation
|
||||
|
||||
this.initialize()
|
||||
}
|
||||
|
||||
async initialize() {
|
||||
await this.DB.connect()
|
||||
await this.initializeConfigDB()
|
||||
|
||||
await this.checkSetup()
|
||||
await this.initPassport()
|
||||
await this.initWebsockets()
|
||||
|
||||
await this.server.initialize()
|
||||
}
|
||||
|
||||
initializeConfigDB = async () => {
|
||||
let serverConfig = await Config.findOne({ key: "server" }).catch(() => {
|
||||
return false
|
||||
})
|
||||
|
||||
if (!serverConfig) {
|
||||
serverConfig = new Config({
|
||||
key: "server",
|
||||
value: {
|
||||
setup: false,
|
||||
},
|
||||
})
|
||||
|
||||
|
||||
await serverConfig.save()
|
||||
}
|
||||
}
|
||||
|
||||
checkSetup = async () => {
|
||||
return new Promise(async (resolve, reject) => {
|
||||
let setupOk = (await Config.findOne({ key: "server" })).value?.setup ?? false
|
||||
|
||||
if (!setupOk) {
|
||||
console.log("⚠️ Server setup is not complete, running setup proccess.")
|
||||
let setupScript = await import("./setup")
|
||||
|
||||
setupScript = setupScript.default ?? setupScript
|
||||
|
||||
try {
|
||||
for await (let script of setupScript) {
|
||||
await script()
|
||||
}
|
||||
|
||||
console.log("✅ Server setup complete.")
|
||||
|
||||
await Config.updateOne({ key: "server" }, { value: { setup: true } })
|
||||
|
||||
return resolve()
|
||||
} catch (error) {
|
||||
console.log("❌ Server setup failed.")
|
||||
console.error(error)
|
||||
process.exit(1)
|
||||
}
|
||||
}
|
||||
|
||||
return resolve()
|
||||
})
|
||||
}
|
||||
|
||||
initPassport() {
|
||||
this.server.middlewares["useJwtStrategy"] = (req, res, next) => {
|
||||
req.jwtStrategy = this.options.jwtStrategy
|
||||
next()
|
||||
}
|
||||
|
||||
passport.use(new LocalStrategy({
|
||||
usernameField: "username",
|
||||
passwordField: "password",
|
||||
session: false
|
||||
}, (username, password, done) => {
|
||||
User.findOne({ username }).select("+password")
|
||||
.then((data) => {
|
||||
if (data === null) {
|
||||
return done(null, false, this.options.jwtStrategy)
|
||||
} else if (!bcrypt.compareSync(password, data.password)) {
|
||||
return done(null, false, this.options.jwtStrategy)
|
||||
}
|
||||
|
||||
// create a token
|
||||
return done(null, data, this.options.jwtStrategy, { username, password })
|
||||
})
|
||||
.catch(err => done(err, null, this.options.jwtStrategy))
|
||||
}))
|
||||
|
||||
this.server.engineInstance.use(passport.initialize())
|
||||
}
|
||||
|
||||
initWebsockets() {
|
||||
this.server.middlewares["useWS"] = (req, res, next) => {
|
||||
req.ws = global.wsInterface
|
||||
next()
|
||||
}
|
||||
|
||||
const onAuthenticated = (socket, user_id) => {
|
||||
this.attachClientSocket(socket, user_id)
|
||||
socket.emit("authenticated")
|
||||
}
|
||||
|
||||
const onAuthenticatedFailed = (socket, error) => {
|
||||
this.detachClientSocket(socket)
|
||||
socket.emit("authenticateFailed", {
|
||||
error,
|
||||
})
|
||||
}
|
||||
|
||||
this.server.wsInterface.eventsChannels.push(["/main", "authenticate", async (socket, token) => {
|
||||
const session = await Session.findOne({ token }).catch(err => {
|
||||
return false
|
||||
})
|
||||
|
||||
if (!session) {
|
||||
return onAuthenticatedFailed(socket, "Session not found")
|
||||
}
|
||||
|
||||
this.verifyJwt(token, async (err, decoded) => {
|
||||
if (err) {
|
||||
return onAuthenticatedFailed(socket, err)
|
||||
} else {
|
||||
const user = await User.findById(decoded.user_id).catch(err => {
|
||||
return false
|
||||
})
|
||||
|
||||
if (!user) {
|
||||
return onAuthenticatedFailed(socket, "User not found")
|
||||
}
|
||||
|
||||
return onAuthenticated(socket, user)
|
||||
}
|
||||
})
|
||||
}])
|
||||
}
|
||||
|
||||
onWSClientConnection = async (socket) => {
|
||||
console.log(`🌐 Client connected: ${socket.id}`)
|
||||
}
|
||||
|
||||
onWSClientDisconnection = async (socket) => {
|
||||
console.log(`🌐 Client disconnected: ${socket.id}`)
|
||||
this.detachClientSocket(socket)
|
||||
}
|
||||
|
||||
attachClientSocket = async (client, userData) => {
|
||||
const socket = this.server.wsInterface.clients.find(c => c.id === client.id)
|
||||
|
||||
if (socket) {
|
||||
socket.socket.disconnect()
|
||||
}
|
||||
|
||||
const clientObj = {
|
||||
id: client.id,
|
||||
socket: client,
|
||||
userId: userData._id.toString(),
|
||||
user: userData,
|
||||
}
|
||||
|
||||
this.server.wsInterface.clients.push(clientObj)
|
||||
|
||||
this.server.wsInterface.io.emit("userConnected", userData)
|
||||
}
|
||||
|
||||
detachClientSocket = async (client) => {
|
||||
const socket = this.server.wsInterface.clients.find(c => c.id === client.id)
|
||||
|
||||
if (socket) {
|
||||
socket.socket.disconnect()
|
||||
this.server.wsInterface.clients = this.server.wsInterface.clients.filter(c => c.id !== client.id)
|
||||
}
|
||||
|
||||
this.server.wsInterface.io.emit("userDisconnect", client.id)
|
||||
}
|
||||
|
||||
verifyJwt = (token, callback) => {
|
||||
jwt.verify(token, this.options.jwtStrategy.secretOrKey, async (err, decoded) => {
|
||||
if (err) {
|
||||
return callback(err)
|
||||
}
|
||||
|
||||
return callback(null, decoded)
|
||||
})
|
||||
}
|
||||
console.log("\n▶️ Initializing main API...\n")
|
||||
await mainAPI.initialize()
|
||||
}
|
||||
|
||||
new Server()
|
||||
main().catch((error) => {
|
||||
console.error(`🆘 [FATAL ERROR] >`, error)
|
||||
})
|
||||
Loading…
x
Reference in New Issue
Block a user