summaryrefslogtreecommitdiffstats
path: root/W.hs
blob: 016baaa725c8a0a914b59ef6fb02ad71d78141ac (plain)
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
-----------------------------------------------------------------------------
-- |
-- Module      :  W.hs
-- Copyright   :  (c) Spencer Janssen 2007
-- License     :  BSD3-style (see LICENSE)
-- 
-- Maintainer  :  sjanssen@cse.unl.edu
-- Stability   :  unstable
-- Portability :  not portable, uses cunning newtype deriving
--
-----------------------------------------------------------------------------
--
-- The W monad, a state monad transformer over IO, for the window
-- manager state, and support routines.
--

module W where

import Data.Sequence
import Control.Monad.State
import System.IO (hFlush, hPutStrLn, stderr)
import Graphics.X11.Xlib

--
-- | WState, the window manager state.
-- Just the display, width, height and a window list
--
data WState = WState
    { display       :: Display
    , screenWidth   :: !Int
    , screenHeight  :: !Int
    , windows       :: Seq Window
    }

-- | The W monad, a StateT transformer over IO encapuslating the window
-- manager state
--
newtype W a = W (StateT WState IO a)
    deriving (Functor, Monad, MonadIO)

-- | Run the W monad, given a chunk of W monad code, and an initial state
-- Return the result, and final state
--
runW :: W a -> WState -> IO (a, WState)
runW (W m) = runStateT m

withIO :: (forall b. (a -> IO b) -> IO b) -> (a -> W c) -> W c
withIO f g = do
    s <- W get
    (y, t) <- io (f (flip runW s . g))
    W (put t)
    return y

--
-- | Lift an IO action into the W monad
--
io :: IO a -> W a
io = liftIO

--
-- | Lift an IO action into the W monad, discarding any result
--
io_ :: IO a -> W ()
io_ f = liftIO f >> return ()

--
-- | A 'trace' for the W monad
--
trace :: String -> W ()
trace msg = io $ do
    hPutStrLn stderr msg
    hFlush stderr

--
-- | Run an action forever
--
forever :: (Monad m) => m a -> m b
forever a = a >> forever a

-- ---------------------------------------------------------------------
-- Getting at the window manager state

-- | Return the current dispaly
getDisplay          :: W Display
getDisplay          = W (gets display)

-- | Return the current windows
getWindows          :: W (Seq Window)
getWindows          = W (gets windows)

-- | Return the screen width
getScreenWidth      :: W Int
getScreenWidth      = W (gets screenWidth)

-- | Return the screen height
getScreenHeight     :: W Int
getScreenHeight     = W (gets screenHeight)

-- | Set the current window list
setWindows          :: Seq Window -> W ()
setWindows x        = W (modify (\s -> s {windows = x}))

-- | Modify the current window list
modifyWindows       :: (Seq Window -> Seq Window) -> W ()
modifyWindows f     = W (modify (\s -> s {windows = f (windows s)}))