refactor(modules)!: allow multiple instances for taskfile & process-compose

docs/index.md

@@ -13,6 +13,8 @@ devshell.mkShell {
 
   # use the modules
   lefthook.enable = true;
+
+  task.",".tasks = { ... };
 }
 ```
 

lib/modules/process_compose.nix

@@ -4,32 +4,33 @@
   config,
   ...
 }: let
-  inherit (lib) mkEnableOption mkOption mkIf types;
+  inherit (lib) mkOption types;
-  cfg = config.process-compose;
-
-  configFile = (pkgs.formats.yaml {}).generate "process-compose.yaml" cfg.config;
-  pcAlias = pkgs.writeTextFile {
-    name = "pc-alias";
-    destination = "/bin/${cfg.alias}";
-    executable = true;
-    text =
-      # sh
-      ''
-        ${pkgs.process-compose}/bin/process-compose --config "${configFile}" ''${@:1}
-      '';
-  };
 in {
-  options.process-compose = {
+  options.process-compose = mkOption {
-    enable =
+    type = types.attrsOf (types.submodule ({
-      mkEnableOption "Process-Compose"
+      config,
-      // {
+      name,
-        default = cfg.config != {};
+      ...
-      };
+    }: {
+      options = {
         alias = mkOption {
           type = types.str;
-      default = "pc";
+          default =
+            if name == "default"
+            then "pc"
+            else name;
           description = ''
             Alias for `process-compose`.
+            Defaults to `"pc"` if `${name}` is `"default"`
+          '';
+        };
+        lazy = mkOption {
+          type = types.bool;
+          default = true;
+          description = ''
+            Whether the process compose config should be built on-demand/lazily.
+            It will probably not land in the gcroot and thus might get cleaned up with every gc.
+            On the other hand, this way loading the devshell is faster. Decide for yourself :)
           '';
         };
         config = mkOption {
@@ -39,9 +40,43 @@ in {
             Configure process-compose here.
           '';
         };
+
+        configFile = mkOption {
+          internal = true;
+          type = types.package;
+        };
+        pcAlias = mkOption {
+          internal = true;
+          type = types.package;
+        };
+      };
+      config = rec {
+        configFile = (pkgs.formats.yaml {}).generate "process-compose.yaml" config.config;
+        pcAlias = pkgs.writeTextFile {
+          name = "pc-alias";
+          destination = "/bin/${config.alias}";
+          executable = true;
+          text = let
+            configScript =
+              if config.lazy
+              then
+                # sh
+                "$(nix build '${builtins.unsafeDiscardOutputDependency configFile.drvPath}^*' --no-link --print-out-paths)"
+              else configFile;
+          in
+            # sh
+            ''
+              CONFIG_FILE="${configScript}"
+              ${pkgs.process-compose}/bin/process-compose --config "$CONFIG_FILE" ''${@:1}
+            '';
+        };
+      };
+    }));
+    default = {};
+    description = ''
+      Define your process-compose instances here.
+    '';
   };
 
-  config = mkIf cfg.enable {
+  config.packages = map (val: val.pcAlias) (builtins.attrValues config.process-compose);
-    packages = [pcAlias];
-  };
 }

lib/modules/process_compose_test.nix

@@ -14,7 +14,7 @@ in {
         script = let
           shell = devshell.mkShell {
             imports = [module];
-            process-compose.enable = true;
+            process-compose."default" = {};
           };
         in
           # sh
@@ -23,6 +23,21 @@ in {
             assert "-f ${shell}/bin/pc" "/bin/pc should exist"
           '';
       }
+      {
+        name = "alias";
+        type = "script";
+        script = let
+          shell = devshell.mkShell {
+            imports = [module];
+            process-compose."pctest" = {};
+          };
+        in
+          # sh
+          ''
+            ${ntlib.helpers.scriptHelpers}
+            assert "-f ${shell}/bin/pctest" "/bin/pctest should exist"
+          '';
+      }
     ];
   };
 }

lib/modules/taskfile.nix

@@ -4,66 +4,21 @@
   config,
   ...
 }: let
-  inherit (lib) mapAttrs mkEnableOption mkOption mkIf types;
+  inherit (lib) mapAttrs mkOption types;
-  cfg = config.task;
-
-  patchedTasks = mapAttrs (_name: value: let
-    taskDir = value.dir or "";
-    absolutePathOrTemplate = (builtins.substring 0 1 taskDir) == "/" || (builtins.substring 0 1 taskDir) == "{";
-  in
-    value
-    // {
-      dir =
-        if absolutePathOrTemplate
-        then taskDir
-        else ''{{env "TASK_ROOT_DIR" | default .USER_WORKING_DIR}}/${taskDir}'';
-    })
-  cfg.tasks;
 
   generator = name: value:
     pkgs.writeTextFile {
       inherit name;
       text = builtins.toJSON value;
     };
-  # NOTE: this requires python just to convert json to yaml, since json is valid yaml we just ignore that
-  # generator = (pkgs.formats.yaml {}).generate;
-  taskfile = generator "taskfile" {
-    version = 3;
-    inherit (cfg) interval;
-    tasks = patchedTasks;
-  };
-  # when using a , as alias for example the store path looks weird.
-  # This way it can be identified as being the task alias
-  taskAlias = pkgs.writeTextFile {
-    name = "task-alias";
-    destination = "/bin/${cfg.alias}";
-    executable = true;
-    text = let
-      taskfileScript =
-        if cfg.lazy
-        then
-          # sh
-          "$(nix build '${builtins.unsafeDiscardOutputDependency taskfile.drvPath}^*' --no-link --print-out-paths)"
-        else taskfile;
-    in
-      # sh
-      ''
-        TASKFILE="${taskfileScript}"
-        STATE_DIR="''${REN_STATE:-''${DEVENV_STATE:-''${PRJ_CACHE_HOME}}}"
-        ROOT_DIR="''${REN_ROOT:-''${DEVENV_ROOT:-''${PRJ_ROOT}}}"
-
-        TASK_TEMP_DIR="''${STATE_DIR}/.task" \
-          TASK_ROOT_DIR="$ROOT_DIR" \
-          ${pkgs.go-task}/bin/task --taskfile "$TASKFILE" ''${@:1}
-      '';
-  };
 in {
-  options.task = {
+  options.task = mkOption {
-    enable =
+    type = types.attrsOf (types.submodule ({
-      mkEnableOption "Task"
+      config,
-      // {
+      name,
-        default = cfg.tasks != {};
+      ...
-      };
+    }: {
+      options = {
         lazy = mkOption {
           type = types.bool;
           default = true;
@@ -75,9 +30,13 @@ in {
         };
         alias = mkOption {
           type = types.str;
-      default = "task";
+          default =
+            if name == "default"
+            then "task"
+            else name;
           description = ''
             Alias for `task`, eg. set to `,` to be able to run `, --list-all`.
+            Defaults to `"task"` if `${name}` is `"default"`
           '';
         };
         interval = mkOption {
@@ -94,9 +53,69 @@ in {
             Configure all your tasks here.
           '';
         };
+
+        taskfile = mkOption {
+          internal = true;
+          type = types.package;
+        };
+        taskAlias = mkOption {
+          internal = true;
+          type = types.package;
+        };
+      };
+      config = let
+        patchedTasks = mapAttrs (_name: value: let
+          taskDir = value.dir or "";
+          absolutePathOrTemplate = (builtins.substring 0 1 taskDir) == "/" || (builtins.substring 0 1 taskDir) == "{";
+        in
+          value
+          // {
+            dir =
+              if absolutePathOrTemplate
+              then taskDir
+              else ''{{env "TASK_ROOT_DIR" | default .USER_WORKING_DIR}}/${taskDir}'';
+          })
+        config.tasks;
+      in rec {
+        # NOTE: this requires python just to convert json to yaml, since json is valid yaml we just ignore that
+        # generator = (pkgs.formats.yaml {}).generate;
+        taskfile = generator "taskfile" {
+          version = 3;
+          inherit (config) interval;
+          tasks = patchedTasks;
+        };
+        # when using a , as alias for example the store path looks weird.
+        # This way it can be identified as being the task alias
+        taskAlias = pkgs.writeTextFile {
+          name = "task-alias";
+          destination = "/bin/${config.alias}";
+          executable = true;
+          text = let
+            taskfileScript =
+              if config.lazy
+              then
+                # sh
+                "$(nix build '${builtins.unsafeDiscardOutputDependency taskfile.drvPath}^*' --no-link --print-out-paths)"
+              else taskfile;
+          in
+            # sh
+            ''
+              TASKFILE="${taskfileScript}"
+              STATE_DIR="''${REN_STATE:-''${DEVENV_STATE:-''${PRJ_CACHE_HOME}}}"
+              ROOT_DIR="''${REN_ROOT:-''${DEVENV_ROOT:-''${PRJ_ROOT}}}"
+
+              TASK_TEMP_DIR="''${STATE_DIR}/.task" \
+                TASK_ROOT_DIR="$ROOT_DIR" \
+                ${pkgs.go-task}/bin/task --taskfile "$TASKFILE" ''${@:1}
+            '';
+        };
+      };
+    }));
+    default = {};
+    description = ''
+      Define your Taskfile instances here.
+    '';
   };
 
-  config = mkIf cfg.enable {
+  config.packages = map (val: val.taskAlias) (builtins.attrValues config.task);
-    packages = [taskAlias];
-  };
 }

lib/modules/taskfile_test.nix

@@ -14,7 +14,7 @@ in {
         script = let
           shell = devshell.mkShell {
             imports = [module];
-            task.enable = true;
+            task."default" = {};
           };
         in
           # sh
@@ -29,10 +29,7 @@ in {
         script = let
           shell = devshell.mkShell {
             imports = [module];
-            task = {
+            task."," = {};
-              enable = true;
-              alias = ",";
-            };
           };
         in
           # sh

nix/repo/devShells.nix

@@ -25,14 +25,11 @@ in {
       treefmtWrapper
     ];
 
-    task = {
+    task.",".tasks = {
-      alias = ",";
-      tasks = {
       "hello" = {
         cmd = "echo world!";
       };
     };
-    };
 
     lefthook.config = {
       skip_output = ["meta" "execution_out"];
@@ -54,7 +51,7 @@ in {
       };
     };
 
-    process-compose.config.processes = {
+    process-compose."default".config.processes = {
       hello.command = "echo 'Hello World'";
       pc = {
         command = "echo 'From Process Compose'";