]> git.madduck.net Git - etc/awesome.git/blob - widgets/fs.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:

Merge pull request #206 from Kirogo/master
[etc/awesome.git] / widgets / fs.lua
1
2 --[[
3                                                       
4      Licensed under GNU General Public License v2     
5       * (c) 2013, Luke Bonham                         
6       * (c) 2010, Adrian C.      <anrxc@sysphere.org> 
7       * (c) 2009, Lucas de Vries <lucas@glacicle.com> 
8                                                       
9 --]]
10
11 local helpers      = require("lain.helpers")
12
13 local beautiful    = require("beautiful")
14 local wibox        = require("wibox")
15 local naughty      = require("naughty")
16
17 local io           = { popen  = io.popen }
18 local pairs        = pairs
19 local mouse        = mouse
20 local string       = { match  = string.match,
21                        format = string.format }
22 local tonumber     = tonumber
23
24 local setmetatable = setmetatable
25
26 -- File system disk space usage
27 -- lain.widgets.fs
28 local fs = {}
29 local fs_notification  = nil
30
31 function fs:hide()
32     if fs_notification ~= nil then
33         naughty.destroy(fs_notification)
34         fs_notification = nil
35     end
36 end
37
38 function fs:show(t_out)
39     fs:hide()
40
41     local ws = helpers.read_pipe(helpers.scripts_dir .. "dfs"):gsub("\n*$", "")
42
43     if fs.followmouse then
44         fs.notification_preset.screen = mouse.screen
45     end
46
47     fs_notification = naughty.notify({
48         preset  = fs.notification_preset,
49         text    = ws,
50         timeout = t_out
51     })
52 end
53
54 -- Unit definitions
55 local unit = { ["mb"] = 1024, ["gb"] = 1024^2 }
56
57 local function worker(args)
58     local args             = args or {}
59     local timeout          = args.timeout or 600
60     local partition        = args.partition or "/"
61     local settings         = args.settings or function() end
62
63     fs.followmouse         = args.followmouse or false
64     fs.notification_preset = args.notification_preset or { fg = beautiful.fg_normal }
65
66     fs.widget = wibox.widget.textbox('')
67
68     helpers.set_map(partition, false)
69
70     function update()
71         fs_info = {}
72         fs_now  = {}
73         local f = assert(io.popen("LC_ALL=C df -kP"))
74
75         for line in f:lines() do -- Match: (size) (used)(avail)(use%) (mount)
76             local s     = string.match(line, "^.-[%s]([%d]+)")
77             local u,a,p = string.match(line, "([%d]+)[%D]+([%d]+)[%D]+([%d]+)%%")
78             local m     = string.match(line, "%%[%s]([%p%w]+)")
79
80             if u and m then -- Handle 1st line and broken regexp
81                 fs_info[m .. " size_mb"]  = string.format("%.1f", tonumber(s) / unit["mb"])
82                 fs_info[m .. " size_gb"]  = string.format("%.1f", tonumber(s) / unit["gb"])
83                 fs_info[m .. " used_p"]   = tonumber(p)
84                 fs_info[m .. " avail_p"]  = 100 - tonumber(p)
85             end
86         end
87
88         f:close()
89
90         fs_now.used      = tonumber(fs_info[partition .. " used_p"])  or 0
91         fs_now.available = tonumber(fs_info[partition .. " avail_p"]) or 0
92         fs_now.size_mb   = tonumber(fs_info[partition .. " size_mb"]) or 0
93         fs_now.size_gb   = tonumber(fs_info[partition .. " size_gb"]) or 0
94
95         notification_preset = fs.notification_preset
96         widget = fs.widget
97         settings()
98
99         if fs_now.used >= 99 and not helpers.get_map(partition)
100         then
101             naughty.notify({
102                 title = "warning",
103                 text = partition .. " ran out!\nmake some room",
104                 timeout = 8,
105                 fg = "#000000",
106                 bg = "#FFFFFF",
107             })
108             helpers.set_map(partition, true)
109         else
110             helpers.set_map(partition, false)
111         end
112     end
113
114     fs.widget:connect_signal('mouse::enter', function () fs:show(0) end)
115     fs.widget:connect_signal('mouse::leave', function () fs:hide() end)
116
117     helpers.newtimer(partition, timeout, update)
118
119     return setmetatable(fs, { __index = fs.widget })
120 end
121
122 return setmetatable(fs, { __call = function(_, ...) return worker(...) end })