shed/modules/home/environment/zsh/extraconfig.nix

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

let
  shellsound = "${pkgs.myScripts.playshellsound}/bin/playshellsound";
  color-commit = "${pkgs.myScripts.color-commit}/bin/color-commit";
  fzf-tab = "${pkgs.myPkgs.fzf-tab}";
  sndpath = "${self}/assets/sound";
in
{
  options = {
    movOpts.envConfig.zshConfig.extraConfig.enable = lib.mkEnableOption "enables my extra shell configurations";
  };
  config = lib.mkIf config.movOpts.envConfig.zshConfig.extraConfig.enable {
    programs.zsh = {
      initContent = ''
        build-drv() { # Put the derivation path in $RESULT instead of making a 'result' symlink
          RESULT=$(nix-build "$@" --no-link)
          if [ -z "$RESULT" ]; then
            return 1
          fi
          export RESULT
          echo "\$RESULT = $RESULT"
        }
        escaped() {
          # escapes regex chars
          sed 's/[^^]/[&]/g; s/\^/\\^/g' <<<"$1"
        }
        nix-shell() {
          NIX_SHELL=true command nix-shell "$@" --run zsh
        }
        nsp() {
          NIX_SHELL=true command nix-shell -p "$@" --run zsh
        }
        viflake() {
          (
            while ! [ -f ./flake.nix ]; do
              builtin cd ..
              if [ "$PWD" = "/" ]; then
                echo "No flake.nix found in this directory, or any parent directories."
                return 1
              fi
            done
            nvim ./flake.nix
          )
        }
        nvim() {
          ${shellsound} ${sndpath}/nvim.wav
          command nvim "$@"
        }
        neovide() {
          ${shellsound} ${sndpath}/nvim.wav
          command neovide "$@"
        }
        alias vi="nvim"
        grimblast() {
          if command grimblast "$@"; then
            ${shellsound} ${sndpath}/screenshot.wav
          fi
        }
        gitcheckout_sfx() {
          if git checkout "$@"; then
            ${shellsound} ${sndpath}/gitcheckout.wav
            return 0
          else
            ${shellsound} ${sndpath}/error.wav
            return 1
          fi
        }
        gitrebase_sfx() {
          if git rebase "$@"; then
            ${shellsound} ${sndpath}/gitrebase.wav
            return 0
          else
            ${shellsound} ${sndpath}/error.wav
            return 1
          fi
        }
        gitcommit_sfx() {
          output=$(git commit "$@")
          if [ -n "$output" ]; then
            ${shellsound} ${sndpath}/gitcommit.wav
            echo "$output" | ${color-commit}
            return 0
          else
            ${shellsound} ${sndpath}/error.wav
            echo "$output"
            return 1
          fi
        }
        gitpush_sfx() {
          if git push "$@"; then
            ${shellsound} ${sndpath}/gitpush.wav
            return 0
          else
            ${shellsound} ${sndpath}/error.wav
            return 1
          fi
        }
        gitpull_sfx() {
          if git pull "$@"; then
            ${shellsound} ${sndpath}/gitpull.wav
            return 0
          else
            ${shellsound} ${sndpath}/error.wav
            return 1
          fi
        }
        unalias ls
        ls() {
          eza -1 --group-directories-first --icons "$@"
          ${shellsound} ${sndpath}/ls.wav
          return 0
        }

        mkcd() {
          mkdir -p "$1" && cd "$1"
        }

        y() {
          local tmp="$(mktemp -t "yazi-cwd.XXXXXX")"
          yazi "$@" --cwd-file="$tmp"
          if cwd="$(command cat -- "$tmp")" && [ -n "$cwd" ] && [ "$cwd" != "$PWD" ]; then
            builtin cd -- "$cwd"
          fi
          rm -f -- "$tmp"
        }

        cd() {
          local prev_sounds_enabled="$SOUNDS_ENABLED"
          SOUNDS_ENABLED=0
          eza -1 --group-directories-first --icons "$@"
          SOUNDS_ENABLED=$prev_sounds_enabled
          builtin cd "$@"
          ${shellsound} ${sndpath}/cd.wav
          return 0
        }

        if [ ! -e $HOME/.zsh_history ]; then
          touch $HOME/.zsh_history
          chmod 600 $HOME/.zsh_history
        fi
        if [ "$TERM" = "linux" ]; then
          echo -en "\e]P0101010"
          setfont ter-v32b
        fi
        setopt APPEND_HISTORY     # Append history to the history file (don't overwrite)
        setopt INC_APPEND_HISTORY # Append to the history file incrementally
        setopt SHARE_HISTORY      # Share history between all zsh sessions

        setopt CORRECT
        setopt NO_NOMATCH
        setopt LIST_PACKED
        setopt ALWAYS_TO_END
        setopt GLOB_COMPLETE
        setopt COMPLETE_ALIASES
        setopt COMPLETE_IN_WORD
        setopt AUTO_CD
        setopt AUTO_CONTINUE
        setopt LONG_LIST_JOBS
        setopt HIST_VERIFY
        setopt SHARE_HISTORY
        setopt HIST_IGNORE_SPACE
        setopt HIST_SAVE_NO_DUPS
        setopt HIST_IGNORE_ALL_DUPS
        setopt EXTENDED_GLOB
        setopt TRANSIENT_RPROMPT
        setopt INTERACTIVE_COMMENTS

        autoload -U compinit     # completion
        autoload -U terminfo     # terminfo keys
        zmodload -i zsh/complist # menu completion
        autoload -U promptinit   # prompt

        autoload -U up-line-or-beginning-search; zle -N up-line-or-beginning-search
        autoload -U down-line-or-beginning-search; zle -N down-line-or-beginning-search

        compinit
        source ${fzf-tab}/fzf-tab.plugin.zsh

        bindkey -v
        type starship_zle-keymap-select >/dev/null || \
        {
          eval "$(starship init zsh)"
        }
        ${shellsound} ${sndpath}/sh-source.wav
        [ ! -f $FLAKEPATH/flake.nix ] && echo "WARNING: flake.nix not found at \$FLAKEPATH. Shell aliases for editing config files won't work correctly!" && echo "Edit the FLAKEPATH session variable in zsh/env.nix to point to the path where you saved the system configuration flake."
      '';
    };
  };
}