From e7aa2bb510152e64f93be135301372f6c9b7b696 Mon Sep 17 00:00:00 2001
From: ItsDrike <itsdrike@protonmail.com>
Date: Fri, 3 Feb 2023 04:47:38 +0100
Subject: [PATCH] Add quick-record script for easy screen capture recordings

---
 home/.config/hypr/hyprland.conf          |  2 +
 home/.local/bin/scripts/gui/quick-record | 90 ++++++++++++++++++++++++
 2 files changed, 92 insertions(+)
 create mode 100755 home/.local/bin/scripts/gui/quick-record

diff --git a/home/.config/hypr/hyprland.conf b/home/.config/hypr/hyprland.conf
index 12f1482..585afcd 100644
--- a/home/.config/hypr/hyprland.conf
+++ b/home/.config/hypr/hyprland.conf
@@ -195,6 +195,8 @@ binde = , XF86MonBrightnessUp, exec, brightness -i 5% -n
 binde = , XF86MonBrightnessDown, exec, brightness -d 5% -n
 
 # Screenshots (with custom screenshot script)
+bind = , 248, exec, quick-record --notify toggle
+bind = SUPER, 248, exec, quick-record toggle
 bind = ALT, Print, exec, wl-copy --type image/png "$(hyprpicker)" && notify-send 'Picked color' "$(wl-paste) (saved to clipboard)"
 bind = , Print, exec, screenshot --notify --copy --target area
 bind = SUPER, Print, exec, screenshot --notify --copy --target area --edit
diff --git a/home/.local/bin/scripts/gui/quick-record b/home/.local/bin/scripts/gui/quick-record
new file mode 100755
index 0000000..0c5b7e2
--- /dev/null
+++ b/home/.local/bin/scripts/gui/quick-record
@@ -0,0 +1,90 @@
+#!/bin/bash
+
+EXTENSION="mp4"
+NOTIFY=0
+
+save_file() {
+    wl-copy -t text/uri-list "file://${file}"
+    [ "$NOTIFY" -eq 1 ] && notify-send -a "quick-record" "Recording saved" "$file <file://${file}>"
+    echo "Recording saved: $file"
+}
+
+stop_recording() {
+    if pidof -s wf-recorder >/dev/null 2>&1; then
+        [ "$NOTIFY" -eq 1 ] && notify-send -a "quick-record" "Recording stopped"
+        killall -s SIGINT wf-recorder
+    else
+        >&2 echo "No active recording to stop"
+        return 1
+    fi
+}
+
+start_recording() {
+    # Remove all previous recordings
+    # No need to clutter /tmp, since this is used for copying the recording
+    # as a file, it's unlikely that we'll need any of the old recordings
+    # when a new one is requested
+    rm ${TMPDIR:-/tmp}/wf-recorder-video-*.$EXTENSION
+
+    file="$(mktemp -t wf-recorder-video-XXXXX.$EXTENSION)"
+    geom="$(slurp)"
+
+    [ "$NOTIFY" -eq 1 ] && notify-send "quick-record" "Recording starting"
+
+    trap save_file SIGINT
+    trap save_file SIGTERM
+    trap save_file SIGHUP
+
+    # Wee need 'y' stdin to confirm that we want to override the file
+    # since mktemp creates a blank file there already
+    echo "y" | wf-recorder -g "$geom" -f "$file"
+
+    # If wf-recorder process ends directly, rather than a trap being hit
+    # we also want to run the save_file func
+    save_file
+}
+
+# Parse any CLI flags first, before getting to positional args
+# (As long as we have $2, meaning there's at least 2 args, treat
+
+# $1 arg as CLI flag``)
+while [ "$2" ]; do
+    case "$1" in
+        -h | --help)
+            cat << EOF
+quick-record is a simple tool for performing quick screen capture recordings
+on wayland WMs easily, using wf-recorder.
+
+Optional flags:
+-n | --notify: Produce notifications on recording start/end
+-e | --extension [extension]: Use a different file extension (default: mp4)
+
+Required positional arguments:
+{start|stop|toggle}: Action which to perform.
+    - start: Start a new recording
+    - stop: Stop any already running recording(s)
+    - toggle: If there is a running recording, stop it, otherwise start a new one
+EOF
+            exit 0
+            ;;
+        -n | --notify) NOTIFY=1; shift ;;
+        -e | --extension) EXTENSION="$2"; shift; shift ;;
+    esac
+done
+
+
+if [ "$1" = "start" ]; then
+    start_recording
+elif [ "$1" = "stop" ]; then
+    stop_recording
+elif [ "$1" = "toggle" ]; then
+    if stop_recording 2>/dev/null; then
+        exit 0
+    else
+        start_recording
+    fi
+else
+    >&2 echo "Error: No argument provided!"
+    >&2 echo "Expected one of: start, stop, toggle"
+    exit 1
+fi