import requests
from datetime import datetime

class AtonReferralsAPI:
    def __init__(self, base_url, auth_client):
        self.base_url = base_url
        self.auth_client = auth_client

    def get_all_referrals(self):
        """Retrieve all referrals."""
        token = self.auth_client.get_access_token()

        response = requests.get(
            f"{self.base_url}/referrals/",
            headers={
                "Authorization": f"Bearer {token}",
                "Content-Type": "application/json"
            }
        )
        response.raise_for_status()
        return response.json()

    def get_referral_by_id(self, trial_id, outcomes_id):
        """Get a specific referral by trial ID and outcomes ID."""
        token = self.auth_client.get_access_token()

        response = requests.get(
            f"{self.base_url}/referrals/{trial_id}/{outcomes_id}",
            headers={
                "Authorization": f"Bearer {token}",
                "Content-Type": "application/json"
            }
        )
        response.raise_for_status()
        return response.json()

    def submit_referral(self, referral_data):
        """Submit a new referral."""
        token = self.auth_client.get_access_token()

        response = requests.post(
            f"{self.base_url}/referrals/",
            headers={
                "Authorization": f"Bearer {token}",
                "Content-Type": "application/json"
            },
            json=referral_data
        )
        response.raise_for_status()
        return response.json()

    def submit_qualified_referral(self, qualified_referral_data):
        """Submit a new qualified referral."""
        token = self.auth_client.get_access_token()

        response = requests.post(
            f"{self.base_url}/referrals/qualified",
            headers={
                "Authorization": f"Bearer {token}",
                "Content-Type": "application/json"
            },
            json=qualified_referral_data
        )
        response.raise_for_status()
        return response.json()

    def get_all_qualified_referrals(self):
        """Retrieve all qualified referrals."""
        token = self.auth_client.get_access_token()

        response = requests.get(
            f"{self.base_url}/referrals/qualified",
            headers={
                "Authorization": f"Bearer {token}",
                "Content-Type": "application/json"
            }
        )
        response.raise_for_status()
        return response.json()

    def get_qualified_referral_by_external_id(self, external_source_id):
        """Get a specific qualified referral by external source ID."""
        token = self.auth_client.get_access_token()

        response = requests.get(
            f"{self.base_url}/referrals/qualified/{external_source_id}",
            headers={
                "Authorization": f"Bearer {token}",
                "Content-Type": "application/json"
            }
        )
        response.raise_for_status()
        return response.json()

# Usage example
from aton_auth import AtonHealthAuth  # Your auth implementation

auth = AtonHealthAuth()
referrals_api = AtonReferralsAPI("https://api.aton.health/api/v1", auth)

# Get all referrals
all_referrals = referrals_api.get_all_referrals()
print(f"Found {len(all_referrals)} referrals")

# Submit a new referral
new_referral = {
    "outcomesId": "12345678-1234-1234-1234-123456789012",
    "trialId": "Aton_ColonoscopyPrep_20250930",
    "versionId": "Aton_ColonoscopyPrep_20250930v2",
    "firstName": "John",
    "lastName": "Smith",
    "phoneNumber": "+1-555-987-6543",
    "email": "john.smith@example.com",
    "zipCode": "54321",
    "prescriberNPI": "0987654321",
    "referralTimestamp": datetime.now().strftime("%m/%d/%Y")
}

result = referrals_api.submit_referral(new_referral)

# Submit a qualified referral
qualified_referral = {
    "firstName": "Jane",
    "lastName": "Doe",
    "mobilePhone": "+1-555-987-6543",
    "email": "jane.doe@example.com",
    "birthday": "1975-03-20",
    "gender": "Female",
    "addressStreet1": "456 Oak Avenue",
    "addressCity": "Springfield",
    "addressState": "IL",
    "addressZip": "62701",
    "providerNpi": "0987654321",
    "socAppointmentDate": "2025-12-01T14:30:00Z",
    "colonoscopyNaive": false,
    "dateOfLastColonoscopy": "2020-05-15",
    "ibdCrohns": false,
    "externalSourceId": "DELFA-67890",
    "noteTitle": "Pre-qualified Study Patient",
    "noteDescription": "Patient has been pre-screened and qualified for colonoscopy prep study"
}

qualified_result = referrals_api.submit_qualified_referral(qualified_referral)
print("Qualified referral submitted:", qualified_result["message"])