This commit is contained in:
2026-02-01 23:09:36 -08:00
parent f301f16949
commit d32f79f406
2 changed files with 108 additions and 106 deletions

View File

@@ -1,6 +1,8 @@
-- mpv_websocket -- mpv_websocket
-- https://github.com/kuroahna/mpv_websocket -- https://github.com/kuroahna/mpv_websocket
local use_secondary_subs = false
local utils = require("mp.utils") local utils = require("mp.utils")
local platform = mp.get_property_native("platform") local platform = mp.get_property_native("platform")
@@ -8,76 +10,101 @@ local platform = mp.get_property_native("platform")
local config_file_path = mp.find_config_file("mpv.conf") local config_file_path = mp.find_config_file("mpv.conf")
local config_folder_path, config_file = utils.split_path(config_file_path) local config_folder_path, config_file = utils.split_path(config_file_path)
local mpv_websocket_path = local mpv_websocket_path =
utils.join_path(config_folder_path, platform == "windows" and "mpv_websocket.exe" or "mpv_websocket") utils.join_path(config_folder_path, platform == "windows" and "mpv_websocket.exe" or "mpv_websocket")
local initialised_websocket local initialised_websocket
local _, err = utils.file_info(config_file_path) local _, err = utils.file_info(config_file_path)
if err then if err then
error("failed to open mpv config file `" .. config_file_path .. "`") error("failed to open mpv config file `" .. config_file_path .. "`")
end end
local _, err = utils.file_info(mpv_websocket_path) local _, err = utils.file_info(mpv_websocket_path)
if err then if err then
error("failed to open mpv_websocket") error("failed to open mpv_websocket")
end end
local function find_mpv_socket(config_file_path) local function find_mpv_socket(config_file_path)
local file = io.open(config_file_path, "r") -- First, try to get from mpv properties (includes command-line options)
if file == nil then local mpv_socket = mp.get_property("input-ipc-server")
error("failed to read mpv config file `" .. config_file_path .. "`") if mpv_socket and mpv_socket ~= "" then
end return mpv_socket
end
local mpv_socket -- Fall back to reading config file
for line in file:lines() do local file = io.open(config_file_path, "r")
mpv_socket = line:match("^input%-ipc%-server%s*=%s*(%g+)%s*") if file == nil then
if mpv_socket then error("failed to read mpv config file `" .. config_file_path .. "`")
break end
end
end
file:close() for line in file:lines() do
mpv_socket = line:match("^input%-ipc%-server%s*=%s*(%g+)%s*")
if mpv_socket then
break
end
end
if not mpv_socket then file:close()
error("input-ipc-server option does not exist in `" .. config_file_path .. "`")
end
return mpv_socket if not mpv_socket then
error("input-ipc-server option does not exist in `" .. config_file_path .. "`")
end
return mpv_socket
end end
local mpv_socket = find_mpv_socket(config_file_path) local mpv_socket = find_mpv_socket(config_file_path)
if platform == "windows" then if platform == "windows" then
mpv_socket = "\\\\.\\pipe" .. mpv_socket:gsub("/", "\\") mpv_socket = "\\\\.\\pipe" .. mpv_socket:gsub("/", "\\")
end end
local function start_websocket() local function start_websocket()
initialised_websocket = mp.command_native_async({ local args = {
name = "subprocess", mpv_websocket_path,
playback_only = false, "-m",
capture_stdout = true, mpv_socket,
capture_stderr = true, "-w",
args = { "6677",
mpv_websocket_path, }
"-m",
mpv_socket, if use_secondary_subs then
"-w", table.insert(args, "-s")
"6677", end
},
}) initialised_websocket = mp.command_native_async({
name = "subprocess",
playback_only = false,
capture_stdout = true,
capture_stderr = true,
args = args,
})
end end
local function end_websocket() local function end_websocket()
mp.abort_async_command(initialised_websocket) mp.abort_async_command(initialised_websocket)
initialised_websocket = nil initialised_websocket = nil
end end
local function toggle_websocket() local function toggle_websocket()
local paused = mp.get_property_bool("pause") local paused = mp.get_property_bool("pause")
if initialised_websocket and paused then if initialised_websocket and paused then
end_websocket() end_websocket()
elseif not initialised_websocket and not paused then elseif not initialised_websocket and not paused then
start_websocket() start_websocket()
end end
end
local function toggle_subs_type()
if use_secondary_subs then
use_secondary_subs = false
else
use_secondary_subs = true
end
if initialised_websocket then
end_websocket()
start_websocket()
end
end end
mp.register_script_message("togglewebsocket", toggle_websocket) mp.register_script_message("togglewebsocket", toggle_websocket)
mp.register_script_message("togglesubstype", toggle_subs_type)
start_websocket() start_websocket()

View File

@@ -1,9 +1,9 @@
#!/usr/bin/env bash #!/usr/bin/env bash
THEME="${THEME:-$HOME/.config/rofi/launchers/type-2/style-2.rasi}" THEME="${THEME:-/opt/mpv-yomitan/catppuccin-macchiato.rasi}"
THUMBNAIL_PATH="/tmp/rmpv-thumbnail.jpg"
FONTCONFIG_FILE=$HOME/.config/mpv/mpv-fonts.conf FONTCONFIG_FILE=$HOME/.config/mpv/mpv-fonts.conf
COMMAND=mpv COMMAND=mpv
VIDEO_EXTENSIONS="mkv|mp4|avi|webm|mov|flv|wmv|m4v|ts|m2ts"
# Parse command-line options first # Parse command-line options first
while getopts ":it:" opt; do while getopts ":it:" opt; do
@@ -26,81 +26,56 @@ while getopts ":it:" opt; do
done done
shift $((OPTIND - 1)) shift $((OPTIND - 1))
generate_thumbnail() { find_videos() {
local video_file="$1" find "$PWD" -maxdepth 1 -type f -regextype posix-extended \
local temp_thumb="/tmp/rmpv-thumbnail-$$.jpg" -iregex ".*\.($VIDEO_EXTENSIONS)$" 2>/dev/null | sort -V
local thumbnail_file="${video_file%.*}.jpg" }
# Clean up previous thumbnail build_rofi_menu() {
rm -f "$THUMBNAIL_PATH" while IFS= read -r video; do
[ -z "$video" ] && continue
local display_name
display_name=$(basename "$video")
printf '%s\0icon\x1fthumbnail://%s\n' "$display_name" "$video"
done < <(find_videos)
}
# Validate input get_video_thumbnail() {
if [[ -z "$video_file" ]]; then local video="$1"
echo "Error: No video file specified" >&2 local thumb_dir="$HOME/.cache/thumbnails/large"
return 1 local video_uri="file://$(realpath "$video")"
local thumb_hash=$(echo -n "$video_uri" | md5sum | cut -d' ' -f1)
local thumb_path="$thumb_dir/$thumb_hash.png"
if [[ -f "$thumb_path" ]]; then
echo "$thumb_path"
return 0
fi fi
if [[ ! -f "$video_file" ]]; then local tmp_thumb="/tmp/rmpv-thumb-$$.jpg"
echo "Error: Video file '$video_file' not found" >&2 if command -v ffmpegthumbnailer &>/dev/null; then
return 1 ffmpegthumbnailer -i "$video" -o "$tmp_thumb" -s 512 -q 5 2>/dev/null && echo "$tmp_thumb"
fi elif command -v ffmpeg &>/dev/null; then
ffmpeg -i "$video" -ss 00:00:30 -vframes 1 -vf "scale=512:-1" "$tmp_thumb" 2>/dev/null && echo "$tmp_thumb"
# Check if it's a video file
if ! file "$video_file" | grep -qE "(video|Video)"; then
echo "Error: '$video_file' doesn't appear to be a video file" >&2
return 1
fi
# Generate thumbnail if it doesn't exist
if [[ ! -f "$thumbnail_file" ]]; then
echo "Generating thumbnail for $(basename "$video_file")..."
# Try generating thumbnail side-by-side
if ! ffmpeg -ss 00:00:01 -i "$video_file" \
-vf "scale=320:240:force_original_aspect_ratio=decrease,pad=320:240:(ow-iw)/2:(oh-ih)/2" \
-frames:v 1 \
-q:v 4 \
"$thumbnail_file" \
-loglevel error -y 2>/dev/null; then
# Fallback to temp file if side-by-side fails (e.g. read-only fs)
echo "Warning: Failed to write to $thumbnail_file, trying temp location" >&2
thumbnail_file="$temp_thumb"
if ! ffmpeg -ss 00:00:01 -i "$video_file" \
-vf "scale=320:240:force_original_aspect_ratio=decrease,pad=320:240:(ow-iw)/2:(oh-ih)/2" \
-frames:v 1 \
-q:v 4 \
"$thumbnail_file" \
-loglevel error -y 2>/dev/null; then
echo "Error: Failed to generate thumbnail" >&2
return 1
fi
fi
fi
# Copy to consistent location for notify-send
# We use a fixed path so notify-send always finds it
if cp "$thumbnail_file" "$THUMBNAIL_PATH" 2>/dev/null; then
echo "Thumbnail ready at: $THUMBNAIL_PATH"
ls -l "$THUMBNAIL_PATH"
file "$THUMBNAIL_PATH"
else
echo "Error: Failed to copy thumbnail to $THUMBNAIL_PATH" >&2
fi fi
} }
choice="$(find . -iname "*[.mkv|.mp4]" | sort -h | rofi -dmenu -i -theme "$THEME" -theme-str 'listview {columns: 1; lines: 15;} window {width: 88%;}' -p "Choose Video")" selection=$(build_rofi_menu | rofi -dmenu -i -show-icons -theme "$THEME" \
if [[ -z "$choice" ]]; then -theme-str 'listview {columns: 1; lines: 15;} window {width: 88%;}' -p "Choose Video ")
if [[ -z "$selection" ]]; then
echo "No video selected." echo "No video selected."
exit 1 exit 1
fi fi
generate_thumbnail "$choice" choice="./$selection"
if [[ ! -f "$THUMBNAIL_PATH" ]]; then
echo "Warning: Thumbnail not created, notification will have no icon" >&2 THUMBNAIL_PATH=$(get_video_thumbnail "$choice")
if [[ -n "$THUMBNAIL_PATH" && -f "$THUMBNAIL_PATH" ]]; then
notify-send -i "$THUMBNAIL_PATH" "Playing Video" "$(basename "$choice")"
else
notify-send "Playing Video" "$(basename "$choice")"
fi fi
notify-send -i "$THUMBNAIL_PATH" "Playing Video" "$(basename "$choice")"
$COMMAND "$choice" & $COMMAND "$choice" &
# vim: ft=sh # vim: ft=sh