Update middleware/midware.js

middleware/midware.js

@@ -1,9 +1,33 @@
 import { Database } from '../database/database.js';
 import { MongoStorage } from "@canmertinyo/rate-limiter-mongo";
 import { rateLimiter } from "@canmertinyo/rate-limiter-core";
-
+import { ApiKey } from '../models.js';
 import * as config from '../config.js';
 
+const authenticateApiKeyPremium = async (req, res, next) => {
+  const apiKey = req.headers['x-api-key'];
+
+  if (!apiKey) {
+    return res.status(401).json({ error: 'Premium API Key required' });
+  }
+
+  try {
+    const keyData = await ApiKey.findOne({ key: apiKey, type: "premium" });
+
+    if (!keyData) {
+      return res.status(403).json({ error: 'Invalid or non-premium API Key' });
+    }
+
+    if (keyData.expiresAt && new Date() > keyData.expiresAt) {
+      return res.status(403).json({ error: 'Premium API Key has expired' });
+    }
+
+    next();
+  } catch (err) {
+    res.status(500).json({ error: 'Server error' });
+  }
+};
+
 const authenticateApiKey = async (req, res, next) => {
   const apiKey = req.headers['x-api-key'];
   const dbClient = new Database("AkenoXJs");
@@ -102,5 +126,6 @@ class CheckMilWare {
 export { 
   CheckMilWare,
   authenticateApiKey,
+  authenticateApiKeyPremium,
   apiLimiter
 };