home-manager/modules/programs/mise.nix

{ config, lib, pkgs, ... }:

with lib;

let
  cfg = config.programs.mise;
  tomlFormat = pkgs.formats.toml { };
in {
  meta.maintainers = [ hm.maintainers.pedorich-n ];

  imports = let
    mkRemovedWarning = opt:
      (mkRemovedOptionModule [ "programs" "rtx" opt ] ''
        The `rtx` package has been replaced by `mise`, please switch over to
        using the options under `programs.mise.*` instead.
      '');

  in map mkRemovedWarning [
    "enable"
    "package"
    "enableBashIntegration"
    "enableZshIntegration"
    "enableFishIntegration"
    "enableNushellIntegration"
    "settings"
  ];

  options = {
    programs.mise = {
      enable = mkEnableOption "mise";

      package = mkPackageOption pkgs "mise" { nullable = true; };

      enableBashIntegration =
        lib.hm.shell.mkBashIntegrationOption { inherit config; };

      enableFishIntegration =
        lib.hm.shell.mkFishIntegrationOption { inherit config; };

      enableZshIntegration =
        lib.hm.shell.mkZshIntegrationOption { inherit config; };

      enableNushellIntegration =
        lib.hm.shell.mkNushellIntegrationOption { inherit config; };

      globalConfig = mkOption {
        type = tomlFormat.type;
        default = { };
        example = literalExpression ''
          tools = {
            node = "lts";
            python = ["3.10" "3.11"];
          };

          aliases = {
            my_custom_node = "20";
          };
        '';
        description = ''
          Config written to {file}`$XDG_CONFIG_HOME/mise/config.toml`.

          See <https://mise.jdx.dev/configuration.html#global-config-config-mise-config-toml>
          for details on supported values.
        '';
      };

      settings = mkOption {
        type = tomlFormat.type;
        default = { };
        example = literalExpression ''
          verbose = false;
          experimental = false;
          disable_tools = ["node"];
        '';
        description = ''
          Settings written to {file}`$XDG_CONFIG_HOME/mise/settings.toml`.

          See <https://mise.jdx.dev/configuration.html#settings-file-config-mise-settings-toml>
          for details on supported values.
        '';
      };
    };
  };

  config = mkIf cfg.enable {
    warnings = optional (cfg.package == null && (cfg.enableBashIntegration
      || cfg.enableZshIntegration || cfg.enableFishIntegration
      || cfg.enableNushellIntegration)) ''
        You have enabled shell integration for `mise` but have not set `package`.

        The shell integration will not be added.
      '';

    home.packages = lib.mkIf (cfg.package != null) [ cfg.package ];

    xdg.configFile = {
      "mise/config.toml" = mkIf (cfg.globalConfig != { }) {
        source = tomlFormat.generate "mise-config" cfg.globalConfig;
      };

      "mise/settings.toml" = mkIf (cfg.settings != { }) {
        source = tomlFormat.generate "mise-settings" cfg.settings;
      };
    };

    programs = {
      bash.initExtra = mkIf cfg.enableBashIntegration ''
        eval "$(${getExe cfg.package} activate bash)"
      '';

      zsh.initExtra = mkIf cfg.enableZshIntegration ''
        eval "$(${getExe cfg.package} activate zsh)"
      '';

      fish.interactiveShellInit = mkIf cfg.enableFishIntegration ''
        ${getExe cfg.package} activate fish | source
      '';

      nushell = mkIf cfg.enableNushellIntegration {
        extraEnv = ''
          let mise_path = $nu.default-config-dir | path join mise.nu
          ^mise activate nu | save $mise_path --force
        '';
        extraConfig = ''
          use ($nu.default-config-dir | path join mise.nu)
        '';
      };
    };
  };
}