#!/bin/bash

# omarchy:summary=Set and launch the default coding agent
# omarchy:args=[pi|omp|opencode|ori|claude|codex|grok|agy|copilot|crush|kimi|deepcode]
# omarchy:examples=omarchy default agent | omarchy default agent codex | omarchy default agent kimi

installing=false
if [[ ${1:-} == "--install" ]]; then
  installing=true
  shift
fi

agent_file="$HOME/.config/omarchy/defaults/agent"

if (($# == 0)); then
  if [[ -f $agent_file ]]; then
    read -r agent <"$agent_file"
  fi

  # Silent when unset rather than defaulting: Omarchy picks no agent for you, so
  # the menu leaves every entry unchecked until one is chosen.
  [[ -n ${agent:-} ]] && echo "$agent"
  exit 0
fi

case "$1" in
pi) agent="pi"; name="Pi" ;;
omp | oh-my-pi) agent="omp"; name="Oh My Pi"; agent_package="github:can1357/oh-my-pi" ;;
opencode | open-code) agent="opencode"; name="OpenCode" ;;
ori | openrouter) agent="ori"; name="Ori"; agent_package="github:OpenRouterLabs/ori-releases" ;;
claude | claude-code) agent="claude"; name="Claude Code" ;;
codex) agent="codex"; name="Codex" ;;
crush) agent="crush"; name="Crush" ;;
grok) agent="grok"; name="Grok"; agent_package="npm:@xai-official/grok" ;;
agy | antigravity | antigravity-cli | gemini | gemini-cli) agent="agy"; name="Antigravity"; agent_package="antigravity-cli" ;;
copilot | github-copilot) agent="copilot"; name="GitHub Copilot" ;;
kimi | kimi-code) agent="kimi"; name="Kimi Code" ;;
deepcode | deep-code) agent="deepcode"; name="Deep Code"; agent_package="npm:@vegamo/deepcode-cli" ;;
*)
  echo "Usage: omarchy-default-agent <pi|omp|opencode|ori|claude|codex|grok|agy|copilot|crush|kimi|deepcode>"
  exit 1
  ;;
esac

agent_package=${agent_package:-$agent}

# Kimi Code installs via its official script, not mise
if [[ $agent == "kimi" ]]; then
  if omarchy-cmd-missing kimi; then
    if [[ $installing == "false" ]]; then
      exec omarchy-launch-floating-terminal-with-presentation omarchy-default-agent --install kimi
    fi
    if ! curl -fsSL https://code.kimi.com/kimi-code/install.sh | bash; then
      echo "Could not install $name with the official installer" >&2
      exit 1
    fi
  fi
else
  if [[ $installing == "false" ]] && ! mise where "$agent_package" &>/dev/null; then
    exec omarchy-launch-floating-terminal-with-presentation omarchy-default-agent --install "$agent"
  fi

  if ! mise use -g "$agent_package"; then
    if [[ $installing == "true" ]]; then
      echo "Could not install $name with mise" >&2
    else
      echo "Could not set $name as the default coding agent" >&2
    fi
    exit 1
  fi
fi

mkdir -p "$(dirname "$agent_file")"
printf '%s\n' "$agent" >"$agent_file"

if [[ $installing == "true" ]]; then
  printf '\033[2J\033[3J\033[H'
  exec omarchy-agent --inline
else
  exec omarchy-agent
fi
