Query agentapi for status

Author: code-asher

registry/coder/modules/claude-code/main.tf

@@ -72,6 +72,12 @@ variable "experiment_report_tasks" {
   default     = false
 }
 
+variable "experiment_report_tasks_interval" {
+  type        = number
+  description = "How often to poll task status in seconds, or zero to disable."
+  default     = 0
+}
+
 variable "experiment_pre_install_script" {
   type        = string
   description = "Custom script to run before installing Claude Code."
@@ -247,6 +253,19 @@ resource "coder_app" "claude_code_web" {
   subdomain    = true
 }
 
+resource "coder_script" "task_reporter" {
+  agent_id     = var.agent_id
+  display_name = "Task reporter"
+  icon         = var.icon
+  # TODO: Should this be added into `coder exp` instead of being a bash script?
+  script = templatefile("${path.module}/reporter.bash", {
+    # Using the `cron` option would be nice, but it does not support sub-minute
+    # resolutions, so the script will loop itself.
+    INTERVAL : var.experiment_report_tasks_interval,
+  })
+  run_on_start = true
+}
+
 resource "coder_app" "claude_code" {
   slug         = "claude-code"
   display_name = "Claude Code"

registry/coder/modules/claude-code/reporter.bash

@@ -0,0 +1,157 @@
+#!/usr/bin/env bash
+set -e
+
+lastmessage=""
+laststatus=""
+
+# json-stringify returns a json string for updating the status.
+# $1: status slug
+# $2: status
+# $3: message
+# $4: uri
+function json-stringify() {
+  node -e "process.stdout.write(JSON.stringify({
+    app_slug: process.argv[1],
+    state: process.argv[2],
+    message: process.argv[3].substring(0, 160),
+    uri: process.argv[4],
+  }))" "$@"
+}
+
+# json-parse returns the specified key from the provided json.
+# $1: the key
+# $2: the json
+function json-parse() {
+  node -e "process.stdout.write(JSON.parse(process.argv[2])[process.argv[1]])" "$@"
+}
+
+# run queries for the status then relays it to coderd.
+# $1: workspace agent uri
+# $2: workspace token
+# $3: llm agent api uri
+# $4: status slug
+function run() {
+  local agenturi=$1 ; shift
+  local token=$1 ; shift
+  local agentapiuri=$1 ; shift
+  local slug=$1 ; shift
+  local resp
+  # TODO: this only returns 'stable' or 'running'.
+  # 1. We need a message.
+  # 2. We need 'working', 'complete', or 'failure' for the status.
+  # 3. Not sure yet what the URI is but maybe we need that.
+  # Do we add a new endpoint or modify this one?
+  local status
+  local message
+  local uri=""
+  if ! resp=$(curl --fail-with-body --silent --show-error --location \
+                   "$agentapiuri/status" 2>&1) ; then
+    message=$(echo "$resp" | xargs)
+    status=failure
+  elif ! status=$(json-parse "status" "$resp" 2>&1) ; then
+    message=$(echo "$status" | xargs)
+    status=failure
+  elif [[ $status == stable ]] ; then
+    message=complete
+    status=complete
+  else
+    message=working
+    status=working
+  fi
+  if [[ $message != $lastmessage ]] || [[ $status != $laststatus ]] ; then
+    local data
+    if ! data=$(json-stringify "$slug" "$status" "$message" "$uri" 2>&1) ; then
+      >&2 echo "Failed to update status"
+      >&2 echo "$data"
+    elif ! resp=$(curl --request PATCH \
+                       --fail-with-body --silent --show-error --location \
+                       --data "$data" \
+                       --header "Content-Type: application/json" \
+                       --header "Coder-Session-Token: $token" \
+                       "$agenturi/api/v2/workspaceagents/me/app-status" 2>&1) ; then
+      >&2 echo "Failed to update status"
+      >&2 echo "$resp"
+      >&2 echo "message: $message"
+      >&2 echo "status: $status"
+    else
+      lastmessage=$message
+      laststatus=$status
+    fi
+  fi
+}
+
+# loop runs the status updater every $1 seconds.
+# $1: interval in seconds
+# $2: workspace agent uri
+# $3: workspace agent token
+# $4: llm agent api uri
+# $5: status slug
+function loop() {
+  local interval=$1 ; shift
+  local agenturi=$1 ; shift
+  local token=$1 ; shift
+  local agentapiuri=$1 ; shift
+  local slug=$1 ; shift
+  while sleep "$interval" ; do
+    run "$agenturi" "$token" "$agentapiuri" "$slug"
+  done
+}
+
+# configure checks for required vars then starts polling in the background.
+# $1: interval in seconds
+function configure() {
+  local interval=$1 ; shift
+  echo "Configuring task reporting via agent polling..."
+  # Node will be used to handle JSON.  Since the module already requires npm,
+  # this should not be an extra requirement.
+  if ! command -v "node" >/dev/null 2>&1 ; then
+    >&2 echo "Node must be installed"
+    exit 1
+  fi
+  local agenturi=${CODER_AGENT_URL-}
+  if [[ -z $agenturi ]] ; then
+    >&2 echo "CODER_AGENT_URL must be set"
+    exit 1
+  fi
+  local token=${CODER_AGENT_TOKEN-}
+  if [[ -z $token ]] ; then
+    if [[ -z ${CODER_AGENT_TOKEN_FILE-} ]] ; then
+      >&2 echo "Either CODER_AGENT_TOKEN or CODER_AGENT_TOKEN_FILE must be set"
+      exit 1
+    elif [[ -f $CODER_AGENT_TOKEN_FILE ]] ; then
+      if ! token=$(<"$CODER_AGENT_TOKEN_FILE") ; then
+        >&2 echo "$CODER_AGENT_TOKEN_FILE could not be read"
+        exit 1
+      fi
+    else
+      >&2 echo "$CODER_AGENT_TOKEN_FILE does not exist or could not be accessed"
+      exit 1
+    fi
+  fi
+  local slug=${CODER_MCP_APP_STATUS_SLUG-}
+  if [[ -z $slug ]] ; then
+    >&2 echo "$CODER_MCP_APP_STATUS_SLUG must be set"
+    exit 1
+  fi
+  local agentapiuri=${CODER_AGENT_API_URL-}
+  if [[ -z $agentapiuri ]] ; then
+    agentapiuri="http://localhost:3284"
+  fi
+  echo "Configured a $interval second poll"
+  loop "$interval" "$agenturi" "$token" "$agentapiuri" "$slug" &
+}
+
+# main configures polling if enabled.
+# $1: interval in seconds
+function main() {
+  local interval=$1 ; shift
+  if [[ -z "$interval" ]] ; then
+    echo "Will not poll since interval is undefined or blank"
+  elif [[ "$interval" == "0" ]] ; then
+    echo "Will not poll since interval is zero"
+  else
+    configure "$interval"
+  fi
+}
+
+main "${INTERVAL}"