69 lines
2.1 KiB
TypeScript
69 lines
2.1 KiB
TypeScript
import { Request, Response, NextFunction } from "express";
|
|
import FileManager from "./data/FileManager.ts";
|
|
import TokenManager from "./api/TokenManager.ts";
|
|
import UserChatLinker from "./data/UserChatLinker.ts";
|
|
import fileUpload from "express-fileupload";
|
|
|
|
export class Middleware {
|
|
static Authroize(req: Request, res: Response, chat_id: string | undefined) {
|
|
const userToken = TokenManager.decode(req.headers.token || req.cookies.token);
|
|
if (!TokenManager.checkToken(userToken, req.headers["device-id"] || req.cookies.device_id)) {
|
|
res.status(401).send({
|
|
msg: "401 UnAuthorized",
|
|
});
|
|
return false;
|
|
}
|
|
if (chat_id && !UserChatLinker.checkUserIsLinkedToChat(userToken.author, chat_id)) {
|
|
res.status(403).send({
|
|
msg: "403 Forbidden",
|
|
});
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
static Get_uploaded_files(req: Request, res: Response, next: NextFunction) {
|
|
const hash = req.params.hash as string;
|
|
if (hash == null) {
|
|
res.status(404).send({
|
|
msg: "404 Not Found",
|
|
});
|
|
return;
|
|
}
|
|
const file = FileManager.findByHash(hash);
|
|
|
|
if (file == null) {
|
|
res.status(404).send({
|
|
msg: "404 Not Found",
|
|
});
|
|
return;
|
|
}
|
|
|
|
if (file.getChatId() != null) {
|
|
if (!Middleware.Authroize(req, res, file.getChatId() as string)) {
|
|
return;
|
|
}
|
|
}
|
|
next();
|
|
}
|
|
|
|
static Post_upload_file(req: Request, res: Response, next: NextFunction) {
|
|
if (!Middleware.Authroize(req, res, req.body.chat_id)) {
|
|
return;
|
|
}
|
|
const file = req.files?.file as fileUpload.UploadedFile;
|
|
if (file?.data == null) {
|
|
res.status(400).send({
|
|
msg: "No file was found or multiple files were uploaded",
|
|
});
|
|
return;
|
|
}
|
|
if (req.body.file_name == null) {
|
|
res.status(400).send({
|
|
msg: "Filename is required",
|
|
});
|
|
return;
|
|
}
|
|
}
|
|
}
|