Initial commit

lib/lain/widget/fs.lua

@@ -0,0 +1,174 @@
+--[[
+
+     Licensed under GNU General Public License v2
+      * (c) 2018, Uli Schlacter
+      * (c) 2018, Otto Modinos
+      * (c) 2013, Luca CPZ
+
+--]]
+
+local helpers = require('lain.helpers')
+local Gio = require('lgi').Gio
+local focused = require('awful.screen').focused
+local wibox = require('wibox')
+local naughty = require('naughty')
+local gears = require('gears')
+local math = math
+local string = string
+local tconcat = table.concat
+local type = type
+local query_size = Gio.FILE_ATTRIBUTE_FILESYSTEM_SIZE
+local query_free = Gio.FILE_ATTRIBUTE_FILESYSTEM_FREE
+local query_used = Gio.FILE_ATTRIBUTE_FILESYSTEM_USED
+local query = query_size .. ',' .. query_free .. ',' .. query_used
+
+-- File systems info
+-- lain.widget.fs
+
+local function factory(args)
+	args = args or {}
+
+	local fs = {
+		widget = args.widget or wibox.widget.textbox(),
+		units = {
+			[1] = 'Kb',
+			[2] = 'Mb',
+			[3] = 'Gb',
+			[4] = 'Tb',
+			[5] = 'Pb',
+			[6] = 'Eb',
+			[7] = 'Zb',
+			[8] = 'Yb',
+		},
+	}
+
+	function fs.hide()
+		if not fs.notification then
+			return
+		end
+		naughty.destroy(fs.notification)
+		fs.notification = nil
+	end
+
+	function fs.show(seconds, scr)
+		fs.hide()
+		fs.update(function()
+			fs.notification_preset.screen = fs.followtag and focused() or scr or 1
+			fs.notification = naughty.notify({
+				preset = fs.notification_preset,
+				timeout = type(seconds) == 'number' and seconds or 5,
+			})
+		end)
+	end
+
+	local timeout = args.timeout or 600
+	local partition = args.partition
+	local threshold = args.threshold or 99
+	local showpopup = args.showpopup or 'on'
+	local settings = args.settings or function() end
+
+	fs.followtag = args.followtag or false
+	fs.notification_preset = args.notification_preset
+
+	if not fs.notification_preset then
+		fs.notification_preset = {
+			font = 'Monospace 10',
+			fg = '#FFFFFF',
+			bg = '#000000',
+		}
+	end
+
+	local function update_synced()
+		local pathlen = 10
+		fs_now = {}
+
+		local notifypaths = {}
+		for _, mount in ipairs(Gio.unix_mounts_get()) do
+			local path = Gio.unix_mount_get_mount_path(mount)
+			local root = Gio.File.new_for_path(path)
+			local info = root:query_filesystem_info(query)
+
+			if info then
+				local size = info:get_attribute_uint64(query_size)
+				local used = info:get_attribute_uint64(query_used)
+				local free = info:get_attribute_uint64(query_free)
+
+				if size > 0 then
+					local units = math.floor(math.log(size) / math.log(1024))
+
+					fs_now[path] = {
+						units = fs.units[units],
+						percentage = math.floor(100 * used / size), -- used percentage
+						size = size / math.pow(1024, units),
+						used = used / math.pow(1024, units),
+						free = free / math.pow(1024, units),
+					}
+
+					if fs_now[path].percentage > 0 then -- don't notify unused file systems
+						notifypaths[#notifypaths + 1] = path
+
+						if #path > pathlen then
+							pathlen = #path
+						end
+					end
+				end
+			end
+		end
+
+		widget = fs.widget
+		settings()
+
+		if partition and fs_now[partition] and fs_now[partition].percentage >= threshold then
+			if not helpers.get_map(partition) then
+				naughty.notify({
+					preset = naughty.config.presets.critical,
+					title = 'Warning',
+					text = string.format('%s is above %d%% (%d%%)', partition, threshold, fs_now[partition].percentage),
+				})
+				helpers.set_map(partition, true)
+			else
+				helpers.set_map(partition, false)
+			end
+		end
+
+		local fmt = '%-' .. tostring(pathlen) .. 's %4s\t%6s\t%6s\n'
+		local notifytable = { [1] = string.format(fmt, 'path', 'used', 'free', 'size') }
+		fmt = '\n%-' .. tostring(pathlen) .. 's %3s%%\t%6.2f\t%6.2f %s'
+		for _, path in ipairs(notifypaths) do
+			notifytable[#notifytable + 1] = string.format(
+				fmt,
+				path,
+				fs_now[path].percentage,
+				fs_now[path].free,
+				fs_now[path].size,
+				fs_now[path].units
+			)
+		end
+
+		fs.notification_preset.text = tconcat(notifytable)
+	end
+
+	function fs.update(callback)
+		Gio.Async.start(gears.protected_call.call)(function()
+			update_synced()
+			if type(callback) == 'function' and callback then
+				callback()
+			end
+		end)
+	end
+
+	if showpopup == 'on' then
+		fs.widget:connect_signal('mouse::enter', function()
+			fs.show(0)
+		end)
+		fs.widget:connect_signal('mouse::leave', function()
+			fs.hide()
+		end)
+	end
+
+	helpers.newtimer(partition or 'fs', timeout, fs.update)
+
+	return fs
+end
+
+return factory