feat(graylog): new Graylog log-search integration

Graylog REST API, 5 commands: relative message search, list/get streams, get
message. API-token (Basic) auth, stdlib-only.

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>

integrations/graylog/scripts/get_message.py

@@ -0,0 +1,56 @@
+import json, os, sys, base64, 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 _auth(cfg):
+    raw = str(cfg.get("token", "")) + ":token"
+    return "Basic " + base64.b64encode(raw.encode("utf-8")).decode("utf-8")
+
+
+def request(method, path, cfg, params=None):
+    url = str(cfg.get("base_url", "")).rstrip("/") + "/api" + path
+    if params:
+        clean = {k: v for k, v in params.items() if v not in (None, "")}
+        if clean:
+            url += "?" + urllib.parse.urlencode(clean)
+    headers = {"Authorization": _auth(cfg), "Accept": "application/json", "X-Requested-By": "riposte"}
+    req = urllib.request.Request(url, headers=headers, method=method)
+    with urllib.request.urlopen(req, timeout=60) as r:
+        raw = r.read()
+        return json.loads(raw) if raw else {}
+
+
+def _run(fn):
+    try:
+        print(json.dumps(fn(_cfg(), _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(cfg, inputs):
+    index = inputs.get("index")
+    if not index:
+        raise Exception("index is required")
+
+    message_id = inputs.get("message_id")
+    if not message_id:
+        raise Exception("message_id is required")
+
+    return request("GET", "/messages/" + q(index) + "/" + q(message_id), cfg)
+
+
+_run(main)

integrations/graylog/scripts/get_stream.py

@@ -0,0 +1,52 @@
+import json, os, sys, base64, 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 _auth(cfg):
+    raw = str(cfg.get("token", "")) + ":token"
+    return "Basic " + base64.b64encode(raw.encode("utf-8")).decode("utf-8")
+
+
+def request(method, path, cfg, params=None):
+    url = str(cfg.get("base_url", "")).rstrip("/") + "/api" + path
+    if params:
+        clean = {k: v for k, v in params.items() if v not in (None, "")}
+        if clean:
+            url += "?" + urllib.parse.urlencode(clean)
+    headers = {"Authorization": _auth(cfg), "Accept": "application/json", "X-Requested-By": "riposte"}
+    req = urllib.request.Request(url, headers=headers, method=method)
+    with urllib.request.urlopen(req, timeout=60) as r:
+        raw = r.read()
+        return json.loads(raw) if raw else {}
+
+
+def _run(fn):
+    try:
+        print(json.dumps(fn(_cfg(), _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(cfg, inputs):
+    stream_id = inputs.get("stream_id")
+    if not stream_id:
+        raise Exception("stream_id is required")
+
+    return request("GET", "/streams/" + q(stream_id), cfg)
+
+
+_run(main)

integrations/graylog/scripts/list_streams.py

@@ -0,0 +1,45 @@
+import json, os, sys, base64, 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 _auth(cfg):
+    raw = str(cfg.get("token", "")) + ":token"
+    return "Basic " + base64.b64encode(raw.encode("utf-8")).decode("utf-8")
+
+
+def request(method, path, cfg, params=None):
+    url = str(cfg.get("base_url", "")).rstrip("/") + "/api" + path
+    if params:
+        clean = {k: v for k, v in params.items() if v not in (None, "")}
+        if clean:
+            url += "?" + urllib.parse.urlencode(clean)
+    headers = {"Authorization": _auth(cfg), "Accept": "application/json", "X-Requested-By": "riposte"}
+    req = urllib.request.Request(url, headers=headers, method=method)
+    with urllib.request.urlopen(req, timeout=60) as r:
+        raw = r.read()
+        return json.loads(raw) if raw else {}
+
+
+def _run(fn):
+    try:
+        print(json.dumps(fn(_cfg(), _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(cfg, inputs):
+    return request("GET", "/streams", cfg)
+
+
+_run(main)

integrations/graylog/scripts/search_messages.py

@@ -0,0 +1,60 @@
+import json, os, sys, base64, 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 _auth(cfg):
+    raw = str(cfg.get("token", "")) + ":token"
+    return "Basic " + base64.b64encode(raw.encode("utf-8")).decode("utf-8")
+
+
+def request(method, path, cfg, params=None):
+    url = str(cfg.get("base_url", "")).rstrip("/") + "/api" + path
+    if params:
+        clean = {k: v for k, v in params.items() if v not in (None, "")}
+        if clean:
+            url += "?" + urllib.parse.urlencode(clean)
+    headers = {"Authorization": _auth(cfg), "Accept": "application/json", "X-Requested-By": "riposte"}
+    req = urllib.request.Request(url, headers=headers, method=method)
+    with urllib.request.urlopen(req, timeout=60) as r:
+        raw = r.read()
+        return json.loads(raw) if raw else {}
+
+
+def _run(fn):
+    try:
+        print(json.dumps(fn(_cfg(), _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(cfg, inputs):
+    query = inputs.get("query")
+    if not query:
+        raise Exception("query is required")
+
+    range_secs = inputs.get("range")
+    range_secs = int(range_secs) if range_secs not in (None, "") else 3600
+
+    limit = inputs.get("limit")
+    limit = int(limit) if limit not in (None, "") else 50
+
+    return request(
+        "GET",
+        "/search/universal/relative",
+        cfg,
+        params={"query": query, "range": range_secs, "limit": limit},
+    )
+
+
+_run(main)

integrations/graylog/scripts/test_connection.py

@@ -0,0 +1,46 @@
+import json, os, sys, base64, 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 _auth(cfg):
+    raw = str(cfg.get("token", "")) + ":token"
+    return "Basic " + base64.b64encode(raw.encode("utf-8")).decode("utf-8")
+
+
+def request(method, path, cfg, params=None):
+    url = str(cfg.get("base_url", "")).rstrip("/") + "/api" + path
+    if params:
+        clean = {k: v for k, v in params.items() if v not in (None, "")}
+        if clean:
+            url += "?" + urllib.parse.urlencode(clean)
+    headers = {"Authorization": _auth(cfg), "Accept": "application/json", "X-Requested-By": "riposte"}
+    req = urllib.request.Request(url, headers=headers, method=method)
+    with urllib.request.urlopen(req, timeout=60) as r:
+        raw = r.read()
+        return json.loads(raw) if raw else {}
+
+
+def _run(fn):
+    try:
+        print(json.dumps(fn(_cfg(), _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(cfg, inputs):
+    request("GET", "/system", cfg)
+    return {"ok": True}
+
+
+_run(main)