From 94379347dfd0c83e3c0355d9deeaba419d50edb3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sa=C5=A1a=20=C4=8Cavara?= <sasa.cavara@reversinglabs.com>
Date: Sun, 20 Nov 2022 11:43:35 +0100
Subject: [PATCH] add systemctl device type configuration option

---
 scrutiny_fa/config.json                       |  2 ++
 .../etc/cont-init.d/01-configuration.sh       | 26 +++++++++++++++++++
 2 files changed, 28 insertions(+)

diff --git a/scrutiny_fa/config.json b/scrutiny_fa/config.json
index 78f445339..13b322ca7 100644
--- a/scrutiny_fa/config.json
+++ b/scrutiny_fa/config.json
@@ -39,6 +39,8 @@
     "Mode": "list(Collector+WebUI|Collector)?",
     "TZ": "str?",
     "Updates": "list(Hourly|Daily|Weekly)"
+    "SMARTCTL_COMMAND_DEVICE_TYPE": "list(auto|ata|scsi|sat|usbcypress|usbjmicron|usbsunplus|marvell|megaraid)?",
+    "SMARTCTL_MEGARAID_DISK_NUM":"int?"
   },
   "slug": "scrutiny_fa",
   "udev": true,
diff --git a/scrutiny_fa/rootfs/etc/cont-init.d/01-configuration.sh b/scrutiny_fa/rootfs/etc/cont-init.d/01-configuration.sh
index e4a1a685b..64de19ca8 100644
--- a/scrutiny_fa/rootfs/etc/cont-init.d/01-configuration.sh
+++ b/scrutiny_fa/rootfs/etc/cont-init.d/01-configuration.sh
@@ -62,3 +62,29 @@ case "$FREQUENCY" in
         sed -i "1a export COLLECTOR_CRON_SCHEDULE=\"0 0 * * 0\"" /etc/cont-init.d/50-cron-config
         ;;
 esac
+############################
+# SMARTCTL COMMAND OPTIONS #
+############################
+
+# Alignt with smartctl commands options
+if bashio::config.has_value "SMARTCTL_COMMAND_DEVICE_TYPE"; then
+    device_type="$(bashio::config 'SMARTCTL_COMMAND_DEVICE_TYPE')"
+    if ! bashio::config.has_value "SMARTCTL_MEGARAID_DISK_NUM"; then
+        megaraid_disk_num="$(bashio::config 'SMARTCTL_MEGARAID_DISK_NUM')"
+        {
+        echo "commands:"
+        echo "  metrics_smartctl_bin: '/usr/sbin/smartctl'"
+        echo "  metrics_scan_args: '--scan --json --dev ${device_type}'"
+        echo "  metrics_info_args: '--info --json --dev ${device_type}'"
+        echo "  metrics_smart_args: '--xall --json --dev ${device_type}'"
+        } > /opt/scrutiny/config/collector.yaml
+    else
+        {
+        echo "commands:"
+        echo "  metrics_smartctl_bin: '/usr/sbin/smartctl'"
+        echo "  metrics_scan_args: '--scan --json --dev ${device_type},${megaraid_disk_num}'"
+        echo "  metrics_info_args: '--info --json --dev ${device_type},${megaraid_disk_num}'"
+        echo "  metrics_smart_args: '--xall --json --dev ${device_type},${megaraid_disk_num}'"
+        } > /opt/scrutiny/config/collector.yaml
+    fi
+fi