Auto-Create & Associate Company from Domain (Python)

def main(input_data, hs_client, http_client, secret, console):
    raw = input_data[0] if isinstance(input_data, list) and input_data else (input_data or {})
    contact_id = raw.get('contact_id') or raw.get('contactId') or raw.get('objectId') or raw.get('hs_object_id')
    email = raw.get('email')

    freemail = {
        'gmail.com', 'yahoo.com', 'hotmail.com', 'outlook.com', 'icloud.com',
        'aol.com', 'protonmail.com', 'zoho.com', 'mail.com', 'gmx.com'
    }

    if contact_id and not email:
        c = hs_client.crm.contacts.basic_api.get_by_id(contact_id, properties=['email'])
        email = c.properties.get('email')

    # Trigger-agnostic fallback: Scan recent contacts
    if not email:
        recent = hs_client.crm.contacts.basic_api.get_page(limit=25, properties=['email'])
        enriched = 0
        for c in (recent.results or []):
            c_email = c.properties.get('email')
            if not c_email or '@' not in c_email:
                continue
            c_domain = c_email.split('@')[1].lower().strip()
            if c_domain in freemail:
                continue
            existing = hs_client.crm.companies.basic_api.get_page(
                limit=1,
                properties=['name', 'domain'],
                filters=[{"value": c_domain, "propertyName": "domain", "operator": "EQ"}]
            )
            if existing.results:
                comp_id = existing.results[0].id
            else:
                c_name = c_domain.split('.')[0].capitalize()
                new_comp = hs_client.crm.companies.basic_api.create(properties={"name": c_name, "domain": c_domain})
                comp_id = new_comp.id
            hs_client.crm.associations.v4.basic_api.create(
                from_object_type='contacts',
                object_id=c.id,
                to_object_type='companies',
                to_object_id=comp_id,
                types=[{"associationCategory": "HUBSPOT_DEFINED", "associationTypeId": 1}]
            )
            enriched += 1
        return {"associated": enriched > 0, "message": f"Trigger-agnostic scan: enriched {enriched} contact(s)"}

    domain = email.split('@')[1].lower().strip()
    if domain in freemail:
        return {"associated": False, "domain": domain, "message": f"Skipped freemail: {domain}"}

    existing = hs_client.crm.companies.basic_api.get_page(
        limit=1,
        properties=['name', 'domain'],
        filters=[{"value": domain, "propertyName": "domain", "operator": "EQ"}]
    )

    if existing.results:
        company_id = existing.results[0].id
    else:
        name = domain.split('.')[0].capitalize()
        company = hs_client.crm.companies.basic_api.create(properties={"name": name, "domain": domain})
        company_id = company.id

    if contact_id:
        hs_client.crm.associations.v4.basic_api.create(
            from_object_type='contacts',
            object_id=contact_id,
            to_object_type='companies',
            to_object_id=company_id,
            types=[{"associationCategory": "HUBSPOT_DEFINED", "associationTypeId": 1}]
        )

    return {"associated": True, "companyId": company_id, "domain": domain, "message": f"Linked to company {company_id}"}