refactor(dns_records): simplify DNS resolution by removing custom nameserver support and unused record types

Author: ayanrajpoot10

bugscanx/modules/others/dns_records.py

@@ -1,39 +1,17 @@
-import os
-from typing import List, Optional
-
 from dns import resolver as dns_resolver
 from rich import print
 
-from bugscanx.utils.common import get_confirm, get_input
-
-
-def configure_resolver(custom_nameservers: Optional[List[str]] = None):
-    dns_obj = dns_resolver.Resolver()
-
-    if custom_nameservers and len(custom_nameservers) > 0:
-        dns_obj.nameservers = custom_nameservers
-        print(
-            f"[yellow]Using custom DNS servers: "
-            f"{', '.join(custom_nameservers)}[/yellow]"
-        )
-        return dns_obj
-
-    is_termux = "com.termux" in os.environ.get(
-        "PREFIX", ""
-    ) or os.path.exists("/data/data/com.termux")
+from bugscanx.utils.common import get_input
 
-    if is_termux:
-        dns_obj.nameservers = ["8.8.8.8", "8.8.4.4", "1.1.1.1"]
 
-    return dns_obj
+def configure_resolver():
+    return dns_resolver.Resolver()
 
 
-def resolve_and_print(
-    domain: str, record_type: str, custom_nameservers: Optional[List[str]] = None
-):
+def resolve_and_print(domain, record_type):
     print(f"\n[green] {record_type} Records:[/green]")
     try:
-        dns_obj = configure_resolver(custom_nameservers)
+        dns_obj = configure_resolver()
         answers = dns_obj.resolve(domain, record_type)
         found = False
         for answer in answers:
@@ -43,20 +21,6 @@ def resolve_and_print(
                     f"[cyan]- {answer.exchange} "
                     f"(priority: {answer.preference})[/cyan]"
                 )
-            elif record_type == "SOA":
-                print(f"[cyan]- Primary NS: {answer.mname}")
-                print(f"  Responsible: {answer.rname}")
-                print(f"  Serial: {answer.serial}")
-                print(f"  Refresh: {answer.refresh}")
-                print(f"  Retry: {answer.retry}")
-                print(f"  Expire: {answer.expire}")
-                print(f"  Minimum TTL: {answer.minimum}[/cyan]")
-            elif record_type == "SRV":
-                print(
-                    f"[cyan]- Priority: {answer.priority}, "
-                    f"Weight: {answer.weight}"
-                )
-                print(f"  Port: {answer.port}, Target: {answer.target}[/cyan]")
             else:
                 print(f"[cyan]- {answer.to_text()}[/cyan]")
         if not found:
@@ -67,7 +31,7 @@ def resolve_and_print(
         print(f"[yellow] Error fetching {record_type} record[/yellow]")
 
 
-def nslookup(domain: str, custom_nameservers: Optional[List[str]] = None):
+def nslookup(domain):
     print(f"[cyan]\n Performing NSLOOKUP for: {domain}[/cyan]")
 
     record_types = [
@@ -77,41 +41,16 @@ def nslookup(domain: str, custom_nameservers: Optional[List[str]] = None):
         "MX",
         "NS",
         "TXT",
-        "SOA",
-        "PTR",
-        "SRV",
-        "CAA",
-        "DNSKEY",
-        "TLSA",
     ]
 
     for record_type in record_types:
-        resolve_and_print(domain, record_type, custom_nameservers)
+        resolve_and_print(domain, record_type)
 
 
 def main():
-    domain = get_input("Enter the domain to lookup")
-    if not domain:
-        print("[red] Please enter a valid domain.[/red]")
-        return
-
-    use_custom_dns = get_confirm(" Want to use custom DNS servers?")
-    custom_nameservers = None
-
-    if use_custom_dns:
-        nameservers_input = get_input(
-            "Enter DNS servers separated by commas (e.g., 8.8.8.8,1.1.1.1)"
-        )
-        if nameservers_input:
-            custom_nameservers = [
-                server.strip() for server in nameservers_input.split(",")
-            ]
-            print(
-                f"[cyan] Will use custom DNS servers: "
-                f"{', '.join(custom_nameservers)}[/cyan]"
-            )
+    domain = get_input("Enter target")
 
     try:
-        nslookup(domain, custom_nameservers)
+        nslookup(domain)
     except Exception as e:
         print(f"[red] An error occurred during DNS lookup: {str(e)}[/red]")