#!/usr/bin/env bash

# Generic
alias sb="source ~/.bashrc"
alias bashreload="source ~/.bashrc"
alias sz="source ~/.zshrc"
alias zshreload="source ~/.zshrc"
alias bashrc="${EDITOR:-nano} +116 ~/.bashrc"
alias zshrc="${EDITOR:-nano} ~/.zshrc"
alias free="free -ht"
alias type="type -a"
alias grep='grep --color=auto'
alias egrep='egrep --color=auto'
alias fgrep='fgrep --color=auto'
alias usersearch="awk -F: '{print \"UserName: \" \$1 \", UserID: \" \$3 \", Home Dir: \" \$6 \", Shell Used: \" \$7}' /etc/passwd | grep"
# alias code="codium"


# History
alias histsearch="history | grep"
alias hs="histsearch"
alias hsi="histsearch"
alias untar='tar -zxvf '


# Directories and Directory listings
alias ~="cd ~"
alias ..="cd .."
alias ...='cd ../../../'             # Go back 3 directory levels
alias ..4='cd ../../../../'          # Go back 4 directory levels
alias ..5='cd ../../../../../'       # Go back 5 directory levels
alias ..6='cd ../../../../../../'    # Go back 6 directory levels
alias lsc='ls --color=auto'
alias ll='lsc -alhF'
alias la='lsc -Ah'
alias l='lsc -CF'
alias lsa="lsc -lAFhZ"
alias mkdir="mkdir -pv"
alias df="df -h"
dir_size(){
	local dir
	if [[ -z "$1" ]]; then
		dir="${PWD}"
	else
		dir="$1"
	fi

	du -ah "${dir}" --max-depth=1 | sort -hr
}
mkcd () {
    mkdir "$1"
    cd "$1" || exit
}


# Network
alias ping="ping -c 10"
alias ping8="ping 8.8.8.8"
alias ping1="ping 1.1.1.1"
alias p8="ping8"
alias p1="ping1"
alias flush-dns="sudo systemd-resolve --flush-caches"
alias dnsreset="sudo systemctl restart dnscrypt-proxy"
alias dnscheck="dnscrypt-proxy -resolve google.com"
alias pubip="curl https://ipinfo.io/ip; echo"
alias ips='printf "Local IP:- "; hostname -I | cut -f1 -d " "; printf "Public IP:- "; curl -s https://ipinfo.io/ip'
alias ipdetails='printf "Local IP:- "; hostname -I | cut -f1 -d " "; printf "Public IP Details:- \n"; geoip'
alias speedtest="speedtest-cli --secure"			# needs speedtest-cli installed
geoip () {
	curl -s https://ipinfo.io | sed '/readme/d;/loc/d;/postal/d;s/org/ISP/' | tr -d {},\" | awk -F ':' 'NF {printf ("%10s: %.25s \n", $1, $2)}'
}
alias listening-apps="sudo netstat -nutlp | grep ':' | awk '{print \$1,\$4,\$NF}' | awk -F: '{print \$1,\$(NF-1),\$NF}' | awk -v OFS=\"\t\" 'BEGIN {printf (\"%s\t%s\t\t%s \n\", \"PROTO\", \"PORT\", \"APPLICATION\")} {print \$1 , \$(NF-1) ,\" \" , \$NF}' | (read -r; printf \"%s\n\" \"\$REPLY\"; sort -k2 -n)"


# Update & Upgrades
alias up="sudo apt-get update && sudo apt-get upgrade -y && distup && autorem"
alias distup="sudo apt dist-upgrade"
alias autorem="sudo apt autoremove"
alias update="sudo apt-get update"
alias install="sudo apt-get install "
alias remove="sudo apt-get remove "


# Git
# To use this - Ensure all git server SSH are in ~/.ssh
alias git_signin='(for i in ~/.ssh/{*github*,*bitbucket*,*gitea*,*gitlab*}; do ssh-add -k $i; done; ) && (echo; echo Identities added successfully)'
git_push_all_changes(){
	git add . && git commit -am "${1}" && git push
}


# For servers
alias ngt="sudo nginx -t"
alias ngrestart="sudo systemctl restart nginx"
alias ngreload="sudo systemctl reload nginx"
alias ngstop="sudo systemctl stop nginx"
alias fpmreset71="sudo systemctl restart php7.1-fpm"
alias fpmreset72="sudo systemctl restart php7.2-fpm"
alias fpmreset73="sudo systemctl restart php7.3-fpm"
alias fpmreset74="sudo systemctl restart php7.4-fpm"
alias ta="tmux a"
alias tat="tmux a -t"
alias tls="tmux ls"
alias tnew="tmux new"
alias tnewt="tmux new -t"

f2b_banned_ips() {
	local provided_jail=$1

	if [[ -n "${provided_jail// /}" ]]; then
		for ip in $(sudo fail2ban-client status "${provided_jail}" | tail -1 | sed 's/[^:]*://;s/\s*//')
		do
			printf "%17s\n" "$ip"
		done
	else
		local total_ips_banned=0
		for JAIL in $(sudo fail2ban-client status | tail -1 | sed 's/[^:]*://;s/\s*//;s/,//g')
		do
			local banned_ip_count
			banned_ip_count=$(sudo fail2ban-client status "${JAIL}" | grep -oP 'Currently banned:\s*\K\d+')

			if [[ "${banned_ip_count}" -gt 0 ]]; then
				echo "${JAIL}: ${banned_ip_count}"

				for ip in $(sudo fail2ban-client status "${JAIL}" | tail -1 | sed 's/[^:]*://;s/\s*//')
				do
					printf "%17s\n" "[$ip]"
				done

				total_ips_banned=$(( total_ips_banned + banned_ip_count ))

				echo
			else
				echo -e "${JAIL}:\n    -None-\n"
			fi
		done

		echo "Total IPs banned across all jails - ${total_ips_banned}"
	fi
}

f2b_unban_ip() {
    local ip_to_unban="$1"
    local jail="$2"

    # If jail is provided - use that jail to directly unban
    if [[ -n "${jail// /}" ]]; then
        sudo fail2ban-client set "${jail}" unbanip "${ip_to_unban}" > /dev/null && echo "Successfully released ban"
    else
        # Find all JAILS this IP belong to
        # Unban the ip where ever it is found

        for JAIL in $(sudo fail2ban-client status | tail -1 | sed 's/[^:]*://;s/\s*//;s/,//g')
        do
            local banned_ip_count
			banned_ip_count=$(sudo fail2ban-client status "${JAIL}" | grep -oP 'Currently banned:\s*\K\d+')
            if [[ "$banned_ip_count" -gt 0 ]] && [[ $(sudo fail2ban-client status "${JAIL}") == *"${ip_to_unban}"* ]]; then
                local found_ip="true"
                echo "Unbanning from ${JAIL}:"
                sudo fail2ban-client set "${JAIL}" unbanip "${ip_to_unban}" > /dev/null && echo "Successfully released ban"
            fi
        done

        if [[ -z "${found_ip// /}" ]]; then
                echo "${ip_to_unban} was not found in any banned lists."
                echo "No action taken."
        fi
    fi
}

f2b_ban_an_ip(){
	local ip_to_ban=$1
	local ban_jail=$2

	if [[ ( -z "${ip_to_ban// /}" ) || ( -z "${ban_jail// /}" ) ]]; then
		echo "Please provide an IP and a Jail (in that order)"
		echo -e "eg -\n\t f2b_ban_an_ip 1.1.1.1 sshd"
		return 1
	fi

	sudo fail2ban-client set "${ban_jail}" banip "${ip_to_ban}" > /dev/null && echo "Ban successful"
}