336 lines
12 KiB
Python
336 lines
12 KiB
Python
|
from typing import Any, Dict, Sequence, List, Optional
|
||
|
|
import sys
|
||
|
|
import json
|
||
|
|
import platform
|
||
|
|
import socket
|
||
|
|
import re
|
||
|
|
import subprocess
|
||
|
|
from ._shared import Cmdlet, CmdletArg, parse_cmdlet_args
|
||
|
|
from helper.logger import log
|
||
|
|
from result_table import ResultTable
|
||
|
|
from .get_file import _get_fixed_ipc_pipe
|
||
|
|
import pipeline as ctx
|
||
|
|
|
||
|
|
def _send_ipc_command(command: Dict[str, Any]) -> Optional[Any]:
|
||
|
|
"""Send a command to the MPV IPC pipe and return the response."""
|
||
|
|
ipc_pipe = _get_fixed_ipc_pipe()
|
||
|
|
request = json.dumps(command) + "\n"
|
||
|
|
|
||
|
|
try:
|
||
|
|
if platform.system() == 'Windows':
|
||
|
|
# Windows named pipe
|
||
|
|
# Opening in r+b mode to read response
|
||
|
|
try:
|
||
|
|
with open(ipc_pipe, 'r+b', buffering=0) as pipe:
|
||
|
|
pipe.write(request.encode('utf-8'))
|
||
|
|
pipe.flush()
|
||
|
|
|
||
|
|
# Read response
|
||
|
|
# We'll try to read a line. This might block if MPV is unresponsive.
|
||
|
|
response_line = pipe.readline()
|
||
|
|
if response_line:
|
||
|
|
return json.loads(response_line.decode('utf-8'))
|
||
|
|
except FileNotFoundError:
|
||
|
|
return None # MPV not running
|
||
|
|
except Exception as e:
|
||
|
|
log(f"Windows IPC Error: {e}", file=sys.stderr)
|
||
|
|
return None
|
||
|
|
else:
|
||
|
|
# Unix socket
|
||
|
|
af_unix = getattr(socket, 'AF_UNIX', None)
|
||
|
|
if af_unix is None:
|
||
|
|
log("Unix sockets not supported on this platform", file=sys.stderr)
|
||
|
|
return None
|
||
|
|
|
||
|
|
try:
|
||
|
|
sock = socket.socket(af_unix, socket.SOCK_STREAM)
|
||
|
|
sock.settimeout(2.0)
|
||
|
|
sock.connect(ipc_pipe)
|
||
|
|
sock.sendall(request.encode('utf-8'))
|
||
|
|
|
||
|
|
# Read response
|
||
|
|
response_data = b""
|
||
|
|
while True:
|
||
|
|
try:
|
||
|
|
chunk = sock.recv(4096)
|
||
|
|
if not chunk:
|
||
|
|
break
|
||
|
|
response_data += chunk
|
||
|
|
if b"\n" in chunk:
|
||
|
|
break
|
||
|
|
except socket.timeout:
|
||
|
|
break
|
||
|
|
|
||
|
|
sock.close()
|
||
|
|
|
||
|
|
if response_data:
|
||
|
|
# Parse lines, look for response to our request
|
||
|
|
lines = response_data.decode('utf-8').strip().split('\n')
|
||
|
|
for line in lines:
|
||
|
|
try:
|
||
|
|
resp = json.loads(line)
|
||
|
|
# If it has 'error' field, it's a response
|
||
|
|
if 'error' in resp:
|
||
|
|
return resp
|
||
|
|
except:
|
||
|
|
pass
|
||
|
|
except (FileNotFoundError, ConnectionRefusedError):
|
||
|
|
return None # MPV not running
|
||
|
|
except Exception as e:
|
||
|
|
log(f"Unix IPC Error: {e}", file=sys.stderr)
|
||
|
|
return None
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
log(f"IPC Error: {e}", file=sys.stderr)
|
||
|
|
return None
|
||
|
|
|
||
|
|
return None
|
||
|
|
|
||
|
|
def _get_playlist() -> List[Dict[str, Any]]:
|
||
|
|
"""Get the current playlist from MPV."""
|
||
|
|
cmd = {"command": ["get_property", "playlist"], "request_id": 100}
|
||
|
|
resp = _send_ipc_command(cmd)
|
||
|
|
if resp and resp.get("error") == "success":
|
||
|
|
return resp.get("data", [])
|
||
|
|
return []
|
||
|
|
|
||
|
|
def _run(result: Any, args: Sequence[str], config: Dict[str, Any]) -> int:
|
||
|
|
"""Manage and play items in the MPV playlist via IPC."""
|
||
|
|
|
||
|
|
parsed = parse_cmdlet_args(args, CMDLET)
|
||
|
|
|
||
|
|
# Handle positional index argument if provided
|
||
|
|
index_arg = parsed.get("index")
|
||
|
|
|
||
|
|
clear_mode = parsed.get("clear")
|
||
|
|
list_mode = parsed.get("list")
|
||
|
|
|
||
|
|
# Handle piped input (add to playlist)
|
||
|
|
if result:
|
||
|
|
# If result is a list of items, add them to playlist
|
||
|
|
items_to_add = []
|
||
|
|
if isinstance(result, list):
|
||
|
|
items_to_add = result
|
||
|
|
elif isinstance(result, dict):
|
||
|
|
items_to_add = [result]
|
||
|
|
|
||
|
|
added_count = 0
|
||
|
|
for i, item in enumerate(items_to_add):
|
||
|
|
# Extract URL/Path
|
||
|
|
target = None
|
||
|
|
title = None
|
||
|
|
|
||
|
|
if isinstance(item, dict):
|
||
|
|
target = item.get("target") or item.get("url") or item.get("path")
|
||
|
|
title = item.get("title") or item.get("name")
|
||
|
|
elif hasattr(item, "target"):
|
||
|
|
target = item.target
|
||
|
|
title = getattr(item, "title", None)
|
||
|
|
elif isinstance(item, str):
|
||
|
|
target = item
|
||
|
|
|
||
|
|
if target:
|
||
|
|
# Add to MPV playlist
|
||
|
|
# We use loadfile with append flag
|
||
|
|
# Configure 1080p limit for streams (bestvideo<=1080p + bestaudio)
|
||
|
|
options = {
|
||
|
|
"ytdl-format": "bestvideo[height<=?1080]+bestaudio/best[height<=?1080]"
|
||
|
|
}
|
||
|
|
|
||
|
|
if title:
|
||
|
|
options["force-media-title"] = title
|
||
|
|
|
||
|
|
cmd = {"command": ["loadfile", target, "append", options], "request_id": 200}
|
||
|
|
resp = _send_ipc_command(cmd)
|
||
|
|
|
||
|
|
if resp is None:
|
||
|
|
# MPV not running (or died)
|
||
|
|
# Start MPV with remaining items
|
||
|
|
_start_mpv(items_to_add[i:])
|
||
|
|
return 0
|
||
|
|
elif resp.get("error") == "success":
|
||
|
|
added_count += 1
|
||
|
|
if title:
|
||
|
|
log(f"Queued: {title}")
|
||
|
|
else:
|
||
|
|
log(f"Queued: {target}")
|
||
|
|
|
||
|
|
if added_count > 0:
|
||
|
|
# If we added items, we might want to play the first one if nothing is playing?
|
||
|
|
# For now, just list the playlist
|
||
|
|
pass
|
||
|
|
|
||
|
|
# Get playlist from MPV
|
||
|
|
items = _get_playlist()
|
||
|
|
|
||
|
|
if not items:
|
||
|
|
log("MPV playlist is empty or MPV is not running.")
|
||
|
|
return 0
|
||
|
|
|
||
|
|
# If index is provided, perform action (Play or Clear)
|
||
|
|
if index_arg is not None:
|
||
|
|
try:
|
||
|
|
# Handle 1-based index
|
||
|
|
idx = int(index_arg) - 1
|
||
|
|
|
||
|
|
if idx < 0 or idx >= len(items):
|
||
|
|
log(f"Index {index_arg} out of range (1-{len(items)}).")
|
||
|
|
return 1
|
||
|
|
|
||
|
|
item = items[idx]
|
||
|
|
title = item.get("title") or item.get("filename") or "Unknown"
|
||
|
|
|
||
|
|
if clear_mode:
|
||
|
|
# Remove item
|
||
|
|
cmd = {"command": ["playlist-remove", idx], "request_id": 101}
|
||
|
|
resp = _send_ipc_command(cmd)
|
||
|
|
if resp and resp.get("error") == "success":
|
||
|
|
log(f"Removed: {title}")
|
||
|
|
# Refresh items for listing
|
||
|
|
items = _get_playlist()
|
||
|
|
list_mode = True
|
||
|
|
index_arg = None
|
||
|
|
else:
|
||
|
|
log(f"Failed to remove item: {resp.get('error') if resp else 'No response'}")
|
||
|
|
return 1
|
||
|
|
else:
|
||
|
|
# Play item
|
||
|
|
cmd = {"command": ["playlist-play-index", idx], "request_id": 102}
|
||
|
|
resp = _send_ipc_command(cmd)
|
||
|
|
if resp and resp.get("error") == "success":
|
||
|
|
log(f"Playing: {title}")
|
||
|
|
return 0
|
||
|
|
else:
|
||
|
|
log(f"Failed to play item: {resp.get('error') if resp else 'No response'}")
|
||
|
|
return 1
|
||
|
|
|
||
|
|
except ValueError:
|
||
|
|
log(f"Invalid index: {index_arg}")
|
||
|
|
return 1
|
||
|
|
|
||
|
|
# List items (Default action or after clear)
|
||
|
|
if list_mode or index_arg is None:
|
||
|
|
if not items:
|
||
|
|
log("MPV playlist is empty.")
|
||
|
|
return 0
|
||
|
|
|
||
|
|
table = ResultTable("MPV Playlist")
|
||
|
|
|
||
|
|
for i, item in enumerate(items):
|
||
|
|
is_current = item.get("current", False)
|
||
|
|
title = item.get("title") or ""
|
||
|
|
filename = item.get("filename") or ""
|
||
|
|
|
||
|
|
# Special handling for memory:// M3U playlists (used to pass titles via IPC)
|
||
|
|
if "memory://" in filename and "#EXTINF:" in filename:
|
||
|
|
try:
|
||
|
|
# Extract title from #EXTINF:-1,Title
|
||
|
|
# Use regex to find title between #EXTINF:-1, and newline
|
||
|
|
match = re.search(r"#EXTINF:-1,(.*?)(?:\n|\r|$)", filename)
|
||
|
|
if match:
|
||
|
|
extracted_title = match.group(1).strip()
|
||
|
|
if not title or title == "memory://":
|
||
|
|
title = extracted_title
|
||
|
|
|
||
|
|
# Extract actual URL
|
||
|
|
# Find the first line that looks like a URL and not a directive
|
||
|
|
lines = filename.splitlines()
|
||
|
|
for line in lines:
|
||
|
|
line = line.strip()
|
||
|
|
if line and not line.startswith('#') and not line.startswith('memory://'):
|
||
|
|
filename = line
|
||
|
|
break
|
||
|
|
except Exception:
|
||
|
|
pass
|
||
|
|
|
||
|
|
# Truncate if too long
|
||
|
|
if len(title) > 57:
|
||
|
|
title = title[:57] + "..."
|
||
|
|
if len(filename) > 27:
|
||
|
|
filename = filename[:27] + "..."
|
||
|
|
|
||
|
|
row = table.add_row()
|
||
|
|
row.add_column("#", str(i + 1))
|
||
|
|
row.add_column("Current", "*" if is_current else "")
|
||
|
|
row.add_column("Title", title)
|
||
|
|
row.add_column("Filename", filename)
|
||
|
|
|
||
|
|
table.set_row_selection_args(i, [str(i + 1)])
|
||
|
|
|
||
|
|
table.set_source_command(".pipe")
|
||
|
|
|
||
|
|
# Register results with pipeline context so @N selection works
|
||
|
|
ctx.set_last_result_table_overlay(table, items)
|
||
|
|
ctx.set_current_stage_table(table)
|
||
|
|
|
||
|
|
print(table)
|
||
|
|
|
||
|
|
return 0
|
||
|
|
|
||
|
|
def _start_mpv(items: List[Any]) -> None:
|
||
|
|
"""Start MPV with a list of items."""
|
||
|
|
ipc_pipe = _get_fixed_ipc_pipe()
|
||
|
|
|
||
|
|
cmd = ['mpv', f'--input-ipc-server={ipc_pipe}']
|
||
|
|
cmd.append('--ytdl-format=bestvideo[height<=?1080]+bestaudio/best[height<=?1080]')
|
||
|
|
|
||
|
|
# Add items
|
||
|
|
first_title_set = False
|
||
|
|
|
||
|
|
for item in items:
|
||
|
|
target = None
|
||
|
|
title = None
|
||
|
|
|
||
|
|
if isinstance(item, dict):
|
||
|
|
target = item.get("target") or item.get("url") or item.get("path")
|
||
|
|
title = item.get("title") or item.get("name")
|
||
|
|
elif hasattr(item, "target"):
|
||
|
|
target = item.target
|
||
|
|
title = getattr(item, "title", None)
|
||
|
|
elif isinstance(item, str):
|
||
|
|
target = item
|
||
|
|
|
||
|
|
if target:
|
||
|
|
if not first_title_set and title:
|
||
|
|
cmd.append(f'--force-media-title={title}')
|
||
|
|
first_title_set = True
|
||
|
|
cmd.append(target)
|
||
|
|
|
||
|
|
if len(cmd) > 3: # mpv + ipc + format + at least one file
|
||
|
|
try:
|
||
|
|
kwargs = {}
|
||
|
|
if platform.system() == 'Windows':
|
||
|
|
kwargs['creationflags'] = 0x00000008 # DETACHED_PROCESS
|
||
|
|
|
||
|
|
subprocess.Popen(cmd, stdin=subprocess.DEVNULL, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL, **kwargs)
|
||
|
|
log(f"Started MPV with {len(cmd)-3} items")
|
||
|
|
except Exception as e:
|
||
|
|
log(f"Error starting MPV: {e}", file=sys.stderr)
|
||
|
|
|
||
|
|
CMDLET = Cmdlet(
|
||
|
|
name=".pipe",
|
||
|
|
aliases=["pipe", "playlist", "queue", "ls-pipe"],
|
||
|
|
summary="Manage and play items in the MPV playlist via IPC",
|
||
|
|
usage=".pipe [index] [-clear]",
|
||
|
|
args=[
|
||
|
|
CmdletArg(
|
||
|
|
name="index",
|
||
|
|
type="int",
|
||
|
|
description="Index of item to play or clear",
|
||
|
|
required=False
|
||
|
|
),
|
||
|
|
CmdletArg(
|
||
|
|
name="clear",
|
||
|
|
type="flag",
|
||
|
|
description="Remove the selected item from the playlist"
|
||
|
|
),
|
||
|
|
CmdletArg(
|
||
|
|
name="list",
|
||
|
|
type="flag",
|
||
|
|
description="List items (default)"
|
||
|
|
),
|
||
|
|
],
|
||
|
|
exec=_run
|
||
|
|
)
|
||
|
|
|