+
+disks = if node[:hardware][:disk]
+ node[:hardware][:disk][:disks]
+ else
+ []
+ end
+
+intel_ssds = disks.select { |d| d[:vendor] == "INTEL" && d[:model] =~ /^SSD/ }
+
+nvmes = if node[:hardware][:pci]
+ node[:hardware][:pci].values.select { |pci| pci[:driver] == "nvme" }
+ else
+ []
+ end
+
+intel_nvmes = nvmes.select { |pci| pci[:vendor_name] == "Intel Corporation" }
+
+if !intel_ssds.empty? || !intel_nvmes.empty?
+ package "unzip"
+
+ intel_ssd_tool_version = "3.0.26"
+
+ remote_file "#{Chef::Config[:file_cache_path]}/Intel_SSD_Data_Center_Tool_#{intel_ssd_tool_version}_Linux.zip" do
+ source "https://downloadmirror.intel.com/29720/eng/Intel_SSD_DCT_#{intel_ssd_tool_version}_Linux.zip"
+ end
+
+ execute "#{Chef::Config[:file_cache_path]}/Intel_SSD_Data_Center_Tool_#{intel_ssd_tool_version}_Linux.zip" do
+ command "unzip Intel_SSD_Data_Center_Tool_#{intel_ssd_tool_version}_Linux.zip isdct_#{intel_ssd_tool_version}.400-1_amd64.deb"
+ cwd Chef::Config[:file_cache_path]
+ user "root"
+ group "root"
+ not_if { ::File.exist?("#{Chef::Config[:file_cache_path]}/isdct_#{intel_ssd_tool_version}.400-1_amd64.deb") }
+ end
+
+ dpkg_package "isdct" do
+ version "#{intel_ssd_tool_version}-1"
+ source "#{Chef::Config[:file_cache_path]}/isdct_#{intel_ssd_tool_version}.400-1_amd64.deb"
+ end
+end
+
+disks = disks.map do |disk|
+ next if disk[:state] == "spun_down" || %w[unconfigured failed].any?(disk[:status])
+
+ if disk[:smart_device]
+ controller = node[:hardware][:disk][:controllers][disk[:controller]]
+
+ if controller && controller[:device]
+ device = controller[:device].sub("/dev/", "")
+ smart = disk[:smart_device]
+
+ if device.start_with?("cciss/") && smart =~ /^cciss,(\d+)$/
+ array = node[:hardware][:disk][:arrays][disk[:arrays].first]
+ munin = "cciss-3#{array[:wwn]}-#{Regexp.last_match(1)}"
+ elsif smart =~ /^.*,(\d+)$/
+ munin = "#{device}-#{Regexp.last_match(1)}"
+ elsif smart =~ %r{^.*,(\d+)/(\d+)$}
+ munin = "#{device}-#{Regexp.last_match(1)}:#{Regexp.last_match(2)}"
+ end
+ elsif disk[:device]
+ device = disk[:device].sub("/dev/", "")
+ smart = disk[:smart_device]
+
+ if smart =~ /^.*,(\d+),(\d+),(\d+)$/
+ munin = "#{device}-#{Regexp.last_match(1)}:#{Regexp.last_match(2)}:#{Regexp.last_match(3)}"
+ end
+ end
+ elsif disk[:device] =~ %r{^/dev/(nvme\d+)n\d+$}
+ device = Regexp.last_match(1)
+ munin = device
+ elsif disk[:device]
+ device = disk[:device].sub("/dev/", "")
+ munin = device
+ end
+
+ next if device.nil?
+
+ Hash[
+ :device => device,
+ :smart => smart,
+ :munin => munin,
+ :hddtemp => munin.tr("-:", "_")
+ ]
+end
+
+disks = disks.compact.uniq
+
+if disks.count.positive?
+ package "smartmontools"
+
+ template "/etc/cron.daily/update-smart-drivedb" do
+ source "update-smart-drivedb.erb"
+ owner "root"
+ group "root"
+ mode "755"
+ end
+
+ template "/usr/local/bin/smartd-mailer" do
+ source "smartd-mailer.erb"
+ owner "root"
+ group "root"
+ mode "755"
+ end
+
+ template "/etc/smartd.conf" do
+ source "smartd.conf.erb"
+ owner "root"
+ group "root"
+ mode "644"
+ variables :disks => disks
+ end
+
+ template "/etc/default/smartmontools" do
+ source "smartmontools.erb"
+ owner "root"
+ group "root"
+ mode "644"
+ end
+
+ service "smartmontools" do
+ action [:enable, :start]
+ subscribes :reload, "template[/etc/smartd.conf]"
+ subscribes :restart, "template[/etc/default/smartmontools]"
+ end
+
+ template "/etc/prometheus/collectors/smart.devices" do
+ source "smart.devices.erb"
+ owner "root"
+ group "root"
+ mode "644"
+ variables :disks => disks
+ end
+
+ prometheus_collector "smart" do
+ interval "15m"
+ end
+
+ # Don't try and do munin monitoring of disks behind
+ # an Areca controller as they only allow one thing to
+ # talk to the controller at a time and smartd will
+ # throw errors if it clashes with munin
+ disks = disks.reject { |disk| disk[:smart]&.start_with?("areca,") }
+
+ disks.each do |disk|
+ munin_plugin "smart_#{disk[:munin]}" do
+ target "smart_"
+ conf "munin.smart.erb"
+ conf_variables :disk => disk
+ end
+ end
+else
+ service "smartd" do
+ action [:stop, :disable]
+ end
+end
+
+if disks.count.positive?
+ munin_plugin "hddtemp_smartctl" do
+ conf "munin.hddtemp.erb"
+ conf_variables :disks => disks
+ end
+else
+ munin_plugin "hddtemp_smartctl" do
+ action :delete
+ conf "munin.hddtemp.erb"
+ end
+end
+
+plugins = Dir.glob("/etc/munin/plugins/smart_*").map { |p| File.basename(p) } -
+ disks.map { |d| "smart_#{d[:munin]}" }
+
+plugins.each do |plugin|
+ munin_plugin plugin do
+ action :delete
+ conf "munin.smart.erb"
+ end
+end
+
+if File.exist?("/etc/mdadm/mdadm.conf")
+ mdadm_conf = edit_file "/etc/mdadm/mdadm.conf" do |line|
+ line.gsub!(/^MAILADDR .*$/, "MAILADDR admins@openstreetmap.org")
+
+ line
+ end
+
+ file "/etc/mdadm/mdadm.conf" do
+ owner "root"
+ group "root"
+ mode "644"
+ content mdadm_conf
+ end
+
+ service "mdadm" do
+ action :nothing
+ subscribes :restart, "file[/etc/mdadm/mdadm.conf]"
+ end
+end
+
+template "/etc/modules" do
+ source "modules.erb"
+ owner "root"
+ group "root"
+ mode "644"
+end
+
+service "kmod" do
+ action :nothing
+ subscribes :start, "template[/etc/modules]"
+end
+
+if node[:hardware][:watchdog]
+ package "watchdog"
+
+ template "/etc/default/watchdog" do
+ source "watchdog.erb"
+ owner "root"
+ group "root"
+ mode "644"
+ variables :module => node[:hardware][:watchdog]
+ end
+
+ service "watchdog" do
+ action [:enable, :start]
+ end
+end
+
+unless Dir.glob("/sys/class/hwmon/hwmon*").empty?
+ package "lm-sensors"
+
+ Dir.glob("/sys/devices/platform/coretemp.*").each do |coretemp|
+ cpu = File.basename(coretemp).sub("coretemp.", "").to_i
+ chip = format("coretemp-isa-%04d", cpu)
+
+ temps = if File.exist?("#{coretemp}/name")
+ Dir.glob("#{coretemp}/temp*_input").map do |temp|
+ File.basename(temp).sub("temp", "").sub("_input", "").to_i
+ end.sort
+ else
+ Dir.glob("#{coretemp}/hwmon/hwmon*/temp*_input").map do |temp|
+ File.basename(temp).sub("temp", "").sub("_input", "").to_i
+ end.sort
+ end
+
+ if temps.first == 1
+ node.default[:hardware][:sensors][chip][:temps][:temp1][:label] = "CPU #{cpu}"
+ temps.shift
+ end
+
+ temps.each_with_index do |temp, index|
+ node.default[:hardware][:sensors][chip][:temps]["temp#{temp}"][:label] = "CPU #{cpu} Core #{index}"
+ end
+ end
+
+ execute "/etc/sensors.d/chef.conf" do
+ action :nothing
+ command "/usr/bin/sensors -s"
+ user "root"
+ group "root"
+ end
+
+ template "/etc/sensors.d/chef.conf" do
+ source "sensors.conf.erb"
+ owner "root"
+ group "root"
+ mode "644"
+ notifies :run, "execute[/etc/sensors.d/chef.conf]"
+ end
+end
+
+if node[:hardware][:shm_size]
+ execute "remount-dev-shm" do
+ action :nothing
+ command "/bin/mount -o remount /dev/shm"
+ user "root"
+ group "root"
+ end
+
+ mount "/dev/shm" do
+ action :enable
+ device "tmpfs"
+ fstype "tmpfs"
+ options "rw,nosuid,nodev,size=#{node[:hardware][:shm_size]}"
+ notifies :run, "execute[remount-dev-shm]"
+ end
+end