#!/usr/bin/env bash

# This work and script was adapted by the work of user LinSoftWin. If it wasn't for him - it wouldn't have happened.
# This script downloads a PIRATE version of Photoshop, because it is not possible to run the actual version from Adobe Creative Cloud.
# Use it at your own risk. The license applies only to the files in this repository.
# I am not responsible for any other files downloaded from other links using the script.
# If the link becomes inactive - it will be replaced by another hosting. Checksums of uploaded files will also be updated.

# In case the user does not use the XDG Base Directory Specification
# https://specifications.freedesktop.org/basedir-spec/latest
XDG_DATA_HOME="$HOME/.local/share"
XDG_CACHE_HOME="$HOME/.cache"

#                            LOGGING SYSTEM
# #####################################################################
#

BLUE="\e[1;34m"   # Blue
RED="\e[1;31m"    # Red
YELLOW="\e[1;33m" # Yellow
GREEN="\e[1;32m"  # Green
RESET="\e[0m"     # Reset colors

LOG="${BLUE}[LOG]${RESET}"
WARNING="${YELLOW}[WARNING]${RESET}"
ERROR="${RED}[ERROR]${RESET}"
SUCCES="${GREEN}[SUCCES]${RESET}"
CHECK="${GREEN}[CHECK]${RESET}"

#						CHECKS & OTHER FUNCTIONS
# ###################################################################
#

if [ -z "$XDG_DATA_HOME" ] && [ -z "$XDG_CACHE_HOME" ]; then
	echo -e "$WARNING Please set variables ${YELLOW}XDG_DATA_HOME${RESET}, ${YELLOW}XDG_CACHE_HOME${RESET} and others ${YELLOW}XDG_*${RESET} according to the XDG Base Directory specification."
fi

# Photoshop URL
PHOTOSHOP_URL="https://spyderrock.com/kMnq2220-AdobePhotoshop2021.xz"

# sha256 checksum
CHECKSUM="8321b969161f2d2ad736067320d493c5b6ae579eaab9400cd1fda6871af2c033"

LAUNCHER="$HOME/.local/bin/photoshop/photoshop.sh"
LOCAL_ARCHIVE=""
ICON=""

trap on_interrupt SIGINT

on_interrupt() {
	trap "exit 1" SIGINT

	echo -e "\n$WARNING User intrrupt!"

	if [ -d "$INSTALL_PATH" ]; then
		if ask_user "Do you want to ${RED}delete${RESET} a newly created folder?"; then
			if rm -rfv "${INSTALL_PATH:?}" 2>>./install_log.log; then
				exit 0
			else
				echo -e "$ERORR The last command ended with an error."
				exit 1
			fi
		else
			echo -e "$LOG Exiting."
			exit 1
		fi
	else
		exit 1
	fi
}

get_help() {
	echo "Usage: ./install.sh [options...] <absolute path>"
	echo "  -a    Use already existing Photoshop.tar.xz"
	echo "  -i    Install Photoshop"
	echo "  -h    Show this help"
}

ask_user() {
	while true; do
		read -r -p "$(echo -e "${WARNING} $* (yes/no): ")" answer

		case "$answer" in
		[yY] | [yY][eE][sS])
			return 0
			;;
		[nN] | [nN][oO])
			return 1
			;;

		*)
			echo "Invalid input, try again"
			;;
		esac
	done
}

OS_ID=$(source /etc/os-release && echo "${ID}")
DEPENDENCIES=()

case "${OS_ID}" in
"arch")
	DEPENDENCIES=(curl wine winetricks imagemagick)
	;;
"redos")
	winetricks_opensuse="https://ftp.lysator.liu.se/pub/opensuse/distribution/leap/15.6/repo/oss/x86_64/winetricks-20240105-bp156.1.1.x86_64.rpm"
	DEPENDENCIES=(curl wine ImageMagick zstd "${winetricks_opensuse}")
	;;
*)
	echo -e "$ERROR For now only ${BLUE}Arch Linux${RESET} is supported."
	exit 1
	;;
esac

# Prints error message and exits with code 1
# $1 - message
print_err() {
  local msg=$1
  echo -e "${ERROR} ${msg}" >&2
  exit 1
}

install_deps() {

  _bold=$(tput bold)

	# Refer to /etc/os-release for more info
	case "${OS_ID}" in
	"arch")
	  if ! ask_user "Script will use '${_bold}${RED}sudo${RESET}'. Proceed?"; then
		  echo -e "$LOG Exiting."
		  exit 1
	  fi
		if ! sudo pacman -S "${DEPENDENCIES[@]}"; then
			print_err "Pacman terminated with an error."
    fi
		;;
  "redos")
    if ! pkexec dnf install "${DEPENDENCIES[@]}" -y --comment "Installed via 'photoshop-linux' script" ;then 
			print_err "${_bold}DNF${RESET} terminated with an error."
    fi
    ;;
	*)
		print_err "For now only ${BLUE}Arch Linux${RESET} and ${RED}RED OS${RESET} is supported."
		;;
	esac
	echo -e "$LOG Missing dependencies was installed"
}

#							MAIN SCRIPT
# ###################################################################
#

is_path_exists() {
	if [ -d "$1" ]; then
		# BUG
		# echo -e "$WARNING The specified path '$1' already exists."
		echo -e "$WARNING The specified path '${YELLOW}${1}${RESET}' already exists."

		if ask_user "Do you want to ${RED}delete${RESET} previous installation?"; then
			if rm -rfv "${1:?}" 2>>./install_log.log; then
				echo -e "$LOG Deleted old installation."
			else
				echo -e "$ERROR Something went wrong."
				exit 1
			fi
		else
			echo -e "$LOG Exiting."
			exit 1
		fi
	fi
}

setup_wine() {
	export WINEPREFIX="$INSTALL_PATH"
	local vc_libraries=("vcrun2003" "vcrun2005" "vcrun2010" "vcrun2012" "vcrun2013" "vcrun2022")

	echo -e "$LOG Setting up wine prefix."
	winecfg /v win10 2>/dev/null

	echo -e "$LOG Downloading and installing core components for wine prefix. This could take some time."

	if ! winetricks --unattended corefonts win10 vkd3d dxvk2030 msxml3 msxml6 gdiplus &>./install_log.log; then
		echo -e "$ERORR Winetricks terminated with an error."
		echo -e "$ERROR Please open an issue by mentioning the contents of ${YELLOW}./install_log.log${RESET}."
		exit 1
	fi

	{
		echo "---------------------------------------------------------------------"
		echo "                  Downloading Visual C++ Libraries				   "
		echo "---------------------------------------------------------------------"
	} >>./install_log.log # Thanks to Katy248 for the idea.

	echo -e "$LOG Downloading and installing Visual C++ libraries."
	if ! winetricks --unattended "${vc_libraries[@]}" &>>./install_log.log; then
		echo -e "$ERROR Winetricks terminated with an error. Please, refer to ${YELLOW}install_log.log${RESET} for more info."
		echo -e "$ERROR If you can't solve the issue yourself, please, open an issue on the GitHub."
		exit 1
	fi
}

download_photoshop() {
	local archive_name="Photoshop.tar.xz"

	if [ -f "./${archive_name}" ]; then
		echo -e "$LOG Found existing archive in current folder."
		echo -e "$LOG Comparing checksums."
		# TODO:
		# separate function to avoid repeating this task
		local local_checksum
		local_checksum="$(sha256sum "$archive_name" | awk '{print  $1}')"

		if [[ "$CHECKSUM" != "$local_checksum" ]]; then
			echo -e "$LOG Checksums don't match!"
			echo -e "$LOG Deleting corrupted archive."
			rm -v "${archive_name:?}" &>>./install_log.log
		fi

		return 0
	fi

	echo -e "$LOG Downloading Photoshop (1.1G)."
	if ! curl --progress-bar "$PHOTOSHOP_URL" -o "$archive_name"; then
		# TODO:
		# separate function to avoid repeating
		echo -e "$ERROR An error occurred during the download. Please, refer to ${YELLOW}install_log.log${RESET} for more info."
		echo -e "$ERROR If you can't solve the issue yourself, please, open an issue on the GitHub."
		exit 1
	fi

	# TODO:
	# A separate function so you don't have to write this code multiple times
	echo -e "$LOG Comparing checksums."
	local local_checksum
	local_checksum="$(sha256sum "$archive_name" | awk '{print  $1}')"

	if [[ "$CHECKSUM" != "$local_checksum" ]]; then
		echo -e "$ERROR Checksums don't match!"
		exit 1

	# TODO
	# 	while true; do
	# 		read -p "$(echo -e "$LOG_WARNING")[WARNING]$(echo -e "$LOG_RESET") Do you want to redownload it again? (yes/no): " answer
	# 		case "$answer" in
	# 			pattern)
	# 				command ...
	# 				;;
	# 			*)
	# 				command ...
	# 				;;
	# 		esac
	# 	done
	fi
}

verify_path() {
	local path="$1"

	# Check the validity of the path if the user has specified the absolute path manually. This is necessary in case the user accidentally misspells $HOME paths.
	# https://github.com/shvedes/photoshop-linux/issues/1
	if [[ ! "$path" == "$HOME"* ]]; then
		echo -e "$ERROR Cannot validade ${YELLOW}\$HOME${RESET} path."
		exit 1
	fi

	# Fix trailing slashes if needed
	path="$(echo "$path" | sed 's/\/\+$//')"
	INSTALL_PATH="$path"

	# Remove the last folder from the given path (as it will be created by wineprefix) and check the remaining path for validity.
	local reformatted_path
	reformatted_path="$(echo "$path" | sed 's/\/[^\/]*$//')"

	if [ -d "$reformatted_path" ]; then
		if [[ "$reformatted_path" == "$HOME" ]]; then
			return 0
		else
			echo -e "$CHECK Directory '${YELLOW}${reformatted_path}${RESET}' exist."
		fi
	else
		echo -e "$ERORR Path $reformatted_path does not exist!"
		exit 1
	fi
}

# TODO:
# Do not use the same checks multiple times
install_photoshop() {
	if [ -z "$LOCAL_ARCHIVE" ]; then
		# echo -e "${LOG_NORMAL}[LOG]${LOG_RESET} Installing Photoshop."
		local filename="Photoshop.tar.xz"

		echo -e "$LOG Extracting Photoshop."
		if ! tar xvf "$filename" &>>./install_log.log; then
			echo -e "$ERORR An error occurred while unpacking the archive."
			exit 1
			# TODO:
			# A separate function so you don't have to write this code multiple times
			# while true; do
			# 	read -p "Delete wine prefix?"
			# 	command ...
			# done
		fi

		echo -e "$LOG Installing Photoshop."
		if ! mv "./Adobe Photoshop 2021" "$INSTALL_PATH/drive_c/Program Files"; then
			echo -e "$ERROR An error occurred during installation."
			exit 1
		fi
	else
		echo -e "$LOG Using given local Photoshop archive."

		if [[ ! "$LOCAL_ARCHIVE" = *.tar.xz ]]; then
			echo -e "$ERORR Only tar.xz is accepted for now."
			exit 1
			# TODO:
			# Allow user to use not only tar.xz / archive from another sources
		fi

		# TODO:
		# A separate function so you don't have to write this code multiple times
		echo -e "$LOG Comparing checksums."

		local local_checksum
		local_checksum="$(sha256sum "$LOCAL_ARCHIVE" | awk '{print  $1}')"

		# TODO:
		# Allow user to skip checksum comparing
		if [[ "$CHECKSUM" != "$local_checksum" ]]; then
			echo -e "$ERROR Checksums don't match!"
			exit 1
		fi

		echo -e "$LOG Extracting Photoshop."
		if ! tar xvf "$LOCAL_ARCHIVE" &>>./install_log.log; then
			echo -e "$ERROR An error occurred while unpacking the archive."
			exit 1
			# TODO:
			# A separate function so you don't have to write this code multiple times
			# while true; do
			# 	read -p "Delete wine prefix?"
			# 	command ...
			# done
		fi

		echo -e "$LOG Installing Photoshop."
		if ! mv "./Adobe Photoshop 2021" "$INSTALL_PATH/drive_c/Program Files"; then
			echo -e "$ERROR An error occurred during installation. Please, refer to ${YELLOW}install_log.log${RESET} for more info."
			echo -e "$ERROR If you can't solve the issue yourself, please, open an issue on the GitHub."
			exit 1
		fi
	fi
}

install_icon() {
	# Papirus Icon Theme already has a Photoshop icon in it.
	# The script will check if you have Papirus installed and use its icon. If Papirus is not installed, the script will download the icon from the Internet and use it.
	if find /usr/share/icons -name "Papirus*" &>/dev/null; then
		ICON="photoshop"
	else
		if [ -d "$XDG_DATA_HOME/icons" ]; then
			if find "$XDG_DATA_HOME/icons" -name "Papirus*" &>/dev/null; then
				ICON="photoshop"
			fi
		else
			mkdir "$XDG_DATA_HOME/icons"
		fi
	fi

	if [ -z "$ICON" ]; then
		local icon_url="https://cdn3d.iconscout.com/3d/premium/thumb/adobe-photoshop-file-3d-icon-download-in-png-blend-fbx-gltf-formats--logo-format-graphic-design-pack-development-icons-9831950.png"
		if ! curl "$icon_url" -o "icon.webp" &>>./install_log.log; then
			echo -e "$ERROR Failed to download icon. Please refer ${YELLOW}install_log.log${RESET} for info."
			exit 1
		fi

		magick "icon.webp" "icon.png"
		rm "./icon.webp"

		echo -e "$LOG Installing icon for .desktop entry."
		mv "./icon.png" "$XDG_DATA_HOME/icons/photoshop.png"
		ICON="$XDG_DATA_HOME/icons/photoshop.png"
	fi
}

install_desktop_entry() {
	if [ ! -d "$XDG_DATA_HOME/applications" ]; then
		mkdir "$XDG_DATA_HOME/applications"
	fi

	local path="$XDG_DATA_HOME/applications/photoshop.desktop"

	echo -e "$LOG Genarating application menu item."

	echo "[Desktop Entry]" >"$path"
	echo "Name=Adobe Photoshop CC 2021" >>"$path"
	echo "Exec=bash -c "$HOME/.local/bin/photoshop/photoshop.sh %F"" >>"$path"
	echo "Type=Application" >>"$path"
	echo "Comment=The industry-standard photo editing software (Wine" >>"$path"
	echo "Categories=Graphics" >>"$path"
	echo "Icon=$ICON" >>"$path"
	echo "MimeType=image/psd;image/x-psd;image/png;image/jpg;image/jpeg;image/webp;image/heif;image/raw" >>"$path"
	echo "StartupWMClass=photoshop.exe" >>"$path"
}

install_launcher() {
	mkdir -p "$HOME/.local/bin/photoshop"
	echo -e "$LOG Installing launcher."

	# Thanks to Katy248 (https://github.com/Katy248) for the idea.
	{
		echo "#!/usr/bin/env bash"
		echo ""
		echo "export WINEPREFIX=\"$WINEPREFIX\""
		echo "LOG_FILE=\"\$XDG_CACHE_HOME/photoshop.log\""
		echo "DXVK_LOG_PATH=\"\$WINEPREFIX/dxvk_cache\""
		echo "DXVK_STATE_CACHE_PATH=\"\$WINEPREFIX/dxvk_cache\""
		echo "PHOTOSHOP=\"\$WINEPREFIX/drive_c/Program Files/Adobe Photoshop 2021/photoshop.exe\""
		echo ""
		echo "echo -e \"All logs are saved in \$LOG_FILE\""
		echo "wine64 \"\$PHOTOSHOP\" \"\$@\" &> \"\$LOG_FILE\" "
	} >"$LAUNCHER"

	chmod +x "$LAUNCHER"
}

main() {
	install_deps

	verify_path "$INSTALL_PATH"
	is_path_exists "$INSTALL_PATH"
	setup_wine

	if [ -z "$LOCAL_ARCHIVE" ]; then
		download_photoshop
		install_photoshop
	else
		install_photoshop
	fi

	install_icon
	install_desktop_entry
	install_launcher

	echo -e "$SUCCES Photoshop is successfully installed."
}

if [[ $# -eq 0 ]]; then
	get_help
	exit 0
fi

while getopts "a:i:h" flag; do
	case $flag in
	a)
		LOCAL_ARCHIVE="$OPTARG"
		;;
	h)
		get_help
		;;
	i)
		INSTALL_PATH="$OPTARG"
		;;
	\?)
		echo "Invalid option: -$OPTARG Use -h for help."
		exit 1
		;;
	esac
done

main