screenshot-service/Docker/utils.sh
2022-06-17 12:09:32 +02:00

203 lines
4.1 KiB
Bash

#!/bin/bash
timestamp(){
echo $((${EPOCHREALTIME//.} / 1000))
}
_get_pixel_color(){
# old version that takes up to .5s to run
START=$(timestamp)
x=$1;
y=$2;
output_path="/tmp/$(mktemp -u XXXXXX).png"
scrot $output_path
magick $output_path -format "%[hex:p{$x,$y}]" info:
END=$(timestamp)
DIFF=$(echo "$END - $START" | bc)
echo "{\"getting_pixel_color_took\": \"${DIFF}ms\"}" > /dev/stderr
}
get_pixel_color(){
x=$1;
y=$2;
FIREFOX_ID=$(wmctrl -lp | grep Firefox | awk '{print $1}')
if [ "$FIREFOX_ID" = "" ]
then
echo "NONE"
return
fi
grabc -w "$FIREFOX_ID" -l +${x}+${y}
}
extract_text(){
output_path="/tmp/$(mktemp -u XXXXXX).png"
cropped_path=$output_path--cropped.png
scrot $output_path
left=$1
top=$2
width=$3
height=$4
# https://www.libvips.org/API/current/libvips-conversion.html#vips-extract-area
vips extract_area "$output_path" "$cropped_path" $left $top $width $height
tesseract "$cropped_path" stdout -l eng
}
click (){
xdotool mousemove $1 $2
xdotool click 1
}
grab_no=0
grab(){
NOTE="$grab_no $@"
#echo $@
((grab_no++))
# scrot --note "-f '/usr/share/fonts/noto/NotoSansMono-Medium.ttf/40' -c 255,0,0,255 -t '$NOTE'" "/opt/static/$(date -Iminutes).png"
#echo "not grabbing"
}
grab_screen_to_public(){
id=$1
filepath="/opt/static/$id/preview.png"
tempfile="/opt/static/$id/temp.png"
mkdir -p "/opt/static/$id"
rm -f "$tempfile"
scrot "$tempfile"
mv -f "$tempfile" "$filepath"
}
keycombo(){
for key in "$@"
do
xdotool keydown $key
sleep 0.01
done
for key in "$@"
do
xdotool keyup $key
sleep 0.01
done
sleep 0.05
}
start_firefox(){
rm -rf /root/.mozilla/firefox/bifup8k5.docker/sessionstore-backups
#echo 'user_pref("layout.css.devPixelsPerPx", "1.5");' >> /root/.mozilla/firefox/bifup8k5.docker/prefs.js
firefox --devtools > /dev/null &
FIREFOX_PID=$!
while [ $(get_pixel_color 100 100) = "NONE" ]
do
sleep 0.1
done;
}
prepare_firefox(){
# xdotool key Return # in case there's a "firefox closed unexpectedly" button
# sleep 1
# grab "after sleep 1"
click 1886 13 #maximize button
grab "after maximize"
FIREFOX_WIDTH=$(wmctrl -lpG | grep Firefox | awk '{print $6}')
while [ "$FIREFOX_WIDTH" != "2880" ]
do
sleep 0.1
done
#grab "after sleep 1"
# no need to reopen the network tab, as it's almost certainly already open
# keycombo Control_L Shift_L e
# grab "after ctrl+shift+e"
# keycombo Shift_L F5 #profiler
# grab "after shift-c F5"
# keycombo Control_L Shift_L e
# sleep 5 # it needs some time
}
wait_for_pixel_color(){
x=$1;
y=$2;
color=$3;
timeout=$4;
start=$(timestamp)
while [ $(get_pixel_color 143 122) != "$color" ]
do
sleep 0.1
ts=$(timestamp)
if [ $(( ts - start )) -gt $(( timeout * 1000 )) ]
then
return
fi
done
}
load_website(){
URL=$1
keycombo Control_L l
grab ctrl+l
xdotool type "$URL"
grab type url
xdotool key Return
grab enter
sleep 1
echo "{\"current_action\": \"waiting for $URL to load...\"}"
times=0
while [ $(get_pixel_color 143 122) = "#2e3436" ] # the center of the X icon that becomes a "refresh" icon once the website is finished loading
do
sleep 0.5
#printf "."
#scrot
times=$((times + 1))
if [ $times -eq 30 ]
then
echo "{\"current_action\": \"website load timeout, proceeding anyway...\"}"
break;
fi
done
echo "{\"current_action\": \"website loaded\"}"
}
open_network_inspector(){
keycombo Shift_L F5 #profiler
wait_for_pixel_color 1777 152 0a84ff 3
keycombo Control_L Shift_L e
}
network_inspector_search(){
query=$1
keycombo Control_L f
sleep 0.2
keycombo Control_L a
sleep 0.1
xdotool type "$query"
#xdotool key Escape
sleep 0.3
seq 28 | xargs -I {} xdotool key Tab
xdotool key Down
sleep 0.1
xdotool key Up
sleep 0.1
}
network_inspector_has_more_entries(){
grab network_inspector_has_more_entries
[ $(get_pixel_color 1267 1572) = "#f9f9fa" ]
}
network_inspector_next_entry(){
xdotool key Tab
sleep 0.05
xdotool key Down
sleep 0.1
}
# last_message="_____"
# measure(){
# }