]> git.madduck.net Git - etc/awesome.git/blob - widgets/weather.lua

madduck's git repository

Every one of the projects in this repository is available at the canonical URL git://git.madduck.net/madduck/pub/<projectpath> — see each project's metadata for the exact URL.

All patches and comments are welcome. Please squash your changes to logical commits before using git-format-patch and git-send-email to patches@git.madduck.net. If you'd read over the Git project's submission guidelines and adhered to them, I'd be especially grateful.

SSH access, as well as push access can be individually arranged.

If you use my repositories frequently, consider adding the following snippet to ~/.gitconfig and using the third clone URL listed for each project:

[url "git://git.madduck.net/madduck/"]
  insteadOf = madduck:

3512f229ad7cb89b0d57bb725c85c510b24ff8c2
[etc/awesome.git] / widgets / weather.lua
1
2 --[[
3                                                   
4      Licensed under GNU General Public License v2 
5       * (c) 2015, Luke Bonham                     
6                                                   
7 --]]
8
9 local newtimer     = require("lain.helpers").newtimer
10 local read_pipe    = require("lain.helpers").read_pipe
11
12 local async        = require("lain.asyncshell")
13 local json         = require("lain.util").dkjson
14 local lain_icons   = require("lain.helpers").icons_dir
15
16 local naughty      = require("naughty")
17 local wibox        = require("wibox")
18
19 local math         = { floor  = math.floor }
20 local mouse        = mouse
21 local os           = { time   = os.time }
22 local string       = { format = string.format,
23                        gsub   = string.gsub }
24
25 local tonumber     = tonumber
26 local setmetatable = setmetatable
27
28 -- OpenWeatherMap
29 -- current weather and X-days forecast
30 -- lain.widgets.weather
31
32 local function worker(args)
33     local weather               = {}
34     local args                  = args or {}
35     local APPID                 = args.APPID or "3e321f9414eaedbfab34983bda77a66e" -- lain default
36     local timeout               = args.timeout or 900   -- 15 min
37     local timeout_forecast      = args.timeout or 86400 -- 24 hrs
38     local current_call          = args.current_call  or "curl -s 'http://api.openweathermap.org/data/2.5/weather?id=%s&units=%s&lang=%s&APPID=%s'"
39     local forecast_call         = args.forecast_call or "curl -s 'http://api.openweathermap.org/data/2.5/forecast/daily?id=%s&units=%s&lang=%s&cnt=%s&APPID=%s'"
40     local city_id               = args.city_id or 0 -- placeholder
41     local utc                   = args.utc or 0
42     local units                 = args.units or "metric"
43     local lang                  = args.lang or "en"
44     local cnt                   = args.cnt or 5
45     local date_cmd              = args.date_cmd or "date -u -d @%d +'%%a %%d'"
46     local icons_path            = args.icons_path or lain_icons .. "openweathermap/"
47     local notification_preset   = args.notification_preset or {}
48     local notification_text_fun = args.notification_text_fun or
49                                   function (wn)
50                                       local day = string.gsub(read_pipe(string.format(date_cmd, wn["dt"])), "\n", "")
51                                       local tmin = math.floor(wn["temp"]["min"])
52                                       local tmax = math.floor(wn["temp"]["max"])
53                                       local desc = wn["weather"][1]["description"]
54
55                                       return string.format("<b>%s</b>: %s, %d - %d ", day, desc, tmin, tmax)
56                                   end
57     local weather_na_markup     = args.weather_na_markup or " N/A "
58     local followmouse           = args.followmouse or false
59     local settings              = args.settings or function() end
60
61     weather.widget    = wibox.widget.textbox(weather_na_markup)
62     weather.icon_path = icons_path .. "na.png"
63     weather.icon      = wibox.widget.imagebox(weather.icon_path)
64
65     function weather.show(t_out)
66         weather.hide()
67
68         if followmouse then
69             notification_preset.screen = mouse.screen
70         end
71
72         if not weather.notification_text then
73             weather.forecast_update()
74         end
75
76
77         weather.notification = naughty.notify({
78             text    = weather.notification_text,
79             icon    = weather.icon_path,
80             timeout = t_out,
81             preset  = notification_preset
82         })
83     end
84
85     function weather.hide()
86         if weather.notification then
87             naughty.destroy(weather.notification)
88             weather.notification = nil
89         end
90     end
91
92     function weather.attach(obj)
93         obj:connect_signal("mouse::enter", function()
94             weather.show(0)
95         end)
96         obj:connect_signal("mouse::leave", function()
97             weather.hide()
98         end)
99     end
100
101     function weather.forecast_update()
102         local cmd = string.format(forecast_call, city_id, units, lang, cnt, APPID)
103         async.request(cmd, function(f)
104             local pos, err
105             weather_now, pos, err = json.decode(f, 1, nil)
106
107             if not err and weather_now and tonumber(weather_now["cod"]) == 200 then
108                 weather.notification_text = ''
109                 for i = 1, weather_now["cnt"] do
110                     weather.notification_text = weather.notification_text ..
111                                                 notification_text_fun(weather_now["list"][i])
112
113                     if i < weather_now["cnt"] then
114                         weather.notification_text = weather.notification_text .. "\n"
115                     end
116                 end
117             end
118         end)
119     end
120
121     function weather.update()
122         local cmd = string.format(current_call, city_id, units, lang, APPID)
123         async.request(cmd, function(f)
124             local pos, err, icon
125             weather_now, pos, err = json.decode(f, 1, nil)
126
127             if not err and weather_now and tonumber(weather_now["cod"]) == 200 then
128                 -- weather icon based on localtime
129                 now     = os.time()
130                 sunrise = tonumber(weather_now["sys"]["sunrise"])
131                 sunset  = tonumber(weather_now["sys"]["sunset"])
132                 icon    = weather_now["weather"][1]["icon"]
133
134                 if sunrise <= (now-86400) then now = now - 86400 end
135
136                 if sunrise <= now and now <= sunset then
137                     icon = string.gsub(icon, "n", "d")
138                 else
139                     icon = string.gsub(icon, "d", "n")
140                 end
141
142                 weather.icon_path = icons_path .. icon .. ".png"
143
144                 widget = weather.widget
145                 settings()
146             else
147                 weather.icon_path = icons_path .. "na.png"
148                 weather.widget:set_markup(weather_na_markup)
149             end
150
151             weather.icon:set_image(weather.icon_path)
152         end)
153     end
154
155     weather.attach(weather.widget)
156
157     newtimer("weather-" .. city_id, timeout, weather.update)
158     newtimer("weather_forecast-" .. city_id, timeout, weather.forecast_update)
159
160     return setmetatable(weather, { __index = weather.widget })
161 end
162
163 return setmetatable({}, { __call = function(_, ...) return worker(...) end })