#!/usr/bin/env bash
# Waybar GPU usage module - NVIDIA
set -euo pipefail

GPU=$(nvidia-smi --query-gpu=utilization.gpu --format=csv,noheader,nounits 2>/dev/null | head -1 | tr -d ' ')
TEMP=$(nvidia-smi --query-gpu=temperature.gpu --format=csv,noheader,nounits 2>/dev/null | head -1 | tr -d ' ')

if [ -z "$GPU" ]; then
    echo '{"text": "GPU N/A", "percentage": 0}'
    exit 0
fi

if [ "$GPU" -ge 80 ]; then
    ICON=""
elif [ "$GPU" -ge 50 ]; then
    ICON=""
else
    ICON="󰢮"
fi

echo "{\"text\": \"${ICON} ${GPU}%\", \"tooltip\": \"GPU Usage: ${GPU}% | Temp: ${TEMP}°C\", \"percentage\": ${GPU}}"