module Xmobar.Plugins.Monitors.Batt.Linux (readBatteries) where
import Xmobar.Plugins.Monitors.Batt.Common ( BattOpts(..)
, Result(..)
, Status(..)
, maybeAlert)
import Control.Monad (unless)
import Control.Exception (SomeException, handle)
import System.FilePath ((</>))
import System.IO (IOMode(ReadMode), hGetLine, withFile, Handle)
import Data.List (sort, sortBy, group)
import Data.Maybe (fromMaybe)
import Data.Ord (comparing)
import Text.Read (readMaybe)
data Files = Files
{ Files -> String
fEFull :: String
, Files -> String
fCFull :: String
, Files -> String
fEFullDesign :: String
, Files -> String
fCFullDesign :: String
, Files -> String
fENow :: String
, Files -> String
fCNow :: String
, Files -> String
fVoltage :: String
, Files -> String
fVoltageMin :: String
, Files -> String
fCurrent :: String
, Files -> String
fPower :: String
, Files -> String
fStatus :: String
, Files -> String
fBat :: String
} deriving Files -> Files -> Bool
(Files -> Files -> Bool) -> (Files -> Files -> Bool) -> Eq Files
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
$c== :: Files -> Files -> Bool
== :: Files -> Files -> Bool
$c/= :: Files -> Files -> Bool
/= :: Files -> Files -> Bool
Eq
defaultFiles :: Files
defaultFiles :: Files
defaultFiles = Files
{ fEFull :: String
fEFull = String
"energy_full"
, fCFull :: String
fCFull = String
"charge_full"
, fEFullDesign :: String
fEFullDesign = String
"energy_full_design"
, fCFullDesign :: String
fCFullDesign = String
"charge_full_design"
, fENow :: String
fENow = String
"energy_now"
, fCNow :: String
fCNow = String
"charge_now"
, fVoltage :: String
fVoltage = String
"voltage_now"
, fVoltageMin :: String
fVoltageMin = String
"voltage_min_design"
, fCurrent :: String
fCurrent = String
"current_now"
, fPower :: String
fPower = String
"power_now"
, fStatus :: String
fStatus = String
"status"
, fBat :: String
fBat = String
"BAT0"
}
type FilesAccessor = Files -> String
sysDir :: FilePath
sysDir :: String
sysDir = String
"/sys/class/power_supply"
battFile :: FilesAccessor -> Files -> FilePath
battFile :: (Files -> String) -> Files -> String
battFile Files -> String
accessor Files
files = String
sysDir String -> String -> String
</> Files -> String
fBat Files
files String -> String -> String
</> Files -> String
accessor Files
files
grabNumber :: (Num a, Read a) => FilesAccessor -> Files -> IO (Maybe a)
grabNumber :: forall a.
(Num a, Read a) =>
(Files -> String) -> Files -> IO (Maybe a)
grabNumber = (Handle -> IO a) -> (Files -> String) -> Files -> IO (Maybe a)
forall a.
(Handle -> IO a) -> (Files -> String) -> Files -> IO (Maybe a)
grabFile ((String -> a) -> IO String -> IO a
forall a b. (a -> b) -> IO a -> IO b
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap String -> a
forall a. Read a => String -> a
read (IO String -> IO a) -> (Handle -> IO String) -> Handle -> IO a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Handle -> IO String
hGetLine)
grabString :: FilesAccessor -> Files -> IO (Maybe String)
grabString :: (Files -> String) -> Files -> IO (Maybe String)
grabString = (Handle -> IO String)
-> (Files -> String) -> Files -> IO (Maybe String)
forall a.
(Handle -> IO a) -> (Files -> String) -> Files -> IO (Maybe a)
grabFile Handle -> IO String
hGetLine
grabFile :: (Handle -> IO a) -> FilesAccessor -> Files -> IO (Maybe a)
grabFile :: forall a.
(Handle -> IO a) -> (Files -> String) -> Files -> IO (Maybe a)
grabFile Handle -> IO a
readMode Files -> String
accessor Files
files =
(SomeException -> IO (Maybe a)) -> IO (Maybe a) -> IO (Maybe a)
forall e a. Exception e => (e -> IO a) -> IO a -> IO a
handle (Maybe a -> SomeException -> IO (Maybe a)
forall a. a -> SomeException -> IO a
onFileError Maybe a
forall a. Maybe a
Nothing) (String -> IOMode -> (Handle -> IO (Maybe a)) -> IO (Maybe a)
forall r. String -> IOMode -> (Handle -> IO r) -> IO r
withFile String
f IOMode
ReadMode ((a -> Maybe a) -> IO a -> IO (Maybe a)
forall a b. (a -> b) -> IO a -> IO b
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap a -> Maybe a
forall a. a -> Maybe a
Just (IO a -> IO (Maybe a))
-> (Handle -> IO a) -> Handle -> IO (Maybe a)
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Handle -> IO a
readMode))
where f :: String
f = (Files -> String) -> Files -> String
battFile Files -> String
accessor Files
files
onFileError :: a -> SomeException -> IO a
onFileError :: forall a. a -> SomeException -> IO a
onFileError a
returnOnError = IO a -> SomeException -> IO a
forall a b. a -> b -> a
const (a -> IO a
forall a. a -> IO a
forall (m :: * -> *) a. Monad m => a -> m a
return a
returnOnError)
batteryFiles :: String -> Files
batteryFiles :: String -> Files
batteryFiles String
bat = Files
defaultFiles { fBat = bat }
data Battery = Battery
{ Battery -> Float
full :: !Float
, Battery -> Float
now :: !Float
, Battery -> Float
power :: !Float
, Battery -> String
status :: !String
}
haveAc :: FilePath -> IO Bool
haveAc :: String -> IO Bool
haveAc String
f =
(SomeException -> IO Bool) -> IO Bool -> IO Bool
forall e a. Exception e => (e -> IO a) -> IO a -> IO a
handle (Bool -> SomeException -> IO Bool
forall a. a -> SomeException -> IO a
onFileError Bool
False) (IO Bool -> IO Bool) -> IO Bool -> IO Bool
forall a b. (a -> b) -> a -> b
$
String -> IOMode -> (Handle -> IO Bool) -> IO Bool
forall r. String -> IOMode -> (Handle -> IO r) -> IO r
withFile (String
sysDir String -> String -> String
</> String
f) IOMode
ReadMode ((String -> Bool) -> IO String -> IO Bool
forall a b. (a -> b) -> IO a -> IO b
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap (String -> String -> Bool
forall a. Eq a => a -> a -> Bool
== String
"1") (IO String -> IO Bool)
-> (Handle -> IO String) -> Handle -> IO Bool
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Handle -> IO String
hGetLine)
readBatPower :: Float -> Files -> IO (Maybe Float)
readBatPower :: Float -> Files -> IO (Maybe Float)
readBatPower Float
sc Files
f =
do pM <- (Files -> String) -> Files -> IO (Maybe Float)
forall a.
(Num a, Read a) =>
(Files -> String) -> Files -> IO (Maybe a)
grabNumber Files -> String
fPower Files
f
cM <- grabNumber fCurrent f
vM <- grabNumber fVoltage f
return $ case (pM, cM, vM) of
(Just Float
pVal, Maybe Float
_, Maybe Float
_) -> Float -> Maybe Float
forall a. a -> Maybe a
Just (Float -> Maybe Float) -> Float -> Maybe Float
forall a b. (a -> b) -> a -> b
$ Float
pVal Float -> Float -> Float
forall a. Fractional a => a -> a -> a
/ Float
sc
(Maybe Float
_, Just Float
cVal, Just Float
vVal) -> Float -> Maybe Float
forall a. a -> Maybe a
Just (Float -> Maybe Float) -> Float -> Maybe Float
forall a b. (a -> b) -> a -> b
$ Float
cVal Float -> Float -> Float
forall a. Num a => a -> a -> a
* Float
vVal Float -> Float -> Float
forall a. Fractional a => a -> a -> a
/ (Float
sc Float -> Float -> Float
forall a. Num a => a -> a -> a
* Float
sc)
(Maybe Float
_, Maybe Float
_, Maybe Float
_) -> Maybe Float
forall a. Maybe a
Nothing
readBatCapacityFull :: Float -> Files -> IO (Maybe Float)
readBatCapacityFull :: Float -> Files -> IO (Maybe Float)
readBatCapacityFull Float
sc Files
f =
do cM <- (Files -> String) -> Files -> IO (Maybe Float)
forall a.
(Num a, Read a) =>
(Files -> String) -> Files -> IO (Maybe a)
grabNumber Files -> String
fCFull Files
f
eM <- grabNumber fEFull f
cdM <- grabNumber fCFullDesign f
edM <- grabNumber fEFullDesign f
vM <- grabNumber fVoltageMin f
return $ case (eM, cM, edM, cdM, vM) of
(Just Float
eVal, Maybe Float
_, Maybe Float
_, Maybe Float
_, Maybe Float
_) -> Float -> Maybe Float
forall a. a -> Maybe a
Just (Float -> Maybe Float) -> Float -> Maybe Float
forall a b. (a -> b) -> a -> b
$ Float
eVal Float -> Float -> Float
forall a. Fractional a => a -> a -> a
/ Float
sc
(Maybe Float
_, Just Float
cVal, Maybe Float
_, Maybe Float
_, Just Float
vVal) -> Float -> Maybe Float
forall a. a -> Maybe a
Just (Float -> Maybe Float) -> Float -> Maybe Float
forall a b. (a -> b) -> a -> b
$ Float
cVal Float -> Float -> Float
forall a. Num a => a -> a -> a
* Float
vVal Float -> Float -> Float
forall a. Fractional a => a -> a -> a
/ (Float
sc Float -> Float -> Float
forall a. Num a => a -> a -> a
* Float
sc)
(Maybe Float
_, Maybe Float
_, Just Float
eVal, Maybe Float
_, Maybe Float
_) -> Float -> Maybe Float
forall a. a -> Maybe a
Just (Float -> Maybe Float) -> Float -> Maybe Float
forall a b. (a -> b) -> a -> b
$ Float
eVal Float -> Float -> Float
forall a. Fractional a => a -> a -> a
/ Float
sc
(Maybe Float
_, Maybe Float
_, Maybe Float
_, Just Float
cVal, Just Float
vVal) -> Float -> Maybe Float
forall a. a -> Maybe a
Just (Float -> Maybe Float) -> Float -> Maybe Float
forall a b. (a -> b) -> a -> b
$ Float
cVal Float -> Float -> Float
forall a. Num a => a -> a -> a
* Float
vVal Float -> Float -> Float
forall a. Fractional a => a -> a -> a
/ (Float
sc Float -> Float -> Float
forall a. Num a => a -> a -> a
* Float
sc)
(Maybe Float
_, Maybe Float
_, Maybe Float
_, Maybe Float
_, Maybe Float
_) -> Maybe Float
forall a. Maybe a
Nothing
readBatCapacityNow :: Float -> Files -> IO (Maybe Float)
readBatCapacityNow :: Float -> Files -> IO (Maybe Float)
readBatCapacityNow Float
sc Files
f =
do cM <- (Files -> String) -> Files -> IO (Maybe Float)
forall a.
(Num a, Read a) =>
(Files -> String) -> Files -> IO (Maybe a)
grabNumber Files -> String
fCNow Files
f
eM <- grabNumber fENow f
vM <- grabNumber fVoltageMin f
return $ case (eM, cM, vM) of
(Just Float
eVal, Maybe Float
_, Maybe Float
_) -> Float -> Maybe Float
forall a. a -> Maybe a
Just (Float -> Maybe Float) -> Float -> Maybe Float
forall a b. (a -> b) -> a -> b
$ Float
eVal Float -> Float -> Float
forall a. Fractional a => a -> a -> a
/ Float
sc
(Maybe Float
_, Just Float
cVal, Just Float
vVal) -> Float -> Maybe Float
forall a. a -> Maybe a
Just (Float -> Maybe Float) -> Float -> Maybe Float
forall a b. (a -> b) -> a -> b
$ Float
cVal Float -> Float -> Float
forall a. Num a => a -> a -> a
* Float
vVal Float -> Float -> Float
forall a. Fractional a => a -> a -> a
/ (Float
sc Float -> Float -> Float
forall a. Num a => a -> a -> a
* Float
sc)
(Maybe Float
_, Maybe Float
_, Maybe Float
_) -> Maybe Float
forall a. Maybe a
Nothing
readBatStatus :: Files -> IO (Maybe String)
readBatStatus :: Files -> IO (Maybe String)
readBatStatus = (Files -> String) -> Files -> IO (Maybe String)
grabString Files -> String
fStatus
readBattery :: Float -> Files -> IO Battery
readBattery :: Float -> Files -> IO Battery
readBattery Float
sc Files
files =
do cFull <- Float -> (Float -> Files -> IO (Maybe Float)) -> IO Float
forall {f :: * -> *} {b}.
Functor f =>
b -> (Float -> Files -> f (Maybe b)) -> f b
withDef Float
0 Float -> Files -> IO (Maybe Float)
readBatCapacityFull
cNow <- withDef 0 readBatCapacityNow
pwr <- withDef 0 readBatPower
s <- withDef "Unknown" (const readBatStatus)
let cFull' = Float -> Float -> Float
forall a. Ord a => a -> a -> a
max Float
cFull Float
cNow
return $ Battery (3600 * cFull')
(3600 * cNow)
(abs pwr)
s
where withDef :: b -> (Float -> Files -> f (Maybe b)) -> f b
withDef b
d Float -> Files -> f (Maybe b)
reader = b -> Maybe b -> b
forall a. a -> Maybe a -> a
fromMaybe b
d (Maybe b -> b) -> f (Maybe b) -> f b
forall a b. (a -> b) -> f a -> f b
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
`fmap` Float -> Files -> f (Maybe b)
reader Float
sc Files
files
sortOn :: Ord b => (a -> b) -> [a] -> [a]
sortOn :: forall b a. Ord b => (a -> b) -> [a] -> [a]
sortOn a -> b
f =
((b, a) -> a) -> [(b, a)] -> [a]
forall a b. (a -> b) -> [a] -> [b]
map (b, a) -> a
forall a b. (a, b) -> b
snd ([(b, a)] -> [a]) -> ([a] -> [(b, a)]) -> [a] -> [a]
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ((b, a) -> (b, a) -> Ordering) -> [(b, a)] -> [(b, a)]
forall a. (a -> a -> Ordering) -> [a] -> [a]
sortBy (((b, a) -> b) -> (b, a) -> (b, a) -> Ordering
forall a b. Ord a => (b -> a) -> b -> b -> Ordering
comparing (b, a) -> b
forall a b. (a, b) -> a
fst) ([(b, a)] -> [(b, a)]) -> ([a] -> [(b, a)]) -> [a] -> [(b, a)]
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (a -> (b, a)) -> [a] -> [(b, a)]
forall a b. (a -> b) -> [a] -> [b]
map (\a
x -> let y :: b
y = a -> b
f a
x in b
y b -> (b, a) -> (b, a)
forall a b. a -> b -> b
`seq` (b
y, a
x))
mostCommonDef :: Eq a => a -> [a] -> a
mostCommonDef :: forall a. Eq a => a -> [a] -> a
mostCommonDef a
x [a]
xs = [a] -> a
forall a. HasCallStack => [a] -> a
head ([a] -> a) -> [a] -> a
forall a b. (a -> b) -> a -> b
$ [[a]] -> [a]
forall a. HasCallStack => [a] -> a
last ([[a]] -> [a]) -> [[a]] -> [a]
forall a b. (a -> b) -> a -> b
$ [a
x] [a] -> [[a]] -> [[a]]
forall a. a -> [a] -> [a]
: ([a] -> Int) -> [[a]] -> [[a]]
forall b a. Ord b => (a -> b) -> [a] -> [a]
sortOn [a] -> Int
forall a. [a] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length ([a] -> [[a]]
forall a. Eq a => [a] -> [[a]]
group [a]
xs)
readBatteries :: BattOpts -> [String] -> IO Result
readBatteries :: BattOpts -> [String] -> IO Result
readBatteries BattOpts
opts [String]
bfs =
do let bfs'' :: [Files]
bfs'' = (String -> Files) -> [String] -> [Files]
forall a b. (a -> b) -> [a] -> [b]
map String -> Files
batteryFiles [String]
bfs
bats <- (Files -> IO Battery) -> [Files] -> IO [Battery]
forall (t :: * -> *) (m :: * -> *) a b.
(Traversable t, Monad m) =>
(a -> m b) -> t a -> m (t b)
forall (m :: * -> *) a b. Monad m => (a -> m b) -> [a] -> m [b]
mapM (Float -> Files -> IO Battery
readBattery (BattOpts -> Float
scale BattOpts
opts)) (Int -> [Files] -> [Files]
forall a. Int -> [a] -> [a]
take Int
3 [Files]
bfs'')
ac <- haveAc (onlineFile opts)
let sign = if Bool
ac then Float
1 else -Float
1
ft = [Float] -> Float
forall a. Num a => [a] -> a
forall (t :: * -> *) a. (Foldable t, Num a) => t a -> a
sum ((Battery -> Float) -> [Battery] -> [Float]
forall a b. (a -> b) -> [a] -> [b]
map Battery -> Float
full [Battery]
bats)
left = if Float
ft Float -> Float -> Bool
forall a. Ord a => a -> a -> Bool
> Float
0 then [Float] -> Float
forall a. Num a => [a] -> a
forall (t :: * -> *) a. (Foldable t, Num a) => t a -> a
sum ((Battery -> Float) -> [Battery] -> [Float]
forall a b. (a -> b) -> [a] -> [b]
map Battery -> Float
now [Battery]
bats) Float -> Float -> Float
forall a. Fractional a => a -> a -> a
/ Float
ft else Float
0
watts = Float
sign Float -> Float -> Float
forall a. Num a => a -> a -> a
* [Float] -> Float
forall a. Num a => [a] -> a
forall (t :: * -> *) a. (Foldable t, Num a) => t a -> a
sum ((Battery -> Float) -> [Battery] -> [Float]
forall a b. (a -> b) -> [a] -> [b]
map Battery -> Float
power [Battery]
bats)
time = if Float
watts Float -> Float -> Bool
forall a. Eq a => a -> a -> Bool
== Float
0 then Float
0 else Float -> Float -> Float
forall a. Ord a => a -> a -> a
max Float
0 ([Float] -> Float
forall a. Num a => [a] -> a
forall (t :: * -> *) a. (Foldable t, Num a) => t a -> a
sum ([Float] -> Float) -> [Float] -> Float
forall a b. (a -> b) -> a -> b
$ (Battery -> Float) -> [Battery] -> [Float]
forall a b. (a -> b) -> [a] -> [b]
map Battery -> Float
time' [Battery]
bats)
mwatts = if Float
watts Float -> Float -> Bool
forall a. Eq a => a -> a -> Bool
== Float
0 then Float
1 else Float
sign Float -> Float -> Float
forall a. Num a => a -> a -> a
* Float
watts
time' Battery
b = (if Bool
ac then Battery -> Float
full Battery
b Float -> Float -> Float
forall a. Num a => a -> a -> a
- Battery -> Float
now Battery
b else Battery -> Float
now Battery
b) Float -> Float -> Float
forall a. Fractional a => a -> a -> a
/ Float
mwatts
statuses :: [Status]
statuses = (String -> Status) -> [String] -> [Status]
forall a b. (a -> b) -> [a] -> [b]
map (Status -> Maybe Status -> Status
forall a. a -> Maybe a -> a
fromMaybe Status
Unknown (Maybe Status -> Status)
-> (String -> Maybe Status) -> String -> Status
forall b c a. (b -> c) -> (a -> b) -> a -> c
. String -> Maybe Status
forall a. Read a => String -> Maybe a
readMaybe)
([String] -> [String]
forall a. Ord a => [a] -> [a]
sort ((Battery -> String) -> [Battery] -> [String]
forall a b. (a -> b) -> [a] -> [b]
map Battery -> String
status [Battery]
bats))
acst = Status -> [Status] -> Status
forall a. Eq a => a -> [a] -> a
mostCommonDef Status
Unknown ([Status] -> Status) -> [Status] -> Status
forall a b. (a -> b) -> a -> b
$ (Status -> Bool) -> [Status] -> [Status]
forall a. (a -> Bool) -> [a] -> [a]
filter (Status
UnknownStatus -> Status -> Bool
forall a. Eq a => a -> a -> Bool
/=) [Status]
statuses
racst | Status
acst Status -> Status -> Bool
forall a. Eq a => a -> a -> Bool
/= Status
Unknown = Status
acst
| Float
time Float -> Float -> Bool
forall a. Eq a => a -> a -> Bool
== Float
0 = Status
Idle
| Bool
ac = Status
Charging
| Bool
otherwise = Status
Discharging
unless ac (maybeAlert opts left)
return $ if isNaN left then NA else Result left watts time racst