#!/usr/bin/env python3
"""
HopSaaS Automated CRM Migration Engine
Exports Salesforce Contacts/Accounts & upserts cleanly into HubSpot CRM v3 API.
"""
import os
import json
import requests

SF_ACCESS_TOKEN = os.getenv("SF_ACCESS_TOKEN", "00D5g00000...")
SF_INSTANCE_URL = os.getenv("SF_INSTANCE_URL", "https://your-org.my.salesforce.com")
HUBSPOT_TOKEN = os.getenv("HUBSPOT_ACCESS_TOKEN", "pat-na1-...")

SF_HEADERS = {"Authorization": f"Bearer {SF_ACCESS_TOKEN}"}
HS_HEADERS = {
    "Authorization": f"Bearer {HUBSPOT_TOKEN}",
    "Content-Type": "application/json"
}

def export_salesforce_contacts():
    query = "SELECT Id, FirstName, LastName, Email, Title, Account.Name FROM Contact WHERE Email != null LIMIT 500"
    url = f"{SF_INSTANCE_URL}/services/data/v59.0/query/?q={query}"
    res = requests.get(url, headers=SF_HEADERS)
    res.raise_for_status()
    return res.json().get("records", [])

def migrate_to_hubspot(contacts):
    print(f"[*] Migrating {len(contacts)} contacts into HubSpot v3 API...")
    batch_inputs = []
    for c in contacts:
        batch_inputs.append({
            "properties": {
                "email": c.get("Email"),
                "firstname": c.get("FirstName", ""),
                "lastname": c.get("LastName", ""),
                "jobtitle": c.get("Title", ""),
                "hs_lead_status": "OPEN",
                "salesforce_legacy_id": c.get("Id")
            }
        })

    # HubSpot Batch Upsert (Max 100 per call)
    for i in range(0, len(batch_inputs), 100):
        chunk = batch_inputs[i:i + 100]
        res = requests.post(
            "https://api.hubapi.com/crm/v3/objects/contacts/batch/create",
            headers=HS_HEADERS,
            json={"inputs": chunk}
        )
        if res.status_code in [200, 201, 207]:
            print(f"  [+] Ingested chunk {i//100 + 1}: {len(chunk)} contacts mapped successfully.")
        else:
            print(f"  [!] Batch warning: {res.text}")

if __name__ == "__main__":
    records = export_salesforce_contacts()
    migrate_to_hubspot(records)
    print("[✓] Salesforce to HubSpot Contact migration completed.")