feat: add new cwd urxvt starter

herbstluftwm/hl_start_urxvt_cwd.py

@@ -1,55 +1,79 @@
-#!/usr/bin/python3
-
 import subprocess
 import datetime
 import sys
-import os
 import shlex
 import psutil
-from hl_utils import error, shexec
 
-out = shexec('xdpyinfo').split('\n')
+WHITESPACE = " "
+NEWLINE = "\n"
+CONSOLE_TYPE = "urxvt"
+HEX_BASE = 16
 
-############ GET FOCUSED WINDOW ID ##############
+def get_focused_window():
-window = -1
+    '''Ask the X-Server for the window with focus'''
-for l in out:
-    if l.startswith('focus:'):
-        for field in l.split(' '):
-            field = field.rstrip(',')
-            if(field.startswith('0x')):
-                window = int(field,16)
-        if window == -1:
-            error("Failed to get window_focus\n")
-if window==-1:
-    sys.exit()
 
-############ IF URXVT GET PID ###########
+    xdpyinfo = subprocess.check_output('xdpyinfo', universal_newlines=True)
-pid = -1
-out = subprocess.check_output(['xprop','-id',hex(window)]).decode().split('\n')
-#out = shexec('xprop -id '+hex(window)).split('\n') #not working
-for l in out:
-    if l.startswith('WM_CLASS(STRING)') and 'urxvt' in l:
-        break;
-else:
-    sys.exit();
 
-for l in out:
+    for line in xdpyinfo.split(NEWLINE):
-    if l.startswith('_NET_WM_PID(CARDINAL)'):
+        print(line)
-        pid = int(l.split(' ')[-1])
+        if line.startswith("focus:"):
-if pid==-1:
+            print("focus")
-    error("Failed to get PID")
 
-############ START NEW TERMINAL ############
+            # this line will look something like 'focus:  window 0x2200009, revert to PointerRoot' #
-process = psutil.Process(pid)
+            window_id = 0
-for p in process.children(): #recursive=false
+            for field in line.split(WHITESPACE):
-    if p.name() in ['zsh','bash']:
+                if field.startswith("0x"):
+                    window_id = int(field.rstrip(","), HEX_BASE)
+                    break
+            
+            if window_id <= 0:
+                print("No Valid Focused Window Found", file=sys.stderr)
+                sys.exit(1)
 
-        # check for ssh #
+            return window_id
-        for sp in p.children(recursive=True):
-            print(sp.name())
-            if sp.name() == 'ssh':
-                print(sp)
 
-        subprocess.Popen(shlex.split('urxvt -cd ' + p.cwd()))
+    print("No focus line found in xdpyinfo", file=sys.stderr)
-        break
+    sys.exit(1)
-sys.exit()
+
+def get_window_console_pid(window_id):
+    '''Retrive the PID for a given window_id if the windows contains a console'''
+
+    x_properties = subprocess.check_output(['xprop','-id', hex(window_id)], universal_newlines=True)
+    
+    selected = None
+    pid = -1
+
+    for line in x_properties.split(NEWLINE):
+
+        if line.startswith("WM_CLASS(STRING)") and CONSOLE_TYPE in line:
+            # looks like this: WM_CLASS(STRING) = "urxvt", "URxvt #
+            selected = line
+
+        if line.startswith('_NET_WM_PID(CARDINAL)'):
+            # looks like this: _NET_WM_PID(CARDINAL) = 91129 #
+            pid = int(line.split(" = ")[-1])
+
+    
+    if not selected or pid <= 0:
+        print(f"Window does not exist or does not container f{CONSOLE_TYPE}", file=sys.stderr)
+        sys.exit(1)
+
+    return pid
+
+def get_path_for_console_pid(pid):
+    '''Retrieve the currently open path for a given console'''
+
+    process = psutil.Process(pid)
+    for p in process.children(): #recursive=false
+        if p.name() in ['zsh','bash']:
+            return p.cwd()
+
+
+if __name__ == "__main__":
+    
+    window_id = get_focused_window()
+    pid = get_window_console_pid(window_id)
+    path = get_path_for_console_pid(pid)
+    
+    # run the new urxvt #
+    subprocess.Popen(["urxvt", "-cd", path])

herbstluftwm/start_urxvt_child.py

@@ -0,0 +1,79 @@
+import subprocess
+import datetime
+import sys
+import shlex
+import psutil
+
+WHITESPACE = " "
+NEWLINE = "\n"
+CONSOLE_TYPE = "urxvt"
+HEX_BASE = 16
+
+def get_focused_window():
+    '''Ask the X-Server for the window with focus'''
+
+    xdpyinfo = subprocess.check_output('xdpyinfo', universal_newlines=True)
+
+    for line in xdpyinfo.split(NEWLINE):
+        print(line)
+        if line.startswith("focus:"):
+            print("focus")
+
+            # this line will look something like 'focus:  window 0x2200009, revert to PointerRoot' #
+            window_id = 0
+            for field in line.split(WHITESPACE):
+                if field.startswith("0x"):
+                    window_id = int(field.rstrip(","), HEX_BASE)
+                    break
+            
+            if window_id <= 0:
+                print("No Valid Focused Window Found", file=sys.stderr)
+                sys.exit(1)
+
+            return window_id
+
+    print("No focus line found in xdpyinfo", file=sys.stderr)
+    sys.exit(1)
+
+def get_window_console_pid(window_id):
+    '''Retrive the PID for a given window_id if the windows contains a console'''
+
+    x_properties = subprocess.check_output(['xprop','-id', hex(window_id)], universal_newlines=True)
+    
+    selected = None
+    pid = -1
+
+    for line in x_properties.split(NEWLINE):
+
+        if line.startswith("WM_CLASS(STRING)") and CONSOLE_TYPE in line:
+            # looks like this: WM_CLASS(STRING) = "urxvt", "URxvt #
+            selected = line
+
+        if line.startswith('_NET_WM_PID(CARDINAL)'):
+            # looks like this: _NET_WM_PID(CARDINAL) = 91129 #
+            pid = int(line.split(" = ")[-1])
+
+    
+    if not selected or pid <= 0:
+        print(f"Window does not exist or does not container f{CONSOLE_TYPE}", file=sys.stderr)
+        sys.exit(1)
+
+    return pid
+
+def get_path_for_console_pid(pid):
+    '''Retrieve the currently open path for a given console'''
+
+    process = psutil.Process(pid)
+    for p in process.children(): #recursive=false
+        if p.name() in ['zsh','bash']:
+            return p.cwd()
+
+
+if __name__ == "__main__":
+    
+    window_id = get_focused_window()
+    pid = get_window_console_pid(window_id)
+    path = get_path_for_console_pid(pid)
+    
+    # run the new urxvt #
+    subprocess.Popen(["urxvt", "-cd", path])