#!/bin/env bash
#
## Calculate DPI, update settings if necessary, then start desktop apps

MONITOR=$(xrandr --listmonitors | grep -E '^\s+[0-9]' | head -1 | sed -r 's/^.*\s+(.*)$/\1/')
REGEX_XRANDR='^.* ([0-9]+)x([0-9]+)\+[0-9]+\+[0-9]+.* ([0-9]+)mm x ([0-9]+)mm.*$'
TEST_STATION_WALLPAPERS='/usr/share/wallpaper/test-stations'

# Resize screen in VMs
if lsmod | grep -Eq 'qxl|virtio_gpu'; then
  echo -n "Starting VM guest services..."
  spice-vdagent
  sleep 0.5
  xrandr --output "${MONITOR}" --auto
fi

echo -n "Getting display details... "

# Get screen data
xrandr_str="$(xrandr | grep mm | head -1)"
width_px="$(echo "${xrandr_str}" | sed -r "s/${REGEX_XRANDR}/\1/")"
height_px="$(echo "${xrandr_str}" | sed -r "s/${REGEX_XRANDR}/\2/")"
width_mm="$(echo "${xrandr_str}" | sed -r "s/${REGEX_XRANDR}/\3/")"
height_mm="$(echo "${xrandr_str}" | sed -r "s/${REGEX_XRANDR}/\4/")"

# Convert to in
width_in="$(echo "${width_mm} * 0.03937" | bc)"
height_in="$(echo "${height_mm} * 0.03937" | bc)"

# Calculate diagonals
diag_px="$(echo "sqrt(${width_px}^2 + ${height_px}^2)" | bc)"
diag_in="$(echo "sqrt(${width_in}^2 + ${height_in}^2)" | bc)"

# Calculate DPI
dpi="$(echo "${diag_px} / ${diag_in}" | bc 2>/dev/null || True)"
dpi="${dpi:-0}"

# Save data
echo "width_px=$width_px" > "$HOME/.screen_data"
echo "height_px=$height_px" >> "$HOME/.screen_data"
echo "dpi=$dpi" >> "$HOME/.screen_data"

echo "Done"

# Update settings if necessary
if [[ "${dpi}" -ge 192 ]]; then
  echo -n "Updating settings for HiDPI... "

  # Fonts
  sed -i 's/!Xft.dpi: 192/Xft.dpi: 192/' "${HOME}/.Xresources"
  sed -i 's/pixelsize=12/pixelsize=24/' "${HOME}/.Xresources"

  # GDK
  export GDK_SCALE=2
  export GDK_DPI_SCALE=0.5

  # Rofi
  sed -i -r 's/}/  dpi: 1;\n}/' "${HOME}/.config/rofi/config.rasi"
  sed -i 's/10px/20px/' "${HOME}/.config/rofi/applets/menu/configs/rounded/powermenu.rasi"
  sed -i 's/12px/24px/' "${HOME}/.config/rofi/applets/menu/configs/rounded/powermenu.rasi"
  sed -i 's/15px/30px/' "${HOME}/.config/rofi/applets/menu/configs/rounded/powermenu.rasi"
  sed -i 's/20px/40px/' "${HOME}/.config/rofi/applets/menu/configs/rounded/powermenu.rasi"
  sed -i 's/40px/80px/' "${HOME}/.config/rofi/applets/menu/configs/rounded/powermenu.rasi"
  sed -i 's/800px/2000px/' "${HOME}/.config/rofi/applets/menu/configs/rounded/powermenu.rasi"

  # Tint2
  sed -i 's/panel_size = 99% 30/panel_size = 99% 60/' \
    "${HOME}/.config/tint2/tint2rc"
  sed -i 's/Hack 10/Hack 20/g' \
    "${HOME}/.config/tint2/tint2rc"
  sed -i 's/Hack 12/Hack 24/g' \
    "${HOME}/.config/tint2/tint2rc"
  sed -i 's/systray_icon_size = 24/systray_icon_size = 48/' \
    "${HOME}/.config/tint2/tint2rc"

  # Done
  echo "Done"
fi

# Update conky
echo -n "Updating conky... "
$HOME/.update_conky
echo "Done"

# Update X
echo -n "Updating X... "
xset s off
xset -dpms
xrdb -merge $HOME/.Xresources
echo "Done"

# Set wallpaper
echo -n "Setting wallpaper... "
_hostname="$(hostnamectl --static | sed 's/.1201.com//')"
_ip="$(ip a show scope global \
  | grep inet \
  | head -1 \
  | sed -r 's#.*inet ([0-9]+.[0-9]+.[0-9]+.[0-9]+.)/.*#\1#')"
if [[ -e "$TEST_STATION_WALLPAPERS/${_hostname:+x}" ]]; then
  ln -sf "$TEST_STATION_WALLPAPERS/${_hostname:+x}" "$HOME/.wallpaper"
elif [[ -e "$TEST_STATION_WALLPAPERS/${_ip:+x}" ]]; then
  ln -sf "$TEST_STATION_WALLPAPERS/${_ip:+x}" "$HOME/.wallpaper"
fi
feh --bg-fill "$HOME/.wallpaper"
echo "Done"