Update subcmd virtmon

sdata/subcmd-virtmon/0.run.sh

@@ -3,70 +3,67 @@
 
 # shellcheck shell=bash
 
-readarray -t vmons < <(hyprctl -j monitors all | jq -r '.[] | select(.name | test("^HEADLESS-")) | .name')
+ensure_cmds wayvnc lsof jq ip
 
-ensure_cmds wayvnc lsof jq
+readarray -t vmon_ids < <(hyprctl -j monitors all | jq -r '.[] | select(.name | test("^TESTER-")) | .name | sub("^TESTER-"; "")')
 
-if [[ "${CLEAN_VIRTUAL_MONITORS}" = true ]]; then
-  echo "Cleaning virtual monitors..."
-  for i in "${vmons[@]}"; do
+if [[ "${CLEAN_TESTER_MONITORS}" = true ]]; then
+  echo "Cleaning tester monitors..."
+  for i in "TESTER-${vmon_ids[@]}"; do
     x hyprctl output remove "$i"
   done
+  echo "Cleaning tester wayvnc sessions..."
+  for tester_socket in /tmp/wayvncctl_tester_* ; do
+    x wayvncctl --socket=$tester_socket -r wayvnc-exit
+  done
+  echo "Cleaning complete, exit..."
   exit 0
 fi
-if [ "${#vmons[@]}" -gt 0 ]; then
-  echo "headless monitors found:"
-  printf '%s\n' "${vmons[@]}"
-  if [[ ! "${KEEP_VIRTUAL_MONITORS}" = true ]]; then
-    echo "Cleaning..."
-    for i in "${vmons[@]}"; do
-      x hyprctl output remove "$i"
-    done
-  fi
-else
-  echo "No headless monitors found."
-fi
 
-echo "Creating tester monitor..."
-readarray -t vmons_old < <(hyprctl -j monitors all | jq -r '.[] | select(.name | test("^HEADLESS-")) | .name')
-x hyprctl output create headless
-readarray -t vmons_new < <(hyprctl -j monitors all | jq -r '.[] | select(.name | test("^HEADLESS-")) | .name')
-declare -A seen
-for e in "${vmons_old[@]}"; do
-  seen["$e"]=1
-done
-deltas=()
-for e in "${vmons_new[@]}"; do
-  if [[ -z "${seen[$e]+_}" ]]; then
-    deltas+=("$e")
-  fi
-done
-if (( ${#deltas[@]} == 1 )); then
-  vmon_tester="${deltas[0]}"
-  echo "tester monitor found: $vmons_tester"
-elif (( ${#deltas[@]} == 0 )); then
-  echo "Error: No tester monitor found"
-  exit 1
-else
-  echo "Error: multiple tester monitor found: ${deltas[*]}"
-  exit 1
-fi
-# TODO: Implement setting geometry
-#echo "Setting geometry..."
-#${vmon_tester}
 
-echo "Using wayvnc to share monitor $vmons_tester..."
+echo "Finding an unused port..."
 for port in {5900..5999}; do
   if ! lsof -nP -iTCP:"$port" -sTCP:LISTEN -t >/dev/null 2>&1; then
     vnc_port="$port"
     break
   fi
 done
-# TODO: Allow running in background and implement --stop to stop it
 if [ -z "$vnc_port" ];then
   echo "No available port for vnc server, aborting..."; exit 1
 fi
-wayvnc -S -o=${vmon_tester} --log-level=trace 0.0.0.0 $vnc_port
+# The name of tester_socket can be anything, the following just borrows $vnc_port as ID
+tester_socket=/tmp/wayvncctl_tester_$vnc_port
+# In case this exists for some reason
+try rm $tester_socket
+vmon_tester=TESTER-$vnc_port;break
 
-echo "Cleaning the tester monitor..."
-hyprctl output remove "${vmon_tester}"
+echo "Creating tester monitor..."
+x hyprctl output create headless ${vmon_tester}
+
+echo "Setting geometry..."
+x hyprctl keyword monitor ${vmon_tester},${VMON_RESOLUTION}@${VMON_FPS},${VMON_POSITION},${VMON_SCALE}${VMON_EXTRA}
+
+echo "========================================="
+echo "Use a VNC client to connect to the virtual monitor."
+echo "  Port: $vnc_port"
+echo "  IP: choose a suitable one from below:"
+ip -o addr show up | grep -v -E 'docker|veth|virbr' | awk '{split($4,a,"/"); print $2"\t"a[1]}'
+echo "The status of the virtual monitor:"
+echo "  Resolution: ${VMON_RESOLUTION}"
+echo "  Frame rate: ${VMON_FPS}"
+echo "Hint:"
+echo "  The VNC client will ask you about server address,"
+echo "  either joined as <IP>:<Port> or separately."
+echo "  As for username and password, just leave them as empty."
+echo "========================================="
+echo "Launching wayvnc..."
+
+if [ "$RUNNING_IN_BACKGROUND" = true ];then
+  nohup wayvnc --socket=$tester_socket -f=${VMON_FPS} -o=${vmon_tester} --log-level=${WAYVNC_LOGLEVEL} 0.0.0.0 $vnc_port > $(mktemp) 2>&1 &
+  disown
+else
+  echo "wayvnc now running, press Ctrl-C to quit."
+  wayvnc --socket=$tester_socket -f=${VMON_FPS} -o=${vmon_tester} --log-level=${WAYVNC_LOGLEVEL} 0.0.0.0 $vnc_port
+  echo "wayvnc stopped. Cleaning..."
+  hyprctl output remove "${vmon_tester}"
+fi