blob: b20957478a1681a49ffd7a5ba84affc080c023ff (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
#!/bin/sh
# === Config: directory for QEMU sockets ===
QEMU_SOCK_DIR="/home/$(whoami)/virt/machines"
# === Send 'print' key to QEMU monitor ===
send_print_to_qemu() {
monitor_socket=$(find "$QEMU_SOCK_DIR" -maxdepth 1 -type s -name '*-monitor.socket' | head -n1)
if [ "$monitor_socket" = "" ]; then
echo "No QEMU monitor socket found."
return 1
fi
echo "Using QEMU monitor socket: $monitor_socket"
if ! printf "sendkey print\n" | socat - UNIX-CONNECT:"$monitor_socket"; then
echo "Failed to send key via socat."
return 1
fi
echo "Sent 'sendkey print' to QEMU successfully."
return 0
}
# === Detect if current window is QEMU (X11 only) ===
if [ "$DISPLAY" != "" ] && command -v xprop >/dev/null 2>&1; then
win_id=$(xprop -root _NET_ACTIVE_WINDOW | awk -F' ' '{print $5}')
class=$(xprop -id "$win_id" WM_CLASS 2>/dev/null | awk -F'"' '{print $4}')
if [ "$class" = "qemu-system-x86_64" ]; then
send_print_to_qemu && exit 0
# if sending fails, optionally fallback or exit anyway
exit 1
fi
fi
#if command -v hyprctl >/dev/null 2>&1 && command -v jq >/dev/null 2>&1; then
# focused_class="$(hyprctl activewindow -j | jq -r '.class' 2>/dev/null)"
# if [ "$focused_class" = "qemu-system-x86_64" ]; then
# send_print_to_qemu && exit 0
# fi
#fi
DIR="$HOME/pictures/screenshots"
OUTPUT_DIR="$HOME/documents/ocr_output"
# Create the directories if they don't exist
[ ! -d "$DIR" ] && mkdir -pv "$DIR"
[ ! -d "$OUTPUT_DIR" ] && mkdir -pv "$OUTPUT_DIR"
file="$DIR/screenshot_$(date '+%Y-%m-%d_%H-%M-%S').png"
text_file="$OUTPUT_DIR/ocr_output_$(date '+%Y-%m-%d_%H-%M-%S')"
copy_to_clipboard() {
if [ "$WAYLAND_DISPLAY" != "" ]; then
wl-copy <"$1"
elif [ "$DISPLAY" != "" ]; then
xclip -selection clipboard -i <"$1"
else
echo "No display server detected. Cannot copy to clipboard."
return 1
fi
}
case "$1" in
screen) grim "$file" ;;
output) slurp -o -r | grim -g - "$file" ;;
area) slurp | grim -g - "$file" ;;
output-area) slurp -o | grim -g - "$file" ;;
ocr) slurp | grim -g - "$file" && tesseract "$file" "$text_file" ;;
ocr-clipboard)
slurp | grim -g - "$file" && tesseract "$file" "$text_file"
if [ -f "$text_file.txt" ]; then
copy_to_clipboard "$text_file.txt"
CLIP_STATUS=$?
if [ "$CLIP_STATUS" -eq 0 ]; then
command rm -f "$text_file.txt"
notify-send -t 10000 --app-name "Screenshot" "OCR to Clipboard" "Text copied to clipboard."
echo "OCR output copied to clipboard and file deleted."
else
notify-send -t 10000 --app-name "Screenshot" "Clipboard Copy Failed" "Failed to copy text to clipboard."
echo "Failed to copy text to clipboard."
fi
else
notify-send -t 10000 --app-name "Screenshot" "OCR Error" "OCR process failed."
exit 1
fi
;;
*)
echo "Invalid argument"
notify-send -t 10000 --app-name "Screenshot" "Screenshot" "Something went wrong."
exit 1
;;
esac
if [ "$1" = "ocr" ]; then
if [ -f "$text_file" ]; then
notify-send -t 10000 --app-name "Screenshot" "OCR Complete" "Text saved to $text_file"
echo "OCR output saved to: $text_file"
else
notify-send -t 10000 --app-name "Screenshot" "OCR Error" "OCR process failed."
exit 1
fi
else
notify-send -t 10000 --app-name "Screenshot" "Screenshot" "Saved as $file"
echo "$file"
fi
|
