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
|
{- This Source Code Form is subject to the terms of the Mozilla Public License,
v. 2.0. If a copy of the MPL was not distributed with this file, You can
obtain one at https://mozilla.org/MPL/2.0/. -}
module TeaCleaner.Configuration
( ProgramOptions(..)
, Settings(..)
, decodeSettingsFile
, commandLineInfo
, execParser
) where
import GHC.Records (HasField(..))
import Data.Text (StrictText)
import qualified Toml
import Toml ((.=))
import Options.Applicative
( Parser
, ParserInfo
, (<**>)
, execParser
, fullDesc
, help
, helper
, info
, long
, progDesc
, switch
)
import Text.URI (URI)
import qualified Text.URI as URI
import Data.Time (UTCTime(..), getCurrentTime)
import Data.IORef (IORef, newIORef)
data ConfigFile = ConfigFile
{ token :: StrictText
, server :: StrictText
, spamWords :: [StrictText]
, mailDomains :: [StrictText]
, noLogin :: Word
} deriving (Eq, Show)
configFileCodec :: Toml.TomlCodec ConfigFile
configFileCodec = ConfigFile
<$> Toml.text "token" .= getField @"token"
<*> Toml.text "server" .= getField @"server"
<*> Toml.arrayOf Toml._Text "spam_words" .= getField @"spamWords"
<*> Toml.arrayOf Toml._Text "mail_domains" .= getField @"mailDomains"
<*> Toml.word "no_login" .= getField @"noLogin"
data Settings = Settings
{ token :: StrictText
, server :: URI
, now :: UTCTime
, spamWords :: [StrictText]
, mailDomains :: [StrictText]
, statistics :: IORef Int
, noLogin :: Word
} deriving Eq
decodeSettingsFile :: FilePath -> IO Settings
decodeSettingsFile configPath = do
ConfigFile{..} <- Toml.decodeFile configFileCodec configPath
parsedServer <- URI.mkURI server
now <- getCurrentTime
ioRef <- newIORef 0
pure $ Settings
{ token = token
, server = parsedServer
, now = now
, spamWords = spamWords
, mailDomains = mailDomains
, noLogin = noLogin
, statistics = ioRef
}
newtype ProgramOptions = ProgramOptions
{ live :: Bool
} deriving (Eq, Show)
commandLineInfo :: ParserInfo ProgramOptions
commandLineInfo = info (commandLine <**> helper)
$ fullDesc <> progDesc "Helps to detect some spam gitea accounts"
commandLine :: Parser ProgramOptions
commandLine = ProgramOptions
<$> switch (long "live" <> help "Purge suspicious users")
|
