1
0
Fork 0
mirror of https://github.com/romkatv/powerlevel10k.git synced 2024-11-25 13:30:07 +00:00

Match VPN Interface fuzzy

This commit is contained in:
Dominik Ritter 2019-01-23 18:56:07 +01:00
parent 54a985cb11
commit bde4337992
2 changed files with 49 additions and 4 deletions

View file

@ -600,9 +600,25 @@ prompt_public_ip() {
# Check VPN is on if VPN interface is set # Check VPN is on if VPN interface is set
if [[ -n $POWERLEVEL9K_PUBLIC_IP_VPN_INTERFACE ]]; then if [[ -n $POWERLEVEL9K_PUBLIC_IP_VPN_INTERFACE ]]; then
if [[ "$OS" == "OSX" ]]; then if [[ "$OS" == "OSX" ]]; then
local interface="$(${ROOT_PREFIX}/sbin/ifconfig $POWERLEVEL9K_PUBLIC_IP_VPN_INTERFACE)" # Get a plain list of all interfaces
# Check if interface is UP. local rawInterfaces="$(${ROOT_PREFIX}/sbin/ifconfig -l)"
[[ "$interface" =~ "<UP," ]] && icon='VPN_ICON' # Parse into array (split by whitespace)
local -a interfaces
interfaces=(${=rawInterfaces})
# Parse only relevant interface names
local pattern="${POWERLEVEL9K_PUBLIC_IP_VPN_INTERFACE}[^ ]?"
local -a relevantInterfaces
for rawInterface in $interfaces; do
[[ "$rawInterface" =~ $pattern ]] && relevantInterfaces+=( $MATCH )
done
for interfaceName in $relevantInterfaces; do
local interface="$(${ROOT_PREFIX}/sbin/ifconfig $interfaceName)"
# Check if interface is UP.
if [[ "$interface" =~ "<UP," ]]; then
icon='VPN_ICON'
break
fi
done
else else
local interface=$(${ROOT_PREFIX}/sbin/ip -brief -4 a show "${POWERLEVEL9K_PUBLIC_IP_VPN_INTERFACE}") local interface=$(${ROOT_PREFIX}/sbin/ip -brief -4 a show "${POWERLEVEL9K_PUBLIC_IP_VPN_INTERFACE}")
[[ -n "$interface" ]] && icon='VPN_ICON' [[ -n "$interface" ]] && icon='VPN_ICON'

View file

@ -42,7 +42,12 @@ function fakeIfconfig() {
cat > $FOLDER/sbin/ifconfig <<EOF cat > $FOLDER/sbin/ifconfig <<EOF
#!/usr/bin/env zsh #!/usr/bin/env zsh
if [[ "\$#" -gt 0 ]]; then if [[ "\$*" == "-l" ]]; then
echo "lo0 gif0 stf0 EHC250 EHC253 tun1 tun0 ${INTERFACE} fw0 en0 en2 en1 p2p0 bridge0 utun0"
exit 0
fi
if [[ "\$*" == "${INTERFACE}" ]]; then
cat <<INNER cat <<INNER
${INTERFACE}: flags=4099<UP,BROADCAST,MULTICAST> mtu 1500 ${INTERFACE}: flags=4099<UP,BROADCAST,MULTICAST> mtu 1500
inet 1.2.3.4 txqueuelen 1000 (Ethernet) inet 1.2.3.4 txqueuelen 1000 (Ethernet)
@ -55,6 +60,7 @@ INNER
exit 0 exit 0
fi fi
if [[ "\$#" == "0" ]]; then
cat <<INNER cat <<INNER
docker0: flags=4099<UP,BROADCAST,MULTICAST> mtu 1500 docker0: flags=4099<UP,BROADCAST,MULTICAST> mtu 1500
inet 172.17.0.1 netmask 255.255.0.0 broadcast 172.17.255.255 inet 172.17.0.1 netmask 255.255.0.0 broadcast 172.17.255.255
@ -81,6 +87,7 @@ lo: flags=73<UP,LOOPBACK,RUNNING> mtu 65536
TX packets 5136 bytes 328651 (320.9 KiB) TX packets 5136 bytes 328651 (320.9 KiB)
TX errors 0 dropped 0 overruns 0 carrier 0 collisions 0 TX errors 0 dropped 0 overruns 0 carrier 0 collisions 0
INNER INNER
fi
EOF EOF
chmod +x $FOLDER/sbin/ifconfig chmod +x $FOLDER/sbin/ifconfig
} }
@ -346,4 +353,26 @@ function testPublicIpSegmentWithVPNTurnedOnOsx() {
unfunction stat unfunction stat
} }
function testPublicIpSegmentWithVPNTurnedOnAndFuzzyMatchingOnOsx() {
typeset -F now
now=$(date +%s)
local OS='OSX'
echo "1.2.3.4" > $POWERLEVEL9K_PUBLIC_IP_FILE
local POWERLEVEL9K_PUBLIC_IP_VPN_INTERFACE="tun"
# Fake stat call
function stat() {
echo $now
}
# Fake ifconfig
fakeIfconfig "tun3"
assertEquals "%K{000} %F{007}(vpn) %f%F{007}1.2.3.4 " "$(prompt_public_ip left 1 false "$FOLDER")"
unfunction stat
}
source shunit2/shunit2 source shunit2/shunit2