#!/bin/sh
set -eu

# Install or upgrade oy, OpenCode 2, and compact context helpers through mise.
#
# Intended curl usage:
#   curl -fsSL https://oy.adonm.dev/install.sh | sh
#   curl -fsSL https://oy.adonm.dev/install.sh | sh -s -- --workspace
#
# Environment knobs:
#   OY_INSTALL_SCOPE  global or workspace; an explicit flag wins
#   OY_SKIP_SETUP     1/true to skip `oy setup` and runtime load checks

oy_version="0.14.4"
oy_tool="github:adonm/oy-cli@$oy_version"
node_tool="node@latest"
opencode_package="@opencode-ai/cli@next"
tokei_tool="aqua:XAMPPRocky/tokei@12.1.2"
ctags_tool="github:universal-ctags/ctags-nightly-build[matching=.release.tar.gz]"

log() {
  printf '%s\n' "$*" >&2
}

die() {
  log "error: $*"
  exit 1
}

show_help() {
  cat <<'EOF'
Install oy, OpenCode 2, and context helpers with mise.

Usage:
  install.sh [--global|--workspace] [--yes]

Scope:
  --global     Add tools to mise's global config
  --workspace  Add tools to mise.toml in the current directory
  --yes        Skip the scope prompt; defaults to global when no scope is given

Environment:
  OY_INSTALL_SCOPE  global or workspace; an explicit flag wins
  OY_SKIP_SETUP     1/true skips OpenCode integration setup and load checks
EOF
}

scope_flag=
assume_yes=0
while [ "$#" -gt 0 ]; do
  case "$1" in
  --global) requested_scope=global ;;
  --workspace) requested_scope=workspace ;;
  --yes | -y) assume_yes=1 ;;
  -h | --help)
    show_help
    exit 0
    ;;
  *) die "unknown installer option: $1" ;;
  esac
  if [ -n "${requested_scope:-}" ] && [ -n "$scope_flag" ]; then
    die "choose only one of --global or --workspace"
  fi
  if [ -n "${requested_scope:-}" ]; then
    scope_flag=$requested_scope
    requested_scope=
  fi
  shift
done

tty_available() {
  [ "$assume_yes" -eq 0 ] || return 1
  [ -t 1 ] || return 1
  [ -r /dev/tty ] && [ -w /dev/tty ] || return 1
  ( : </dev/tty ) 2>/dev/null
}

scope=${scope_flag:-${OY_INSTALL_SCOPE:-}}
case "$scope" in
global | workspace) ;;
"")
  if tty_available; then
    printf 'Install mise-managed tools globally or in the current workspace? [G/w] ' >/dev/tty
    read -r scope_reply </dev/tty || scope_reply=
    case "$(printf '%s' "$scope_reply" | tr '[:upper:]' '[:lower:]')" in
    w | workspace) scope=workspace ;;
    "" | g | global) scope=global ;;
    *) die "choose global or workspace" ;;
    esac
  else
    scope=global
  fi
  ;;
*) die "OY_INSTALL_SCOPE must be global or workspace" ;;
esac

case "$(uname -s)" in
Linux | Darwin) ;;
*) die "oy supports Linux and macOS only; Windows users should run the installer in WSL2" ;;
esac

find_mise() {
  if command -v mise >/dev/null 2>&1; then
    command -v mise
  elif [ -n "${MISE_INSTALL_PATH:-}" ] && [ -x "$MISE_INSTALL_PATH" ]; then
    printf '%s\n' "$MISE_INSTALL_PATH"
  elif [ -x "$HOME/.local/bin/mise" ]; then
    printf '%s\n' "$HOME/.local/bin/mise"
  else
    return 1
  fi
}

case "${SHELL:-}" in
*/bash | bash) shell_target=bash ;;
*/zsh | zsh) shell_target=zsh ;;
*/fish | fish) shell_target=fish ;;
*) shell_target= ;;
esac

install_mise() {
  if [ -n "$shell_target" ]; then
    mise_url="https://mise.run/$shell_target"
    log "Installing mise and configuring $shell_target activation..."
  else
    mise_url="https://mise.run"
    log "Installing mise (shell activation skipped because SHELL=${SHELL:-unset})..."
  fi
  if command -v curl >/dev/null 2>&1; then
    curl -fsSL "$mise_url" | sh
  elif command -v wget >/dev/null 2>&1; then
    wget -qO- "$mise_url" | sh
  else
    die "mise is not installed and neither curl nor wget is available"
  fi
}

mise_bin="$(find_mise || true)"
installed_mise=0
if [ -z "$mise_bin" ]; then
  install_mise
  installed_mise=1
  mise_bin="$(find_mise || true)"
fi
[ -n "$mise_bin" ] || die "mise installed, but no mise executable was found on PATH, at MISE_INSTALL_PATH, or at ~/.local/bin/mise"

if [ "$installed_mise" -eq 0 ]; then
  log "Updating mise itself when supported..."
  if "$mise_bin" self-update --yes; then
    mise_bin="$(find_mise || true)"
    [ -n "$mise_bin" ] || die "mise self-update completed, but no mise executable was found"
  else
    log "Skipping mise self-update; this is normal for package-manager installs."
  fi
fi

mise_use() {
  if [ "$scope" = "global" ]; then
    "$mise_bin" use --global --yes --minimum-release-age 0 "$@"
  else
    "$mise_bin" use --yes --minimum-release-age 0 "$@"
  fi
}

mise_unuse() {
  if [ "$scope" = "global" ]; then
    "$mise_bin" unuse --global --yes --no-prune "$@"
  else
    "$mise_bin" unuse --yes --no-prune "$@"
  fi
}

log "Mise scope: $scope"
log "Installing/upgrading oy and OpenCode with mise..."
mise_use "$oy_tool" "$node_tool"

log "Installing OpenCode 2 with npm as documented upstream..."
"$mise_bin" exec "$node_tool" -- npm install -g "$opencode_package"

log "Installing optional prebuilt context helpers..."
if ! mise_use "$tokei_tool" "$ctags_tool"; then
  log "Warning: optional context helpers could not be installed; rerun this installer later."
fi

log "Removing superseded source/package-manager tool entries..."
mise_unuse \
  cargo:oy-cli \
  "npm:@opencode-ai/cli" \
  cargo:tokei \
  github:universal-ctags/ctags
"$mise_bin" reshim

installed_oy_version=$("$mise_bin" exec "$oy_tool" -- oy --version 2>/dev/null) \
  || die "oy installed, but oy --version failed"
case "$installed_oy_version" in
*"$oy_version"*) ;;
*) die "expected oy $oy_version after install, got: $installed_oy_version" ;;
esac

installed_opencode_version=$("$mise_bin" exec "$node_tool" -- opencode2 --version 2>/dev/null) \
  || die "OpenCode 2 installed, but opencode2 --version failed"
case "$installed_opencode_version" in
*"0.0.0-next-"[0-9]*) ;;
*) die "expected an OpenCode 2 next-channel build after install, got: $installed_opencode_version" ;;
esac

log "Stopping any older OpenCode background service..."
if ! "$mise_bin" exec "$node_tool" -- opencode2 service stop >/dev/null 2>&1; then
  log "No running OpenCode service needed stopping."
fi

log "Pruning unreferenced old tool versions..."
prune_status=0
"$mise_bin" prune --yes --tools \
  github:adonm/oy-cli \
  cargo:oy-cli \
  "npm:@opencode-ai/cli" \
  cargo:tokei \
  github:universal-ctags/ctags || prune_status=$?
if [ "$prune_status" -ne 0 ]; then
  log "Warning: mise could not prune old versions; the newly installed versions remain active."
fi

case "${OY_SKIP_SETUP:-}" in
1 | true | TRUE | yes | YES)
  log "Skipping oy setup because OY_SKIP_SETUP is set."
  ;;
*)
  log "Installing the OpenCode integration with oy setup..."
  "$mise_bin" exec "$oy_tool" "$node_tool" -- oy setup
  log "Starting OpenCode so it can install the version-matched oy plugin..."
  "$mise_bin" exec "$node_tool" -- opencode2 service start >/dev/null \
    || die "OpenCode could not start after oy setup"
  workspace=$(pwd)
  log "Waiting for OpenCode to resolve and load the oy commands..."
  plugin_loaded=0
  attempts=0
  while [ "$attempts" -lt 60 ]; do
    loaded_plugins=$("$mise_bin" exec "$node_tool" -- opencode2 api v2.command.list \
      --param "location[directory]=$workspace" 2>/dev/null || true)
    case "$loaded_plugins" in
    *'"name":"oy-audit"'* | *'"name": "oy-audit"'*)
      plugin_loaded=1
      break
      ;;
    esac
    attempts=$((attempts + 1))
    sleep 2
  done
  [ "$plugin_loaded" -eq 1 ] \
    || die "OpenCode started, but the oy commands did not load within 120 seconds; run 'oy doctor --check' for details"
  log "Verified OpenCode loaded the oy commands."
  ;;
esac

log "Done."
if [ "$installed_mise" -eq 1 ] && [ -n "$shell_target" ]; then
  log "Restart your shell to load the mise activation configured by https://mise.run/$shell_target."
fi
if [ "$scope" = "workspace" ]; then
  log "Mise tools were added to the current workspace."
else
  log "Mise tools were added to the global config."
fi
log "Then run: oy doctor"