Example notification daemon for rebootmgr notify strategy

rebootmgr_notify.go

package main

import (
	"io"
	"log"
	"net"
	"os"
	"reflect"
	"strconv"

	"github.com/godbus/dbus/v5"
)

var sockAddr string = "/run/user/" +
	strconv.Itoa(os.Geteuid()) +
	"/rebootnotification.socket"

func notifySend() {
	conn, err := dbus.ConnectSessionBus()
	if err != nil {
		panic(err)
	}
	defer conn.Close()

	obj := conn.Object(
		"org.freedesktop.Notifications",
		"/org/freedesktop/Notifications",
	)
	call := obj.Call(
		"org.freedesktop.Notifications.Notify",
		0,
		"",
		uint32(0),
		"",
		"Test",
		"This is a test of the DBus bindings for go.",
		[]string{},
		map[string]dbus.Variant{},
		int32(5000),
	)
	if call.Err != nil {
		panic(call.Err)
	}
}

func handleMessage(c net.Conn) {
	log.Printf("Client connected [%s]", c.RemoteAddr().Network())

	inputBuffer := make([]byte, 1024)
	data, err := c.Read(inputBuffer)
	if err != nil {
		panic("Receiving error")
	}

	if reflect.DeepEqual(inputBuffer[:data-2], []byte("Reboot")) {
		notifySend()
	}
	io.Copy(c, c)
	c.Close()
}

func main() {
	if err := os.RemoveAll(sockAddr); err != nil {
		log.Fatal(err)
	}

	l, err := net.Listen("unix", sockAddr)
	if err != nil {
		log.Fatal("listen error:", err)
	}
	defer l.Close()

	for {
		// Accept new connections, dispatching them to echoServer
		// in a goroutine.
		conn, err := l.Accept()
		if err != nil {
			log.Fatal("accept error:", err)
		}

		go handleMessage(conn)
	}
}