Highlight windows for marking too

1 files changed, 9 insertions, 7 deletions

diff --git a/src/Rahm/Desktop/Keys.hs b/src/Rahm/Desktop/Keys.hs
index 69873e4..9ae9c30 100644
--- a/src/Rahm/Desktop/Keys.hs
+++ b/src/Rahm/Desktop/Keys.hs
@@ -354,14 +354,16 @@ keymap = runKeys $ do
       doc "For mosaic layout, shrink the size-share of the current window" $
       sendMessage =<< shrinkPositionAlt
 
-  bind xK_m $ do
+  bind xK_m $
     justMod $
-      doc "Mark the current window with the next typed character." $
-      runMaybeT_ $ do
-       locs <- readNextLocationSet
-       mapNextString $ \_ str -> lift $
-        case str of
-          [ch] | isAlpha ch -> markAllLocations str locs
+      doc "Mark the current window with the next typed character." $ do
+      locs <- fromMaybe [] <$> runMaybeT readNextLocationSet
+      withBorderColor "#00ffff" (mapMaybe locationWindow locs) $ do
+        runMaybeT_ $ do
+         mapNextString $ \_ str -> lift $
+          case str of
+            [ch] | isAlpha ch -> markAllLocations str locs
+            _ -> return ()
 
   bind xK_plus $ do
     justMod $