# ─< Helper functions >─────────────────────────────────────────────────────────────────
function echo_error() { echo -e "\033[0;1;31merror:\033[0;31m\t${*}\033[0m"; }
function echo_binfo() { echo -e "\033[0;1;34mInfo:\033[0;34m\t${*}\033[0m"; }
function echo_info() { echo -e "\033[0;35m${*}\033[0m"; }

# ─< Check if the given command exists silently >─────────────────────────────────────────
command_exists() {
  command -v "$@" >/dev/null 2>&1
}

# Check if the user is root and set sudo variable if necessary
check_root() {
  if [[ "${EUID}" -ne 0 ]]; then
    if command_exists sudo; then
      echo_binfo "User is not root. Using sudo for privileged operations."
      _sudo="sudo"
    else
      echo_error "No sudo found and you're not root! Can't install packages."
      return 1
    fi
  else
    echo_binfo "Root access confirmed."
    _sudo=""
  fi
}

# Load completions
autoload -Uz compinit && compinit

bindkey -e

setopt appendhistory
setopt sharehistory
setopt hist_ignore_space
setopt hist_ignore_all_dups
setopt hist_save_no_dups
setopt hist_ignore_dups
setopt hist_find_no_dups
autoload -Uz select-word-style
select-word-style shell
# Huge history. Doesn't appear to slow things down, so why not?
HISTSIZE=500000
HISTFILESIZE=100000

HISTFILE=~/.zsh_history
SAVEHIST=$HISTSIZE
HISTDUP=erase
# Completion styling
zstyle :compinstall filename '/home/pika/.zshrc'
zstyle ':completion:*' matcher-list 'm:{a-z}={A-Za-z}'
zstyle ':completion:*' list-colors "${(s.:.)LS_COLORS}"
zstyle ':completion:*' menu no
zstyle ':fzf-tab:complete:cd:*' fzf-preview 'ls --color $realpath'
zstyle ':fzf-tab:complete:__zoxide_z:*' fzf-preview 'ls --color $realpath'
setopt autocd notify
# End of lines configured by zsh-newuser-install

_init (){
  if command_exists oh-my-posh; then
    # eval "$(oh-my-posh init zsh --config 'https://raw.githubusercontent.com/JanDeDobbeleer/oh-my-posh/main/themes/amro.omp.json')"
    eval "$(oh-my-posh init zsh --config 'https://git.k4li.de/dotfiles/oh-my-posh/raw/branch/main/amro.toml')"
    # eval "$(oh-my-posh init zsh --config '~/gitea/dotfiles/oh-my-posh/amro.toml')"
  else
    curl -s https://ohmyposh.dev/install.sh | sudo bash --norc -s -- -d /usr/bin/
  fi
  if command_exists zoxide; then
    eval "$(zoxide init zsh)"
  fi
  local zconf="$HOME/.zsh"
  local zAutosg="$zconf/autosuggestions/zsh-autosuggestions.zsh"
  [[ -f "$zAutosg" ]] &&
    . $zAutosg
}

_alias(){
  alias please="sudo"

  # ─< easier dir up >────────────────────────────────────────────────────────────────────────
  alias ..="cd .."

  # ─< weather >──────────────────────────────────────────────────────────────────────────────
  alias www="curl wttr.in/Ulm"

  # ─< colored ip >───────────────────────────────────────────────────────────────────
  alias ip="ip --color=always"

  # ─< check for rg >─────────────────────────────────────────────────────────────────────────
  if command_exists rg; then
    alias grep="rg --color=always"
  else
    alias grep="grep --color=always"
  fi

  # ─< linutil >────────────────────────────────────────────────────────────────────────────
  alias linutil="curl -fsSL https://christitus.com/linux | sh"

  # ─< telnet (starwars) >────────────────────────────────────────────────────────────────────
  if command_exists telnet; then
    alias starwars="telnet -a telehack.com"
  fi

  [[ -f "$HOME/go/bin/lazygit" ]] &&
    alias lazygit="$HOME/go/bin/lazygit" &&
    alias lg="lazygit"

  # ─< colorized ls >─────────────────────────────────────────────────────────────────────────
  if command_exists exa; then
    alias ls="exa --icons -l --git"
    alias l="exa --long --no-filesize --no-permissions --no-time --git --colour-scale --icons"
    alias ll="exa --all --long --no-filesize --no-permissions --no-time --git --colour-scale --icons"
    alias tree="exa --icons -l --tree"
  # elif command_exists lsd; then
  #   alias ls="lsd -l -1 -h1 --almost-all --git"
  #   alias l="lsd -1"
  #   alias ll="lsd -1 --almost-all"
  #   alias clearl="command clear && l"
  #   alias tree="lsd --tree"
  # elif command_exists eza; then
  #   alias ls="eza --icons --long --git"
  #   alias l="eza --icons -l"
  #   alias ll="eza --icons -laa"
  #   alias tree="eza --icons -l --tree"
  else
    alias ls="ls --color=always -lAph"
    alias l="ls --color=always -lph -w1"
    alias ll="ls --color=always -lph"
  fi

  # ─< t stands for tmux >────────────────────────────────────────────────────────────────────
  if command_exists tmux; then
    tmux_y="-- tmux-session active! | connecting to active session --"
    tmux_n="-- no tmux-session found! | creating one --"
    ta() {
      command tmux list-sessions >/dev/null 2>&1
      if [ $? -eq 0 ]; then
        if command_exists notify-send; then
          notify-send "$tmux_y"
          sleep 0.5
          tmux attach
        else
          echo_info "$tmux_y"
          sleep 0.5
          tmux attach
        fi
      else
        if command_exists notify-send; then
          notify-send "$tmux_n"
          sleep 0.5
          tmux
        else
          echo_info "$tmux_n"
          sleep 0.5
          tmux
        fi
      fi
    }
    alias ts="tmux source $HOME/.tmux.conf"
  fi

# ─< t stands for trash(-cli) >───────────────────────────────────────────────────────────────
  if command_exists trash; then
    alias rm="trash"
    alias t="trash"
  elif command_exists trash-cli; then
    alias rm="trash-cli"
    alias t="trash-cli"
  else
    echo_error "-- You do not have trash or trash-cli installed! Your 'rm' will be permanent! --"
  fi

  # ─< bat alias >────────────────────────────────────────────────────────────────────────────
  if command_exists batcat; then
    alias cat="batcat --color=always -p --paging=never"
    alias less="batcat --paging always --color=always"
    alias gd="batcat --diff"
  elif command_exists bat; then
    alias cat="bat --color=always -p"
    alias less="bat --paging always --color=always"
    alias gd="bat --diff"
  fi

  # ─< fastfetch >────────────────────────────────────────────────────────────────────────────
  if command_exists fastfetch; then
    alias ff="fastfetch"
    alias clearff="command clear & fastfetch"
    alias clearf="command clear & fastfetch"
    if fastfetch --config os >/dev/null 2>&1; then
      alias f="fastfetch --config os"
    else
      git clone https://git.k4li.de/mirrors/fastfetch.git "$HOME/.local/share/fastfetch" >/dev/null 2>&1
      exec "$SHELL"
    fi
    command clear &
    fastfetch
  fi

  # ─< d stands for docker >──────────────────────────────────────────────────────────────────
  if command_exists docker; then
    alias up="docker compose up"
    alias down="docker compose down"
    alias pull="docker compose pull"
    alias d="docker"
    alias dr="docker run --rm -it"
    alias drs="docker compose down && docker compose up -d --remove-orphans --force-recreate"
    alias ds="docker ps -a"
    alias dc="docker compose"
    alias appupdate="docker compose pull && docker compose up -d --force-recreate"
  fi
  
  # ─< g stands for GIT >─────────────────────────────────────────────────────────────────────
  if command_exists git; then
    alias g="git"
    alias gs="git status"
    alias gm='git checkout main && git merge'
    alias gc="git clone --recurse-submodule"
    gcl() {
      git clone --depth=1 "https://github.com/$*"
    }
    alias ga="git add"
    alias gp="git pull --recurse-submodule"
    alias gms='git maintenance start'
    alias gsu="git submodule foreach git pull && git add . && git commit -m '  updated 📌submodules' && echo '-- Committed changes, pushing now..' && sleep 1 && git push"
    alias gcm="git commit -m"
    alias gpu="git push --recurse-submodule=on-demand"
    if command_exists lazygit; then
      alias lg="lazygit"
    fi
  fi
}

_coding_() {
  # ─< h stands for HUGO >──────────────────────────────────────────────────────────────────
  if command_exists hugo; then
    alias h='hugo'
    alias hs='hugo server -D --noHTTPCache --disableFastRender'
  fi
  # ─< c stands for bin/cake >──────────────────────────────────────────────────────────────
  # alias cake='bin/cake'
  # alias c='cake'
  # alias cs='c server -p 1313'
  # ─< VSCodium >─────────────────────────────────────────────────────────────────────────────
  if command_exists codium; then
    alias code="codium"
    export EDITOR="codium"
  fi
  # ─< neovide, the best frontend for any neovim-config >───────────────────────────────────
  if [ -d "$HOME/.local/share/neovide/" ]; then
    if command_exists neovide; then
      alias nvim='neovide --fork'
    else
      if [ -n "$neovide_path" ]; then
        alias nvim="$neovide_path --fork"
      else
        neovide_path=$(find "$HOME/" -name "*neovide*.appimage" 2>/dev/null)
        alias nvim="$neovide_path --fork"
      fi
    fi
  fi
  # Function to get the IP address
  get_ip() {
    ip a | grep 'inet ' | grep -v '127.0.0.1' | awk '{print $2}' | cut -d/ -f1 | head -n 1
  }

  # Check if php is available, then create the alias
  if command -v php >/dev/null 2>&1; then
    alias phprun="php artisan serve --host=$(get_ip) --port=8000"
  fi

  # Check if npm is available, then create the alias
  if command -v npm >/dev/null 2>&1; then
    alias npmrun="npm run dev -- --host=$(get_ip) --port=8001"
  fi
}

# ─< rsync >────────────────────────────────────────────────────────────────────────────────
_rsync_() {
  _rrsync() {
    if command_exists find; then
      numbers=0
      for source in "${@:1:$#-1}"; do
        numbers=$(($numbers + $(find "$source" -type f | wc -l)))
      done
      destination="${@: -1}"

      rsync -avP --info=progress2 "${@:1:$#-1}" "$destination" | pv -lpes $numbers
    else
      echo_error "-- We could not find 'find'. Please install 'find' to enable the progress bar. (hit 'CTRL + C' to exit now) --"
      sleep 5
      rsync -avP --info=progress2 "$@"
    fi
  }
  if command_exists rsync; then
    alias cp="_rrsync"
    alias scp="rsync -avP"
  fi
}

get_packager() {
  search=""
  install=""
  update=""
  upgrade=""
  refresh=""
  remove=""
  . /etc/os-release
  case "$ID" in
  ubuntu | debian | pop)
    if command_exists nala; then
      search="nala search"
      install="nala install --assume-yes"
      refresh="nala update"
      upgrade="nala upgrade --full"
      remove="nala purge"
      clean="nala autoremove --assume-yes"
      alias update="$_sudo $refresh && $_sudo $upgrade"
      alias install="$_sudo $refresh && $_sudo $install"
      alias remove="$_sudo $remove"
      alias search="$search"
    elif command_exists apt-get; then
      search="apt-cache search"
      install="apt-get install --yes"
      refresh="apt-get update"
      upgrade="apt-get upgrade"
      remove="apt-get purge"
      clean="apt-get autoremove"
      alias update="$_sudo $refresh && $_sudo $upgrade"
      alias install="$_sudo $refresh && $_sudo $install"
      alias remove="$_sudo $remove"
      alias search="$search"
    fi
    ;;
  arch | manjaro | endevouros)
    if command_exists yay; then
      alias install="yay -S --noconfirm"
      alias update="yay -Syu"
      alias remove="yay -R"
      alias search="yay -Ss"
    elif command_exists paru; then
      alias install="paru -S --noconfirm"
      alias update="paru -Syu"
      alias remove="paru -R"
      alias search="paru -Ss"
    elif command_exists pacman; then
      alias install="$_sudo pacman -S --noconfirm"
      alias update="$_sudo pacman -Syu"
      alias remove="$_sudo pacman -R"
      alias search="$_sudo pacman -Ss"
    fi
    ;;
  fedora | centos)
    alias install="dnf install --yes"
    alias update="dnf update"
    alias remove="dnf remove"
    alias search="dnf search"
    ;;
  alpine)
    install="apk add"
    update="apk update"
    upgrade="apk upgrade"
    remove="apk del"
    alias install="$_sudo $install"
    alias update="$_sudo $update && $_sudo $upgrade"
    alias remove="$_sudo $remove"
    ;;
  esac
}

_tools_(){
  if ! command_exists has; then
    $(git clone https://github.com/kdabir/has.git /tmp/has && cd /tmp/has && sudo make install)
  else
    tools() {
      local pkgs="bash fish git curl make cmake gcc g++ rg docker composer node npm php jre python3 go cargo"
      has $pkgs
    }
  fi

}

main(){
  get_packager
  _init
  _alias
  _coding_
  _rsync_
}
if check_root; then
  main
fi
source /usr/share/zsh-syntax-highlighting/zsh-syntax-highlighting.zsh