aboutsummaryrefslogtreecommitdiff
path: root/cmd/Main.hs
blob: 645ac4bbb3452d60a7acbf8ebdf0928ef2ea9c8d (plain)
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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
{-# LANGUAGE LambdaCase #-}

module Main (
  main
) where

import Control.Monad (when)
import Data.List (intercalate)
import Data.Semigroup ((<>))
import Data.Version (showVersion)
import System.Exit (die)
import System.IO (IOMode(ReadMode), hGetLine, hPrint, stderr, withFile)

import Options.Applicative (
    (<**>), (<|>), Parser, auto, execParser, fullDesc, header, help, helper,
    info, long, metavar, option, optional, short, strOption, switch
  )
import System.Directory (createDirectoryIfMissing)
import System.Environment.XDG.BaseDir (getUserConfigDir, getUserConfigFile)

import Paths_openweathermap (version) -- from cabal
import qualified Web.OpenWeatherMap.Client as Client
import qualified Web.OpenWeatherMap.Types.Coord as Coord
import qualified Web.OpenWeatherMap.Types.CurrentWeather as CurrentWeather
import qualified Web.OpenWeatherMap.Types.Main as Main
import qualified Web.OpenWeatherMap.Types.Sys as Sys
import qualified Web.OpenWeatherMap.Types.Weather as Weather
import qualified Web.OpenWeatherMap.Types.Wind as Wind

appName :: String
appName = "openweathermap"

parseLocation :: Parser Client.Location
parseLocation = byName <|> byCoord
  where
    byName = Client.Name <$> strOption
      (  long "city"
      <> short 'c'
      <> metavar "CITY"
      <> help "City name" )

    byCoord = Client.Coord
      <$> option auto
        (  long "lat"
        <> metavar "NUM"
        <> help "Latitude in decimal degrees" )
      <*> option auto
        (  long "lon"
        <> metavar "NUM"
        <> help "Longitude in decimal degrees" )


data ApiKey
  = ApiKeyFile FilePath
  | ApiKey String

parseApiKey :: Parser ApiKey
parseApiKey = fromFile <|> inCmdLine
  where
    fromFile = ApiKeyFile <$> strOption
      (  long "api-key-file"
      <> short 'K'
      <> metavar "APIKEYFILE"
      <> help "Read API key from this file" )

    inCmdLine = ApiKey <$> strOption
      (  long "api-key"
      <> short 'k'
      <> metavar "APIKEY"
      <> help "API key" )


data Config = Config
  { apikey :: Maybe ApiKey
  , location :: Client.Location
  , debug :: Bool
  }

parseConfig :: Parser Config
parseConfig = Config
          <$> optional parseApiKey
          <*> parseLocation
          <*> switch (long "debug" <> short 'd' <> help "Enable debug")

getApiKey :: Maybe ApiKey -> IO String
getApiKey (Just (ApiKey key)) = return key
getApiKey (Just (ApiKeyFile f)) = withFile f ReadMode hGetLine
getApiKey Nothing = do
  createDirectoryIfMissing True =<< getUserConfigDir appName
  getUserConfigFile appName "key" >>= getApiKey . Just . ApiKeyFile


showLocation :: CurrentWeather.CurrentWeather -> String
showLocation w = city ++ maybe "" ("," ++) country ++ " " ++ coords
  where
    name = CurrentWeather.name w
    coord = CurrentWeather.coord w
    country = Sys.country . CurrentWeather.sys $ w
    city = if name /= "" then name else "<unknown>"
    coords = "("++ show (Coord.lat coord) ++ "°, "
                ++ show (Coord.lon coord) ++ "°)"

showWeather :: [Weather.Weather] -> String
showWeather w = intercalate "," $ Weather.main <$> w

showHumidity :: Main.Main -> String
showHumidity m = "H " ++ show hm ++ " %"
  where
    hm :: Int
    hm = round . Main.humidity $ m

-- https://en.wikipedia.org/wiki/Millimeter_of_mercury
showPressure :: Main.Main -> String
showPressure m = "P " ++ show p ++ " mmHg"
  where
    hPa2mmHg hpa = hpa * 0.750061561303
    p :: Int
    p = round . hPa2mmHg . Main.pressure $ m

-- https://stackoverflow.com/q/7490660/933161
showWind :: Wind.Wind -> String
showWind w = dir ++ " " ++ show speed ++ " m/s"
  where
    speed :: Int
    speed = round . Wind.speed $ w
    deg = Wind.deg w
    --     [ "N", "NE", "E", "SE", "S", "SW", "W", "NW" ]
    dirs = [ "↓", "↙",  "←", "↖",  "↑", "↗",  "→", "↘"  ]
    l = length dirs
    sector = round $ (deg * fromIntegral l) / 360.0
    dir = dirs !! (sector `rem` l)

showTemp :: Main.Main -> String
showTemp m = "T " ++ temp ++ " °C"
  where
    k2c k = k - 273.15 -- Kelvin to Celsius
    tmax :: Int
    tmin :: Int
    tmax = round . k2c . Main.temp_max $ m
    tmin = round . k2c . Main.temp_min $ m
    show' t = if t > 0 then "+" ++ show t else show t
    temp = if tmax /= tmin
           then show' tmin ++ ".." ++ show' tmax
           else show' tmin

printWeather :: CurrentWeather.CurrentWeather -> IO ()
printWeather w = putStrLn out
  where
    weather = showWeather $ CurrentWeather.weather w
    place = showLocation w
    mainw = CurrentWeather.main w
    wind = CurrentWeather.wind w
    out = place ++ ": " ++ intercalate ",  "
      [ weather, showHumidity mainw, showPressure mainw, showTemp mainw,
        showWind wind ]

run :: Config -> IO ()
run cfg = do
  appid <- getApiKey . apikey $ cfg
  Client.getWeather appid (location cfg) >>= \case
    Left err      -> die $ show err
    Right weather -> do
      when (debug cfg) $ hPrint stderr weather
      printWeather weather

main :: IO ()
main = run =<< execParser opts
  where
    opts = info (parseConfig <**> helper) (fullDesc <> header desc)
    desc = "openweathermap "
        ++ showVersion version
        ++ " - command-line client for https://openweathermap.org/api"