Initial commit

recorder/recorder.sh

@@ -0,0 +1,69 @@
+#!/bin/env bash
+
+# depends: wf-recorder, libnotify
+
+TEMPDIR="/tmp/niri-recorder"
+mkdir -p "$TEMPDIR"
+LOCK="$TEMPDIR/lock"
+RFMT=".mkv"
+OFMT=".mp4"
+RAW="$TEMPDIR/raw$RFMT"
+OUTDIR="$HOME/Videos/niri-recorder"
+mkdir -p "$OUTDIR"
+
+# sends current recording state to socket
+function sig {
+  echo "$1" | socat - UNIX-CONNECT:/tmp/recorder-sock.sock
+}
+
+# function generates unique name for recording
+function compname {
+  now=$(date +"%y-%m-%d-%H:%M:%S")
+  echo "$OUTDIR/$now$OFMT"
+}
+
+# First we need to check if the recording
+# lockfile exists.
+if [[ -f "$LOCK" ]]; then
+  # Stop the recording
+  sig "STP"
+  kill "$(cat "$LOCK")"
+  # remove lockfile
+  rm "$LOCK"
+  outpath="$(compname)"
+  sig "CMP"
+  # compress the recording
+  ffmpeg -y -i "$RAW" -c:v libx264 -preset slow -crf 21 -r 30 -b:v 2M -maxrate 3M -bufsize 4M -c:a aac -b:a 96k -movflags +faststart "$outpath" || (sig "ERR"; exit 1)
+  # copy URI path to clipboard
+  wl-copy -t 'text/uri-list' <<<"file://$outpath" || (sig "ERR"; exit 1)
+  sig "CPD"
+  # delete the raw recording
+  rm "$RAW"
+
+else
+  # count how many monitors are attached
+  num_mon=$(niri msg --json outputs | jq 'keys | length')
+  wf_flags="-Dyf"
+  if [ "$1" = "region" ];then
+    # select a screen region
+    sel=$(slurp) || exit 1
+    wf-recorder -g "$sel" "$wf_flags" "$RAW" &
+  elif [ "$1" = "screen" ] || [ "$1" = "" ] && (( num_mon > 1 )); then
+    # select entire screen
+    sel=$(slurp -o) || exit 1
+    wf-recorder -g "$sel" "$wf_flags" "$RAW" &
+  else
+    # this runs when screen is specified and there's only one monitor
+    # it also runs with no args bc screen is default
+    wf-recorder "$wf_flags" "$RAW" &
+  fi
+
+  sig "REC"
+  # create lockfile
+  touch "$LOCK"
+
+  # save recorder's process id to lockfile
+  PID=$!
+  echo "$PID" > "$LOCK"
+fi
+