.bashrc_ext

# Check if the command 'exa' exists
command -v exa >/dev/null 2>&1 && {
    alias ls='exa --long --header --git'
    alias l='exa --all --long --header --git'
}

# Check if the command 'ghq' exists
command -v ghq >/dev/null 2>&1 && {
    alias g='ghq'
}

# Check if the command 'batcat' exists
command -v batcat >/dev/null 2>&1 && {
    alias bat='batcat'
}

alias lzd='lazydocker'
alias dcul='docker compose up -d && docker compose logs -f'
alias dcup='docker compose pull && docker compose down && dcul'
alias dcd='docker compose down'
alias dcb='docker compose build'
alias dclf='docker compose logs -f'
alias update='sudo apt update && sudo apt upgrade -y'

cd() {
    if [[ $1 == ..[0-9]* ]]; then
        local count="${1:2}"
        local target=""
        for ((i=0; i<count; i++)); do
            target+="../"
        done
        builtin cd "$target"
    else
        builtin cd "$@"
    fi
    ls
}

declare -A saved_dirs
spwd() {
    local name="default"
    if [[ ! -z "$1" ]]; then
        name="$1"
    fi
    saved_dirs["$name"]=$(pwd)
    echo "Saved current directory as '$name'."
}

rpwd() {
    local name="default"
    if [[ ! -z "$1" ]]; then
        name="$1"
    fi

    if [[ ! -z "${saved_dirs["$name"]}" ]]; then
        cd "${saved_dirs["$name"]}" || return 1
    else
        echo "No saved directory found for '$name'."
    fi
}

lpwd() {
    for name in "${!saved_dirs[@]}"; do
        echo "$name: ${saved_dirs[$name]}"
    done
}

ex() {
	if [ -f "$1" ] ; then
		case $1 in
			*.tar.bz2)   tar xjf $1   ;;
			*.tar.gz)    tar xzf $1   ;;
			*.bz2)       bunzip2 $1   ;;
			*.rar)       unrar x $1   ;;
			*.gz)        gunzip $1    ;;
			*.tar)       tar xf $1    ;;
			*.tbz2)      tar xjf $1   ;;
			*.tgz)       tar xzf $1   ;;
			*.zip)       unzip $1     ;;
			*.Z)         uncompress $1;;
			*.7z)        7z x $1      ;;
			*.deb)       ar x $1      ;;
			*.tar.xz)    tar xf $1    ;;
			*.tar.zst)   unzstd $1    ;;
			*)           echo "'$1' cannot be extracted via ex()" ;;
		esac
	else
		echo "'$1' is not a valid file"
	fi
}

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

dotfiles_update() {
    git pull
    ARCH=$(uname -m); if [ "$ARCH" = "x86_64" ]; then ARCH="amd64"; else ARCH="arm64"; fi; URL=$(curl -s https://api.github.com/repos/Zelak312/dotfiles/releases/latest | jq -r --arg ARCH "$ARCH" '.assets[] | select(.name | contains("dotfiles-linux-"+$ARCH)) | .browser_download_url'); TEMP=$(mktemp) && curl -L -o $TEMP $URL && chmod +x $TEMP && $TEMP && rm $TEMP
}

backup() {
    local backup_name=""
    local files=()

    # Loop through the arguments
    for arg in "$@"; do
        if [[ "$arg" == "--name" ]]; then
            backup_name=true
        elif [[ "$backup_name" == true ]]; then
            backup_name="$arg"
            backup_name="${backup_name}.tar.gz" # Ensuring the file extension
        else
            files+=("$arg")
        fi
    done

    # Check if files are provided
    if [ ${#files[@]} -eq 0 ]; then
        echo "Usage: backup [--name custom_name] [folder1] [folder2] [file1] ..."
        return 1
    fi

    # Set default name if not provided
    if [[ -z "$backup_name" ]]; then
        backup_name="backup_$(date +%Y%m%d_%H%M%S).tar.gz"
    fi

    # Create the backup
    tar -czf "$backup_name" -- "${files[@]}"
    echo "Backup created: $backup_name"
}

check_ports() {
    if [ $# -eq 0 ]; then
        echo "Usage: check_ports <range>"
        return 1
    fi
    ss -tuln | grep ":$1"
}

git_stash_diff() {
    if [ $# -eq 0 ]; then
        git stash show -p
    else
        git stash show -p stash@{"$1"}
    fi
}

# on startup
# Check if tmux is installed and automatically start tmux on SSH login
if command -v tmux &> /dev/null && [ -z "$TMUX" ] && [ -n "$SSH_CONNECTION" ]; then
    # Check if connection doesn't come from vscode
    if [[ -z "$VSCODE_IPC_HOOK_CLI" && "$TERM_PROGRAM" != "vscode" ]]; then
        tmux -u attach-session -t ssh_tmux || tmux -u new-session -s ssh_tmux
    fi
fi