1
0
mirror of https://github.com/sredevopsorg/deepseek-kubernetes-sandbox.git synced 2026-11-03 01:13:24 +00:00
Files
deepseek-kubernetes-sandbox/.opencode/plugins/entire.ts
SREDevOpsAgents a585e6be13 Enable Entire
2026-10-31 12:25:45 -03:00

586 lines
26 KiB
TypeScript

// Entire CLI plugin for OpenCode
// Auto-generated by `entire enable --agent opencode`
// Do not edit manually — changes will be overwritten on next install.
//
// Uses node:child_process so hooks work under both Bun (OpenCode CLI/TUI)
// and Node (OpenCode Desktop's Electron sidecar). See #2014.
import { spawn, spawnSync } from "node:child_process"
import type { Plugin } from "@opencode-ai/plugin"
export const EntirePlugin: Plugin = async ({ directory }) => {
// Track seen user messages to fire turn-start only once per message
const seenUserMessages = new Set<string>()
// Track current session ID for message events (which don't include sessionID)
let currentSessionID: string | null = null
// Track the model used by the most recent assistant message
let currentModel: string | null = null
// In-memory store for message metadata (role, tokens, etc.)
const messageStore = new Map<string, any>()
// One-time model-context injection captured from the turn-start hook's stdout,
// applied on the next LLM call via experimental.chat.system.transform.
let pendingInjection: string | null = null
// Child (subagent) sessions. OpenCode's task tool runs each subagent as a
// real session, so without this set a child would register as the user's
// session: its write would take the checkpoint and the parent would log
// "no files modified". Learned from session.* events' `parentID`, and from the child
// ID surfaced in the parent's task part / tool.execute.after — the latter
// two also cover a child session resumed via `task_id` from an earlier
// process, whose own session.created predates this plugin instance. These
// sets live for the process and are cleared only on server.instance.disposed.
const childSessions = new Set<string>()
// child session ID -> the top-level (user's) session it descends from. With
// subagent_depth > 1 a child can itself call the task tool; that
// grandchild's task is recorded on the top-level session, the only one
// Entire tracks, keyed by its own callID like any other task.
const rootOf = new Map<string, string>()
// child session ID -> the session that launched it, so aborting a session
// can end the background calls it launched.
const parentOf = new Map<string, string>()
// The per-call maps and sets below are keyed by callKey: the calling
// session plus the hook key. A callID is unique only within one session
// (some providers number calls per conversation), so a child's own task
// call can share one with its parent's running call.
// Call keys already announced via subagent-start (the running part update
// repeats), so a late running update cannot announce a task again after its
// stop. Cleared for a key when a new call starts under it (a reused callID).
const announcedTasks = new Set<string>()
// Call key -> Date.now() at tool.execute.before, sent as started_at. A
// child resumed via `task_id` holds every earlier call's messages too; this
// is what lets the CLI keep only the ones this call produced. The hook key
// is the part's callID, except for a command subtask (`subtask: true`),
// whose before/after hooks carry the task part's own id instead.
const taskStartedAt = new Map<string, number>()
// Announced tasks not yet stopped: call key -> its subagent-stop payload.
// Every stop path takes the entry out, so a task stops at most once.
const liveTasks = new Map<string, Record<string, unknown>>()
// Call key of a task part id -> call key of its callID, to map a command
// subtask's hook key to the callID its subagent-start used.
const partCallID = new Map<string, string>()
// Call keys whose stop fired before their running part was announced. The
// late announcement is then skipped rather than opening a task that would
// never stop.
const stoppedBeforeStart = new Set<string>()
// Call keys of tasks stopped from their error part, so a
// tool.execute.after that still arrives does not stop them again.
const stoppedOnError = new Set<string>()
// Background children (OPENCODE_EXPERIMENTAL_BACKGROUND_SUBAGENTS, or a
// foreground task promoted to the background): child ID -> subagent-stop
// payloads held back until the child's own session goes idle. A queue: a
// resume via `task_id` that joins a running child is queued behind the
// current run and runs as its own busy -> idle cycle, so each idle ends the
// oldest held call.
const backgroundTasks = new Map<string, Record<string, unknown>[]>()
// Sessions whose last session.status was not idle. An idle only ends a
// held call when it follows a busy period: an errored run reports idle
// twice.
const busySessions = new Set<string>()
// Children whose held stops all end at their next idle: an abort reached
// them mid-run. Cancelling a job does not cancel a joined run already
// executing, so the child's own idle, not the abort, is when it ends.
const drainAtIdle = new Set<string>()
/**
* Build the shell command for a hook invocation.
* Uses sh -c so the command is guarded by a `command -v` probe: when the
* entire binary is not on PATH the hook exits 0 rather than failing the
* surrounding OpenCode operation.
*/
function hookCmd(hookName: string): string[] {
return ["sh", "-c", `if ! command -v entire >/dev/null 2>&1; then exit 0; fi; exec entire hooks opencode ${hookName}`]
}
/**
* Pipe JSON payload to an entire hooks command (async).
* Errors are logged but never thrown — plugin failures must not crash OpenCode.
*/
async function callHook(hookName: string, payload: Record<string, unknown>) {
try {
const json = JSON.stringify(payload)
const [cmd, ...args] = hookCmd(hookName)
await new Promise<void>((resolve) => {
const proc = spawn(cmd, args, {
cwd: directory,
stdio: ["pipe", "ignore", "ignore"],
})
proc.on("error", () => resolve())
proc.on("close", () => resolve())
proc.stdin?.end(json + "\n")
})
} catch {
// Silently ignore — plugin failures must not crash OpenCode
}
}
/**
* Synchronous variant for hooks that must complete before subsequent agent work
* or process exit. `turn-start` must finish initializing session state before a
* fast mid-turn commit can hit git hooks, and `turn-end` / `session-end` must
* finish before `opencode run` tears down its event loop.
*/
function callHookSync(hookName: string, payload: Record<string, unknown>) {
try {
const json = JSON.stringify(payload)
const [cmd, ...args] = hookCmd(hookName)
spawnSync(cmd, args, {
cwd: directory,
input: json + "\n",
stdio: ["pipe", "ignore", "ignore"],
})
} catch {
// Silently ignore — plugin failures must not crash OpenCode
}
}
// parseInjectedContext scans a hook's stdout for Entire's injection envelope
// ({"inject_context":"..."}) and returns the text to inject, or null.
function parseInjectedContext(stdout: string): string | null {
if (!stdout) return null
for (const line of stdout.split("\n")) {
const trimmed = line.trim()
if (!trimmed.startsWith("{")) continue
try {
const parsed = JSON.parse(trimmed) as { inject_context?: unknown }
if (typeof parsed.inject_context === "string" && parsed.inject_context.length > 0) {
return parsed.inject_context
}
} catch {
// not our envelope — ignore
}
}
return null
}
// fireTurnStart fires turn-start synchronously (state must be ready before
// mid-turn commits) and stashes any model-context injection emitted on stdout
// for experimental.chat.system.transform to apply. Entire emits the injection
// at most once per session, so pendingInjection is set on the first turn only.
function fireTurnStart(payload: Record<string, unknown>) {
try {
const json = JSON.stringify(payload)
const [cmd, ...args] = hookCmd("turn-start")
const proc = spawnSync(cmd, args, {
cwd: directory,
input: json + "\n",
encoding: "utf8",
stdio: ["pipe", "pipe", "ignore"],
})
const out = typeof proc.stdout === "string" ? proc.stdout : ""
const injected = parseInjectedContext(out)
if (injected) pendingInjection = injected
} catch {
// Silently ignore — plugin failures must not crash OpenCode
}
}
// trackChild records a child and the top-level session it descends from.
function trackChild(childID: string, parentID: string) {
childSessions.add(childID)
if (!rootOf.has(childID)) rootOf.set(childID, rootOf.get(parentID) ?? parentID)
if (!parentOf.has(childID)) parentOf.set(childID, parentID)
}
// callKey scopes a hook key (a callID or a task part id) to the session
// that made the call.
function callKey(sessionID: string, hookKey: string): string {
return sessionID + "\0" + hookKey
}
// topLevelSession maps a session to the top-level session it belongs to.
function topLevelSession(sessionID: string): string {
return rootOf.get(sessionID) ?? sessionID
}
// announceTask fires subagent-start the first time a task part is running
// with its child bound (state.metadata.sessionId). tool.execute.before fires
// earlier but has no child ID yet, and session.created for the child can
// interleave with a sibling's, so neither is a safe join. The running update
// repeats, hence once per call key. Called for task parts in child sessions
// too, so a nested subagent is announced on the top-level session.
function announceTask(part: any) {
if (!(part?.type === "tool" && part.tool === "task" && part.callID &&
part.state?.status === "running" && part.state?.metadata?.sessionId)) return
const sessionID = part.sessionID ?? currentSessionID
if (!sessionID) return
const key = callKey(sessionID, part.callID)
const partKey = callKey(sessionID, part.id)
if (announcedTasks.has(key)) return
announcedTasks.add(key)
trackChild(part.state.metadata.sessionId, sessionID)
// Without the before hook (the plugin loaded mid-call), the part's own
// start, OpenCode's clock before the child's prompt, still bounds it.
const startedAt = taskStartedAt.get(key) ?? taskStartedAt.get(partKey) ?? part.state?.time?.start ?? 0
if (stoppedBeforeStart.delete(key) || stoppedBeforeStart.delete(partKey)) {
forgetTaskKeys(part, sessionID)
return
}
const payload = {
session_id: topLevelSession(sessionID),
tool_use_id: part.callID,
subagent_id: part.state.metadata.sessionId,
subagent_type: part.state?.input?.subagent_type ?? "",
task_description: part.state?.input?.description ?? "",
started_at: startedAt,
}
liveTasks.set(key, payload)
if (part.id) partCallID.set(partKey, key)
callHookSync("subagent-start", payload)
}
// announcedKey maps a call key to the call key the task was announced under.
function announcedKey(key: string): string {
return partCallID.get(key) ?? key
}
// takeLiveTask removes and returns the announced, not yet stopped task a
// call key belongs to.
function takeLiveTask(key: string): Record<string, unknown> | undefined {
const announced = announcedKey(key)
partCallID.delete(key)
const payload = liveTasks.get(announced)
liveTasks.delete(announced)
return payload
}
function forgetTaskKeys(part: any, sessionID: string) {
taskStartedAt.delete(callKey(sessionID, part.callID))
if (part.id) taskStartedAt.delete(callKey(sessionID, part.id))
}
// stopFailedTask fires subagent-stop for an announced task whose part
// ended in error without tool.execute.after: OpenCode skips that hook when
// the call is aborted (Esc, `opencode run` teardown) or its execute throws.
// A task that failed before its child was bound was never announced, so
// nothing fires for it.
function stopFailedTask(part: any) {
if (!(part?.type === "tool" && part.tool === "task" && part.callID &&
part.state?.status === "error")) return
const sessionID = part.sessionID ?? currentSessionID ?? ""
const key = callKey(sessionID, part.callID)
const partKey = callKey(sessionID, part.id)
const payload = takeLiveTask(key)
forgetTaskKeys(part, sessionID)
stoppedBeforeStart.delete(key)
if (part.id) {
partCallID.delete(partKey)
stoppedBeforeStart.delete(partKey)
}
if (!payload) return
stoppedOnError.add(key)
if (part.id) stoppedOnError.add(partKey)
callHookSync("subagent-stop", payload)
}
// finishBackgroundTask fires the oldest held subagent-stop for a background
// child whose run went idle: its work finished, failed, or was aborted
// (`opencode run` exiting with the child still running). Each case ends the
// call; a later call queued on the same child ends at its own run's idle.
function finishBackgroundTask(childID: string) {
const payloads = backgroundTasks.get(childID)
const payload = payloads?.shift()
if (!payload) return
if (payloads.length === 0) backgroundTasks.delete(childID)
callHookSync("subagent-stop", payload)
}
// drainBackgroundTasks fires every held subagent-stop for a child whose
// background job is over. A job that fails or is cancelled drops the runs
// still queued on it, so their calls get no idle of their own; left held,
// the next resume of the child would end them at its idle instead.
function drainBackgroundTasks(childID: string) {
const payloads = backgroundTasks.get(childID)
if (!payloads) return
backgroundTasks.delete(childID)
for (const payload of payloads) callHookSync("subagent-stop", payload)
}
// A background job's end reaches the session that launched it as a
// synthetic text part: <task id="<child>" state="completed|error">.
const backgroundResult = /^<task id="([^"]+)" state="(completed|error)">/
// backgroundSessionAborted ends the held calls of an aborted session and of
// the children it launched (Esc cancels its background jobs and reports no
// result): at once for an idle child, at its next idle for a busy one.
function backgroundSessionAborted(sessionID: string) {
const targets = [sessionID]
for (const [childID, parentID] of parentOf) if (parentID === sessionID) targets.push(childID)
for (const target of targets) {
if (!backgroundTasks.has(target)) continue
if (busySessions.has(target)) drainAtIdle.add(target)
else drainBackgroundTasks(target)
}
}
function resetSessionTracking(sessionID: string) {
if (currentSessionID === sessionID) {
return false
}
seenUserMessages.clear()
messageStore.clear()
currentModel = null
currentSessionID = sessionID
// Drop any turn-start injection captured for the prior session so it
// can't leak into the new session's system prompt.
pendingInjection = null
return true
}
return {
// Apply the one-time Entire context injection captured at turn-start by
// appending it to the system prompt for this LLM call.
"experimental.chat.system.transform": async (input: { sessionID?: string }, output: { system: string[] }) => {
if (input?.sessionID && childSessions.has(input.sessionID)) return
if (pendingInjection && Array.isArray(output.system)) {
output.system.push(pendingInjection)
pendingInjection = null
}
},
// Subagent call start. No child ID exists yet, so this only records the
// time; subagent-start fires from the running task part below.
"tool.execute.before": async (input) => {
try {
if (input.tool !== "task") return
const key = callKey(input.sessionID, input.callID)
// A new call under this key: some providers number calls per
// conversation, so an earlier call's markers must not dedupe or
// swallow this one's start and stop.
announcedTasks.delete(key)
stoppedBeforeStart.delete(key)
stoppedOnError.delete(key)
taskStartedAt.set(key, Date.now())
} catch {
// Silently ignore — plugin failures must not crash OpenCode
}
},
// Subagent completion. tool.execute.after for the task tool fires once, with
// the child session ID in output.metadata: at true completion for a
// foreground task, immediately for a background one (metadata.background),
// whose stop is held until the child goes idle. Synchronous: `opencode run`
// exits on the parent's idle right after this, and an async hook would be
// killed before it finished.
"tool.execute.after": async (input, output) => {
try {
if (input.tool !== "task") return
const key = callKey(input.sessionID, input.callID)
const announced = takeLiveTask(key)
if (stoppedOnError.delete(key) && !announced) {
// Already stopped: its part ended in error first.
taskStartedAt.delete(key)
return
}
// A failed command subtask fires this hook with no output; its
// announced child still ends here.
const childID = output?.metadata?.sessionId ?? announced?.subagent_id
if (!childID) {
taskStartedAt.delete(key)
return
}
trackChild(childID, input.sessionID)
const startedAt = taskStartedAt.get(key) ?? 0
taskStartedAt.delete(key)
if (!announced) stoppedBeforeStart.add(key)
// A child's own task call (subagent_depth > 1) is recorded on the
// top-level session: the child has no Entire session of its own.
const payload = announced ?? {
session_id: topLevelSession(input.sessionID),
tool_use_id: input.callID,
subagent_id: childID,
subagent_type: input.args?.subagent_type ?? "",
task_description: input.args?.description ?? "",
started_at: startedAt,
}
if (output?.metadata?.background === true) {
backgroundTasks.set(childID, [...(backgroundTasks.get(childID) ?? []), payload])
return
}
callHookSync("subagent-stop", payload)
} catch {
// Silently ignore — plugin failures must not crash OpenCode
}
},
event: async ({ event }) => {
try {
const props = (event as any).properties
const info = props?.info
if (event.type.startsWith("session.") && info?.parentID && info?.id) trackChild(info.id, info.parentID)
if (event.type === "message.part.updated") {
announceTask(props?.part)
stopFailedTask(props?.part)
const part = props?.part
// Only a part in a tracked session: synthetic text also carries
// attachment contents, which could spell the same envelope.
const sid = part?.sessionID
const tracked = sid === currentSessionID || childSessions.has(sid) || [...parentOf.values()].includes(sid)
if (tracked && part?.type === "text" && part.synthetic === true && typeof part.text === "string") {
const result = backgroundResult.exec(part.text)
if (result) drainBackgroundTasks(result[1])
}
}
// An aborted message, reported on session.error, or on the message
// itself when the run was interrupted outside the processor.
if (event.type === "session.error" && props?.error?.name === "MessageAbortedError" && props?.sessionID) {
backgroundSessionAborted(props.sessionID)
}
if (event.type === "message.updated" && info?.error?.name === "MessageAbortedError" && info?.sessionID) {
backgroundSessionAborted(info.sessionID)
}
if (event.type === "session.status" && props?.sessionID) {
if (props?.status?.type !== "idle") busySessions.add(props.sessionID)
else if (busySessions.delete(props.sessionID)) {
if (drainAtIdle.delete(props.sessionID)) drainBackgroundTasks(props.sessionID)
else finishBackgroundTask(props.sessionID)
}
}
const eventSessionID: string | undefined =
props?.sessionID ?? info?.sessionID ?? info?.id ?? props?.part?.sessionID
if (eventSessionID && childSessions.has(eventSessionID)) return
switch (event.type) {
case "session.created": {
const session = info
if (!session?.id) break
// Reset per-session tracking state when switching sessions.
if (resetSessionTracking(session.id)) {
await callHook("session-start", {
session_id: session.id,
})
}
break
}
case "message.updated": {
const msg = info
if (!msg) break
if (msg.sessionID && resetSessionTracking(msg.sessionID)) {
callHookSync("session-start", {
session_id: msg.sessionID,
})
}
// Store message metadata (role, time, tokens, etc.)
messageStore.set(msg.id, msg)
// Track model from assistant messages
if (msg.role === "assistant" && msg.modelID) {
currentModel = msg.modelID
}
// Fallback: some opencode run flows commit before any message.part.updated
// event is delivered for the user's prompt. Start the turn from the
// user message itself so git hooks see an ACTIVE session in time.
if (msg.role === "user" && !seenUserMessages.has(msg.id)) {
seenUserMessages.add(msg.id)
const sessionID = msg.sessionID ?? currentSessionID
if (sessionID) {
fireTurnStart({
session_id: sessionID,
prompt: "",
model: currentModel ?? "",
})
}
}
break
}
case "message.part.updated": {
const part = props?.part
if (!part?.messageID) break
// Fire turn-start on the first text part of a new user message
const msg = messageStore.get(part.messageID)
if (msg?.role === "user" && part.type === "text" && !seenUserMessages.has(msg.id)) {
seenUserMessages.add(msg.id)
const sessionID = msg.sessionID ?? currentSessionID
if (sessionID) {
// A synthetic part is OpenCode writing, not the user: a
// background task's result arrives this way. The parent does
// take a turn on it, so the turn still starts, with no prompt.
fireTurnStart({
session_id: sessionID,
prompt: part.synthetic === true ? "" : (part.text ?? ""),
model: currentModel ?? "",
})
}
}
break
}
case "session.status": {
// session.status fires in both TUI and non-interactive (run) mode.
// session.idle is deprecated and not reliably emitted in run mode.
if (props?.status?.type !== "idle") break
const sessionID = props?.sessionID ?? currentSessionID
if (!sessionID) break
// Use sync variant: `opencode run` exits on the same idle event,
// so an async hook would be killed before completing.
callHookSync("turn-end", {
session_id: sessionID,
model: currentModel ?? "",
})
break
}
case "session.compacted": {
const sessionID = props?.sessionID
if (!sessionID) break
await callHook("compaction", {
session_id: sessionID,
})
break
}
case "session.deleted": {
const session = info
if (!session?.id) break
seenUserMessages.clear()
messageStore.clear()
currentSessionID = null
pendingInjection = null
// childSessions/announcedTasks are NOT cleared here: a child's own
// deletion never reaches this case (the guard above returns first),
// so this only ever fires for an unrelated top-level session, and
// clearing here would drop a live child back to top-level mid-task.
// Use sync variant: session-end may fire during shutdown.
callHookSync("session-end", {
session_id: session.id,
})
break
}
case "server.instance.disposed": {
// Fires when OpenCode shuts down (TUI close or `opencode run` exit).
// session.deleted only fires on explicit user deletion, not on quit,
// so this is the only reliable way to end sessions on exit.
if (!currentSessionID) break
const sessionID = currentSessionID
seenUserMessages.clear()
messageStore.clear()
currentSessionID = null
pendingInjection = null
childSessions.clear()
rootOf.clear()
parentOf.clear()
announcedTasks.clear()
taskStartedAt.clear()
liveTasks.clear()
partCallID.clear()
stoppedBeforeStart.clear()
stoppedOnError.clear()
// A background child still running here is ended by the Go side's
// SessionEnd sweep, which completes every live task record.
backgroundTasks.clear()
busySessions.clear()
drainAtIdle.clear()
// Use sync variant: this is the last event before process exit.
callHookSync("session-end", {
session_id: sessionID,
})
break
}
}
} catch {
// Silently ignore — plugin failures must not crash OpenCode
}
},
}
}