#!/bin/bash

# Install or update Ollama on Debian/Ubuntu systems.
set -euo pipefail

# Allow running directly as root (no sudo needed) or as a normal user.
if [[ "${EUID}" -eq 0 ]]; then
  sudo() { while [[ "$1" == -* ]]; do shift; done; "$@"; }
  export -f sudo
fi

echo "==> Installing base packages for install script"
sudo apt-get update
sudo apt-get install -y --no-install-recommends curl ca-certificates

if command -v ollama >/dev/null 2>&1; then
  echo "==> Updating existing Ollama installation"
else
  echo "==> Installing Ollama"
fi

curl -fsSL https://ollama.com/install.sh | sh

ollama --version