From 2923d4d3f4221213f2547008ef57cfa83293a586 Mon Sep 17 00:00:00 2001
From: Gregory Ballantine <gballantine@metaunix.net>
Date: Tue, 2 May 2023 21:34:02 -0600
Subject: [PATCH] Added sleep states for FreeBSD

---
 system/sleep.sls         | 38 +++++---------------------------------
 system/sleep/freebsd.sls | 16 ++++++++++++++++
 system/sleep/linux.sls   | 38 ++++++++++++++++++++++++++++++++++++++
 3 files changed, 59 insertions(+), 33 deletions(-)
 create mode 100644 system/sleep/freebsd.sls
 create mode 100644 system/sleep/linux.sls

diff --git a/system/sleep.sls b/system/sleep.sls
index a9d5af7..5eb3d51 100644
--- a/system/sleep.sls
+++ b/system/sleep.sls
@@ -1,38 +1,10 @@
 {% 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'
+{% if salt['grains']['get']('os_family') == 'FreeBSD' %}
+include:
+  - system.sleep.freebsd
 {% 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'
+include:
+  - system.sleep.linux
 {% endif %}
diff --git a/system/sleep/freebsd.sls b/system/sleep/freebsd.sls
new file mode 100644
index 0000000..6b383f9
--- /dev/null
+++ b/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 sysrc
+sleep_disable_sysrc:
+  sysrc:
+    - name: 'kern.suspend_blocked'
+    - value: '1'
+{% else %}
+# Enable sleep via sysrc
+sleep_enable_sysrc:
+  sysrc:
+    - name: 'kern.suspend_blocked'
+    - value: '0'
+{% endif %}
diff --git a/system/sleep/linux.sls b/system/sleep/linux.sls
new file mode 100644
index 0000000..a9d5af7
--- /dev/null
+++ b/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 %}