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
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
|
{-# HLINT ignore "Use camelCase" #-}
module Wetterhorn.Core
( WState (..),
WConfig (..),
SurfaceState (..),
W,
getWConfig,
getWState,
runW,
Wetterhorn,
initWetterhorn,
wio,
incrementState,
readWState,
defaultConfig,
requestHotReload,
ctxConfig,
KeyEvent (..),
KeyState (..),
)
where
import Control.Arrow (first)
import Control.Exception
import Control.Monad (when)
import Data.ByteString (ByteString)
import qualified Data.ByteString.Char8 as CH
import Data.Map (Map)
import qualified Data.Map as Map
import Data.Maybe (fromMaybe)
import Foreign (Ptr, StablePtr, Word32, castForeignPtr, newStablePtr, ptrToIntPtr)
import Numeric (showHex)
import Text.Printf
import Wetterhorn.Core.ForeignInterface (ForeignInterface)
import qualified Wetterhorn.Core.ForeignInterface as ForeignInterface
import Wetterhorn.WlRoots (WlrInputDevice, wlrSeatKeyboardNotifyKey, wlrSeatSetKeyboard)
import Data.Char (chr, ord)
data WContext = WContext
{ ctxForeignInterface :: ForeignInterface,
ctxConfig :: WConfig
}
-- This is the OpaqueState passed to the harness.
type Wetterhorn = StablePtr (WContext, WState)
requestHotReload :: W ()
requestHotReload = do
fi <- ctxForeignInterface <$> getWContext
wio $ ForeignInterface.requestHotReload fi
requestLog :: String -> W ()
requestLog str = do
fi <- ctxForeignInterface <$> getWContext
wio $ ForeignInterface.requestLog fi str
requestExit :: Int -> W ()
requestExit ec = do
fi <- ctxForeignInterface <$> getWContext
wio $ ForeignInterface.requestExit fi ec
initWetterhorn :: WConfig -> IO Wetterhorn
initWetterhorn conf = do
foreignInterface <- ForeignInterface.getForeignInterface
newStablePtr (WContext foreignInterface conf, WState "this is a string" 0)
data WState = WState
{ someString :: String,
integer :: Int
}
deriving (Show, Read)
data SurfaceState = Map | Unmap | Destroy deriving (Eq, Ord, Show, Enum)
data KeyState = KeyPressed | KeyReleased deriving (Show, Read, Eq, Enum, Ord)
data KeyEvent = KeyEvent
{ timeMs :: Word32,
keycode :: Word32,
state :: KeyState,
modifiers :: Word32,
keysym :: Word32,
codepoint :: Char,
device :: Ptr WlrInputDevice
}
deriving (Show, Ord, Eq)
data WConfig = WConfig
{ keybindingHandler :: KeyEvent -> W Bool,
surfaceHandler :: SurfaceState -> Ptr () -> W ()
}
defaultBindings :: Map (KeyState, Word32, Word32) (W ())
defaultBindings =
Map.fromList
[ ((KeyPressed, 0x8, sym 'r'), requestHotReload),
((KeyPressed, 0x8, sym 'l'), requestLog "This is a log statement!\n"),
((KeyPressed, 0x8, sym 't'), wio $ ForeignInterface.doShellExec "alacritty"),
((KeyPressed, 0x9, sym 'T'), wio $ ForeignInterface.doShellExec "gnome-terminal"),
((KeyPressed, 0x8, sym 'c'), wio $ ForeignInterface.doShellExec "pavucontrol"),
((KeyPressed, 0x8, sym 'q'), requestExit 0)
]
where
sym = fromIntegral . ord
defaultConfig :: WConfig
defaultConfig =
WConfig
{ keybindingHandler = \keyEvent -> do
i <- incrementState
seatPtr <- (wio . ForeignInterface.getSeat . ctxForeignInterface) =<< getWContext
wio $ printf "%d - Got %s\n" i (show keyEvent)
maybe
( wio $ do
wlrSeatSetKeyboard seatPtr (device keyEvent)
wlrSeatKeyboardNotifyKey
seatPtr
(timeMs keyEvent)
(keycode keyEvent)
( case state keyEvent of
KeyReleased -> 0
_ -> 1
)
return True
)
(fmap (const True))
$ Map.lookup
(state keyEvent, modifiers keyEvent, keysym keyEvent)
defaultBindings,
surfaceHandler = \state ptr -> wio (printf "Surface %s is %s\n" (showHex (ptrToIntPtr ptr) "") (show state))
}
readWState :: ByteString -> IO WState
readWState bs =
catch
(return $ read (CH.unpack bs))
( \e ->
let _ = (e :: SomeException) in return (WState "" 0)
)
newtype W a = W ((WContext, WState) -> IO (a, WState))
instance Functor W where
fmap mfn (W fn) = W $ fmap (first mfn) <$> fn
instance Applicative W where
pure a = W $ \(_, s) -> return (a, s)
mfn <*> ma = do
fn <- mfn
fn <$> ma
instance Monad W where
(W fntoa) >>= fnmb = W $ \(config, state) -> do
(a, state') <- fntoa (config, state)
let W fntob = fnmb a
fntob (config, state')
getWContext :: W WContext
getWContext = W pure
getWConfig :: W WConfig
getWConfig = ctxConfig <$> getWContext
getWState :: W WState
getWState = W $ \(_, s) -> pure (s, s)
runW :: W a -> (WContext, WState) -> IO (a, WState)
runW (W fn) = fn
incrementState :: W Int
incrementState = W $ \(_, WState s i) -> return (i, WState s (i + 1))
wio :: IO a -> W a
wio fn = W $ \(_, b) -> fn >>= \a -> return (a, b)
|
