#!/usr/bin/env bash

updates=()

if [[ -f /etc/debian_version ]]; then
    APT_UPDATES=$(sudo apt update 2>/dev/null | grep packages | cut -d '.' -f 1 | awk '{print $1}')
    if [ "$APT_UPDATES" = "All" ]; then
        NUM_UPDATES=0
    else
        NUM_UPDATES="$APT_UPDATES"
    fi
    updates+=("[ APT: ${NUM_UPDATES} ]")
fi

if [[ -f /etc/redhat-release ]]; then
    updates+=("[ DNF: $(sudo dnf check-update | wc -l) ]")
fi

if command -v flatpak >/dev/null; then
    updates+=("[ Flatpak: $(flatpak remote-ls --updates | wc -l) ]")
fi

echo "${updates[*]}"