Save/restore also linked windows

Author: hlintala

scripts/helpers.sh

@@ -78,6 +78,17 @@ is_session_grouped() {
 	[[ "$GROUPED_SESSIONS" == *"${d}${session_name}${d}"* ]]
 }
 
+get_linked_windows() {
+	local linked_windows_dump="$1"
+	export LINKED_WINDOWS="${d}$(echo "$linked_windows_dump" | cut -f2-3 -d"$d" | tr "$d\\n" ":$d")"
+}
+
+is_window_linked() {
+	local session_name="$1"
+	local window_index="$2"
+	[[ "$LINKED_WINDOWS" = *"${d}${session_name}:${window_index}${d}"* ]]
+}
+
 # pane content file helpers
 
 pane_contents_create_archive() {

scripts/restore.sh

@@ -284,6 +284,13 @@ handle_session_0() {
 	fi
 }
 
+restore_linked_windows() {
+	\grep '^linked_window' $(last_resurrect_file) |
+		while IFS=$d read line_type session window_index original_session original_window_index; do
+			tmux link-window -s "${original_session}:${original_window_index}" -t "${session}:${window_index}"
+		done
+}
+
 restore_window_properties() {
 	local window_name
 	\grep '^window' $(last_resurrect_file) |
@@ -369,6 +376,7 @@ main() {
 		execute_hook "pre-restore-all"
 		restore_all_panes
 		handle_session_0
+		restore_linked_windows
 		restore_window_properties >/dev/null 2>&1
 		execute_hook "pre-restore-pane-processes"
 		restore_all_pane_processes

scripts/save.sh

@@ -25,6 +25,18 @@ grouped_sessions_format() {
 	echo "$format"
 }
 
+linked_windows_format() {
+	local format
+	format+="#{window_linked}"
+	format+="${delimiter}"
+	format+="#{window_id}"
+	format+="${delimiter}"
+	format+="#{session_name}"
+	format+="${delimiter}"
+	format+="#{window_index}"
+	echo "$format"
+}
+
 pane_format() {
 	local format
 	format+="pane"
@@ -185,13 +197,45 @@ fetch_and_dump_grouped_sessions(){
 	fi
 }
 
+dump_linked_windows() {
+	local current_window_id=""
+	local original_session
+	local original_window_index
+	tmux list-windows -a -F "$(linked_windows_format)" |
+		grep "^1" |
+		cut -c 3- |
+		sort |
+		while IFS=$d read window_id session_name window_index; do
+			if is_session_grouped "$session_name"; then
+				continue
+			fi
+			if [ "$window_id" != "$current_window_id" ]; then
+				# this window is the original/first of linked windows
+				original_session="$session_name"
+				original_window_index="$window_index"
+				current_window_id="$window_id"
+			else
+				# this window is linked to the original window
+				echo "linked_window${d}${session_name}${d}${window_index}${d}${original_session}${d}${original_window_index}"
+			fi
+		done
+}
+
+fetch_and_dump_linked_windows() {
+	local linked_windows_dump="$(dump_linked_windows)"
+	get_linked_windows "$linked_windows_dump"
+	if [ -n "$linked_windows_dump" ]; then
+		echo "$linked_windows_dump"
+	fi
+}
+
 # translates pane pid to process command running inside a pane
 dump_panes() {
 	local full_command
 	dump_panes_raw |
 		while IFS=$d read line_type session_name window_number window_active window_flags pane_index pane_title dir pane_active pane_command pane_pid history_size; do
 			# not saving panes from grouped sessions
-			if is_session_grouped "$session_name"; then
+			if is_session_grouped "$session_name" || is_window_linked "$session_name" "$window_number"; then
 				continue
 			fi
 			full_command="$(pane_full_command $pane_pid)"
@@ -240,6 +284,7 @@ save_all() {
 	local last_resurrect_file="$(last_resurrect_file)"
 	mkdir -p "$(resurrect_dir)"
 	fetch_and_dump_grouped_sessions > "$resurrect_file_path"
+	fetch_and_dump_linked_windows >> "$resurrect_file_path"
 	dump_panes   >> "$resurrect_file_path"
 	dump_windows >> "$resurrect_file_path"
 	dump_state   >> "$resurrect_file_path"