# Enable Powerlevel10k instant prompt. Should stay close to the top of ~/.zshrc.
# Initialization code that may require console input (password prompts, [y/n]
# confirmations, etc.) must go above this block; everything else may go below.
if [[ -r "${XDG_CACHE_HOME:-$HOME/.cache}/p10k-instant-prompt-${(%):-%n}.zsh" ]]; then
  source "${XDG_CACHE_HOME:-$HOME/.cache}/p10k-instant-prompt-${(%):-%n}.zsh"
fi

# Set PATH
export PATH="/usr/local:$PATH"

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

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

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

fpath=("$HOME/.zsh/zsh-completions/src" $fpath)

# Editor
export EDITOR='nvim'
export SUDO_EDITOR='nvim'
export VISUAL='nvim'

export ZSH="$HOME/.oh-my-zsh"
export GPG_TTY=$(tty)
export BAT_THEME="Nord"

export FZF_DEFAULT_COMMAND="fd --hidden --strip-cwd-prefix --exclude .git"
export FZF_CTRL_T_COMMAND="$FZF_DEFAULT_COMMAND"
export FZF_ALT_C_COMMAND="fd --type=d --hidden --strip-cwd-prefix --exclude .git"
export FZF_CTRL_T_OPTS="--preview 'bat -n --color=always --line-range :500 {}'"
export FZF_ALT_C_OPTS="--preview 'eza --tree --color=always {} | head -200'"

zstyle ':omz:plugins:alias-finder' autoload yes
zstyle ':omz:plugins:alias-finder' exact yes
zstyle ':omz:plugins:ssh-agent' agent-forwarding yes
zstyle ':omz:plugins:ssh-agent' lifetime 1h
zstyle ':omz:plugins:ssh-agent' quiet yes

# Only export SSH_AUTH_SOCK if agent is running
if [ -S "$SSH_AUTH_SOCK" ]; then
  export SSH_AUTH_SOCK
fi

# DISABLE_AUTO_TITLE="true"
ENABLE_CORRECTION="true"

# Command History
HIST_STAMPS="yyyy-mm-dd"
HISTSIZE=128000
SAVEHIST=128000
setopt hist_verify
setopt share_history
setopt hist_ignore_all_dups
setopt pushd_ignore_dups

export FZF_DEFAULT_OPTS=$FZF_DEFAULT_OPTS'
    --color=fg:#e5e9f0,bg:#3b4252,hl:#81a1c1
    --color=fg+:#e5e9f0,bg+:#3b4252,hl+:#81a1c1
    --color=info:#eacb8a,prompt:#bf6069,pointer:#b48dac
    --color=marker:#a3be8b,spinner:#b48dac,header:#a3be8b'

zvm_after_init_commands+=(init_env)

function zvm_config() {
   export ZVM_INIT_MODE='sourcing'
   export ZVM_LINE_INIT_MODE=$ZVM_MODE_INSERT
   export ZVM_VI_EDITOR="$VISUAL"
   export ZVM_VI_SURROUND_BINDKEY='classic'
   export ZVM_VI_INSERT_ESCAPE_BINDKEY='jk'
   export ZVM_ESCAPE_KEYTIMEOUT=0.03
}

plugins=(
   alias-finder
   aliases
   colored-man-pages
   genpass
   ssh
   ssh-agent
   sudo
   web-search
)

# set up dircolors
case $(uname -s) in

   Darwin)
      plugins+=(
         brew
         copyfile
         copypath
         macos
      )
      
      zstyle ':omz:plugins:ssh-agent' ssh-add-args --apple-load-keychain

      eval "$(/opt/homebrew/bin/brew shellenv)"

      if command -v gdircolors >/dev/null 2>&1; then
         test -f "$HOME/.dircolors" && eval "$(gdircolors "$HOME/.dircolors")" || eval "$(gdircolors)"
      fi
      ;;

   Linux)
      plugins+=(
         archlinux
         command-not-found
         systemd
         ubuntu
      )

      test -f "$HOME/.dircolors" && eval "$(dircolors -b "$HOME/.dircolors")" || eval "$(dircolors -b)"
      ;;

   *) ;;
esac

function init_env() {
   eval "$(zoxide init zsh)"
   eval "$(fzf --zsh)"
   
   # To customize prompt, run `p10k configure` or edit ~/.p10k.zsh.
   [[ ! -f "$HOME/.p10k.zsh" ]] || source "$HOME/.zsh/powerlevel10k/powerlevel10k.zsh-theme" && source "$HOME/.p10k.zsh"
   
   # Load aliases
   [[ ! -f "$HOME/.aliases" ]] || source "$HOME/.aliases"

   # temporarily used to load nvm on my desktop
   [[ ! -f "$HOME/.zinit.$(hostname)" ]] || source "$HOME/.zinit.$(hostname)"
}

[[ ! -f "$ZSH/oh-my-zsh.sh" ]] || source "$ZSH/oh-my-zsh.sh"

function zvm_after_init() {
   autoload -Uz add-zle-hook-widget
   # add-zle-hook-widget zle-line-pre-redraw zvm_zle-line-pre-redraw
}

_fzf_compgen_path() {
   fd --hidden --exclude .git . "$1"
}

_fzf_compgen_dir() {
   fd --type=d --hidden --exclude .git . "$1"
}

_fzf_comprun() {
   local command=$1
   shift

   case "$command" in
      cd)
         fzf --preview 'eza --tree --color=always {} | head -200' "$@"
         ;;
      export|unset)
         fzf --preview "eval 'echo \$' {}" "$@"
         ;;
      ssh)
         fzf --preview 'dig {}' "$@"
         ;;
      *)
         fzf --preview "--preview 'bat -n --color=always --line-range :500'" "$@"
         ;;
   esac
}

[[ ! -f "$HOME/.zsh/zsh-syntax-highlighting/zsh-syntax-highlighting.zsh" ]] || source "$HOME/.zsh/zsh-syntax-highlighting/zsh-syntax-highlighting.zsh"
[[ ! -f "$HOME/.zsh/zsh-autosuggestions/zsh-autosuggestions.zsh" ]] || source "$HOME/.zsh/zsh-autosuggestions/zsh-autosuggestions.zsh"
[[ ! -f "$HOME/.zsh/zsh-vi-mode/zsh-vi-mode.plugin.zsh" ]] || source "$HOME/.zsh/zsh-vi-mode/zsh-vi-mode.plugin.zsh"