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, 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,128 @@
+#!/usr/bin/env bash
+set -e
+
+lastmessage=""
+laststatus=""
+
+# run queries for the status then relays it to coderd.
+# $1: workspace agent url
+# $2: workspace token
+# $3: llm agent api url
+# $4: status slug
+function run() {
+  local agenturl=$1 ; shift
+  local token=$1 ; shift
+  local agentapiurl=$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 --request GET --silent --show-error --location \
+     "$agentapiurl/status") ; then
+    message="Failed to fetch status"
+    status=failure
+  elif ! status=$(node -e "process.stdout.write(JSON.parse('$resp').status)" 2>&1) ; then
+    message="Failed to parse status"
+    status=failure
+  elif [[ $status == "stable" ]] ; then
+    message=complete
+    status=complete
+  else
+    message=working
+    status=working
+  fi
+  if [[ $message != $lastmessage ]] || [[ $status != $laststatus ]] ; then
+    if ! resp=$(curl --request PATCH --silent --show-error --location \
+                     --data "{\"app_slug\":\"$slug\",\"state\":\"$status\",\"message\":\"$message\",\"uri\": \"$uri\"}" \
+                     --header "Content-Type: application/json" \
+                     --header "Coder-Session-Token: $token" \
+                     "$agenturl/api/v2/workspaceagents/me/app-status") ; then
+      >&2 echo "Failed to update status"
+      >&2 echo "cURL response: $resp"
+    else
+      lastmessage=$message
+      laststatus=$status
+    fi
+  fi
+}
+
+# loop runs the status updater every $1 seconds.
+# $1: interval in seconds
+# $2: workspace agent url
+# $3: workspace agent token
+# $4: llm agent api url
+# $5: status slug
+function loop() {
+  local interval=$1 ; shift
+  local agenturl=$1 ; shift
+  local token=$1 ; shift
+  local agentapiurl=$1 ; shift
+  local slug=$1 ; shift
+  while sleep "$interval" ; do
+    run "$agenturl" "$token" "$agentapiurl" "$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 agenturl=${CODER_AGENT_URL-}
+  if [[ -z $agenturl ]] ; 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 agentapiurl=${CODER_AGENT_API_URL-}
+  if [[ -z $agentapiurl ]] ; then
+    agentapiurl="http://localhost:3284"
+  fi
+  echo "Configured a $interval second poll"
+  loop "$interval" "$agenturl" "$token" "$agentapiurl" "$slug" &
+}
+
+# main configures polling if enabled.
+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-}"