From ccd2f3a464285a896c974127a2824b8aeeeab42c Mon Sep 17 00:00:00 2001
From: omark96 <43138339+omark96@users.noreply.github.com>
Date: Wed, 23 Jul 2025 13:15:13 +0200
Subject: [PATCH] fix(wm): move last_focused_workspace logic to focus_workspace
 method

There are currently a number of commands that do not update the previously focused workspace
correctly. The previous method relied on handling each case, this change makes it so that
last_focused_workspace is always updated whenever the focus is changed.
---
 komorebi/src/monitor.rs         |  2 +-
 komorebi/src/process_command.rs | 19 -------------------
 2 files changed, 1 insertion(+), 20 deletions(-)

diff --git a/komorebi/src/monitor.rs b/komorebi/src/monitor.rs
index ac0efbc7..3eef9ffe 100644
--- a/komorebi/src/monitor.rs
+++ b/komorebi/src/monitor.rs
@@ -500,7 +500,7 @@ impl Monitor {
             if workspaces.get(idx).is_none() {
                 workspaces.resize(idx + 1, Workspace::default());
             }
-
+            self.set_last_focused_workspace(Some(self.workspaces.focused_idx()));
             self.workspaces.focus(idx);
         }
 
diff --git a/komorebi/src/process_command.rs b/komorebi/src/process_command.rs
index 9eaa6177..b01be330 100644
--- a/komorebi/src/process_command.rs
+++ b/komorebi/src/process_command.rs
@@ -209,25 +209,6 @@ impl WindowManager {
         // We don't have From implemented for &mut WindowManager
         let initial_state = State::from(self.as_ref());
 
-        match message {
-            SocketMessage::CycleFocusEmptyWorkspace(_)
-            | SocketMessage::CycleFocusWorkspace(_)
-            | SocketMessage::FocusWorkspaceNumber(_) => {
-                if let Some(monitor) = self.focused_monitor_mut() {
-                    let idx = monitor.focused_workspace_idx();
-                    monitor.set_last_focused_workspace(Option::from(idx));
-                }
-            }
-            SocketMessage::FocusMonitorWorkspaceNumber(target_monitor_idx, _) => {
-                let idx = self.focused_workspace_idx_for_monitor_idx(target_monitor_idx)?;
-                if let Some(monitor) = self.monitors_mut().get_mut(target_monitor_idx) {
-                    monitor.set_last_focused_workspace(Option::from(idx));
-                }
-            }
-
-            _ => {}
-        };
-
         let mut force_update_borders = false;
         match message {
             SocketMessage::Promote => self.promote_container_to_front()?,