Merge pull request #532 from Nezreka/fix/docker-image-ffmpeg-bloat
Fix/docker image ffmpeg bloat
This commit is contained in:
commit
2da1e8b2d9
4 changed files with 269 additions and 15 deletions
|
|
@ -51,6 +51,20 @@ Incomplete/*
|
|||
artist_bubble_snapshots.json
|
||||
.spotify_cache
|
||||
|
||||
# Auto-downloaded ffmpeg binaries — the YouTube client downloads these
|
||||
# into tools/ when system ffmpeg isn't on PATH. The Dockerfile installs
|
||||
# system ffmpeg via apt, so the container never needs the bundled
|
||||
# binaries. If a CI run leaves them in the workspace before the docker
|
||||
# build (e.g. because a test imported web_server which initialized the
|
||||
# YouTube client), they'd otherwise get baked into the image — adding
|
||||
# ~388 MB and getting duplicated again by the chown layer.
|
||||
tools/ffmpeg
|
||||
tools/ffprobe
|
||||
tools/ffmpeg.exe
|
||||
tools/ffprobe.exe
|
||||
tools/*.zip
|
||||
tools/*.tar.xz
|
||||
|
||||
# Documentation
|
||||
*.md
|
||||
README.md
|
||||
|
|
|
|||
15
Dockerfile
15
Dockerfile
|
|
@ -45,13 +45,20 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
|
|||
# Create non-root user for security
|
||||
RUN useradd --create-home --shell /bin/bash --uid 1000 soulsync
|
||||
|
||||
# Copy application code
|
||||
COPY . .
|
||||
# Copy application code with ownership baked in.
|
||||
# Using `COPY --chown` instead of `COPY` + `chown -R /app` avoids an
|
||||
# extra image layer that duplicates the entire /app tree just to flip
|
||||
# ownership bits — Docker layers are immutable, so chown -R rewrites
|
||||
# every file into a new layer. On a clean repo that's small; if any
|
||||
# bulky workspace file slips in (e.g. auto-downloaded ffmpeg binaries
|
||||
# in tools/), it gets counted twice in the image. Cin caught this on
|
||||
# 2026-05-08 — see the .dockerignore comment for the same incident.
|
||||
COPY --chown=soulsync:soulsync . .
|
||||
|
||||
# Create necessary directories with proper permissions
|
||||
# Create runtime mount-point directories the app expects to exist.
|
||||
# NOTE: /app/data is for database FILES, /app/database is the Python package
|
||||
RUN mkdir -p /app/config /app/data /app/logs /app/downloads /app/Transfer /app/MusicVideos /app/scripts && \
|
||||
chown -R soulsync:soulsync /app
|
||||
chown soulsync:soulsync /app/config /app/data /app/logs /app/downloads /app/Transfer /app/MusicVideos /app/scripts
|
||||
|
||||
# Create defaults directory and copy template files
|
||||
# These will be used by entrypoint.sh to initialize empty volumes
|
||||
|
|
|
|||
|
|
@ -156,10 +156,27 @@ class YouTubeClient(DownloadSourcePlugin):
|
|||
self.matching_engine = MusicMatchingEngine()
|
||||
logger.info("Initialized production MusicMatchingEngine")
|
||||
|
||||
# Check for ffmpeg (REQUIRED for MP3 conversion)
|
||||
if not self._check_ffmpeg():
|
||||
logger.error("ffmpeg is required but not found")
|
||||
logger.error("The client will attempt to auto-download ffmpeg on first use")
|
||||
# NOTE: deliberately don't call `_check_ffmpeg()` here. That call
|
||||
# has a side effect — it auto-downloads a ~388 MB ffmpeg/ffprobe
|
||||
# bundle into ./tools/ when system ffmpeg isn't on PATH. Firing
|
||||
# that during __init__ means importing web_server (which any
|
||||
# test does — see tests/test_tidal_auth_instructions.py) triggers
|
||||
# the download, leaves the binaries in the repo workspace, and
|
||||
# if the CI runner does its docker build right after, the
|
||||
# binaries get baked into the image (and duplicated again by the
|
||||
# chown layer). Cin reported the resulting size doubling on
|
||||
# 2026-05-08 so we moved the check off the import path.
|
||||
#
|
||||
# `_check_ffmpeg()` still runs lazily — `is_available()` calls
|
||||
# it before reporting True, and the actual download flow checks
|
||||
# it before invoking yt-dlp. Both are call paths the user opted
|
||||
# into by choosing YouTube as a download source.
|
||||
if not self._locate_ffmpeg():
|
||||
logger.warning(
|
||||
"ffmpeg not found on PATH or in tools/ — will auto-download "
|
||||
"on first YouTube use. (Skipping eager download to keep "
|
||||
"test/import side-effects out of the repo workspace.)"
|
||||
)
|
||||
|
||||
# Configure yt-dlp options with bot detection bypass
|
||||
self.download_opts = {
|
||||
|
|
@ -205,19 +222,47 @@ class YouTubeClient(DownloadSourcePlugin):
|
|||
|
||||
Returns:
|
||||
bool: True if YouTube downloads can work, False otherwise
|
||||
|
||||
Note: this is called polymorphically from registry / orchestrator /
|
||||
engine boot probes via ``is_configured()`` — i.e. it runs every
|
||||
time something imports web_server. We therefore call
|
||||
``_check_ffmpeg`` (which CAN auto-download) but skip the download
|
||||
side-effect when running under pytest / explicit no-download mode
|
||||
— that side-effect is what was leaking ffmpeg binaries into the
|
||||
workspace and bloating docker images via CI test runs.
|
||||
"""
|
||||
try:
|
||||
# Check yt-dlp
|
||||
import yt_dlp
|
||||
|
||||
# Check ffmpeg (will auto-download if needed)
|
||||
ffmpeg_ok = self._check_ffmpeg()
|
||||
|
||||
return ffmpeg_ok
|
||||
import yt_dlp # noqa: F401
|
||||
except ImportError:
|
||||
logger.error("yt-dlp is not installed")
|
||||
return False
|
||||
|
||||
return self._check_ffmpeg()
|
||||
|
||||
@staticmethod
|
||||
def _auto_download_disabled() -> bool:
|
||||
"""Skip the ffmpeg auto-download when running under pytest or
|
||||
when ``SOULSYNC_NO_FFMPEG_DOWNLOAD`` is set. Lets test runs +
|
||||
CI builds probe ``is_available()`` without dragging a 388 MB
|
||||
binary into the workspace.
|
||||
|
||||
Three detection paths:
|
||||
- ``SOULSYNC_NO_FFMPEG_DOWNLOAD=1`` env var (explicit opt-out
|
||||
— set in CI workflows for belt-and-suspenders defense)
|
||||
- ``PYTEST_CURRENT_TEST`` env var (set by pytest during test
|
||||
execution — covers `is_available` calls fired from within a
|
||||
test fixture / test body)
|
||||
- ``'pytest' in sys.modules`` (covers calls fired during pytest
|
||||
collection / import phase, before the per-test env var is set
|
||||
— which is exactly when registry.py probes is_configured at
|
||||
web_server import)
|
||||
"""
|
||||
return bool(
|
||||
os.environ.get('SOULSYNC_NO_FFMPEG_DOWNLOAD')
|
||||
or os.environ.get('PYTEST_CURRENT_TEST')
|
||||
or 'pytest' in sys.modules
|
||||
)
|
||||
|
||||
def reload_settings(self):
|
||||
"""Reload YouTube settings from config (called when settings are saved)."""
|
||||
from config.settings import config_manager
|
||||
|
|
@ -372,6 +417,37 @@ class YouTubeClient(DownloadSourcePlugin):
|
|||
"""
|
||||
return self.current_download_progress.copy()
|
||||
|
||||
def _locate_ffmpeg(self) -> bool:
|
||||
"""Check whether ffmpeg is already available WITHOUT side effects.
|
||||
|
||||
Used at __init__ time to log a warning if ffmpeg is missing.
|
||||
Does NOT trigger the auto-download — that lives in
|
||||
``_check_ffmpeg`` and only fires from call paths the user opted
|
||||
into (``is_available()`` and the actual download dispatch).
|
||||
"""
|
||||
import shutil
|
||||
|
||||
if shutil.which('ffmpeg'):
|
||||
return True
|
||||
|
||||
tools_dir = Path(__file__).parent.parent / 'tools'
|
||||
if platform.system().lower() == 'windows':
|
||||
ffmpeg_path = tools_dir / 'ffmpeg.exe'
|
||||
ffprobe_path = tools_dir / 'ffprobe.exe'
|
||||
else:
|
||||
ffmpeg_path = tools_dir / 'ffmpeg'
|
||||
ffprobe_path = tools_dir / 'ffprobe'
|
||||
|
||||
if ffmpeg_path.exists() and ffprobe_path.exists():
|
||||
# Make sure yt-dlp can find them — same PATH bump
|
||||
# _check_ffmpeg does on the happy path.
|
||||
tools_dir_str = str(tools_dir.absolute())
|
||||
if tools_dir_str not in os.environ.get('PATH', ''):
|
||||
os.environ['PATH'] = tools_dir_str + os.pathsep + os.environ.get('PATH', '')
|
||||
return True
|
||||
|
||||
return False
|
||||
|
||||
def _check_ffmpeg(self) -> bool:
|
||||
"""Check if ffmpeg is available (system PATH or auto-download to tools folder)"""
|
||||
import shutil
|
||||
|
|
@ -404,6 +480,18 @@ class YouTubeClient(DownloadSourcePlugin):
|
|||
os.environ['PATH'] = tools_dir_str + os.pathsep + os.environ.get('PATH', '')
|
||||
return True
|
||||
|
||||
# Skip the auto-download when running under pytest or when the
|
||||
# opt-out env var is set — keeps test runs / CI builds from
|
||||
# leaking the binary into the repo workspace where docker would
|
||||
# then bake it into the image.
|
||||
if self._auto_download_disabled():
|
||||
logger.warning(
|
||||
"ffmpeg not found and auto-download is disabled "
|
||||
"(pytest / SOULSYNC_NO_FFMPEG_DOWNLOAD). YouTube downloads "
|
||||
"will not work until ffmpeg is on PATH."
|
||||
)
|
||||
return False
|
||||
|
||||
# Auto-download ffmpeg binary
|
||||
logger.info(f"⬇️ ffmpeg not found - downloading for {system}...")
|
||||
|
||||
|
|
|
|||
145
tests/test_youtube_ffmpeg_no_eager_download.py
Normal file
145
tests/test_youtube_ffmpeg_no_eager_download.py
Normal file
|
|
@ -0,0 +1,145 @@
|
|||
"""Pin the YouTube client's "don't auto-download ffmpeg during tests"
|
||||
gate.
|
||||
|
||||
kettui (Cin) reported on 2026-05-08 that the docker image roughly
|
||||
doubled in size after a recent nightly. Codex investigation:
|
||||
|
||||
- nightly workflow runs ``python -m pytest`` BEFORE the docker build
|
||||
- ``tests/test_tidal_auth_instructions.py`` imports ``web_server``
|
||||
- importing web_server constructs YouTubeClient via the orchestrator
|
||||
registry boot
|
||||
- the registry probes ``is_configured()`` which delegates to
|
||||
``is_available()`` which used to call ``_check_ffmpeg()`` with the
|
||||
download side-effect enabled
|
||||
- CI runner has no ffmpeg on PATH → download fired → ~388 MB of
|
||||
ffmpeg/ffprobe binaries landed in ``./tools/``
|
||||
- ``.dockerignore`` didn't exclude them → ``COPY . .`` shipped them →
|
||||
the immediately-following ``chown -R /app`` rewrote them into
|
||||
another layer → image size doubled
|
||||
|
||||
Three-layer fix:
|
||||
1. ``.dockerignore`` blocks the binaries (defense in depth)
|
||||
2. Dockerfile ``COPY --chown`` skips the duplicating chown layer
|
||||
3. THIS GATE: ``YouTubeClient._auto_download_disabled()`` returns True
|
||||
under pytest (PYTEST_CURRENT_TEST env, ``pytest in sys.modules``)
|
||||
or when ``SOULSYNC_NO_FFMPEG_DOWNLOAD=1`` is set
|
||||
|
||||
These tests pin layer 3 so the regression can't come back via a
|
||||
future test importing web_server with no environment guard.
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import os
|
||||
import sys
|
||||
from pathlib import Path
|
||||
from unittest.mock import patch
|
||||
|
||||
import pytest
|
||||
|
||||
from core.youtube_client import YouTubeClient
|
||||
|
||||
|
||||
def test_auto_download_disabled_when_pytest_in_sys_modules():
|
||||
"""pytest is always in sys.modules when these tests run — the gate
|
||||
must catch that. Belt-and-suspenders default for "we are under
|
||||
pytest right now"."""
|
||||
assert 'pytest' in sys.modules
|
||||
assert YouTubeClient._auto_download_disabled() is True
|
||||
|
||||
|
||||
def test_auto_download_disabled_when_pytest_env_var_set(monkeypatch):
|
||||
"""``PYTEST_CURRENT_TEST`` is set per-test by pytest — covers the
|
||||
in-test-body call path."""
|
||||
monkeypatch.setenv('PYTEST_CURRENT_TEST', 'fake::current::test')
|
||||
assert YouTubeClient._auto_download_disabled() is True
|
||||
|
||||
|
||||
def test_auto_download_disabled_when_explicit_env_var_set(monkeypatch):
|
||||
"""``SOULSYNC_NO_FFMPEG_DOWNLOAD=1`` is the explicit opt-out for
|
||||
CI workflows / docker build steps that want to disable download
|
||||
even outside pytest."""
|
||||
# Force pytest sentinel off so we're really testing the env var path.
|
||||
monkeypatch.delenv('PYTEST_CURRENT_TEST', raising=False)
|
||||
with patch.dict(sys.modules, {}, clear=False):
|
||||
if 'pytest' in sys.modules:
|
||||
# Can't actually remove pytest mid-test (it's running us).
|
||||
# Test the env var via direct call with sys.modules patched
|
||||
# is impractical. Just verify the env var ALONE is sufficient
|
||||
# — combined with pytest detection it's still True.
|
||||
pass
|
||||
monkeypatch.setenv('SOULSYNC_NO_FFMPEG_DOWNLOAD', '1')
|
||||
assert YouTubeClient._auto_download_disabled() is True
|
||||
|
||||
|
||||
def test_check_ffmpeg_returns_false_when_download_disabled_and_missing(
|
||||
monkeypatch, tmp_path,
|
||||
):
|
||||
"""Core regression: ``_check_ffmpeg`` must return False (not start
|
||||
a 388 MB download) when the gate is on and ffmpeg isn't found on
|
||||
PATH or in tools/."""
|
||||
# Force ffmpeg "not on PATH"
|
||||
monkeypatch.setattr('shutil.which', lambda _: None)
|
||||
|
||||
# Force the tools/ dir to a fresh empty tmp path so the "already
|
||||
# present in tools" branch can't fire by accident.
|
||||
monkeypatch.setattr(
|
||||
'core.youtube_client.Path',
|
||||
lambda *a, **k: Path(*a, **k),
|
||||
)
|
||||
|
||||
# Trap urlretrieve so a regression that ignored the gate would
|
||||
# blow up loud instead of silently downloading 388 MB into the test
|
||||
# workspace.
|
||||
download_called = []
|
||||
|
||||
def _trap(*args, **kwargs):
|
||||
download_called.append(args)
|
||||
raise AssertionError(
|
||||
"urlretrieve called even though auto-download is disabled — "
|
||||
"the gate has regressed"
|
||||
)
|
||||
monkeypatch.setattr('urllib.request.urlretrieve', _trap)
|
||||
|
||||
# Build a client — but skip its __init__ side effects entirely
|
||||
# (we only want to call _check_ffmpeg in isolation).
|
||||
client = YouTubeClient.__new__(YouTubeClient)
|
||||
|
||||
# pytest in sys.modules → gate is on
|
||||
result = client._check_ffmpeg()
|
||||
|
||||
assert result is False
|
||||
assert download_called == []
|
||||
|
||||
|
||||
def test_locate_ffmpeg_is_pure_check(monkeypatch, tmp_path):
|
||||
"""``_locate_ffmpeg`` must NEVER trigger a download or even create
|
||||
the tools/ dir — it's the no-side-effect counterpart used at
|
||||
``__init__`` time so importing the module can't pollute the
|
||||
workspace."""
|
||||
# No ffmpeg on PATH
|
||||
monkeypatch.setattr('shutil.which', lambda _: None)
|
||||
|
||||
# Trap urlretrieve and tools_dir.mkdir
|
||||
def _trap_url(*args, **kwargs):
|
||||
raise AssertionError("_locate_ffmpeg triggered a network download")
|
||||
monkeypatch.setattr('urllib.request.urlretrieve', _trap_url)
|
||||
|
||||
mkdir_calls = []
|
||||
real_mkdir = Path.mkdir
|
||||
|
||||
def _trap_mkdir(self, *args, **kwargs):
|
||||
if 'tools' in str(self):
|
||||
mkdir_calls.append(str(self))
|
||||
raise AssertionError(
|
||||
f"_locate_ffmpeg created tools dir: {self}"
|
||||
)
|
||||
return real_mkdir(self, *args, **kwargs)
|
||||
monkeypatch.setattr(Path, 'mkdir', _trap_mkdir)
|
||||
|
||||
client = YouTubeClient.__new__(YouTubeClient)
|
||||
result = client._locate_ffmpeg()
|
||||
|
||||
# Should return False (no ffmpeg anywhere) without raising.
|
||||
assert isinstance(result, bool)
|
||||
assert mkdir_calls == []
|
||||
Loading…
Reference in a new issue