#!/bin/sh
#
#    byobu-reconnect-sockets - source this file to re-establish
#                              GPG_AGENT_INFO and DBUS_SESSION_BUS_ADDRESS,
#                              useful when reconnecting to an existing
#                              byobu session.
#
#    Copyright (C) 2009 Canonical Ltd.
#    Copyright (C) 2012-2014 Dustin Kirkland <kirkland@byobu.org>
#
#    Authors: Dustin Kirkland <kirkland@byobu.org>
#             Ryan C. Thompson <rct@thompsonclan.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/>.

PKG="byobu"
[ -r "$HOME/.byoburc" ] && . "$HOME/.byoburc"
[ -z "${BYOBU_PREFIX}" ] && export BYOBU_PREFIX="@prefix@" || export BYOBU_PREFIX
. "${BYOBU_PREFIX}/lib/${PKG}/include/common"

case "$-" in
	*i*)
		# Interactive shell — sourced normally
	;;
	*)
		# Non-interactive. Allow when BYOBU_BACKEND is already set in the
		# environment (e.g. fish users sourcing via the bass shim).
		# Refuse only when there is no sign of a running byobu context,
		# which most likely means the script is being executed directly.
		if [ -z "$BYOBU_BACKEND" ]; then
			echo 2>&1
			echo "ERROR: You must source this file, rather than execute it." 2>&1
			echo "  . $0" 2>&1
			echo 2>&1
			exit 1
		fi
	;;
esac

export_and_send () {
	var="$1"
	value="$(eval "echo \$$var")"
	export "$var"
	case $BYOBU_BACKEND in
		tmux)
			tmux setenv "$var" "$value"
		;;
		screen)
			screen -X setenv "$var" "$value"
		;;
	esac
}

screen_update () {
	# Ensure that screen's environment variables/values get propagated here
	# Enable word splitting for zsh:
	[ "x$ZSH_VERSION" != x ] && setopt local_options sh_word_split
	tempfile=$(mktemp -q ${BYOBU_RUN_DIR}/sockets-XXXXXXXX) && {
		for var in $VARS_TO_UPDATE; do
			screen sh -c "echo export $var=\$$var >> \"$tempfile\""
		done
		. "$tempfile"
		rm -f "$tempfile"
	}
}

tmux_update () {
	# Ensure that tmux's environment variables/values get propagated here
	# Enable word splitting for zsh:
	[ "x$ZSH_VERSION" != x ] && setopt local_options sh_word_split
	for var in $VARS_TO_UPDATE; do
		expr="$(tmux showenv | grep "^$var=")"
		if [ -n "$expr" ]; then
			export "$expr"
		fi
	done
}

# Pull environment variables/values from backend and update/export here
VARS_TO_UPDATE="DISPLAY DBUS_SESSION_BUS_ADDRESS SESSION_MANAGER GPG_AGENT_INFO XDG_SESSION_COOKIE XDG_SESSION_PATH GNOME_KEYRING_CONTROL GNOME_KEYRING_PID GPG_AGENT_INFO SSH_ASKPASS SSH_AUTH_SOCK SSH_AGENT_PID WINDOWID UPSTART_JOB UPSTART_EVENTS UPSTART_SESSION UPSTART_INSTANCE"

case $BYOBU_BACKEND in
	tmux)
		tmux_update
	;;
	screen)
		screen_update
	;;
esac

# Establish gpg-agent socket, helps when reconnecting to a detached session
newest "$HOME/.gnupg/gpg-agent-info-*" && . "$_RET" && export_and_send GPG_AGENT_INFO

# Reconnect dbus, source the most recently touched session-bus
# Sorry, ls -t is needed here, to sort by time
newest "$HOME/.dbus/session-bus/*" && . "$_RET"

[ -r "$BYOBU_RUN_DIR/sockets" ] && . "$BYOBU_RUN_DIR/sockets"
export_and_send DBUS_SESSION_BUS_ADDRESS
export_and_send SESSION_MANAGER

# vi: syntax=sh ts=4 noexpandtab
