summaryrefslogtreecommitdiffhomepage
path: root/Plugins/Monitors/Batt.hs
blob: 434145296040f9cbd94d46570baf568c7bcf4441 (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
-----------------------------------------------------------------------------
-- |
-- Module      :  Plugins.Monitors.Batt
-- Copyright   :  (c) Andrea Rossato
-- License     :  BSD-style (see LICENSE)
-- 
-- Maintainer  :  Andrea Rossato <andrea.rossato@unibz.it>
-- Stability   :  unstable
-- Portability :  unportable
--
-- A battery monitor for Xmobar
--
-----------------------------------------------------------------------------

module Plugins.Monitors.Batt where

import qualified Data.ByteString.Lazy.Char8 as B
import qualified Data.Map as M
import Data.Char
import Data.List
import Data.Maybe
import Control.Monad
import System.Posix.Files

import Plugins.Monitors.Common

type BattMap = M.Map String Integer

battConfig :: IO MConfig
battConfig = mkMConfig
       "Batt: <left>" -- template
       ["left"]       -- available replacements

fileB1 :: (String, String)
fileB1 = ("/proc/acpi/battery/BAT1/info", "/proc/acpi/battery/BAT1/state")

fileB2 :: (String, String)
fileB2 = ("/proc/acpi/battery/BAT2/info", "/proc/acpi/battery/BAT2/state")

checkFileBatt :: (String, String) -> IO Bool
checkFileBatt (i,_) =
    fileExist i

readFileBatt :: (String, String) -> IO BattMap
readFileBatt (i,s) = 
    do a <- catch (B.readFile i) (const $ return B.empty)
       b <- catch (B.readFile s) (const $ return B.empty)
       return $ mkMap a b

mkMap :: B.ByteString -> B.ByteString -> BattMap
mkMap a b = M.fromList . mapMaybe parseLine $ concatMap B.lines [a, b]

parseLine :: B.ByteString -> Maybe (String, Integer)
parseLine s =
    let (k, r) = B.break (==':') s
        (v, _) = B.span (isDigit) (B.tail r)
        (ks, vs) = (B.unpack k, B.unpack v)
    in case (ks, vs) of
        ([], _) -> Nothing
        (_, []) -> Nothing
        _ -> Just (ks, read vs)

parseBATT :: IO Float
parseBATT =
    do m1 <- readFileBatt fileB1
       m2 <- readFileBatt fileB2
       let pr1 = M.findWithDefault 0 remKey m1
           fu1 = M.findWithDefault 0 fullKey m1
           pr2 = M.findWithDefault 0 remKey m2
           fu2 = M.findWithDefault 0 fullKey m2
           pr  = pr1 + pr2
           fu  = fu1 + fu2
           pc  = if fu /= 0 then fromInteger pr / fromInteger fu else 0.0
       return pc
  where
    remKey = "remaining capacity"
    fullKey = "last full capacity"

formatBatt :: Float -> Monitor [String] 
formatBatt x =
    do let f s = floatToPercent (s / 100)
       l <- showWithColors f (x * 100)
       return [l]

runBatt :: [String] -> Monitor String
runBatt _ =
    do c <- io $ parseBATT
       l <- formatBatt c
       parseTemplate l