{ config, lib, ... }:

{
  options.zsh.enable = lib.mkEnableOption "Configure zsh";
  config = lib.mkIf config.zsh.enable {
    programs.zsh = {
      enable = true;
      enableCompletion = true;
      autosuggestion.enable = true;
      syntaxHighlighting.enable = true;

      initContent = ''
        # view man pages with nvim
        export MANPAGER="nvim +Man!"

        # Directory completion with trailing slash
        zstyle ':completion:*' list-dirs-first true
        zstyle ':completion:*' special-dirs true
        zstyle ':completion:*' squeeze-slashes true
        zstyle ':completion:*' add-space false

        # Case-insensitive completion
        zstyle ':completion:*' matcher-list 'm:{a-z}={A-Z}'

        # vim keybindings
        bindkey -v

        # imagemagick wrapper function to remove background
        remove_bg() {
          if [[ $# -lt 2 ]]; then
            echo 'Usage: remove_bg <input_file> <fuzz_percentage> [transparent_color] [output_file]'
            return 1
          fi

          local input_file=$1
          local fuzz=$2
          local transparent_color=$3
          local output_file=$4

					magick "$input_file" \
						-fuzz $fuzz -transparent "$transparent_color" \
						-alpha on -channel A -blur 0x2 -level 5%,100% +channel \
						"$output_file"

          echo "Saved transparent image to: $output_file"
        }

        eval "$(zoxide init zsh)"
      '';

      shellAliases = {
        ls = "ls --color=auto";
				nixceck = "nix flake check --all-systems";
      };

      zplug = {
        enable = true;
        plugins = [
          { name = "mafredri/zsh-async"; }
          { name = "sindresorhus/pure"; tags = [ "as:theme" "use:pure.zsh" ]; }
          { name = "zdharma-continuum/fast-syntax-highlighting"; }
          { name = "zsh-users/zsh-autosuggestions"; }
        ];
      };
    };
  };
}