docker_cli.py

#!/usr/bin/env python
# coding: utf-8
"""Headless CLI downloader for Docker / NAS use (issue #28) — no GUI.

Downloads one or more supported URLs to $DOWNLOAD_DIR (default /downloads).
URLs are taken from, in order: command-line args, the $URLS / $URL env var
(space/comma separated), and a mounted text file ($URLS_FILE, default
/downloads/urls.txt — one URL per line, '#' comments allowed).

Examples:
  docker run --rm -v /nas/videos:/downloads ghcr.io/alos21750/jabletv https://jable.tv/videos/abc-123/
  docker run --rm -v /nas/videos:/downloads -e URLS_FILE=/downloads/urls.txt ghcr.io/alos21750/jabletv
"""
import os
import sys

# --- issue #23: point SSL/curl_cffi at certifi's ASCII-safe CA bundle BEFORE any
# curl_cffi import, so a non-UTF-8 default cert path can't crash the resolver. ---
try:
    import certifi as _certifi
    _ca = _certifi.where()
    if _ca and os.path.exists(_ca):
        os.environ.setdefault('SSL_CERT_FILE', _ca)
        os.environ.setdefault('SSL_CERT_DIR', os.path.dirname(_ca))
except Exception:
    pass

import M3U8Sites
import config
from M3U8Sites import M3U8Crawler


def gather_urls():
    urls = [u for u in sys.argv[1:] if u.strip()]
    env = (os.environ.get('URLS') or os.environ.get('URL') or '')
    urls += [u.strip() for u in env.replace(',', ' ').split() if u.strip()]
    path = os.environ.get('URLS_FILE', os.path.join(os.environ.get('DOWNLOAD_DIR', '/downloads'), 'urls.txt'))
    if os.path.isfile(path):
        with open(path, encoding='utf-8') as fh:
            urls += [ln.strip() for ln in fh if ln.strip() and not ln.lstrip().startswith('#')]
    seen, out = set(), []
    for u in urls:
        if u not in seen:
            seen.add(u); out.append(u)
    return out


def main():
    dest = os.environ.get('DOWNLOAD_DIR', '/downloads')
    os.makedirs(dest, exist_ok=True)
    res = os.environ.get('RESOLUTION')
    if res:
        try:
            M3U8Crawler.set_resolution_pref(res)
        except Exception:
            pass

    max_workers = None
    raw_workers = os.environ.get('MAX_WORKERS_PER_VIDEO')
    if raw_workers is not None and raw_workers.strip():
        max_workers = config.normalize_max_workers_per_video(raw_workers)

    urls = gather_urls()
    if not urls:
        print("no URL provided. Pass URL(s) as arguments, set the URLS env var, or add a"
              " urls.txt to the download dir.\n"
              "  docker run --rm -v /nas/videos:/downloads ghcr.io/alos21750/jabletv <URL>", flush=True)
        return 2

    worker_label = (
        max_workers if max_workers is not None
        else config.get_max_workers_per_video())
    print(f"下載目錄: {dest} | 解析度: {M3U8Crawler.get_resolution_pref()} | "
          f"每片執行緒: {worker_label} | 共 {len(urls)} 個網址", flush=True)
    ok = fail = 0
    for url in urls:
        print(f"\n========== {url} ==========", flush=True)
        try:
            site = M3U8Sites.CreateSite(
                url, dest, max_workers=max_workers)
            if site is None:
                print(f"[跳過] 不支援的網址: {url}", flush=True); fail += 1; continue
            if not site.is_url_vaildate():
                print(f"[失敗] 無法解析影片（版面改版 / 被 Cloudflare 阻擋 / 影片不存在）: {url}", flush=True); fail += 1; continue
            site.start_download()
            print(f"[完成] {url}", flush=True); ok += 1
        except Exception as exc:
            print(f"[錯誤] {url}: {exc}", flush=True); fail += 1
    print(f"\n===== 完成 {ok} 個，失敗 {fail} 個 =====", flush=True)
    return 0 if fail == 0 else 1


if __name__ == '__main__':
    sys.exit(main())