1 files changed, 28 insertions, 0 deletions
diff --git a/src/Rahm/Desktop/Common.hs b/src/Rahm/Desktop/Common.hs
index 5c29a1c..71ffad0 100644
--- a/src/Rahm/Desktop/Common.hs
+++ b/src/Rahm/Desktop/Common.hs
@@ -2,6 +2,7 @@ module Rahm.Desktop.Common
   ( focusLocation,
     masterWindow,
     windowsInWorkspace,
+    pointerWorkspace,
     getString,
     askWindowId,
     windowJump,
@@ -17,6 +18,9 @@ module Rahm.Desktop.Common
     runMaybeT_,
     setBorderColor,
     click,
+    pointerLocation,
+    pointerWindow,
+    getDisplayAndRoot,
     Location (..),
   )
 where
@@ -225,3 +229,27 @@ click = do
   (dpy, root) <- asks $ (,) <$> display <*> X.theRoot
   (_, _, window, _, _, _, _, _) <- io $ X.queryPointer dpy root
   focus window
+
+getDisplayAndRoot :: X (X.Display, X.Window)
+getDisplayAndRoot = X.asks $ (,) <$> X.display <*> X.theRoot
+
+pointerLocation :: (Integral a, Integral b) => X (a, b)
+pointerLocation = do
+  (dpy, root) <- getDisplayAndRoot
+  (_, _, _, fromIntegral -> x, fromIntegral -> y, _, _, _) <-
+    io $ X.queryPointer dpy root
+  return (x, y)
+
+pointerWindow :: X X.Window
+pointerWindow = do
+  (dpy, root) <- getDisplayAndRoot
+  (_, _, w, _, _, _, _, _) <-
+    io $ X.queryPointer dpy root
+  return w
+
+pointerWorkspace :: X (Maybe WorkspaceId)
+pointerWorkspace = runMaybeT $ do
+  (x, y) <- lift pointerLocation
+  (S.Screen (S.tag -> ws1) _ _) <- MaybeT $ X.pointScreen x y
+  return ws1
+