feat: use a custom loader for haumea

This patch replaces the scoped loader used in `load` with a custom
version that can resolve module arguments as well.

flake.nix

@@ -51,27 +51,51 @@
       src,
     }:
     # modules/profiles are always functions
-    args @ {pkgs, ...}: let
+    {
+      config,
+      pkgs,
+      ...
+    }: let
       cr = cell.__cr ++ [(baseNameOf src)];
       file = "${self.outPath}#${lib.concatStringsSep "/" cr}";
 
-      i = args // {inherit cell inputs;};
-      defaultWith = import (haumea + /src/loaders/__defaultWith.nix) {inherit lib;};
-      loader = defaultWith (scopedImport i) i;
+      defaultWith = let
+        inherit
+          (lib)
+          functionArgs
+          pipe
+          toFunction
+          ;
+      in (importer: inputs: path: let
+        f = toFunction (importer path);
+      in
+        pipe f [
+          functionArgs
+          (let
+            context = name: ''while evaluating the argument `${name}' in "${file}":'';
+          in
+            builtins.mapAttrs (
+              name: _:
+                builtins.addErrorContext (context name)
+                (inputs.${name} or config._module.args.${name})
+            ))
+          f
+        ]);
+      loader = inputs: defaultWith (scopedImport inputs) inputs;
+      args = {inherit cell inputs;};
     in
       if lib.pathIsDirectory src
       then
         lib.setDefaultModuleLocation file (haumea.lib.load {
-          inherit src;
-          loader = haumea.lib.loaders.scoped;
+          inherit loader src;
           transformer = with haumea.lib.transformers; [
             liftDefault
             (hoistLists "_imports" "imports")
           ];
-          inputs = i;
+          inputs = args;
         })
       # Mimic haumea for a regular file
-      else lib.setDefaultModuleLocation file (loader src);
+      else lib.setDefaultModuleLocation file (loader args src);
 
     findLoad = {
       inputs,