flocon/shared/modules/hjem/collection/environment.nix

{
  lib,
  pkgs,
  config,
  ...
}: let
  inherit (lib.attrsets) mapAttrsToList;
  inherit (lib.lists) isList;
  inherit (lib.modules) mkIf;
  inherit (lib.options) mkOption;
  inherit (lib.strings) concatStringsSep;
  inherit (lib.types) attrsOf listOf oneOf int path str;

  inherit (pkgs) writeShellScript;

  cfg = config.environment;

  writeEnvScript = let
    toEnv = env:
      if isList env
      then concatStringsSep ":" (map toString env)
      else toString env;
  in
    attrs:
      writeShellScript "load-env"
      (concatStringsSep "\n"
        (mapAttrsToList (name: value: "export ${name}=\"${toEnv value}\"") attrs));
in {
  options.environment = {
    sessionVariables = mkOption {
      type = attrsOf (oneOf [(listOf (oneOf [int str path])) int str path]);
      default = {};
      example = {
        EDITOR = "nvim";
        VISUAL = "nvim";
      };
      description = ''
        A set of environment variables used in the user environment.
        These variables will be set as systemd user environment
        variables, using `~/.profile`. The value of each
        variable can be either a string or a list of strings. The
        latter is concatenated, interspersed with colon
        characters.
      '';
    };
  };

  config.files.".profile".text = mkIf (cfg.sessionVariables != {}) ''
    . ${writeEnvScript cfg.sessionVariables}
  '';
}