Add displaying attached container host and user

Author: tony-sol

scripts/shared.sh

@@ -14,20 +14,42 @@ set_tmux_option() {
 }
 
 get_remote_info() {
-	local command=$1
-	local pane_pid=$(tmux display-message -p "#{pane_pid}")
-
 	# First get the current pane command pid to get the full command with arguments
-	local cmd=$({ pgrep -flaP $pane_pid ; ps -o command -p $pane_pid ; } | xargs -I{} echo {} | grep ssh | sed -E 's/^[0-9]*[[:blank:]]*ssh //')
+	local cmd=$(_current_pane_command | grep -E 'ssh' | sed -E 's/^[0-9]*[[:blank:]]*ssh //')
 	# Fetch configuration with given cmd
 	ssh -G $cmd 2>/dev/null | grep -E -e '^host\s' -e '^port\s' -e '^user\s' | sort | cut -f 2 -d ' ' | xargs
 }
 
+get_container_info() {
+	local cmd=$(_current_pane_command | grep -E 'docker|podman' | sed -E 's/^[0-9]*[[:blank:]]* //')
+
+	local runner=${cmd%% *}
+	# @TODO get dynamic named container
+	if [[ $cmd =~ ' --name' ]]; then
+		local container=$(echo ${cmd##* --name} | cut -f 1 -d ' ')
+		container=${container##*=}
+	else
+		local all_running_containers=$(docker ps -q | xargs docker inspect)
+		return
+	fi
+
+	local info=$($runner inspect --format '{{ .Config.Hostname }}/{{ .Config.Domainname }}/{{ .Config.User }}' $container)
+	local host=$(echo $info | cut -f 1 -d '/')
+	local domain=$(echo $info | cut -f 2 -d '/')
+	local user=$(echo $info | cut -f 3 -d '/')
+	# @TODO `port` is not applicable with container for now
+	echo "${host}${domain:+.$domain} 0 ${user}"
+}
+
 get_info() {
+	local host port user
 	# If command is ssh, fetch connection info
 	if ssh_connected; then
 		read -r host port user <<<$(get_remote_info)
 	fi
+	if containered; then
+		read -r host port user <<<$(get_container_info)
+	fi
 	# Return requested info
 	case "$1" in
 		"user") # user from ssh info or `whoami`
@@ -49,10 +71,20 @@ get_info() {
 }
 
 ssh_connected() {
-	local pane_pid=$(tmux display-message -p "#{pane_pid}")
-
-	# Get current pane command
-	local cmd=$(pgrep -flaP $pane_pid)
+	local cmd=$(_current_pane_command)
 
 	[[ $cmd =~ " ssh " ]] || [[ $cmd =~ " sshpass " ]]
 }
+
+containered() {
+	local cmd=$(_current_pane_command)
+
+	[[ $cmd =~ " docker " ]] || [[ $cmd =~ " podman " ]]
+}
+
+_current_pane_command() {
+	local pane_pid=$(tmux display-message -p "#{pane_pid}")
+
+	# @TODO research, maybe `pgrep -flaP $pane_pid` is enough
+	{ pgrep -flaP $pane_pid; ps -o command -p $pane_pid; } | xargs -I{} echo {}
+}