#!/usr/bin/env bash

# Script Configuration
# Load configuration file if available
# this is useful if you want to source keys from a secret file
CONFIG_FILE="config.sh"
if [ -f "$CONFIG_FILE" ]; then
	# shellcheck disable=1090
	source "${CONFIG_FILE}"
fi

LOG_DIR=$(pwd)
if [ ! -d "$LOG_DIR" ]; then
	echo "The directory you have specified to save the logs does not exist."
	echo "Please create the directory with the following command:"
	echo "mkdir -p $LOG_DIR"
	echo -en "Or specify a different LOG_DIR\n"
	exit 1
fi

IMAGE_PATH="$HOME/Pictures"
if [ ! -d "$IMAGE_PATH" ]; then
	echo "The directory you have specified to save the screenshot does not exist."
	echo "Please create the directory with the following command:"
	echo "mkdir -p $IMAGE_PATH"
	echo -en "Or specify a different IMAGE_PATH\n"
	exit 1
fi

IMAGE_NAME="ass"
FILE="${IMAGE_PATH}/${IMAGE_NAME}.png"

# Function to check if a tool is installed
check_tool() {
	command -v "$1" >/dev/null 2>&1
}

# Function to take Flameshot screenshots
takeFlameshot() {
	# check if flameshot tool is installed
	REQUIRED_TOOLS=("flameshot")

	for tool in "${REQUIRED_TOOLS[@]}"; do
		if ! check_tool "$tool"; then
			echo "Error: $tool is not installed. Please install it before using this script."
			exit 1
		fi
	done

	flameshot config -f "${IMAGE_NAME}"
	flameshot gui -r -p "${IMAGE_PATH}" >/dev/null
}

# Function to take Wayland screenshots using grim + slurp
takeGrimshot() {
	# check if grim and slurp are installed
	REQUIRED_TOOLS=("grim" "slurp")
	for tool in "${REQUIRED_TOOLS[@]}"; do
		if ! check_tool "$tool"; then
			echo "Error: $tool is not installed. Please install it before using this script."
			exit 1
		fi
	done

	grim -g "$(slurp)" "${FILE}" >/dev/null
}

# Function to remove the taken screenshot
removeTargetFile() {
	echo -en "Process complete.\nRemoving image.\n"
	rm -v "${FILE}"
}

# Function to upload target image to your ass instance
uploadScreenshot() {
	echo -en "KEY & DOMAIN are set. Attempting to upload to your ass instance.\n"
	URL=$(curl -X POST \
		-H "Content-Type: multipart/form-data" \
		-H "Accept: application/json" \
		-H "User-Agent: ShareX/13.4.0" \
		-H "Authorization: $KEY" \
		-F "file=@${FILE}" "https://$DOMAIN/" | grep -Po '(?<="resource":")[^"]+')
	if [[ "${XDG_SESSION_TYPE}" == x11 ]]; then
		printf "%s" "$URL" | xclip -sel clip
	elif [[ "${XDG_SESSION_TYPE}" == wayland ]]; then
		printf "%s" "$URL" | wl-copy
	else
		echo -en "Invalid desktop session!\nExiting.\n"
		exit 1
	fi
}

localScreenshot() {
	echo -en "KEY & DOMAIN variables are not set. Attempting local screenshot.\n"
	if [[ "${XDG_SESSION_TYPE}" == x11 ]]; then
		xclip -sel clip -target image/png <"${FILE}"
	elif [[ "${XDG_SESSION_TYPE}" == wayland ]]; then
		wl-copy <"${FILE}"
	else
		echo -en "Unknown display backend. Assuming Xorg and using xclip.\n"
		xclip -sel clip -target image/png <"${FILE}"
	fi
}

# Check if the screenshot tool based on display backend
if [[ "${XDG_SESSION_TYPE}" == x11 ]]; then
	echo -en "Display backend detected as Xorg (x11), using Flameshot\n"
	takeFlameshot
elif [[ "${XDG_SESSION_TYPE}" == wayland ]]; then
	echo -en "Display backend detected as Wayland, using grim & slurp\n"
	takeGrimshot
else
	echo -en "Unknown display backend. Assuming Xorg and using Flameshot\n"
	takeFlameshot >"${LOG_DIR}/flameshot.log"
	echo -en "Done. Make sure you check for any errors and report them.\nLogfile located in '${LOG_DIR}'\n"
fi

# Check if the screenshot file exists before proceeding
if [[ -f "${FILE}" ]]; then
	if [[ -n "$KEY" && -n "$DOMAIN" ]]; then
		# Upload the file to the ass instance
		uploadImage

		# Remove image
		removeTargetFile
	else
		# Take a screenshot locally
		localScreenshot

		# Remove image
		removeTargetFile
	fi
else
	echo -en "Target file ${FILE} was not found. Aborting screenshot.\n"
	exit 1
fi