Add GPU support loading DRI devices and assigning related groupse (#543)

Author: rjtokenring

internal/orchestrator/orchestrator.go

@@ -823,20 +823,26 @@ func getDevices() []string {
 		panic(err)
 	}
 
-	// Get video devices
-	tmpFilter := deviceList
-	tmpFilter.FilterPrefix("video")
-	videoDevices := tmpFilter.AsStrings()
-
-	// Get audio devices
-	tmpFilter = deviceList
-	tmpFilter.FilterPrefix("snd")
-	soundDevices := []string{}
-	if len(tmpFilter.AsStrings()) > 0 {
-		soundDevices = append(soundDevices, "/dev/snd") // Add /dev/snd as a sound device
+	devices := []string{}
+	addSoundDevice, addGPUDevice := false, false
+
+	for _, p := range deviceList {
+		switch {
+		case p.HasPrefix("video"):
+			devices = append(devices, p.String())
+		case p.HasPrefix("snd"):
+			addSoundDevice = true
+		case p.HasPrefix("dri"):
+			addGPUDevice = true
+		}
 	}
-
-	return append(videoDevices, soundDevices...)
+	if addSoundDevice {
+		devices = append(devices, "/dev/snd")
+	}
+	if addGPUDevice {
+		devices = append(devices, "/dev/dri")
+	}
+	return devices
 }
 
 func disconnectSerialFromRPCRouter(ctx context.Context, portAddress string, cfg config.Configuration) func() {

internal/orchestrator/provision.go

@@ -310,7 +310,7 @@ func generateMainComposeFile(
 	}
 
 	devices := getDevices()
-	groups := []string{"dialout", "video", "audio"}
+	groups := []string{"dialout", "video", "audio", "render"}
 
 	mainAppCompose.Services = &mainService{
 		Main: service{
@@ -407,11 +407,11 @@ func generateServicesOverrideFile(services []string, servicesThatRequireDevices
 	overrideCompose.Services = make(map[string]serviceOverride, len(services))
 	for _, svc := range services {
 		override := serviceOverride{
-			User:     user,
-			GroupAdd: groups,
+			User: user,
 		}
 		if slices.Contains(servicesThatRequireDevices, svc) {
 			override.Devices = &devices
+			override.GroupAdd = groups
 		}
 		overrideCompose.Services[svc] = override
 	}