mirror of
https://github.com/spacebarchat/server.git
synced 2024-11-11 05:02:37 +01:00
✨ rate limit now configurable in the config
This commit is contained in:
parent
63c1956dd9
commit
202cc24811
@ -12,7 +12,7 @@ import { BodyParser } from "./middlewares/BodyParser";
|
||||
import express, { Router, Request, Response } from "express";
|
||||
import mongoose from "mongoose";
|
||||
import path from "path";
|
||||
import RateLimit from "./middlewares/RateLimit";
|
||||
import { initRateLimits } from "./middlewares/RateLimit";
|
||||
import TestClient from "./middlewares/TestClient";
|
||||
|
||||
// this will return the new updated document for findOneAndUpdate
|
||||
@ -86,12 +86,8 @@ export class FosscordServer extends Server {
|
||||
const api = Router();
|
||||
// @ts-ignore
|
||||
this.app = api;
|
||||
api.use(RateLimit({ bucket: "global", count: 10, window: 5, bot: 250 }));
|
||||
api.use(RateLimit({ bucket: "error", count: 5, error: true, window: 5, bot: 15, onlyIp: true }));
|
||||
api.use("/guilds/:id", RateLimit({ count: 5, window: 5 }));
|
||||
api.use("/webhooks/:id", RateLimit({ count: 5, window: 5 }));
|
||||
api.use("/channels/:id", RateLimit({ count: 5, window: 5 }));
|
||||
|
||||
initRateLimits(api);
|
||||
this.routes = await this.registerRoutes(path.join(__dirname, "routes", "/"));
|
||||
app.use("/api/v8", api);
|
||||
app.use("/api/v9", api);
|
||||
|
@ -1,5 +1,5 @@
|
||||
import { db, MongooseCache, Bucket } from "@fosscord/server-util";
|
||||
import { IRouterHandler, NextFunction, Request, Response } from "express";
|
||||
import { db, MongooseCache, Bucket, Config } from "@fosscord/server-util";
|
||||
import { NextFunction, Request, Response, Router } from "express";
|
||||
import { getIpAdress } from "../util/ipAddress";
|
||||
import { API_PREFIX_TRAILING_SLASH } from "./Authentication";
|
||||
|
||||
@ -65,7 +65,7 @@ export default function RateLimit(opts: {
|
||||
const global = bucket_id === "global";
|
||||
|
||||
if (resetAfterMs > 0) {
|
||||
console.log("blocked", { resetAfterMs });
|
||||
console.log("blocked bucket: " + bucket_id, { resetAfterMs });
|
||||
return (
|
||||
res
|
||||
.status(429)
|
||||
@ -105,6 +105,32 @@ export default function RateLimit(opts: {
|
||||
};
|
||||
}
|
||||
|
||||
export function initRateLimits(app: Router) {
|
||||
const { routes, global, ip, error } = Config.get().limits.rate;
|
||||
|
||||
app.use(
|
||||
RateLimit({
|
||||
bucket: "global",
|
||||
onlyIp: true,
|
||||
...ip
|
||||
})
|
||||
);
|
||||
app.use(RateLimit({ bucket: "global", ...global }));
|
||||
app.use(
|
||||
RateLimit({
|
||||
bucket: "error",
|
||||
error: true,
|
||||
onlyIp: true,
|
||||
...error
|
||||
})
|
||||
);
|
||||
app.use("/guilds/:id", RateLimit(routes.guild));
|
||||
app.use("/webhooks/:id", RateLimit(routes.webhook));
|
||||
app.use("/channels/:id", RateLimit(routes.channel));
|
||||
app.use("/auth/login", RateLimit(routes.auth.login));
|
||||
app.use("/auth/register", RateLimit({ onlyIp: true, success: true, ...routes.auth.register }));
|
||||
}
|
||||
|
||||
function hitRoute(opts: { user_id: string; bucket_id: string; max_hits: number; window: number }) {
|
||||
return db.collection("ratelimits").updateOne(
|
||||
{ id: opts.bucket_id, user_id: opts.user_id },
|
||||
|
@ -13,7 +13,6 @@ export default router;
|
||||
|
||||
router.post(
|
||||
"/",
|
||||
RateLimit({ count: 5, window: 60, onlyIp: true }),
|
||||
check({
|
||||
login: new Length(String, 2, 100), // email or telephone
|
||||
password: new Length(String, 8, 72),
|
||||
|
@ -12,7 +12,6 @@ const router: Router = Router();
|
||||
|
||||
router.post(
|
||||
"/",
|
||||
RateLimit({ count: 2, window: 60 * 60 * 12, onlyIp: true, success: true }),
|
||||
check({
|
||||
username: new Length(String, 2, 32),
|
||||
// TODO: check min password length in config
|
||||
|
Loading…
Reference in New Issue
Block a user