stock-widget/stock-widget.lua at master · whwright/stock-widget · GitHub

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
local wibox = require("wibox")
local awful = require("awful")
awful.util = require("awful.util")
local os = require("os")
local naughty = require("naughty")


local function color_tags(color)
    if color
    then return '<span color="' .. color .. '">', '</span>'
    else return '', ''
    end
end

local function substitute(template, context)
    if type(template) == "string" then
        return (template:gsub("%${([%w_]+)}", function(key)
            return tostring(context[key] or "Err!")
        end))
    else
        -- function / functor:
        return template(context)
    end
end


local stock_widget = {}

function stock_widget:new(args)
    return setmetatable({}, {__index = self}):init(args)
end

function stock_widget:init(args)
    if args.symbol == nil then
        error("symbol is require")
    end

    self.symbol = args.symbol
    self.widget = wibox.widget.textbox()

    self.text_format = args.text_format or (
        "${symbol_color_on}${symbol}${symbol_color_off} ${price_color_on}\$${price}${price_color_off} ${change_color_on}${change}%${change_color_off}")
    self.symbol_color = args.symbol_color or "white"
    self.price_color = args.price_color or nil
    self.gain_color = args.gain_color or "green"
    self.loss_color = args.loss_color or "red"

    self.get_stock_price_location = args.get_stock_price_location
    if self.get_stock_price_location == nil then
        local home = os.getenv("HOME")
        local base_config_location = home .. "/.config/awesome"
        local sub_module_location = home .. "/.config/awesome/stock-widget"
        if awful.util.file_readable(base_config_location .. "/get_stock_price.py") then
            self.get_stock_price_location = base_config_location
        elseif awful.util.file_readable(sub_module_location .. "/get_stock_price.py") then
            self.get_stock_price_location = sub_module_location
        else
            error("could not find get_stock_price.py")
        end
    end

    -- set default text before response has come back
    self:update_widget({ price = "-", change = "-"})
    self.timer = timer({ timeout = args.timeout or 60 })
    self.timer:connect_signal("timeout", function() self:update() end)
    self.timer:start()
    self:update()

    return self
end

function stock_widget:update()
    local cmd = self.get_stock_price_location .. "/get_stock_price.py" .. " " .. self.symbol
    awful.spawn.with_line_callback(cmd, {
            stdout = function(line)
                local items = {}
                for item in line:gmatch("%S+") do table.insert(items, item) end

                data = {
                    price  = items[1],
                    change = items[2]
                }
                self:update_widget(data)
            end,
            stderr = function(line)
                self:update_widget({ price = "ERR", change = "ERR"})
                -- naughty.notify({ text = "ERR: " .. line })
            end
        })
end

function stock_widget:update_widget(data)
    data.symbol = self.symbol
    data.symbol_color_on, data.symbol_color_off = color_tags(self.symbol_color)
    data.price_color_on, data.price_color_off = color_tags(self.price_color)

    local change_color = self.gain_color
    if data.change:sub(1,1) == "-" then
        change_color = self.loss_color
    end
    data.change_color_on, data.change_color_off = color_tags(change_color)

    self.widget:set_markup(substitute(self.text_format, data))
end

return setmetatable(stock_widget, {__call = stock_widget.new})