1 files changed, 19 insertions, 0 deletions
diff --git a/lib/types/populate.nix b/lib/types/populate.nix
index 6264f99..a194831 100644
--- a/lib/types/populate.nix
+++ b/lib/types/populate.nix
@@ -31,6 +31,14 @@
           source-types.file
         ]);
       };
+      flake = lib.mkOption {
+        apply = x:
+          if lib.types.str.check x
+            then { flake = x; }
+            else x;
+        default = null;
+        type = lib.types.nullOr (lib.types.either lib.types.str source-types.flake);
+      };
       git = lib.mkOption {
         default = null;
         type = lib.types.nullOr source-types.git;
@@ -137,6 +145,17 @@
         };
       };
     };
+    flake = lib.types.submodule {
+      options = {
+        flake = lib.mkOption {
+          type = lib.types.str; # TODO lib.types.flake-ref
+        };
+        input = lib.mkOption {
+          default = null;
+          type = lib.types.nullOr lib.types.str;
+        };
+      };
+    };
     git = lib.types.submodule {
       options = {
         clean = {