Better Rofi integration

1 files changed, 12 insertions, 2 deletions

diff --git a/src/Main.hs b/src/Main.hs
index da3b4f1..f70496c 100644
--- a/src/Main.hs
+++ b/src/Main.hs
@@ -13,9 +13,10 @@ import Data.Monoid
 import Internal.XMobarLog
 import Internal.Keys
 import Internal.Layout
+import Internal.DMenu (menuCommandString)
 
 import qualified XMonad as X
-import qualified XMonad.StackSet as S
+import qualified XMonad.StackSet as W
 
 main = do
 
@@ -24,6 +25,7 @@ main = do
   let fp = homeDir </> ".xmonad" </> "startup"
 
   setEnv "SUDO_ASKPASS" "/usr/bin/ssh-askpass"
+  setEnv "ROFI" menuCommandString
 
   xmobar <- spawnXMobar
 
@@ -43,7 +45,8 @@ main = do
          , className =? "yakuake" --> doFloat
          , className =? "MPlayer" --> doFloat
          , title =? "Event Tester" --> doFloat
-         , title =? "Floating Term" --> doFloat
+         , title =? "Floating Term" --> doCenterFloat
+         , title =? "Notes" --> doCenterFloat
          , title =? "xmessage" --> doFloat
          , title =? "gxmessage" --> doFloat
          , className =? "mpv" --> doFloat
@@ -55,3 +58,10 @@ main = do
        , clickJustFocuses = False
        , logHook = xMobarLogHook xmobar
        }
+
+doCenterFloat :: ManageHook
+doCenterFloat =
+  ask >>= \w -> doF . W.float w . centerRect . snd =<< liftX (floatLocation w)
+
+centerRect :: W.RationalRect -> W.RationalRect
+centerRect (W.RationalRect x y w h) = W.RationalRect ((1 - w) / 2) ((1 - h) / 2) w h