Adding Throttle Middleware

lib/src/exception/http_exception.dart

@@ -1,4 +1,5 @@
-import 'package:vania/vania.dart';
+import '../http/response/response.dart';
+import 'base_http_exception.dart';
 
 class HttpException extends BaseHttpException {
   HttpException(

lib/src/exception/internal_server_error.dart

@@ -1,5 +1,4 @@
 import 'dart:io';
-
 import 'base_http_exception.dart';
 
 class InternalServerError extends BaseHttpException {

lib/src/exception/not_found_exception.dart

@@ -1,6 +1,6 @@
 import 'dart:io';
-
-import 'package:vania/vania.dart';
+import '../http/response/response.dart';
+import 'base_http_exception.dart';
 
 class NotFoundException extends BaseHttpException {
   NotFoundException({

lib/src/exception/throttle_exception.dart

@@ -0,0 +1,10 @@
+import '../http/response/response.dart';
+import 'base_http_exception.dart';
+
+class ThrottleException extends BaseHttpException {
+  ThrottleException(
+      {required super.message,
+      required super.code,
+      super.responseType = ResponseType.json,
+      super.errorCode = 'Rate limiting'});
+}

lib/src/route/middleware/throttle.dart

@@ -0,0 +1,39 @@
+import 'dart:io';
+
+import 'package:vania/vania.dart';
+
+import '../../exception/throttle_exception.dart';
+import '../throttle_requests.dart';
+
+class Throttle extends Middleware {
+  final int maxAttempts;
+  final Duration duration;
+
+  Throttle({
+    this.maxAttempts = 6,
+    this.duration = const Duration(seconds: 60),
+  }) {
+    throttle = ThrottleRequests(maxAttempts: maxAttempts, duration: duration);
+  }
+
+  late ThrottleRequests throttle;
+
+  @override
+  Future handle(Request req) async {
+    final clientIp = req.ip;
+    if (clientIp == null) {
+      req.response.statusCode = HttpStatus.internalServerError;
+      req.response.write('Error determining client IP');
+      await req.response.close();
+      return;
+    }
+
+    if (!throttle.request(clientIp)) {
+      throw ThrottleException(
+        message: 'Too Many Requests.',
+        code: HttpStatus.tooManyRequests,
+      );
+    }
+    next?.handle(req);
+  }
+}

lib/src/route/throttle_requests.dart

@@ -0,0 +1,33 @@
+class ThrottleRequests {
+  final int maxAttempts;
+  final Duration duration;
+  final Map<String, _RateLimit> _limits = {};
+
+  ThrottleRequests({required this.maxAttempts, required this.duration});
+
+  bool request(String ip) {
+    final currentTime = DateTime.now();
+
+    _limits.putIfAbsent(ip, () => _RateLimit(0, currentTime));
+    final limit = _limits[ip]!;
+
+    if (currentTime.difference(limit.windowStart).compareTo(duration) >= 0) {
+      limit.count = 0;
+      limit.windowStart = currentTime;
+    }
+
+    if (limit.count < maxAttempts) {
+      limit.count++;
+      return true;
+    } else {
+      return false;
+    }
+  }
+}
+
+class _RateLimit {
+  int count;
+  DateTime windowStart;
+
+  _RateLimit(this.count, this.windowStart);
+}

lib/vania.dart

@@ -20,6 +20,7 @@ export 'src/websocket/websocket_event.dart';
 
 export 'src/route/router.dart';
 export 'src/route/route.dart';
+export 'src/route/middleware/throttle.dart';
 
 export 'src/cache/cache_driver.dart';
 export 'src/cache/cache.dart';