Update plugins/antiban.js

plugins/antiban.js

@@ -3,11 +3,10 @@ import axios from 'axios';
 import { Database } from '../database/database.js';
 import { predictCreationDate } from '../lib/create-date.js';
 import { authenticateApiKey, apiLimiter } from '../middleware/midware.js';
+import { Job } from '../models.js';
 const AntibanRoutes = express.Router();
 import * as uuid from 'uuid';
 
-const myUUID = uuid.v4();
-
 const protectedUsers = [6477856957, 1191668125, 1448273246, 1054295664, 6444305696]; 
 
 const TelegramUser = async (user_id) => {
@@ -232,15 +231,35 @@ AntibanRoutes.get("/api/v1/user/story-dl", authenticateApiKey, apiLimiter, async
     if (!Links) {
         return res.status(400).json({ error: "Invalid or missing link" });
     }
-
-    try {
-        const result = await TelegramUserStoryDL(Links);
-        res.json(result);
-    } catch (error) {
-        res.status(500).json({ error: "Failed to fetch user info" });
-    }
+    const job_id = uuid.v4();
+    const newJob = new Job({ job_id, status: "pending" });
+    await newJob.save();
+    setTimeout(async () => {
+        try {
+            const result = await TelegramUserStoryDL(Links);
+            const story_bytes = result.download;
+
+            await Job.findOneAndUpdate(
+                { job_id },
+                { status: "completed", story_bytes }
+            );
+        } catch (err) {
+            await Job.findOneAndUpdate({ job_id }, { status: "failed" });
+        }
+    }, 10000);
+    res.json({ message: "Download started", job_id });
 });
 
+app.get("/api/v1/user/story/:job_id", authenticateApiKey, apiLimiter, async (req, res) => {
+    const job = await Job.findOne({ job_id: req.params.job_id });
+
+    if (!job) return res.status(404).json({ error: "Job not found" });
+    res.json({
+        job_id: job.job_id,
+        status: job.status,
+        story_bytes: job.story_bytes
+    });
+});
 
 /**
  * @swagger