dfdsf

2025-12-11 19:04:02 -08:00
parent 6863c6c7ea
commit 16d8a763cd
103 changed files with 4759 additions and 9156 deletions
--- a/Provider/youtube.py
+++ b/Provider/youtube.py
@@ -0,0 +1,94 @@
+from __future__ import annotations
+
+import json
+import shutil
+import subprocess
+import sys
+from typing import Any, Dict, List, Optional
+
+from Provider._base import SearchProvider, SearchResult
+from SYS.logger import log
+
+
+class YouTube(SearchProvider):
+    """Search provider for YouTube using yt-dlp."""
+
+    def search(
+        self,
+        query: str,
+        limit: int = 10,
+        filters: Optional[Dict[str, Any]] = None,
+        **kwargs: Any,
+    ) -> List[SearchResult]:
+        ytdlp_path = shutil.which("yt-dlp")
+        if not ytdlp_path:
+            log("[youtube] yt-dlp not found in PATH", file=sys.stderr)
+            return []
+
+        search_query = f"ytsearch{limit}:{query}"
+        cmd = [ytdlp_path, "--dump-json", "--flat-playlist", "--no-warnings", search_query]
+
+        try:
+            process = subprocess.run(
+                cmd,
+                capture_output=True,
+                text=True,
+                encoding="utf-8",
+                errors="replace",
+            )
+
+            if process.returncode != 0:
+                log(f"[youtube] yt-dlp failed: {process.stderr}", file=sys.stderr)
+                return []
+
+            results: List[SearchResult] = []
+            for line in process.stdout.splitlines():
+                if not line.strip():
+                    continue
+
+                try:
+                    video_data = json.loads(line)
+                except json.JSONDecodeError:
+                    continue
+
+                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="youtube",
+                        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,
+                        },
+                    )
+                )
+
+            return results
+
+        except Exception as exc:
+            log(f"[youtube] Error: {exc}", file=sys.stderr)
+            return []
+
+    def validate(self) -> bool:
+        return shutil.which("yt-dlp") is not None