From 13c3169b413c0075a9ceb06b014f0e4b2810127e Mon Sep 17 00:00:00 2001
From: Johannes Kirschbauer <hsjobeki@gmail.com>
Date: Mon, 24 Jun 2024 15:47:25 +0200
Subject: [PATCH] lib: eval clan module as lib function

---
 lib/default.nix                   |  1 +
 lib/eval-clan-modules/default.nix | 34 +++++++++++++++++++++++++++++++
 lib/flake-module.nix              |  2 +-
 pkgs/schemas/flake-module.nix     | 28 ++++++-------------------
 4 files changed, 42 insertions(+), 23 deletions(-)
 create mode 100644 lib/eval-clan-modules/default.nix

diff --git a/lib/default.nix b/lib/default.nix
index e9fbde4d..ef5fe7ac 100644
--- a/lib/default.nix
+++ b/lib/default.nix
@@ -5,6 +5,7 @@
   ...
 }:
 {
+  evalClanModules = import ./eval-clan-modules { inherit clan-core nixpkgs lib; };
   jsonschema = import ./jsonschema { inherit lib; };
   modules = import ./description.nix { inherit clan-core lib; };
   buildClan = import ./build-clan { inherit clan-core lib nixpkgs; };
diff --git a/lib/eval-clan-modules/default.nix b/lib/eval-clan-modules/default.nix
new file mode 100644
index 00000000..814388d7
--- /dev/null
+++ b/lib/eval-clan-modules/default.nix
@@ -0,0 +1,34 @@
+{
+  nixpkgs,
+  clan-core,
+  lib,
+}:
+let
+  inherit (import nixpkgs { system = "x86_64-linux"; }) pkgs;
+
+  inherit (clan-core) clanModules;
+
+  baseModule = {
+    imports = (import (pkgs.path + "/nixos/modules/module-list.nix")) ++ [
+      {
+        nixpkgs.hostPlatform = "x86_64-linux";
+        clan.core.clanName = "dummy";
+      }
+    ];
+  };
+
+  # This function takes a list of module names and evaluates them
+  # evalClanModules :: [ String ] -> { config, options, ... }
+  evalClanModules =
+    modulenames:
+    let
+      evaled = lib.evalModules {
+        modules = [
+          baseModule
+          clan-core.nixosModules.clanCore
+        ] ++ (map (name: clanModules.${name}) modulenames);
+      };
+    in
+    evaled;
+in
+evalClanModules
diff --git a/lib/flake-module.nix b/lib/flake-module.nix
index 48437bb1..ef93467b 100644
--- a/lib/flake-module.nix
+++ b/lib/flake-module.nix
@@ -7,7 +7,7 @@
 {
   imports = [ ./jsonschema/flake-module.nix ];
   flake.lib = import ./default.nix {
-    inherit lib;
+    inherit lib inputs;
     inherit (inputs) nixpkgs;
     clan-core = self;
   };
diff --git a/pkgs/schemas/flake-module.nix b/pkgs/schemas/flake-module.nix
index 9ba6f4ef..59c25d01 100644
--- a/pkgs/schemas/flake-module.nix
+++ b/pkgs/schemas/flake-module.nix
@@ -10,37 +10,21 @@
       #   borgbackup = self.clanModules.borgbackup;
       # };
 
-      baseModule = {
-        imports = (import (pkgs.path + "/nixos/modules/module-list.nix")) ++ [
-          {
-            nixpkgs.hostPlatform = "x86_64-linux";
-            clan.core.clanName = "dummy";
-          }
-        ];
-      };
-
       optionsFromModule =
-        modulename: module:
+        mName:
         let
-          evaled = lib.evalModules {
-            modules = [
-              module
-              baseModule
-              self.nixosModules.clanCore
-            ];
-          };
+          eval = self.lib.evalClanModules [ mName ];
         in
-        # Filter out "injected" options that are not part of the module
-        if (evaled.options.clan ? "${modulename}") then evaled.options.clan.${modulename} else { };
+        if (eval.options.clan ? "${mName}") then eval.options.clan.${mName} else { };
 
       clanModuleSchemas = lib.mapAttrs (
-        modulename: module: self.lib.jsonschema.parseOptions (optionsFromModule modulename module)
+        modulename: _: self.lib.jsonschema.parseOptions (optionsFromModule modulename)
       ) clanModules;
 
-      clanModuleFunctionSchemas = lib.mapAttrsFlatten (modulename: module: {
+      clanModuleFunctionSchemas = lib.mapAttrsFlatten (modulename: _: {
         name = modulename;
         description = self.lib.modules.getShortDescription modulename;
-        parameters = self.lib.jsonschema.parseOptions (optionsFromModule modulename module);
+        parameters = self.lib.jsonschema.parseOptions (optionsFromModule modulename);
       }) clanModules;
     in
     rec {