dotfiles/.zshrc

# Install tmux plugin manager and tmux plugins if not installed yet.
if [[ -n $TMUX ]]; then 
    TMUX_PLUG_DIR="$HOME/.tmux/plugins/tpm/"
    [[ -d $TMUX_PLUG_DIR ]] || \
        (mkdir -p $TMUX_PLUG_DIR && \
        git clone https://github.com/tmux-plugins/tpm $TMUX_PLUG_DIR)

    if [[ $(ls $TMUX_PLUG_DIR | wc -l) -lt 2 ]]; then
        $TMUX_PLUG_DIR/bin/install_plugins
    fi
fi

# Install oh-my-zsh if not installed yet
if [[ -z $(ls $HOME/.oh-my-zsh) ]]; then
    sh -c "$(curl -fsSL https://raw.githubusercontent.com/ohmyzsh/ohmyzsh/master/tools/install.sh)"
    mv $HOME/.zshrc.pre-oh-my-zsh $HOME/.zshrc

    # Install command-time if not installed yet
    [[ -d $HOME/.oh-my-zsh/custom/plugins/command-time ]] || 
        git clone https://github.com/popstas/zsh-command-time $HOME/.oh-my-zsh/custom/plugins/command-time
fi

# Install vim-plug for nvim (vim version is automagically symlinked)
NVIMPLUG="$HOME/.local/share/nvim/site/autoload/plug.vim"
[[ -f $NVIMPLUG ]] || \
    curl -fLo $NVIMPLUG --create-dirs https://raw.githubusercontent.com/junegunn/vim-plug/master/plug.vim && \

# If you come from bash you might have to change your $PATH.
export PATH=$HOME/bin:/usr/local/bin:/snap/bin:$PATH

# Path to your oh-my-zsh installation.
export ZSH=$HOME/.oh-my-zsh

# Set name of the theme to load. Optionally, if you set this to "random"
# it'll load a random theme each time that oh-my-zsh is loaded.
# See https://github.com/robbyrussell/oh-my-zsh/wiki/Themes
ZSH_THEME="agnoster"

# Uncomment the following line to use case-sensitive completion.
CASE_SENSITIVE="false"

# Uncomment the following line to use hyphen-insensitive completion. Case
# sensitive completion must be off. _ and - will be interchangeable.
HYPHEN_INSENSITIVE="true"

# Uncomment the following line to disable bi-weekly auto-update checks.
DISABLE_AUTO_UPDATE="true"

# Uncomment the following line to change how often to auto-update (in days).
# export UPDATE_ZSH_DAYS=13

# Uncomment the following line to disable colors in ls.
DISABLE_LS_COLORS="false"

# Uncomment the following line to disable auto-setting terminal title.
DISABLE_AUTO_TITLE="false"

# Uncomment the following line to enable command auto-correction.
ENABLE_CORRECTION="true"

# Uncomment the following line to display red dots whilst waiting for completion.
# COMPLETION_WAITING_DOTS="true"

# Uncomment the following line if you want to disable marking untracked files
# under VCS as dirty. This makes repository status check for large repositories
# much, much faster.
# DISABLE_UNTRACKED_FILES_DIRTY="true"

# Uncomment the following line if you want to change the command execution time
# stamp shown in the history command output.
# The optional three formats: "mm/dd/yyyy"|"dd.mm.yyyy"|"yyyy-mm-dd"
HIST_STAMPS="yyyy-mm-dd"

# Would you like to use another custom folder than $ZSH/custom?
# ZSH_CUSTOM=/path/to/new-custom-folder

# Which plugins would you like to load? (plugins can be found in ~/.oh-my-zsh/plugins/*)
# Custom plugins may be added to ~/.oh-my-zsh/custom/plugins/
# Example format: plugins=(rails git textmate ruby lighthouse)
# Add wisely, as too many plugins slow down shell startup.
plugins=(
   git
   common-aliases
   gpg-agent
 #  gradle
 #  python
   tmux
   ssh-agent
   sudo
   systemd
   wd
#   archlinux
   command-time
   shrink-path
  
)

source $ZSH/oh-my-zsh.sh

# Other oh-my-zsh plugins
ZSH_AUTOSUGGESTIONS="/usr/share/zsh/plugins/zsh-autosuggestions/zsh-autosuggestions.zsh"
[[ -f $ZSH_AUTOSUGGESTIONS ]] && source $ZSH_AUTOSUGGESTIONS

ZSH_SYNTAX_HIGHLIGHTING="/usr/share/zsh/plugins/zsh-syntax-highlighting/zsh-syntax-highlighting.zsh"
[[ -f $ZSH_SYNTAX_HIGHLIGHTING ]] && source $ZSH_SYNTAX_HIGHLIGHTING

# User configuration

# Fix ZSH keys; stolen from https://github.com/m-ou-se/config
# typeset -A key

# bindkey -e
# bindkey '^[[Z' reverse-menu-complete
# bindkey '^[[1;5C' emacs-forward-word
# bindkey '^[[1;5D' emacs-backward-word
# bindkey '^[[H' beginning-of-line
# bindkey '^[[F' end-of-line
# bindkey -s '^[Oo' '/'
# bindkey -s '^[Oj' '*'
# bindkey -s '^[Om' '-'
# bindkey -s '^[Ok' '+'

# key[Home]=${terminfo[khome]}
# key[End]=${terminfo[kend]}
# key[Insert]=${terminfo[kich1]}
# key[Delete]=${terminfo[kdch1]}
# key[Up]=${terminfo[kcuu1]}
# key[Down]=${terminfo[kcud1]}
# key[Left]=${terminfo[kcub1]}
# key[Right]=${terminfo[kcuf1]}
# key[PageUp]=${terminfo[kpp]}
# key[PageDown]=${terminfo[knp]}

# [[ -n "${key[Home]}"    ]]  && bindkey  "${key[Home]}"    beginning-of-line
# [[ -n "${key[End]}"     ]]  && bindkey  "${key[End]}"     end-of-line
# [[ -n "${key[Insert]}"  ]]  && bindkey  "${key[Insert]}"  overwrite-mode
# [[ -n "${key[Delete]}"  ]]  && bindkey  "${key[Delete]}"  delete-char
# [[ -n "${key[Up]}"      ]]  && bindkey  "${key[Up]}"      up-line-or-history
# [[ -n "${key[Down]}"    ]]  && bindkey  "${key[Down]}"    down-line-or-history
# [[ -n "${key[Left]}"    ]]  && bindkey  "${key[Left]}"    backward-char
# [[ -n "${key[Right]}"   ]]  && bindkey  "${key[Right]}"   forward-char
#####


export MANPATH="/usr/local/man:$MANPATH"

# You may need to manually set your language environment
export LANG=en_US.UTF-8

# Prompt stuff
# Make sure the path in prompt does not get too long
prompt_dir() { 
    prompt_segment blue black "$(shrink_path -l -t -s)"
}

export DEFAULT_USER='gregory' # for a shortened username

arch() {
    echo "x86_64"
}

prompt_context () {
	if [[ "$USER" != "$DEFAULT_USER" || -n "$SSH_CLIENT" ]]
	then
		prompt_segment black default "%(!.%{%F{yellow}%}.)$USER@%m"
    else 
        # Only display first character of username
        prompt_segment black default "%(!.%{%F{yellow}%}.)${USER:0:1}@%m"
    fi
}

# Preferred editor for local and remote sessions
# if [[ -n $SSH_CONNECTION ]]; then
#   export EDITOR='vim'
# else
#   export EDITOR='mvim'
# fi

# Compilation flags
export ARCHFLAGS="-arch x86_64"

# Set personal aliases, overriding those provided by oh-my-zsh libs,
# plugins, and themes. Aliases can be placed here, though oh-my-zsh
# users are encouraged to define aliases within the ZSH_CUSTOM folder.
# For a full list of active aliases, run `alias`.
#
# Example aliases
alias zshconfig="vim ~/.zshrc"
alias vimconfig="vim ~/.vimrc"
#

# Useful functions
hex() {
    emulate -L zsh
    if [[ -n "$1" ]]; then
        printf "%x\n" $1
    else
        print 'Usage: hex <number-to-convert>'
        return 1
    fi
}

# Show message of the day (if it exists)
[[ -f '/etc/motd' ]] && cat /etc/motd

# wttr.in weather application
weather () {
    if [[ -z $1 ]]; then
        PLACE="Delft"
    else
        PLACE="$1"
    fi
	curl "wttr.in/$PLACE"
}

texdoc () {
    echo "Trying texdoc.net/pkg/$1..."
    xdg-open "http://texdoc.net/pkg/$1"
}

dotfiles_setup() {
    # List of all installed needed for this setup
    echo "Please turn on Community repository if on Arch"

    PKGLIST="\
        alacritty alacritty-terminfo \
        vim vim-spell-en vim-spell-nl \
        tmux \
        neovim python-neovim \
        mosh \
        rustup \
        zsh-syntax-highlighting zsh-autosuggestions \
        firefox powertop \
        exa ripgrep \
        xclip xsel \
        zathura zathura-pdf-mupdf zathura-ps xdotool \
        "

    sudo pacman -S --needed `echo $PKGLIST` 


    # # Install oh-my-zsh
    # echo "Installing oh-my-zsh"
    # OMZSH_DIR="$HOME/.oh-my-zsh" 
    # [[ -d $OMZSH_DIR ]] || \
    #     bash -c "$(curl -fsSL https://raw.githubusercontent.com/robbyrussell/oh-my-zsh/master/tools/install.sh)"

    # echo "Installing plugins for oh-my-zsh.."
    # # Install command-time plugin for oh-my-zsh
    # CMD_TIME_DIR="$HOME/.oh-my-zsh/custom/plugins/command-time"
    # [[ -d $CMD_TIME_DIR ]] || \
    #     git clone https://github.com/popstas/zsh-command-time.git $CMD_TIME_DIR

    # Setup Rust
    echo 'Installing rust stable, nightly'
    rustup install stable nightly
    rustup default stable
    
    echo 'Installing cargo-update'
    cargo install cargo-update 
    
    echo 'Generating completions'
    mkdir -p ~/.zfunc
    rustup completions zsh cargo > $HOME/.zfunc/_cargo

    # Install vim-plug
    # NVIMPLUG="$HOME/.local/share/nvim/site/autoload/plug.vim"
    # [[ -f $NVIMPLUG ]] || \
    #     curl -fLo $NVIMPLUG --create-dirs https://raw.githubusercontent.com/junegunn/vim-plug/master/plug.vim

    # Setup Plug and install plugins
    nvim +PlugInstall +PlugUpdate +qa!

    # Installing aurto
    echo "Installing aurto"
    mkdir -p /tmp/aurutils /tmp/aurto
    ( 
        cd /tmp/aurutils
        curl -L https://aur.archlinux.org/cgit/aur.git/snapshot/aurutils.tar.gz | tar xz
        cd aurutils
        gpg --recv-keys DBE7D3DD8C81D58D0A13D0E76BC26A17B9B7018A
        makepkg -srci
    )
    (
        cd /tmp/aurto
        curl -L https://aur.archlinux.org/cgit/aur.git/snapshot/aurto.tar.gz | tar xz
        cd aurto
        makepkg -srci
    )
    aurto add aurto
    systemctl disable --now update-aurto.timer update-aurto-startup.timer
}


# Upgrade system and all relevant plugins / packages / applications / things
upgrade() {
    # Cache sudo password, such that `pacman` and `aurget` don't ask for one
    # Which may go unnoticed
    # sudo true >/dev/null 2>&1
    if [[ -n $(command -v pacman) ]]; then
        echo 'Upgrading system packages'
        sudo pacman -Syu
    fi

    if [[ -n $(command -v apt) ]]; then
        echo 'Upgrading system packages'
        sudo apt update 
        sudo apt full-upgrade
    fi


    # Update dotfiles
    git --git-dir=$HOME/.dotfiles/ --work-tree=$HOME pull

    if [[ -n $(command -v dconf) ]]; then
        # Export dconf settings to a backup
        DCONF_FILE="$HOME/.dconf.org.dump"
        mv $DCONF_FILE{,-.bak} >/dev/null 2>&1
        dconf dump /org/ > $DCONF_FILE
    fi

    if [[ -n $(command -v nvim) ]]; then
        echo 'Updating vim Plug plugin and Plugged packages'
        nvim +PlugInstall +PlugUpgrade +qa! 
        nvim +PlugUpdate +qa! 
    elif [[ -n $(command -v vim) ]]; then
        echo 'Updating vim Plug plugin and Plugged packages'
        vim +PlugInstall +PlugUpgrade +qa! 
        vim +PlugUpdate +qa! 
    fi

    echo 'Updating tmux plugins'
    $HOME/.tmux/plugins/tpm/bin/update_plugins all

    if [[ -n $(command -v pacman) ]]; then
        echo 'Upgrading system packages'
        sudo pacman -Syu
    fi

    if [[ -n $(command -v apt) ]]; then
        echo 'Upgrading system packages'
        sudo apt update 
        sudo apt full-upgrade
    fi

    if [[ -n $(command -v rustup) ]]; then
        echo 'Updating rustup'
        rustup self upgrade-data

        echo 'Updating rust'
        rustup update 
    fi

    if [[ -n $(command -v cargo-install-update) ]]; then
        echo 'Recompiling all rust binaries'
        cargo install-update -a 
    fi

    if [[ -n $(command -v omz) ]]; then 
        echo 'Updating oh-my-zsh'
        omz update
    fi

    if [[ -n $(command -v paru) ]]; then
        echo 'Updating paru'
        paru -Syu
    fi

    UPDATE_AURTO='/usr/lib/aurto/update-aurto'
    if [[ -f $UPDATE_AURTO ]]; then
        echo 'Upgrading aur packages'
        $UPDATE_AURTO
        sudo pacman -Syu

        update_aurto () {
            /usr/lib/aurto/update-aurto
        }
    fi
}

# Create dir and cd to it
mkcd() { mkdir -p "$@" && cd "$@"; }

# Keychain for caching (gpg and) ssh keys
if [[ -n $(command -v keychain) ]]; then
    eval $(keychain --eval  --quiet id_ed25519 --noask --timeout 10)
fi

# Wegwezen met je stomme beep
# setterm -blength 0 >/dev/null 2>&1

# Command-line fuzzy finder
[ -f ~/.fzf.zsh ] && source ~/.fzf.zsh

############################
# Exports
############################

# Disable telemetry for several tools
export DOTNET_CLI_TELEMETRY_OPTOUT='1'
export GATSBY_TELEMETRY_DISABLED=1
export GAUGE_TELEMETRY_ENABLED=false
export HASURA_GRAPHQL_ENABLE_TELEMETRY=false
export NEXT_TELEMETRY_DISABLED=1
export AUTOMATEDLAB_TELEMETRY_OPTOUT=1
export POWERSHELL_TELEMETRY_OPTOUT=1

export GOPATH=$HOME/go
export GOBIN=$GOPATH/bin
export PATH="\
$HOME/bin:\
$HOME/.local/bin:\
$HOME/opt/bin:\
$HOME/.cargo/bin:\
$GOBIN:\
$HOME/.cabal/bin:\
$HOME/.npm-global/bin:\
$HOME/intelFPGA_lite/18.0/quartus/bin:\
$HOME/esp/xtensa-esp32-elf/bin:\
/opt/vivado/Vivado/2019.1/bin:\
$PATH"

export WORKSPACE="$HOME/Documents/Gregorius/Projects/workspace"
export THIS_ULX3S_DEVICE="LFE5U-85F"
export ULX3S_COM="/dev/ttyUSB0"

if [[ -n $(command -v nvim) ]]; then 
    export EDITOR=nvim
elif [[ -n $(command -v vim) ]]; then  
    export EDITOR=vim
else
    export EDITOR=vi
fi

# LEGACY: May be redundant.
export MOZILLA_FIVE_HOME=/usr/lib/mozilla
# export LD_LIBRARY_PATH=${MOZILLA_FIVE_HOME}:${LD_LIBRARY_PATH}

# License file for Xilinx proprietary toolchain
export XILINXD_LICENSE_FILE=$HOME/Downloads/Xilinx.lic

# ESP IDF development toolchain
_ZSH_IDF_PATH=$HOME/esp/esp-idf
[[ -d $_ZSH_IDF_PATH ]] && export IDF_PATH=$_ZSH_IDF_PATH

# ssh key
_ZSH_SSH_KEY_PATH="$HOME/.ssh/id_ed25519"
[[ -f _ZSH_SSH_KEY_PATH ]] && export SSH_KEY_PATH=_ZSH_SSH_KEY_PATH

# RUSTFLAGS
# export RUSTFLAGS='-C target-cpu=native'

# Python rc file
_ZSH_PYTHONSTARTUP="$HOME/.config/pyrc.py"
[[ -f $_ZSH_PYTHONSTARTUP ]] && export PYTHONSTARTUP=$_ZSH_PYTHONSTARTUP

_ZSH_PYTHONPYCACHEPREFIX="$HOME/.pycache/"
export PYTHONPYCACHEPREFIX=$_ZSH_PYTHONPYCACHEPREFIX
[[ -d $_ZSH_PYTHONPYCACHEPREFIX ]] || mkdir $_ZSH_PYTHONPYCACHEPREFIX -p

############################
## Aliases ##
############################
alias rpi='ssh gregory@192.168.2.42' 
alias chk='ping -c3 -i0.2 1.1' 
alias vpsander='ssh kevin@papegaai.vps.inthout.me'
alias s='ssh kevinhill.nl -p 22002'
alias force_max_brightness='cat /sys/class/backlight/intel_backlight/max_brightness | sudo tee /sys/class/backlight/intel_backlight/brightness'
BATCAPFILE='/sys/class/power_supply/BAT0/capacity'
[[ -f $BATCAPFILE ]] && \
    alias batcap="cat $BATCAPFILE"

[[ -n $(command -v firefox-nightly) ]] && alias firefox='firefox-nightly'

[[ -n $(command -v powertop) ]] && alias ptop='sudo powertop' # Needs sudo rights
alias wttr="weather"

# CPU frequency, only work on ZBook (I hope)
alias savepower="sudo cpupower frequency-set -u 2GHz"
alias maxpower="sudo cpupower frequency-set -u 4.1GHz"
[[ -n $(command -v exa) ]] && alias ls='exa'
[[ -n $(command -v nvim) ]] && alias vim='nvim'
alias das="ssh ihpc1855@fs3.das4.tudelft.nl"
alias t='$HOME/Documents/Gregorius/Uni/CE/Thesis'
alias note='$EDITOR $(date +"%Y-%m-%d.txt")'

alias :q="exit"
alias :wq="exit"
alias :x="exit"

# Typos
alias sl="ls"
[[ -n $(command -v git) ]] && alias gti="git"

alias mn="man"
alias clc="clear"

# If the end of an alias is a space, try to resolve next word as well as an alias
alias sudo="sudo " 

# For my dotfiles repo
alias dotgit='git --git-dir=$HOME/.dotfiles/ --work-tree=$HOME'
alias dg=dotgit

alias d='/mnt/c/Users/gkh99/stack/Documents/Gregorius/werk/Brunel-XDelft/daadwerkelijkwerk/'

# Cargo stuff
if [[ -n $(command -v cargo) ]]; then
    alias b="cargo build"
    alias bb="cargo build --release"
    alias r="cargo run"
    alias rr="cargo run --release"
fi

# gheghe
[[ -n $(command -v python3) ]] && alias pieton='/usr/bin/env python3'
[[ -n $(command -v python3) ]] && alias py='python3'
[[ -n $(command -v python2) ]] && alias py2='python2'

if [[ -n $(command -v intel-virtual-output) ]]; then
    function sii() {
        systemctl suspend
        intel-virtual-output
        intel-virtual-output
    }
fi

# Auto-install new gitea version
if [[ -n $(command -v gitea) ]]; then 
    function get_gitea() {
        VERSION=$1
        echo "Downloading Gitea version $VERSION"
        wget https://github.com/go-gitea/gitea/releases/download/v$VERSION/gitea-$VERSION-linux-amd64 -O /tmp/gitea-bin
        sudo mv /tmp/gitea-bin /usr/local/bin/gitea
        sudo chmod +x /usr/local/bin/gitea
    }
fi

if [[ -n $(command -v hugo) ]]; then
    function get_hugo() {
        VERSION=$1
        echo "Downloading Hugo version $VERSION"
        wget "https://github.com/gohugoio/hugo/releases/download/v$VERSION/hugo_extended_$(echo $VERSION)_Linux-64bit.deb" -O /tmp/hugo.deb
        sudo dpkg -i /tmp/hugo.deb
    }
fi

[[ -f .zshrc.local ]] && source .zshrc.local
pkill -9 tracker-miner-f