import httpx
from sqlalchemy.orm import Session
import models


async def import_contacts_from_google(access_token: str, db: Session, owner: str = None) -> int:
    """
    Import contacts from Google People API
    
    Args:
        access_token: OAuth 2.0 access token from Google
        db: Database session
        owner: Name of the person importing (e.g., 'me', 'fianc\u00e9')
        
    Returns:
        Number of contacts imported
    """
    headers = {
        "Authorization": f"Bearer {access_token}"
    }
    
    # Google People API endpoint
    url = "https://people.googleapis.com/v1/people/me/connections"
    params = {
        "personFields": "names,phoneNumbers,emailAddresses",
        "pageSize": 1000
    }
    
    imported_count = 0
    
    async with httpx.AsyncClient() as client:
        response = await client.get(url, headers=headers, params=params)
        
        if response.status_code != 200:
            raise Exception(f"Failed to fetch contacts: {response.text}")
        
        data = response.json()
        connections = data.get("connections", [])
        
        for connection in connections:
            # Extract name
            names = connection.get("names", [])
            if not names:
                continue
                
            name = names[0]
            first_name = name.get("givenName", "")
            last_name = name.get("familyName", "")
            
            if not first_name and not last_name:
                continue
            
            # Extract email
            emails = connection.get("emailAddresses", [])
            email = emails[0].get("value") if emails else None
            
            # Extract phone number
            phones = connection.get("phoneNumbers", [])
            phone_number = phones[0].get("value") if phones else None
            
            # Check if contact already exists by email OR phone number
            existing = None
            if email:
                existing = db.query(models.Guest).filter(models.Guest.email == email).first()
            if not existing and phone_number:
                existing = db.query(models.Guest).filter(models.Guest.phone_number == phone_number).first()
            
            if existing:
                # Contact exists - merge owners
                if existing.owner and owner not in existing.owner.split(","):
                    # Add current owner to existing owners
                    existing.owner = f"{existing.owner},{owner}"
                    db.add(existing)
            else:
                # Create new guest
                guest = models.Guest(
                    first_name=first_name or "Unknown",
                    last_name=last_name or "",
                    email=email,
                    phone_number=phone_number,
                    rsvp_status="pending",
                    owner=owner
                )
                db.add(guest)
                imported_count += 1
        
        db.commit()
    
    return imported_count