add locks plugin

Signed-off-by: Mattias Andrée <maandree@operamail.com>

1 files changed, 83 insertions, 0 deletions

diff --git a/examples/plugins/locks b/examples/plugins/locks
new file mode 100644
index 0000000..b24f870
--- /dev/null
+++ b/examples/plugins/locks
@@ -0,0 +1,83 @@
+# -*- python -*-
+
+# A xpybar configuration example testing the features of plugins.locks
+
+import time
+import threading
+
+from plugins.locks import Locks
+
+
+OUTPUT, HEIGHT, YPOS, TOP = 0, 12, 24, True
+
+
+locks = Locks().locks
+headers = ('Id', 'Syscall', 'Level', 'Shared', 'Pid', 'Device', 'Inode', 'Range', 'Mountpoint', 'Pathname')
+def convert(lock):
+    rc = []
+    rc.append(str(lock.lock_id))
+    rc.append({'FLOCK' : 'flock', 'POSIX' : 'lockf'}[lock.lock_type])
+    rc.append('mandatory' if lock.mandatory else 'advisory')
+    rc.append('yes' if lock.shared else 'no')
+    rc.append(str(lock.pid))
+    rc.append('%i:%i' % (lock.major, lock.minor))
+    rc.append(str(lock.inode))
+    rc.append('%i..%s' % (lock.start, 'eof' if lock.end is None else str(lock.end)))
+    rc += ['...'] * 2
+    return rc
+text_locks = [convert(lock) for lock in locks]
+
+HEIGHT *= len(text_locks)
+
+
+def locate_files():
+    dev_cache = {}
+    for i, lock in enumerate(locks):
+        if (lock.major, lock.minor) in dev_cache:
+            mountpoint, device = dev_cache[(lock.major, lock.minor)]
+            text_locks[i][-2] = mountpoint
+            text_locks[i][5] = '%s(%s)' % (text_locks[i][5], device)
+        else:
+            device = lock.find_device()
+            mountpoint = device[1] if device is not None else '(not found)'
+            device     = device[2] if device is not None else None
+            dev_cache[(lock.major, lock.minor)] = (mountpoint, device)
+            text_locks[i][-2] = mountpoint
+            if device is not None:
+                text_locks[i][5] = '%s(%s)' % (text_locks[i][5], device)
+        if not mountpoint.startswith('/'):
+            text_locks[i][-1] = '(not found)'
+        try:
+            with open('/proc/%i/comm' % lock.pid, 'rb') as file:
+                cmd = file.read()
+            cmd = cmd.decode('utf-8', 'replace')[:-1]
+            text_locks[i][4] = '%s(%s)' % (text_locks[i][4], cmd)
+        except:
+            pass
+    bar.invalidate()
+    for i, lock in enumerate(locks):
+        mountpoint = text_locks[i][-2]
+        if not mountpoint.startswith('/'):
+            continue
+        pathnames = lock.find_pathname(mountpoint)
+        if len(pathnames) == 0:
+            text_locks[i][-1] = '(not found)'
+        else:
+            text_locks[i][-1] = pathnames[0]
+        bar.invalidate()
+
+
+start_ = start
+def start():
+    start_()
+    async(locate_files)
+
+
+semaphore = threading.Semaphore()
+def redraw():
+    semaphore.acquire(blocking = True)
+    text = '\n'.join(' │ '.join('%s: %s' % (h, v) for (h, v) in zip(headers, lock)) for lock in text_locks)
+    bar.clear()
+    bar.draw_coloured_text(0, 10, 0, 2, text)
+    semaphore.release()
+