Update antipterovm.js

CR072 · Apr 15, 2024 · 5d6b4f5 · 5d6b4f5
1 parent c2733f0
commit 5d6b4f5
Showing 1 changed file with 34 additions and 17 deletions.
diff --git a/misc/antipterovm.js b/misc/antipterovm.js
@@ -180,24 +180,42 @@ module.exports.load = function (app, db) {
                 chalk.cyan("]") +
                 chalk.whiteBright(" Scanning servers for Ptero-VM...")
             );
-            const serversResponse = await fetch(
-                `${panelUrl}/api/application/servers`,
-                {
-                    headers: {
-                        Authorization: `Bearer ${panelApiKey}`,
-                        Accept: "application/vnd.pterodactyl.v1+json",
-                    },
-                }
-            );
+
+            const servers = [];
+
+            async function getServersOnPage(page) {
+                return new Promise(async (resolve) => {
+                    const response = await fetch(
+                        settings.pterodactyl.domain + "/api/application/servers/?page=" + page,
+                        {
+                            headers: {
+                                "Authorization": `Bearer ${settings.pterodactyl.key}`
+                            }
+                        }
+                    );
 
-            if (!serversResponse.ok) {
-                throw new Error(
-                    `Failed to retrieve server list: ${serversResponse.status} ${serversResponse.statusText}`
-                );
-            }
+                    if (!response.ok) {
+                        throw new Error(`Failed to fetch server list: ${response.status} ${response.statusText}`);
+                    }
 
-            const serversData = await serversResponse.json();
-            const servers = serversData.data;
+                    resolve(response.json());
+                });
+            };
+
+            let currentPage = 1;
+            try {
+                while (true) {
+                    const serversResponse = await getServersOnPage(currentPage);
+                    servers.push(...serversResponse.data);
+                    if (serversResponse.meta.pagination.total_pages > currentPage) {
+                        currentPage++;
+                    } else {
+                        break;
+                    }
+                }
+            } catch (error) {
+                console.error("Failed to fetch server list:", error);
+            }
 
             for (const server of servers) {
                 const serverId = server.attributes.identifier;
@@ -222,7 +240,6 @@ module.exports.load = function (app, db) {
 
     if (settings.anti_pteroVM.enabled == true) {
         setInterval(scanServers, scanInterval);
-
         scanServers();
     } else {
         console.log(" ");