Make script more robus

Fix few bugs and make it more generic

PKGBUILD

@@ -2,36 +2,25 @@
 
 # Maintainer: Daniele Fucini <dfucini [at] gmail [dot] com>
 
-pkgname=update-notify-git
+pkgname=update-notify
-_pkgname=update-notify
+pkgver=2.0
-pkgdesc='Notify when updates are available'
-pkgver=0.0.1
 pkgrel=1
-url="https://github.com/Fuxino/${_pkgname}"
+pkgdesc='Notify when updates are available'
 arch=('any')
+url="https://git.shouldnt.work/fuxino/${pkgname}"
 license=('GPL-3.0-or-later')
-makedepends=('git')
 depends=('pacman-contrib'
          'aurutils')
-source=(git+${url}.git?signed#branch=master)
+makedepends=('git')
+optdepends=('git-monitor: to check git repos for updates')
+source=(git+${url}?signed#tag=${pkgver})
 validpgpkeys=('7E12BC1FF3B6EDB2CD8053EB981A2B2A3BBF5514')
-sha256sums=('SKIP')
+sha256sums=('e96dbcc2b77750be03d89cdf3e1351c20c9079b7de1e129da0188dd3b30f6a03')
-
-pkgver() 
-{  
-    cd ${_pkgname}
-    git describe --long --tags | sed 's/\([^-]*-g\)/r\1/;s/-/./g'
-}
-
-prepare()
-{
-    git -C ${srcdir}/${_pkgname} clean -dfx
-}
 
 package()
 {
-    cd ${srcdir}/${_pkgname}
+    cd ${srcdir}/${pkgname}
-    install -Dm755 ${_pkgname} ${pkgdir}/usr/bin/${_pkgname}
+    install -Dm755 ${pkgname} ${pkgdir}/usr/bin/${pkgname}
-    install -Dm644 ${_pkgname}.service ${pkgdir}/usr/lib/systemd/user/${_pkgname}.service
+    install -Dm644 ${pkgname}.service ${pkgdir}/usr/lib/systemd/user/${pkgname}.service
-    install -Dm644 ${_pkgname}.timer ${pkgdir}/usr/lib/systemd/user/${_pkgname}.timer
+    install -Dm644 ${pkgname}.timer ${pkgdir}/usr/lib/systemd/user/${pkgname}.timer
 }

update-notify

@@ -1,21 +1,117 @@
 #!/bin/bash
+set -u
 
-if [ "$#" -eq 1 ]; then
+# ---------------------------------------------------------------------------
-    threshold=$1
+# Config
+# ---------------------------------------------------------------------------
+threshold=${1:-30}
+AUR_REPO="aur-local"
+# stock kernel package -> your custom build of it. Add or remove entries freely.
+declare -A KERNELS=(
+    [linux]="linux-custom"
+    [linux-lts]="linux-lts-custom"
+    [linux-zen]="linux-zen-custom"
+)
+state_dir="${XDG_STATE_HOME:-$HOME/.local/state}/check-updates"
+
+[[ $threshold =~ ^[0-9]+$ ]] || { echo "Threshold must be a number" >&2; exit 1; }
+mkdir -p "$state_dir"
+
+# notify-send from cron/systemd needs the session bus address
+export DBUS_SESSION_BUS_ADDRESS="${DBUS_SESSION_BUS_ADDRESS:-unix:path=/run/user/$(id -u)/bus}"
+
+# ---------------------------------------------------------------------------
+# Helpers
+# ---------------------------------------------------------------------------
+
+# Send a notification only if its body changed since last time for this key.
+notify_once() {
+    local key=$1 title=$2 body=$3
+    local file="$state_dir/$key"
+    if [[ -f $file && $(<"$file") == "$body" ]]; then
+        return 0
+    fi
+    printf '%s' "$body" > "$file"
+    notify-send "$title" "$body"
+}
+
+# Forget a key so the next occurrence notifies again.
+clear_state() {
+    rm -f "$state_dir/$1"
+}
+
+# If $1 has a pending update, print its new version and return 0.
+# checkupdates lines look like: "pkgname oldver -> newver"
+pending_version() {
+    local pkg=$1 item new
+    for item in "${arch_list[@]}"; do
+        if [[ $item == "$pkg "* ]]; then
+            read -r _ _ _ new <<< "$item"
+            printf '%s\n' "$new"
+            return 0
+        fi
+    done
+    return 1
+}
+
+# ---------------------------------------------------------------------------
+# Gather updates
+# ---------------------------------------------------------------------------
+
+# Fetch checkupdates once. Exit codes: 0 = updates, 2 = none, 1 = error.
+checkupdates_out=$(checkupdates 2>/dev/null)
+checkupdates_rc=$?
+if (( checkupdates_rc == 1 )); then
+    echo "checkupdates failed (network or sync problem?)" >&2
+fi
+
+arch_list=()
+if [[ -n $checkupdates_out ]]; then
+    mapfile -t arch_list <<< "$checkupdates_out"
+fi
+arch_updates=${#arch_list[@]}
+
+aur_updates=0
+if command -v aur >/dev/null 2>&1; then
+    aur_updates=$(aur repo -u -d "$AUR_REPO" 2>/dev/null | wc -l)
+fi
+updates=$((arch_updates + aur_updates))
+
+# ---------------------------------------------------------------------------
+# Kernel check: the stock kernel updated, but your custom build didn't
+# ---------------------------------------------------------------------------
+for stock_pkg in "${!KERNELS[@]}"; do
+    custom_pkg=${KERNELS[$stock_pkg]}
+
+    # Only relevant if the custom kernel is installed and has no pending update itself
+    if pacman -Q "$custom_pkg" >/dev/null 2>&1 && ! pending_version "$custom_pkg" >/dev/null; then
+        if stock_ver=$(pending_version "$stock_pkg"); then
+            notify_once "kernel-$stock_pkg" "Kernel update available: $stock_pkg" \
+                "Rebuild $custom_pkg ($stock_pkg $stock_ver)"
+            continue
+        fi
+    fi
+    clear_state "kernel-$stock_pkg"
+done
+
+# ---------------------------------------------------------------------------
+# General updates threshold
+# ---------------------------------------------------------------------------
+if (( updates >= threshold )); then
+    notify_once "updates" "Updates available" "Arch: $arch_updates, AUR: $aur_updates"
+
+    hints=(
+        "git-monitor|Check git packages|Use git-monitor for git packages updates"
+        "fwupdmgr|Check firmware|Use fwupdmgr to check for firmware updates"
+    )
+    for hint in "${hints[@]}"; do
+        IFS='|' read -r cmd title body <<< "$hint"
+        if command -v "$cmd" >/dev/null 2>&1; then
+            notify_once "hint-$cmd" "$title" "$body"
+        fi
+    done
 else
-    threshold=30
+    clear_state "updates"
-fi
+    clear_state "hint-git-monitor"
-
+    clear_state "hint-fwupdmgr"
-arch_updates=$(checkupdates | wc -l)
-aur_updates=$(aur repo -u -d aur-local | wc -l)
-kernel_update=$(checkupdates | grep linux | grep -v custom | grep -v archlinux | grep -v api | grep -v firmware | wc -l)
-
-updates=$(($arch_updates + $aur_updates))
-
-if [ "$kernel_update" -ge 1 ]; then
-    notify-send "Kernel update available" "Rebuild custom kernel to update"
-fi
-
-if [ "$updates" -ge "$threshold" ]; then
-    notify-send "Updates available" "Arch: $arch_updates, AUR: $aur_updates"
 fi

update-notify.timer

@@ -3,7 +3,7 @@ Description=Run update-notify every 6 hours
 
 [Timer]
 OnBootSec=5min
-OnUnitActiveSec=6h
+OnUnitActiveSec=12h
 Persistent=true
 
 [Install]