#!/usr/bin/env bash
##makedesc: Install ollama

# https://github.com/ollama/ollama

set -o pipefail

mkdir -p "$HOME/install"

install() {
    echo
    echo "==============================================="
    echo "Installing ollama (latest)"
    echo "==============================================="
    echo
    curl -fsSL https://ollama.com/install.sh | sh
    echo
    echo "Finish!"
}

upgrade() {
    echo
    echo "==============================================="
    echo "Upgrading ollama"
    echo "==============================================="
    echo
    curl -fsSL https://ollama.com/install.sh | sh
    echo
    echo "Finish!"
}

remove() {
    # https://docs.ollama.com/linux#uninstall
    echo
    echo "==============================================="
    echo "Removing ollama"
    echo "==============================================="
    echo
    sudo systemctl stop ollama
    sudo systemctl disable ollama
    sudo rm /etc/systemd/system/ollama.service
    sudo rm -r "$(which ollama | tr 'bin' 'lib')"
    sudo rm "$(which ollama)"
    sudo userdel ollama
    sudo groupdel ollama
    sudo rm -rf \
        /usr/bin/ollama \
        /usr/share/ollama \
        /usr/local/lib/ollama \
        /usr/local/bin/ollama
    echo
    echo "Finish!"
    echo
}

case "$1" in
    u|upgrade) upgrade ;;
    r|remove|d|delete|p|purge) remove ;;
    *) install ;;
esac