chore: limit number of visits returned from backend

fingerprintjs · Jan 14, 2024 · 8bbffd5 · 8bbffd5
1 parent 39d40c2
commit 8bbffd5
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 5 deletions.
diff --git a/src/pages/api/bot-firewall/get-bot-visits.ts b/src/pages/api/bot-firewall/get-bot-visits.ts
@@ -1,10 +1,11 @@
 import { NextApiRequest, NextApiResponse } from 'next';
 import { BotVisit, getBotVisits } from '../../../server/botd-firewall/botVisitDatabase';
 
-export default async function handler(_req: NextApiRequest, res: NextApiResponse<BotVisit[]>) {
+export default async function handler(req: NextApiRequest, res: NextApiResponse<BotVisit[]>) {
   try {
-    const blockedIps = await getBotVisits();
-    res.status(200).json(blockedIps);
+    const limit = Number(req.query.limit);
+    const botVisits = await getBotVisits(limit);
+    res.status(200).json(botVisits);
   } catch (error) {
     console.error(error);
     res.statusMessage = `Something went wrong ${error}`;

diff --git a/src/server/botd-firewall/botVisitDatabase.ts b/src/server/botd-firewall/botVisitDatabase.ts
@@ -1,4 +1,4 @@
-import { Attributes, DataTypes, InferAttributes, InferCreationAttributes, Model } from 'sequelize';
+import { Attributes, DataTypes, FindOptions, InferAttributes, InferCreationAttributes, Model } from 'sequelize';
 import { sequelize } from '../server';
 import { EventResponseBotData } from '../../shared/types';
 
@@ -62,4 +62,12 @@ export const saveBotVisit = async (botData: EventResponseBotData, visitorId: str
   });
 };
 
-export const getBotVisits = async () => await BotVisitDbModel.findAll({ order: [['timestamp', 'DESC']] });
+export const getBotVisits = async (limit?: number) => {
+  const options: FindOptions = {
+    order: [['timestamp', 'DESC']],
+  };
+  if (limit) {
+    options.limit = limit;
+  }
+  return await BotVisitDbModel.findAll(options);
+};