refactor(sway): rework the whole screen capture part
This commit is contained in:
parent
2dca1fb254
commit
0fc804a42d
3 changed files with 184 additions and 70 deletions
160
modules/home-manager/desktop/sway/includes/files/screencapt.sh
Executable file
160
modules/home-manager/desktop/sway/includes/files/screencapt.sh
Executable file
|
@ -0,0 +1,160 @@
|
|||
#!/usr/bin/env bash
|
||||
set -eu -o pipefail
|
||||
|
||||
APP_NAME="ScreenCapt"
|
||||
|
||||
declare -A COMMAND
|
||||
COMMAND=(
|
||||
[screenshot]="grim"
|
||||
[video]="wl-screenrec"
|
||||
)
|
||||
declare -A ICONS
|
||||
ICONS=(
|
||||
["screenshot"]="accessories-screenshot"
|
||||
["video"]="record-desktop"
|
||||
)
|
||||
|
||||
declare -A OUTPUT_DIR
|
||||
OUTPUT_DIR=(
|
||||
["screenshot"]="${XDG_PICTURES_DIR:-"~/pictures"}/screenshots"
|
||||
["video"]="${XDG_VIDEOS_DIR:="~/videos"}/screenrecords"
|
||||
)
|
||||
|
||||
declare -A OUTPUT_FILE_EXT
|
||||
OUTPUT_FILE_EXT=(
|
||||
["screenshot"]="png"
|
||||
["video"]="mp4"
|
||||
)
|
||||
|
||||
TO_FILE=0
|
||||
RECORD_AUDIO=0
|
||||
ACTION="screenshot"
|
||||
REGION_TYPE="window"
|
||||
|
||||
notify() {
|
||||
local summary message command
|
||||
filename="${1:-""}"
|
||||
summary="New ${ACTION}!"
|
||||
if [[ $TO_FILE -eq 1 ]]; then
|
||||
message+="Available in <i>${filename##*/}</i>"
|
||||
else
|
||||
message+="Available in the clipboard"
|
||||
fi
|
||||
command=(notify-send "${summary}" --app-name="$APP_NAME")
|
||||
if [[ -n "$filename" && "$ACTION" == "screenshots" ]]; then
|
||||
command+=(--icon="${filename}")
|
||||
else
|
||||
command+=("--icon=${ICONS[$ACTION]}")
|
||||
fi
|
||||
command+=("$message")
|
||||
"${command[@]}"
|
||||
}
|
||||
|
||||
process_args() {
|
||||
while :; do
|
||||
case ''${1:-""} in
|
||||
screenshot)
|
||||
ACTION="$1"
|
||||
;;
|
||||
|
||||
video)
|
||||
# If a video record is in progress Stop it
|
||||
if pgrep "wl-screenrec"; then
|
||||
kill -s SIGINT $(pgrep "wl-screenrec")
|
||||
exit 0
|
||||
fi
|
||||
TO_FILE=1
|
||||
ACTION="$1"
|
||||
;;
|
||||
-r | --region)
|
||||
if [[ "${2:-""}" =~ ^(r|region|s|screen|w|window)$ ]]; then
|
||||
REGION_TYPE="$2"
|
||||
else
|
||||
exit 1
|
||||
fi
|
||||
;;
|
||||
-f | --file)
|
||||
TO_FILE=1
|
||||
;;
|
||||
-a | --with-audio)
|
||||
RECORD_AUDIO=1
|
||||
;;
|
||||
*)
|
||||
break
|
||||
;;
|
||||
esac
|
||||
shift
|
||||
done
|
||||
}
|
||||
|
||||
get_app_name() {
|
||||
local appname
|
||||
case "$REGION_TYPE" in
|
||||
r | region)
|
||||
appname="region"
|
||||
;;
|
||||
s | screen)
|
||||
appname="screen"
|
||||
;;
|
||||
w | window)
|
||||
appname=$(swaymsg -t get_tree | jq -r '.. | ((.nodes? // empty) + (.floating_nodes? // empty))[] | select(.visible and .pid and .focused) | "\(.app_id)"')
|
||||
;;
|
||||
esac
|
||||
printf "%s" "${appname:-}"
|
||||
}
|
||||
|
||||
get_region() {
|
||||
local region switch
|
||||
case "$REGION_TYPE" in
|
||||
r | region)
|
||||
region=$(slurp)
|
||||
switch="-g"
|
||||
;;
|
||||
s | screen)
|
||||
region=$(swaymsg -t get_outputs | jq -r '.[] | select(.focused) | "\(.name)"')
|
||||
switch="-o"
|
||||
;;
|
||||
w | window)
|
||||
region=$(swaymsg -t get_tree | jq -r '.. | ((.nodes? // empty) + (.floating_nodes? // empty))[] | select(.visible and .pid and .focused) | .rect | "\(.x),\(.y) \(.width)x\(.height)"')
|
||||
switch="-g"
|
||||
;;
|
||||
esac
|
||||
printf "%s%s" "${switch:-}" "${region:-}"
|
||||
}
|
||||
|
||||
get_output_filename() {
|
||||
local date
|
||||
printf -v date '%(%F_%H.%M.%S)T'
|
||||
printf "%s/%s-%s.%s" "${OUTPUT_DIR[$ACTION]}" "$date" "$(get_app_name)" "${OUTPUT_FILE_EXT[$ACTION]}"
|
||||
}
|
||||
|
||||
ensure_directories_exist() {
|
||||
for dir in "${!OUTPUT_DIR[@]}"; do
|
||||
mkdir -p "${OUTPUT_DIR[$dir]}"
|
||||
done
|
||||
}
|
||||
|
||||
main() {
|
||||
process_args "$@"
|
||||
ensure_directories_exist
|
||||
local command
|
||||
command=("${COMMAND[$ACTION]}" "$(get_region)")
|
||||
if [[ "$TO_FILE" -eq 1 ]]; then
|
||||
local filename
|
||||
filename=$(get_output_filename)
|
||||
if [[ "$ACTION" = "video" ]]; then
|
||||
if [[ "$RECORD_AUDIO" -eq 1 ]]; then
|
||||
command+=(--audio)
|
||||
fi
|
||||
command+=(-f)
|
||||
fi
|
||||
command+=("$filename")
|
||||
"${command[@]}"
|
||||
else
|
||||
"${command[@]}" - | wl-copy
|
||||
fi
|
||||
notify "${filename:-""}"
|
||||
exit 0
|
||||
}
|
||||
|
||||
main "$@"
|
Loading…
Add table
Add a link
Reference in a new issue