dotfiles/modules/home/shell.nix

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

let
  cfg = config.my.shell;
in {
  options.my.shell = {
    aliases.enable = lib.mkEnableOption null;
    direnv.enable = lib.mkEnableOption null;
    fish.enable = lib.mkEnableOption null;
    globalNpmPackages.enable = lib.mkEnableOption null;
    starship.enable = lib.mkEnableOption null;
    zsh.enable = lib.mkEnableOption null;
  };

  config = lib.mkMerge [
    (lib.mkIf cfg.aliases.enable {
      home.shellAliases = {
        e = "emacsclient -n -c";
        et = "emacsclient -nw";
        v = lib.mkDefault "vim";
        jq = "jaq";
        java = "java -Dawt.useSystemAAFontSettings=lcd";
        j = "javac *.java && java Main";
        syu = "nh os switch --ask --update -- --accept-flake-config";
        nswitch = "nh os switch --ask -- --accept-flake-config";
      };
    })

    (lib.mkIf cfg.direnv.enable {
      programs.direnv = {
        enable = true;
        nix-direnv.enable = true;
      };
    })

    (lib.mkIf cfg.fish.enable {
      programs.fish = {
        enable = true;
        interactiveShellInit = let
          catppuccinMochaSrc = pkgs.fetchFromGitHub {
            owner = "catppuccin";
            repo = "fish";
            rev = "91e6d6721362be05a5c62e235ed8517d90c567c9";
            hash = "sha256-l9V7YMfJWhKDL65dNbxaddhaM6GJ0CFZ6z+4R6MJwBA=";
          };
          catppuccinMochaVars = pkgs.runCommandLocal "catppuccin-mocha-vars" { } ''
            ${pkgs.ed}/bin/ed '${catppuccinMochaSrc}/themes/Catppuccin Mocha.theme' <<- EOF
              v/^fish_/d
              %s/^/set -g /
              w $out
            EOF
          '';
        in ''
          set -g fish_greeting

          ${lib.optionalString config.my.graphical.wayland.foot.enable ''
            if [ "$TERM" = "foot" ]
              function mark_prompt_start --on-event fish_prompt
                echo -en "\e]133;A\e\\"
              end
            end
          ''}
        '';
        plugins = with pkgs.fishPlugins; [
          {
            name = "autopair";
            src = autopair.src;
          }
          {
            name = "sponge";
            src = sponge.src;
          }
        ];
      };
    })

    (lib.mkIf cfg.globalNpmPackages.enable {
      home = let
        npmPackagesPath = "${config.home.homeDirectory}/.npm-packages";
      in {
        file.".npmrc".text = ''
          prefix = ${npmPackagesPath}
        '';
        sessionPath = [
          "${npmPackagesPath}/bin"
        ];
        sessionVariables.NODE_PATH = "${npmPackagesPath}/lib/node_modules";
      };
    })

    (lib.mkIf cfg.zsh.enable {
      programs.zsh = {
        enable = true;
        dotDir = ".config/zsh";
        autocd = true;
        localVariables = {
          ZVM_VI_HIGHLIGHT_FOREGROUND = "black";
          ZVM_VI_HIGHLIGHT_BACKGROUND = "red";
        };
        initExtraFirst = ''
          if [[ -n $INSIDE_EMACS ]]; then
            bindkey -e
          else
            AUTOPAIR_INHIBIT_INIT=true
          fi
        '';
        initExtra = ''
          zstyle ':completion:*' menu select
          zstyle ':completion:*' matcher-list ''\'' \
            'm:{[:lower:][:upper:]}={[:upper:][:lower:]}' \
            '+r:|[._-]=* r:|=*' \
            '+l:|=* r:|=*'
          zmodload zsh/complist
          bindkey -M menuselect "''${terminfo[kcbt]}" reverse-menu-complete
          if [[ -n $INSIDE_EMACS ]]; then
            autoload -Uz add-zsh-hook
            add-zsh-hook -d precmd zvm_init
            bindkey "^W" vi-backward-kill-word
          else
            zle -N autopair-insert
            zle -N autopair-close
            zle -N autopair-delete

            zvm_after_init_commands+=('
              for p in ''${(@k)AUTOPAIR_PAIRS}; do
                zvm_bindkey viins "$p" autopair-insert
                bindkey -M isearch "$p" self-insert

                local rchar="$(_ap-get-pair $p)"
                if [[ $p != $rchar ]]; then
                    zvm_bindkey viins "$rchar" autopair-close
                    bindkey -M isearch "$rchar" self-insert
                fi
              done

              zvm_bindkey viins "''${terminfo[kbs]}" autopair-delete
              zvm_bindkey viins "^h" autopair-delete
              bindkey -M isearch "''${terminfo[kbs]}" backward-delete-char
              bindkey -M isearch "^h" backward-delete-char
              zvm_bindkey viins "^[''${terminfo[kbs]}" vi-backward-kill-word
              zvm_bindkey viins "^W" vi-backward-kill-word
            ')
          fi
        '';
        plugins = with pkgs; [
          {
            name = "zsh-autopair";
            src = zsh-autopair;
            file = "share/zsh/zsh-autopair/autopair.zsh";
          }
          {
            name = "zsh-vi-mode";
            src = zsh-vi-mode;
            file = "share/zsh-vi-mode/zsh-vi-mode.plugin.zsh";
          }
          {
            name = "fast-syntax-highlighting";
            src = zsh-fast-syntax-highlighting;
            file = "share/zsh/site-functions/fast-syntax-highlighting.plugin.zsh";
          }
        ];
      };
    })

    (lib.mkIf cfg.starship.enable {
      programs.starship = {
        enable = true;
        settings = {
          add_newline = false;
          continuation_prompt = "[>](bold green) ";
          character = {
            disabled = false;
            success_symbol = "[%](bold green)";
            error_symbol = "[%](bold yellow)";
            vimcmd_symbol = "[%](bold blue)";
          };
          status = {
            disabled = false;
            format = "[$symbol$common_meaning$status]($style) ";
            symbol = "";
            map_symbol = true;
            not_executable_symbol = "+x ";
            not_found_symbol = "?? ";
            sigint_symbol = "XX ";
            signal_symbol = "!! ";
            pipestatus = true;
            pipestatus_separator = "[|](bold yellow) ";
            pipestatus_format = "[\\[](bold yellow) $pipestatus"
                                + "[\\] => ](bold yellow)"
                                + "[$symbol$common_meaning$status]($style) ";
          };
          directory = {
            style = "bold blue";
            truncation_length = 6;
            truncate_to_repo = false;
            truncation_symbol = ".../";
          };
          nix_shell.format = "[$state( \\($name\\))]($style) ";
          git_branch.format = "[$symbol$branch(:$remote_branch)]($style) ";
          line_break.disabled = true;
          c.disabled = true;
          cmake.disabled = true;
          golang.disabled = true;
          haskell.disabled = true;
          java.disabled = false; # the only marginally useful one
          nodejs.disabled = true;
          lua.disabled = true;
          nim.disabled = true;
          package.disabled = true;
          perl.disabled = true;
          php.disabled = true;
          python.disabled = true;
          ruby.disabled = true;
          rust.disabled = true;
        };
      };
    })
  ];
}