From 4020e13df9021b6b5df07fea1b797abf677d9347 Mon Sep 17 00:00:00 2001 From: Guillaume BOURGEOIS Date: Sun, 12 Jul 2026 14:10:25 +0200 Subject: [PATCH] feat(microsoft-purview): new Microsoft Purview DLP integration DLP alerts via Microsoft Graph Security API, 6 commands: list/get DLP alerts, update alert (status/assignee/determination), list/get incidents. Azure AD OAuth2 client-credentials auth, stdlib-only. Co-Authored-By: Claude Opus 4.8 (1M context) --- integrations/microsoft-purview/manifest.yaml | 85 ++++++++++++++++++ .../microsoft-purview/scripts/get_alert.py | 70 +++++++++++++++ .../microsoft-purview/scripts/get_incident.py | 70 +++++++++++++++ .../scripts/list_dlp_alerts.py | 69 ++++++++++++++ .../scripts/list_incidents.py | 71 +++++++++++++++ .../scripts/test_connection.py | 65 ++++++++++++++ .../microsoft-purview/scripts/update_alert.py | 89 +++++++++++++++++++ 7 files changed, 519 insertions(+) create mode 100644 integrations/microsoft-purview/manifest.yaml create mode 100644 integrations/microsoft-purview/scripts/get_alert.py create mode 100644 integrations/microsoft-purview/scripts/get_incident.py create mode 100644 integrations/microsoft-purview/scripts/list_dlp_alerts.py create mode 100644 integrations/microsoft-purview/scripts/list_incidents.py create mode 100644 integrations/microsoft-purview/scripts/test_connection.py create mode 100644 integrations/microsoft-purview/scripts/update_alert.py diff --git a/integrations/microsoft-purview/manifest.yaml b/integrations/microsoft-purview/manifest.yaml new file mode 100644 index 0000000..51c167c --- /dev/null +++ b/integrations/microsoft-purview/manifest.yaml @@ -0,0 +1,85 @@ +id: microsoft_purview +name: Microsoft Purview +version: 1.0.0 +description: "Microsoft Purview DLP alerts (via Microsoft Graph Security API) — data-loss-prevention triage: list and read DLP security alerts, update alert status/assignment, and list/read security incidents. Azure AD OAuth2 client-credentials authentication; stdlib-only, no extra Python dependencies." +changelog: "1.0.0 — Initial release: list/get DLP alerts, update alert, list/get incidents." +category: dlp + +# Per-instance configuration. Application (client-credentials) OAuth2. Needs Graph +# application permissions SecurityAlert.ReadWrite.All and SecurityIncident.Read.All. +config_schema: + properties: + tenant_id: + type: string + description: "Azure AD tenant ID" + client_id: + type: string + description: "App registration (client) ID" + client_secret: + type: string + description: "App registration client secret" + x-soar-sensitive: true + required: + - tenant_id + - client_id + - client_secret + +commands: + - id: list_dlp_alerts + name: purview-list-dlp-alerts + description: "List Data Loss Prevention alerts." + risk: read + inputs_schema: + properties: + status: { type: string, description: "Filter by status (new, inProgress, resolved)" } + limit: { type: number, description: "Max alerts (default 25)" } + required: [] + outputs_schema: { properties: {} } + - id: get_alert + name: purview-get-alert + description: "Get a single security alert by ID." + risk: read + inputs_schema: + properties: + alert_id: { type: string, description: "Alert ID" } + required: [alert_id] + outputs_schema: { properties: {} } + - id: update_alert + name: purview-update-alert + description: "Update an alert's status and/or assignee." + inputs_schema: + properties: + alert_id: { type: string, description: "Alert ID" } + status: { type: string, description: "new, inProgress, or resolved" } + assigned_to: { type: string, description: "Assignee (UPN)" } + determination: { type: string, description: "Optional determination (e.g. truePositive, falsePositive)" } + required: [alert_id] + outputs_schema: { properties: {} } + - id: list_incidents + name: purview-list-incidents + description: "List security incidents." + risk: read + inputs_schema: + properties: + status: { type: string, description: "Filter by status (active, resolved, redirected)" } + limit: { type: number, description: "Max incidents (default 25)" } + required: [] + outputs_schema: { properties: {} } + - id: get_incident + name: purview-get-incident + description: "Get a single security incident by ID." + risk: read + inputs_schema: + properties: + incident_id: { type: string, description: "Incident ID" } + required: [incident_id] + outputs_schema: { properties: {} } + + - id: test_connection + name: purview-test-connection + description: "Verify connectivity and the app credentials (used by the Test button)." + risk: read + inputs_schema: + properties: {} + required: [] + outputs_schema: { properties: {} } diff --git a/integrations/microsoft-purview/scripts/get_alert.py b/integrations/microsoft-purview/scripts/get_alert.py new file mode 100644 index 0000000..f74f101 --- /dev/null +++ b/integrations/microsoft-purview/scripts/get_alert.py @@ -0,0 +1,70 @@ +import json, os, sys, urllib.parse, urllib.request, urllib.error + +GRAPH = "https://graph.microsoft.com/v1.0" + + +def _cfg(): + return json.loads(os.environ.get("INTEGRATION_SECRETS", "{}")) + + +def _inputs(): + return json.loads(os.environ.get("INTEGRATION_INPUTS", "{}")) + + +def _token(cfg): + url = "https://login.microsoftonline.com/" + str(cfg.get("tenant_id", "")) + "/oauth2/v2.0/token" + form = urllib.parse.urlencode({ + "grant_type": "client_credentials", + "client_id": str(cfg.get("client_id", "")), + "client_secret": str(cfg.get("client_secret", "")), + "scope": "https://graph.microsoft.com/.default", + }).encode("utf-8") + req = urllib.request.Request(url, data=form, headers={"Content-Type": "application/x-www-form-urlencoded"}, method="POST") + with urllib.request.urlopen(req, timeout=60) as r: + tok = json.loads(r.read()) + if not tok.get("access_token"): + raise Exception("Token request failed: " + json.dumps(tok)) + return tok["access_token"] + + +def graph(method, path, token, body=None, params=None): + url = GRAPH + path + if params: + clean = {k: v for k, v in params.items() if v not in (None, "")} + if clean: + url += "?" + urllib.parse.urlencode(clean) + data = json.dumps(body).encode("utf-8") if body is not None else None + headers = {"Authorization": "Bearer " + token, "Accept": "application/json"} + if data is not None: + headers["Content-Type"] = "application/json" + req = urllib.request.Request(url, 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(fn): + try: + cfg = _cfg() + inputs = _inputs() + token = _token(cfg) + print(json.dumps(fn(token, inputs))) + 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) + + +q = lambda v: urllib.parse.quote(str(v), safe="") + + +def main(token, inputs): + alert_id = inputs.get("alert_id") + if not alert_id: + raise Exception("alert_id is required") + return graph("GET", "/security/alerts_v2/" + q(alert_id), token) + + +_run(main) diff --git a/integrations/microsoft-purview/scripts/get_incident.py b/integrations/microsoft-purview/scripts/get_incident.py new file mode 100644 index 0000000..19c806e --- /dev/null +++ b/integrations/microsoft-purview/scripts/get_incident.py @@ -0,0 +1,70 @@ +import json, os, sys, urllib.parse, urllib.request, urllib.error + +GRAPH = "https://graph.microsoft.com/v1.0" + + +def _cfg(): + return json.loads(os.environ.get("INTEGRATION_SECRETS", "{}")) + + +def _inputs(): + return json.loads(os.environ.get("INTEGRATION_INPUTS", "{}")) + + +def _token(cfg): + url = "https://login.microsoftonline.com/" + str(cfg.get("tenant_id", "")) + "/oauth2/v2.0/token" + form = urllib.parse.urlencode({ + "grant_type": "client_credentials", + "client_id": str(cfg.get("client_id", "")), + "client_secret": str(cfg.get("client_secret", "")), + "scope": "https://graph.microsoft.com/.default", + }).encode("utf-8") + req = urllib.request.Request(url, data=form, headers={"Content-Type": "application/x-www-form-urlencoded"}, method="POST") + with urllib.request.urlopen(req, timeout=60) as r: + tok = json.loads(r.read()) + if not tok.get("access_token"): + raise Exception("Token request failed: " + json.dumps(tok)) + return tok["access_token"] + + +def graph(method, path, token, body=None, params=None): + url = GRAPH + path + if params: + clean = {k: v for k, v in params.items() if v not in (None, "")} + if clean: + url += "?" + urllib.parse.urlencode(clean) + data = json.dumps(body).encode("utf-8") if body is not None else None + headers = {"Authorization": "Bearer " + token, "Accept": "application/json"} + if data is not None: + headers["Content-Type"] = "application/json" + req = urllib.request.Request(url, 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(fn): + try: + cfg = _cfg() + inputs = _inputs() + token = _token(cfg) + print(json.dumps(fn(token, inputs))) + 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) + + +q = lambda v: urllib.parse.quote(str(v), safe="") + + +def main(token, inputs): + incident_id = inputs.get("incident_id") + if not incident_id: + raise Exception("incident_id is required") + return graph("GET", "/security/incidents/" + q(incident_id), token) + + +_run(main) diff --git a/integrations/microsoft-purview/scripts/list_dlp_alerts.py b/integrations/microsoft-purview/scripts/list_dlp_alerts.py new file mode 100644 index 0000000..bf08bc0 --- /dev/null +++ b/integrations/microsoft-purview/scripts/list_dlp_alerts.py @@ -0,0 +1,69 @@ +import json, os, sys, urllib.parse, urllib.request, urllib.error + +GRAPH = "https://graph.microsoft.com/v1.0" + + +def _cfg(): + return json.loads(os.environ.get("INTEGRATION_SECRETS", "{}")) + + +def _inputs(): + return json.loads(os.environ.get("INTEGRATION_INPUTS", "{}")) + + +def _token(cfg): + url = "https://login.microsoftonline.com/" + str(cfg.get("tenant_id", "")) + "/oauth2/v2.0/token" + form = urllib.parse.urlencode({ + "grant_type": "client_credentials", + "client_id": str(cfg.get("client_id", "")), + "client_secret": str(cfg.get("client_secret", "")), + "scope": "https://graph.microsoft.com/.default", + }).encode("utf-8") + req = urllib.request.Request(url, data=form, headers={"Content-Type": "application/x-www-form-urlencoded"}, method="POST") + with urllib.request.urlopen(req, timeout=60) as r: + tok = json.loads(r.read()) + if not tok.get("access_token"): + raise Exception("Token request failed: " + json.dumps(tok)) + return tok["access_token"] + + +def graph(method, path, token, body=None, params=None): + url = GRAPH + path + if params: + clean = {k: v for k, v in params.items() if v not in (None, "")} + if clean: + url += "?" + urllib.parse.urlencode(clean) + data = json.dumps(body).encode("utf-8") if body is not None else None + headers = {"Authorization": "Bearer " + token, "Accept": "application/json"} + if data is not None: + headers["Content-Type"] = "application/json" + req = urllib.request.Request(url, 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(fn): + try: + cfg = _cfg() + inputs = _inputs() + token = _token(cfg) + print(json.dumps(fn(token, inputs))) + 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) + + +def main(token, inputs): + status = inputs.get("status") + limit = inputs.get("limit") or 25 + filter_str = "serviceSource eq 'microsoftDataLossPrevention'" + if status: + filter_str += " and status eq '" + str(status) + "'" + return graph("GET", "/security/alerts_v2", token, params={"$filter": filter_str, "$top": int(limit)}) + + +_run(main) diff --git a/integrations/microsoft-purview/scripts/list_incidents.py b/integrations/microsoft-purview/scripts/list_incidents.py new file mode 100644 index 0000000..5548922 --- /dev/null +++ b/integrations/microsoft-purview/scripts/list_incidents.py @@ -0,0 +1,71 @@ +import json, os, sys, urllib.parse, urllib.request, urllib.error + +GRAPH = "https://graph.microsoft.com/v1.0" + + +def _cfg(): + return json.loads(os.environ.get("INTEGRATION_SECRETS", "{}")) + + +def _inputs(): + return json.loads(os.environ.get("INTEGRATION_INPUTS", "{}")) + + +def _token(cfg): + url = "https://login.microsoftonline.com/" + str(cfg.get("tenant_id", "")) + "/oauth2/v2.0/token" + form = urllib.parse.urlencode({ + "grant_type": "client_credentials", + "client_id": str(cfg.get("client_id", "")), + "client_secret": str(cfg.get("client_secret", "")), + "scope": "https://graph.microsoft.com/.default", + }).encode("utf-8") + req = urllib.request.Request(url, data=form, headers={"Content-Type": "application/x-www-form-urlencoded"}, method="POST") + with urllib.request.urlopen(req, timeout=60) as r: + tok = json.loads(r.read()) + if not tok.get("access_token"): + raise Exception("Token request failed: " + json.dumps(tok)) + return tok["access_token"] + + +def graph(method, path, token, body=None, params=None): + url = GRAPH + path + if params: + clean = {k: v for k, v in params.items() if v not in (None, "")} + if clean: + url += "?" + urllib.parse.urlencode(clean) + data = json.dumps(body).encode("utf-8") if body is not None else None + headers = {"Authorization": "Bearer " + token, "Accept": "application/json"} + if data is not None: + headers["Content-Type"] = "application/json" + req = urllib.request.Request(url, 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(fn): + try: + cfg = _cfg() + inputs = _inputs() + token = _token(cfg) + print(json.dumps(fn(token, inputs))) + 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) + + +def main(token, inputs): + status = inputs.get("status") + limit = inputs.get("limit") or 25 + + params = {"$top": int(limit)} + if status: + params["$filter"] = "status eq '" + str(status) + "'" + + return graph("GET", "/security/incidents", token, params=params) + + +_run(main) diff --git a/integrations/microsoft-purview/scripts/test_connection.py b/integrations/microsoft-purview/scripts/test_connection.py new file mode 100644 index 0000000..3e07b74 --- /dev/null +++ b/integrations/microsoft-purview/scripts/test_connection.py @@ -0,0 +1,65 @@ +import json, os, sys, urllib.parse, urllib.request, urllib.error + +GRAPH = "https://graph.microsoft.com/v1.0" + + +def _cfg(): + return json.loads(os.environ.get("INTEGRATION_SECRETS", "{}")) + + +def _inputs(): + return json.loads(os.environ.get("INTEGRATION_INPUTS", "{}")) + + +def _token(cfg): + url = "https://login.microsoftonline.com/" + str(cfg.get("tenant_id", "")) + "/oauth2/v2.0/token" + form = urllib.parse.urlencode({ + "grant_type": "client_credentials", + "client_id": str(cfg.get("client_id", "")), + "client_secret": str(cfg.get("client_secret", "")), + "scope": "https://graph.microsoft.com/.default", + }).encode("utf-8") + req = urllib.request.Request(url, data=form, headers={"Content-Type": "application/x-www-form-urlencoded"}, method="POST") + with urllib.request.urlopen(req, timeout=60) as r: + tok = json.loads(r.read()) + if not tok.get("access_token"): + raise Exception("Token request failed: " + json.dumps(tok)) + return tok["access_token"] + + +def graph(method, path, token, body=None, params=None): + url = GRAPH + path + if params: + clean = {k: v for k, v in params.items() if v not in (None, "")} + if clean: + url += "?" + urllib.parse.urlencode(clean) + data = json.dumps(body).encode("utf-8") if body is not None else None + headers = {"Authorization": "Bearer " + token, "Accept": "application/json"} + if data is not None: + headers["Content-Type"] = "application/json" + req = urllib.request.Request(url, 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(fn): + try: + cfg = _cfg() + inputs = _inputs() + token = _token(cfg) + print(json.dumps(fn(token, inputs))) + 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) + + +def main(token, inputs): + graph("GET", "/security/alerts_v2", token, params={"$top": 1}) + return {"ok": True} + + +_run(main) diff --git a/integrations/microsoft-purview/scripts/update_alert.py b/integrations/microsoft-purview/scripts/update_alert.py new file mode 100644 index 0000000..a27c427 --- /dev/null +++ b/integrations/microsoft-purview/scripts/update_alert.py @@ -0,0 +1,89 @@ +import json, os, sys, urllib.parse, urllib.request, urllib.error + +GRAPH = "https://graph.microsoft.com/v1.0" + + +def _cfg(): + return json.loads(os.environ.get("INTEGRATION_SECRETS", "{}")) + + +def _inputs(): + return json.loads(os.environ.get("INTEGRATION_INPUTS", "{}")) + + +def _token(cfg): + url = "https://login.microsoftonline.com/" + str(cfg.get("tenant_id", "")) + "/oauth2/v2.0/token" + form = urllib.parse.urlencode({ + "grant_type": "client_credentials", + "client_id": str(cfg.get("client_id", "")), + "client_secret": str(cfg.get("client_secret", "")), + "scope": "https://graph.microsoft.com/.default", + }).encode("utf-8") + req = urllib.request.Request(url, data=form, headers={"Content-Type": "application/x-www-form-urlencoded"}, method="POST") + with urllib.request.urlopen(req, timeout=60) as r: + tok = json.loads(r.read()) + if not tok.get("access_token"): + raise Exception("Token request failed: " + json.dumps(tok)) + return tok["access_token"] + + +def graph(method, path, token, body=None, params=None): + url = GRAPH + path + if params: + clean = {k: v for k, v in params.items() if v not in (None, "")} + if clean: + url += "?" + urllib.parse.urlencode(clean) + data = json.dumps(body).encode("utf-8") if body is not None else None + headers = {"Authorization": "Bearer " + token, "Accept": "application/json"} + if data is not None: + headers["Content-Type"] = "application/json" + req = urllib.request.Request(url, 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(fn): + try: + cfg = _cfg() + inputs = _inputs() + token = _token(cfg) + print(json.dumps(fn(token, inputs))) + 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) + + +q = lambda v: urllib.parse.quote(str(v), safe="") + + +def main(token, inputs): + alert_id = inputs.get("alert_id") + if not alert_id: + raise Exception("alert_id is required") + + status = inputs.get("status") + assigned_to = inputs.get("assigned_to") + determination = inputs.get("determination") + + body = {} + if status: + body["status"] = status + if assigned_to: + body["assignedTo"] = assigned_to + if determination: + body["determination"] = determination + + if not body: + raise Exception("at least one field to update is required") + + result = graph("PATCH", "/security/alerts_v2/" + q(alert_id), token, body=body) + if not result: + return {"ok": True, "alert_id": alert_id} + return result + + +_run(main)