From 990e5ce6791ff1f497a61280a82eb66e3789e0e9 Mon Sep 17 00:00:00 2001
From: will <will.bradshaw50@gmail.com>
Date: Sun, 28 Sep 2025 09:45:58 +1000
Subject: [PATCH] docker-cli: add docker contexts support

Adds docker-cli.contexts support. This allows declarative configuration
of [docker
contexts](https://docs.docker.com/engine/manage-resources/contexts/).
---
 modules/programs/docker-cli.nix               | 46 ++++++++++++++++++-
 tests/modules/programs/docker-cli/default.nix |  1 +
 .../programs/docker-cli/example-contexts.json | 11 +++++
 .../programs/docker-cli/example-contexts.nix  | 36 +++++++++++++++
 4 files changed, 93 insertions(+), 1 deletion(-)
 create mode 100644 tests/modules/programs/docker-cli/example-contexts.json
 create mode 100644 tests/modules/programs/docker-cli/example-contexts.nix

diff --git a/modules/programs/docker-cli.nix b/modules/programs/docker-cli.nix
index 668ef7f7c..937b91013 100644
--- a/modules/programs/docker-cli.nix
+++ b/modules/programs/docker-cli.nix
@@ -32,6 +32,38 @@ in
       '';
     };
 
+    contexts = mkOption {
+      type = lib.types.attrsOf (
+        lib.types.submodule (
+          { name, config, ... }:
+          {
+            freeformType = jsonFormat.type;
+            options = {
+              Name = mkOption {
+                type = lib.types.str;
+                readOnly = true;
+                description = "Name of the Docker context. Defaults to the attribute name (the <name> in programs.docker-cli.contexts.<name>). Overriding requires lib.mkForce.";
+              };
+            };
+            config.Name = name;
+          }
+        )
+      );
+      default = { };
+      example = lib.literalExpression ''
+        {
+          example = {
+            Metadata = { Description = "example1"; };
+            Endpoints.docker.Host = "unix://example2";
+          };
+        }
+      '';
+      description = ''
+        Attribute set of Docker context configurations. Each attribute name becomes the context Name; overriding requires lib.mkForce. See:
+        <https://docs.docker.com/engine/manage-resources/contexts/
+      '';
+    };
+
     settings = mkOption {
       type = jsonFormat.type;
       default = { };
@@ -62,7 +94,19 @@ in
         "${cfg.configDir}/config.json" = {
           source = jsonFormat.generate "config.json" cfg.settings;
         };
-      };
+      }
+      // lib.mapAttrs' (
+        n: ctx:
+        let
+          path = "${cfg.configDir}/contexts/meta/${builtins.hashString "sha256" ctx.Name}/meta.json";
+        in
+        {
+          name = path;
+          value = {
+            source = jsonFormat.generate "config.json" (ctx);
+          };
+        }
+      ) cfg.contexts;
     };
   };
 }
diff --git a/tests/modules/programs/docker-cli/default.nix b/tests/modules/programs/docker-cli/default.nix
index e852cf142..fa0ef54a9 100644
--- a/tests/modules/programs/docker-cli/default.nix
+++ b/tests/modules/programs/docker-cli/default.nix
@@ -1,4 +1,5 @@
 {
   docker-cli = ./example-config.nix;
   docker-cli-empty-config = ./empty-config.nix;
+  docker-cli-contexts = ./example-contexts.nix;
 }
diff --git a/tests/modules/programs/docker-cli/example-contexts.json b/tests/modules/programs/docker-cli/example-contexts.json
new file mode 100644
index 000000000..35d25cafb
--- /dev/null
+++ b/tests/modules/programs/docker-cli/example-contexts.json
@@ -0,0 +1,11 @@
+{
+  "Endpoints": {
+    "docker": {
+      "Host": "unix://example2"
+    }
+  },
+  "Metadata": {
+    "Description": "example1"
+  },
+  "Name": "example"
+}
diff --git a/tests/modules/programs/docker-cli/example-contexts.nix b/tests/modules/programs/docker-cli/example-contexts.nix
new file mode 100644
index 000000000..f76aa4556
--- /dev/null
+++ b/tests/modules/programs/docker-cli/example-contexts.nix
@@ -0,0 +1,36 @@
+{
+  config,
+  ...
+}:
+{
+  programs.docker-cli = {
+    enable = true;
+
+    configDir = ".docker2";
+
+    contexts = {
+      example = {
+        Metadata = {
+          Description = "example1";
+        };
+        Endpoints = {
+          docker = {
+            Host = "unix://example2";
+          };
+        };
+      };
+    };
+  };
+
+  nmt.script =
+    let
+      cfgDocker = config.programs.docker-cli;
+      configTestPath = "home-files/${cfgDocker.configDir}/contexts/meta/50d858e0985ecc7f60418aaf0cc5ab587f42c2570a884095a9e8ccacd0f6545c/meta.json";
+    in
+    ''
+      assertPathNotExists home-files/.docker/config.json
+      assertFileExists ${configTestPath}
+      assertFileContent ${configTestPath} \
+        ${./example-contexts.json}
+    '';
+}