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

# Map a target status to the Sekoia workflow transition name.
TRANSITIONS = {"Ongoing": "Validate", "Acknowledged": "Acknowledge", "Rejected": "Reject", "Closed": "Close"}


def _cfg():
    s = json.loads(os.environ.get("INTEGRATION_SECRETS", "{}"))
    base = str(s.get("url") or "https://api.sekoia.io").rstrip("/")
    headers = {"Authorization": "Bearer " + s.get("api_key", ""), "Content-Type": "application/json", "Accept": "application/json"}
    return base, headers


def _inputs():
    return json.loads(os.environ.get("INTEGRATION_INPUTS", "{}"))


def request(method, path, body=None):
    base, headers = _cfg()
    data = json.dumps(body).encode("utf-8") if body is not None else None
    req = urllib.request.Request(base + "/" + path.lstrip("/"), data=data, headers=headers, method=method)
    with urllib.request.urlopen(req, timeout=90) as r:
        raw = r.read()
        return json.loads(raw) if raw else {}


def run():
    inp = _inputs()
    alert_id = urllib.parse.quote(inp.get("id", ""), safe="")
    status = inp.get("status", "")
    transition = TRANSITIONS.get(status)
    workflow = request("GET", "/v1/sic/alerts/" + alert_id + "/workflow")
    action = next((a for a in workflow.get("actions", []) if a.get("name") == transition), None)
    if not action:
        print(json.dumps({"error": "No workflow transition available for status '" + status + "'.",
                          "available": [a.get("name") for a in workflow.get("actions", [])]}))
        sys.exit(1)
    body = {"action_uuid": action["id"]}
    if inp.get("comment"):
        body["comment"] = inp["comment"]
    result = request("PATCH", "/v1/sic/alerts/" + alert_id + "/workflow", body=body)
    print(json.dumps({"updated": True, "status": status, "result": result}))


try:
    run()
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)
