Add devShell option (#2)

srid · Mar 30, 2023 · 6000244 · 6000244
1 parent dc7ba61
commit 6000244
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -20,7 +20,7 @@ A `flake-parts` module for finding your way to the project root directory
     }; 
 ```
 
-Now you have access to the program that returns the absolute path to the project root via `${lib.getExe config.flake-root.package}`.
+Now you have access to the program that returns the absolute path to the project root via `${lib.getExe config.flake-root.package}`. There is also `config.flake-root.devShell` which exposes a `shellHook` providing the `$FLAKE_ROOT` environment variable.
 
 ## Examples
 

diff --git a/flake-module.nix b/flake-module.nix
@@ -18,9 +18,23 @@ in
                 description = lib.mdDoc "The name of the unique file that exists only at project root";
                 default = "flake.nix";
               };
-              # Functions
+              devShell = mkOption {
+                type = types.package;
+                readOnly = true;
+                description = lib.mdDoc ''
+                  Devshell providing a shellHook setting $FLAKE_ROOT
+                '';
+                default = pkgs.mkShell {
+                  name = "flake-root-devshell";
+                  shellHook = ''
+                    FLAKE_ROOT="''$(${lib.getExe config.flake-root.package})"
+                    export FLAKE_ROOT
+                  '';
+                };
+              };
               package = mkOption {
                 type = types.package;
+                readOnly = true;
                 description = lib.mdDoc ''
                   The Nix package providing the command to find the project root.
                 '';
@@ -57,7 +71,7 @@ in
             description = lib.mdDoc ''
               flake-root module options
             '';
-            default = {};
+            default = { };
           };
         });
   };