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
|
{-# LANGUAGE OverloadedStrings, LambdaCase #-}
module Config (
Config(..),
getConfig,
makeOpenSearch
) where
import Prelude hiding (lookup)
import Control.Applicative ((<|>))
import Data.Aeson.KeyMap (empty, lookup)
import Data.Functor ((<&>))
import Data.Maybe (fromMaybe)
import Data.String (fromString, IsString)
import Data.Text (Text)
import Data.Yaml
import Network.Wai.Handler.Warp (HostPreference)
import System.Environment (lookupEnv)
data Config = Config {
confPort :: Int,
confHost :: HostPreference,
confBaseUrl :: Text,
confFavicon :: Text,
confUser :: Text,
confPwHash :: Text
} deriving (Show, Eq)
getConfig :: IO Config
getConfig = do
confFile <- decodeFileEither "fastbangs.yaml" <&> \case
Right ob -> ob
Left _ -> empty
Config
<$> (read <$> resolveVal (lookup "port" confFile) "PORT" "20546")
<*> resolveVal (lookup "bind-addr" confFile) "BIND_ADDR" "*6"
<*> resolveVal (lookup "base-url" confFile) "BASE_URL" "http://localhost:20546"
<*> resolveVal (lookup "favicon-url" confFile) "FAVICON_URL" ""
<*> resolveVal (lookup "admin-user" confFile) "ADMIN_USER" "bleb"
<*> resolveVal (lookup "admin-pw-hash" confFile) "ADMIN_PW_HASH" "" -- prevent login without manual pw
where resolveVal :: IsString s => Maybe String -> String -> String -> IO s
resolveVal mayConf q def = do
mayEnv <- lookupEnv q
return $ fromString $ fromMaybe def $ mayEnv <|> mayConf
makeOpenSearch :: Config -> Maybe Text -> Text
makeOpenSearch cfg defBang =
"<OpenSearchDescription xmlns=\"http://a9.com/-/spec/opensearch/1.1/\"\n"
<> " xmlns:moz=\"http://www.mozilla.org/2006/browser/search/\">\n"
<> " <ShortName>" <> name <> "</ShortName>\n"
<> " <Description>Handles search bangs (mostly) locally.</Description>\n"
<> " <InputEncoding>UTF-8</InputEncoding>\n"
<> " <Image width=\"16\" height=\"16\" type=\"image/x-icon\">" <> confFavicon cfg <> "</Image>\n"
<> " <Url type=\"text/html\" template=\"" <> searchUrl <> "\"/>\n"
<> "</OpenSearchDescription>\n"
where (searchUrl, name) = case defBang of
Nothing -> (confBaseUrl cfg <> "/#{searchTerms}", "FastBangs")
Just b -> (confBaseUrl cfg <> "/#" <> b <> "#{searchTerms}", "FastBangs (" <> b <> ")")
|
