ollie/data/scripts/acme/AddTests

42 lines
971 B
Bash
Executable File

#!/bin/bash
# AddTests - Ask ollie to write unit tests for the selected code
#
# Usage: Select code, then middle-click AddTests in the tag
set -euo pipefail
if [[ -z "${winid:-}" ]]; then
echo "AddTests must be run from within acme" >&2
exit 1
fi
# Get the window name (filename) from the tag
filename=$(9p read acme/$winid/tag | awk '{print $1}')
# Read the current selection
echo 'addr=dot' | 9p write acme/$winid/ctl
selection=$(9p read acme/$winid/xdata 2>/dev/null || true)
if [[ -z "$selection" ]]; then
echo "No selection. Select some code first." >&2
exit 1
fi
# Ensure acme session/agent exists
eval $(acme-ollie-ensure)
# Build location string
location=""
if [[ -n "$filename" && "$filename" != *"+Errors" && "$filename" != "+*" ]]; then
location=" in \`$filename\`"
fi
# Build and send the prompt
prompt="Write unit tests for this code${location}:
\`\`\`
$selection
\`\`\`"
echo "$prompt" | o "$OLLIE_SESSION/$OLLIE_AGENT" prompt