#!/bin/bash -e
#
#    ip_address: report a host's ip address
#    Note: Uses decimal sleep (0.02, 0.1) which requires bash
#
#    Copyright (C) 2008-2011 Canonical Ltd.
#    Copyright (C) 2011-2014 Dustin Kirkland
#
#    Authors: Dustin Kirkland <kirkland@byobu.org>
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU General Public License as published by
#    the Free Software Foundation, version 3 of the License.
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU General Public License for more details.
#
#    You should have received a copy of the GNU General Public License
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.

__ip_address_detail() {
	if [ -x /sbin/ip ]; then
		/sbin/ip -4 addr list
		/sbin/ip -6 addr list
	fi
}

__ip_address() {
	local interface ipaddr cache="$BYOBU_RUN_DIR/cache.$BYOBU_BACKEND/ip_address"
	# Allow interface overrides in $BYOBU_CONFIG_DIR/statusrc
	if [ -n "$MONITORED_NETWORK" ]; then
		interface="$MONITORED_NETWORK"
	else
		case "$IPV6" in
			1|true|yes)
				if [ -r /proc/net/ipv6_route ]; then
					interface=$(awk '$10 != "lo" { iface=$10 ; }; END { print iface; }' /proc/net/ipv6_route)
				else
					get_network_interface; interface="$_RET"
				fi
			;;
			*) get_network_interface; interface="$_RET";;
		esac
	fi
	case "$IPV6" in
		1|true|yes)
			if [ "$IP_EXTERNAL" = "1" ]; then
				# Background an update
				timeout 1 wget -q -O- http://v6.ipv6-test.com/api/myip.php </dev/null >"$cache" 2>/dev/null &
				# Brief sleep for cold-start display optimization (requires bash)
				sleep 0.02
			else
				# Background an update
				if [ -x /sbin/ip ]; then
					LC_ALL=C /sbin/ip -6 addr list dev "$interface" scope global </dev/null >"$cache" 2>/dev/null &
				elif eval $BYOBU_TEST ifconfig >/dev/null 2>&1; then
					LC_ALL=c ifconfig "$interface" | grep "inet6 " | awk '{print $2}' | sed -e "s/%.*//" >"$cache" 2>/dev/null &

				fi
			fi
			[ -s "$cache" ] && read ipaddr < "$cache"
			# Print 'None' if we have no global address
			[ -z "$ipaddr" ] && ipaddr="None"
			ipaddr=${ipaddr#* inet6 }
			ipaddr=${ipaddr%%/*}
		;;
		*)
			if [ "$IP_EXTERNAL" = "1" ]; then
				if [ -n "$EXTERNAL_IP_SOURCE" ]; then
					# User-specified URL (e.g. https://api.ipify.org). LP: #1869483
					timeout 1 wget -q -O- "$EXTERNAL_IP_SOURCE" </dev/null >"$cache" 2>/dev/null &
					sleep 0.02
				elif command -v dig >/dev/null 2>&1; then
					# Try dig with OpenDNS first (faster), fall back to wget
					timeout 1 dig +short myip.opendns.com @resolver1.opendns.com </dev/null 2>/dev/null | tr -d '\n' >"$cache" &
					# Brief sleep for cold-start display optimization (requires bash)
					sleep 0.1
				fi
				# Fall back to wget if primary source didn't work
				if [ ! -s "$cache" ]; then
					timeout 1 wget -q -O- http://v4.ipv6-test.com/api/myip.php </dev/null >"$cache" 2>/dev/null &
					# Brief sleep for cold-start display optimization (requires bash)
					sleep 0.02
				fi
				[ -s "$cache" ] && read ipaddr < "$cache"
			elif metadata_available; then
				# We're in EC2, so get our public IP address
				timeout 0.2 wget -q -O- http://169.254.169.254/latest/meta-data/public-ipv4 </dev/null >"$cache" 2>/dev/null &
				# Brief sleep for cold-start display optimization (requires bash)
				sleep 0.02
				[ -s "$cache" ] && read ipaddr < "$cache"
			else
				if [ -x /sbin/ip ]; then
					# 'ip route get' reports the preferred source for outbound
					# traffic without needing to guess the interface. LP: #2015819
					ipaddr=$(LC_ALL=C /sbin/ip route get 1.1.1.1 2>/dev/null | \
						awk '/src/{for(i=1;i<=NF;i++) if($i=="src"){print $(i+1); exit}}')
					# Fall back to interface-based detection (no default route, VPN, etc.)
					if [ -z "$ipaddr" ]; then
						ipaddr=$(LC_ALL=C /sbin/ip -4 addr list dev "$interface" scope global 2>/dev/null)
						ipaddr=${ipaddr#* inet }
						ipaddr=${ipaddr%%/*}
					fi
				elif eval $BYOBU_TEST ifconfig >/dev/null 2>&1; then
					ipaddr=$(ifconfig "$interface" | grep "inet " | awk '{print $2}')
				fi
			fi
		;;
	esac
	if [ -n "$ipaddr" ]; then
		if [ "$1" = "t" ]; then
			printf "%s" "$ipaddr"
		else
			color b w k; printf "%s" "$ipaddr"; color --
		fi
	fi
}

# vi: syntax=sh ts=4 noexpandtab
