aboutsummaryrefslogtreecommitdiff
path: root/src/Rahm/Desktop/RebindKeys.hs
blob: fb8899b064c45344f1e1589b20e102304871f880 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
-- Module for intercepting key presses not explicity mapped in the key bindings.
-- This uses some deep magic with grabKey and windows and everything else, but
-- it makes window-specific key bindings awesome!
module Rahm.Desktop.RebindKeys
  ( remapHook,
    disableKey,
    remapKey,
    sendKeyQ,
    sendKey,
    rebindKey,
    WindowHook (..),
  )
where

import Control.Applicative ((<|>))
import Control.Monad (forM, forM_)
import Control.Monad.Trans.Class (lift)
import Data.Default (def)
import Data.Map (Map)
import qualified Data.Map as Map
  ( delete,
    findWithDefault,
    fromListWith,
    insert,
    lookup,
  )
import Data.Monoid (All (..))
import Rahm.Desktop.Keys.Grab (KeySymOrKeyCode (..), doGrab, getKeyCodesForKeysym)
import XMonad
  ( Default (def),
    Display,
    Event (KeyEvent, ev_event_type, ev_keycode, ev_state, ev_window),
    ExtensionClass (initialValue),
    KeyCode,
    KeyMask,
    KeySym,
    MonadReader (ask),
    Query (..),
    Window,
    X,
    XConf (XConf, display, theRoot),
    allocaXEvent,
    displayKeycodes,
    extraModifiers,
    grabKey,
    grabModeAsync,
    io,
    keyPress,
    keyPressMask,
    keyRelease,
    keyReleaseMask,
    keycodeToKeysym,
    liftX,
    noSymbol,
    none,
    sendEvent,
    setEventType,
    setKeyEvent,
    (.|.),
  )
import qualified XMonad.Util.ExtensibleState as XS (get, modify)

type WindowHook = Query ()

newtype InterceptState
  = InterceptState (Map (KeyMask, KeySymOrKeyCode) (X ()))

newtype RemapState
  = RemapState (Map (Window, (KeyMask, KeySymOrKeyCode)) (X ()))

instance ExtensionClass InterceptState where
  initialValue = InterceptState def

instance ExtensionClass RemapState where
  initialValue = RemapState def

remapHook :: Event -> X All
remapHook event = do
  RemapState map <- XS.get

  case event of
    KeyEvent {ev_window = win, ev_event_type = typ, ev_keycode = code, ev_state = m}
      | typ == keyPress -> do
        XConf {display = dpy, theRoot = rootw} <- ask
        keysym <- io $ keycodeToKeysym dpy code 0

        case Map.lookup (win, (m, Kc code)) map
          <|> Map.lookup (win, (m, Ks keysym)) map of
          Just xdo -> do
            xdo
            return (All False)
          Nothing -> return (All True)
    _ -> return (All True)

disableKey :: (KeyMask, KeySym) -> WindowHook
disableKey key = remapKey (fmap Ks key) (return ())

remapKey :: (KeyMask, KeySymOrKeyCode) -> X () -> WindowHook
remapKey keyFrom action = do
  window <- ask
  Query $
    lift $ do
      XConf {display = disp, theRoot = rootw} <- ask
      doGrab disp window keyFrom

      XS.modify $ \(RemapState keyMap) ->
        RemapState $
          Map.insert (window, keyFrom) action keyMap

-- sendKey, but as a query.
sendKeyQ :: (KeyMask, KeySym) -> Query ()
sendKeyQ key = do
  win <- ask
  liftX (sendKey key win)

sendKey :: (KeyMask, KeySym) -> Window -> X ()
sendKey (keymask, keysym) w = do
  XConf {display = disp, theRoot = rootw} <- ask

  codes <- io $ getKeyCodesForKeysym disp keysym

  case codes of
    (keycode : _) ->
      io $
        allocaXEvent $ \xEv -> do
          setEventType xEv keyPress
          setKeyEvent xEv w rootw none keymask keycode True
          sendEvent disp w True keyPressMask xEv

          setEventType xEv keyRelease
          sendEvent disp w True keyReleaseMask xEv
    _ -> return ()

rebindKey :: (KeyMask, KeySym) -> (KeyMask, KeySym) -> WindowHook
rebindKey keyFrom keyTo =
  (remapKey (fmap Ks keyFrom) . sendKey keyTo) =<< ask