feat: adds postBuild argument and formats justfile after generation

lib/eval.nix

@@ -2,15 +2,15 @@
   Evaluates the given input files and data and returns a derivation which builds the result.
 */
 { pkgs, lib }:
-{ inputFiles, outputFile, data ? { }, cue ? pkgs.cue, ... }@args:
+{ inputFiles, outputFile, postBuild ? "", data ? { }, cue ? pkgs.cue, ... }@args:
 with pkgs.lib;
 let
   json = optionalString (data != { }) (builtins.toJSON data);
 
   defaultFlags = {
     outfile = "$out";
   };
-  extraFlags = removeAttrs args [ "inputFiles" "outputFile" "data" "cue" ];
+  extraFlags = removeAttrs args [ "inputFiles" "outputFile" "postBuild" "data" "cue" ];
 
   allFlags = defaultFlags // extraFlags;
   allInputs = inputFiles ++ optionals (json != "") [ "json: $jsonPath" ];
@@ -29,6 +29,7 @@ let
     ''
       echo "nixago: Rendering output..."
       ${cueEvalCmd}
+      ${postBuild}
     '';
 in
 result

lib/template.nix

@@ -9,14 +9,14 @@
   the module.
 */
 { pkgs, lib }:
-{ data, files, output, shellHookExtra ? "", flags ? { } }:
+{ data, files, output, postBuild ? "", shellHookExtra ? "", flags ? { } }:
 with pkgs.lib;
 let
   result = evalModules {
     modules = [
       ../modules/template.nix
       {
-        inherit data files output shellHookExtra;
+        inherit data files output postBuild shellHookExtra;
       }
     ];
     specialArgs = ({ inherit pkgs; flakeLib = lib; } // flags);

modules/template.nix

@@ -46,6 +46,7 @@ in
           inputFiles = config.files;
           outputFile = config.output;
           data = config.data;
+          postBuild = config.postBuild;
         } // flags);
     };
     data = mkOption {
@@ -60,6 +61,11 @@ in
       type = types.str;
       description = "The name of the file to generate";
     };
+    postBuild = mkOption {
+      type = types.str;
+      description = "Commands to run after the configuration is built";
+      default = "";
+    };
     shellHook = mkOption {
       type = types.str;
       description = "Shell hook for linking generated configuration file";

plugins/just/make.nix

@@ -5,6 +5,9 @@ let
     data = { data = config; };
     files = [ ./template.cue ];
     output = ".justfile";
+    postBuild = ''
+      ${pkgs.just}/bin/just --unstable --fmt -f $out
+    '';
     flags = {
       expression = "rendered";
       out = "text";

tests/just/expected.txt

@@ -1,5 +1,5 @@
 var := "value"
+
 task1:
     echo "Doing the thing"
     @doThing
-