391 lines
15 KiB
Python
391 lines
15 KiB
Python
"""ytdlp format selector provider.
|
|
|
|
When a URL is passed through download-file, this provider displays available formats
|
|
in a table and routes format selection back to download-file with the chosen format
|
|
already specified via -format, skipping the format table on the second invocation.
|
|
|
|
This keeps format selection logic in ytdlp and leaves add-file plug-and-play.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
from typing import Any, Dict, Iterable, List, Optional
|
|
|
|
from ProviderCore.base import Provider, SearchResult
|
|
from SYS.provider_helpers import TableProviderMixin
|
|
from SYS.logger import log, debug
|
|
from tool.ytdlp import list_formats, is_url_supported_by_ytdlp
|
|
|
|
|
|
class ytdlp(TableProviderMixin, Provider):
|
|
"""ytdlp format selector and video search provider.
|
|
|
|
DUAL FUNCTIONALITY:
|
|
1. FORMAT SELECTION: When download-file is used with a yt-dlp supported URL,
|
|
displays available formats in a table for user selection.
|
|
2. SEARCH: When search-file is used with -provider ytdlp, searches YouTube
|
|
(and other yt-dlp supported sites) for videos.
|
|
|
|
FORMAT SELECTION USAGE:
|
|
- User runs: download-file "https://example.com/video"
|
|
- If URL is ytdlp-supported and no format specified, displays format table
|
|
- User selects @N (e.g., @3 for format index 3)
|
|
- Selection args include -format <format_id>, re-invoking download-file
|
|
- Second download-file call sees -format and skips table, downloads directly
|
|
|
|
SEARCH USAGE:
|
|
- User runs: search-file -provider ytdlp "linux tutorial"
|
|
- Shows YouTube search results as a table
|
|
- User selects @1 to download that video
|
|
- Selection args route to download-file for streaming download
|
|
|
|
SELECTION FLOW (Format):
|
|
1. download-file receives URL without -format
|
|
2. Calls ytdlp to list formats
|
|
3. Returns formats as ResultTable (from this provider)
|
|
4. User selects @N
|
|
5. Selection args: ["-format", "<format_id>"] route back to download-file
|
|
6. Second download-file invocation with -format skips table
|
|
|
|
SELECTION FLOW (Search):
|
|
1. search-file lists YouTube videos via yt_dlp
|
|
2. Returns videos as ResultTable (from this provider)
|
|
3. User selects @N
|
|
4. Selection args: ["-url", "<youtube_url>"] route to download-file
|
|
5. download-file downloads the selected video
|
|
|
|
TABLE AUTO-STAGES:
|
|
- Format selection: ytdlp.formatlist -> download-file (with -format)
|
|
- Video search: ytdlp.search -> download-file (with -url)
|
|
|
|
SUPPORTED URLS:
|
|
This provider dynamically discovers all yt-dlp supported sites via yt_dlp.gen_extractors().
|
|
"""
|
|
|
|
# Dynamically load URL domains from yt-dlp's extractors
|
|
# This enables provider auto-discovery for format selection routing
|
|
@property
|
|
def URL(self) -> List[str]:
|
|
"""Get list of supported domains from yt-dlp extractors."""
|
|
try:
|
|
import yt_dlp
|
|
# Build a comprehensive list from known extractors and fallback domains
|
|
domains = set(self._fallback_domains)
|
|
|
|
# Try to get extractors and extract domain info
|
|
try:
|
|
extractors = yt_dlp.gen_extractors()
|
|
for extractor_class in extractors:
|
|
# Get extractor name and try to convert to domain
|
|
name = getattr(extractor_class, 'IE_NAME', '')
|
|
if name and name not in ('generic', 'http'):
|
|
# Convert extractor name to domain (e.g., 'YouTube' -> 'youtube.com')
|
|
name_lower = name.lower().replace('ie', '').strip()
|
|
if name_lower and len(name_lower) > 2:
|
|
domains.add(f"{name_lower}.com")
|
|
except Exception:
|
|
pass
|
|
|
|
return list(domains) if domains else self._fallback_domains
|
|
except Exception:
|
|
return self._fallback_domains
|
|
|
|
# Fallback common domains in case extraction fails
|
|
_fallback_domains = [
|
|
"youtube.com", "youtu.be",
|
|
"bandcamp.com",
|
|
"vimeo.com",
|
|
"twitch.tv",
|
|
"dailymotion.com",
|
|
"rumble.com",
|
|
"odysee.com",
|
|
]
|
|
|
|
TABLE_AUTO_STAGES = {
|
|
"ytdlp.formatlist": ["download-file"],
|
|
"ytdlp.search": ["download-file"],
|
|
}
|
|
# Forward selection args (including -format or -url) to the next stage
|
|
AUTO_STAGE_USE_SELECTION_ARGS = True
|
|
|
|
def search(
|
|
self,
|
|
query: str,
|
|
limit: int = 50,
|
|
filters: Optional[Dict[str, Any]] = None,
|
|
**kwargs: Any,
|
|
) -> List[SearchResult]:
|
|
"""
|
|
NOT USED: This provider is invoked via ResultTable integration, not search.
|
|
Formats are fetched directly in download-file and returned as ResultTable rows
|
|
with this provider registered as the handler.
|
|
"""
|
|
return []
|
|
|
|
def search(
|
|
self,
|
|
query: str,
|
|
limit: int = 10,
|
|
filters: Optional[Dict[str, Any]] = None,
|
|
**kwargs: Any,
|
|
) -> List[SearchResult]:
|
|
"""Search YouTube and other yt-dlp supported sites for videos.
|
|
|
|
Uses yt-dlp's ytsearch capability to find videos, then returns them
|
|
as SearchResult rows for table display and selection.
|
|
"""
|
|
try:
|
|
import yt_dlp # type: ignore
|
|
|
|
ydl_opts: Dict[str, Any] = {
|
|
"quiet": True,
|
|
"skip_download": True,
|
|
"extract_flat": True
|
|
}
|
|
with yt_dlp.YoutubeDL(ydl_opts) as ydl: # type: ignore[arg-type]
|
|
search_query = f"ytsearch{limit}:{query}"
|
|
info = ydl.extract_info(search_query, download=False)
|
|
entries = info.get("entries") or []
|
|
results: List[SearchResult] = []
|
|
for video_data in entries[:limit]:
|
|
title = video_data.get("title", "Unknown")
|
|
video_id = video_data.get("id", "")
|
|
url = video_data.get(
|
|
"url"
|
|
) or f"https://youtube.com/watch?v={video_id}"
|
|
uploader = video_data.get("uploader", "Unknown")
|
|
duration = video_data.get("duration", 0)
|
|
view_count = video_data.get("view_count", 0)
|
|
|
|
duration_str = (
|
|
f"{int(duration // 60)}:{int(duration % 60):02d}"
|
|
if duration else ""
|
|
)
|
|
views_str = f"{view_count:,}" if view_count else ""
|
|
|
|
results.append(
|
|
SearchResult(
|
|
table="ytdlp.search",
|
|
title=title,
|
|
path=url,
|
|
detail=f"By: {uploader}",
|
|
annotations=[duration_str, f"{views_str} views"],
|
|
media_kind="video",
|
|
columns=[
|
|
("Title", title),
|
|
("Uploader", uploader),
|
|
("Duration", duration_str),
|
|
("Views", views_str),
|
|
],
|
|
full_metadata={
|
|
"video_id": video_id,
|
|
"uploader": uploader,
|
|
"duration": duration,
|
|
"view_count": view_count,
|
|
# Selection metadata for table system and @N expansion
|
|
"_selection_args": ["-url", url],
|
|
},
|
|
)
|
|
)
|
|
return results
|
|
except Exception:
|
|
debug("[ytdlp] yt_dlp import or search failed")
|
|
return []
|
|
|
|
def validate(self) -> bool:
|
|
"""Validate yt-dlp availability."""
|
|
try:
|
|
import yt_dlp # type: ignore
|
|
return True
|
|
except Exception:
|
|
return False
|
|
|
|
|
|
# Minimal provider registration for the new table system
|
|
try:
|
|
from SYS.result_table_adapters import register_provider
|
|
from SYS.result_table_api import ResultModel, ColumnSpec, metadata_column, title_column
|
|
|
|
def _convert_format_result_to_model(sr: Any) -> ResultModel:
|
|
"""Convert format dict to ResultModel for strict table display."""
|
|
d = sr.to_dict() if hasattr(sr, "to_dict") else (sr if isinstance(sr, dict) else {})
|
|
title = d.get("title") or f"Format {d.get('format_id', 'unknown')}"
|
|
|
|
# Extract metadata from columns and full_metadata
|
|
metadata: Dict[str, Any] = {}
|
|
columns = d.get("columns") or []
|
|
for name, value in columns:
|
|
key = str(name or "").strip().lower()
|
|
if key in ("id", "resolution", "ext", "size", "video", "audio", "format_id"):
|
|
metadata[key] = value
|
|
|
|
try:
|
|
fm = d.get("full_metadata") or {}
|
|
if isinstance(fm, dict):
|
|
for k, v in fm.items():
|
|
metadata[str(k).strip().lower()] = v
|
|
except Exception:
|
|
pass
|
|
|
|
return ResultModel(
|
|
title=str(title),
|
|
path=d.get("url") or d.get("target"),
|
|
ext=d.get("ext"),
|
|
size_bytes=None,
|
|
metadata=metadata,
|
|
source="ytdlp"
|
|
)
|
|
|
|
def _adapter(items: Iterable[Any]) -> Iterable[ResultModel]:
|
|
"""Adapter to convert format results to ResultModels."""
|
|
for it in items:
|
|
try:
|
|
yield _convert_format_result_to_model(it)
|
|
except Exception:
|
|
continue
|
|
|
|
def _has_metadata(rows: List[ResultModel], key: str) -> bool:
|
|
"""Check if any row has a given metadata key with a non-empty value."""
|
|
for row in rows:
|
|
md = row.metadata or {}
|
|
if key in md:
|
|
val = md[key]
|
|
if val is None:
|
|
continue
|
|
if isinstance(val, str) and not val.strip():
|
|
continue
|
|
return True
|
|
return False
|
|
|
|
def _columns_factory(rows: List[ResultModel]) -> List[ColumnSpec]:
|
|
"""Build column specifications from available metadata in rows."""
|
|
cols = [title_column()]
|
|
if _has_metadata(rows, "resolution"):
|
|
cols.append(metadata_column("resolution", "Resolution"))
|
|
if _has_metadata(rows, "ext"):
|
|
cols.append(metadata_column("ext", "Ext"))
|
|
if _has_metadata(rows, "size"):
|
|
cols.append(metadata_column("size", "Size"))
|
|
if _has_metadata(rows, "video"):
|
|
cols.append(metadata_column("video", "Video"))
|
|
if _has_metadata(rows, "audio"):
|
|
cols.append(metadata_column("audio", "Audio"))
|
|
return cols
|
|
|
|
def _selection_fn(row: ResultModel) -> List[str]:
|
|
"""Return selection args for format selection.
|
|
|
|
When user selects @N, these args are passed to download-file which sees
|
|
the -format specifier and skips the format table, downloading directly.
|
|
"""
|
|
metadata = row.metadata or {}
|
|
|
|
# Check for explicit selection args first
|
|
args = metadata.get("_selection_args") or metadata.get("selection_args")
|
|
if isinstance(args, (list, tuple)) and args:
|
|
result_args = [str(x) for x in args if x is not None]
|
|
debug(f"[ytdlp] Selection routed with args: {result_args}")
|
|
return result_args
|
|
|
|
# Fallback: use format_id
|
|
format_id = metadata.get("format_id") or metadata.get("id")
|
|
if format_id:
|
|
result_args = ["-format", str(format_id)]
|
|
debug(f"[ytdlp] Selection routed with format_id: {format_id}")
|
|
return result_args
|
|
|
|
debug(f"[ytdlp] Warning: No selection args or format_id found in row")
|
|
return []
|
|
|
|
register_provider(
|
|
"ytdlp.formatlist",
|
|
_adapter,
|
|
columns=_columns_factory,
|
|
selection_fn=_selection_fn,
|
|
metadata={"description": "ytdlp format selector for streaming media"},
|
|
)
|
|
debug("[ytdlp] Provider registered successfully with TABLE_AUTO_STAGES routing to download-file")
|
|
|
|
# Also register the search table
|
|
def _convert_search_result_to_model(sr: Any) -> ResultModel:
|
|
"""Convert YouTube SearchResult to ResultModel for strict table display."""
|
|
d = sr.to_dict() if hasattr(sr, "to_dict") else (sr if isinstance(sr, dict) else {"title": getattr(sr, "title", str(sr))})
|
|
title = d.get("title") or ""
|
|
path = d.get("path") or None
|
|
columns = d.get("columns") or getattr(sr, "columns", None) or []
|
|
|
|
# Extract metadata from columns and full_metadata
|
|
metadata: Dict[str, Any] = {}
|
|
for name, value in columns:
|
|
key = str(name or "").strip().lower()
|
|
if key in ("uploader", "duration", "views", "video_id"):
|
|
metadata[key] = value
|
|
|
|
try:
|
|
fm = d.get("full_metadata") or {}
|
|
if isinstance(fm, dict):
|
|
for k, v in fm.items():
|
|
metadata[str(k).strip().lower()] = v
|
|
except Exception:
|
|
pass
|
|
|
|
return ResultModel(
|
|
title=str(title),
|
|
path=str(path) if path else None,
|
|
ext=None,
|
|
size_bytes=None,
|
|
metadata=metadata,
|
|
source="ytdlp"
|
|
)
|
|
|
|
def _search_adapter(items: Iterable[Any]) -> Iterable[ResultModel]:
|
|
"""Adapter to convert search results to ResultModels."""
|
|
for it in items:
|
|
try:
|
|
yield _convert_search_result_to_model(it)
|
|
except Exception:
|
|
continue
|
|
|
|
def _search_columns_factory(rows: List[ResultModel]) -> List[ColumnSpec]:
|
|
"""Build column specifications for search results."""
|
|
cols = [title_column()]
|
|
if _has_metadata(rows, "uploader"):
|
|
cols.append(metadata_column("uploader", "Uploader"))
|
|
if _has_metadata(rows, "duration"):
|
|
cols.append(metadata_column("duration", "Duration"))
|
|
if _has_metadata(rows, "views"):
|
|
cols.append(metadata_column("views", "Views"))
|
|
return cols
|
|
|
|
def _search_selection_fn(row: ResultModel) -> List[str]:
|
|
"""Return selection args for search results.
|
|
|
|
When user selects @N from search results, route to download-file with -url.
|
|
"""
|
|
metadata = row.metadata or {}
|
|
|
|
# Check for explicit selection args first
|
|
args = metadata.get("_selection_args") or metadata.get("selection_args")
|
|
if isinstance(args, (list, tuple)) and args:
|
|
return [str(x) for x in args if x is not None]
|
|
|
|
# Fallback to direct URL
|
|
if row.path:
|
|
return ["-url", row.path]
|
|
|
|
return ["-title", row.title or ""]
|
|
|
|
register_provider(
|
|
"ytdlp.search",
|
|
_search_adapter,
|
|
columns=_search_columns_factory,
|
|
selection_fn=_search_selection_fn,
|
|
metadata={"description": "ytdlp video search using yt-dlp"},
|
|
)
|
|
debug("[ytdlp] Search provider registered successfully")
|
|
except Exception as e:
|
|
# best-effort registration
|
|
debug(f"[ytdlp] Provider registration note: {e}")
|
|
pass
|