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 string       = { format = string.format,
  22                        gsub   = string.gsub }
  23
  24 local setmetatable = setmetatable
  25
  26 -- OpenWeatherMap
  27 -- current weather and X-days forecast
  28 -- lain.widgets.weather
  29
  30 local function worker(args)
  31     local weather             = {}
  32     local args                = args or {}
  33     local timeout             = args.timeout or 900   -- 15 min
  34     local timeout_forecast    = args.timeout or 86400 -- 24 hrs
  35     local current_call        = "curl -s 'http://api.openweathermap.org/data/2.5/weather?id=%s&units=%s&lang=%s'"
  36     local forecast_call       = "curl -s 'http://api.openweathermap.org/data/2.5/forecast/daily?id=%s&units=%s&lang=%s&cnt=%s'"
  37     local city_id             = args.city_id or 0 -- placeholder
  38     local units               = args.units or "metric"
  39     local lang                = args.lang or "en"
  40     local cnt                 = args.cnt or 7
  41     local date_cmd            = args.date_cmd or "date -u -d @%d +'%%a %%d'"
  42     local icons_path          = args.icons_path or lain_icons .. "openweathermap/"
  43     local notification_preset = args.notification_preset or {}
  44     local followmouse         = args.followmouse or false
  45     local settings            = args.settings or function() end
  46
  47     weather.widget = wibox.widget.textbox('')
  48     weather.icon   = wibox.widget.imagebox()
  49
  50     function weather.show(t_out)
  51         weather.hide()
  52
  53         if followmouse then
  54             notification_preset.screen = mouse.screen
  55         end
  56
  57         weather.notification = naughty.notify({
  58             text    = weather.notification_text,
  59             icon    = weather.icon_path,
  60             timeout = t_out,
  61             preset  = notification_preset
  62         })
  63     end
  64
  65     function weather.hide()
  66         if weather.notification ~= nil then
  67             naughty.destroy(weather.notification)
  68             weather.notification = nil
  69         end
  70     end
  71
  72     function weather.attach(obj)
  73         obj:connect_signal("mouse::enter", function()
  74             weather.show(0)
  75         end)
  76         obj:connect_signal("mouse::leave", function()
  77             weather.hide()
  78         end)
  79     end
  80
  81     function weather.forecast_update()
  82         local cmd = string.format(forecast_call, city_id, units, lang, cnt)
  83         async.request(cmd, function(f)
  84             weather_now, pos, err = json.decode(f, 1, nil)
  85
  86             if not err and weather_now ~= nil and tonumber(weather_now["cod"]) == 200 then
  87                 weather.notification_text = ''
  88                 for i = 1, weather_now["cnt"] do
  89                     day = string.gsub(read_pipe(string.format(date_cmd, weather_now["list"][i]["dt"])), "\n", "")
  90
  91                     tmin = math.floor(weather_now["list"][i]["temp"]["min"])
  92                     tmax = math.floor(weather_now["list"][i]["temp"]["max"])
  93                     desc = weather_now["list"][i]["weather"][1]["description"]
  94
  95                     weather.notification_text = weather.notification_text ..
  96                                                 string.format("<b>%s</b>: %s, %d - %d  ", day, desc, tmin, tmax)
  97
  98                     if i < weather_now["cnt"] then
  99                         weather.notification_text = weather.notification_text .. "\n"
 100                     end
 101                 end
 102             else
 103                 weather.icon_path = icons_path .. "na.png"
 104                 weather.notification_text = "API/connection error or bad/not set city ID"
 105             end
 106         end)
 107     end
 108
 109     function weather.update()
 110         local cmd = string.format(current_call, city_id, units, lang)
 111         async.request(cmd, function(f)
 112             weather_now, pos, err = json.decode(f, 1, nil)
 113
 114             if not err and weather_now ~= nil and tonumber(weather_now["cod"]) == 200 then
 115                 weather.icon_path = icons_path .. weather_now["weather"][1]["icon"] .. ".png"
 116                 weather.icon:set_image(weather.icon_path)
 117                 widget = weather.widget
 118                 settings()
 119             else
 120                 weather.widget._layout.text = " N/A " -- tries to avoid textbox bugs
 121                 weather.icon:set_image(icons_path .. "na.png")
 122             end
 123         end)
 124     end
 125
 126     weather.attach(weather.widget)
 127
 128     newtimer("weather-" .. city_id, timeout, weather.update)
 129     newtimer("weather_forecast" .. city_id, timeout, weather.forecast_update)
 130
 131     return setmetatable(weather, { __index = weather.widget })
 132 end
 133
 134 return setmetatable({}, { __call = function(_, ...) return worker(...) end })