import json, os, sys, urllib.request, urllib.parse, urllib.error


def request(method, url, headers, body=None):
    data = json.dumps(body).encode("utf-8") if body is not None else None
    req = urllib.request.Request(url, data=data, headers=headers, method=method)
    with urllib.request.urlopen(req, timeout=60) as resp:
        raw = resp.read()
        return json.loads(raw) if raw else {}


def csv(v):
    return [x.strip() for x in str(v or "").split(",") if x.strip()]


def flag(v):
    return str(v).lower() in ("true", "1", "yes")


def main():
    inputs = json.loads(os.environ.get("INTEGRATION_INPUTS", "{}"))
    # PowerQuery runs against the Singularity Data Lake, not the management console.
    sdl_url = str(inputs.get("singularity_xdr_url", "")).rstrip("/")
    if not sdl_url.startswith("https://"):
        raise ValueError("singularity_xdr_url must start with https://")
    headers = {
        "Authorization": "Bearer " + str(inputs.get("singularity_xdr_api_key", "")),
        "Accept": "application/json",
        "Content-Type": "application/json",
    }
    payload = {"query": inputs.get("query")}
    if inputs.get("start_time"):
        payload["startTime"] = inputs["start_time"]
    if inputs.get("end_time"):
        payload["endTime"] = inputs["end_time"]
    if inputs.get("priority"):
        payload["priority"] = inputs["priority"]
    if inputs.get("team_emails"):
        payload["teamEmails"] = csv(inputs["team_emails"])
    if inputs.get("recurring") is not None:
        payload["recurring"] = flag(inputs["recurring"])
    print(json.dumps(request("POST", sdl_url + "/api/powerQuery", headers, payload)))


try:
    main()
except urllib.error.HTTPError as e:
    print(json.dumps({"error": "HTTP " + str(e.code), "detail": e.read().decode("utf-8", "replace")}))
    sys.exit(1)
except Exception as e:
    print(json.dumps({"error": str(e)}))
    sys.exit(1)
