marleyos/modules/home/options/shellAbbrs/default.nix

{
  config,
  lib,
  ...
}: let
  # https://github.com/nix-community/home-manager/blob/master/modules/programs/fish.nix
  abbrModule = lib.types.submodule {
    options = {
      expansion = lib.mkOption {
        type = with lib.types; nullOr str;
        default = null;
        description = "The command expanded by an abbreviation.";
      };

      position = lib.mkOption {
        type = with lib.types; nullOr str;
        default = null;
        example = "anywhere";
        description = ''
          If the position is "command", the abbreviation expands only if the
          position is a command. If it is "anywhere", the abbreviation expands
          anywhere.
        '';
      };

      regex = lib.mkOption {
        type = with lib.types; nullOr str;
        default = null;
        description = ''
          The regular expression pattern matched instead of the literal name.
        '';
      };

      setCursor = lib.mkOption {
        type = with lib.types; (either bool str);
        default = false;
        description = ''
          The marker indicates the position of the cursor when the abbreviation
          is expanded. When setCursor is true, the marker is set with a default
          value of "%".
        '';
      };

      function = lib.mkOption {
        type = with lib.types; nullOr str;
        default = null;
        description = "The fish function expanded instead of a literal string.";
      };
    };
  };

  removeFishOnly = lib.attrsets.filterAttrs (
    _: v:
      if (builtins.isAttrs v)
      then
        !(
          (v ? regex) || (v ? setCursor) || (v ? function) || ((v ? position) && (v.position == "anywhere"))
        )
      else true
  );

  abbrToAlias =
    lib.attrsets.mapAttrs (
      _: v:
        if (builtins.isAttrs v)
        then v.expansion
        else v
    )
    removeFishOnly;
in {
  options = {
    # https://github.com/nix-community/home-manager/blob/master/modules/programs/fish.nix
    home.shellAbbrs = lib.mkOption {
      type = with lib.types; attrsOf (either str abbrModule);
      default = {};
      example = lib.literalExpression ''
        {
          l = "less";
          gco = "git checkout";
          "-C" = {
            position = "anywhere";
            expansion = "--color";
          };
        }
      '';
      description = ''
        An attribute set that maps aliases (the top level attribute names in
        this option) to abbreviations. Abbreviations are expanded with the
        longer phrase after they are entered.
      '';
    };
  };

  config = let
    fishCfg = config.programs.fish;
    inherit (config.home) shellAbbrs;
  in {
    programs.fish.shellAbbrs = lib.mkIf fishCfg.enable shellAbbrs;

    home.shellAliases = lib.mkIf (!fishCfg.enable) (abbrToAlias shellAbbrs);
  };
}