ytui-media 0.1.0__py3-none-any.whl

app.py

@@ -0,0 +1,364 @@
+"""Terminal TUI Application for yt-dlp."""
+
+import sys
+from typing import Optional
+
+from textual.app import App, ComposeResult
+from textual.binding import Binding
+from textual.screen import Screen
+from textual.widgets import Footer, Header
+
+from config import AppConfig
+from ffmpeg_utils import ensure_ffmpeg_in_path, is_ffmpeg_available
+from history import HistoryManager
+from manager import DownloadManager
+from rtl_utils import set_rtl_mode
+from screens.download_screen import DownloadScreen
+from screens.ffmpeg_modal import FfmpegSetupModal
+from screens.history_screen import HistoryScreen
+from screens.search_screen import SearchScreen
+from screens.settings_screen import SettingsScreen
+from themes import register_all_themes
+
+
+
+class YtDlpApp(App):
+    """Clean, high-density terminal tool with customizable theme profiles."""
+
+    TITLE = "YTUI"
+    SUB_TITLE = "Terminal Media Downloader"
+    ENABLE_COMMAND_PALETTE = False
+
+    CSS = """
+    * {
+        scrollbar-background: $surface;
+        scrollbar-background-hover: $surface;
+        scrollbar-background-active: $panel;
+        scrollbar-color: $boost;
+        scrollbar-color-hover: $secondary;
+        scrollbar-color-active: $primary;
+        scrollbar-size-vertical: 1;
+        scrollbar-size-horizontal: 1;
+    }
+    Screen {
+        background: $background;
+        color: $foreground;
+        layout: vertical;
+        width: 100%;
+        height: 100%;
+        margin: 0;
+        padding: 0;
+        scrollbar-background: $surface;
+        scrollbar-color: $boost;
+        scrollbar-color-hover: $secondary;
+        scrollbar-color-active: $primary;
+        scrollbar-size-vertical: 1;
+        scrollbar-size-horizontal: 1;
+    }
+    ScrollBar {
+        background: $surface;
+        color: $boost;
+    }
+    ScrollBar.-vertical {
+        width: 1;
+        background: $surface;
+    }
+    ScrollBar.-horizontal {
+        height: 1;
+        background: $surface;
+    }
+    ScrollBar > .scrollbar--bar {
+        color: $boost;
+        background: $surface;
+    }
+    ScrollBar > .scrollbar--bar:hover {
+        color: $secondary;
+        background: $surface;
+    }
+    ScrollBar.-active > .scrollbar--bar {
+        color: $primary;
+        background: $panel;
+    }
+    ScrollBarCorner {
+        background: $surface;
+    }
+    Header {
+        background: $surface;
+        color: $primary;
+        height: 1;
+        dock: top;
+    }
+    Footer {
+        background: $surface;
+        color: $text-muted;
+        height: 1;
+        dock: bottom;
+    }
+    Footer > .footer--key {
+        background: $panel;
+        color: $primary;
+        text-style: bold;
+        padding: 0 1;
+    }
+    Footer > .footer--description {
+        color: $text-muted;
+        padding: 0 1 0 0;
+    }
+    Footer > .footer--highlight {
+        background: $boost;
+        color: $foreground;
+    }
+    Input {
+        background: $surface;
+        border: none;
+        height: 1;
+        padding: 0 1;
+        color: $foreground;
+    }
+    Input:focus {
+        background: $panel;
+        color: $foreground;
+        text-style: bold;
+    }
+    Checkbox {
+        background: transparent;
+        color: $foreground;
+        border: none;
+        height: auto;
+        padding: 0;
+    }
+    Checkbox:focus {
+        background: transparent;
+        color: $primary;
+        text-style: bold;
+    }
+    Checkbox > .toggle--label {
+        color: $foreground;
+        background: transparent;
+    }
+    Checkbox:focus > .toggle--label {
+        color: $primary;
+    }
+    Checkbox > .toggle--button {
+        color: $surface;
+        background: $boost;
+    }
+    Checkbox:focus > .toggle--button {
+        color: $background;
+        background: $primary;
+    }
+    Checkbox.-on > .toggle--button {
+        color: $background;
+        background: $primary;
+    }
+    DataTable {
+        background: $background;
+        border: none;
+        padding: 0;
+        color: $foreground;
+    }
+    DataTable > .datatable--header {
+        background: $surface;
+        color: $primary;
+        text-style: bold;
+        border-bottom: solid $border;
+    }
+    DataTable > .datatable--cursor {
+        background: $panel;
+        color: $primary;
+        text-style: bold;
+    }
+    DataTable > .datatable--hover {
+        background: $boost;
+    }
+    ListView {
+        background: $background;
+        padding: 0;
+    }
+    ListItem {
+        background: transparent;
+        margin: 0;
+        padding: 0 1;
+        border: none;
+        color: $foreground;
+    }
+    ListItem:focus {
+        background: $panel;
+        color: $primary;
+        text-style: bold;
+    }
+    ProgressBar {
+        height: 1;
+        padding: 0;
+        border: none;
+    }
+    ProgressBar > .progressbar--bar {
+        color: $primary;
+        background: $panel;
+    }
+    ProgressBar > .progressbar--complete {
+        color: $success;
+    }
+    Checkbox {
+        background: transparent;
+        border: none;
+        padding: 0 1;
+        height: 1;
+    }
+    Checkbox > .toggle--label {
+        color: $foreground;
+    }
+    Select {
+        background: $surface;
+        border: none;
+        height: 1;
+        padding: 0 1;
+    }
+    SelectCurrent {
+        border: none;
+        height: 1;
+        padding: 0 1;
+    }
+    Button {
+        height: 1;
+        min-width: 6;
+        padding: 0 1;
+        border: none;
+        background: $panel;
+        color: $foreground;
+    }
+    Button:focus, Button:hover {
+        background: $primary;
+        color: $background;
+        text-style: bold;
+    }
+    Button.-primary {
+        background: $primary;
+        color: $background;
+        text-style: bold;
+    }
+    Button.-success {
+        background: $success;
+        color: $background;
+        text-style: bold;
+    }
+    Button.-warning {
+        background: $warning;
+        color: $background;
+        text-style: bold;
+    }
+    Button.-error {
+        background: $error;
+        color: $background;
+        text-style: bold;
+    }
+    ToastHolder {
+        dock: top;
+        align: right top;
+        height: auto;
+        width: auto;
+    }
+    Toast {
+        background: $surface;
+        color: $foreground;
+        border: solid $border;
+        border-left: thick $primary;
+        padding: 0 1;
+        width: auto;
+        min-width: 32;
+        max-width: 60;
+    }
+    Toast > .toast--title {
+        color: $primary;
+        text-style: bold;
+    }
+    Toast > .toast--message {
+        color: $text-muted;
+    }
+    Toast.-information {
+        border-left: thick $primary;
+    }
+    Toast.-warning {
+        border-left: thick $warning;
+    }
+    Toast.-error {
+        border-left: thick $error;
+    }
+    LoadingIndicator {
+        color: $primary;
+        background: transparent;
+        height: 1;
+    }
+    """
+
+    BINDINGS = [
+        Binding("ctrl+s", "switch_to_search", "Search", show=True),
+        Binding("ctrl+j", "switch_to_downloads", "Queue", show=True),
+        Binding("ctrl+y", "switch_to_history", "History", show=True),
+        Binding("ctrl+o", "switch_to_settings", "Config", show=True),
+        Binding("ctrl+q", "quit_app", "Quit", show=True),
+    ]
+
+    SCREENS = {
+        "search_screen": SearchScreen,
+        "download_screen": DownloadScreen,
+        "history_screen": HistoryScreen,
+        "settings_screen": SettingsScreen,
+    }
+
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+        self.config = AppConfig.load()
+        self.history = HistoryManager()
+        self.manager = DownloadManager.get_instance(config=self.config, history=self.history)
+
+    def on_mount(self) -> None:
+        # Register all custom theme profiles
+        register_all_themes(self)
+
+        # Apply persisted theme and RTL mode
+        try:
+            self.theme = self.config.theme
+        except Exception:
+            pass
+        try:
+            set_rtl_mode(getattr(self.config, "rtl_mode", "reshaped_bidi"))
+        except Exception:
+            pass
+
+        # Ensure FFmpeg is configured in PATH
+        ensure_ffmpeg_in_path(getattr(self.config, "ffmpeg_location", None))
+
+        self.push_screen("search_screen")
+
+        # Prompt for FFmpeg download on first run / if missing
+        if not is_ffmpeg_available(getattr(self.config, "ffmpeg_location", None)) and not getattr(self.config, "skip_ffmpeg_check", False):
+            self.push_screen(FfmpegSetupModal(is_first_launch=True))
+
+    def action_switch_to_search(self) -> None:
+        self.switch_screen("search_screen")
+
+    def action_switch_to_downloads(self) -> None:
+        self.switch_screen("download_screen")
+
+    def action_switch_to_history(self) -> None:
+        self.switch_screen("history_screen")
+
+    def action_switch_to_settings(self) -> None:
+        self.switch_screen("settings_screen")
+
+    def on_unmount(self) -> None:
+        self.manager.shutdown()
+
+    def action_quit_app(self) -> None:
+        self.manager.shutdown()
+        self.exit()
+
+
+def main():
+    app = YtDlpApp()
+    app.run()
+
+
+if __name__ == "__main__":
+    main()

config.py

@@ -0,0 +1,232 @@
+"""Configuration manager for YTUI."""
+
+import json
+import os
+import sys
+from dataclasses import asdict, dataclass, field
+from pathlib import Path
+from typing import Any, Dict, Optional
+
+
+def sanitize_path(raw_path: Optional[str]) -> str:
+    """Sanitize user-entered path strings: strip quotes, expand variables, resolve tildes."""
+    if not raw_path:
+        return ""
+    # Strip surrounding single/double quotes, whitespace, and smart quotes
+    cleaned = str(raw_path).strip().strip('"\'“”‘’').strip()
+    if not cleaned:
+        return ""
+    # Expand environment variables (e.g. %USERPROFILE%, %APPDATA%)
+    cleaned = os.path.expandvars(cleaned)
+    # Expand user home tilde (~)
+    cleaned = os.path.expanduser(cleaned)
+    return cleaned
+
+
+DEFAULT_FILENAME_TEMPLATE = "%(title)s [%(id)s].%(ext)s"
+SPONSORBLOCK_CATEGORIES = (
+    "sponsor",
+    "selfpromo",
+    "interaction",
+    "intro",
+    "outro",
+    "preview",
+    "filler",
+    "music_offtopic",
+    "hook",
+)
+DEFAULT_SPONSORBLOCK_CATEGORIES = "sponsor,selfpromo"
+
+
+def sanitize_filename_template(template: Optional[str]) -> str:
+    """Sanitize and validate filename template to prevent directory traversal outside download directory."""
+    if not template or not isinstance(template, str):
+        return DEFAULT_FILENAME_TEMPLATE
+    cleaned = str(template).strip().strip('"\'“”‘’').strip()
+    if not cleaned:
+        return DEFAULT_FILENAME_TEMPLATE
+    # Block absolute paths (POSIX / Windows) or drive specifications
+    if cleaned.startswith(("/", "\\")) or (len(cleaned) >= 2 and cleaned[1] == ":"):
+        return DEFAULT_FILENAME_TEMPLATE
+    p = Path(cleaned)
+    if p.is_absolute():
+        return DEFAULT_FILENAME_TEMPLATE
+    # Block path traversal segments (..)
+    parts = cleaned.replace("\\", "/").split("/")
+    if ".." in parts:
+        return DEFAULT_FILENAME_TEMPLATE
+    return cleaned
+
+
+def sanitize_sponsorblock_categories(categories: Optional[str]) -> str:
+    """Return a normalized comma-separated list of removable SponsorBlock categories."""
+    requested = str(categories or "").lower().split(",")
+    valid_categories = []
+    for category in requested:
+        clean_category = category.strip()
+        if clean_category in SPONSORBLOCK_CATEGORIES and clean_category not in valid_categories:
+            valid_categories.append(clean_category)
+    return ",".join(valid_categories) or DEFAULT_SPONSORBLOCK_CATEGORIES
+
+
+def atomic_json_save(path: Path, data: Any, indent: int = 2, file_mode: int = 0o600, dir_mode: int = 0o700) -> None:
+    """Atomically write JSON data to path using a temporary file and os.replace."""
+    target_path = Path(path).resolve() if not isinstance(path, Path) else path
+    parent_dir = target_path.parent
+    parent_dir.mkdir(parents=True, exist_ok=True)
+    if os.name == "posix":
+        try:
+            os.chmod(parent_dir, dir_mode)
+        except OSError:
+            pass
+
+    tmp_path = parent_dir / f"{target_path.name}.tmp"
+    try:
+        with open(tmp_path, "w", encoding="utf-8") as f:
+            json.dump(data, f, indent=indent)
+        if os.name == "posix":
+            try:
+                os.chmod(tmp_path, file_mode)
+            except OSError:
+                pass
+        os.replace(tmp_path, target_path)
+    except Exception:
+        if tmp_path.exists():
+            try:
+                tmp_path.unlink()
+            except OSError:
+                pass
+        raise
+
+
+def get_default_download_dir() -> str:
+    """Return default user Downloads folder or current directory."""
+    home = Path.home()
+    downloads = home / "Downloads"
+    if downloads.exists():
+        return str(downloads)
+    return str(home)
+
+
+def get_config_dir() -> Path:
+    """Return platform-appropriate configuration directory."""
+    if sys.platform == "win32":
+        app_data = os.environ.get("APPDATA")
+        if app_data:
+            base = Path(app_data)
+        else:
+            base = Path.home() / "AppData" / "Roaming"
+    else:
+        base = Path.home() / ".config"
+    legacy_dir = base / "yt-dlp-tui"
+    cfg_dir = base / "ytui"
+    if not cfg_dir.exists() and legacy_dir.exists():
+        cfg_dir = legacy_dir
+    else:
+        cfg_dir.mkdir(parents=True, exist_ok=True)
+    if os.name == "posix":
+        try:
+            os.chmod(cfg_dir, 0o700)
+        except OSError:
+            pass
+    return cfg_dir
+
+
+@dataclass
+class AppConfig:
+    """Strongly typed application configuration settings."""
+
+    # Download paths & naming
+    download_dir: str = field(default_factory=get_default_download_dir)
+    filename_template: str = DEFAULT_FILENAME_TEMPLATE
+
+    def __post_init__(self) -> None:
+        if self.cookies_file:
+            self.cookies_file = sanitize_path(self.cookies_file)
+        if self.download_dir:
+            self.download_dir = sanitize_path(self.download_dir) or get_default_download_dir()
+        self.filename_template = sanitize_filename_template(self.filename_template)
+        self.sponsorblock_categories = sanitize_sponsorblock_categories(self.sponsorblock_categories)
+
+    # Bandwidth & concurrency
+    max_concurrent_downloads: int = 3
+    rate_limit: str = "0"  # e.g., '0' for unlimited, '2M', '500K'
+
+    # Network resiliency & auto-resume
+    retries: int = 10
+    fragment_retries: int = 10
+    continuedl: bool = True
+
+    # Authentication & Cookies
+    browser_cookies: str = "none"  # none, chrome, firefox, edge, brave, opera, vivaldi, safari
+    cookies_file: str = ""
+
+    # Subtitles
+    download_subtitles: bool = False
+    auto_generated_subtitles: bool = False
+    subtitle_mode: str = "embed"  # 'embed', 'external', or 'both'
+    subtitle_langs: str = "en"
+
+    # Thumbnails & Chapters
+    download_thumbnail: bool = False
+    thumbnail_mode: str = "embed"  # 'embed' or 'file'
+    embed_chapters: bool = False
+    split_chapters: bool = False
+    remove_sponsor_segments: bool = False
+    sponsorblock_categories: str = DEFAULT_SPONSORBLOCK_CATEGORIES
+
+    # Metadata
+    embed_metadata: bool = True
+
+    # Media Engine & FFmpeg
+    ffmpeg_location: str = ""
+    skip_ffmpeg_check: bool = False
+
+    # Proxy & Geo-bypass
+    proxy: str = ""
+    geo_bypass: bool = True
+
+    # Appearance & Display
+    theme: str = "shadcn-zinc"
+    rtl_mode: str = "reshaped_bidi"  # 'reshaped_bidi', 'native_raw', 'bidi_only', 'disabled'
+
+    @classmethod
+    def config_file_path(cls) -> Path:
+        return get_config_dir() / "config.json"
+
+    @classmethod
+    def load(cls) -> "AppConfig":
+        path = cls.config_file_path()
+        if not path.exists():
+            config = cls()
+            config.save()
+            return config
+        try:
+            with open(path, "r", encoding="utf-8") as f:
+                data = json.load(f)
+            # Filter out keys that don't match dataclass fields
+            valid_keys = {f.name for f in cls.__dataclass_fields__.values()}
+            filtered = {k: v for k, v in data.items() if k in valid_keys}
+            return cls(**filtered)
+        except Exception:
+            return cls()
+
+    def save(self) -> None:
+        path = self.config_file_path()
+        try:
+            atomic_json_save(path, asdict(self), indent=2, file_mode=0o600, dir_mode=0o700)
+        except Exception as e:
+            print(f"Warning: Could not save config: {e}", file=sys.stderr)
+
+    def to_dict(self) -> Dict[str, Any]:
+        return asdict(self)
+
+    def update(self, **kwargs: Any) -> None:
+        for k, v in kwargs.items():
+            if hasattr(self, k):
+                setattr(self, k, v)
+        if "filename_template" in kwargs:
+            self.filename_template = sanitize_filename_template(self.filename_template)
+        if "sponsorblock_categories" in kwargs:
+            self.sponsorblock_categories = sanitize_sponsorblock_categories(self.sponsorblock_categories)
+        self.save()