{-#LANGUAGE CPP #-}
{-#LANGUAGE RecordWildCards#-}
module Xmobar.Plugins.Monitors.Cpu
( startCpu
, runCpu
, cpuConfig
, MC.CpuDataRef
, CpuOpts
, CpuArguments
, MC.parseCpu
, getArguments
) where
import Xmobar.Plugins.Monitors.Common
import Data.IORef (newIORef)
import System.Console.GetOpt
import Xmobar.Run.Timer (doEveryTenthSeconds)
import Control.Monad (void)
import Xmobar.Plugins.Monitors.Cpu.Common (CpuData(..))
#if defined(freebsd_HOST_OS)
import qualified Xmobar.Plugins.Monitors.Cpu.FreeBSD as MC
#else
import qualified Xmobar.Plugins.Monitors.Cpu.Linux as MC
#endif
newtype CpuOpts = CpuOpts
{ CpuOpts -> Maybe IconPattern
loadIconPattern :: Maybe IconPattern
}
defaultOpts :: CpuOpts
defaultOpts :: CpuOpts
defaultOpts = CpuOpts
{ loadIconPattern :: Maybe IconPattern
loadIconPattern = forall a. Maybe a
Nothing
}
options :: [OptDescr (CpuOpts -> CpuOpts)]
options :: [OptDescr (CpuOpts -> CpuOpts)]
options =
[ forall a. String -> [String] -> ArgDescr a -> String -> OptDescr a
Option String
"" [String
"load-icon-pattern"] (forall a. (String -> a) -> String -> ArgDescr a
ReqArg (\String
x CpuOpts
o ->
CpuOpts
o { loadIconPattern :: Maybe IconPattern
loadIconPattern = forall a. a -> Maybe a
Just forall a b. (a -> b) -> a -> b
$ String -> IconPattern
parseIconPattern String
x }) String
"") String
""
]
barField :: String
barField :: String
barField = String
"bar"
vbarField :: String
vbarField :: String
vbarField = String
"vbar"
ipatField :: String
ipatField :: String
ipatField = String
"ipat"
totalField :: String
totalField :: String
totalField = String
"total"
userField :: String
userField :: String
userField = String
"user"
niceField :: String
niceField :: String
niceField = String
"nice"
systemField :: String
systemField :: String
systemField = String
"system"
idleField :: String
idleField :: String
idleField = String
"idle"
iowaitField :: String
iowaitField :: String
iowaitField = String
"iowait"
cpuConfig :: IO MConfig
cpuConfig :: IO MConfig
cpuConfig =
String -> [String] -> IO MConfig
mkMConfig
String
"Cpu: <total>%"
[ String
barField
, String
vbarField
, String
ipatField
, String
totalField
, String
userField
, String
niceField
, String
systemField
, String
idleField
, String
iowaitField
]
data Field = Field {
Field -> String
fieldName :: !String,
Field -> ShouldCompute
fieldCompute :: !ShouldCompute
} deriving (Field -> Field -> Bool
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: Field -> Field -> Bool
$c/= :: Field -> Field -> Bool
== :: Field -> Field -> Bool
$c== :: Field -> Field -> Bool
Eq, Eq Field
Field -> Field -> Bool
Field -> Field -> Ordering
Field -> Field -> Field
forall a.
Eq a
-> (a -> a -> Ordering)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> a)
-> (a -> a -> a)
-> Ord a
min :: Field -> Field -> Field
$cmin :: Field -> Field -> Field
max :: Field -> Field -> Field
$cmax :: Field -> Field -> Field
>= :: Field -> Field -> Bool
$c>= :: Field -> Field -> Bool
> :: Field -> Field -> Bool
$c> :: Field -> Field -> Bool
<= :: Field -> Field -> Bool
$c<= :: Field -> Field -> Bool
< :: Field -> Field -> Bool
$c< :: Field -> Field -> Bool
compare :: Field -> Field -> Ordering
$ccompare :: Field -> Field -> Ordering
Ord, Int -> Field -> ShowS
[Field] -> ShowS
Field -> String
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [Field] -> ShowS
$cshowList :: [Field] -> ShowS
show :: Field -> String
$cshow :: Field -> String
showsPrec :: Int -> Field -> ShowS
$cshowsPrec :: Int -> Field -> ShowS
Show)
data ShouldCompute = Compute | Skip deriving (ShouldCompute -> ShouldCompute -> Bool
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: ShouldCompute -> ShouldCompute -> Bool
$c/= :: ShouldCompute -> ShouldCompute -> Bool
== :: ShouldCompute -> ShouldCompute -> Bool
$c== :: ShouldCompute -> ShouldCompute -> Bool
Eq, Eq ShouldCompute
ShouldCompute -> ShouldCompute -> Bool
ShouldCompute -> ShouldCompute -> Ordering
ShouldCompute -> ShouldCompute -> ShouldCompute
forall a.
Eq a
-> (a -> a -> Ordering)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> a)
-> (a -> a -> a)
-> Ord a
min :: ShouldCompute -> ShouldCompute -> ShouldCompute
$cmin :: ShouldCompute -> ShouldCompute -> ShouldCompute
max :: ShouldCompute -> ShouldCompute -> ShouldCompute
$cmax :: ShouldCompute -> ShouldCompute -> ShouldCompute
>= :: ShouldCompute -> ShouldCompute -> Bool
$c>= :: ShouldCompute -> ShouldCompute -> Bool
> :: ShouldCompute -> ShouldCompute -> Bool
$c> :: ShouldCompute -> ShouldCompute -> Bool
<= :: ShouldCompute -> ShouldCompute -> Bool
$c<= :: ShouldCompute -> ShouldCompute -> Bool
< :: ShouldCompute -> ShouldCompute -> Bool
$c< :: ShouldCompute -> ShouldCompute -> Bool
compare :: ShouldCompute -> ShouldCompute -> Ordering
$ccompare :: ShouldCompute -> ShouldCompute -> Ordering
Ord, Int -> ShouldCompute -> ShowS
[ShouldCompute] -> ShowS
ShouldCompute -> String
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [ShouldCompute] -> ShowS
$cshowList :: [ShouldCompute] -> ShowS
show :: ShouldCompute -> String
$cshow :: ShouldCompute -> String
showsPrec :: Int -> ShouldCompute -> ShowS
$cshowsPrec :: Int -> ShouldCompute -> ShowS
Show)
formatField :: MonitorConfig -> CpuOpts -> CpuData -> Field -> IO String
formatField :: MonitorConfig -> CpuOpts -> CpuData -> Field -> IO String
formatField MonitorConfig
cpuParams CpuOpts
cpuOpts cpuInfo :: CpuData
cpuInfo@CpuData {Float
cpuTotal :: CpuData -> Float
cpuIowait :: CpuData -> Float
cpuIdle :: CpuData -> Float
cpuSystem :: CpuData -> Float
cpuNice :: CpuData -> Float
cpuUser :: CpuData -> Float
cpuTotal :: Float
cpuIowait :: Float
cpuIdle :: Float
cpuSystem :: Float
cpuNice :: Float
cpuUser :: Float
..} Field {String
ShouldCompute
fieldCompute :: ShouldCompute
fieldName :: String
fieldCompute :: Field -> ShouldCompute
fieldName :: Field -> String
..}
| String
fieldName forall a. Eq a => a -> a -> Bool
== String
barField =
if ShouldCompute
fieldCompute forall a. Eq a => a -> a -> Bool
== ShouldCompute
Compute
then forall (m :: * -> *).
MonadIO m =>
MonitorConfig -> Float -> Float -> m String
pShowPercentBar MonitorConfig
cpuParams (Float
100 forall a. Num a => a -> a -> a
* Float
cpuTotal) Float
cpuTotal
else forall (f :: * -> *) a. Applicative f => a -> f a
pure []
| String
fieldName forall a. Eq a => a -> a -> Bool
== String
vbarField =
if ShouldCompute
fieldCompute forall a. Eq a => a -> a -> Bool
== ShouldCompute
Compute
then forall (m :: * -> *).
MonadIO m =>
MonitorConfig -> Float -> Float -> m String
pShowVerticalBar MonitorConfig
cpuParams (Float
100 forall a. Num a => a -> a -> a
* Float
cpuTotal) Float
cpuTotal
else forall (f :: * -> *) a. Applicative f => a -> f a
pure []
| String
fieldName forall a. Eq a => a -> a -> Bool
== String
ipatField =
if ShouldCompute
fieldCompute forall a. Eq a => a -> a -> Bool
== ShouldCompute
Compute
then Maybe IconPattern -> Float -> IO String
pShowIconPattern (CpuOpts -> Maybe IconPattern
loadIconPattern CpuOpts
cpuOpts) Float
cpuTotal
else forall (f :: * -> *) a. Applicative f => a -> f a
pure []
| Bool
otherwise =
if ShouldCompute
fieldCompute forall a. Eq a => a -> a -> Bool
== ShouldCompute
Compute
then forall (m :: * -> *).
MonadIO m =>
MonitorConfig -> Float -> m String
pShowPercentWithColors MonitorConfig
cpuParams (String -> CpuData -> Float
getFieldValue String
fieldName CpuData
cpuInfo)
else forall (f :: * -> *) a. Applicative f => a -> f a
pure []
getFieldValue :: String -> CpuData -> Float
getFieldValue :: String -> CpuData -> Float
getFieldValue String
field CpuData{Float
cpuTotal :: Float
cpuIowait :: Float
cpuIdle :: Float
cpuSystem :: Float
cpuNice :: Float
cpuUser :: Float
cpuTotal :: CpuData -> Float
cpuIowait :: CpuData -> Float
cpuIdle :: CpuData -> Float
cpuSystem :: CpuData -> Float
cpuNice :: CpuData -> Float
cpuUser :: CpuData -> Float
..}
| String
field forall a. Eq a => a -> a -> Bool
== String
barField = Float
cpuTotal
| String
field forall a. Eq a => a -> a -> Bool
== String
vbarField = Float
cpuTotal
| String
field forall a. Eq a => a -> a -> Bool
== String
ipatField = Float
cpuTotal
| String
field forall a. Eq a => a -> a -> Bool
== String
totalField = Float
cpuTotal
| String
field forall a. Eq a => a -> a -> Bool
== String
userField = Float
cpuUser
| String
field forall a. Eq a => a -> a -> Bool
== String
niceField = Float
cpuNice
| String
field forall a. Eq a => a -> a -> Bool
== String
systemField = Float
cpuSystem
| String
field forall a. Eq a => a -> a -> Bool
== String
idleField = Float
cpuIdle
| Bool
otherwise = Float
cpuIowait
computeFields :: [String] -> [String] -> [Field]
computeFields :: [String] -> [String] -> [Field]
computeFields [] [String]
_ = []
computeFields (String
x:[String]
xs) [String]
inputFields =
if String
x forall (t :: * -> *) a. (Foldable t, Eq a) => a -> t a -> Bool
`elem` [String]
inputFields
then (Field {fieldName :: String
fieldName = String
x, fieldCompute :: ShouldCompute
fieldCompute = ShouldCompute
Compute}) forall a. a -> [a] -> [a]
:
[String] -> [String] -> [Field]
computeFields [String]
xs [String]
inputFields
else (Field {fieldName :: String
fieldName = String
x, fieldCompute :: ShouldCompute
fieldCompute = ShouldCompute
Skip}) forall a. a -> [a] -> [a]
:
[String] -> [String] -> [Field]
computeFields [String]
xs [String]
inputFields
formatCpu :: CpuArguments -> CpuData -> IO [String]
formatCpu :: CpuArguments -> CpuData -> IO [String]
formatCpu CpuArguments{[String]
[(String, [(String, String, String)])]
[(String, String, String)]
[Field]
CpuDataRef
MonitorConfig
CpuOpts
cpuFields :: CpuArguments -> [Field]
cpuAllTemplate :: CpuArguments -> [(String, [(String, String, String)])]
cpuInputTemplate :: CpuArguments -> [(String, String, String)]
cpuOpts :: CpuArguments -> CpuOpts
cpuArgs :: CpuArguments -> [String]
cpuParams :: CpuArguments -> MonitorConfig
cpuDataRef :: CpuArguments -> CpuDataRef
cpuFields :: [Field]
cpuAllTemplate :: [(String, [(String, String, String)])]
cpuInputTemplate :: [(String, String, String)]
cpuOpts :: CpuOpts
cpuArgs :: [String]
cpuParams :: MonitorConfig
cpuDataRef :: CpuDataRef
..} CpuData
cpuInfo = do
[String]
strs <- forall (t :: * -> *) (m :: * -> *) a b.
(Traversable t, Monad m) =>
(a -> m b) -> t a -> m (t b)
mapM (MonitorConfig -> CpuOpts -> CpuData -> Field -> IO String
formatField MonitorConfig
cpuParams CpuOpts
cpuOpts CpuData
cpuInfo) [Field]
cpuFields
forall (f :: * -> *) a. Applicative f => a -> f a
pure forall a b. (a -> b) -> a -> b
$ forall a. (a -> Bool) -> [a] -> [a]
filter (Bool -> Bool
not forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall (t :: * -> *) a. Foldable t => t a -> Bool
null) [String]
strs
getInputFields :: CpuArguments -> [String]
getInputFields :: CpuArguments -> [String]
getInputFields CpuArguments{[String]
[(String, [(String, String, String)])]
[(String, String, String)]
[Field]
CpuDataRef
MonitorConfig
CpuOpts
cpuFields :: [Field]
cpuAllTemplate :: [(String, [(String, String, String)])]
cpuInputTemplate :: [(String, String, String)]
cpuOpts :: CpuOpts
cpuArgs :: [String]
cpuParams :: MonitorConfig
cpuDataRef :: CpuDataRef
cpuFields :: CpuArguments -> [Field]
cpuAllTemplate :: CpuArguments -> [(String, [(String, String, String)])]
cpuInputTemplate :: CpuArguments -> [(String, String, String)]
cpuOpts :: CpuArguments -> CpuOpts
cpuArgs :: CpuArguments -> [String]
cpuParams :: CpuArguments -> MonitorConfig
cpuDataRef :: CpuArguments -> CpuDataRef
..} = forall a b. (a -> b) -> [a] -> [b]
map (\(String
_,String
f,String
_) -> String
f) [(String, String, String)]
cpuInputTemplate
optimizeAllTemplate :: CpuArguments -> CpuArguments
optimizeAllTemplate :: CpuArguments -> CpuArguments
optimizeAllTemplate args :: CpuArguments
args@CpuArguments {[String]
[(String, [(String, String, String)])]
[(String, String, String)]
[Field]
CpuDataRef
MonitorConfig
CpuOpts
cpuFields :: [Field]
cpuAllTemplate :: [(String, [(String, String, String)])]
cpuInputTemplate :: [(String, String, String)]
cpuOpts :: CpuOpts
cpuArgs :: [String]
cpuParams :: MonitorConfig
cpuDataRef :: CpuDataRef
cpuFields :: CpuArguments -> [Field]
cpuAllTemplate :: CpuArguments -> [(String, [(String, String, String)])]
cpuInputTemplate :: CpuArguments -> [(String, String, String)]
cpuOpts :: CpuArguments -> CpuOpts
cpuArgs :: CpuArguments -> [String]
cpuParams :: CpuArguments -> MonitorConfig
cpuDataRef :: CpuArguments -> CpuDataRef
..} =
let inputFields :: [String]
inputFields = CpuArguments -> [String]
getInputFields CpuArguments
args
allTemplates :: [(String, [(String, String, String)])]
allTemplates =
forall a. (a -> Bool) -> [a] -> [a]
filter (\(String
field, [(String, String, String)]
_) -> String
field forall (t :: * -> *) a. (Foldable t, Eq a) => a -> t a -> Bool
`elem` [String]
inputFields) [(String, [(String, String, String)])]
cpuAllTemplate
in CpuArguments
args {cpuAllTemplate :: [(String, [(String, String, String)])]
cpuAllTemplate = [(String, [(String, String, String)])]
allTemplates}
data CpuArguments =
CpuArguments
{ CpuArguments -> CpuDataRef
cpuDataRef :: !MC.CpuDataRef
, CpuArguments -> MonitorConfig
cpuParams :: !MonitorConfig
, CpuArguments -> [String]
cpuArgs :: ![String]
, CpuArguments -> CpuOpts
cpuOpts :: !CpuOpts
, CpuArguments -> [(String, String, String)]
cpuInputTemplate :: ![(String, String, String)]
, CpuArguments -> [(String, [(String, String, String)])]
cpuAllTemplate :: ![(String, [(String, String, String)])]
, CpuArguments -> [Field]
cpuFields :: ![Field]
}
getArguments :: [String] -> IO CpuArguments
getArguments :: [String] -> IO CpuArguments
getArguments [String]
cpuArgs = do
[Int]
initCpuData <- IO [Int]
MC.cpuData
CpuDataRef
cpuDataRef <- forall a. a -> IO (IORef a)
newIORef [Int]
initCpuData
forall (f :: * -> *) a. Functor f => f a -> f ()
void forall a b. (a -> b) -> a -> b
$ CpuDataRef -> IO CpuData
MC.parseCpu CpuDataRef
cpuDataRef
MonitorConfig
cpuParams <- [String] -> IO MConfig -> IO MonitorConfig
computeMonitorConfig [String]
cpuArgs IO MConfig
cpuConfig
[(String, String, String)]
cpuInputTemplate <- MonitorConfig -> IO [(String, String, String)]
runTemplateParser MonitorConfig
cpuParams
[(String, [(String, String, String)])]
cpuAllTemplate <- [String] -> IO [(String, [(String, String, String)])]
runExportParser (MonitorConfig -> [String]
pExport MonitorConfig
cpuParams)
[String]
nonOptions <-
case forall a.
ArgOrder a -> [OptDescr a] -> [String] -> ([a], [String], [String])
getOpt forall a. ArgOrder a
Permute [OptDescr Opts]
pluginOptions [String]
cpuArgs of
([Opts]
_, [String]
n, []) -> forall (f :: * -> *) a. Applicative f => a -> f a
pure [String]
n
([Opts]
_, [String]
_, [String]
errs) -> forall a. HasCallStack => String -> a
error forall a b. (a -> b) -> a -> b
$ String
"getArguments: " forall a. Semigroup a => a -> a -> a
<> forall a. Show a => a -> String
show [String]
errs
CpuOpts
cpuOpts <-
case forall a.
ArgOrder a -> [OptDescr a] -> [String] -> ([a], [String], [String])
getOpt forall a. ArgOrder a
Permute [OptDescr (CpuOpts -> CpuOpts)]
options [String]
nonOptions of
([CpuOpts -> CpuOpts]
o, [String]
_, []) -> forall (f :: * -> *) a. Applicative f => a -> f a
pure forall a b. (a -> b) -> a -> b
$ forall (t :: * -> *) a b.
Foldable t =>
(a -> b -> b) -> b -> t a -> b
foldr forall a. a -> a
id CpuOpts
defaultOpts [CpuOpts -> CpuOpts]
o
([CpuOpts -> CpuOpts]
_, [String]
_, [String]
errs) -> forall a. HasCallStack => String -> a
error forall a b. (a -> b) -> a -> b
$ String
"getArguments options: " forall a. Semigroup a => a -> a -> a
<> forall a. Show a => a -> String
show [String]
errs
let cpuFields :: [Field]
cpuFields =
[String] -> [String] -> [Field]
computeFields
(forall a b. (a -> b) -> [a] -> [b]
map forall a b. (a, b) -> a
fst [(String, [(String, String, String)])]
cpuAllTemplate)
(forall a b. (a -> b) -> [a] -> [b]
map (\(String
_, String
f, String
_) -> String
f) [(String, String, String)]
cpuInputTemplate)
forall (f :: * -> *) a. Applicative f => a -> f a
pure forall a b. (a -> b) -> a -> b
$ CpuArguments -> CpuArguments
optimizeAllTemplate CpuArguments {[String]
[(String, [(String, String, String)])]
[(String, String, String)]
[Field]
CpuDataRef
MonitorConfig
CpuOpts
cpuFields :: [Field]
cpuOpts :: CpuOpts
cpuAllTemplate :: [(String, [(String, String, String)])]
cpuInputTemplate :: [(String, String, String)]
cpuParams :: MonitorConfig
cpuDataRef :: CpuDataRef
cpuArgs :: [String]
cpuFields :: [Field]
cpuAllTemplate :: [(String, [(String, String, String)])]
cpuInputTemplate :: [(String, String, String)]
cpuOpts :: CpuOpts
cpuArgs :: [String]
cpuParams :: MonitorConfig
cpuDataRef :: CpuDataRef
..}
runCpu :: CpuArguments -> IO String
runCpu :: CpuArguments -> IO String
runCpu args :: CpuArguments
args@CpuArguments {[String]
[(String, [(String, String, String)])]
[(String, String, String)]
[Field]
CpuDataRef
MonitorConfig
CpuOpts
cpuFields :: [Field]
cpuAllTemplate :: [(String, [(String, String, String)])]
cpuInputTemplate :: [(String, String, String)]
cpuOpts :: CpuOpts
cpuArgs :: [String]
cpuParams :: MonitorConfig
cpuDataRef :: CpuDataRef
cpuFields :: CpuArguments -> [Field]
cpuAllTemplate :: CpuArguments -> [(String, [(String, String, String)])]
cpuInputTemplate :: CpuArguments -> [(String, String, String)]
cpuOpts :: CpuArguments -> CpuOpts
cpuArgs :: CpuArguments -> [String]
cpuParams :: CpuArguments -> MonitorConfig
cpuDataRef :: CpuArguments -> CpuDataRef
..} = do
CpuData
cpuValue <- CpuDataRef -> IO CpuData
MC.parseCpu CpuDataRef
cpuDataRef
[String]
temMonitorValues <- CpuArguments -> CpuData -> IO [String]
formatCpu CpuArguments
args CpuData
cpuValue
let templateInput :: TemplateInput
templateInput =
TemplateInput
{ temInputTemplate :: [(String, String, String)]
temInputTemplate = [(String, String, String)]
cpuInputTemplate
, temAllTemplate :: [(String, [(String, String, String)])]
temAllTemplate = [(String, [(String, String, String)])]
cpuAllTemplate
, [String]
temMonitorValues :: [String]
temMonitorValues :: [String]
..
}
MonitorConfig -> TemplateInput -> IO String
pureParseTemplate MonitorConfig
cpuParams TemplateInput
templateInput
startCpu :: [String] -> Int -> (String -> IO ()) -> IO ()
startCpu :: [String] -> Int -> (String -> IO ()) -> IO ()
startCpu [String]
args Int
refreshRate String -> IO ()
cb = do
CpuArguments
cpuArgs <- [String] -> IO CpuArguments
getArguments [String]
args
Int -> IO () -> IO ()
doEveryTenthSeconds Int
refreshRate (CpuArguments -> IO String
runCpu CpuArguments
cpuArgs forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= String -> IO ()
cb)