mirror of
https://github.com/wizbright/waybox.git
synced 2025-10-29 05:40:20 -04:00
45 lines
1.9 KiB
Text
45 lines
1.9 KiB
Text
|
#!/bin/env python
|
||
|
|
|
||
|
|
import os, subprocess, sys
|
||
|
|
import xml.dom.minidom as dom
|
||
|
|
|
||
|
|
def get_waybox_pid(name='waybox'):
|
||
|
|
lines = subprocess.Popen(['pgrep', '-x', name], stdout=subprocess.PIPE).stdout.read().splitlines()
|
||
|
|
if lines:
|
||
|
|
return int(lines[0])
|
||
|
|
else:
|
||
|
|
return 0
|
||
|
|
|
||
|
|
def get_menu_items(menu, indent):
|
||
|
|
print("%s⇢ %s\0nonselectable\x1ftrue\x1ficon\x1fgo-next-symbolic" % (indent, menu.getAttribute('label').replace('_', '') or menu.getAttribute('id')))
|
||
|
|
indent = indent + " "
|
||
|
|
children = menu.childNodes
|
||
|
|
for i in range(0, children.length):
|
||
|
|
if children[i].nodeType == dom.Node.ELEMENT_NODE and children[i].tagName == 'item':
|
||
|
|
action = children[i].getElementsByTagName('action')[0].getAttribute('name')
|
||
|
|
cmd = ""
|
||
|
|
icon = children[i].getAttribute('icon')
|
||
|
|
label = children[i].getAttribute('label').replace('_', '')
|
||
|
|
if action == 'Execute':
|
||
|
|
cmd = children[i].getElementsByTagName('execute')[0].firstChild.nodeValue
|
||
|
|
elif action == 'Exit':
|
||
|
|
cmd = "kill -s SIGTERM %d" % get_waybox_pid()
|
||
|
|
elif action == 'Reconfigure':
|
||
|
|
cmd = "kill -s SIGUSR2 %d" % get_waybox_pid()
|
||
|
|
print("%s %s\0info\x1f%s\x1ficon\x1f%s" % (indent, label, cmd, icon))
|
||
|
|
elif children[i].nodeType == dom.Node.ELEMENT_NODE and children[i].tagName == 'menu':
|
||
|
|
get_menu_items(children[i], indent)
|
||
|
|
|
||
|
|
menu_xml = os.getenv("WB_MENU_XML") or "menu.xml"
|
||
|
|
# If ran as a rofi script (not possible with wofi)
|
||
|
|
if not os.getenv('ROFI_RETV') is None:
|
||
|
|
import shlex
|
||
|
|
print("\0message\x1f%s" % os.path.abspath(menu_xml))
|
||
|
|
if not (info := os.getenv('ROFI_INFO')) is None:
|
||
|
|
subprocess.Popen(shlex.split(info), stdout=subprocess.PIPE)
|
||
|
|
sys.exit(0)
|
||
|
|
|
||
|
|
document = dom.parse(menu_xml)
|
||
|
|
root_menu = document.getElementsByTagName('menu')[0]
|
||
|
|
get_menu_items(root_menu, "")
|