#!/bin/bash
#
# Check if the X11 socket is present and if not, link with wslg mount
#
function main() {

  # shellcheck disable=SC2001,SC2155
  local display_number=$(echo "${1}" | sed 's/[^0-9]*//g')

  if [ -S /tmp/.X11-unix/X"${display_number}" ]; then
    echo "${display_number}"
  else
    umount /tmp/.X11-unix 2>/dev/null
    rm -rf /tmp/.X11-unix
    ln -s /mnt/wslg/.X11-unix /tmp/.X11-unix

    display_number=$(ls /tmp/.X11-unix/ | sed 's/[^0-9]*//g')

    if [ -S /tmp/.X11-unix/X"${display_number}" ]; then
      echo "${display_number}"
    else
      echo "0"
    fi
  fi
}

main "$@"
