fix(yggctl): prefer system bus when run as root

yggctl will attempt to connect to the session bus only if
DBUS_SESSION_BUS_ADDRESS is non-empty and the effective UID of the
process is greater than zero. This results in root invocations of yggctl
preferring the system bus over the session bus.

Signed-off-by: Link Dupont <link@sub-pop.net>

cmd/yggctl/actions.go

@@ -43,14 +43,7 @@ func generateControlMessageAction(c *cli.Context) error {
 }
 
 func workersAction(c *cli.Context) error {
-	var conn *dbus.Conn
-	var err error
-
-	if os.Getenv("DBUS_SESSION_BUS_ADDRESS") != "" {
-		conn, err = dbus.ConnectSessionBus()
-	} else {
-		conn, err = dbus.ConnectSystemBus()
-	}
+	conn, err := connectBus()
 	if err != nil {
 		return cli.Exit(fmt.Errorf("cannot connect to bus: %w", err), 1)
 	}
@@ -91,14 +84,7 @@ func workersAction(c *cli.Context) error {
 }
 
 func dispatchAction(c *cli.Context) error {
-	var conn *dbus.Conn
-	var err error
-
-	if os.Getenv("DBUS_SESSION_BUS_ADDRESS") != "" {
-		conn, err = dbus.ConnectSessionBus()
-	} else {
-		conn, err = dbus.ConnectSystemBus()
-	}
+	conn, err := connectBus()
 	if err != nil {
 		return cli.Exit(fmt.Errorf("cannot connect to bus: %w", err), 1)
 	}
@@ -136,14 +122,7 @@ func dispatchAction(c *cli.Context) error {
 }
 
 func listenAction(ctx *cli.Context) error {
-	var conn *dbus.Conn
-	var err error
-
-	if os.Getenv("DBUS_SESSION_BUS_ADDRESS") != "" {
-		conn, err = dbus.ConnectSessionBus()
-	} else {
-		conn, err = dbus.ConnectSystemBus()
-	}
+	conn, err := connectBus()
 	if err != nil {
 		return cli.Exit(fmt.Errorf("cannot connect to bus: %w", err), 1)
 	}
@@ -209,3 +188,10 @@ func generateMessage(messageType, responseTo, directive, content string, metadat
 		return nil, fmt.Errorf("unsupported message type: %v", messageType)
 	}
 }
+
+func connectBus() (*dbus.Conn, error) {
+	if os.Getenv("DBUS_SESSION_BUS_ADDRESS") != "" && os.Geteuid() > 0 {
+		return dbus.ConnectSessionBus()
+	}
+	return dbus.ConnectSystemBus()
+}