#!/usr/bin/env bash
# onw - Ore no NPU ga konna ni ugoku wake nai: install the resident top-bar app (Ubuntu).
# Installs the system pieces the indicator and the window need (AppIndicator, WebKit; asks for sudo once), creates .venv,
# installs onw + the tray, adds "onw" to the application list and starts it in the top bar.
# GNOME shows indicators through the "Ubuntu AppIndicators" extension, which Ubuntu enables by default.
set -e
cd "$(dirname "$0")"
export PYTHONUTF8=1

if ! python3 -c "import sys; sys.exit(0 if sys.version_info >= (3, 10) else 1)" 2>/dev/null; then
    echo "Python 3.10 or newer is needed (Ubuntu 22.04 and later ship it):  sudo apt install python3"
    exit 1
fi

need=""
# the window is a WebKit web view (pywebview): 4.1 on Ubuntu 24.04 and later, 4.0 before
webkit=gir1.2-webkit2-4.1
apt-cache show $webkit >/dev/null 2>&1 || webkit=gir1.2-webkit2-4.0
# wl-clipboard / xclip: "Copy the OpenAI API URL" in the top-bar menu (Wayland / X11)
for p in python3-venv python3-gi gir1.2-ayatanaappindicator3-0.1 $webkit wl-clipboard xclip; do
    dpkg -s "$p" >/dev/null 2>&1 || need="$need $p"
done
if [ -n "$need" ]; then
    echo "[setup] installing system packages:$need"
    sudo apt-get install -y $need
fi

install_onw() {
    # the bundled pip is fine (a newer pip once failed to install packages that have scripts, so it is not upgraded)
    .venv/bin/python -m pip install -e ".[tray]" \
        --extra-index-url https://storage.openvinotoolkit.org/simple/wheels/pre-release
}
if [ ! -f .venv/installed-tray.txt ] || ! .venv/bin/python -c "import webview" >/dev/null 2>&1; then
    if [ ! -f .venv/bin/python ]; then
        echo "[setup] creating Python environment in .venv ..."
        # system site packages: the GObject bindings for the indicator and WebKit come from apt (python3-gi)
        python3 -m venv --system-site-packages .venv
    fi
    if ! install_onw; then
        echo "[setup] installation failed; rebuilding the Python environment once ..."
        rm -rf .venv
        python3 -m venv --system-site-packages .venv
        install_onw || exit 1
    fi
    echo ok > .venv/installed.txt
    echo ok > .venv/installed-tray.txt
fi

mkdir -p ~/.local/share/applications
cat > ~/.local/share/applications/onw.desktop <<EOF
[Desktop Entry]
Type=Application
Name=onw
Comment=LLMs on the Intel NPU (俺のNPUがこんなに動くわけない)
Exec="$PWD/.venv/bin/onw" tray
Path=$PWD
Icon=$PWD/onw/web/icon.png
Terminal=false
Categories=Utility;
EOF
echo "[setup] application list: onw"

echo "[start] onw is starting in the top bar (first model start: download + NPU compile) ..."
nohup .venv/bin/onw tray >/dev/null 2>&1 &