Sway : Add fullscreenrec & autorotate.

home/config/sway/module/Screenshot.nix

@@ -1,4 +1,4 @@
-{ config, ... }: let
+{ config, pkgs, lib, ... }: let
 	codec     = "libsvtav1";
 	color     = config.style.color;
 	container = "mp4";
@@ -8,14 +8,100 @@
 	selection = "slurp -d -b ${color.bg.light}${opacity} -c ${color.fg.light} -w 0 -s 00000000";
 	pixfmt    = "yuv420p10le";
 in {
-	text = ''
+	text = let
+		fullscr = pkgs.writeShellScriptBin "FullscreenScreenshot" ''
+			scrFile="''${XDG_PICTURES_DIR[0]}/$(date +${format}).png"
+
+			grim \
+				-o $(swaymsg -t get_outputs | jq -r ".[] | select(.focused) | .name") - \
+			| tee "''${scrFile}" \
+			| wl-copy -t image/png
+		'';
+
+		selectscr = pkgs.writeShellScriptBin "SelectScreenshot" ''
+			scrSelection=$(${selection})
+			[[ -n "''${scrSelection}" ]] || exit
+
+			scrFile="''${XDG_PICTURES_DIR[0]}/$(date +${format}).png"
+
+			grim \
+				-g "''${scrSelection}" - \
+			| tee "''${scrFile}" \
+			| wl-copy -t image/png
+		'';
+
+		selectrec = pkgs.writeShellScriptBin "SelectRecording" ''
+			pkill -SIGINT wf-recorder \
+			|| {
+				scrSelection=$(${selection})
+				[[ -n "''${scrSelection}" ]] || exit
+
+				scrRecFile="''${XDG_VIDEOS_DIR[0]}/$(date +${format}).${container}"
+
+				scrTransform="$(swaymsg -t get_outputs | jq -r '.[] | select(.focused) | .transform')"
+				[[ "''${scrTransform}" = "normal" ]] && scrTransform=""
+
+				wf-recorder \
+					--geometry "''${scrSelection}" \
+					--codec ${codec} \
+					--file "''${scrRecFile}" \
+					--framerate ${toString framerate} \
+					--pixel-format ${pixfmt} \
+				&& ffmpeg \
+					-f lavfi \
+					-i anullsrc=channel_layout=stereo:sample_rate=44100 \
+					-i "''${scrRecFile}" \
+					-c:v copy \
+					-c:a libopus \
+					-shortest \
+					-f ${container} "''${scrRecFile}_" \
+				&& mv "''${scrRecFile}_" "''${scrRecFile}" \
+				&& [[ -n "''${scrTransform}" ]] \
+				&& ffmpeg -display_rotation ''${scrTransform} -i ''${scrRecFile} -c copy -f ${container} ''${scrRecFile}_ \
+				&& mv ''${scrRecFile}_ ''${scrRecFile} \
+				|| rm ''${scrRecFile}_
+			};
+		'';
+
+		fullrec = pkgs.writeShellScriptBin "FullscreenRecording" ''
+			pkill -SIGINT wf-recorder \
+			|| {
+				scrRecFile="''${XDG_VIDEOS_DIR[0]}/$(date +${format}).${container}"
+				scrTransform="$(swaymsg -t get_outputs | jq -r '.[] | select(.focused) | .transform')"
+				[[ "''${scrTransform}" = "normal" ]] && scrTransform=""
+
+				wf-recorder \
+					-o $(swaymsg -t get_outputs | jq -r ".[] | select(.focused) | .name") - \
+					--codec ${codec} \
+					--file "''${scrRecFile}" \
+					--framerate ${toString framerate} \
+					--pixel-format ${pixfmt} \
+				&& ffmpeg \
+					-f lavfi \
+					-i anullsrc=channel_layout=stereo:sample_rate=44100 \
+					-i "''${scrRecFile}" \
+					-c:v copy \
+					-c:a libopus \
+					-shortest \
+					-f ${container} "''${scrRecFile}_" \
+				&& mv "''${scrRecFile}_" "''${scrRecFile}" \
+				&& [[ -n "''${scrTransform}" ]] \
+				&& ffmpeg -display_rotation ''${scrTransform} -i ''${scrRecFile} -c copy -f ${container} ''${scrRecFile}_ \
+				&& mv ''${scrRecFile}_ ''${scrRecFile} \
+				|| rm ''${scrRecFile}_
+			};
+		'';
+	in ''
 		# Fullscreen screenshot.
-		bindsym --to-code $mod+y exec 'export scrFile="''${XDG_PICTURES_DIR[0]}/$(date +${format}).png"; grim -o $(swaymsg -t get_outputs | jq -r ".[] | select(.focused) | .name") - | tee "''${scrFile}" | wl-copy -t image/png'
+		bindsym --to-code $mod+y exec ${lib.getExe fullscr}
+
+		# Fullscreen recording.
+		bindsym --to-code $mod+shift+y exec ${lib.getExe fullrec}
 
 		# Select screenshot.
-		bindsym --to-code $mod+v exec 'export scrFile="''${XDG_PICTURES_DIR[0]}/$(date +${format}).png"; { grim -g "$(${selection})" - || rm "''${scrFile}"; } | tee "''${scrFile}" | wl-copy -t image/png'
+		bindsym --to-code $mod+v exec ${lib.getExe selectscr}
 
 		# Select recording.
-		bindsym --to-code $mod+shift+v exec 'pkill -SIGINT wf-recorder || { export scrRecFile="''${XDG_VIDEOS_DIR[0]}/$(date +${format}).${container}"; wf-recorder --geometry "$(${selection})" --codec ${codec} --file "''${scrRecFile}" --framerate ${toString framerate} --pixel-format ${pixfmt} && ffmpeg -f lavfi -i anullsrc=channel_layout=stereo:sample_rate=44100 -i "''${scrRecFile}" -c:v copy -c:a libopus -shortest -f ${container} "''${scrRecFile}_" && mv "''${scrRecFile}_" "''${scrRecFile}"; }'
+		bindsym --to-code $mod+shift+v exec ${lib.getExe selectrec}
 	'';
 }