78 lines
2.8 KiB
Haskell
78 lines
2.8 KiB
Haskell
module Main
|
|
( main
|
|
) where
|
|
|
|
import Data.Vector (Vector)
|
|
import qualified Data.Text as Text
|
|
import qualified Data.Vector as Vector
|
|
import qualified Data.Text.IO as Text.IO
|
|
import qualified Data.Text.Lazy.IO as Text.Lazy.IO
|
|
import qualified Data.Text.Lazy.Builder as Text.Builder
|
|
import qualified Data.Text.Lazy.Builder.Int as Text.Builder
|
|
import GHC.Records (HasField(..))
|
|
import TeaCleaner.Filter
|
|
( UserFilter(..)
|
|
, FilterResult(..)
|
|
, filterByActivities
|
|
, filterByUserProperties
|
|
)
|
|
import TeaCleaner.Client (User(..), getUsers, purgeUser)
|
|
import TeaCleaner.Configuration
|
|
( ProgramOptions(..)
|
|
, Settings(..)
|
|
, decodeSettingsFile
|
|
, commandLineInfo
|
|
, execParser
|
|
)
|
|
import Control.Monad (when)
|
|
import Data.IORef (modifyIORef, readIORef)
|
|
|
|
printStatistics :: Settings -> IO ()
|
|
printStatistics Settings{ statistics } =
|
|
readIORef statistics >>= printCount
|
|
where
|
|
printCount count =
|
|
let count' = Text.Builder.decimal count
|
|
in Text.Lazy.IO.putStrLn
|
|
$ Text.Builder.toLazyText
|
|
$ "Count: " <> count'
|
|
|
|
handleResults :: Settings -> Bool -> Vector FilterResult -> IO (Vector User)
|
|
handleResults settings liveRun =
|
|
Vector.foldM' handleResult Vector.empty
|
|
where
|
|
handleResult accumulator (FilterResult user FailedFilter)
|
|
= handleFailedFilter settings liveRun user
|
|
>> pure accumulator
|
|
handleResult accumulator (FilterResult _ PassFilter) = pure accumulator
|
|
handleResult accumulator (FilterResult user SuspiciousFilter) = pure
|
|
$ Vector.snoc accumulator user
|
|
|
|
handleFailedFilter :: Settings -> Bool -> User -> IO ()
|
|
handleFailedFilter settings liveRun user = Text.IO.putStrLn buildValue
|
|
>> modifyIORef (getField @"statistics" settings) (+ 1)
|
|
>> when liveRun (purgeUser settings user)
|
|
where
|
|
buildValue = getField @"username" user <> "\n"
|
|
<> " Website: " <> getField @"website" user <> "\n"
|
|
<> " Created: " <> Text.pack (show $ getField @"created" user) <> "\n"
|
|
<> " Last login: " <> Text.pack (show $ getField @"lastLogin" user) <> "\n"
|
|
<> " Email: " <> getField @"email" user <> "\n"
|
|
<> " Website: " <> getField @"website" user <> "\n"
|
|
<> " Description: " <> getField @"description" user <> "\n"
|
|
<> " Avatar: " <> getField @"avatarUrl" user <> "\n"
|
|
<> "\n"
|
|
|
|
main :: IO ()
|
|
main = execParser commandLineInfo >>= withArguments
|
|
where
|
|
withArguments ProgramOptions{ liveRun } = decodeSettingsFile "config/tea-cleaner.toml"
|
|
>>= withSettings liveRun
|
|
|
|
withSettings :: Bool -> Settings -> IO ()
|
|
withSettings liveRun settings = getUsers settings
|
|
>>= handleResults settings liveRun . fmap (filterByUserProperties settings)
|
|
>>= traverse (filterByActivities settings)
|
|
>>= handleResults settings liveRun
|
|
>> printStatistics settings
|