Upload app.py

app.py

@@ -33,6 +33,35 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+@tool
+def subfinder_tool(domain: str, subdomains: str) -> str:
+    """A tool that checks which subdomains of a given domain exist by performing DNS lookups.
+    Args:
+        domain: The main domain (e.g., 'example.com').
+        subdomains: A comma-separated list of subdomains to check (e.g., 'www,mail,blog').
+    """
+    valid_subdomains = []
+    subdomain_list = subdomains.split(',')
+    
+    for sub in subdomain_list:
+        sub = sub.strip()
+        if not sub:
+            continue
+        full_domain = f"{sub}.{domain}"
+        try:
+            # Perform DNS lookup
+            socket.gethostbyname(full_domain)
+            valid_subdomains.append(full_domain)
+        except socket.gaierror:
+            continue  # If domain doesn't resolve
+        except Exception as e:
+            return f"Error checking {full_domain}: {str(e)}"
+    
+    if valid_subdomains:
+        return f"Valid subdomains: {', '.join(valid_subdomains)}"
+    else:
+        return "No valid subdomains found"
+
 
 final_answer = FinalAnswerTool()