#!/bin/sh
# yeaboi installer — https://yeaboi.ai/install.sh
#
#   curl -LsSf https://yeaboi.ai/install.sh | sh
#
# Installs uv (if missing), then installs yeaboi as an isolated uv tool on a
# Python that uv fetches itself. The point is that the Python already on the
# machine is irrelevant: `pip` and `pipx` both use the interpreter they are run
# with and hard-fail when it is too old, which is the single most common reason
# a first-time user never sees the product.
#
# Overridable, all optional:
#   YEABOI_PYTHON         Python to build the tool env on   (default: >=3.10)
#   YEABOI_PACKAGE        what to install                   (default: yeaboi)
#   YEABOI_UV_ARGS        extra args for `uv tool install`  (default: none)
#   UV_INSTALLER_VERSION  uv release used when uv is absent
#
#   curl -LsSf https://yeaboi.ai/install.sh | YEABOI_PACKAGE='yeaboi[voice]' sh
#
# Constraints this file must keep, both of them load-bearing:
#
#   * It never reads stdin. Under `curl | sh` the script *is* stdin, so a single
#     `read` would swallow the rest of itself and execute a truncated program.
#   * POSIX sh only — no [[, no arrays, no `source`, no `function`. The documented
#     invocation pipes into `sh`, which on Debian and Ubuntu is dash, not bash.
#
# It writes only under $HOME and never calls sudo.

set -eu

# A version specifier, not a version. `--python 3.10` pins every user to the
# oldest supported runtime and downloads a ~30 MB interpreter onto machines that
# already have a perfectly good one; `--python '>=3.10'` reuses what is there and
# downloads only when nothing qualifies. Keep this byte-identical to
# `requires-python` in pyproject.toml — tests/unit/test_install_script.py asserts it.
YEABOI_PYTHON="${YEABOI_PYTHON:->=3.10}"
YEABOI_PACKAGE="${YEABOI_PACKAGE:-yeaboi}"
YEABOI_UV_ARGS="${YEABOI_UV_ARGS:-}"
UV_INSTALLER_VERSION="${UV_INSTALLER_VERSION:-0.11.2}"

say() { printf '%s\n' "$*"; }
die() { printf 'error: %s\n' "$*" >&2; exit 1; }

# ---------------------------------------------------------------------------
# Platform
# ---------------------------------------------------------------------------
# Native Windows is refused rather than half-served: src/yeaboi/ui/shared/_input.py
# imports termios and tty at module scope, so the TUI cannot start there at all.
# Installing successfully into a shell that can never run the program is a worse
# failure than refusing with a pointer.
case "$(uname -s 2>/dev/null || echo unknown)" in
    MINGW* | MSYS* | CYGWIN*)
        die "yeaboi's terminal UI needs a POSIX terminal and cannot run on native Windows.
       Install inside WSL (https://learn.microsoft.com/windows/wsl/install) and
       re-run this command from your WSL shell."
        ;;
esac

# ---------------------------------------------------------------------------
# uv
# ---------------------------------------------------------------------------
# uv gained PEP 440 specifier support for `--python` in 0.4.30 (measured, not
# assumed). An older uv already on PATH would reject `--python '>=3.10'` outright,
# so it is replaced rather than used — the whole premise of this script is that
# what the machine already has must not decide whether the install works.
uv_predates_specifiers() {
    _v=$(uv --version 2>/dev/null | awk '{print $2}')
    [ -n "$_v" ] || return 0
    _major=${_v%%.*}
    _rest=${_v#*.}
    _minor=${_rest%%.*}
    _patch=${_rest#*.}
    _patch=${_patch%%[!0-9]*}
    case "$_major$_minor$_patch" in *[!0-9]*) return 0 ;; esac
    [ "$_major" -gt 0 ] && return 1
    [ "$_minor" -gt 4 ] && return 1
    [ "$_minor" -lt 4 ] && return 0
    [ "${_patch:-0}" -lt 30 ] && return 0
    return 1
}

if ! command -v uv >/dev/null 2>&1 || uv_predates_specifiers; then
    if command -v uv >/dev/null 2>&1; then
        say "Your uv ($(uv --version 2>/dev/null)) predates \`--python\` version specifiers; installing a newer one..."
    else
        say "Installing uv (the Python package manager yeaboi installs through)..."
    fi
    uv_installer="https://astral.sh/uv/${UV_INSTALLER_VERSION}/install.sh"
    # Pinned rather than tracking latest, for the same reason this repo pins its
    # GitHub Actions by SHA. uv's own installer checksum-verifies the binary it
    # downloads, so that half is already covered.
    # Downloaded to a file and checked, never piped straight into `sh`. POSIX sh
    # has no `pipefail` and `set -e` takes the LAST command's status, so
    # `curl ... | sh` reports success when curl 404s and `sh` reads an empty
    # stream — and feeds `sh` a truncated script when a connection drops
    # mid-transfer. The user would then be told uv "is not on PATH", which sends
    # them to do something that cannot help.
    # `mktemp` under $HOME, not a PID-named path in /tmp. Fetch-then-execute is
    # this file's entire job, so the gap between the two is the thing to protect:
    # a predictable name in a world-writable directory lets a local attacker
    # pre-create it as a symlink (curl -o and wget -O both follow, writing the
    # installer through the link) or swap the file after the download and before
    # `sh` reads it. mktemp is O_EXCL, mode 0600 and unguessable. Under $HOME so
    # the header's "writes only under $HOME" is literally true.
    uv_script=$(mktemp "$HOME/.yeaboi-uv-installer.XXXXXX") || die "could not create a temporary file under $HOME"
    trap 'rm -f "$uv_script"' EXIT INT TERM
    if command -v curl >/dev/null 2>&1; then
        curl -LsSf "$uv_installer" -o "$uv_script" || die "could not download the uv installer from $uv_installer
       Check your network or a proxy, or install uv yourself:
       https://docs.astral.sh/uv/getting-started/installation/"
    elif command -v wget >/dev/null 2>&1; then
        wget -qO "$uv_script" "$uv_installer" || die "could not download the uv installer from $uv_installer
       Check your network or a proxy, or install uv yourself:
       https://docs.astral.sh/uv/getting-started/installation/"
    else
        die "neither curl nor wget is available — install one, or install uv yourself:
       https://docs.astral.sh/uv/getting-started/installation/"
    fi
    # A short file is a truncated transfer, not an installer.
    [ -s "$uv_script" ] || die "the uv installer downloaded empty from $uv_installer — try again."
    sh "$uv_script" || die "the uv installer failed. Install uv yourself and re-run:
       https://docs.astral.sh/uv/getting-started/installation/"
    rm -f "$uv_script"
    trap - EXIT INT TERM

    # Make uv usable in *this* run rather than telling the user to open a new
    # shell. The uv installer writes this env file for exactly this purpose.
    if [ -f "$HOME/.local/bin/env" ]; then
        # shellcheck disable=SC1091
        . "$HOME/.local/bin/env"
    fi
    command -v uv >/dev/null 2>&1 || die "uv installed but is not on PATH — open a new shell and re-run."
    # The predicate again, not just `command -v`. If the installer honoured a
    # custom UV_INSTALL_DIR it may not have written the env file above, leaving
    # the *old* uv still first on PATH — which passes `command -v` and then
    # rejects `--python '>=3.10'` with a message that explains nothing.
    if uv_predates_specifiers; then
        die "the uv on PATH ($(uv --version 2>/dev/null)) is older than 0.4.30 and cannot
       resolve a Python version specifier. A newer uv was installed but is not
       first on PATH — open a new shell and re-run, or upgrade with 'uv self update'."
    fi
fi

# ---------------------------------------------------------------------------
# yeaboi
# ---------------------------------------------------------------------------
# This is the line the whole script exists for, and it is the one setting that
# must not be left to a default: automatic downloads are uv's default, but a
# ~/.config/uv/uv.toml or a corporate image setting python-downloads = "never"
# collapses the install straight back onto system Python — the exact failure
# being fixed here.
export UV_PYTHON_DOWNLOADS=automatic

say "Installing ${YEABOI_PACKAGE} on Python ${YEABOI_PYTHON}..."
# YEABOI_UV_ARGS is deliberately unquoted so callers can pass more than one flag.
# shellcheck disable=SC2086
uv tool install --python "$YEABOI_PYTHON" $YEABOI_UV_ARGS "$YEABOI_PACKAGE"

# ---------------------------------------------------------------------------
# PATH and next steps
# ---------------------------------------------------------------------------
if ! command -v yeaboi >/dev/null 2>&1; then
    say ""
    say "yeaboi is installed but not yet on your PATH."
    uv tool update-shell >/dev/null 2>&1 || true
    say "Open a new terminal, or add this to your shell profile:"
    say ""
    say "    export PATH=\"\$HOME/.local/bin:\$PATH\""
fi

say ""
say "Done. Next:"
say ""
say "    yeaboi --setup      # add your API key"
say "    yeaboi              # launch the TUI"
say ""