1 files changed, 11 insertions, 7 deletions

diff --git a/fireslay b/fireslay
index 3e52a84..9485cd4 100755
--- a/fireslay
+++ b/fireslay
@@ -80,14 +80,18 @@ web_content_pids()
         then
             echo $pid
         fi
-    done < $FIREFOX_GROUP/cgroup.procs
+    done < $FIREFOX_GROUP_PROCS
 }
 
-CURRENT_GROUP=$(get_current_group)
-FIREFOX_GROUP=${CURRENT_GROUP%/*}/firefox
-FIREFOX_GROUP_PROCS=$FIREFOX_GROUP/cgroup.procs
+uid=$(getent passwd $SUDO_USER | (IFS=: read _ _ id _; echo $id))
+group=user.slice/user-$uid.slice/firefox
+group_procs=/sys/fs/cgroup/$group/cgroup.procs
 
-[ -e "$FIREFOX_GROUP_PROCS" ]
-
-main "$@"
+FIREFOX_GROUP_PROCS=$group_procs
+if [ -e "$FIREFOX_GROUP_PROCS" ]
+then
+    main "$@"
+else
+    echo "Error: Firefox group not found ($FIREFOX_GROUP_PROCS)" >&2
+fi