From 1a85c2571690ba592ac5183d5eadaf9846fe532b Mon Sep 17 00:00:00 2001
From: Factiven <factiven.org@gmail.com>
Date: Mon, 25 Sep 2023 00:44:40 +0700
Subject: Update v4.1.0 (#79)

* Update v4.1.0

* Update pages/_app.js
---
 pages/api/v2/source/index.js | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'pages/api/v2/source/index.js')

diff --git a/pages/api/v2/source/index.js b/pages/api/v2/source/index.js
index 51ac5ec..74a63cb 100644
--- a/pages/api/v2/source/index.js
+++ b/pages/api/v2/source/index.js
@@ -1,3 +1,4 @@
+import { rateLimiterRedis, redis } from "@/lib/redis";
 import axios from "axios";
 
 const CONSUMET_URI = process.env.API_URI;
@@ -33,6 +34,17 @@ export default async function handler(req, res) {
     return res.status(405).json({ message: "Method not allowed" });
   }
 
+  if (redis) {
+    try {
+      const ipAddress = req.socket.remoteAddress;
+      await rateLimiterRedis.consume(ipAddress);
+    } catch (error) {
+      return res.status(429).json({
+        error: `Too Many Requests, retry after ${error.msBeforeNext / 1000}`,
+      });
+    }
+  }
+
   const { source, providerId, watchId, episode, id, sub = "sub" } = req.body;
 
   if (source === "anify") {
-- 
cgit v1.2.3