Tidy some common code for destroying sessions into a new function.

3 files changed, 25 insertions, 30 deletions

diff --git a/server-fn.c b/server-fn.c
index d3cdbd60..d39108dd 100644
--- a/server-fn.c
+++ b/server-fn.c
@@ -272,10 +272,7 @@ server_kill_window(struct window *w)
 {
 	struct session	*s;
 	struct winlink	*wl;
-	struct client	*c;
-	u_int		 i, j;
-	int		 destroyed;
-
+	u_int		 i;
 	
 	for (i = 0; i < ARRAY_LENGTH(&sessions); i++) {
 		s = ARRAY_ITEM(&sessions, i);
@@ -284,23 +281,30 @@ server_kill_window(struct window *w)
 		if ((wl = winlink_find_by_window(&s->windows, w)) == NULL)
 			continue;
 		
-		destroyed = session_detach(s, wl);
-		for (j = 0; j < ARRAY_LENGTH(&clients); j++) {
-			c = ARRAY_ITEM(&clients, j);
-			if (c == NULL || c->session != s)
-				continue;
-			
-			if (destroyed) {
-				c->session = NULL;
-				server_write_client(c, MSG_EXIT, NULL, 0);
-			} else
-				server_redraw_client(c);
-		}
+		if (session_detach(s, wl))
+			server_destroy_session(s);
+		else
+			server_redraw_session(s);
 	}
 	recalculate_sizes();
 }
 
 void
+server_destroy_session(struct session *s)
+{
+	struct client	*c;
+	u_int		 i;
+	
+	for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
+		c = ARRAY_ITEM(&clients, i);
+		if (c == NULL || c->session != s)
+			continue;
+		c->session = NULL;
+		server_write_client(c, MSG_EXIT, NULL, 0);
+	}
+}
+
+void
 server_set_identify(struct client *c)
 {
 	struct timeval	tv;
diff --git a/server.c b/server.c
index ce2d984a..09ce8b36 100644
--- a/server.c
+++ b/server.c
@@ -1211,21 +1211,11 @@ server_check_window(struct window *w)
 		RB_FOREACH(wl, winlinks, &s->windows) {
 			if (wl->window != w)
 				continue;
-			destroyed = session_detach(s, wl);
-			for (j = 0; j < ARRAY_LENGTH(&clients); j++) {
-				c = ARRAY_ITEM(&clients, j);
-				if (c == NULL || c->session != s)
-					continue;
-				if (!destroyed) {
-					server_redraw_client(c);
-					continue;
-				}
-				c->session = NULL;
-				server_write_client(c, MSG_EXIT, NULL, 0);
-			}
-			/* If the session was destroyed, bail now. */
-			if (destroyed)
+			if (session_detach(s, wl)) {
+				server_destroy_session(s);
 				break;
+			}
+			server_redraw_session(s);
 			goto restart;
 		}
 	}
diff --git a/tmux.h b/tmux.h
index a1b7fea2..ed7242c5 100644
--- a/tmux.h
+++ b/tmux.h
@@ -1457,6 +1457,7 @@ void	 server_status_window(struct window *);
 void	 server_lock(void);
 int	 server_unlock(const char *);
 void	 server_kill_window(struct window *);
+void	 server_destroy_session(struct session *);
 void	 server_set_identify(struct client *);
 void	 server_clear_identify(struct client *);