import httpx
from .exceptions import TRIGGERcmdConnectionError

def trigger(token, data):
    url = "https://www.triggercmd.com/api/run/trigger"
    headers = {"Authorization": "Bearer " + token}
    r = httpx.post(url, headers=headers, data=data)
    return r

def list(token):
    url = "https://www.triggercmd.com/api/command/simplelist"
    headers = {"Authorization": "Bearer " + token}
    r = httpx.get(url, headers=headers)
    return r

# For Home Assistant integration:
async def async_trigger(token, data, httpx_client):
    url = "https://www.triggercmd.com/api/run/trigger"
    headers = {"Authorization": "Bearer " + token}
    r = await httpx_client.post(url, headers=headers, data=data)
    return r

async def async_list(token, httpx_client):
    url = "https://www.triggercmd.com/api/command/simplelist"
    headers = {"Authorization": "Bearer " + token}
    r = await httpx_client.get(url, headers=headers)
    return r
    
async def async_connection_test(token, httpx_client):
    url = "https://www.triggercmd.com/api/ha/connection_test"
    headers = {"Authorization": "Bearer " + token}
    try:
        r = await httpx_client.get(url, headers=headers)
    except Exception as e:
        raise TRIGGERcmdConnectionError
    else:
        return r.status_code