#!/bin/sh
# tasksel Test- hook: pre-tick the task on NVIDIA DGX Spark class hardware
# (GB10; ASUS GX10 and Gigabyte AI TOP ATOM both report this family).
# Stanza keyword:  Test-dgx-spark: mark show
#   $1 = task name, $2 = action if hardware matches, $3 = action otherwise.
# Exit codes (tasksel contract): 0 install silently, 1 hide, 2 show marked,
# 3 show unmarked.
fam=$(cat "${DMI_ROOT:-/sys/class/dmi/id}/product_family" 2>/dev/null)
if [ "$fam" = "DGX Spark" ]; then var=$2; else var=$3; fi
case "$var" in
	install) exit 0 ;;
	skip)    exit 1 ;;
	mark)    exit 2 ;;
	show)    exit 3 ;;
	*)       exit 1 ;;
esac
