mirror of https://github.com/tycrek/ass
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
137 lines
3.8 KiB
137 lines
3.8 KiB
#!/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
|