lazy/files/zsh/20-function.zsh

106 lines
3.2 KiB
Bash

#!/bin/zsh
__zsh_is_public_ipv4() {
local ip=$1
if [[ $ip =~ ^[0-9]+\.[0-9]+\.[0-9]+\.[0-9]+$ ]]; then
IFS='.' read -r -A ADDR <<< "$ip"
if [[ ${ADDR[1]} -eq 10 ]]; then
return 1
elif [[ ${ADDR[1]} -eq 172 && ${ADDR[2]} -ge 16 && ${ADDR[2]} -le 31 ]]; then
return 1
elif [[ ${ADDR[1]} -eq 192 && ${ADDR[2]} -eq 168 ]]; then
return 1
else
return 0
fi
else
return 1
fi
}
__zsh_get_ipv4() {
local ipv4=$(ip route get 1.1.1.1 | head -1 | awk '{print $7}')
if __zsh_is_public_ipv4 $ipv4; then echo "$ipv4"; return 0; fi
ipv4=$(ip route get 1.1.1.1 | head -1 | awk '{print $NF}')
if __zsh_is_public_ipv4 $ipv4; then echo "$ipv4"; return 0; fi
ipv4=$(curl -4 -s ifconfig.me)
if __zsh_is_public_ipv4 $ipv4; then echo "$ipv4"; return 0; fi
ipv4=$(host -4 myip.opendns.com resolver1.opendns.com | grep "myip.opendns.com has" | awk '{print $4}')
if __zsh_is_public_ipv4 $ipv4; then echo "$ipv4"; return 0; fi
# echo "No public IPv4 address found"
return 1
}
__zsh_get_ipv6() {
if [[ "$(ip -6 route get 2001:4860:4860::8888 2>&1 | awk '{print $NF}')" == "unreachable" ]]; then
echo "No IPv6 connectivity"
return 1
fi
local ipv6=$(ip route get 2001:4860:4860::8888 | head -1 | awk '{print $11}')
if [[ $ipv6 =~ ^([0-9a-fA-F]{0,4}:){7}[0-9a-fA-F]{0,4}$ ]]; then echo "$ipv6"; return 0; fi
ipv6=$(ip route get 2001:4860:4860::8888 | head -1 | awk '{print $NF}')
if [[ $ipv6 =~ ^([0-9a-fA-F]{0,4}:){7}[0-9a-fA-F]{0,4}$ ]]; then echo "$ipv6"; return 0; fi
ipv6=$(curl -6 -s ifconfig.me)
if [[ $ipv6 =~ ^([0-9a-fA-F]{0,4}:){7}[0-9a-fA-F]{0,4}$ ]]; then echo "$ipv6"; return 0; fi
# echo "No public IPv6 address found"
return 1
}
__zsh_get_ip() {
__zsh_get_ipv4
__zsh_get_ipv6
}
__zsh_history_remove() {
if [ -z "$1" ]; then
echo "Please specify a keyword to remove"
return 1
fi
# Check if HISTFILE exists
if [ -z "$HISTFILE" ]; then
echo "Error: HISTFILE is not set"
return 1
fi
local keyword="$1"
local backup_file="${HISTFILE}.backup"
# Check if there are any matching entries
if ! grep -q "$keyword" "$HISTFILE"; then
echo "No matching entries found for: $keyword"
return 0
fi
# Show matching entries
echo "The following history entries will be removed:"
echo "----------------------------------------"
grep --color=always "$keyword" "$HISTFILE"
echo "----------------------------------------"
# Confirmation
echo -n "Are you sure you want to remove these entries? (y/N) "
read confirm
if [[ ! "$confirm" =~ ^[yY](es)?$ ]]; then
echo "Operation cancelled"
return 0
fi
# Create backup
cp "$HISTFILE" "$backup_file"
echo "Backup created at: $backup_file"
# Remove entries containing keyword
if [[ "$OSTYPE" == "darwin"* ]]; then
# macOS
sed -i '' "/${keyword}/d" "$HISTFILE"
else
# Linux
sed -i "/${keyword}/d" "$HISTFILE"
fi
echo "Removal completed"
echo "If needed, you can restore from backup using:"
echo "cp \"$backup_file\" \"$HISTFILE\""
}