zshrc

# Shell options
# Init prompt and colours
autoload -U compinit promptinit
autoload -U colors && colors
compinit
promptinit

setopt AUTO_PUSHD
setopt SHARE_HISTORY
setopt completealiases
setopt extendedglob

unsetopt nomatch
unsetopt correct
unsetopt correct_all

zstyle ':completion:*' matcher-list '' 'm:{a-zA-Z}={A-Za-z}' 'r:|[._-]=* r:|=*' 'l:|=* r:|=*'
zstyle ':completion:*' menu select
HISTFILE=~/.zsh_history
HISTSIZE=10000000
SAVEHIST=10000

# Path
# Global paths: end of $PATH
if [ -d "/opt/homebrew/bin" ]; then
    # Special case - homebrew should be first, after nix
    # FIXME: this super doesn't work
    PATH="/opt/homebrew/bin:$PATH"
fi

if [ -e "$HOME/.nix-profile/etc/profile.d/nix.sh" ]; then
    # Nix single-user
    source "$HOME/.nix-profile/etc/profile.d/nix.sh"
fi

if [ -e '/nix/var/nix/profiles/default/etc/profile.d/nix-daemon.sh' ]; then
    # Nix multi-user
    . '/nix/var/nix/profiles/default/etc/profile.d/nix-daemon.sh'
fi
if [ -d "/ubin" ]; then
    PATH="$PATH:/ubin"
fi

if [ -d "/opt/cuda/bin" ]; then
    PATH="$PATH:/opt/cuda/bin"
fi

if [ -d "/sbin" ] && [ ! -L "/sbin" ]; then
    PATH="$PATH:/sbin"
fi

if [ -d "/usr/sbin" ] && [ ! -L "/sbin" ]; then
    PATH="$PATH:/usr/sbin"
fi

# Local paths - beginning of $PATH
if [ -d "$HOME/.gem" ]; then
    for i in $HOME/.gem/ruby/*; do
        PATH="$i/bin:$PATH"
    done
fi

if [ -d "$HOME/.yarn/bin" ]; then
    PATH="$HOME/.yarn/bin:$PATH"
fi

if [ -d "$HOME/.config/yarn/global/node_modules/.bin" ]; then
    PATH="$HOME/.config/yarn/global/node_modules/.bin:$PATH"
fi

if [ -d "$HOME/.cargo/bin" ]; then
    PATH="$HOME/.cargo/bin:$PATH"
fi

if [ -d "$HOME/.local/bin" ]; then
    PATH="$HOME/.local/bin:$PATH"
fi

if [ -d "$HOME/.bin" ]; then
    PATH="$HOME/.bin:$PATH"
fi

if [ -d "$HOME/bin" ]; then
    PATH="$HOME/bin:$PATH"
fi

# Prompt
if type starship >/dev/null; then
    eval "$(starship init zsh)"
    RPROMPT=""
else
    prompt walters
fi

# Aliases
source ~/.zsh/aliases.zsh

# Functions
source ~/.zsh/functions.zsh

# Environment
export GPG_TTY=$(tty)
export MAKEFLAGS="-j$(nproc)"
export NODE_OPTIONS="--max_old_space_size=16384"
export RUST_BACKTRACE=1
export XDG_DATA_DIRS=$HOME/.nix-profile/share:$HOME/.share:"${XDG_DATA_DIRS:-/usr/local/share/:/usr/share/}"

if [[ $(uname) == "Darwin" ]]; then
    if [ -f "/etc/static/zshrc" ]; then
        source "/etc/static/zshrc"
    fi
fi

# Do not set GIT_PAGER if diff.external is set
if type git >/dev/null && [[ $(git config diff.external) == "" ]]; then
    if type bat >/dev/null && type delta >/dev/null; then
        export GIT_PAGER='delta | bat --style=numbers,header'
    fi
fi

if type rg >/dev/null; then
    export FZF_DEFAULT_COMMAND='rg --files'
fi

if type clang >/dev/null; then
    export CC=clang
    export CXX=clang++
fi

if [ -f "$HOME/.config/python/startup.py" ]; then
    export PYTHONSTARTUP="$HOME/.config/python/startup.py"
fi

if type bat >/dev/null; then
    export MANPAGER="sh -c 'col -bx | bat -l man -p'"
fi

if [ -d "/etc/nix/path" ]; then
    export NIX_PATH="/etc/nix/path"
fi

# esc-esc sudo
sudo-command-line() {
    [[ -z $BUFFER ]] && zle up-history
    if [[ $BUFFER == sudo\ * ]]; then
        LBUFFER="${LBUFFER#sudo }"
    elif [[ $BUFFER == $EDITOR\ * ]]; then
        LBUFFER="${LBUFFER#$EDITOR }"
        LBUFFER="sudoedit $LBUFFER"
    elif [[ $BUFFER == vim\ * ]]; then
        LBUFFER="${LBUFFER#vim }"
        LBUFFER="sudoedit $LBUFFER"
    elif [[ $BUFFER == sudoedit\ * ]]; then
        LBUFFER="${LBUFFER#sudoedit }"
        LBUFFER="$EDITOR $LBUFFER"
    else
        LBUFFER="sudo $LBUFFER"
    fi
}

zle -N sudo-command-line
bindkey "\e\e" sudo-command-line
bindkey -M vicmd '\e\e' sudo-command-line

if [ -d /etc/zsh/zshrc.d ]; then
    for file in /etc/zsh/zshrc.d/*; do
        source $file
    done
fi

eval $(ssh-agent) >/dev/null

# Command-not-found
if [ -f ~/.zsh/command-not-found.zsh ]; then
    source ~/.zsh/command-not-found.zsh
fi

# Plugins
if [ -f ~/.zsh/plugins/git.plugin.zsh ]; then
    source ~/.zsh/plugins/git.plugin.zsh
else
    echo "git plugin not loaded"
fi

if [ -f ~/.zsh/plugins/zsh-syntax-highlighting/zsh-syntax-highlighting.zsh ]; then
    # zsh-syntax-highlighting colours
    typeset -A ZSH_HIGHLIGHT_STYLES
    ZSH_HIGHLIGHT_STYLES[unknown-token]=fg=#E06C75,bold
    ZSH_HIGHLIGHT_STYLES[reserved-word]=fg=#EFC07B
    ZSH_HIGHLIGHT_STYLES[suffix-alias]=fg=#98C379,underline
    ZSH_HIGHLIGHT_STYLES[global-alias]=fg=#56B6C2
    ZSH_HIGHLIGHT_STYLES[precommand]=fg=#98C379,underline
    ZSH_HIGHLIGHT_STYLES[autodirectory]=fg=#98C379,underline
    ZSH_HIGHLIGHT_STYLES[globbing]=fg=#61AFEF
    ZSH_HIGHLIGHT_STYLES[history-expansion]=fg=#61AFEF
    ZSH_HIGHLIGHT_STYLES[command-substitution-delimiter]=fg=#C678DD
    ZSH_HIGHLIGHT_STYLES[process-substitution-delimiter]=fg=#C678DD
    ZSH_HIGHLIGHT_STYLES[back-quoted-argument-delimiter]=fg=#C678DD
    ZSH_HIGHLIGHT_STYLES[single-quoted-argument]=fg=#EFC07B
    ZSH_HIGHLIGHT_STYLES[double-quoted-argument]=fg=#EFC07B
    ZSH_HIGHLIGHT_STYLES[dollar-quoted-argument]=fg=#EFC07B
    ZSH_HIGHLIGHT_STYLES[rc-quote]=fg=#56B6C2
    ZSH_HIGHLIGHT_STYLES[dollar-double-quoted-argument]=fg=#56B6C2
    ZSH_HIGHLIGHT_STYLES[back-double-quoted-argument]=fg=#56B6C2
    ZSH_HIGHLIGHT_STYLES[back-dollar-quoted-argument]=fg=#56B6C2
    ZSH_HIGHLIGHT_STYLES[redirection]=fg=#EFC07B
    ZSH_HIGHLIGHT_STYLES[comment]=fg=#282C34,bold
    ZSH_HIGHLIGHT_STYLES[arg0]=fg=#98C379
    source ~/.zsh/plugins/zsh-syntax-highlighting/zsh-syntax-highlighting.zsh
else
    echo "zsh-syntax-highlighting plugin not loaded"
fi

if [ -f ~/.zsh/plugins/zsh-autosuggestions/zsh-autosuggestions.zsh ]; then
    source ~/.zsh/plugins/zsh-autosuggestions/zsh-autosuggestions.zsh
else
    echo "zsh-autosuggestions plugin not loaded"
fi

if grep -Fxq "arch" /etc/os-release 2>/dev/null; then
    if [ -f /usr/share/doc/pkgfile/command-not-found.zsh ]; then
        source /usr/share/doc/pkgfile/command-not-found.zsh
    else
        echo "pkgfile plugin not loaded"
    fi
fi

if [ -f ~/.zsh/plugins/zsh-vi-mode/zsh-vi-mode.plugin.zsh ]; then
    source ~/.zsh/plugins/zsh-vi-mode/zsh-vi-mode.plugin.zsh
    bindkey -r '^R'
    # KEYTIMEOUT=20
    # if type starship >/dev/null; then
        # RPROMPT=""
    # fi
else
    echo "vi-mode plugin not loaded"
fi

if [ -f ~/.zsh/plugins/zsh-nix-shell/nix-shell.plugin.zsh ]; then
    source ~/.zsh/plugins/zsh-nix-shell/nix-shell.plugin.zsh
else
    echo "zsh-nix-shell plugin not loaded"
fi

if [ -f ~/.zsh/plugins/globalias.plugin.zsh ]; then
    source ~/.zsh/plugins/globalias.plugin.zsh
    GLOBALIAS_FILTER_VALUES=(
        ls
        lsa
        lsat
        lstg
        vim

        cp
        df
        du
        e
        feh
        ffmpeg
        ffplay
        ffprobe
        less
        mkdir
        mv
        rg
        rm
        rsync
        sudo
    )
else
    echo "globalias plugin not loaded"
fi

# Largely taken from
# https://github.com/ellie/atuin/blob/aa556fa0883d1a5c8f35960c9d6e8966e4307896/src/shell/atuin.zsh
if type atuin >/dev/null; then
    autoload -U add-zsh-hook

    export ATUIN_SESSION=$(atuin uuid)
    export ATUIN_HISTORY="atuin history list"
    export ATUIN_BINDKEYS="true"

    _atuin_preexec() {
        id=$(atuin history start -- "$1")
        export ATUIN_HISTORY_ID="$id"
    }

    _atuin_precmd() {
        local EXIT="$?"

        [[ -z "${ATUIN_HISTORY_ID}" ]] && return

        (RUST_LOG=error atuin history end --exit $EXIT -- $ATUIN_HISTORY_ID &) >/dev/null 2>&1
    }

    _atuin_search() {
        emulate -L zsh
        zle -I

        # Switch to cursor mode, then back to application
        echoti rmkx
        # swap stderr and stdout, so that the tui stuff works
        # TODO: not this
        output=$(RUST_LOG=error atuin search -i -- $BUFFER 3>&1 1>&2 2>&3)
        echoti smkx

        if [[ -n $output ]]; then
            LBUFFER=$output
        fi

        zle reset-prompt
    }

    add-zsh-hook preexec _atuin_preexec
    add-zsh-hook precmd _atuin_precmd

    zle -N _atuin_search_widget _atuin_search

    if [[ $ATUIN_BINDKEYS == "true" ]]; then
        bindkey '^r' _atuin_search_widget
    fi
fi

function zvm_after_init() {
  zvm_bindkey viins '^R' _atuin_search_widget
  zvm_bindkey vicmd '^R' _atuin_search_widget
}

zvm_after_init_commands+=("bindkey '^[[A' up-line-or-search" "bindkey '^[[B' down-line-or-search")

alias luamake=/home/jamie/Git/dotfiles/nvim/lua-language-server/3rd/luamake/luamake