From 7f443cfdd46096543a5b94245cd6ee04d400de3d Mon Sep 17 00:00:00 2001
From: mwiegand <mwiegand@seibert-media.net>
Date: Sat, 30 Oct 2021 20:13:24 +0200
Subject: [PATCH] smartctl

---
 bundles/grafana/items.py                      |  1 +
 .../files/{hdd_temp => telegraf_plugin}       | 16 +++++++++++++--
 bundles/smartctl/items.py                     |  3 ++-
 bundles/smartctl/metadata.py                  |  4 ++--
 data/grafana/flux.mako                        |  3 +++
 data/grafana/rows/smartctl.py                 | 20 ++++++++++++++++++-
 nodes/home.backups.py                         |  6 +++---
 7 files changed, 44 insertions(+), 9 deletions(-)
 rename bundles/smartctl/files/{hdd_temp => telegraf_plugin} (55%)

diff --git a/bundles/grafana/items.py b/bundles/grafana/items.py
index c4e765a..6e80882 100644
--- a/bundles/grafana/items.py
+++ b/bundles/grafana/items.py
@@ -136,6 +136,7 @@ for dashboard_id, monitored_node in enumerate(monitored_nodes, start=1):
                         bucket=bucket,
                         host=monitored_node.name,
                         negative=query_config.get('negative', False),
+                        boolean_to_int=query_config.get('boolean_to_int', False),
                         resolution=query_config.get('resolution', 1) * 4,
                         filters={
                             'host': monitored_node.name,
diff --git a/bundles/smartctl/files/hdd_temp b/bundles/smartctl/files/telegraf_plugin
similarity index 55%
rename from bundles/smartctl/files/hdd_temp
rename to bundles/smartctl/files/telegraf_plugin
index 01a1cd2..8099131 100644
--- a/bundles/smartctl/files/hdd_temp
+++ b/bundles/smartctl/files/telegraf_plugin
@@ -2,11 +2,23 @@
 
 for device in $(smartctl --scan | cut -d' ' -f1)
 do
-  temp=$(smartctl -n idle -A --json=c $device | jq .temperature.current)
+  # POWER STATUS
 
+  if $(smartctl -n idle -n sleep -i $device | grep "Power mode" | grep -q "STANDBY")
+  then
+    active=False
+  else
+    active=True
+  fi
+  echo "smartctl,host=${node.name},device=$device active=$active $(date --utc +%s%N)"
+
+  # TEMPS
+
+  temp=$(smartctl -n idle -A --json=c $device | jq .temperature.current)
+  
   if [[ $temp == ?(-)+([0-9]) ]]
   then
-    echo "smartctl,host=${node.name},device=$device temperature=$temp $(date --utc +%s%N)"
+    echo "smartctl,host=${node.name},device=$device temperature=<%text>${temp}</%text>i $(date --utc +%s%N)"
   elif [[ $temp == null ]]
   then
     # hdd might be sleeping
diff --git a/bundles/smartctl/items.py b/bundles/smartctl/items.py
index 5922f7d..5c070d9 100644
--- a/bundles/smartctl/items.py
+++ b/bundles/smartctl/items.py
@@ -1,5 +1,6 @@
 files = {
-    '/usr/local/share/icinga/plugins/hdd_temp': {
+    '/usr/local/share/icinga/plugins/smartctl': {
+        'source': 'telegraf_plugin',
         'content_type': 'mako',
         'mode': '0755',
     },
diff --git a/bundles/smartctl/metadata.py b/bundles/smartctl/metadata.py
index 3079b77..6d7794c 100644
--- a/bundles/smartctl/metadata.py
+++ b/bundles/smartctl/metadata.py
@@ -13,7 +13,7 @@ defaults = {
             'inputs': {
                 'exec': [{
                     'commands': [
-                        f'sudo /usr/local/share/icinga/plugins/hdd_temp',
+                        f'sudo /usr/local/share/icinga/plugins/smartctl',
                     ],
                     'data_format': 'influx',
                     'interval': '60s',
@@ -22,6 +22,6 @@ defaults = {
         },
     },
     'sudoers': {
-        'telegraf': ['/usr/local/share/icinga/plugins/hdd_temp'],
+        'telegraf': ['/usr/local/share/icinga/plugins/smartctl'],
     },
 }
diff --git a/data/grafana/flux.mako b/data/grafana/flux.mako
index 42de3f6..c95ee18 100644
--- a/data/grafana/flux.mako
+++ b/data/grafana/flux.mako
@@ -6,6 +6,9 @@ from(bucket: "${bucket}")
 % endfor
 % if function == 'derivative':
   |> derivative(nonNegative: true)
+% endif
+% if boolean_to_int:
+  |> map(fn: (r) => ({r with _value: if r._value == true then 1 else 0 }))
 % endif
   |> aggregateWindow(every: duration(v: int(v: v.windowPeriod)*${resolution}), fn: mean, createEmpty: false)
 % if negative:
diff --git a/data/grafana/rows/smartctl.py b/data/grafana/rows/smartctl.py
index 1fc671e..b60bf11 100644
--- a/data/grafana/rows/smartctl.py
+++ b/data/grafana/rows/smartctl.py
@@ -13,6 +13,24 @@
         },
         'display_name': '__field.labels.device',
         'min': 0,
-        'unit': 'degrees',
+        'unit': 'celsius',
+    },
+    'active': {
+        'stacked': True,
+        'queries': {
+            'usage': {
+                'filters': {
+                    '_measurement': 'smartctl',
+                    '_field': [
+                        'active',
+                    ],
+                },
+                'function': 'last',
+                'boolean_to_int': True,
+            },
+        },
+        'display_name': '__field.labels.device',
+        'min': 0,
+        'unit': 'active',
     },
 }
diff --git a/nodes/home.backups.py b/nodes/home.backups.py
index d0b0087..a99315e 100644
--- a/nodes/home.backups.py
+++ b/nodes/home.backups.py
@@ -24,13 +24,13 @@
         },
         'smartctl': {
             '/dev/disk/by-id/ata-HGST_HDN726040ALE614_K3GV6TPL': {
-                'apm': 32,
+                'apm': 1,
             },
             '/dev/disk/by-id/ata-HGST_HDN726040ALE614_K4KAJXEB': {
-                'apm': 32,
+                'apm': 1,
             },
             '/dev/disk/by-id/ata-TOSHIBA_HDWQ140_19VZK0EMFAYG': {
-                'apm': 32,
+                'apm': 1,
             },
         },
         'zfs-mirror': {