{-# LANGUAGE DeriveDataTypeable, CPP #-}
module Xmobar.System.Signal where
import Data.Foldable (for_)
import Data.Typeable (Typeable)
import Control.Concurrent
import Control.Concurrent.STM
import Control.Exception
import System.Posix.Signals
import Graphics.X11.Types (Button)
import Graphics.X11.Xlib.Types (Position)
import System.IO
#ifdef DBUS
import DBus (IsVariant(..))
import Control.Monad ((>=>))
#endif
safeHead :: [a] -> Maybe a
safeHead :: forall a. [a] -> Maybe a
safeHead [] = Maybe a
forall a. Maybe a
Nothing
safeHead (a
x:[a]
_) = a -> Maybe a
forall a. a -> Maybe a
Just a
x
data WakeUp = WakeUp deriving (Int -> WakeUp -> ShowS
[WakeUp] -> ShowS
WakeUp -> String
(Int -> WakeUp -> ShowS)
-> (WakeUp -> String) -> ([WakeUp] -> ShowS) -> Show WakeUp
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
$cshowsPrec :: Int -> WakeUp -> ShowS
showsPrec :: Int -> WakeUp -> ShowS
$cshow :: WakeUp -> String
show :: WakeUp -> String
$cshowList :: [WakeUp] -> ShowS
showList :: [WakeUp] -> ShowS
Show,Typeable)
instance Exception WakeUp
data SignalType = Wakeup
| Reposition
| ChangeScreen
| Hide Int
| Reveal Int
| Toggle Int
| SetAlpha Int
| TogglePersistent
| Action Button Position
deriving (ReadPrec [SignalType]
ReadPrec SignalType
Int -> ReadS SignalType
ReadS [SignalType]
(Int -> ReadS SignalType)
-> ReadS [SignalType]
-> ReadPrec SignalType
-> ReadPrec [SignalType]
-> Read SignalType
forall a.
(Int -> ReadS a)
-> ReadS [a] -> ReadPrec a -> ReadPrec [a] -> Read a
$creadsPrec :: Int -> ReadS SignalType
readsPrec :: Int -> ReadS SignalType
$creadList :: ReadS [SignalType]
readList :: ReadS [SignalType]
$creadPrec :: ReadPrec SignalType
readPrec :: ReadPrec SignalType
$creadListPrec :: ReadPrec [SignalType]
readListPrec :: ReadPrec [SignalType]
Read, Int -> SignalType -> ShowS
[SignalType] -> ShowS
SignalType -> String
(Int -> SignalType -> ShowS)
-> (SignalType -> String)
-> ([SignalType] -> ShowS)
-> Show SignalType
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
$cshowsPrec :: Int -> SignalType -> ShowS
showsPrec :: Int -> SignalType -> ShowS
$cshow :: SignalType -> String
show :: SignalType -> String
$cshowList :: [SignalType] -> ShowS
showList :: [SignalType] -> ShowS
Show)
#ifdef DBUS
instance IsVariant SignalType where
toVariant :: SignalType -> Variant
toVariant = String -> Variant
forall a. IsVariant a => a -> Variant
toVariant (String -> Variant)
-> (SignalType -> String) -> SignalType -> Variant
forall b c a. (b -> c) -> (a -> b) -> a -> c
. SignalType -> String
forall a. Show a => a -> String
show
fromVariant :: Variant -> Maybe SignalType
fromVariant = Variant -> Maybe String
forall a. IsVariant a => Variant -> Maybe a
fromVariant (Variant -> Maybe String)
-> (String -> Maybe SignalType) -> Variant -> Maybe SignalType
forall (m :: * -> *) a b c.
Monad m =>
(a -> m b) -> (b -> m c) -> a -> m c
>=> String -> Maybe SignalType
parseSignalType
#endif
parseSignalType :: String -> Maybe SignalType
parseSignalType :: String -> Maybe SignalType
parseSignalType = ((SignalType, String) -> SignalType)
-> Maybe (SignalType, String) -> Maybe SignalType
forall a b. (a -> b) -> Maybe a -> Maybe b
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap (SignalType, String) -> SignalType
forall a b. (a, b) -> a
fst (Maybe (SignalType, String) -> Maybe SignalType)
-> (String -> Maybe (SignalType, String))
-> String
-> Maybe SignalType
forall b c a. (b -> c) -> (a -> b) -> a -> c
. [(SignalType, String)] -> Maybe (SignalType, String)
forall a. [a] -> Maybe a
safeHead ([(SignalType, String)] -> Maybe (SignalType, String))
-> ReadS SignalType -> String -> Maybe (SignalType, String)
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ReadS SignalType
forall a. Read a => ReadS a
reads
setupSignalHandler :: TMVar SignalType -> IO ()
setupSignalHandler :: TMVar SignalType -> IO ()
setupSignalHandler TMVar SignalType
tid = do
Signal -> Handler -> Maybe SignalSet -> IO Handler
installHandler Signal
sigUSR2 (IO () -> Handler
Catch (IO () -> Handler) -> IO () -> Handler
forall a b. (a -> b) -> a -> b
$ TMVar SignalType -> IO ()
updatePosHandler TMVar SignalType
tid) Maybe SignalSet
forall a. Maybe a
Nothing
Signal -> Handler -> Maybe SignalSet -> IO Handler
installHandler Signal
sigUSR1 (IO () -> Handler
Catch (IO () -> Handler) -> IO () -> Handler
forall a b. (a -> b) -> a -> b
$ TMVar SignalType -> IO ()
changeScreenHandler TMVar SignalType
tid) Maybe SignalSet
forall a. Maybe a
Nothing
() -> IO ()
forall a. a -> IO a
forall (m :: * -> *) a. Monad m => a -> m a
return ()
updatePosHandler :: TMVar SignalType -> IO ()
updatePosHandler :: TMVar SignalType -> IO ()
updatePosHandler TMVar SignalType
sig = do
STM () -> IO ()
forall a. STM a -> IO a
atomically (STM () -> IO ()) -> STM () -> IO ()
forall a b. (a -> b) -> a -> b
$ TMVar SignalType -> SignalType -> STM ()
forall a. TMVar a -> a -> STM ()
putTMVar TMVar SignalType
sig SignalType
Reposition
() -> IO ()
forall a. a -> IO a
forall (m :: * -> *) a. Monad m => a -> m a
return ()
changeScreenHandler :: TMVar SignalType -> IO ()
changeScreenHandler :: TMVar SignalType -> IO ()
changeScreenHandler TMVar SignalType
sig = do
STM () -> IO ()
forall a. STM a -> IO a
atomically (STM () -> IO ()) -> STM () -> IO ()
forall a b. (a -> b) -> a -> b
$ TMVar SignalType -> SignalType -> STM ()
forall a. TMVar a -> a -> STM ()
putTMVar TMVar SignalType
sig SignalType
ChangeScreen
() -> IO ()
forall a. a -> IO a
forall (m :: * -> *) a. Monad m => a -> m a
return ()
withDeferSignals :: IO a -> IO a
withDeferSignals :: forall a. IO a -> IO a
withDeferSignals IO a
thing = do
threadId <- IO ThreadId
myThreadId
caughtSignal <- newEmptyMVar
let signals =
(Signal -> Bool) -> [Signal] -> [Signal]
forall a. (a -> Bool) -> [a] -> [a]
filter (Bool -> Bool
not (Bool -> Bool) -> (Signal -> Bool) -> Signal -> Bool
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (Signal -> SignalSet -> Bool) -> SignalSet -> Signal -> Bool
forall a b c. (a -> b -> c) -> b -> a -> c
flip Signal -> SignalSet -> Bool
inSignalSet SignalSet
reservedSignals)
[ Signal
sigQUIT
, Signal
sigTERM
]
for_ signals $ \Signal
s ->
Signal -> Handler -> Maybe SignalSet -> IO Handler
installHandler Signal
s
(IO () -> Handler
Catch (IO () -> Handler) -> IO () -> Handler
forall a b. (a -> b) -> a -> b
$ do
MVar Signal -> Signal -> IO Bool
forall a. MVar a -> a -> IO Bool
tryPutMVar MVar Signal
caughtSignal Signal
s
Handle -> String -> IO ()
hPutStrLn Handle
stderr (String
"xmobar: Caught signal "String -> ShowS
forall a. [a] -> [a] -> [a]
++Signal -> String
forall a. Show a => a -> String
show Signal
sString -> ShowS
forall a. [a] -> [a] -> [a]
++String
"; exiting...")
ThreadId -> AsyncException -> IO ()
forall e. Exception e => ThreadId -> e -> IO ()
throwTo ThreadId
threadId AsyncException
ThreadKilled)
Maybe SignalSet
forall a. Maybe a
Nothing
thing `finally` do
s0 <- tryReadMVar caughtSignal
case s0 of
Maybe Signal
Nothing -> () -> IO ()
forall a. a -> IO a
forall (f :: * -> *) a. Applicative f => a -> f a
pure ()
Just Signal
s -> do
Signal -> Handler -> Maybe SignalSet -> IO Handler
installHandler Signal
s Handler
Default Maybe SignalSet
forall a. Maybe a
Nothing
Signal -> IO ()
raiseSignal Signal
s