refactor: rename text_ascii to ascii and update references; rename handlers to handler and minor formatings

Author: ayanrajpoot10

bugscanx/__init__.py

@@ -33,7 +33,7 @@ def banner():
     """)
 
 
-def text_ascii(text, color="bold magenta", indentation=2):
+def ascii(text, color="bold magenta", indentation=2):
     clear_screen()
     ascii_banner = figlet.renderText(text)
     shifted_banner = "\n".join((" " * indentation) + line 

bugscanx/handlers.py bugscanx/handler.pybugscanx/handlers.py renamed to bugscanx/handler.py

@@ -1,6 +1,6 @@
 import sys
 from rich import print
-from bugscanx import banner, text_ascii, handlers
+from . import banner, ascii, handler
 
 
 MENU_OPTIONS = {
@@ -34,9 +34,9 @@ def main():
             if choice == '0':
                 return
 
-            text_ascii(MENU_OPTIONS[choice][0])
+            ascii(MENU_OPTIONS[choice][0])
             try:
-                getattr(handlers, f'run_{choice}')()
+                getattr(handler, f'run_{choice}')()
                 print("\n[yellow] Press Enter to continue...", end="")
                 input()
             except KeyboardInterrupt:

bugscanx/main.py

@@ -9,7 +9,7 @@
 from rich.padding import Padding
 from rich.progress import Progress, TimeElapsedColumn
 from bugscanx.utils.prompts import get_input, get_confirm, clear_screen
-from bugscanx import text_ascii
+from bugscanx import ascii
 
 
 def read_lines(file_path):
@@ -287,7 +287,7 @@ def main():
 
         action = options.get(choice)
         if not action:
-            text_ascii("FILE TOOLKIT")
+            ascii("FILE TOOLKIT")
             continue
 
         desc, func, color = action
@@ -304,4 +304,4 @@ def main():
         except KeyboardInterrupt:
             pass
         finally:
-            text_ascii("FILE TOOLKIT")
+            ascii("FILE TOOLKIT")

bugscanx/modules/others/file_toolkit.py

@@ -2,10 +2,8 @@
 import socket
 import http.client
 from concurrent.futures import ThreadPoolExecutor, as_completed
-
 import requests
 from rich import print
-
 from bugscanx.utils.prompts import get_input
 
 

bugscanx/modules/others/host_info.py

@@ -172,7 +172,7 @@ def main():
                     manager.restart_application()
         else:
             manager.console.print(
-                f"[yellow] Stable update available: {version_info.current_version} → {version_info.latest_stable}"
+                f"[yellow] Update available: {version_info.current_version} → {version_info.latest_stable}"
             )
             if not get_confirm(" Update now"):
                 return

bugscanx/modules/others/update.py

@@ -1,5 +1,26 @@
+import ipaddress
+from rich import print
 from bugscanx.utils.prompts import get_input, get_confirm
-from bugscanx.utils.cidr import read_cidrs_from_file
+
+
+def read_cidrs_from_file(filepath):
+    valid_cidrs = []
+    try:
+        with open(filepath, 'r') as file:
+            for line in file:
+                line = line.strip()
+                if not line:
+                    continue
+                try:
+                    ipaddress.ip_network(line, strict=False)
+                    valid_cidrs.append(line)
+                except ValueError:
+                    pass
+            
+        return valid_cidrs
+    except Exception as e:
+        print(f"[bold red]Error reading file: {e}[/bold red]")
+        return []
 
 
 def get_cidr_ranges_from_input(cidr_input):
@@ -26,7 +47,7 @@ def get_host_input():
         cidr = get_input("Enter CIDR range(s)", validators="cidr", mandatory=False)
         if not cidr:
             cidr_file = get_input(
-                "Enter CIDR file", input_type="file")
+                "Enter CIDR file", input_type="file", validators="file")
             cidr = read_cidrs_from_file(cidr_file) if cidr_file else None
         return None, cidr
     return filename, None
@@ -84,14 +105,14 @@ def get_input_proxy():
     if filename is None and cidr is None:
         return None, None, None
 
-    target_url = get_input("Enter target url", default="in1.wstunnel.site")
+    target_url = get_input("Enter target url", default="in1.wstunnel.site", validators="required")
     default_payload = (
         "GET / HTTP/1.1[crlf]"
         "Host: [host][crlf]"
         "Connection: Upgrade[crlf]"
         "Upgrade: websocket[crlf][crlf]"
     )
-    payload = get_input("Enter payload", default=default_payload)
+    payload = get_input("Enter payload", default=default_payload, validators="required")
     port_list = get_input("Enter port(s)", validators="number", default="80").split(',')
     output, threads = get_common_inputs()
 
@@ -139,15 +160,15 @@ def get_input_proxy2():
         transformer=lambda result: ', '.join(result) if isinstance(result, list) else result
     )
 
-    proxy = get_input("Enter proxy", instruction="(proxy:port)")
+    proxy = get_input("Enter proxy", instruction="(proxy:port)", validators="required")
 
     use_auth = get_confirm(" Use proxy authentication?")
     proxy_username = None
     proxy_password = None
 
     if use_auth:
-        proxy_username = get_input("Enter proxy username")
-        proxy_password = get_input("Enter proxy password")
+        proxy_username = get_input("Enter proxy username", validators="required")
+        proxy_password = get_input("Enter proxy password", validators="required")
 
     if cidr:
         try:

bugscanx/modules/scanners/host_scanner.py

@@ -85,6 +85,7 @@ def main():
         ips.extend(process_file(file_path))
         default_output = f"{file_path.rsplit('.', 1)[0]}_domains.txt"
 
-    output_file = get_input("Enter output filename", default=default_output)
+    output_file = get_input("Enter output filename", default=default_output, validators="required")
+    print()
     iplookup = IPLookup()
     iplookup.run(ips, output_file)

bugscanx/modules/scrapers/iplookup/iplookup.py

@@ -91,6 +91,7 @@ def main():
             domains = [d.strip() for d in f if DomainValidator.is_valid_domain(d.strip())]
         default_output = f"{file_path.rsplit('.', 1)[0]}_subdomains.txt"
 
-    output_file = get_input("Enter output filename", default=default_output)
+    output_file = get_input("Enter output filename", default=default_output, validators="required")
+    print()
     subfinder = SubFinder()
     subfinder.run(domains, output_file, sources)