#!/bin/bash
# KO CLI Handoff Helper Script
# Usage: ./ko-handoff.sh <task-description> [--files <glob>] [--dry-run]

set -e

API_KEY="${KO_API_KEY:-}"
BASE_URL="${KO_BASE_URL:-https://api.ko.ai}"

if [ -z "$API_KEY" ]; then
  echo "Error: KO_API_KEY not set"
  echo "Set it with: export KO_API_KEY=your-key-here"
  exit 1
fi

# Parse arguments
TASK_DESC="${1:-}"
FILES_GLOB="${2:-}"
DRY_RUN=false

while [[ $# -gt 1 ]]; do
  case "$2" in
    --files)
      FILES_GLOB="$3"
      shift 2
      ;;
    --dry-run)
      DRY_RUN=true
      shift
      ;;
    *)
      shift
      ;;
  esac
done

if [ -z "$TASK_DESC" ]; then
  echo "Usage: ko-handoff.sh <task-description> [--files <glob>] [--dry-run]"
  echo ""
  echo "Examples:"
  echo "  ko-handoff.sh 'Refactor auth module' --files 'src/auth/**'"
  echo "  ko-handoff.sh 'Optimize database queries' --dry-run"
  exit 1
fi

# Build request
cat > /tmp/ko-request.json <<EOF
{
  "task": "$TASK_DESC",
  "files": "$FILES_GLOB",
  "dry_run": $DRY_RUN,
  "priority": "normal"
}
EOF

echo "📤 Handing off to KO CLI..."
echo "Task: $TASK_DESC"
[ -n "$FILES_GLOB" ] && echo "Files: $FILES_GLOB"
[ "$DRY_RUN" = true ] && echo "Mode: DRY-RUN (no credits consumed)"
echo ""

# Send to KO
RESPONSE=$(curl -s -X POST \
  -H "Authorization: Bearer $API_KEY" \
  -H "Content-Type: application/json" \
  -d @/tmp/ko-request.json \
  "$BASE_URL/v1/tasks")

# Parse response
TASK_ID=$(echo "$RESPONSE" | grep -o '"id":"[^"]*' | cut -d'"' -f4)
STATUS=$(echo "$RESPONSE" | grep -o '"status":"[^"]*' | cut -d'"' -f4)
CREDITS=$(echo "$RESPONSE" | grep -o '"estimated_credits":[0-9]*' | cut -d':' -f2)

if [ -z "$TASK_ID" ]; then
  echo "❌ Failed to hand off task"
  echo "Response: $RESPONSE"
  exit 1
fi

echo "✅ Task submitted"
echo "ID: $TASK_ID"
echo "Status: $STATUS"
[ -n "$CREDITS" ] && echo "Est. Credits: $CREDITS"
echo ""
echo "Check progress: ko status $TASK_ID"

rm -f /tmp/ko-request.json