misc-scripts/screenshot

#!/bin/sh
set -euf

OUTFILE="${HOME}/pics/screenshots/$(date +"%Y-%m-%d_%H-%M-%S").jxl" # output file template, change as needed
# menu + args, split on spaces
MENU="fzfmenu -p screenshot"

clean() {
	[ -n "${freeze_pid:-}" ] && kill "$freeze_pid"
	[ -f "${tmpfreeze:-}" ] && rm -f "$tmpfreeze"
	[ -f "${tmpfp:-}" ] && rm -f "$tmpfp"
}

save() {
	cjxl "$tmpfp" -d 0.0 "$OUTFILE"
	notify-send -u low "$OUTFILE"
	exit 0
}

choose() {
	while true; do
		case "$(printf "save\nedit\nscan" | $MENU)" in
			save) save ;;
			edit)
				h1="$(sha256sum "$tmpfp")"
				$editor "$tmpfp"
				h2="$(sha256sum "$tmpfp")"
				[ "$h1" != "$h2" ] && save # change, save
				exit 0
				;;
			scan)
				zdata="$(zbarimg -1 "$tmpfp")" || zdata="error: scan failed with exit code $?"
				export zdata
				$TERMINAL -e sh -c 'printf "%s\n" "${zdata#*:}"; read -r __' &
				printf '%s\n' "${zdata#*:}"
				;;
			*)	break ;;
		esac
	done
}

freeze() {
	tmpfreeze="$(mktemp -t screenshot_freeze_XXXX.png)"
	$freeze_cmd "$tmpfreeze"
	$freeze_img "$tmpfreeze" &
	freeze_pid=$!
	sleep 0.1
	# x11: window manager makes it fullscreen
	# doesn't work too well on cwm but idc
}

case "${XDG_SESSION_TYPE:-}" in
	wayland)
		editor="swappy -f"
		freeze_img="swayimg -g 0,0,1920,1080"
		freeze_cmd="grim -l 0"
		ss() {
			if [ -n "$REGION" ]; then
				reg="$(slurp -f '-g "%x,%y %wx%h"')" || return 1
			fi
			grim ${REGION:+-g "$reg"} "$tmpfp"
			wl-copy < "$tmpfp"
		}
		;;
	*)
		editor="gimp -n"
		freeze_img="nsxiv -b -N SCREENSHOT_FREEZE"
		freeze_cmd="shotgun"
		ss() {
			if [ -n "$REGION" ]; then
				reg="$(hacksaw -f '-i %i -g %g')" || return 1
			fi
			shotgun ${SCREEN:+-s} ${REGION:+$reg} \
				${WINDOW:+-i $(xdotool getactivewindow)} "$tmpfp"
			xclip -sel clipboard -t image/png < "$tmpfp"
		}
		;;
esac

tmpfp="$(mktemp -t screenshot_XXXX.png)"
trap 'clean' EXIT

SCREEN="" REGION="" WINDOW="" FREEZE="" ssopts=""
while getopts :srwf OPT; do
	case "$OPT" in
		s) SCREEN=1 ;; # screen
		r) REGION=1 ;; # region
		w) WINDOW=1 ;; # active window (don't use with region)
		f) freeze ;; # freeze screen, only useful with -r
		*) echo "invalid option: -$OPTARG" && exit 1 ;;
	esac
done

ss
ec=$?

[ -n "${freeze_pid:-}" ] && kill "$freeze_pid" && freeze_pid="" # unfreeze
[ "$ec" -eq 0 ] && choose