widgets/yawn/init.lua

   1
   2 --[[
   3
   4      Yahoo's Awesome (WM) Weather Notification
   5
   6      Licensed under WTFPL v2
   7       * (c) 2013, Luke Bonham
   8
   9 --]]
  10
  11 local markup       = require("lain.util.markup")
  12
  13 local beautiful    = require("beautiful")
  14 local naughty      = require("naughty")
  15 local wibox        = require("wibox")
  16
  17 local debug        = { getinfo = debug.getinfo }
  18 local io           = io
  19 local os           = { date    = os.date,
  20                        getenv  = os.getenv }
  21 local string       = { find    = string.find,
  22                        match   = string.match,
  23                        gsub    = string.gsub,
  24                        sub     = string.sub }
  25 local tonumber     = tonumber
  26
  27 local setmetatable = setmetatable
  28
  29 -- yawn integration
  30 -- https://github.com/copycat-killer/yawn
  31 -- lain.widgets.yawn
  32 local yawn =
  33 {
  34     units    = "",
  35     forecast = "",
  36     icon     = wibox.widget.imagebox(),
  37     widget   = wibox.widget.textbox()
  38 }
  39
  40 local project_path       = debug.getinfo(1, 'S').source:match[[^@(.*/).*$]]
  41 local localizations_path = project_path .. 'localizations/'
  42 local icon_path          = project_path .. 'icons/'
  43 local api_url            = 'http://weather.yahooapis.com/forecastrss'
  44 local units_set          = '?u=c&w=' -- Default is Celsius
  45 local language           = string.match(os.getenv("LANG"), "(%S*$*)[.]")
  46 local weather_data       = nil
  47 local notification       = nil
  48 local city_id            = nil
  49 local sky                = nil
  50 local settings           = {}
  51 local update_timer       = nil
  52
  53 local function fetch_weather(args)
  54     local toshow = args.toshow or "forecast"
  55     local spr = args.spr or " "
  56     local footer = args.footer or ""
  57
  58     local url = api_url .. units_set .. city_id
  59     local f = io.popen("curl --connect-timeout 1 -fsm 2 '"
  60                        .. url .. "'" )
  61     local text = f:read("*all")
  62     io.close(f)
  63
  64     -- In case of no connection or invalid city ID
  65     -- widgets won't display
  66     if text == "" or text:match("City not found")
  67     then
  68         sky = icon_path .. "na.png"
  69         if text == "" then
  70             weather_data = "Service not available at the moment."
  71             return "N/A"
  72         else
  73             weather_data = "City not found!\n" ..
  74                            "Are you sure " .. city_id ..
  75                            " is your Yahoo city ID?"
  76             return "?"
  77         end
  78     end
  79
  80     -- Processing raw data
  81     weather_data = text:gsub("<.->", "")
  82     weather_data = weather_data:match("Current Conditions:.-Full")
  83     weather_data = weather_data:gsub("Current Conditions:.-\n", "Now: ")
  84     weather_data = weather_data:gsub("Forecast:.-\n", "")
  85     weather_data = weather_data:gsub("\nFull", "")
  86     weather_data = weather_data:gsub("[\n]$", "")
  87     weather_data = weather_data:gsub(" [-] " , ": ")
  88     weather_data = weather_data:gsub("[.]", ",")
  89     weather_data = weather_data:gsub("High: ", "")
  90     weather_data = weather_data:gsub(" Low: ", " - ")
  91
  92     -- Getting info for text widget
  93     local now      = weather_data:sub(weather_data:find("Now:")+5,
  94                      weather_data:find("\n")-1)
  95     local forecast = now:sub(1, now:find(",")-1)
  96     local units    = now:sub(now:find(",")+2, -2)
  97
  98     -- Day/Night icon change
  99     local hour = tonumber(os.date("%H"))
 100     sky = icon_path
 101
 102     if forecast == "Clear"         or
 103        forecast == "Fair"          or
 104        forecast == "Partly Cloudy" or
 105        forecast == "Mostly Cloudy"
 106        then
 107            if hour >= 6 and hour <= 18
 108            then
 109                sky = sky .. "Day"
 110            else
 111                sky = sky .. "Night"
 112            end
 113     end
 114
 115     sky = sky  .. forecast:gsub(" ", ""):gsub("/", "") .. ".png"
 116
 117     -- In case there's no defined icon for current forecast
 118     f = io.popen(sky)
 119     if f == nil then
 120         sky = icon_path .. "na.png"
 121     else
 122         io.close(f)
 123     end
 124
 125     -- Localization
 126     local f = io.open(localizations_path .. language, "r")
 127     if language:find("en_") == nil and f ~= nil
 128     then
 129         io.close(f)
 130         for line in io.lines(localizations_path .. language)
 131         do
 132             word = string.sub(line, 1, line:find("|")-1)
 133             translation = string.sub(line, line:find("|")+1)
 134             weather_data = string.gsub(weather_data, word, translation)
 135         end
 136     end
 137
 138     -- Finally setting infos
 139     forecast = weather_data:match(": %S+"):gsub(": ", ""):gsub(",", "")
 140     yawn.forecast = markup(yawn.forecast_color, markup.font(beautiful.font, forecast))
 141     yawn.units = markup(yawn.units_color, markup.font(beautiful.font, units))
 142     yawn.icon:set_image(sky)
 143
 144     if toshow == "forecast" then
 145         return yawn.forecast
 146     elseif toshow == "units" then
 147         return yawn.units
 148     else -- "both"
 149         return yawn.forecast .. spr
 150                .. yawn.units .. footer
 151     end
 152 end
 153
 154 function yawn.hide()
 155     if notification ~= nil then
 156         naughty.destroy(notification)
 157         notification = nil
 158     end
 159 end
 160
 161 function yawn.show(t_out)
 162     if yawn.widget._layout.text == "?"
 163     then
 164         if update_timer ~= nil
 165         then
 166             update_timer:emit_signal("timeout")
 167         else
 168             fetch_weather(settings)
 169         end
 170     end
 171
 172     yawn.hide()
 173
 174     notification = naughty.notify({
 175         text = weather_data,
 176         icon = sky,
 177         timeout = t_out,
 178         fg = yawn.notification_color
 179     })
 180 end
 181
 182 function yawn.register(id, args)
 183     local args = args or {}
 184
 185     settings = { args.toshow, args.spr, args.footer }
 186
 187     yawn.units_color        = args.units_color or
 188                               beautiful.fg_normal or "#FFFFFF"
 189     yawn.forecast_color     = args.forecast_color or
 190                               yawn.units_color
 191     yawn.notification_color = args.notification_color or
 192                               beautiful.fg_focus or "#FFFFFF"
 193
 194     if args.u == "f" then units_set = '?u=f&w=' end
 195
 196     city_id = id
 197
 198     update_timer = timer({ timeout = 600 }) -- 10 mins
 199     update_timer:connect_signal("timeout", function()
 200         yawn.widget:set_markup(fetch_weather(settings))
 201     end)
 202     update_timer:start()
 203     update_timer:emit_signal("timeout")
 204
 205     yawn.icon:connect_signal("mouse::enter", function()
 206         yawn.show(0)
 207     end)
 208     yawn.icon:connect_signal("mouse::leave", function()
 209         yawn.hide()
 210     end)
 211 end
 212
 213 function yawn.attach(widget, id, args)
 214     yawn.register(id, args)
 215
 216     widget:connect_signal("mouse::enter", function()
 217         yawn.show(0)
 218     end)
 219
 220     widget:connect_signal("mouse::leave", function()
 221         yawn.hide()
 222     end)
 223 end
 224
 225 -- }}}
 226
 227 return setmetatable(yawn, { __call = function(_, ...) return yawn.register(...) end })