Files
riposte-marketplace/integrations/mock-edr-s1/scripts/incident_action.py
T
Guillaume BOURGEOIS 7e96048446 feat(mock-edr-s1): EDR incident integration from OpenAPI spec
Built from the published OpenAPI spec for mock instance s1 (type: edr).
Incident ingestion (list_incidents) with since/after_id paging and an OCSF
mapper + 'Mock EDR Incident' default type, plus an acknowledge/resolve/dismiss
incident action. X-API-Key auth; the instance path segment is configurable.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
2026-06-27 16:00:04 +02:00

41 lines
1.4 KiB
Python

import json, os, sys, urllib.request, urllib.parse, urllib.error
def _cfg():
s = json.loads(os.environ.get("INTEGRATION_SECRETS", "{}"))
base = str(s.get("base_url") or "https://mockprod.riposte-labs.com").rstrip("/")
instance = str(s.get("instance") or "s1").strip("/")
headers = {"X-API-Key": s.get("api_key", ""), "Content-Type": "application/json", "Accept": "application/json"}
return base, instance, 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():
_, instance, _ = _cfg()
inp = _inputs()
inc_id = urllib.parse.quote(str(inp.get("id", "")), safe="")
body = {"action": inp.get("action")}
print(json.dumps(request("POST", "api/" + instance + "/incidents/" + inc_id + "/actions", body=body)))
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)