2169fb00ee
AgentAPI can now save and restore conversation state across workspace restarts. The module exports env vars (AGENTAPI_STATE_FILE, AGENTAPI_SAVE_STATE, AGENTAPI_LOAD_STATE, AGENTAPI_PID_FILE) that the binary reads directly. No consumer module changes needed. New variables: enable_state_persistence (default false), state_file_path, pid_file_path. State and PID files default to $HOME/<module_dir_name>/. Requires agentapi >= v0.12.0. A shared version_at_least function in lib.sh gates the env var exports and SIGUSR1 in the shutdown script. Old binaries get a warning and graceful skip. Shutdown script now does SIGUSR1 (state save), log snapshot capture (existing, now fault-tolerant via subshell), then SIGTERM with wait. Closes coder/internal#1257 Refs coder/internal#1256 Refs #696
130 lines
4.6 KiB
Bash
130 lines
4.6 KiB
Bash
#!/bin/bash
|
|
set -e
|
|
set -x
|
|
|
|
set -o nounset
|
|
MODULE_DIR_NAME="$ARG_MODULE_DIR_NAME"
|
|
WORKDIR="$ARG_WORKDIR"
|
|
PRE_INSTALL_SCRIPT="$ARG_PRE_INSTALL_SCRIPT"
|
|
INSTALL_SCRIPT="$ARG_INSTALL_SCRIPT"
|
|
INSTALL_AGENTAPI="$ARG_INSTALL_AGENTAPI"
|
|
AGENTAPI_VERSION="$ARG_AGENTAPI_VERSION"
|
|
START_SCRIPT="$ARG_START_SCRIPT"
|
|
WAIT_FOR_START_SCRIPT="$ARG_WAIT_FOR_START_SCRIPT"
|
|
POST_INSTALL_SCRIPT="$ARG_POST_INSTALL_SCRIPT"
|
|
AGENTAPI_PORT="$ARG_AGENTAPI_PORT"
|
|
AGENTAPI_CHAT_BASE_PATH="${ARG_AGENTAPI_CHAT_BASE_PATH:-}"
|
|
TASK_ID="${ARG_TASK_ID:-}"
|
|
TASK_LOG_SNAPSHOT="${ARG_TASK_LOG_SNAPSHOT:-true}"
|
|
ENABLE_STATE_PERSISTENCE="${ARG_ENABLE_STATE_PERSISTENCE:-false}"
|
|
STATE_FILE_PATH="${ARG_STATE_FILE_PATH:-}"
|
|
PID_FILE_PATH="${ARG_PID_FILE_PATH:-}"
|
|
set +o nounset
|
|
|
|
# shellcheck source=lib.sh
|
|
source /tmp/agentapi-lib.sh
|
|
|
|
command_exists() {
|
|
command -v "$1" > /dev/null 2>&1
|
|
}
|
|
|
|
module_path="$HOME/${MODULE_DIR_NAME}"
|
|
mkdir -p "$module_path/scripts"
|
|
|
|
# Check for jq dependency if task log snapshot is enabled.
|
|
if [[ $TASK_LOG_SNAPSHOT == true ]] && [[ -n $TASK_ID ]]; then
|
|
if ! command_exists jq; then
|
|
echo "Warning: jq is not installed. Task log snapshot requires jq to capture conversation history."
|
|
echo "Install jq to enable log snapshot functionality when the workspace stops."
|
|
fi
|
|
fi
|
|
if [ ! -d "${WORKDIR}" ]; then
|
|
echo "Warning: The specified folder '${WORKDIR}' does not exist."
|
|
echo "Creating the folder..."
|
|
mkdir -p "${WORKDIR}"
|
|
echo "Folder created successfully."
|
|
fi
|
|
if [ -n "${PRE_INSTALL_SCRIPT}" ]; then
|
|
echo "Running pre-install script..."
|
|
echo -n "${PRE_INSTALL_SCRIPT}" > "$module_path/pre_install.sh"
|
|
chmod +x "$module_path/pre_install.sh"
|
|
"$module_path/pre_install.sh" 2>&1 | tee "$module_path/pre_install.log"
|
|
fi
|
|
|
|
echo "Running install script..."
|
|
echo -n "${INSTALL_SCRIPT}" > "$module_path/install.sh"
|
|
chmod +x "$module_path/install.sh"
|
|
"$module_path/install.sh" 2>&1 | tee "$module_path/install.log"
|
|
|
|
# Install AgentAPI if enabled
|
|
if [ "${INSTALL_AGENTAPI}" = "true" ]; then
|
|
echo "Installing AgentAPI..."
|
|
arch=$(uname -m)
|
|
if [ "$arch" = "x86_64" ]; then
|
|
binary_name="agentapi-linux-amd64"
|
|
elif [ "$arch" = "aarch64" ]; then
|
|
binary_name="agentapi-linux-arm64"
|
|
else
|
|
echo "Error: Unsupported architecture: $arch"
|
|
exit 1
|
|
fi
|
|
if [ "${AGENTAPI_VERSION}" = "latest" ]; then
|
|
# for the latest release the download URL pattern is different than for tagged releases
|
|
# https://docs.github.com/en/repositories/releasing-projects-on-github/linking-to-releases
|
|
download_url="https://github.com/coder/agentapi/releases/latest/download/$binary_name"
|
|
else
|
|
download_url="https://github.com/coder/agentapi/releases/download/${AGENTAPI_VERSION}/$binary_name"
|
|
fi
|
|
curl \
|
|
--retry 5 \
|
|
--retry-delay 5 \
|
|
--fail \
|
|
--retry-all-errors \
|
|
-L \
|
|
-C - \
|
|
-o agentapi \
|
|
"$download_url"
|
|
chmod +x agentapi
|
|
sudo mv agentapi /usr/local/bin/agentapi
|
|
fi
|
|
if ! command_exists agentapi; then
|
|
echo "Error: AgentAPI is not installed. Please enable install_agentapi or install it manually."
|
|
exit 1
|
|
fi
|
|
|
|
echo -n "${START_SCRIPT}" > "$module_path/scripts/agentapi-start.sh"
|
|
echo -n "${WAIT_FOR_START_SCRIPT}" > "$module_path/scripts/agentapi-wait-for-start.sh"
|
|
chmod +x "$module_path/scripts/agentapi-start.sh"
|
|
chmod +x "$module_path/scripts/agentapi-wait-for-start.sh"
|
|
|
|
if [ -n "${POST_INSTALL_SCRIPT}" ]; then
|
|
echo "Running post-install script..."
|
|
echo -n "${POST_INSTALL_SCRIPT}" > "$module_path/post_install.sh"
|
|
chmod +x "$module_path/post_install.sh"
|
|
"$module_path/post_install.sh" 2>&1 | tee "$module_path/post_install.log"
|
|
fi
|
|
|
|
export LANG=en_US.UTF-8
|
|
export LC_ALL=en_US.UTF-8
|
|
|
|
cd "${WORKDIR}"
|
|
|
|
export AGENTAPI_CHAT_BASE_PATH="${AGENTAPI_CHAT_BASE_PATH:-}"
|
|
# Disable host header check since AgentAPI is proxied by Coder (which does its own validation)
|
|
export AGENTAPI_ALLOWED_HOSTS="*"
|
|
export AGENTAPI_PID_FILE="${PID_FILE_PATH:-$module_path/agentapi.pid}"
|
|
# Only set state env vars when persistence is enabled and the binary supports
|
|
# it. State persistence requires agentapi >= v0.12.0.
|
|
if [ "${ENABLE_STATE_PERSISTENCE}" = "true" ]; then
|
|
actual_version=$(agentapi_version)
|
|
if version_at_least 0.12.0 "$actual_version"; then
|
|
export AGENTAPI_STATE_FILE="${STATE_FILE_PATH:-$module_path/agentapi-state.json}"
|
|
export AGENTAPI_SAVE_STATE="true"
|
|
export AGENTAPI_LOAD_STATE="true"
|
|
else
|
|
echo "Warning: State persistence requires agentapi >= v0.12.0 (current: ${actual_version:-unknown}), skipping."
|
|
fi
|
|
fi
|
|
nohup "$module_path/scripts/agentapi-start.sh" true "${AGENTAPI_PORT}" &> "$module_path/agentapi-start.log" &
|
|
"$module_path/scripts/agentapi-wait-for-start.sh" "${AGENTAPI_PORT}"
|