Fixed the state module called in FreeBSD sleep states

system/sleep.sls

@@ -1,20 +1,10 @@
 {% from "system/map.jinja" import system_settings with context %}
 {% set settings = system_settings['sleep'] %}
 
-{% if settings.disable %}
-disable_sleep_target:
-  service.masked:
-    - name: 'sleep.target'
-
-disable_suspend_target:
-  service.masked:
-    - name: 'suspend.target'
-
-disable_hibernate_target:
-  service.masked:
-    - name: 'hibernate.target'
-
-disable_hybrid-sleep_target:
-  service.masked:
-    - name: 'hybrid-sleep.target'
+{% if salt['grains']['get']('os_family') == 'FreeBSD' %}
+include:
+  - system.sleep.freebsd
+{% else %}
+include:
+  - system.sleep.linux
 {% endif %}

system/sleep/freebsd.sls

@@ -0,0 +1,16 @@
+{% from "system/map.jinja" import system_settings with context %}
+{% set settings = system_settings['sleep'] %}
+
+{% if settings.disable %}
+# Disable sleep via sysctl
+sleep_disable_sysctl:
+  sysctl.present:
+    - name: 'kern.suspend_blocked'
+    - value: '1'
+{% else %}
+# Enable sleep via sysctl
+sleep_enable_sysctl:
+  sysctl.present:
+    - name: 'kern.suspend_blocked'
+    - value: '0'
+{% endif %}

system/sleep/linux.sls

@@ -0,0 +1,38 @@
+{% from "system/map.jinja" import system_settings with context %}
+{% set settings = system_settings['sleep'] %}
+
+{% if settings.disable %}
+# Disable common sleep states
+disable_sleep_target:
+  service.masked:
+    - name: 'sleep.target'
+
+disable_suspend_target:
+  service.masked:
+    - name: 'suspend.target'
+
+disable_hibernate_target:
+  service.masked:
+    - name: 'hibernate.target'
+
+disable_hybrid-sleep_target:
+  service.masked:
+    - name: 'hybrid-sleep.target'
+{% else %}
+# Enable sleep states
+enable_sleep_target:
+  service.unmasked:
+    - name: 'sleep.target'
+
+enable_suspend_target:
+  service.unmasked:
+    - name: 'suspend.target'
+
+enable_hibernate_target:
+  service.unmasked:
+    - name: 'hibernate.target'
+
+enable_hybrid-sleep_target:
+  service.unmasked:
+    - name: 'hybrid-sleep.target'
+{% endif %}