adblock2privoxy-3.0.0: src/Statistics.hs

module Statistics
  ( collectStat,
  )
where

import Control.Monad
import Control.Monad.State
import Data.List (foldl')
import qualified Data.Map.Strict as Map
import Data.Maybe
import InputParser

type Stat = Map.Map String Int

collectStat :: [Line] -> [String]
collectStat = fmap resultLine . Map.toAscList . foldl' (flip getStat) Map.empty
  where
    resultLine (name, value) = concat [name, ": ", show value]

increment :: String -> Stat -> Stat
increment key = Map.insertWith (+) key 1

isJustFilled :: Maybe [a] -> Bool
isJustFilled Nothing = False
isJustFilled (Just list) = not . null $ list

getStat :: Line -> Stat -> Stat
getStat (Line _ Comment {}) = increment "Comments"
getStat (Line _ Error {}) = increment "Errors"
getStat (Line _ ElementHide {}) = increment "Elements hiding rules"
getStat (Line _ (RequestBlock policy _ (RequestOptions _ thirdParty domains _ _ _ _ _))) = execState stateState
  where
    incrementState = modify . increment
    stateState = do
      incrementState "Request block rules total"
      when (policy == InputParser.Unblock) $ incrementState "Request block rules for exception"
      when (isJust thirdParty) $ incrementState "Rules with third party option"
      when ((not . null . _negative $ domains) || (isJustFilled . _positive $ domains)) $ incrementState "Request block rules with domain option"
      when ((not . null . _negative $ domains) || (isJustFilled . _positive $ domains)) $ incrementState "Request block rules with request type options"