feat: add reboot and shutdown

pkg/system/dbus.go

@@ -0,0 +1,23 @@
+package system
+
+import (
+	"log"
+
+	"github.com/godbus/dbus/v5"
+)
+
+// WithDbus executes the given function with a D-Bus system connection
+// and handles any connection errors
+func WithDbus(fn func(*dbus.Conn) error) error {
+	conn, err := dbus.SystemBus()
+	if err != nil {
+		log.Printf("Failed to connect to system bus: %v", err)
+		return err
+	}
+	if err := fn(conn); err != nil {
+		log.Print(err)
+		return err
+	}
+	conn.Close()
+	return nil
+}

pkg/system/state.go

@@ -0,0 +1,33 @@
+package system
+
+import (
+	"log"
+
+	"github.com/godbus/dbus/v5"
+)
+
+// Restart restarts the system.
+func Restart() error {
+	return WithDbus(func(conn *dbus.Conn) error {
+		obj := conn.Object("org.freedesktop.systemd1", "/org/freedesktop/systemd1")
+		log.Println("Rebooting system...")
+		call := obj.Call("org.freedesktop.systemd1.Manager.Reboot", 0)
+		if call.Err != nil {
+			log.Fatal(call.Err)
+		}
+		return nil
+	})
+}
+
+// Shutdown shuts down the system.
+func Shutdown() error {
+	return WithDbus(func(conn *dbus.Conn) error {
+		obj := conn.Object("org.freedesktop.systemd1", "/org/freedesktop/systemd1")
+		log.Println("Shutting down system...")
+		call := obj.Call("org.freedesktop.systemd1.Manager.PowerOff", 0)
+		if call.Err != nil {
+			log.Fatal(call.Err)
+		}
+		return nil
+	})
+}