{-#LANGUAGE CPP #-}
{-#LANGUAGE RecordWildCards#-}

-----------------------------------------------------------------------------
-- |
-- Module      :  Plugins.Monitors.Cpu
-- Copyright   :  (c) 2011, 2017, 2022 Jose Antonio Ortega Ruiz
--                (c) 2007-2010 Andrea Rossato
-- License     :  BSD-style (see LICENSE)
--
-- Maintainer  :  Jose A. Ortega Ruiz <jao@gnu.org>
-- Stability   :  unstable
-- Portability :  unportable
--
-- A cpu monitor for Xmobar
--
-----------------------------------------------------------------------------

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)] -- [("Cpu: ","total","% "),("","user","%")]
    , CpuArguments -> [(String, [(String, String, String)])]
cpuAllTemplate :: ![(String, [(String, String, String)])] -- [("bar",[]),("vbar",[]),("ipat",[]),("total",[]),...]
    , 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)