1 files changed, 21 insertions, 3 deletions
diff --git a/lib/types/populate.nix b/lib/types/populate.nix
index b1d4151..e971fd8 100644
--- a/lib/types/populate.nix
+++ b/lib/types/populate.nix
@@ -39,6 +39,10 @@
         default = null;
         type = lib.types.nullOr (lib.types.either lib.types.str source-types.flake);
       };
+      flake-input = lib.mkOption {
+        default = null;
+        type = lib.types.nullOr source-types.flake-input;
+      };
       git = lib.mkOption {
         default = null;
         type = lib.types.nullOr source-types.git;
@@ -166,6 +170,17 @@
     };
     flake = lib.types.submodule {
       options = {
+        flake = lib.mkOption {
+          type = lib.types.str; # TODO lib.types.flake-ref
+        };
+        inputs = lib.mkOption {
+          default = {};
+          type = flake-inputs;
+        };
+      };
+    };
+    flake-input = lib.types.submodule {
+      options = {
         fetchOnTarget = lib.mkOption {
           default = false;
           type = lib.types.bool;
@@ -173,9 +188,12 @@
         flake = lib.mkOption {
           type = lib.types.str; # TODO lib.types.flake-ref
         };
-        input = lib.mkOption {
-          default = null;
-          type = lib.types.nullOr lib.types.str;
+        inputs = lib.mkOption {
+          default = {};
+          type = flake-inputs;
+        };
+        name = lib.mkOption {
+          type = lib.types.str;
         };
       };
     };