diff options
| -rw-r--r-- | src/Xmobar/Plugins/Monitors/Batt.hs | 53 |
1 files changed, 47 insertions, 6 deletions
diff --git a/src/Xmobar/Plugins/Monitors/Batt.hs b/src/Xmobar/Plugins/Monitors/Batt.hs index e29870a..fd02cf1 100644 --- a/src/Xmobar/Plugins/Monitors/Batt.hs +++ b/src/Xmobar/Plugins/Monitors/Batt.hs @@ -45,6 +45,9 @@ data BattOpts = BattOpts , onIconPattern :: Maybe IconPattern , offIconPattern :: Maybe IconPattern , idleIconPattern :: Maybe IconPattern + , lowString :: String + , mediumString :: String + , highString :: String } defaultOpts :: BattOpts @@ -65,6 +68,9 @@ defaultOpts = BattOpts , onIconPattern = Nothing , offIconPattern = Nothing , idleIconPattern = Nothing + , lowString = "" + , mediumString = "" + , highString = "" } options :: [OptDescr (BattOpts -> BattOpts)] @@ -89,6 +95,9 @@ options = o { offIconPattern = Just $ parseIconPattern x }) "") "" , Option "" ["idle-icon-pattern"] (ReqArg (\x o -> o { idleIconPattern = Just $ parseIconPattern x }) "") "" + , Option "" ["lows"] (ReqArg (\x o -> o { lowString = x }) "") "" + , Option "" ["mediums"] (ReqArg (\x o -> o { mediumString = x }) "") "" + , Option "" ["highs"] (ReqArg (\x o -> o { highString = x }) "") "" ] parseOpts :: [String] -> IO BattOpts @@ -125,6 +134,23 @@ data Battery = Battery , status :: !String } +data BatteryStatus + = BattHigh + | BattMedium + | BattLow + +-- | Convert the current battery charge into a 'BatteryStatus' +getBattStatus + :: Float -- ^ Current battery charge, assumed to be in [0,1] + -> BattOpts -- ^ Battery options, including high/low thresholds + -> BatteryStatus +getBattStatus charge opts + | c >= highThreshold opts = BattHigh + | c >= lowThreshold opts = BattMedium + | otherwise = BattLow + where + c = 100 * min 1 charge + safeFileExist :: String -> String -> IO Bool safeFileExist d f = handle noErrors $ fileExist (d </> f) where noErrors = const (return False) :: SomeException -> IO Bool @@ -226,7 +252,9 @@ runBatt' bfs args = do do l <- fmtPercent x ws <- fmtWatts w opts suffix d si <- getIconPattern opts s x - st <- showWithColors' (fmtStatus opts s nas) (100 * x) + st <- showWithColors' + (fmtStatus opts s nas (getBattStatus x opts)) + (100 * x) parseTemplate (l ++ [st, fmtTime $ floor t, ws, si]) NA -> getConfigValue naString where fmtPercent :: Float -> Monitor [String] @@ -244,11 +272,24 @@ runBatt' bfs args = do then minutes else '0' : minutes where hours = show (x `div` 3600) minutes = show ((x `mod` 3600) `div` 60) - fmtStatus opts Idle _ = idleString opts - fmtStatus _ Unknown na = na - fmtStatus opts Full _ = idleString opts - fmtStatus opts Charging _ = onString opts - fmtStatus opts Discharging _ = offString opts + fmtStatus + :: BattOpts + -> Status + -> String -- ^ What to in case battery status is unknown + -> BatteryStatus + -> String + fmtStatus opts Idle _ _ = idleString opts + fmtStatus _ Unknown na _ = na + fmtStatus opts Full _ _ = idleString opts + fmtStatus opts Charging _ _ = onString opts + fmtStatus opts Discharging _ battStatus = + (case battStatus of + BattHigh -> highString + BattMedium -> mediumString + BattLow -> lowString + ) + <> offString + $ opts maybeColor Nothing str = str maybeColor (Just c) str = "<fc=" ++ c ++ ">" ++ str ++ "</fc>" color x o | x >= 0 = maybeColor (posColor o) |