]> 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:

f07ef98daf88eac5b51d5735c1d452212c1eb4ba
[etc/awesome.git] / widgets / fs.lua
1
2 --[[
3                                                   
4      Licensed under GNU General Public License v2 
5       * (c) 2013, Luke Bonham                     
6                                                   
7 --]]
8
9 local helpers      = require("lain.helpers")
10
11 local shell        = require("awful.util").shell
12 local beautiful    = require("beautiful")
13 local focused      = require("awful.screen").focused
14 local wibox        = require("wibox")
15 local naughty      = require("naughty")
16
17 local string       = string
18 local tonumber     = tonumber
19
20 local setmetatable = setmetatable
21
22 -- File system disk space usage
23 -- lain.widgets.fs
24 local fs = {}
25
26 -- Unit definitions
27 fs.unit = { ["mb"] = 1024, ["gb"] = 1024^2 }
28
29 function fs.hide()
30     naughty.destroy(fs.notification)
31     fs.notification = nil
32 end
33
34 function fs.show(seconds, scr)
35     fs.hide()
36
37     if fs.followtag then
38         fs.notification_preset.screen = focused()
39     elseif scr then
40         fs.notification_preset.screen = scr
41     end
42
43     local cmd = (fs.options and string.format("dfs %s", fs.options)) or "dfs"
44
45     helpers.async(helpers.scripts_dir .. cmd, function(ws)
46         fs.notification = naughty.notify({
47             preset      = fs.notification_preset,
48             text        = ws:gsub("\n*$", ""),
49             timeout     = seconds or 5,
50         })
51     end)
52 end
53
54 local function worker(args)
55     local args             = args or {}
56     local timeout          = args.timeout or 600
57     local partition        = args.partition or "/"
58     local showpopup        = args.showpopup or "on"
59     local notify           = args.notify or "on"
60     local settings         = args.settings or function() end
61
62     fs.options             = args.options
63     fs.followtag           = args.followtag 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
71     function update()
72         fs_info, fs_now  = {}, {}
73         helpers.async(string.format("%s -c 'LC_ALL=C df -k --output=target,size,used,avail,pcent'", shell), function(f)
74             for line in string.gmatch(f, "\n[^\n]+") do
75                 local m,s,u,a,p = string.match(line, "(/.-%s).-(%d+).-(%d+).-(%d+).-([%d]+)%%")
76                 m = m:gsub(" ", "") -- clean target from any whitespace
77
78                 fs_info[m .. " size_mb"]  = string.format("%.1f", tonumber(s) / fs.unit["mb"])
79                 fs_info[m .. " size_gb"]  = string.format("%.1f", tonumber(s) / fs.unit["gb"])
80                 fs_info[m .. " used_mb"]  = string.format("%.1f", tonumber(u) / fs.unit["mb"])
81                 fs_info[m .. " used_gb"]  = string.format("%.1f", tonumber(u) / fs.unit["gb"])
82                 fs_info[m .. " used_p"]   = p
83                 fs_info[m .. " avail_mb"] = string.format("%.1f", tonumber(a) / fs.unit["mb"])
84                 fs_info[m .. " avail_gb"] = string.format("%.1f", tonumber(a) / fs.unit["gb"])
85                 fs_info[m .. " avail_p"]  = string.format("%d", 100 - tonumber(p))
86             end
87
88             fs_now.size_mb      = fs_info[partition .. " size_mb"]  or "N/A"
89             fs_now.size_gb      = fs_info[partition .. " size_gb"]  or "N/A"
90             fs_now.used         = fs_info[partition .. " used_p"]   or "N/A"
91             fs_now.used_mb      = fs_info[partition .. " used_mb"]  or "N/A"
92             fs_now.used_gb      = fs_info[partition .. " used_gb"]  or "N/A"
93             fs_now.available    = fs_info[partition .. " avail_p"]  or "N/A"
94             fs_now.available_mb = fs_info[partition .. " avail_mb"] or "N/A"
95             fs_now.available_gb = fs_info[partition .. " avail_gb"] or "N/A"
96
97             notification_preset = fs.notification_preset
98             widget = fs.widget
99             settings()
100
101             if notify == "on" and tonumber(fs_now.used) >= 99 and not helpers.get_map(partition) then
102                 naughty.notify({
103                     title   = "warning",
104                     text    = partition .. " is empty!",
105                     timeout = 8,
106                     fg      = "#000000",
107                     bg      = "#FFFFFF"
108                 })
109                 helpers.set_map(partition, true)
110             else
111                 helpers.set_map(partition, false)
112             end
113         end)
114     end
115
116     if showpopup == "on" then
117        fs.widget:connect_signal('mouse::enter', function () fs.show(0) end)
118        fs.widget:connect_signal('mouse::leave', function () fs.hide() end)
119     end
120
121     helpers.newtimer(partition, timeout, update)
122
123     return setmetatable(fs, { __index = fs.widget })
124 end
125
126 return setmetatable(fs, { __call = function(_, ...) return worker(...) end })