From 1949c98cf79b539d212c80019d70ae361a32936e Mon Sep 17 00:00:00 2001 From: Guillaume BOURGEOIS Date: Sun, 12 Jul 2026 15:04:02 +0200 Subject: [PATCH] feat(sophos-central): new Sophos Central endpoint integration Sophos Central Endpoint API, 7 commands: list/get endpoints, isolate/de-isolate (containment), scan, list alerts. OAuth2 client-credentials + tenant discovery (whoami), stdlib-only. Co-Authored-By: Claude Opus 4.8 (1M context) --- integrations/sophos-central/manifest.yaml | 85 ++++++++++++++++++ .../scripts/deisolate_endpoint.py | 86 +++++++++++++++++++ .../sophos-central/scripts/get_endpoint.py | 85 ++++++++++++++++++ .../scripts/isolate_endpoint.py | 86 +++++++++++++++++++ .../sophos-central/scripts/list_alerts.py | 83 ++++++++++++++++++ .../sophos-central/scripts/list_endpoints.py | 83 ++++++++++++++++++ .../sophos-central/scripts/scan_endpoint.py | 85 ++++++++++++++++++ .../sophos-central/scripts/test_connection.py | 81 +++++++++++++++++ 8 files changed, 674 insertions(+) create mode 100644 integrations/sophos-central/manifest.yaml create mode 100644 integrations/sophos-central/scripts/deisolate_endpoint.py create mode 100644 integrations/sophos-central/scripts/get_endpoint.py create mode 100644 integrations/sophos-central/scripts/isolate_endpoint.py create mode 100644 integrations/sophos-central/scripts/list_alerts.py create mode 100644 integrations/sophos-central/scripts/list_endpoints.py create mode 100644 integrations/sophos-central/scripts/scan_endpoint.py create mode 100644 integrations/sophos-central/scripts/test_connection.py diff --git a/integrations/sophos-central/manifest.yaml b/integrations/sophos-central/manifest.yaml new file mode 100644 index 0000000..642d43a --- /dev/null +++ b/integrations/sophos-central/manifest.yaml @@ -0,0 +1,85 @@ +id: sophos_central +name: Sophos Central +version: 1.0.0 +description: "Sophos Central (Endpoint API) — endpoint containment: list and read endpoints, isolate/de-isolate a host, trigger a scan, and list alerts. OAuth2 client-credentials authentication (with tenant discovery); stdlib-only, no extra Python dependencies." +changelog: "1.0.0 — Initial release: list/get endpoints, isolate/de-isolate, scan, list alerts." +category: endpoint + +# Per-instance configuration. Client credentials are exchanged for a token; the +# whoami endpoint provides the tenant ID and regional API host used thereafter. +config_schema: + properties: + client_id: + type: string + description: "Sophos Central API client ID" + client_secret: + type: string + description: "Sophos Central API client secret" + x-soar-sensitive: true + required: + - client_id + - client_secret + +commands: + - id: list_endpoints + name: sophos-list-endpoints + description: "List endpoints." + risk: read + inputs_schema: + properties: + limit: { type: number, description: "Max endpoints (default 50)" } + required: [] + outputs_schema: { properties: {} } + - id: get_endpoint + name: sophos-get-endpoint + description: "Get a single endpoint by ID." + risk: read + inputs_schema: + properties: + endpoint_id: { type: string, description: "Endpoint ID" } + required: [endpoint_id] + outputs_schema: { properties: {} } + - id: isolate_endpoint + name: sophos-isolate-endpoint + description: "Isolate an endpoint from the network (containment)." + inputs_schema: + properties: + endpoint_id: { type: string, description: "Endpoint ID" } + comment: { type: string, description: "Optional comment" } + required: [endpoint_id] + outputs_schema: { properties: {} } + - id: deisolate_endpoint + name: sophos-deisolate-endpoint + description: "Remove an endpoint from isolation." + inputs_schema: + properties: + endpoint_id: { type: string, description: "Endpoint ID" } + comment: { type: string, description: "Optional comment" } + required: [endpoint_id] + outputs_schema: { properties: {} } + - id: scan_endpoint + name: sophos-scan-endpoint + description: "Trigger a malware scan on an endpoint." + inputs_schema: + properties: + endpoint_id: { type: string, description: "Endpoint ID" } + required: [endpoint_id] + outputs_schema: { properties: {} } + - id: list_alerts + name: sophos-list-alerts + description: "List alerts." + risk: read + inputs_schema: + properties: + limit: { type: number, description: "Max alerts (default 50)" } + required: [] + outputs_schema: { properties: {} } + + - id: test_connection + name: sophos-test-connection + description: "Verify the credentials and tenant discovery (used by the Test button)." + risk: read + inputs_schema: + properties: {} + required: [] + outputs_schema: { properties: {} } diff --git a/integrations/sophos-central/scripts/deisolate_endpoint.py b/integrations/sophos-central/scripts/deisolate_endpoint.py new file mode 100644 index 0000000..54f8d11 --- /dev/null +++ b/integrations/sophos-central/scripts/deisolate_endpoint.py @@ -0,0 +1,86 @@ +import json, os, sys, urllib.parse, urllib.request, urllib.error + + +def _cfg(): + return json.loads(os.environ.get("INTEGRATION_SECRETS", "{}")) + + +def _inputs(): + return json.loads(os.environ.get("INTEGRATION_INPUTS", "{}")) + + +def _token(cfg): + form = urllib.parse.urlencode({ + "grant_type": "client_credentials", + "client_id": str(cfg.get("client_id", "")), + "client_secret": str(cfg.get("client_secret", "")), + "scope": "token", + }).encode("utf-8") + req = urllib.request.Request("https://id.sophos.com/api/v2/oauth2/token", data=form, + headers={"Content-Type": "application/x-www-form-urlencoded", + "Accept": "application/json"}, 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 _whoami(token): + req = urllib.request.Request("https://api.central.sophos.com/whoami/v1", + headers={"Authorization": "Bearer " + token, "Accept": "application/json"}, method="GET") + with urllib.request.urlopen(req, timeout=60) as r: + wi = json.loads(r.read()) + tenant = wi.get("id") + region = (wi.get("apiHosts") or {}).get("dataRegion") + if not tenant or not region: + raise Exception("whoami failed: " + json.dumps(wi)) + return tenant, region.rstrip("/") + + +def _ctx_auth(cfg): + token = _token(cfg) + tenant, region = _whoami(token) + return token, tenant, region + + +def request(method, path, token, tenant, region, body=None, params=None): + url = region + 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, "X-Tenant-ID": tenant, "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, tenant, region = _ctx_auth(cfg) + print(json.dumps(fn(token, tenant, region, 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, tenant, region, inputs): + endpoint_id = inputs.get("endpoint_id") + if not endpoint_id: + raise Exception("endpoint_id is required") + comment = inputs.get("comment") + body = {"enabled": False, "ids": [endpoint_id], "comment": (comment or "De-isolated via Riposte")} + return request("POST", "/endpoint/v1/endpoints/isolation", token, tenant, region, body=body) + + +_run(main) diff --git a/integrations/sophos-central/scripts/get_endpoint.py b/integrations/sophos-central/scripts/get_endpoint.py new file mode 100644 index 0000000..0267b8e --- /dev/null +++ b/integrations/sophos-central/scripts/get_endpoint.py @@ -0,0 +1,85 @@ +import json, os, sys, urllib.parse, urllib.request, urllib.error + + +def _cfg(): + return json.loads(os.environ.get("INTEGRATION_SECRETS", "{}")) + + +def _inputs(): + return json.loads(os.environ.get("INTEGRATION_INPUTS", "{}")) + + +def _token(cfg): + form = urllib.parse.urlencode({ + "grant_type": "client_credentials", + "client_id": str(cfg.get("client_id", "")), + "client_secret": str(cfg.get("client_secret", "")), + "scope": "token", + }).encode("utf-8") + req = urllib.request.Request("https://id.sophos.com/api/v2/oauth2/token", data=form, + headers={"Content-Type": "application/x-www-form-urlencoded", + "Accept": "application/json"}, 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 _whoami(token): + req = urllib.request.Request("https://api.central.sophos.com/whoami/v1", + headers={"Authorization": "Bearer " + token, "Accept": "application/json"}, method="GET") + with urllib.request.urlopen(req, timeout=60) as r: + wi = json.loads(r.read()) + tenant = wi.get("id") + region = (wi.get("apiHosts") or {}).get("dataRegion") + if not tenant or not region: + raise Exception("whoami failed: " + json.dumps(wi)) + return tenant, region.rstrip("/") + + +def _ctx_auth(cfg): + token = _token(cfg) + tenant, region = _whoami(token) + return token, tenant, region + + +def request(method, path, token, tenant, region, body=None, params=None): + url = region + 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, "X-Tenant-ID": tenant, "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, tenant, region = _ctx_auth(cfg) + print(json.dumps(fn(token, tenant, region, 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, tenant, region, inputs): + endpoint_id = inputs.get("endpoint_id") + if not endpoint_id: + raise Exception("endpoint_id is required") + q = lambda v: urllib.parse.quote(str(v), safe="") + return request("GET", "/endpoint/v1/endpoints/" + q(endpoint_id), token, tenant, region) + + +_run(main) diff --git a/integrations/sophos-central/scripts/isolate_endpoint.py b/integrations/sophos-central/scripts/isolate_endpoint.py new file mode 100644 index 0000000..b3a8ab9 --- /dev/null +++ b/integrations/sophos-central/scripts/isolate_endpoint.py @@ -0,0 +1,86 @@ +import json, os, sys, urllib.parse, urllib.request, urllib.error + + +def _cfg(): + return json.loads(os.environ.get("INTEGRATION_SECRETS", "{}")) + + +def _inputs(): + return json.loads(os.environ.get("INTEGRATION_INPUTS", "{}")) + + +def _token(cfg): + form = urllib.parse.urlencode({ + "grant_type": "client_credentials", + "client_id": str(cfg.get("client_id", "")), + "client_secret": str(cfg.get("client_secret", "")), + "scope": "token", + }).encode("utf-8") + req = urllib.request.Request("https://id.sophos.com/api/v2/oauth2/token", data=form, + headers={"Content-Type": "application/x-www-form-urlencoded", + "Accept": "application/json"}, 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 _whoami(token): + req = urllib.request.Request("https://api.central.sophos.com/whoami/v1", + headers={"Authorization": "Bearer " + token, "Accept": "application/json"}, method="GET") + with urllib.request.urlopen(req, timeout=60) as r: + wi = json.loads(r.read()) + tenant = wi.get("id") + region = (wi.get("apiHosts") or {}).get("dataRegion") + if not tenant or not region: + raise Exception("whoami failed: " + json.dumps(wi)) + return tenant, region.rstrip("/") + + +def _ctx_auth(cfg): + token = _token(cfg) + tenant, region = _whoami(token) + return token, tenant, region + + +def request(method, path, token, tenant, region, body=None, params=None): + url = region + 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, "X-Tenant-ID": tenant, "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, tenant, region = _ctx_auth(cfg) + print(json.dumps(fn(token, tenant, region, 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, tenant, region, inputs): + endpoint_id = inputs.get("endpoint_id") + if not endpoint_id: + raise Exception("endpoint_id is required") + comment = inputs.get("comment") + body = {"enabled": True, "ids": [endpoint_id], "comment": (comment or "Isolated via Riposte")} + return request("POST", "/endpoint/v1/endpoints/isolation", token, tenant, region, body=body) + + +_run(main) diff --git a/integrations/sophos-central/scripts/list_alerts.py b/integrations/sophos-central/scripts/list_alerts.py new file mode 100644 index 0000000..ab92755 --- /dev/null +++ b/integrations/sophos-central/scripts/list_alerts.py @@ -0,0 +1,83 @@ +import json, os, sys, urllib.parse, urllib.request, urllib.error + + +def _cfg(): + return json.loads(os.environ.get("INTEGRATION_SECRETS", "{}")) + + +def _inputs(): + return json.loads(os.environ.get("INTEGRATION_INPUTS", "{}")) + + +def _token(cfg): + form = urllib.parse.urlencode({ + "grant_type": "client_credentials", + "client_id": str(cfg.get("client_id", "")), + "client_secret": str(cfg.get("client_secret", "")), + "scope": "token", + }).encode("utf-8") + req = urllib.request.Request("https://id.sophos.com/api/v2/oauth2/token", data=form, + headers={"Content-Type": "application/x-www-form-urlencoded", + "Accept": "application/json"}, 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 _whoami(token): + req = urllib.request.Request("https://api.central.sophos.com/whoami/v1", + headers={"Authorization": "Bearer " + token, "Accept": "application/json"}, method="GET") + with urllib.request.urlopen(req, timeout=60) as r: + wi = json.loads(r.read()) + tenant = wi.get("id") + region = (wi.get("apiHosts") or {}).get("dataRegion") + if not tenant or not region: + raise Exception("whoami failed: " + json.dumps(wi)) + return tenant, region.rstrip("/") + + +def _ctx_auth(cfg): + token = _token(cfg) + tenant, region = _whoami(token) + return token, tenant, region + + +def request(method, path, token, tenant, region, body=None, params=None): + url = region + 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, "X-Tenant-ID": tenant, "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, tenant, region = _ctx_auth(cfg) + print(json.dumps(fn(token, tenant, region, 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, tenant, region, inputs): + limit = inputs.get("limit") + limit = int(limit) if limit not in (None, "") else 50 + return request("GET", "/common/v1/alerts", token, tenant, region, params={"pageSize": limit}) + + +_run(main) diff --git a/integrations/sophos-central/scripts/list_endpoints.py b/integrations/sophos-central/scripts/list_endpoints.py new file mode 100644 index 0000000..d7bd262 --- /dev/null +++ b/integrations/sophos-central/scripts/list_endpoints.py @@ -0,0 +1,83 @@ +import json, os, sys, urllib.parse, urllib.request, urllib.error + + +def _cfg(): + return json.loads(os.environ.get("INTEGRATION_SECRETS", "{}")) + + +def _inputs(): + return json.loads(os.environ.get("INTEGRATION_INPUTS", "{}")) + + +def _token(cfg): + form = urllib.parse.urlencode({ + "grant_type": "client_credentials", + "client_id": str(cfg.get("client_id", "")), + "client_secret": str(cfg.get("client_secret", "")), + "scope": "token", + }).encode("utf-8") + req = urllib.request.Request("https://id.sophos.com/api/v2/oauth2/token", data=form, + headers={"Content-Type": "application/x-www-form-urlencoded", + "Accept": "application/json"}, 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 _whoami(token): + req = urllib.request.Request("https://api.central.sophos.com/whoami/v1", + headers={"Authorization": "Bearer " + token, "Accept": "application/json"}, method="GET") + with urllib.request.urlopen(req, timeout=60) as r: + wi = json.loads(r.read()) + tenant = wi.get("id") + region = (wi.get("apiHosts") or {}).get("dataRegion") + if not tenant or not region: + raise Exception("whoami failed: " + json.dumps(wi)) + return tenant, region.rstrip("/") + + +def _ctx_auth(cfg): + token = _token(cfg) + tenant, region = _whoami(token) + return token, tenant, region + + +def request(method, path, token, tenant, region, body=None, params=None): + url = region + 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, "X-Tenant-ID": tenant, "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, tenant, region = _ctx_auth(cfg) + print(json.dumps(fn(token, tenant, region, 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, tenant, region, inputs): + limit = inputs.get("limit") + limit = int(limit) if limit not in (None, "") else 50 + return request("GET", "/endpoint/v1/endpoints", token, tenant, region, params={"pageSize": limit}) + + +_run(main) diff --git a/integrations/sophos-central/scripts/scan_endpoint.py b/integrations/sophos-central/scripts/scan_endpoint.py new file mode 100644 index 0000000..b2750ab --- /dev/null +++ b/integrations/sophos-central/scripts/scan_endpoint.py @@ -0,0 +1,85 @@ +import json, os, sys, urllib.parse, urllib.request, urllib.error + + +def _cfg(): + return json.loads(os.environ.get("INTEGRATION_SECRETS", "{}")) + + +def _inputs(): + return json.loads(os.environ.get("INTEGRATION_INPUTS", "{}")) + + +def _token(cfg): + form = urllib.parse.urlencode({ + "grant_type": "client_credentials", + "client_id": str(cfg.get("client_id", "")), + "client_secret": str(cfg.get("client_secret", "")), + "scope": "token", + }).encode("utf-8") + req = urllib.request.Request("https://id.sophos.com/api/v2/oauth2/token", data=form, + headers={"Content-Type": "application/x-www-form-urlencoded", + "Accept": "application/json"}, 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 _whoami(token): + req = urllib.request.Request("https://api.central.sophos.com/whoami/v1", + headers={"Authorization": "Bearer " + token, "Accept": "application/json"}, method="GET") + with urllib.request.urlopen(req, timeout=60) as r: + wi = json.loads(r.read()) + tenant = wi.get("id") + region = (wi.get("apiHosts") or {}).get("dataRegion") + if not tenant or not region: + raise Exception("whoami failed: " + json.dumps(wi)) + return tenant, region.rstrip("/") + + +def _ctx_auth(cfg): + token = _token(cfg) + tenant, region = _whoami(token) + return token, tenant, region + + +def request(method, path, token, tenant, region, body=None, params=None): + url = region + 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, "X-Tenant-ID": tenant, "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, tenant, region = _ctx_auth(cfg) + print(json.dumps(fn(token, tenant, region, 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, tenant, region, inputs): + endpoint_id = inputs.get("endpoint_id") + if not endpoint_id: + raise Exception("endpoint_id is required") + q = lambda v: urllib.parse.quote(str(v), safe="") + return request("POST", "/endpoint/v1/endpoints/" + q(endpoint_id) + "/scans", token, tenant, region, body={}) + + +_run(main) diff --git a/integrations/sophos-central/scripts/test_connection.py b/integrations/sophos-central/scripts/test_connection.py new file mode 100644 index 0000000..3c6f7ce --- /dev/null +++ b/integrations/sophos-central/scripts/test_connection.py @@ -0,0 +1,81 @@ +import json, os, sys, urllib.parse, urllib.request, urllib.error + + +def _cfg(): + return json.loads(os.environ.get("INTEGRATION_SECRETS", "{}")) + + +def _inputs(): + return json.loads(os.environ.get("INTEGRATION_INPUTS", "{}")) + + +def _token(cfg): + form = urllib.parse.urlencode({ + "grant_type": "client_credentials", + "client_id": str(cfg.get("client_id", "")), + "client_secret": str(cfg.get("client_secret", "")), + "scope": "token", + }).encode("utf-8") + req = urllib.request.Request("https://id.sophos.com/api/v2/oauth2/token", data=form, + headers={"Content-Type": "application/x-www-form-urlencoded", + "Accept": "application/json"}, 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 _whoami(token): + req = urllib.request.Request("https://api.central.sophos.com/whoami/v1", + headers={"Authorization": "Bearer " + token, "Accept": "application/json"}, method="GET") + with urllib.request.urlopen(req, timeout=60) as r: + wi = json.loads(r.read()) + tenant = wi.get("id") + region = (wi.get("apiHosts") or {}).get("dataRegion") + if not tenant or not region: + raise Exception("whoami failed: " + json.dumps(wi)) + return tenant, region.rstrip("/") + + +def _ctx_auth(cfg): + token = _token(cfg) + tenant, region = _whoami(token) + return token, tenant, region + + +def request(method, path, token, tenant, region, body=None, params=None): + url = region + 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, "X-Tenant-ID": tenant, "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, tenant, region = _ctx_auth(cfg) + print(json.dumps(fn(token, tenant, region, 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, tenant, region, inputs): + return {"ok": True, "tenant": tenant} + + +_run(main)