#!/usr/bin/env bash
set -eu

function remove_logs_and_cache {
  rm -r --interactive=never /var/log/mullvad-vpn/ || \
    echo "Failed to remove mullvad-vpn logs"
  rm -r --interactive=never /var/cache/mullvad-vpn/ || \
    echo "Failed to remove mullvad-vpn cache"
}

function get_home_dirs {
  if [[ -f "/etc/passwd" ]] && command -v cut > /dev/null; then
      cut -d: -f6 /etc/passwd
  fi
}

function clear_gpu_cache {
  local home_dirs
  home_dirs=$(get_home_dirs)

  for home_dir in $home_dirs; do
      local gpu_cache_dir="$home_dir/.config/Mullvad VPN/GPUCache"
      if [[ -d "$gpu_cache_dir" ]]; then
          echo "Clearing GPU cache in $gpu_cache_dir"
          rm -r --interactive=never "$gpu_cache_dir" || \
              echo "Failed to clear GPU cache"
      fi
  done
}

function remove_config {
  rm -r --interactive=never /etc/mullvad-vpn || \
    echo "Failed to remove mullvad-vpn config"

  # Remove app settings and auto-launcher for all users. This doesn't respect XDG_CONFIG_HOME due
  # to the complexity required.
  local home_dirs
  home_dirs=$(get_home_dirs)
  for home_dir in $home_dirs; do
      local mullvad_dir="$home_dir/.config/Mullvad VPN"
      if [[ -d "$mullvad_dir" ]]; then
          echo "Removing mullvad-vpn app settings from $mullvad_dir"
          rm -r --interactive=never "$mullvad_dir" || \
              echo "Failed to remove mullvad-vpn app settings"
      fi

      local autostart_path="$home_dir/.config/autostart/mullvad-vpn.desktop"
      # mullvad-vpn.desktop can be both a file or a symlink.
      if [[ -f "$autostart_path" || -L "$autostart_path" ]]; then
          echo "Removing mullvad-vpn app autostart file $autostart_path"
          rm --interactive=never "$autostart_path" || \
              echo "Failed to remove mullvad-vpn autostart file"
      fi
  done
}

# checking what kind of an action is taking place
case $@ in
  # apt purge passes "purge"
  "purge")
    remove_logs_and_cache
    remove_config
    ;;
  # apt remove passes "remove"
  "remove")
    ;;
  # yum remove passes a 0
  "0")
    remove_logs_and_cache
    remove_config
    ;;
esac

# Different electron versions can have incompatible GPU caches. Clearing it on upgrades makes sure
# the same cache is not used across versions.
clear_gpu_cache

# Remove apparmor profile
if apparmor_parser -R /etc/apparmor.d/mullvad &>/dev/null; then
    echo "Removing apparmor profile"
    rm -f /etc/apparmor.d/mullvad || echo "Failed to delete apparmor profile"
fi