From 117d0863724bb9b883d0754b00a97db1ae6a031d Mon Sep 17 00:00:00 2001
From: Maciek Chudek <mchudek@asu.edu>
Date: Wed, 24 Nov 2021 04:15:13 -0700
Subject: [PATCH] Added option to center the mouse instead of restoring
 position. Useful option for users and improves functionality after
 adding/removing monitors with xrandr.

---
 lib/awful/screen.lua | 31 +++++++++++++++++++------------
 1 file changed, 19 insertions(+), 12 deletions(-)

diff --git a/lib/awful/screen.lua b/lib/awful/screen.lua
index 8e444e521..d36c6dbef 100644
--- a/lib/awful/screen.lua
+++ b/lib/awful/screen.lua
@@ -93,27 +93,34 @@ end
 -- @tparam screen _screen Screen number (defaults / falls back to mouse.screen).
 -- @request client activate screen.focus granted The most recent focused client
 --  for this screen should be re-activated.
-function screen.focus(_screen)
+function screen.focus(_screen, ...)
     client = client or require("awful.client")
     if type(_screen) == "number" and _screen > capi.screen.count() then _screen = screen.focused() end
     _screen = get_screen(_screen)
+    if select('#', ...) == 0 then center_mouse = false else center_mouse = select(1, ...) or false end
 
     -- screen and pos for current screen
     local s = get_screen(capi.mouse.screen)
     local pos
-
-    if not _screen.mouse_per_screen then
-        -- This is the first time we enter this screen,
-        -- keep relative mouse position on the new screen.
+    -- center the mouse if requested, else restore previous position
+    if center_mouse then
         pos = capi.mouse.coords()
-        local relx = (pos.x - s.geometry.x) / s.geometry.width
-        local rely = (pos.y - s.geometry.y) / s.geometry.height
-
-        pos.x = _screen.geometry.x + relx * _screen.geometry.width
-        pos.y = _screen.geometry.y + rely * _screen.geometry.height
+        pos.x = _screen.geometry.x +  _screen.geometry.width/2
+        pos.y = _screen.geometry.y +  _screen.geometry.height/2
     else
-        -- restore mouse position
-        pos = _screen.mouse_per_screen
+        if not _screen.mouse_per_screen then
+            -- This is the first time we enter this screen,
+            -- keep relative mouse position on the new screen.
+            pos = capi.mouse.coords()
+            local relx = (pos.x - s.geometry.x) / s.geometry.width
+            local rely = (pos.y - s.geometry.y) / s.geometry.height
+
+            pos.x = _screen.geometry.x + relx * _screen.geometry.width
+            pos.y = _screen.geometry.y + rely * _screen.geometry.height
+        else
+            -- restore mouse position
+            pos = _screen.mouse_per_screen
+        end
     end
 
     -- save pointer position of current screen