#!/bin/sh
# Based off Bidule0hm's script in https://forums.freenas.org/index.php?threads/scripts-to-report-smart-zpool-and-ups-status-hdd-cpu-t%C2%B0-hdd-identification-and-backup-the-config.27365/
singlepartdrives="ada0 ada1 ada2 ada3 ada4 ada5 ada6 ada7 da1 da2 da3 da4 da5 "
dualpartdrives="da0 da6 da7 da8 da9 da10 da11 da12 da13"
# Due to naming conventions in BSD for NVMe drives, the nvd convention is used for drives, and nvme is used for the controller.
# To use in this script, just use the number associated to get the parent drive, and add p1, p2 etc to get the different partitions. 
# By default, the nvme script uses partition 2 to show the GPTID as this is standard if the entire drive is utilized. 
nvmedrives="0 0p1 0p2 1 1p1 1p2"

echo ""
echo "+=========+============================================+======================+===========================+==========+=====+======+===========+"
echo "| Device  | GPTID                                      | Serial               | Model                     | Capacity | Bay | Tray | Purchased |"
echo "+=========+============================================+======================+===========================+==========+=====+======+===========+"
for drive in $singlepartdrives
do
	gptid=""
	serial=""
	model=""
	capacity=""
    gptid=`glabel status -s "${drive}p1" | awk '{print $1}'`
    serial=`smartctl -i /dev/${drive} | grep "Serial Number" | awk '{print $3}'`
    model=`smartctl -i /dev/${drive} | grep "Device Model" | awk '{print $3, $4}'`
    capacity=`smartctl -i /dev/${drive} | grep "Capacity" | awk '{print $5 $6}' | tr -d "[]"`
    printf "| %-7s | %-42s | %-20s | %-25s | %-8s | %-3s | %-4s | %-9s |\n" "$drive" "$gptid" "$serial" "$model" "$capacity"
    echo "+---------+--------------------------------------------+----------------------+---------------------------+----------+-----+------+-----------+"
done
for drive in $dualpartdrives
do
	gptid=""
	serial=""
	model=""
	capacity=""
    gptid=`glabel status -s "${drive}p2" | awk '{print $1}'`
    serial=`smartctl -i /dev/${drive} | grep "Serial Number" | awk '{print $3}'`
    model=`smartctl -i /dev/${drive} | grep "Device Model" | awk '{print $3, $4}'`
    capacity=`smartctl -i /dev/${drive} | grep "Capacity" | awk '{print $5 $6}' | tr -d "[]"`
    printf "| %-7s | %-42s | %-20s | %-25s | %-8s | %-3s | %-4s | %-9s |\n" "$drive" "$gptid" "$serial" "$model" "$capacity"
    echo "+---------+--------------------------------------------+----------------------+---------------------------+----------+-----+------+-----------+"
done
for drive in $nvmedrives
do
	gptid=""
	serial=""
	model=""
	capacity=""
    if [ ${#drive} -gt 1 ]; then
		gptid=`glabel status -s "nvd${drive}" | awk '{print $1}'`
	fi
    serial=`smartctl -i /dev/nvme${drive} | grep "Serial Number" | awk '{print $3}'`
    model=`smartctl -i /dev/nvme${drive} | grep "Model Number" | awk '{print $3, $4}'`
    capacity=`smartctl -i /dev/nvme${drive} | grep "Capacity" | awk '{print $5 $6}' | tr -d "[]"`
    printf "| %-7s | %-42s | %-20s | %-25s | %-8s | %-3s | %-4s | %-9s |\n" "nvd$drive" "$gptid" "$serial" "$model" "$capacity"
    echo "+---------+--------------------------------------------+----------------------+---------------------------+----------+-----+------+-----------+"
done
echo ""