jpmschweitzerandClaude Opus 4.5
33f31b09a3
Web scraping functionality is no longer needed. Removes: - src/web_scraper/ directory (6 files) - /web-scraper/scrape endpoint from tools controller - References from health controller endpoints list Tools controller now only contains DNS lookup functionality. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
103 lines
3.3 KiB
Python
103 lines
3.3 KiB
Python
"""
|
|
Tools Controller
|
|
|
|
Provides utility tool endpoints including:
|
|
- DNS lookups
|
|
"""
|
|
from fastapi import APIRouter, HTTPException, status
|
|
|
|
from src.controllers.base import BaseController
|
|
from src.logging_config import get_logger
|
|
from src.dns.schemas import DNSLookupRequest, DNSLookupResponse
|
|
from src.dns.service import DNSService
|
|
from src.dns.exceptions import DNSQueryError
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
class ToolsController(BaseController):
|
|
"""
|
|
Controller for utility tools
|
|
|
|
Provides endpoints for:
|
|
- DNS lookups
|
|
"""
|
|
|
|
def __init__(self):
|
|
super().__init__(prefix="/tools", tags=["Tools"])
|
|
self.dns_service = DNSService()
|
|
|
|
def create_router(self) -> APIRouter:
|
|
"""Create and configure the router"""
|
|
router = APIRouter(prefix=self.prefix, tags=self.tags)
|
|
|
|
@router.post(
|
|
"/dns/lookup",
|
|
response_model=DNSLookupResponse,
|
|
status_code=status.HTTP_200_OK,
|
|
summary="Perform DNS lookup",
|
|
description="""
|
|
Perform DNS lookups for various record types.
|
|
|
|
Uses dnspython for reliable DNS queries with support for multiple record types
|
|
and custom nameservers. Perfect for troubleshooting DNS issues and checking
|
|
domain configurations.
|
|
|
|
**Supported Record Types:**
|
|
- A: IPv4 address records
|
|
- AAAA: IPv6 address records
|
|
- MX: Mail exchange records
|
|
- TXT: Text records (SPF, DKIM, etc.)
|
|
- CNAME: Canonical name records
|
|
- NS: Nameserver records
|
|
- SOA: Start of authority records
|
|
- PTR: Pointer records (reverse DNS)
|
|
- CAA: Certification authority authorization
|
|
- SRV: Service records
|
|
|
|
**Features:**
|
|
- Custom nameserver support (e.g., 8.8.8.8, 1.1.1.1)
|
|
- Query time measurement
|
|
- Detailed error messages
|
|
|
|
**Rate Limiting:** None (internal network use only)
|
|
"""
|
|
)
|
|
async def dns_lookup(request: DNSLookupRequest) -> DNSLookupResponse:
|
|
"""
|
|
Perform DNS lookup for a domain
|
|
|
|
Args:
|
|
request: DNS lookup request with domain, record type, and optional nameserver
|
|
|
|
Returns:
|
|
DNS lookup results with records and metadata
|
|
|
|
Raises:
|
|
HTTPException: 400 for invalid queries, 500 for processing errors
|
|
"""
|
|
try:
|
|
logger.info(f"Received DNS lookup request for: {request.domain} ({request.record_type})")
|
|
result = await self.dns_service.lookup(request)
|
|
return result
|
|
|
|
except DNSQueryError as e:
|
|
logger.warning(f"DNS query error: {str(e)}")
|
|
raise HTTPException(
|
|
status_code=status.HTTP_400_BAD_REQUEST,
|
|
detail=f"DNS query failed: {str(e)}"
|
|
)
|
|
|
|
except Exception as e:
|
|
logger.error(f"Unexpected error during DNS lookup: {str(e)}", exc_info=True)
|
|
raise HTTPException(
|
|
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
|
detail="An unexpected error occurred during DNS lookup"
|
|
)
|
|
|
|
return router
|
|
|
|
|
|
# Create controller instance
|
|
tools_controller = ToolsController()
|