from pydantic import BaseModel, HttpUrl
from typing import List

class DiscoverResponse(BaseModel):
    count: int
    domains: List[str]

class BulkScrapeRequest(BaseModel):
    urls: List[HttpUrl]

class LeadData(BaseModel):
    domain: str
    emails: List[str]
    status: str

class BulkScrapeResponse(BaseModel):
    total_processed: int
    successful_leads: int
    results: List[LeadData]


class ScrapeToDBRequest(BaseModel):
    email_limit: int = 1000  # Number of emails to scrape before stopping
    domain_limit: int = 100  # Number of domains to scrape (default 100, can go up to 10000)
    category: str = "WEB"

    class Config:
        from_attribute = True


class ScrapeToDBResponse(BaseModel):
    total_processed: int
    successful_leads: int
    total_emails_found: int
    total_emails_saved: int
    duplicates_skipped: int
    errors: int
    results: List[LeadData]