From 6e551b4111d623521a6bcd4be958f544b298b026 Mon Sep 17 00:00:00 2001
From: piotr <nwg.piotr@gmail.com>
Date: Tue, 14 Feb 2023 01:38:26 +0100
Subject: [PATCH] add processes entry point

---
 nwg_panel/processes.py | 104 +++++++++++++++++++++++++++++++++++++++++
 setup.py               |   5 +-
 2 files changed, 107 insertions(+), 2 deletions(-)
 create mode 100644 nwg_panel/processes.py

diff --git a/nwg_panel/processes.py b/nwg_panel/processes.py
new file mode 100644
index 0000000..900c7ab
--- /dev/null
+++ b/nwg_panel/processes.py
@@ -0,0 +1,104 @@
+import os
+
+import psutil
+from i3ipc import Connection
+import gi
+gi.require_version('Gtk', '3.0')
+from gi.repository import Gtk, Gdk, GLib, Gio
+
+grid = None
+protected = ["systemd", "bash"]
+
+theme = Gtk.IconTheme.get_default()
+
+def handle_keyboard(win, event):
+    if event.type == Gdk.EventType.KEY_RELEASE and event.keyval == Gdk.KEY_Escape:
+        win.destroy()
+
+
+def terminate(btn, pid):
+    print("Killing {}".format(pid))
+    os.kill(pid, 2)
+
+
+def list_processes(box):
+    tree = Connection().get_tree()
+    processes = {}
+
+    for proc in psutil.process_iter(['pid', 'ppid', 'name', 'username']):
+        if proc.info['username'] == os.getenv('USER') and proc.info['ppid'] == 1:
+            processes[proc.info['pid']] = proc.info['name']
+
+    global grid
+    if grid:
+        grid.destroy()
+
+    grid = Gtk.Grid.new()
+    grid.set_row_spacing(6)
+    grid.set_column_spacing(6)
+    box.pack_start(grid, True, True, 0)
+
+    lbl = Gtk.Label()
+    lbl.set_markup("<b>PID</b>")
+    lbl.set_property("halign", Gtk.Align.END)
+    grid.attach(lbl, 0, 0, 1, 1)
+
+    lbl = Gtk.Label()
+    lbl.set_markup("<b>Name</b>")
+    lbl.set_property("halign", Gtk.Align.START)
+    grid.attach(lbl, 2, 0, 1, 1)
+
+    lbl = Gtk.Label()
+    lbl.set_markup("<b>Kill</b>")
+    lbl.set_property("halign", Gtk.Align.START)
+    grid.attach(lbl, 3, 0, 1, 1)
+
+    idx = 1
+    for pid in processes:
+        if not tree.find_by_pid(pid):
+            lbl = Gtk.Label.new(str(pid))
+            lbl.set_property("halign", Gtk.Align.END)
+            grid.attach(lbl, 0, idx, 1, 1)
+
+            name = processes[pid]
+
+            if theme.lookup_icon(name, 16, Gtk.IconLookupFlags.FORCE_SYMBOLIC):
+                img = Gtk.Image.new_from_icon_name(name, Gtk.IconSize.MENU)
+                img.set_property("halign", Gtk.Align.END)
+                grid.attach(img, 1, idx, 1, 1)
+
+            lbl = Gtk.Label.new(name)
+            lbl.set_property("halign", Gtk.Align.START)
+            grid.attach(lbl, 2, idx, 1, 1)
+
+
+
+            if processes[pid] not in protected:
+                btn = Gtk.Button.new_from_icon_name("gtk-close", Gtk.IconSize.MENU)
+                btn.connect("clicked", terminate, pid)
+                grid.attach(btn, 3, idx, 1, 1)
+
+            idx += 1
+    grid.show_all()
+    return True
+
+
+def main():
+    win = Gtk.Window.new(Gtk.WindowType.TOPLEVEL)
+    win.connect('destroy', Gtk.main_quit)
+    win.connect("key-release-event", handle_keyboard)
+
+    box = Gtk.Box.new(Gtk.Orientation.VERTICAL, 0)
+    box.set_property("margin", 12)
+    win.add(box)
+
+    list_processes(box)
+
+    win.show_all()
+
+    GLib.timeout_add(1000, list_processes, box)
+    Gtk.main()
+
+
+if __name__ == '__main__':
+    main()
diff --git a/setup.py b/setup.py
index 4c95050..c2fc389 100644
--- a/setup.py
+++ b/setup.py
@@ -8,7 +8,7 @@ def read(f_name):
 
 setup(
     name='nwg-panel',
-    version='0.7.16',
+    version='0.7.17',
     description='GTK3-based panel for sway window manager',
     packages=find_packages(),
     include_package_data=True,
@@ -25,7 +25,8 @@ setup(
         'gui_scripts': [
             'nwg-panel = nwg_panel.main:main',
             'nwg-panel-config = nwg_panel.config:main',
-            'nwg-dwl-interface = nwg_panel.dwl_interface:main'
+            'nwg-dwl-interface = nwg_panel.dwl_interface:main',
+            'nwg-processes = nwg_panel.processes:main'
         ]
     }
 )