home-manager/modules/programs/sagemath.nix

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

let

  cfg = config.programs.sagemath;

  inherit (lib)
    literalExpression
    mkEnableOption
    mkOption
    types
    ;

in
{
  meta.maintainers = [ lib.maintainers.kirelagin ];

  options.programs.sagemath = {
    enable = mkEnableOption "SageMath, a mathematics software system";

    package = lib.mkPackageOption pkgs "sage" {
      nullable = true;
      extraDescription = ''
        The SageMath package to use.
      '';
    };

    configDir = mkOption {
      type = types.str;
      default = "${config.xdg.configHome}/sage";
      defaultText = literalExpression "\${config.xdg.configHome}/sage";
      description = ''
        Directory where the {file}`sage.init` file will be stored.
        Note that the upstream default is {file}`~/.sage`,
        but our default is to follow XDG.
      '';
    };

    dataDir = mkOption {
      type = types.str;
      default = "${config.xdg.dataHome}/sage";
      defaultText = literalExpression "\${config.xdg.dataHome}/sage";
      description = ''
        Location for {env}`DOT_SAGE`.
        Note that the upstream default is {file}`~/.sage`,
        but our default is to follow XDG.
      '';
    };

    initScript = mkOption {
      type = types.lines;
      default = "";
      example = "%colors linux";
      description = ''
        Contents of the {file}`init.sage` file that is loaded on startup.
      '';
    };
  };

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

    home.file."${cfg.configDir}/init.sage".text = cfg.initScript;

    home.sessionVariables = {
      DOT_SAGE = cfg.dataDir;
      SAGE_STARTUP_FILE = "${cfg.configDir}/init.sage";
    };
  };
}