nixos/lib/default.nix

inputs@{
  unstable,
  self,
  disko,
  ...
}:
let
  inherit (unstable) lib;
in
# Only good use case for rec
rec {
  wrench = lib.flip lib.pipe;

  needsSystem = lib.flip builtins.elem [
    "defaultPackage"
    "devShell"
    "devShells"
    "formatter"
    "legacyPackages"
    "packages"
  ];

  constructInputs' =
    system:
    wrench [
      (lib.filterAttrs (_: lib.isType "flake"))
      (lib.mapAttrs (_: lib.mapAttrs (name: value: if needsSystem name then value.${system} else value)))
    ];

  listNixFilesRecursive = wrench [
    builtins.unsafeDiscardStringContext
    lib.filesystem.listFilesRecursive
    (builtins.filter (x: !lib.hasPrefix "_" (builtins.baseNameOf x) && lib.hasSuffix ".nix" x))
  ];

  mkModules =
    path:
    lib.pipe path [
      listNixFilesRecursive
      (map (name: {
        name = lib.pipe name [
          (lib.removeSuffix ".nix")
          (lib.removePrefix "${path}/")
        ];
        value = addSchizophreniaToModule name;
      }))
      builtins.listToAttrs
    ];

  addSchizophreniaToModule =
    x:
    let
      # the imported module
      imported = import x;
    in
    /*
      If the module isn't a function then
      it doesn't need arguments and error
      message locations will function correctly
    */
    if !lib.isFunction imported then
      x
    else
      let
        # all arguments defined in the module
        funcArgs = lib.functionArgs imported;
        /*
          The names of all arguments which will be
          available to be inserted into the module arguments
        */
        argNames = builtins.attrNames inputs ++ [
          "inputs"
          "inputs'"
          "self'"
          "_dir"
        ];

        /*
          arguments to be passed minus
          per system attributes
          for example flake-parts-esque inputs'
        */
        argsPre = {
          inherit inputs self;
          /*
            _dir is the "self" derived
            path to the directory containing the module
          */
          _dir = builtins.dirOf x;
        };

        /*
          arguments which will be inserted
          set to the before per-system values
        */
        providedArgs = lib.pipe funcArgs [
          (lib.filterAttrs (n: _: builtins.elem n argNames))
          (lib.mapAttrs (n: _: argsPre.${n} or { }))
        ];

        /*
          arguments which the module system
          not provided here. either to be
          provided by the module system or invalid
        */
        neededArgs = lib.filterAttrs (n: _: !builtins.elem n argNames) funcArgs;
      in
      {
        __functionArgs = neededArgs // {
          /*
            always require pkgs to be passed
            to derive system from pkgs.stdenv.system
          */
          pkgs = false;
        };

        __functor =
          /*
            args is specialArgs + _module.args which are needed
            and always pkgs
          */
          _: args:
          imported (
            /*
              take module system provided arguments
              filter them so only the required ones are passed
            */
            (lib.filterAttrs (n: _: neededArgs ? ${n}) args)
            # add needed arguments
            // (
              providedArgs
              # add system dependent arguments
              // (
                let
                  inputs' = constructInputs' args.pkgs.stdenv.system inputs;

                  actuallyAllArgs = inputs' // {
                    inherit inputs';
                    self' = inputs'.self;
                    inherit (inputs) self;
                  };
                in
                lib.filterAttrs (n: _: providedArgs ? ${n}) actuallyAllArgs
              )
            )
          )
          # add _file to the final module attribute set
          // {
            _file = x;
          };
      };

  gerg-utils =
    pkgsf: outputs:
    lib.pipe
      [
        "x86_64-linux"
        "aarch64-linux"
      ]
      [
        (map (
          system:
          builtins.mapAttrs (
            name: value: if needsSystem name then { ${system} = value (pkgsf system); } else value
          ) outputs
        ))
        (lib.foldAttrs lib.mergeAttrs { })
      ];

  mkHosts = wrench [
    builtins.readDir
    (lib.filterAttrs (_: v: v == "directory"))
    builtins.attrNames
    (map (x: {
      name = x;
      value = lib.evalModules {
        specialArgs.modulesPath = "${unstable}/nixos/modules";
        modules = builtins.concatLists [
          (builtins.attrValues self.nixosModules)
          (map addSchizophreniaToModule (listNixFilesRecursive "${self}/hosts/${x}"))
          (import "${unstable}/nixos/modules/module-list.nix")
          (lib.optionals (self.diskoConfigurations ? "disko-${x}") [
            self.diskoConfigurations."disko-${x}"
            disko.nixosModules.default
          ])
        ];
      };
    }))
    builtins.listToAttrs
  ];

  mkDisko =
    path:
    lib.pipe path [
      builtins.readDir
      (lib.filterAttrs (n: v: v == "regular" && lib.hasSuffix ".nix" n))
      builtins.attrNames
      (map (x: {
        name = lib.pipe x [
          (lib.removeSuffix path)
          (lib.removeSuffix ".nix")
          (x: "disko-${x}")
        ];
        value.disko.devices = import "${path}/${x}" lib;
      }))
      builtins.listToAttrs
    ];

  /*
    /<name> -> packages named by directory
    /<name>/call.nix ->  callPackage override imported via import <file> pkgs
    call.nix example
      {python3Packages}: {
        inherit (python3Packages) callPackage;
        args = {};
     }

    /<name>/package.nix -> the package itself
    /<name>/wrapper.nix:
    a optional wrapper for the package
    which is callPackage'd with the original package
    as an argument named <name>-unwrapped
  */
  mkPackages =
    path: pkgs:
    lib.pipe path [
      builtins.readDir
      (lib.filterAttrs (_: v: v == "directory"))
      builtins.attrNames
      (map (
        n:
        let
          p = "${path}/${n}";

          callPackage =
            file: args:
            let
              defaultArgs =
                pkgs
                // pkgs.xorg
                // (
                  let
                    inputs' = constructInputs' pkgs.stdenv.system inputs;
                  in
                  {
                    inherit inputs' inputs;
                    self' = inputs'.self;
                    inherit (inputs) self;
                    # npins sources if i ever use them
                    # sources = lib.mapAttrs (_: pkgs.npins.mkSource) (lib.importJSON "${self}/packages/sources.json").pins;
                  }
                );
              _callPackage = lib.callPackageWith defaultArgs;
              fullPath = "${p}/${file}.nix";
              callPath = "${p}/call.nix";
            in
            assert lib.assertMsg (builtins.pathExists fullPath)
              "File attempting to be callPackage'd '${fullPath}' does not exist";

            if builtins.pathExists callPath then
              let
                x = _callPackage callPath { };
              in
              x.callPackage or _callPackage fullPath (x.args or defaultArgs // args)

            else
              _callPackage fullPath args;
        in

        if builtins.pathExists "${p}/wrapper.nix" then
          # My distaste for rec grows ever stronger
          let
            set."${n}-unwrapped" = callPackage "package" { };
          in
          { ${n} = callPackage "wrapper" set; } // set
        else
          { ${n} = callPackage "package" { }; }

      ))
      lib.mergeAttrsList
    ];
}