updated inspector tool

Author: ThatSINEWAVE

Tool-05-Database-Field-Inspector.py

@@ -2,9 +2,47 @@
 import datetime
 import os
 import re
+import requests
+import sys
+import importlib.util
+
+
+def load_servers_js(file_path):
+    """Load the servers.js file and extract server invites"""
+    try:
+        # Read the file content
+        with open(file_path, "r", encoding="utf-8") as f:
+            content = f.read()
+
+        # Extract invite URLs using regex
+        pattern = r"https://discord\.com/invite/[a-zA-Z0-9]+"
+        invites = re.findall(pattern, content)
+        return invites
+    except Exception as e:
+        print(f"Error loading servers.js: {e}")
+        return []
+
+
+def get_server_member_count(invite_code):
+    """Get member count for a Discord server using its invite code"""
+    try:
+        invite_code = invite_code.split("/")[-1]  # Extract just the code part
+        url = f"https://discord.com/api/v9/invites/{invite_code}?with_counts=true"
+        headers = {
+            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
+        }
+        response = requests.get(url, headers=headers)
+        data = response.json()
+
+        if "approximate_member_count" in data:
+            return data["approximate_member_count"]
+        return 0
+    except Exception as e:
+        print(f"Error fetching member count for {invite_code}: {e}")
+        return 0
 
 
-def process_json(json_file, output_file, date_str):
+def process_json(json_file, output_file, date_str, total_protected_members=0):
     with open(json_file, "r", encoding="utf-8") as f:
         data = json.load(f)
 
@@ -33,6 +71,8 @@ def process_json(json_file, output_file, date_str):
         f.write(f'<div align="center">\n\n')
         f.write(f"# Database Inspection - {date_str}\n\n")
         f.write(f"## Total Cases: {case_count}\n\n")
+        if total_protected_members > 0:
+            f.write(f"## Protected Members: {total_protected_members:,}\n\n")
         f.write(f"</div>\n\n")
 
         for field, values in unique_values.items():
@@ -62,7 +102,7 @@ def append_other_counts(output_file, counts_dict):
 
 
 def update_readme(
-    readme_path, date_str, inspection_filename, case_count, additional_counts
+        readme_path, date_str, inspection_filename, case_count, additional_counts, total_protected_members=0
 ):
     with open(readme_path, "r", encoding="utf-8") as f:
         content = f.read()
@@ -71,12 +111,14 @@ def update_readme(
         f"- **{name}**: {count} entries" for name, count in additional_counts.items()
     )
 
+    protected_members_line = f"**Protected Members**: {total_protected_members:,}" if total_protected_members > 0 else ""
+
     new_section = f"""<!-- INSPECTION-START -->
 ## Latest Database Inspection - {date_str}
 
 **Inspection File**: [`{inspection_filename}`](Inspection-Database/{inspection_filename})  
-**Total Cases**: {case_count}
-
+- **Total Cases**: {case_count}
+- **{protected_members_line}**
 {additional_lines}
 <!-- INSPECTION-END -->"""
 
@@ -90,7 +132,7 @@ def update_readme(
             content += "\n\n" + new_section
         else:
             content = (
-                content[:insert_point] + new_section + "\n\n" + content[insert_point:]
+                    content[:insert_point] + new_section + "\n\n" + content[insert_point:]
             )
 
     with open(readme_path, "w", encoding="utf-8") as f:
@@ -106,18 +148,28 @@ def update_readme(
 
     json_file = "../Database-Files/Edit-Database/Compromised-Discord-Accounts.json"
     readme_path = "../Database-Files/README.md"
+    servers_js_path = "../thedartproject.github.io/assets/js/servers.js"  # Path to servers.js
+
+    # Get server invites and count members
+    total_protected_members = 0
+    invite_links = load_servers_js(servers_js_path)
+
+    print(f"Found {len(invite_links)} server invites in servers.js")
+    for invite in invite_links:
+        members = get_server_member_count(invite)
+        print(f"Server invite {invite}: {members} members")
+        total_protected_members += members
+
+    print(f"Total protected members: {total_protected_members}")
 
     # Count primary file
-    case_count = process_json(json_file, inspection_path, today)
+    case_count = process_json(json_file, inspection_path, today, total_protected_members)
 
     # Count entries from additional files
     additional_counts = {
         "Discord IDs": count_entries_from_file(
             "../Database-Files/Filter-Database/Discord-IDs.json"
         ),
-        "Malicious URLs": count_entries_from_file(
-            "../Database-Files/Filter-Database/Malicious-URLs.json"
-        ),
         "Discord Servers": count_entries_from_file(
             "../Database-Files/Filter-Database/Discord-Servers.json"
         ),
@@ -128,5 +180,10 @@ def update_readme(
 
     append_other_counts(inspection_path, additional_counts)
     update_readme(
-        readme_path, today, inspection_filename, case_count, additional_counts
-    )
+        readme_path,
+        today,
+        inspection_filename,
+        case_count,
+        additional_counts,
+        total_protected_members
+    )