diff options
Diffstat (limited to 'src/integration/assets')
| -rw-r--r-- | src/integration/assets/claude/herdr-agent-state.sh | 70 | ||||
| -rw-r--r-- | src/integration/assets/codex/herdr-agent-state.sh | 70 | ||||
| -rw-r--r-- | src/integration/assets/opencode/herdr-agent-state.js | 99 |
3 files changed, 239 insertions, 0 deletions
diff --git a/src/integration/assets/claude/herdr-agent-state.sh b/src/integration/assets/claude/herdr-agent-state.sh new file mode 100644 index 00000000..3a202a2e --- /dev/null +++ b/src/integration/assets/claude/herdr-agent-state.sh @@ -0,0 +1,70 @@ +#!/bin/sh +# installed by herdr +# safe to edit. this hook only activates inside herdr-managed panes. + +set -eu + +action="${1:-}" +cat >/dev/null 2>/dev/null || true + +case "$action" in + working|idle|blocked|release) ;; + *) exit 0 ;; +esac + +[ "${HERDR_ENV:-}" = "1" ] || exit 0 +[ -n "${HERDR_SOCKET_PATH:-}" ] || exit 0 +[ -n "${HERDR_PANE_ID:-}" ] || exit 0 +command -v python3 >/dev/null 2>&1 || exit 0 + +HERDR_ACTION="$action" python3 - <<'PY' +import json +import os +import random +import socket +import time + +source = "herdr:claude" +action = os.environ.get("HERDR_ACTION", "") +pane_id = os.environ.get("HERDR_PANE_ID") +socket_path = os.environ.get("HERDR_SOCKET_PATH") + +if not pane_id or not socket_path: + raise SystemExit(0) + +request_id = f"{source}:{int(time.time() * 1000)}:{random.randrange(1_000_000):06d}" +if action == "release": + request = { + "id": request_id, + "method": "pane.release_agent", + "params": { + "pane_id": pane_id, + "source": source, + "agent": "claude", + }, + } +else: + request = { + "id": request_id, + "method": "pane.report_agent", + "params": { + "pane_id": pane_id, + "source": source, + "agent": "claude", + "state": action, + }, + } + +try: + client = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) + client.settimeout(0.5) + client.connect(socket_path) + client.sendall((json.dumps(request) + "\n").encode()) + try: + client.recv(4096) + except Exception: + pass + client.close() +except Exception: + pass +PY diff --git a/src/integration/assets/codex/herdr-agent-state.sh b/src/integration/assets/codex/herdr-agent-state.sh new file mode 100644 index 00000000..01510d6d --- /dev/null +++ b/src/integration/assets/codex/herdr-agent-state.sh @@ -0,0 +1,70 @@ +#!/bin/sh +# installed by herdr +# safe to edit. this hook only activates inside herdr-managed panes. + +set -eu + +action="${1:-}" +cat >/dev/null 2>/dev/null || true + +case "$action" in + working|idle|blocked|release) ;; + *) exit 0 ;; +esac + +[ "${HERDR_ENV:-}" = "1" ] || exit 0 +[ -n "${HERDR_SOCKET_PATH:-}" ] || exit 0 +[ -n "${HERDR_PANE_ID:-}" ] || exit 0 +command -v python3 >/dev/null 2>&1 || exit 0 + +HERDR_ACTION="$action" python3 - <<'PY' +import json +import os +import random +import socket +import time + +source = "herdr:codex" +action = os.environ.get("HERDR_ACTION", "") +pane_id = os.environ.get("HERDR_PANE_ID") +socket_path = os.environ.get("HERDR_SOCKET_PATH") + +if not pane_id or not socket_path: + raise SystemExit(0) + +request_id = f"{source}:{int(time.time() * 1000)}:{random.randrange(1_000_000):06d}" +if action == "release": + request = { + "id": request_id, + "method": "pane.release_agent", + "params": { + "pane_id": pane_id, + "source": source, + "agent": "codex", + }, + } +else: + request = { + "id": request_id, + "method": "pane.report_agent", + "params": { + "pane_id": pane_id, + "source": source, + "agent": "codex", + "state": action, + }, + } + +try: + client = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) + client.settimeout(0.5) + client.connect(socket_path) + client.sendall((json.dumps(request) + "\n").encode()) + try: + client.recv(4096) + except Exception: + pass + client.close() +except Exception: + pass +PY diff --git a/src/integration/assets/opencode/herdr-agent-state.js b/src/integration/assets/opencode/herdr-agent-state.js new file mode 100644 index 00000000..8dbaa0d8 --- /dev/null +++ b/src/integration/assets/opencode/herdr-agent-state.js @@ -0,0 +1,99 @@ +import net from "node:net"; + +const SOURCE = "herdr:opencode"; + +function reportState(action) { + const paneId = process.env.HERDR_PANE_ID; + const socketPath = process.env.HERDR_SOCKET_PATH; + + if (!paneId || !socketPath) { + return Promise.resolve(); + } + + const requestId = `${SOURCE}:${Date.now()}:${Math.floor(Math.random() * 1_000_000) + .toString() + .padStart(6, "0")}`; + const request = { + id: requestId, + method: action === "release" ? "pane.release_agent" : "pane.report_agent", + params: + action === "release" + ? { + pane_id: paneId, + source: SOURCE, + agent: "opencode", + } + : { + pane_id: paneId, + source: SOURCE, + agent: "opencode", + state: action, + }, + }; + + return new Promise((resolve) => { + const client = net.createConnection(socketPath, () => { + client.write(`${JSON.stringify(request)}\n`); + }); + + const finish = () => { + client.destroy(); + resolve(); + }; + + client.setTimeout(500, finish); + client.on("data", finish); + client.on("error", finish); + client.on("end", finish); + client.on("close", resolve); + }); +} + +export const HerdrAgentStatePlugin = async () => { + if ( + process.env.HERDR_ENV !== "1" || + !process.env.HERDR_SOCKET_PATH || + !process.env.HERDR_PANE_ID + ) { + return {}; + } + + return { + event: async ({ event }) => { + const type = event?.type; + const properties = event?.properties ?? {}; + + switch (type) { + case "permission.asked": + await reportState("blocked"); + break; + case "permission.replied": { + const reply = properties.reply ?? properties.response; + if (reply === "reject") { + await reportState("idle"); + } else if (reply === "once" || reply === "always") { + await reportState("working"); + } + break; + } + case "session.status": { + const status = + typeof properties.status === "string" + ? properties.status + : properties.status?.type; + if (status === "busy" || status === "retry") { + await reportState("working"); + } else if (status === "idle") { + await reportState("idle"); + } + break; + } + case "session.idle": + await reportState("idle"); + break; + default: + break; + } + }, + }; +}; |
