Automated merge from development to main

Author: github-actions[bot]

src/main/java/io/github/intisy/docker/model/DeviceMapping.java

@@ -0,0 +1,62 @@
+package io.github.intisy.docker.model;
+
+import com.google.gson.annotations.SerializedName;
+
+/**
+ * Represents a device mapping from host to container.
+ *
+ * @author Finn Birich
+ */
+public class DeviceMapping {
+    @SerializedName("PathOnHost")
+    private String pathOnHost;
+
+    @SerializedName("PathInContainer")
+    private String pathInContainer;
+
+    @SerializedName("CgroupPermissions")
+    private String cgroupPermissions;
+
+    public DeviceMapping() {
+    }
+
+    public DeviceMapping(String pathOnHost, String pathInContainer, String cgroupPermissions) {
+        this.pathOnHost = pathOnHost;
+        this.pathInContainer = pathInContainer;
+        this.cgroupPermissions = cgroupPermissions;
+    }
+
+    /**
+     * Create a device mapping with read-write-mknod permissions.
+     */
+    public static DeviceMapping of(String pathOnHost, String pathInContainer) {
+        return new DeviceMapping(pathOnHost, pathInContainer, "rwm");
+    }
+
+    public String getPathOnHost() {
+        return pathOnHost;
+    }
+
+    public DeviceMapping setPathOnHost(String pathOnHost) {
+        this.pathOnHost = pathOnHost;
+        return this;
+    }
+
+    public String getPathInContainer() {
+        return pathInContainer;
+    }
+
+    public DeviceMapping setPathInContainer(String pathInContainer) {
+        this.pathInContainer = pathInContainer;
+        return this;
+    }
+
+    public String getCgroupPermissions() {
+        return cgroupPermissions;
+    }
+
+    public DeviceMapping setCgroupPermissions(String cgroupPermissions) {
+        this.cgroupPermissions = cgroupPermissions;
+        return this;
+    }
+}

src/main/java/io/github/intisy/docker/model/DeviceRequest.java

@@ -0,0 +1,133 @@
+package io.github.intisy.docker.model;
+
+import com.google.gson.annotations.SerializedName;
+
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * Represents a device request for container access to host devices (e.g., GPUs).
+ *
+ * @author Finn Birich
+ */
+public class DeviceRequest {
+    @SerializedName("Driver")
+    private String driver;
+
+    @SerializedName("Count")
+    private Integer count;
+
+    @SerializedName("DeviceIDs")
+    private List<String> deviceIDs;
+
+    @SerializedName("Capabilities")
+    private List<List<String>> capabilities;
+
+    @SerializedName("Options")
+    private Map<String, String> options;
+
+    public DeviceRequest() {
+    }
+
+    /**
+     * Create a device request for all NVIDIA GPUs.
+     */
+    public static DeviceRequest requestAllGpus() {
+        return new DeviceRequest()
+                .withDriver("nvidia")
+                .withCount(-1)
+                .withCapabilities(Collections.singletonList(Collections.singletonList("gpu")));
+    }
+
+    /**
+     * Create a device request for a specific number of GPUs.
+     */
+    public static DeviceRequest requestGpus(int count) {
+        return new DeviceRequest()
+                .withDriver("nvidia")
+                .withCount(count)
+                .withCapabilities(Collections.singletonList(Collections.singletonList("gpu")));
+    }
+
+    /**
+     * Create a device request for specific GPU device IDs.
+     */
+    public static DeviceRequest requestGpusByIds(List<String> deviceIds) {
+        return new DeviceRequest()
+                .withDriver("nvidia")
+                .withDeviceIDs(deviceIds)
+                .withCapabilities(Collections.singletonList(Collections.singletonList("gpu")));
+    }
+
+    public String getDriver() {
+        return driver;
+    }
+
+    public DeviceRequest setDriver(String driver) {
+        this.driver = driver;
+        return this;
+    }
+
+    public DeviceRequest withDriver(String driver) {
+        this.driver = driver;
+        return this;
+    }
+
+    public Integer getCount() {
+        return count;
+    }
+
+    public DeviceRequest setCount(Integer count) {
+        this.count = count;
+        return this;
+    }
+
+    public DeviceRequest withCount(Integer count) {
+        this.count = count;
+        return this;
+    }
+
+    public List<String> getDeviceIDs() {
+        return deviceIDs;
+    }
+
+    public DeviceRequest setDeviceIDs(List<String> deviceIDs) {
+        this.deviceIDs = deviceIDs;
+        return this;
+    }
+
+    public DeviceRequest withDeviceIDs(List<String> deviceIDs) {
+        this.deviceIDs = deviceIDs;
+        return this;
+    }
+
+    public List<List<String>> getCapabilities() {
+        return capabilities;
+    }
+
+    public DeviceRequest setCapabilities(List<List<String>> capabilities) {
+        this.capabilities = capabilities;
+        return this;
+    }
+
+    public DeviceRequest withCapabilities(List<List<String>> capabilities) {
+        this.capabilities = capabilities;
+        return this;
+    }
+
+    public Map<String, String> getOptions() {
+        return options;
+    }
+
+    public DeviceRequest setOptions(Map<String, String> options) {
+        this.options = options;
+        return this;
+    }
+
+    public DeviceRequest withOptions(Map<String, String> options) {
+        this.options = options;
+        return this;
+    }
+}

src/main/java/io/github/intisy/docker/model/HostConfig.java

@@ -79,6 +79,15 @@ public class HostConfig {
     @SerializedName("ShmSize")
     private Long shmSize;
 
+    @SerializedName("DeviceRequests")
+    private List<DeviceRequest> deviceRequests;
+
+    @SerializedName("Devices")
+    private List<DeviceMapping> devices;
+
+    @SerializedName("Runtime")
+    private String runtime;
+
     public HostConfig() {
     }
 
@@ -312,6 +321,49 @@ public HostConfig setShmSize(Long shmSize) {
         return this;
     }
 
+    public List<DeviceRequest> getDeviceRequests() {
+        return deviceRequests;
+    }
+
+    public HostConfig setDeviceRequests(List<DeviceRequest> deviceRequests) {
+        this.deviceRequests = deviceRequests;
+        return this;
+    }
+
+    public HostConfig addDeviceRequest(DeviceRequest deviceRequest) {
+        if (this.deviceRequests == null) {
+            this.deviceRequests = new ArrayList<>();
+        }
+        this.deviceRequests.add(deviceRequest);
+        return this;
+    }
+
+    public List<DeviceMapping> getDevices() {
+        return devices;
+    }
+
+    public HostConfig setDevices(List<DeviceMapping> devices) {
+        this.devices = devices;
+        return this;
+    }
+
+    public HostConfig addDevice(DeviceMapping device) {
+        if (this.devices == null) {
+            this.devices = new ArrayList<>();
+        }
+        this.devices.add(device);
+        return this;
+    }
+
+    public String getRuntime() {
+        return runtime;
+    }
+
+    public HostConfig setRuntime(String runtime) {
+        this.runtime = runtime;
+        return this;
+    }
+
     /**
      * Restart policy for a container.
      */