Compare commits
17
Commits
v0.2.0
...
cdbe471166
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
cdbe471166 | ||
|
|
61e0031262 | ||
|
|
c99b423b72 | ||
|
|
9091c4d049 | ||
|
|
8228c81b5c | ||
|
|
d6ef70922c | ||
|
|
131c80f5de | ||
|
|
37b841f009 | ||
|
|
d5dce9c769 | ||
|
|
8f53a24e1c | ||
|
|
da78f7252c | ||
|
|
ece79515c0 | ||
|
|
46435feebd | ||
|
|
d5f67c6de5 | ||
|
|
8ec3e4c637 | ||
|
|
d6233c2995 | ||
|
|
3f50577de6 |
@@ -26,6 +26,9 @@ ENTRYPOINT ["music-mirror"]
|
|||||||
FROM runtime AS test
|
FROM runtime AS test
|
||||||
|
|
||||||
RUN pip install --no-cache-dir pytest
|
RUN pip install --no-cache-dir pytest
|
||||||
|
# sync-to-ipod.sh and its tests need these; the runtime image deliberately does
|
||||||
|
# not carry them, and neither does the base.
|
||||||
|
RUN apk add --no-cache bash rsync findmnt
|
||||||
COPY pytest.ini ./
|
COPY pytest.ini ./
|
||||||
# Host-side tools; not in the runtime image, but the suite covers them.
|
# Host-side tools; not in the runtime image, but the suite covers them.
|
||||||
COPY tools ./tools
|
COPY tools ./tools
|
||||||
|
|||||||
@@ -150,13 +150,130 @@ tools/sync-to-ipod.sh /mnt/tank/media/music-mp3 /media/IPOD/Music
|
|||||||
tools/sync-to-ipod.sh -n /mnt/tank/media/music-mp3 /media/IPOD/Music # dry run
|
tools/sync-to-ipod.sh -n /mnt/tank/media/music-mp3 /media/IPOD/Music # dry run
|
||||||
```
|
```
|
||||||
|
|
||||||
It refuses to start unless the destination is a mounted FAT filesystem that is
|
The destination is where the artist folders should end up — normally a
|
||||||
its own mount point, because `--delete` aimed at the wrong directory empties it
|
subdirectory such as `/media/IPOD/Music`, not the card root. A subdirectory is
|
||||||
and does not announce itself. It also excludes `/.rockbox`, the scrobbler logs
|
the better target: `--delete` is confined to it, and the device path budget is
|
||||||
|
derived from it rather than configured, so the two cannot disagree.
|
||||||
|
|
||||||
|
It refuses to start unless the destination is on a mounted FAT filesystem. That
|
||||||
|
check is also what catches an unmounted device — `/media/IPOD/Music` then
|
||||||
|
resolves to the host's own root filesystem, and this refuses to empty that.
|
||||||
|
`--help` says all of it. It also excludes `/.rockbox`, the scrobbler logs
|
||||||
and the various filesystem metadata directories from deletion — the mirror does
|
and the various filesystem metadata directories from deletion — the mirror does
|
||||||
not contain them, and without the exclusion a sync to the card root would
|
not contain them, and without the exclusion a sync to the card root would
|
||||||
remove the Rockbox install.
|
remove the Rockbox install.
|
||||||
|
|
||||||
|
Progress is a single line that rewrites itself:
|
||||||
|
|
||||||
|
```
|
||||||
|
[ 24%] 12,345/49,600 3.2 GiB/13.1 GiB 4.4 MiB/s ETA 38m12s King Gizzard / Petro…
|
||||||
|
```
|
||||||
|
|
||||||
|
The estimate comes from rsync's `%l`, which gives each file's size as it
|
||||||
|
completes. Bytes done over time elapsed is the same arithmetic rsync would do,
|
||||||
|
and needs nothing it does not already print. The rate is measured over a
|
||||||
|
trailing thirty seconds rather than the whole run, so it follows a device that
|
||||||
|
slows down instead of averaging the slowdown away — and it is suppressed
|
||||||
|
entirely for the first two seconds, where the window is microseconds wide and
|
||||||
|
would report gigabytes per second.
|
||||||
|
|
||||||
|
rsync says nothing at all while it builds its file list, which on fifty
|
||||||
|
thousand files over USB is minutes of apparent hang, and its own `progress2`
|
||||||
|
percentage is computed against a list it has not finished discovering. So the
|
||||||
|
script counts first — files and bytes both, a second pass over the tree, which
|
||||||
|
is what a percentage and an estimate that mean something cost — and renders the
|
||||||
|
rest itself. Piped to a log it prints a plain line every thirty seconds
|
||||||
|
instead, with no carriage returns, and a summary at the end either way.
|
||||||
|
|
||||||
|
### The Rockbox database
|
||||||
|
|
||||||
|
Point `MUSIC_MIRROR_DATABASE_TOOL` at Rockbox's host-side builder and the sync
|
||||||
|
rebuilds the database itself, so it never has to happen on the device.
|
||||||
|
|
||||||
|
```sh
|
||||||
|
git clone --depth 1 https://github.com/Rockbox/rockbox.git
|
||||||
|
cd rockbox && mkdir build-db && cd build-db
|
||||||
|
../tools/configure --target=ipodvideo --type=d && make -j$(nproc)
|
||||||
|
```
|
||||||
|
|
||||||
|
It needs a native compiler and SDL2 development headers, not the ARM
|
||||||
|
cross-toolchain, and `tools/configure` detects `__aarch64__` correctly. On a
|
||||||
|
distribution without `/usr/bin/perl` or `gcc-ar` — NixOS, say — patch the
|
||||||
|
shebangs in `tools/*.pl` and pass `AR=ar`.
|
||||||
|
|
||||||
|
Building it here rather than on the device is not merely faster. The on-device
|
||||||
|
commit sorts the whole index in whatever memory `core_alloc_maximum()` can
|
||||||
|
scrape together; on a fifty-thousand-track library it runs for hours or aborts
|
||||||
|
outright.
|
||||||
|
|
||||||
|
The scan runs against a scratch root — a real `.rockbox` beside a symlink
|
||||||
|
standing in for wherever the music lands on the device — so the paths recorded
|
||||||
|
are the ones Rockbox will look up, while the **bytes are read from the mirror
|
||||||
|
rather than over USB**. Only the dozen `.tcd` files cross to the card.
|
||||||
|
|
||||||
|
Cost, measured: the parser makes about 49 reads and 43 seeks per file, probing
|
||||||
|
the head for ID3v2 and the tail for ID3v1. On a local disk that is 2,000 files
|
||||||
|
in half a second. Over SMB, readahead absorbs most of the reads but the opens
|
||||||
|
and the head/tail split are real round trips, so a first full scan is minutes
|
||||||
|
rather than seconds. It is a one-time cost: the builder is incremental, and the
|
||||||
|
scratch root is kept between runs, so a later pass over unchanged files does no
|
||||||
|
metadata reads at all.
|
||||||
|
|
||||||
|
If minutes is still too many, run the builder where the mirror is local — on
|
||||||
|
the NAS — and copy the `.tcd` files across. There is nothing to parallelise:
|
||||||
|
the tool is single-threaded, and two instances cannot produce one database.
|
||||||
|
|
||||||
|
### If the sync is interrupted
|
||||||
|
|
||||||
|
No partially copied track is ever left under a name Rockbox would play. rsync
|
||||||
|
writes to a hidden temporary file and only renames it into place once the file
|
||||||
|
is complete, and *"by default, rsync will delete any partially transferred file
|
||||||
|
if the transfer is interrupted"*. `--partial` is deliberately not used, and
|
||||||
|
there is a test asserting it never will be.
|
||||||
|
|
||||||
|
After an unclean kill or a power cut a hidden `.track.mp3.XXXXXX` can survive.
|
||||||
|
It is not playable, it is not in the source, and the next run's `--delete`
|
||||||
|
removes it.
|
||||||
|
|
||||||
|
Interrupting does not, by itself, endanger the filesystem. The kernel flushes
|
||||||
|
dirty pages within `dirty_expire_centisecs` — thirty seconds by default — and
|
||||||
|
`umount` always syncs before it returns. Losing data needs you to interrupt,
|
||||||
|
*and* pull the card inside that window, *and* skip the unmount.
|
||||||
|
|
||||||
|
The script still traps `INT` and `TERM` and flushes and unmounts on the way
|
||||||
|
out, exiting 130. Not because a Ctrl-C is dangerous, but because it removes the
|
||||||
|
manual step and makes the exit deterministic — you get the same "safe to
|
||||||
|
disconnect" either way, rather than having to remember which path you took.
|
||||||
|
Both paths call the same function, so they cannot drift apart.
|
||||||
|
|
||||||
|
What genuinely does lose data is pulling the cable or the card without
|
||||||
|
unmounting at all, interrupted or not. FAT32 has no journal. Wait for the
|
||||||
|
unmount line.
|
||||||
|
|
||||||
|
### Making it faster over a network mount
|
||||||
|
|
||||||
|
The transfer is metadata-bound, not throughput-bound: 49,600 files means 49,600
|
||||||
|
round trips, and the counting pass doubles that. In rough order of what it is
|
||||||
|
worth doing:
|
||||||
|
|
||||||
|
| Lever | Why |
|
||||||
|
| ----- | --- |
|
||||||
|
| Mount the source with `actimeo=60,cache=loose` | SMB defaults to a **one second** attribute cache, so nearly every `stat` goes to the wire — twice, once per pass. This is the single biggest change and it is a mount option, not an rsync flag. |
|
||||||
|
| Put the card in a reader for the first load | USB 2.0 through an iPod in disk mode is the floor for the destination. No amount of source tuning gets past it. |
|
||||||
|
| `-Q` | Skips the counting pass entirely. Costs the percentage and the estimate, saves a whole walk of the tree. |
|
||||||
|
| `--whole-file`, `--omit-dir-times` | Already set. The first stops rsync checksumming destination files it is about to overwrite whole; the second drops a setattr per directory, 6,150 of them. |
|
||||||
|
|
||||||
|
**NFS instead of SMB** is worth trying but is not the big win it looks like.
|
||||||
|
Its attribute caching defaults are far more generous than SMB's — `acregmax` of
|
||||||
|
sixty seconds against `actimeo=1` — which is precisely the gap that
|
||||||
|
`actimeo=60` closes on the mount you already have. Bulk read throughput between
|
||||||
|
the two is much of a muchness on a gigabit link. Try the mount option first; it
|
||||||
|
is one line and needs no change on the NAS.
|
||||||
|
|
||||||
|
And if the destination is the iPod rather than a card reader, none of this
|
||||||
|
matters much: the source can feed data faster than USB 2.0 through an iPod will
|
||||||
|
take it either way.
|
||||||
|
|
||||||
The unmount is the point of doing this in a script. FAT32 has no journal and
|
The unmount is the point of doing this in a script. FAT32 has no journal and
|
||||||
the device is reached through disk mode, so an interrupted write is corruption
|
the device is reached through disk mode, so an interrupted write is corruption
|
||||||
that needs `fsck.vfat` from another machine.
|
that needs `fsck.vfat` from another machine.
|
||||||
@@ -229,6 +346,58 @@ producing files that already exist byte for byte. The run moves them instead,
|
|||||||
and says so. Prune then finds nothing to remove because nothing was left
|
and says so. Prune then finds nothing to remove because nothing was left
|
||||||
behind.
|
behind.
|
||||||
|
|
||||||
|
Renames are counted apart from encodes in the pass summary, and `--dry-run`
|
||||||
|
reports `would rename` rather than `would encode` — the difference between the
|
||||||
|
two is a minute against an afternoon, so a preview that conflated them would be
|
||||||
|
worse than no preview. A dry run also does not list the pre-rename files as
|
||||||
|
orphans: nothing was moved, so they are still there, but they are what a real
|
||||||
|
run would move rather than what it would delete.
|
||||||
|
|
||||||
|
### Path length
|
||||||
|
|
||||||
|
Rockbox's `MAX_PATH` is 260, from `firmware/include/fs_defines.h`, and it bounds
|
||||||
|
the path *as the device sees it*. The directory the mirror is copied into comes
|
||||||
|
out of the same budget, so `--device-prefix` (default `/Music`) is subtracted
|
||||||
|
from `--max-path` to get what a mirror-relative path may spend:
|
||||||
|
|
||||||
|
| Destination on the device | Mirror-relative budget |
|
||||||
|
| ------------------------- | ---------------------- |
|
||||||
|
| `/Music/` | 253 |
|
||||||
|
| the card root | 259 |
|
||||||
|
|
||||||
|
Worth being exact about, because a checker that measures mirror-relative paths
|
||||||
|
against the flat 260 quietly passes everything from 253 to 260 — and those are
|
||||||
|
the paths most likely to be near the edge in the first place.
|
||||||
|
|
||||||
|
Over-budget paths are shortened from the **deepest component outward**: the
|
||||||
|
track name carries the least navigational value and the artist directory the
|
||||||
|
most, so the filename goes first and the artist is touched only if nothing else
|
||||||
|
will do.
|
||||||
|
|
||||||
|
A component is cut **from the middle**, not the end, because of how these names
|
||||||
|
are built. Lidarr writes `Artist - Album - 07 - Flamethrower.mp3` inside a
|
||||||
|
directory already named for that artist and album, so a long album title
|
||||||
|
appears three times in one path and the informative part — the track number and
|
||||||
|
title — is at the very end. Cutting from the end throws exactly that away:
|
||||||
|
|
||||||
|
```
|
||||||
|
before King Gizzard & the Lizard Wizard - PetroDragonic Apocalypse; or, Dawn of Eternal
|
||||||
|
Night - An Annihilation of Planet Earth and the Beginning of Merciless
|
||||||
|
Damnation - 07 - Flamethrower.mp3
|
||||||
|
after King Gizzard & the Lizard~c526~ginning of Merciless Damnation - 07 - Flamethrower.mp3
|
||||||
|
```
|
||||||
|
|
||||||
|
Two thirds of the remaining room goes to the tail, since the head is usually a
|
||||||
|
restatement of the directory it sits in. A shortened component gains four hex
|
||||||
|
digits of the original name: two names sharing both a head and a tail would
|
||||||
|
otherwise produce the same string, and a silent collision between two tracks is
|
||||||
|
worse than an ugly filename.
|
||||||
|
|
||||||
|
The result is stable: the same source always produces the same shortened name,
|
||||||
|
so a pass does not rename what the previous pass wrote. A path too deeply
|
||||||
|
nested to fit without reducing every component to nonsense is left alone and
|
||||||
|
reported instead.
|
||||||
|
|
||||||
### Album art
|
### Album art
|
||||||
|
|
||||||
Rockbox looks for cover art **on the filesystem** — `cover.jpg`, `folder.jpg`
|
Rockbox looks for cover art **on the filesystem** — `cover.jpg`, `folder.jpg`
|
||||||
|
|||||||
+161
-13
@@ -17,6 +17,7 @@ import argparse
|
|||||||
import concurrent.futures
|
import concurrent.futures
|
||||||
import fcntl
|
import fcntl
|
||||||
import functools
|
import functools
|
||||||
|
import hashlib
|
||||||
import logging
|
import logging
|
||||||
import os
|
import os
|
||||||
import re
|
import re
|
||||||
@@ -88,6 +89,17 @@ MTIME_TOLERANCE_SECONDS = 2
|
|||||||
# never arrives -- "Kick Out the Epic Motherf**ker" is a real example.
|
# never arrives -- "Kick Out the Epic Motherf**ker" is a real example.
|
||||||
FAT32_RESERVED = re.compile(r'[<>:"/\\|?*\x00-\x1f]')
|
FAT32_RESERVED = re.compile(r'[<>:"/\\|?*\x00-\x1f]')
|
||||||
|
|
||||||
|
# Rockbox's MAX_PATH, from firmware/include/fs_defines.h. It bounds the whole
|
||||||
|
# path as the device sees it, so the budget for a mirror-relative path is this
|
||||||
|
# less whatever directory the mirror is copied into.
|
||||||
|
MAX_PATH = 260
|
||||||
|
DEVICE_PREFIX = "/Music"
|
||||||
|
|
||||||
|
# A component cut below this is no longer recognisable, and a path that cannot
|
||||||
|
# be brought under the limit without going there is better reported than
|
||||||
|
# mangled.
|
||||||
|
MIN_COMPONENT = 12
|
||||||
|
|
||||||
# The mirror exists to be read back by something else -- an SMB share, another
|
# The mirror exists to be read back by something else -- an SMB share, another
|
||||||
# account on the box -- so everything written into it has to be group-readable.
|
# account on the box -- so everything written into it has to be group-readable.
|
||||||
# Neither writer manages that unaided: tempfile.mkstemp forces 0600 whatever the
|
# Neither writer manages that unaided: tempfile.mkstemp forces 0600 whatever the
|
||||||
@@ -155,11 +167,83 @@ def fat32_safe(component):
|
|||||||
return cleaned or "_"
|
return cleaned or "_"
|
||||||
|
|
||||||
|
|
||||||
def mirror_path_for(source, source_root, mirror_root, safe=False):
|
def device_prefix_length(prefix):
|
||||||
|
"""Return the on-device prefix as it will actually appear, with slashes.
|
||||||
|
|
||||||
|
"/Music" costs seven characters -- the leading slash, the name, and the
|
||||||
|
separator before the mirror's own path -- while an empty prefix costs one.
|
||||||
|
Approximating that loses a character at the root, which is precisely where
|
||||||
|
the longest paths are.
|
||||||
|
"""
|
||||||
|
cleaned = prefix.strip("/")
|
||||||
|
return f"/{cleaned}/" if cleaned else "/"
|
||||||
|
|
||||||
|
|
||||||
|
def shorten_component(component, budget):
|
||||||
|
"""Return a component of at most `budget` characters, cut from the middle.
|
||||||
|
|
||||||
|
From the middle, not the end, because of how these names are built. Lidarr
|
||||||
|
writes "Artist - Album - 07 - Flamethrower.mp3" inside a directory already
|
||||||
|
named for that artist and album, so the informative part -- the track
|
||||||
|
number and title -- is at the very end. Cutting from the end discards it
|
||||||
|
and leaves every track on the record with the same name.
|
||||||
|
|
||||||
|
The four hex digits are of the original component. Two names sharing both a
|
||||||
|
head and a tail would otherwise produce the same string, and a silent
|
||||||
|
collision between two tracks is worse than an ugly filename.
|
||||||
|
"""
|
||||||
|
stem, dot, extension = component.rpartition(".")
|
||||||
|
if not dot or len(extension) > 4:
|
||||||
|
stem, extension = component, ""
|
||||||
|
else:
|
||||||
|
extension = dot + extension
|
||||||
|
|
||||||
|
digest = hashlib.blake2s(component.encode("utf-8"), digest_size=2).hexdigest()
|
||||||
|
marker = f"~{digest}~"
|
||||||
|
room = max(2, budget - len(extension) - len(marker))
|
||||||
|
if room >= len(stem):
|
||||||
|
return stem + extension
|
||||||
|
|
||||||
|
# Two thirds to the tail: the head is usually a restatement of the
|
||||||
|
# directory it sits in, and the tail is what tells two tracks apart.
|
||||||
|
keep_end = min(len(stem), room * 2 // 3)
|
||||||
|
keep_start = max(1, room - keep_end)
|
||||||
|
return stem[:keep_start].rstrip(". ") + marker + stem[len(stem) - keep_end :] + extension
|
||||||
|
|
||||||
|
|
||||||
|
def fit_path(relative, budget):
|
||||||
|
"""Return a relative path within `budget` characters, or the best available.
|
||||||
|
|
||||||
|
Shortened from the deepest component outward. The filename carries the least
|
||||||
|
navigational value and the artist directory the most, so the track name is
|
||||||
|
sacrificed before the album and the album before the artist.
|
||||||
|
"""
|
||||||
|
parts = list(relative.parts)
|
||||||
|
for index in reversed(range(len(parts))):
|
||||||
|
overage = len(str(Path(*parts))) - budget
|
||||||
|
if overage <= 0:
|
||||||
|
break
|
||||||
|
allowed = max(MIN_COMPONENT, len(parts[index]) - overage)
|
||||||
|
if allowed < len(parts[index]):
|
||||||
|
parts[index] = shorten_component(parts[index], allowed)
|
||||||
|
fitted = Path(*parts)
|
||||||
|
if len(str(fitted)) > budget:
|
||||||
|
logger.warning(
|
||||||
|
"%s is still %d characters over the limit after shortening; it is too"
|
||||||
|
" deeply nested to fit",
|
||||||
|
relative,
|
||||||
|
len(str(fitted)) - budget,
|
||||||
|
)
|
||||||
|
return fitted
|
||||||
|
|
||||||
|
|
||||||
|
def mirror_path_for(source, source_root, mirror_root, safe=False, budget=0):
|
||||||
"""Return the mirror path corresponding to a source file."""
|
"""Return the mirror path corresponding to a source file."""
|
||||||
relative = source.relative_to(source_root).with_suffix(MIRROR_SUFFIX)
|
relative = source.relative_to(source_root).with_suffix(MIRROR_SUFFIX)
|
||||||
if safe:
|
if safe:
|
||||||
relative = Path(*(fat32_safe(part) for part in relative.parts))
|
relative = Path(*(fat32_safe(part) for part in relative.parts))
|
||||||
|
if budget > 0 and len(str(relative)) > budget:
|
||||||
|
relative = fit_path(relative, budget)
|
||||||
return mirror_root / relative
|
return mirror_root / relative
|
||||||
|
|
||||||
|
|
||||||
@@ -357,26 +441,39 @@ def copy(source, mirror, dry_run):
|
|||||||
return Result("copied", mirror)
|
return Result("copied", mirror)
|
||||||
|
|
||||||
|
|
||||||
def adopt_existing(source, mirror, previous):
|
def adopt_existing(source, mirror, candidates, dry_run=False):
|
||||||
"""Move an already-encoded file to its new name. Returns whether it moved.
|
"""Move an already-encoded file to its new name. Returns whether it moved.
|
||||||
|
|
||||||
Turning on FAT32-safe naming changes the path of every track whose name
|
Turning on FAT32-safe naming changes the path of every track whose name
|
||||||
held a reserved character. Without this the run would encode them all again
|
held a reserved character. Without this the run would encode them all again
|
||||||
and then prune the originals -- hours of work to produce files that already
|
and then prune the originals -- hours of work to produce files that already
|
||||||
exist, byte for byte, under the old name.
|
exist, byte for byte, under the old name.
|
||||||
|
|
||||||
|
Several candidates are tried because there is more than one previous
|
||||||
|
naming: the original, and the sanitised-but-not-yet-shortened form left by
|
||||||
|
an earlier version.
|
||||||
"""
|
"""
|
||||||
|
for previous in candidates:
|
||||||
if previous == mirror or not previous.is_file() or not is_current(source, previous):
|
if previous == mirror or not previous.is_file() or not is_current(source, previous):
|
||||||
return False
|
continue
|
||||||
|
if dry_run:
|
||||||
|
logger.info("would rename %s -> %s", previous.name, mirror.name)
|
||||||
|
return True
|
||||||
mirror.parent.mkdir(parents=True, exist_ok=True)
|
mirror.parent.mkdir(parents=True, exist_ok=True)
|
||||||
os.replace(previous, mirror)
|
os.replace(previous, mirror)
|
||||||
logger.info("renamed %s -> %s", previous.name, mirror.name)
|
logger.info("renamed %s -> %s", previous.name, mirror.name)
|
||||||
return True
|
return True
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
def process(source, mirror, quality_args, dry_run, previous=None):
|
def process(source, mirror, quality_args, dry_run, previous=None):
|
||||||
"""Bring one source file's mirror entry up to date."""
|
"""Bring one source file's mirror entry up to date."""
|
||||||
if previous is not None and not dry_run and not mirror.exists():
|
# Counted separately from an encode, and reported in a dry run, because the
|
||||||
adopt_existing(source, mirror, previous)
|
# difference between moving a file and re-encoding it is the difference
|
||||||
|
# between a minute and an afternoon.
|
||||||
|
if previous is not None and not mirror.exists():
|
||||||
|
if adopt_existing(source, mirror, previous, dry_run):
|
||||||
|
return Result("renamed", mirror)
|
||||||
if is_current(source, mirror):
|
if is_current(source, mirror):
|
||||||
# A mirror written before this bit was set has a correct mtime, so
|
# A mirror written before this bit was set has a correct mtime, so
|
||||||
# nothing else in the pass would ever revisit it. Top it up here
|
# nothing else in the pass would ever revisit it. Top it up here
|
||||||
@@ -400,7 +497,7 @@ def find_sources(root):
|
|||||||
yield path
|
yield path
|
||||||
|
|
||||||
|
|
||||||
def plan(scan_root, source_root, mirror_root, safe=False):
|
def plan(scan_root, source_root, mirror_root, safe=False, budget=0):
|
||||||
"""Map each mirror path to the one source that should produce it.
|
"""Map each mirror path to the one source that should produce it.
|
||||||
|
|
||||||
Two sources can want the same mirror path -- `01 Song.flac` alongside a
|
Two sources can want the same mirror path -- `01 Song.flac` alongside a
|
||||||
@@ -416,7 +513,7 @@ def plan(scan_root, source_root, mirror_root, safe=False):
|
|||||||
# that now beats discovering it as a silent overwrite during the copy.
|
# that now beats discovering it as a silent overwrite during the copy.
|
||||||
seen = {}
|
seen = {}
|
||||||
for source in find_sources(scan_root):
|
for source in find_sources(scan_root):
|
||||||
mirror = mirror_path_for(source, source_root, mirror_root, safe)
|
mirror = mirror_path_for(source, source_root, mirror_root, safe, budget)
|
||||||
key = str(mirror).casefold() if safe else str(mirror)
|
key = str(mirror).casefold() if safe else str(mirror)
|
||||||
rival_path = seen.get(key)
|
rival_path = seen.get(key)
|
||||||
rival = chosen.get(rival_path) if rival_path else None
|
rival = chosen.get(rival_path) if rival_path else None
|
||||||
@@ -474,7 +571,15 @@ def prune(mirror_root, expected, dry_run):
|
|||||||
|
|
||||||
|
|
||||||
def run_once(
|
def run_once(
|
||||||
scan_root, source_root, mirror_root, quality_args, jobs, dry_run, do_prune, safe=False
|
scan_root,
|
||||||
|
source_root,
|
||||||
|
mirror_root,
|
||||||
|
quality_args,
|
||||||
|
jobs,
|
||||||
|
dry_run,
|
||||||
|
do_prune,
|
||||||
|
safe=False,
|
||||||
|
budget=0,
|
||||||
):
|
):
|
||||||
"""Run a single pass. Returns the number of failures.
|
"""Run a single pass. Returns the number of failures.
|
||||||
|
|
||||||
@@ -483,10 +588,10 @@ def run_once(
|
|||||||
"""
|
"""
|
||||||
started = time.monotonic()
|
started = time.monotonic()
|
||||||
logger.info("pass starting with %d concurrent encoders", jobs)
|
logger.info("pass starting with %d concurrent encoders", jobs)
|
||||||
counts = {"encoded": 0, "copied": 0, "skipped": 0, "failed": 0}
|
counts = {"encoded": 0, "copied": 0, "renamed": 0, "skipped": 0, "failed": 0}
|
||||||
failures = []
|
failures = []
|
||||||
|
|
||||||
work = plan(scan_root, source_root, mirror_root, safe)
|
work = plan(scan_root, source_root, mirror_root, safe, budget)
|
||||||
|
|
||||||
with concurrent.futures.ThreadPoolExecutor(max_workers=jobs) as pool:
|
with concurrent.futures.ThreadPoolExecutor(max_workers=jobs) as pool:
|
||||||
futures = [
|
futures = [
|
||||||
@@ -496,7 +601,14 @@ def run_once(
|
|||||||
mirror,
|
mirror,
|
||||||
quality_args,
|
quality_args,
|
||||||
dry_run,
|
dry_run,
|
||||||
mirror_path_for(source, source_root, mirror_root) if safe else None,
|
(
|
||||||
|
[
|
||||||
|
mirror_path_for(source, source_root, mirror_root),
|
||||||
|
mirror_path_for(source, source_root, mirror_root, True),
|
||||||
|
]
|
||||||
|
if safe
|
||||||
|
else None
|
||||||
|
),
|
||||||
)
|
)
|
||||||
for mirror, source in work.items()
|
for mirror, source in work.items()
|
||||||
]
|
]
|
||||||
@@ -506,16 +618,27 @@ def run_once(
|
|||||||
if result.action == "failed":
|
if result.action == "failed":
|
||||||
failures.append(result)
|
failures.append(result)
|
||||||
|
|
||||||
removed = prune(mirror_root, set(work), dry_run) if do_prune else 0
|
expected = set(work)
|
||||||
|
if safe and dry_run:
|
||||||
|
# Nothing was actually renamed, so the pre-sanitisation files are still
|
||||||
|
# on disk. They are not orphans -- they are the files a real run would
|
||||||
|
# move -- and reporting them for deletion would misrepresent the pass
|
||||||
|
# twice over.
|
||||||
|
for source in work.values():
|
||||||
|
expected.add(mirror_path_for(source, source_root, mirror_root))
|
||||||
|
expected.add(mirror_path_for(source, source_root, mirror_root, True))
|
||||||
|
removed = prune(mirror_root, expected, dry_run) if do_prune else 0
|
||||||
|
|
||||||
for failure in failures:
|
for failure in failures:
|
||||||
logger.error("failed: %s: %s", failure.path, failure.error)
|
logger.error("failed: %s: %s", failure.path, failure.error)
|
||||||
|
|
||||||
logger.info(
|
logger.info(
|
||||||
"pass complete in %.1fs: %d encoded, %d copied, %d up to date, %d removed, %d failed",
|
"pass complete in %.1fs: %d encoded, %d copied, %d renamed, %d up to date,"
|
||||||
|
" %d removed, %d failed",
|
||||||
time.monotonic() - started,
|
time.monotonic() - started,
|
||||||
counts["encoded"],
|
counts["encoded"],
|
||||||
counts["copied"],
|
counts["copied"],
|
||||||
|
counts["renamed"],
|
||||||
counts["skipped"],
|
counts["skipped"],
|
||||||
removed,
|
removed,
|
||||||
counts["failed"],
|
counts["failed"],
|
||||||
@@ -599,6 +722,19 @@ def build_parser():
|
|||||||
help="name mirror files so a FAT32 device will accept them"
|
help="name mirror files so a FAT32 device will accept them"
|
||||||
" (env MUSIC_MIRROR_FAT32_SAFE)",
|
" (env MUSIC_MIRROR_FAT32_SAFE)",
|
||||||
)
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
"--max-path",
|
||||||
|
type=int,
|
||||||
|
default=int(os.getenv("MUSIC_MIRROR_MAX_PATH", str(MAX_PATH))),
|
||||||
|
help=f"longest path the device will take, counted from its root; Rockbox's"
|
||||||
|
f" MAX_PATH is {MAX_PATH} (env MUSIC_MIRROR_MAX_PATH)",
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
"--device-prefix",
|
||||||
|
default=os.getenv("MUSIC_MIRROR_DEVICE_PREFIX", DEVICE_PREFIX),
|
||||||
|
help="directory the mirror is copied into on the device, whose length comes"
|
||||||
|
" out of the path budget (env MUSIC_MIRROR_DEVICE_PREFIX)",
|
||||||
|
)
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--no-prune",
|
"--no-prune",
|
||||||
action="store_true",
|
action="store_true",
|
||||||
@@ -657,6 +793,17 @@ def main(argv=None):
|
|||||||
# A partial pass cannot tell an orphan from a file outside its scope.
|
# A partial pass cannot tell an orphan from a file outside its scope.
|
||||||
do_prune = False
|
do_prune = False
|
||||||
|
|
||||||
|
# The device's limit covers the whole path it will see, so what the mirror
|
||||||
|
# may spend is that less the directory it gets copied into.
|
||||||
|
budget = max(0, args.max_path - len(device_prefix_length(args.device_prefix)))
|
||||||
|
if args.fat32_safe:
|
||||||
|
logger.info(
|
||||||
|
"paths are limited to %d characters, from --max-path %d less the %r prefix",
|
||||||
|
budget,
|
||||||
|
args.max_path,
|
||||||
|
args.device_prefix,
|
||||||
|
)
|
||||||
|
|
||||||
lock = acquire_lock(mirror_root)
|
lock = acquire_lock(mirror_root)
|
||||||
if lock is None:
|
if lock is None:
|
||||||
logger.error("another pass is already running over %s", mirror_root)
|
logger.error("another pass is already running over %s", mirror_root)
|
||||||
@@ -683,6 +830,7 @@ def main(argv=None):
|
|||||||
args.dry_run,
|
args.dry_run,
|
||||||
do_prune,
|
do_prune,
|
||||||
args.fat32_safe,
|
args.fat32_safe,
|
||||||
|
budget,
|
||||||
)
|
)
|
||||||
if interval is None or stopping:
|
if interval is None or stopping:
|
||||||
return 1 if failures else 0
|
return 1 if failures else 0
|
||||||
|
|||||||
+1
-1
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
|
|||||||
|
|
||||||
[project]
|
[project]
|
||||||
name = "music-mirror"
|
name = "music-mirror"
|
||||||
version = "0.2.0"
|
version = "0.4.0"
|
||||||
description = "Maintain a lossy MP3 mirror of a lossless music library"
|
description = "Maintain a lossy MP3 mirror of a lossless music library"
|
||||||
readme = "README.md"
|
readme = "README.md"
|
||||||
requires-python = ">=3.11"
|
requires-python = ">=3.11"
|
||||||
|
|||||||
@@ -558,3 +558,231 @@ def test_a_cover_left_without_tracks_is_pruned(tmp_path, make_flac, make_cover):
|
|||||||
run(source, mirror)
|
run(source, mirror)
|
||||||
|
|
||||||
assert not (mirror / "Gone").exists()
|
assert not (mirror / "Gone").exists()
|
||||||
|
|
||||||
|
|
||||||
|
def test_a_dry_run_reports_a_rename_not_an_encode(tmp_path, make_flac, caplog):
|
||||||
|
"""The difference between moving a file and re-encoding it is the
|
||||||
|
difference between a minute and an afternoon, so a dry run must not
|
||||||
|
describe the first as the second."""
|
||||||
|
source = tmp_path / "src"
|
||||||
|
mirror = tmp_path / "dst"
|
||||||
|
make_flac(source / "Album" / "Where Are You?.flac")
|
||||||
|
run(source, mirror)
|
||||||
|
|
||||||
|
with caplog.at_level("INFO"):
|
||||||
|
run(source, mirror, "--fat32-safe", "--dry-run")
|
||||||
|
|
||||||
|
assert "would rename" in caplog.text
|
||||||
|
assert "would encode" not in caplog.text
|
||||||
|
|
||||||
|
|
||||||
|
def test_a_dry_run_does_not_call_the_old_paths_orphans(tmp_path, make_flac, caplog):
|
||||||
|
"""Nothing was renamed, so they are still there -- but they are the files a
|
||||||
|
real run would move, not files it would delete."""
|
||||||
|
source = tmp_path / "src"
|
||||||
|
mirror = tmp_path / "dst"
|
||||||
|
make_flac(source / "Album" / "Where Are You?.flac")
|
||||||
|
run(source, mirror)
|
||||||
|
|
||||||
|
with caplog.at_level("INFO"):
|
||||||
|
run(source, mirror, "--fat32-safe", "--dry-run")
|
||||||
|
|
||||||
|
assert "would remove orphan" not in caplog.text
|
||||||
|
|
||||||
|
|
||||||
|
def test_a_dry_run_moves_nothing(tmp_path, make_flac):
|
||||||
|
source = tmp_path / "src"
|
||||||
|
mirror = tmp_path / "dst"
|
||||||
|
make_flac(source / "Album" / "Where Are You?.flac")
|
||||||
|
run(source, mirror)
|
||||||
|
|
||||||
|
run(source, mirror, "--fat32-safe", "--dry-run")
|
||||||
|
|
||||||
|
assert (mirror / "Album" / "Where Are You?.mp3").is_file()
|
||||||
|
assert not (mirror / "Album" / "Where Are You_.mp3").exists()
|
||||||
|
|
||||||
|
|
||||||
|
def test_renames_are_counted_separately_from_encodes(tmp_path, make_flac, caplog):
|
||||||
|
source = tmp_path / "src"
|
||||||
|
mirror = tmp_path / "dst"
|
||||||
|
make_flac(source / "Album" / "Where Are You?.flac")
|
||||||
|
run(source, mirror)
|
||||||
|
|
||||||
|
with caplog.at_level("INFO"):
|
||||||
|
run(source, mirror, "--fat32-safe")
|
||||||
|
|
||||||
|
assert "1 renamed" in caplog.text
|
||||||
|
assert "0 encoded" in caplog.text
|
||||||
|
|
||||||
|
|
||||||
|
def test_a_long_path_is_shortened_from_the_deepest_component(tmp_path, make_flac):
|
||||||
|
"""The track name carries the least navigational value and the artist the
|
||||||
|
most, so the filename is sacrificed before the album."""
|
||||||
|
artist = "A" * 60
|
||||||
|
album = "B" * 60
|
||||||
|
title = "C" * 150
|
||||||
|
source = tmp_path / "src"
|
||||||
|
mirror = tmp_path / "dst"
|
||||||
|
make_flac(source / artist / album / f"{title}.flac")
|
||||||
|
|
||||||
|
run(source, mirror, "--fat32-safe", "--max-path", "160", "--device-prefix", "/Music")
|
||||||
|
|
||||||
|
written = list(mirror.rglob("*.mp3"))
|
||||||
|
assert len(written) == 1
|
||||||
|
relative = written[0].relative_to(mirror)
|
||||||
|
assert relative.parts[0] == artist, "the artist directory should be untouched"
|
||||||
|
assert relative.parts[1] == album, "the album directory should be untouched"
|
||||||
|
assert len(str(relative)) <= 160 - len("Music") - 2
|
||||||
|
|
||||||
|
|
||||||
|
def test_shortening_is_stable_across_passes(tmp_path, make_flac):
|
||||||
|
"""An unstable name would rename every file on every pass, for ever."""
|
||||||
|
source = tmp_path / "src"
|
||||||
|
mirror = tmp_path / "dst"
|
||||||
|
make_flac(source / ("D" * 80) / ("E" * 80) / f"{'F' * 120}.flac")
|
||||||
|
|
||||||
|
run(source, mirror, "--fat32-safe", "--max-path", "180")
|
||||||
|
first = sorted(str(p.relative_to(mirror)) for p in mirror.rglob("*.mp3"))
|
||||||
|
stamp = next(mirror.rglob("*.mp3")).stat().st_mtime_ns
|
||||||
|
|
||||||
|
run(source, mirror, "--fat32-safe", "--max-path", "180")
|
||||||
|
|
||||||
|
assert sorted(str(p.relative_to(mirror)) for p in mirror.rglob("*.mp3")) == first
|
||||||
|
assert next(mirror.rglob("*.mp3")).stat().st_mtime_ns == stamp
|
||||||
|
|
||||||
|
|
||||||
|
def test_two_long_names_do_not_collide_after_shortening(tmp_path, make_flac):
|
||||||
|
"""They share a prefix and cut to the same string; the hash is what keeps
|
||||||
|
them apart."""
|
||||||
|
source = tmp_path / "src"
|
||||||
|
mirror = tmp_path / "dst"
|
||||||
|
shared = "G" * 140
|
||||||
|
make_flac(source / "Album" / f"{shared}one.flac")
|
||||||
|
make_flac(source / "Album" / f"{shared}two.flac")
|
||||||
|
|
||||||
|
run(source, mirror, "--fat32-safe", "--max-path", "120")
|
||||||
|
|
||||||
|
assert len(list(mirror.rglob("*.mp3"))) == 2
|
||||||
|
|
||||||
|
|
||||||
|
def test_a_sanitised_mirror_is_renamed_rather_than_re_encoded_when_shortening(
|
||||||
|
tmp_path, make_flac
|
||||||
|
):
|
||||||
|
"""The previous naming is sanitised-but-not-shortened, not the original."""
|
||||||
|
source = tmp_path / "src"
|
||||||
|
mirror = tmp_path / "dst"
|
||||||
|
make_flac(source / "Album" / f"Where Are You? {'H' * 140}.flac")
|
||||||
|
|
||||||
|
run(source, mirror, "--fat32-safe", "--max-path", "400")
|
||||||
|
before = next(mirror.rglob("*.mp3"))
|
||||||
|
contents = before.read_bytes()
|
||||||
|
|
||||||
|
run(source, mirror, "--fat32-safe", "--max-path", "120")
|
||||||
|
|
||||||
|
after = next(mirror.rglob("*.mp3"))
|
||||||
|
assert after != before
|
||||||
|
assert after.read_bytes() == contents, "it was re-encoded rather than moved"
|
||||||
|
|
||||||
|
|
||||||
|
def test_shortening_only_applies_when_over_budget(tmp_path, make_flac):
|
||||||
|
source = tmp_path / "src"
|
||||||
|
mirror = tmp_path / "dst"
|
||||||
|
make_flac(source / "Artist" / "Album" / "Short Name.flac")
|
||||||
|
|
||||||
|
run(source, mirror, "--fat32-safe")
|
||||||
|
|
||||||
|
assert (mirror / "Artist" / "Album" / "Short Name.mp3").is_file()
|
||||||
|
|
||||||
|
|
||||||
|
def test_a_path_that_cannot_be_made_to_fit_is_reported(tmp_path, make_flac, caplog):
|
||||||
|
"""Too deeply nested to shorten without making every component unreadable."""
|
||||||
|
deep = Path(*["I" * 20 for _ in range(10)])
|
||||||
|
source = tmp_path / "src"
|
||||||
|
mirror = tmp_path / "dst"
|
||||||
|
make_flac(source / deep / "track.flac")
|
||||||
|
|
||||||
|
with caplog.at_level("WARNING"):
|
||||||
|
run(source, mirror, "--fat32-safe", "--max-path", "80")
|
||||||
|
|
||||||
|
assert "too" in caplog.text and "nested" in caplog.text
|
||||||
|
|
||||||
|
|
||||||
|
# Lidarr writes "Artist - Album - NN - Title.mp3" inside a directory already
|
||||||
|
# named for that artist and album, so a long album title appears three times in
|
||||||
|
# one path. These are real paths from a real library.
|
||||||
|
GIZZARD_ALBUM = (
|
||||||
|
"PetroDragonic Apocalypse; or, Dawn of Eternal Night - An Annihilation of"
|
||||||
|
" Planet Earth and the Beginning of Merciless Damnation"
|
||||||
|
)
|
||||||
|
GIZZARD_TRACKS = (
|
||||||
|
"01 - Motor Spirit",
|
||||||
|
"02 - Supercell",
|
||||||
|
"03 - Converge",
|
||||||
|
"04 - Witchcraft",
|
||||||
|
"05 - Gila Monster",
|
||||||
|
"06 - Dragon",
|
||||||
|
"07 - Flamethrower",
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def gizzard_path(track):
|
||||||
|
return Path(
|
||||||
|
"King Gizzard & the Lizard Wizard",
|
||||||
|
f"{GIZZARD_ALBUM} (2023)",
|
||||||
|
f"King Gizzard & the Lizard Wizard - {GIZZARD_ALBUM} - {track}.mp3",
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def test_shortening_keeps_the_part_that_tells_tracks_apart():
|
||||||
|
"""Cutting from the end discards the track number and title, which is all
|
||||||
|
that distinguishes one track on the record from another."""
|
||||||
|
for track in GIZZARD_TRACKS:
|
||||||
|
fitted = music_mirror.fit_path(gizzard_path(track), 253)
|
||||||
|
assert len(str(fitted)) <= 253
|
||||||
|
assert fitted.name.endswith(f"{track}.mp3"), fitted.name
|
||||||
|
|
||||||
|
|
||||||
|
def test_every_track_on_a_long_album_keeps_a_distinct_name():
|
||||||
|
fitted = {music_mirror.fit_path(gizzard_path(t), 253).name for t in GIZZARD_TRACKS}
|
||||||
|
|
||||||
|
assert len(fitted) == len(GIZZARD_TRACKS)
|
||||||
|
|
||||||
|
|
||||||
|
def test_a_long_title_keeps_both_ends():
|
||||||
|
"""The Beatles' 'The Long One' is one track whose title is the long part."""
|
||||||
|
path = Path(
|
||||||
|
"The Beatles",
|
||||||
|
"Abbey Road (1969)",
|
||||||
|
"Digital Media 03",
|
||||||
|
"The Beatles - Abbey Road - 09 - The Long One - You Never Give Me Your Money"
|
||||||
|
" + Sun King + Mean Mr Mustard + Her Majesty + Polythene Pam + She Came In"
|
||||||
|
" Through the Bathroom Window+ Golden Slumbers + Carry That Weight + The End.mp3",
|
||||||
|
)
|
||||||
|
|
||||||
|
fitted = music_mirror.fit_path(path, 253)
|
||||||
|
|
||||||
|
assert len(str(fitted)) <= 253
|
||||||
|
assert fitted.name.startswith("The Beatles - Abbey Road - 09 - The Long One")
|
||||||
|
assert fitted.name.endswith("The End.mp3")
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
("prefix", "expected"),
|
||||||
|
[("/Music", 253), ("Music", 253), ("/Music/", 253), ("", 259), ("/", 259)],
|
||||||
|
)
|
||||||
|
def test_the_device_prefix_is_costed_exactly(prefix, expected):
|
||||||
|
"""A mirror-relative path of 253 characters becomes 260 on the device once
|
||||||
|
/Music/ is in front of it, which is the whole of the limit. Approximating
|
||||||
|
the prefix loses a character at the root, where the longest paths are."""
|
||||||
|
assert 260 - len(music_mirror.device_prefix_length(prefix)) == expected
|
||||||
|
|
||||||
|
|
||||||
|
def test_the_budget_is_reported_so_it_can_be_checked(tmp_path, make_flac, caplog):
|
||||||
|
source = tmp_path / "src"
|
||||||
|
mirror = tmp_path / "dst"
|
||||||
|
make_flac(source / "a.flac")
|
||||||
|
|
||||||
|
with caplog.at_level("INFO"):
|
||||||
|
run(source, mirror, "--fat32-safe")
|
||||||
|
|
||||||
|
assert "limited to 253 characters" in caplog.text
|
||||||
|
|||||||
@@ -0,0 +1,180 @@
|
|||||||
|
import io
|
||||||
|
import sys
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
sys.path.insert(0, str(Path(__file__).resolve().parent.parent / "tools"))
|
||||||
|
|
||||||
|
import rsync_progress # noqa: E402
|
||||||
|
|
||||||
|
|
||||||
|
class NotATerminal(io.StringIO):
|
||||||
|
def isatty(self):
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
class Terminal(io.StringIO):
|
||||||
|
def isatty(self):
|
||||||
|
return True
|
||||||
|
|
||||||
|
|
||||||
|
def run(lines, total=0, out=None, bytes_expected=0):
|
||||||
|
out = out or NotATerminal()
|
||||||
|
rsync_progress.main(
|
||||||
|
["--total", str(total), "--bytes", str(bytes_expected)],
|
||||||
|
stream=io.StringIO(lines),
|
||||||
|
out=out,
|
||||||
|
)
|
||||||
|
return out.getvalue()
|
||||||
|
|
||||||
|
|
||||||
|
def test_the_artist_and_album_are_pulled_from_the_path():
|
||||||
|
assert rsync_progress.album_of("Pendulum/Immersion/01 - Watercolour.mp3") == (
|
||||||
|
"Pendulum / Immersion"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def test_a_shallower_path_degrades_rather_than_failing():
|
||||||
|
assert rsync_progress.album_of("Pendulum/loose.mp3") == "Pendulum"
|
||||||
|
assert rsync_progress.album_of("loose.mp3") == ""
|
||||||
|
|
||||||
|
|
||||||
|
def test_directories_are_not_counted():
|
||||||
|
"""rsync reports them too, and counting them puts the percentage past 100."""
|
||||||
|
output = run("Artist/\nArtist/Album/\nArtist/Album/track.mp3\n", total=1)
|
||||||
|
|
||||||
|
assert "1/1" in output
|
||||||
|
assert "100%" in output
|
||||||
|
|
||||||
|
|
||||||
|
def test_the_percentage_tracks_the_total():
|
||||||
|
output = run("".join(f"A/B/{i}.mp3\n" for i in range(5)), total=10)
|
||||||
|
|
||||||
|
assert "5/10" in output
|
||||||
|
assert "50%" in output
|
||||||
|
|
||||||
|
|
||||||
|
def test_without_a_total_it_counts_instead_of_guessing():
|
||||||
|
output = run("A/B/one.mp3\nA/B/two.mp3\n")
|
||||||
|
|
||||||
|
assert "2 files" in output
|
||||||
|
assert "%" not in output
|
||||||
|
|
||||||
|
|
||||||
|
def test_a_final_line_is_always_printed():
|
||||||
|
"""Otherwise the last state of a rewriting line is whatever it happened to
|
||||||
|
be when the interval last elapsed."""
|
||||||
|
output = run("A/B/one.mp3\n", total=1)
|
||||||
|
|
||||||
|
assert output.endswith("\n")
|
||||||
|
assert "1/1" in output
|
||||||
|
|
||||||
|
|
||||||
|
def test_nothing_transferred_still_reports():
|
||||||
|
output = run("", total=0)
|
||||||
|
|
||||||
|
assert "0 files" in output
|
||||||
|
|
||||||
|
|
||||||
|
def test_a_log_gets_no_carriage_returns():
|
||||||
|
"""A non-terminal filling with \\r and escape codes is unreadable."""
|
||||||
|
output = run("".join(f"A/B/{i}.mp3\n" for i in range(50)), total=50)
|
||||||
|
|
||||||
|
assert "\r" not in output
|
||||||
|
assert "\033" not in output
|
||||||
|
|
||||||
|
|
||||||
|
def test_a_terminal_rewrites_one_line():
|
||||||
|
output = run("".join(f"A/B/{i}.mp3\n" for i in range(50)), total=50, out=Terminal())
|
||||||
|
|
||||||
|
assert "\r\033[2K" in output
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
("text", "width", "expected"),
|
||||||
|
[
|
||||||
|
("short", 20, "short"),
|
||||||
|
("King Gizzard / PetroDragonic Apocalypse", 20, "…agonic Apocalypse"),
|
||||||
|
],
|
||||||
|
)
|
||||||
|
def test_long_labels_are_trimmed_from_the_left(text, width, expected):
|
||||||
|
"""The album is the informative end, so the artist is what gets cut."""
|
||||||
|
trimmed = rsync_progress.fit(text, width)
|
||||||
|
|
||||||
|
assert len(trimmed) <= width
|
||||||
|
if len(text) > width:
|
||||||
|
assert trimmed.startswith("…")
|
||||||
|
assert text.endswith(trimmed.lstrip("…"))
|
||||||
|
|
||||||
|
|
||||||
|
def test_the_size_and_path_are_parsed():
|
||||||
|
assert rsync_progress.parse("5000 Artist/Album/Track.mp3\n") == (
|
||||||
|
5000,
|
||||||
|
"Artist/Album/Track.mp3",
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def test_a_filename_containing_spaces_survives():
|
||||||
|
"""Splitting on every space would lose most of the library."""
|
||||||
|
assert rsync_progress.parse("1234 Artist/An Album/A Track With Spaces.mp3") == (
|
||||||
|
1234,
|
||||||
|
"Artist/An Album/A Track With Spaces.mp3",
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def test_a_bare_path_is_tolerated():
|
||||||
|
"""In case this is fed --out-format='%n' by something older."""
|
||||||
|
assert rsync_progress.parse("Artist/Album/Track.mp3") == (0, "Artist/Album/Track.mp3")
|
||||||
|
|
||||||
|
|
||||||
|
def test_directory_sizes_do_not_inflate_the_total():
|
||||||
|
"""rsync reports directories with a 4096 inode size, which is several
|
||||||
|
megabytes of nothing across six thousand albums."""
|
||||||
|
output = run("4096 Artist/\n4096 Artist/Album/\n5000 Artist/Album/t.mp3\n", total=1)
|
||||||
|
|
||||||
|
assert "4.9 KiB" in output
|
||||||
|
assert "12" not in output.split("Artist")[0]
|
||||||
|
|
||||||
|
|
||||||
|
def test_a_rate_is_not_reported_until_it_means_something():
|
||||||
|
"""The first files arrive microseconds apart and would give a rate in the
|
||||||
|
gigabytes per second and an ETA of zero."""
|
||||||
|
rate = rsync_progress.Rate()
|
||||||
|
rate.add(100.0, 0)
|
||||||
|
rate.add(100.5, 5_000_000)
|
||||||
|
|
||||||
|
assert rate.per_second() == 0.0
|
||||||
|
|
||||||
|
|
||||||
|
def test_a_rate_over_a_long_enough_window_is_reported():
|
||||||
|
rate = rsync_progress.Rate()
|
||||||
|
rate.add(100.0, 0)
|
||||||
|
rate.add(110.0, 10_000_000)
|
||||||
|
|
||||||
|
assert rate.per_second() == pytest.approx(1_000_000)
|
||||||
|
|
||||||
|
|
||||||
|
def test_the_window_forgets_the_distant_past():
|
||||||
|
"""So the estimate follows a device that slows down rather than averaging
|
||||||
|
the slowdown away."""
|
||||||
|
rate = rsync_progress.Rate(window=30.0)
|
||||||
|
for second in range(0, 100, 10):
|
||||||
|
rate.add(float(second), second * 1_000_000)
|
||||||
|
rate.add(200.0, 100_000_000)
|
||||||
|
|
||||||
|
assert rate.samples[0][0] >= 90.0
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
("seconds", "expected"),
|
||||||
|
[(0, "0s"), (45, "45s"), (60, "1m00s"), (1092, "18m12s"), (7500, "2h05m")],
|
||||||
|
)
|
||||||
|
def test_durations_read_without_arithmetic(seconds, expected):
|
||||||
|
assert rsync_progress.human_duration(seconds) == expected
|
||||||
|
|
||||||
|
|
||||||
|
def test_a_summary_is_printed_at_the_end():
|
||||||
|
output = run("5000000 A/B/one.mp3\n", total=1, bytes_expected=5000000)
|
||||||
|
|
||||||
|
assert "copied 4.8 MiB in" in output
|
||||||
@@ -0,0 +1,273 @@
|
|||||||
|
"""The guards on sync-to-ipod.sh, which are the substance of the script.
|
||||||
|
|
||||||
|
rsync --delete is being aimed at a whole filesystem, so every refusal here is
|
||||||
|
protecting against emptying the wrong directory -- a mistake that does not
|
||||||
|
announce itself.
|
||||||
|
"""
|
||||||
|
|
||||||
|
import shutil
|
||||||
|
import subprocess
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
SCRIPT = Path(__file__).resolve().parent.parent / "tools" / "sync-to-ipod.sh"
|
||||||
|
|
||||||
|
# Skipped rather than failed where the tools are absent: this is a host-side
|
||||||
|
# script, and a machine without rsync is not a machine that would run it.
|
||||||
|
REQUIRED = ("bash", "rsync", "findmnt")
|
||||||
|
pytestmark = pytest.mark.skipif(
|
||||||
|
not all(shutil.which(tool) for tool in REQUIRED),
|
||||||
|
reason=f"needs {', '.join(REQUIRED)} on PATH",
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def run(*arguments):
|
||||||
|
return subprocess.run(
|
||||||
|
["bash", str(SCRIPT), *arguments], capture_output=True, text=True
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def mirror(tmp_path):
|
||||||
|
source = tmp_path / "mirror"
|
||||||
|
(source / "Album").mkdir(parents=True)
|
||||||
|
(source / "Album" / "track.mp3").write_bytes(b"x")
|
||||||
|
return source
|
||||||
|
|
||||||
|
|
||||||
|
def test_the_host_root_is_refused(mirror):
|
||||||
|
"""Stripping the trailing slash from "/" leaves an empty string, and an
|
||||||
|
earlier version then reported it as "not a directory" instead."""
|
||||||
|
result = run(str(mirror), "/")
|
||||||
|
|
||||||
|
assert result.returncode == 1
|
||||||
|
assert "refusing to sync onto /" in result.stderr
|
||||||
|
|
||||||
|
|
||||||
|
def test_an_empty_mirror_is_refused(tmp_path):
|
||||||
|
"""Mirroring nothing onto the device would delete everything on it."""
|
||||||
|
empty = tmp_path / "empty"
|
||||||
|
empty.mkdir()
|
||||||
|
destination = tmp_path / "dest"
|
||||||
|
destination.mkdir()
|
||||||
|
|
||||||
|
result = run(str(empty), str(destination))
|
||||||
|
|
||||||
|
assert result.returncode == 1
|
||||||
|
assert "refusing to mirror nothing" in result.stderr
|
||||||
|
|
||||||
|
|
||||||
|
def test_syncing_a_directory_onto_itself_is_refused(mirror):
|
||||||
|
result = run(str(mirror), str(mirror))
|
||||||
|
|
||||||
|
assert result.returncode == 1
|
||||||
|
assert "same directory" in result.stderr
|
||||||
|
|
||||||
|
|
||||||
|
def test_a_non_fat_destination_is_refused(mirror, tmp_path):
|
||||||
|
"""Which is also how an unmounted device is caught: /media/IPOD/Music then
|
||||||
|
resolves to the host's own root filesystem."""
|
||||||
|
destination = tmp_path / "dest"
|
||||||
|
destination.mkdir()
|
||||||
|
|
||||||
|
result = run(str(mirror), str(destination))
|
||||||
|
|
||||||
|
assert result.returncode == 1
|
||||||
|
assert "not FAT" in result.stderr
|
||||||
|
assert "Is the device mounted?" in result.stderr
|
||||||
|
|
||||||
|
|
||||||
|
def test_a_missing_destination_is_refused(mirror, tmp_path):
|
||||||
|
result = run(str(mirror), str(tmp_path / "nowhere"))
|
||||||
|
|
||||||
|
assert result.returncode == 1
|
||||||
|
assert "not a directory" in result.stderr
|
||||||
|
|
||||||
|
|
||||||
|
def test_a_subdirectory_of_the_device_is_a_valid_target(mirror, tmp_path):
|
||||||
|
"""The better target, in fact: --delete is confined to it."""
|
||||||
|
destination = tmp_path / "dest" / "Music"
|
||||||
|
destination.mkdir(parents=True)
|
||||||
|
|
||||||
|
result = run("-f", "-n", str(mirror), str(destination))
|
||||||
|
|
||||||
|
assert result.returncode == 0, result.stderr
|
||||||
|
assert "dry run, nothing was written" in result.stderr
|
||||||
|
|
||||||
|
|
||||||
|
def test_the_device_prefix_is_derived_from_the_destination(mirror, tmp_path):
|
||||||
|
"""Derived rather than configured, so it cannot disagree with where the
|
||||||
|
files are actually going -- and the device's path limit applies to it."""
|
||||||
|
destination = tmp_path / "dest" / "Music"
|
||||||
|
destination.mkdir(parents=True)
|
||||||
|
|
||||||
|
result = run("-f", "-n", str(mirror), str(destination))
|
||||||
|
|
||||||
|
assert "the device will see this as /" in result.stderr
|
||||||
|
|
||||||
|
|
||||||
|
def test_a_dry_run_writes_nothing(mirror, tmp_path):
|
||||||
|
destination = tmp_path / "dest"
|
||||||
|
destination.mkdir()
|
||||||
|
|
||||||
|
run("-f", "-n", str(mirror), str(destination))
|
||||||
|
|
||||||
|
assert list(destination.iterdir()) == []
|
||||||
|
|
||||||
|
|
||||||
|
def test_rockbox_is_never_deleted(mirror, tmp_path):
|
||||||
|
"""A sync to the card root would otherwise remove the Rockbox install,
|
||||||
|
since the mirror does not contain it."""
|
||||||
|
destination = tmp_path / "dest"
|
||||||
|
destination.mkdir()
|
||||||
|
(destination / ".rockbox").mkdir()
|
||||||
|
(destination / ".rockbox" / "rockbox.ipod").write_bytes(b"firmware")
|
||||||
|
(destination / ".scrobbler.log").write_bytes(b"#AUDIOSCROBBLER/1.1\n")
|
||||||
|
(destination / "Stale.mp3").write_bytes(b"old")
|
||||||
|
|
||||||
|
result = run("-f", "-S", "-U", str(mirror), str(destination))
|
||||||
|
|
||||||
|
assert result.returncode == 0, result.stderr
|
||||||
|
assert (destination / ".rockbox" / "rockbox.ipod").is_file()
|
||||||
|
assert (destination / ".scrobbler.log").is_file()
|
||||||
|
# But a track whose source has gone is still removed. That is the point.
|
||||||
|
assert not (destination / "Stale.mp3").exists()
|
||||||
|
assert (destination / "Album" / "track.mp3").is_file()
|
||||||
|
|
||||||
|
|
||||||
|
def test_help_goes_to_stdout_and_exits_clean():
|
||||||
|
"""Asking for help is not an error; getting the arguments wrong is."""
|
||||||
|
result = run("--help")
|
||||||
|
|
||||||
|
assert result.returncode == 0
|
||||||
|
assert result.stdout.startswith("usage:")
|
||||||
|
assert result.stderr == ""
|
||||||
|
|
||||||
|
|
||||||
|
def test_short_help_behaves_the_same():
|
||||||
|
result = run("-h")
|
||||||
|
|
||||||
|
assert result.returncode == 0
|
||||||
|
assert result.stdout.startswith("usage:")
|
||||||
|
|
||||||
|
|
||||||
|
def test_misuse_goes_to_stderr_and_does_not():
|
||||||
|
result = run("only-one-argument")
|
||||||
|
|
||||||
|
assert result.returncode == 2
|
||||||
|
assert result.stderr.startswith("usage:")
|
||||||
|
assert result.stdout == ""
|
||||||
|
|
||||||
|
|
||||||
|
def test_the_help_explains_what_the_destination_should_be():
|
||||||
|
"""The question this script actually gets asked."""
|
||||||
|
help_text = run("--help").stdout
|
||||||
|
|
||||||
|
assert "/media/IPOD/Music" in help_text
|
||||||
|
assert "artist folders" in help_text
|
||||||
|
assert ".rockbox" in help_text
|
||||||
|
|
||||||
|
|
||||||
|
def test_the_help_says_how_to_reach_and_leave_disk_mode():
|
||||||
|
help_text = run("--help").stdout
|
||||||
|
|
||||||
|
assert "Menu+Select" in help_text
|
||||||
|
assert "holding Play" in help_text
|
||||||
|
|
||||||
|
|
||||||
|
def test_quick_mode_skips_the_counting_pass(mirror, tmp_path):
|
||||||
|
"""Over SMB the walk is the expensive part, and doing it twice for a
|
||||||
|
percentage is not always the trade you want."""
|
||||||
|
destination = tmp_path / "dest"
|
||||||
|
destination.mkdir()
|
||||||
|
|
||||||
|
result = run("-f", "-S", "-U", "-Q", str(mirror), str(destination))
|
||||||
|
|
||||||
|
assert result.returncode == 0, result.stderr
|
||||||
|
assert "skipping the count" in result.stderr
|
||||||
|
assert "files to copy" not in result.stderr
|
||||||
|
assert (destination / "Album" / "track.mp3").is_file()
|
||||||
|
|
||||||
|
|
||||||
|
def test_the_delta_algorithm_is_disabled(mirror, tmp_path):
|
||||||
|
"""It would read every destination file back over USB to checksum it, to
|
||||||
|
avoid resending an MP3 that has changed in its entirety anyway."""
|
||||||
|
script = SCRIPT.read_text()
|
||||||
|
|
||||||
|
assert "--whole-file" in script
|
||||||
|
|
||||||
|
|
||||||
|
def test_directory_timestamps_are_not_set(mirror, tmp_path):
|
||||||
|
"""One setattr round trip per directory, across six thousand albums, to set
|
||||||
|
timestamps nothing reads."""
|
||||||
|
script = SCRIPT.read_text()
|
||||||
|
|
||||||
|
assert "--omit-dir-times" in script
|
||||||
|
|
||||||
|
|
||||||
|
def test_an_interrupt_is_trapped_so_the_filesystem_is_flushed():
|
||||||
|
"""Ctrl-C during a transfer would otherwise skip the sync and the unmount,
|
||||||
|
leaving a journal-less FAT filesystem with dirty buffers -- which is the
|
||||||
|
corruption this script exists to prevent.
|
||||||
|
|
||||||
|
Structural rather than timed: reproducing a mid-transfer signal needs a
|
||||||
|
payload large enough to be slow, and a test that depends on losing a race
|
||||||
|
is a test that fails in CI for no reason.
|
||||||
|
"""
|
||||||
|
script = SCRIPT.read_text()
|
||||||
|
|
||||||
|
assert "trap interrupted INT TERM" in script
|
||||||
|
assert "exit 130" in script
|
||||||
|
|
||||||
|
|
||||||
|
def test_the_flush_and_unmount_happen_on_every_exit_path():
|
||||||
|
script = SCRIPT.read_text()
|
||||||
|
|
||||||
|
# Both the normal path and the interrupt path go through the same function,
|
||||||
|
# so one cannot drift from the other.
|
||||||
|
assert script.count("finish\n") >= 2
|
||||||
|
assert "--partial" not in script, "rsync must delete partial files, not keep them"
|
||||||
|
|
||||||
|
|
||||||
|
def test_the_database_step_is_skipped_without_a_tool(mirror, tmp_path, monkeypatch):
|
||||||
|
"""Opt-in, like the scrobbler: absent configuration is not an error."""
|
||||||
|
destination = tmp_path / "dest"
|
||||||
|
destination.mkdir()
|
||||||
|
monkeypatch.delenv("MUSIC_MIRROR_DATABASE_TOOL", raising=False)
|
||||||
|
|
||||||
|
result = run("-f", "-S", "-U", str(mirror), str(destination))
|
||||||
|
|
||||||
|
assert result.returncode == 0, result.stderr
|
||||||
|
assert "no database tool configured" in result.stderr
|
||||||
|
|
||||||
|
|
||||||
|
def test_a_missing_database_tool_is_refused(mirror, tmp_path, monkeypatch):
|
||||||
|
destination = tmp_path / "dest"
|
||||||
|
destination.mkdir()
|
||||||
|
monkeypatch.setenv("MUSIC_MIRROR_DATABASE_TOOL", str(tmp_path / "nonexistent"))
|
||||||
|
|
||||||
|
result = run("-f", "-S", "-U", str(mirror), str(destination))
|
||||||
|
|
||||||
|
assert result.returncode == 1
|
||||||
|
assert "not executable" in result.stderr
|
||||||
|
|
||||||
|
|
||||||
|
def test_the_database_step_can_be_skipped(mirror, tmp_path, monkeypatch):
|
||||||
|
destination = tmp_path / "dest"
|
||||||
|
destination.mkdir()
|
||||||
|
monkeypatch.setenv("MUSIC_MIRROR_DATABASE_TOOL", str(tmp_path / "nonexistent"))
|
||||||
|
|
||||||
|
result = run("-f", "-S", "-U", "-B", str(mirror), str(destination))
|
||||||
|
|
||||||
|
assert result.returncode == 0, result.stderr
|
||||||
|
assert "not executable" not in result.stderr
|
||||||
|
|
||||||
|
|
||||||
|
def test_the_scan_reads_from_the_mirror_not_the_device():
|
||||||
|
"""The whole point: tags come off the mirror, only the .tcd files go over
|
||||||
|
USB. Reading 49,600 files through an iPod's USB bridge is the slow path."""
|
||||||
|
script = SCRIPT.read_text()
|
||||||
|
|
||||||
|
assert 'ln -s "$mirror"' in script
|
||||||
|
assert 'cd "$scratch"' in script
|
||||||
+36
-8
@@ -21,12 +21,26 @@ from pathlib import Path
|
|||||||
|
|
||||||
RESERVED = re.compile(r'[<>:"\\|?*\x00-\x1f]')
|
RESERVED = re.compile(r'[<>:"\\|?*\x00-\x1f]')
|
||||||
COMPONENT_LIMIT = 255
|
COMPONENT_LIMIT = 255
|
||||||
# Rockbox builds paths into a fixed buffer; long trees fail on the device even
|
# Rockbox's MAX_PATH, from firmware/include/fs_defines.h. It bounds the path as
|
||||||
# when every individual component is legal.
|
# the device sees it, so the directory the mirror is copied into comes out of
|
||||||
|
# the same budget.
|
||||||
PATH_LIMIT = 260
|
PATH_LIMIT = 260
|
||||||
|
DEVICE_PREFIX = "/Music"
|
||||||
|
|
||||||
|
|
||||||
def problems_with(relative):
|
def device_prefix_length(prefix):
|
||||||
|
"""Return the on-device prefix as it will actually appear, with slashes.
|
||||||
|
|
||||||
|
"/Music" costs seven characters -- the leading slash, the name, and the
|
||||||
|
separator before the mirror's own path -- while an empty prefix costs one.
|
||||||
|
Approximating that loses a character at the root, which is precisely where
|
||||||
|
the longest paths are.
|
||||||
|
"""
|
||||||
|
cleaned = prefix.strip("/")
|
||||||
|
return f"/{cleaned}/" if cleaned else "/"
|
||||||
|
|
||||||
|
|
||||||
|
def problems_with(relative, budget=PATH_LIMIT):
|
||||||
"""Return every reason this relative path is unfit for FAT32."""
|
"""Return every reason this relative path is unfit for FAT32."""
|
||||||
found = []
|
found = []
|
||||||
for part in relative.parts:
|
for part in relative.parts:
|
||||||
@@ -36,8 +50,8 @@ def problems_with(relative):
|
|||||||
found.append(f"trailing dot or space in {part!r}")
|
found.append(f"trailing dot or space in {part!r}")
|
||||||
if len(part) > COMPONENT_LIMIT:
|
if len(part) > COMPONENT_LIMIT:
|
||||||
found.append(f"component of {len(part)} characters")
|
found.append(f"component of {len(part)} characters")
|
||||||
if len(str(relative)) > PATH_LIMIT:
|
if len(str(relative)) > budget:
|
||||||
found.append(f"path of {len(str(relative))} characters")
|
found.append(f"path of {len(str(relative))} characters, over a budget of {budget}")
|
||||||
return found
|
return found
|
||||||
|
|
||||||
|
|
||||||
@@ -52,7 +66,21 @@ def main(argv=None):
|
|||||||
parser = argparse.ArgumentParser(description=__doc__)
|
parser = argparse.ArgumentParser(description=__doc__)
|
||||||
parser.add_argument("root", help="directory to check, e.g. the mirror")
|
parser.add_argument("root", help="directory to check, e.g. the mirror")
|
||||||
parser.add_argument("--limit", type=int, default=0, help="show at most this many")
|
parser.add_argument("--limit", type=int, default=0, help="show at most this many")
|
||||||
|
parser.add_argument(
|
||||||
|
"--max-path",
|
||||||
|
type=int,
|
||||||
|
default=PATH_LIMIT,
|
||||||
|
help=f"longest path the device will take, from its root (default {PATH_LIMIT},"
|
||||||
|
" Rockbox's MAX_PATH)",
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
"--device-prefix",
|
||||||
|
default=DEVICE_PREFIX,
|
||||||
|
help="directory the mirror is copied into on the device; its length comes out"
|
||||||
|
f" of the budget (default {DEVICE_PREFIX})",
|
||||||
|
)
|
||||||
args = parser.parse_args(argv)
|
args = parser.parse_args(argv)
|
||||||
|
budget = max(0, args.max_path - len(device_prefix_length(args.device_prefix)))
|
||||||
|
|
||||||
root = Path(args.root)
|
root = Path(args.root)
|
||||||
if not root.is_dir():
|
if not root.is_dir():
|
||||||
@@ -68,7 +96,7 @@ def main(argv=None):
|
|||||||
# different strings, and the collision check would miss it.
|
# different strings, and the collision check would miss it.
|
||||||
key = unicodedata.normalize("NFC", str(relative)).casefold()
|
key = unicodedata.normalize("NFC", str(relative)).casefold()
|
||||||
by_case[key].append(relative)
|
by_case[key].append(relative)
|
||||||
for problem in problems_with(relative):
|
for problem in problems_with(relative, budget):
|
||||||
faults.append((relative, problem))
|
faults.append((relative, problem))
|
||||||
|
|
||||||
for relative, group in sorted(by_case.items()):
|
for relative, group in sorted(by_case.items()):
|
||||||
@@ -82,8 +110,8 @@ def main(argv=None):
|
|||||||
print(f"\n{len(faults)} problems across {total} files", file=sys.stderr)
|
print(f"\n{len(faults)} problems across {total} files", file=sys.stderr)
|
||||||
if faults:
|
if faults:
|
||||||
print(
|
print(
|
||||||
"Run music-mirror with --fat32-safe to have the mirror named"
|
"Run music-mirror with --fat32-safe to have the mirror named acceptably"
|
||||||
" acceptably in the first place.",
|
" in the first place; it shortens over-long paths as well.",
|
||||||
file=sys.stderr,
|
file=sys.stderr,
|
||||||
)
|
)
|
||||||
return 1 if faults else 0
|
return 1 if faults else 0
|
||||||
|
|||||||
Executable
+190
@@ -0,0 +1,190 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
"""Render rsync's per-file output as a single updating status line.
|
||||||
|
|
||||||
|
Fed the size and path rsync reports with --out-format='%l %n', one per line.
|
||||||
|
The size is what makes an estimate possible: rsync's own rate is not exposed
|
||||||
|
per file, but bytes completed over time elapsed is the same arithmetic and
|
||||||
|
needs nothing rsync does not already print. Prints one
|
||||||
|
line that rewrites itself, showing how far through the transfer is and which
|
||||||
|
album is currently going across, rather than either scrolling fifty thousand
|
||||||
|
filenames past or -- as rsync does while it builds its file list -- saying
|
||||||
|
nothing at all for several minutes.
|
||||||
|
|
||||||
|
Falls back to periodic plain lines when stderr is not a terminal, so a log does
|
||||||
|
not fill up with carriage returns.
|
||||||
|
"""
|
||||||
|
|
||||||
|
import argparse
|
||||||
|
import collections
|
||||||
|
import os
|
||||||
|
import shutil
|
||||||
|
import sys
|
||||||
|
import time
|
||||||
|
|
||||||
|
# The estimate is taken over a trailing window rather than the whole run, so it
|
||||||
|
# follows a device that slows down instead of averaging the slowdown away.
|
||||||
|
RATE_WINDOW_SECONDS = 30.0
|
||||||
|
|
||||||
|
# Below this the window is too narrow to divide by: the first few files arrive
|
||||||
|
# in microseconds and produce a rate in the gigabytes per second, and an ETA of
|
||||||
|
# nothing at all. Better to show neither until the figure means something.
|
||||||
|
RATE_MINIMUM_SPAN_SECONDS = 2.0
|
||||||
|
|
||||||
|
|
||||||
|
def parse(line):
|
||||||
|
"""Return (bytes, path) for one line of rsync output.
|
||||||
|
|
||||||
|
Tolerates a bare path, in case someone runs this against --out-format='%n'.
|
||||||
|
"""
|
||||||
|
line = line.rstrip("\n")
|
||||||
|
size, separator, path = line.partition(" ")
|
||||||
|
if separator and size.isdigit():
|
||||||
|
return int(size), path
|
||||||
|
return 0, line
|
||||||
|
|
||||||
|
|
||||||
|
def human_bytes(count):
|
||||||
|
size = float(count)
|
||||||
|
for unit in ("B", "KiB", "MiB", "GiB", "TiB"):
|
||||||
|
if size < 1024 or unit == "TiB":
|
||||||
|
return f"{size:.1f} {unit}"
|
||||||
|
size /= 1024
|
||||||
|
|
||||||
|
|
||||||
|
def human_duration(seconds):
|
||||||
|
"""Return a duration nobody has to do arithmetic on."""
|
||||||
|
seconds = int(seconds)
|
||||||
|
if seconds < 60:
|
||||||
|
return f"{seconds}s"
|
||||||
|
if seconds < 3600:
|
||||||
|
return f"{seconds // 60}m{seconds % 60:02d}s"
|
||||||
|
return f"{seconds // 3600}h{(seconds % 3600) // 60:02d}m"
|
||||||
|
|
||||||
|
|
||||||
|
class Rate:
|
||||||
|
"""Bytes per second over a trailing window."""
|
||||||
|
|
||||||
|
def __init__(self, window=RATE_WINDOW_SECONDS):
|
||||||
|
self.window = window
|
||||||
|
self.samples = collections.deque()
|
||||||
|
|
||||||
|
def add(self, when, total_bytes):
|
||||||
|
self.samples.append((when, total_bytes))
|
||||||
|
while len(self.samples) > 2 and when - self.samples[0][0] > self.window:
|
||||||
|
self.samples.popleft()
|
||||||
|
|
||||||
|
def per_second(self):
|
||||||
|
if len(self.samples) < 2:
|
||||||
|
return 0.0
|
||||||
|
(first_time, first_bytes), (last_time, last_bytes) = (
|
||||||
|
self.samples[0],
|
||||||
|
self.samples[-1],
|
||||||
|
)
|
||||||
|
elapsed = last_time - first_time
|
||||||
|
if elapsed < RATE_MINIMUM_SPAN_SECONDS:
|
||||||
|
return 0.0
|
||||||
|
return (last_bytes - first_bytes) / elapsed
|
||||||
|
|
||||||
|
|
||||||
|
def album_of(path):
|
||||||
|
"""Return "Artist / Album" for a mirror-relative path."""
|
||||||
|
parts = [part for part in path.strip("/").split("/") if part]
|
||||||
|
if len(parts) >= 3:
|
||||||
|
return f"{parts[0]} / {parts[1]}"
|
||||||
|
if len(parts) == 2:
|
||||||
|
return parts[0]
|
||||||
|
return ""
|
||||||
|
|
||||||
|
|
||||||
|
def fit(text, width):
|
||||||
|
"""Trim to the terminal, from the left: the album matters more than the artist."""
|
||||||
|
if width <= 1 or len(text) <= width:
|
||||||
|
return text
|
||||||
|
return "…" + text[-(width - 1) :]
|
||||||
|
|
||||||
|
|
||||||
|
def render(done, total, copied, expected, rate, label, width):
|
||||||
|
"""Build the status line, giving whatever room is left to the album."""
|
||||||
|
if total > 0:
|
||||||
|
share = min(100, done * 100 // total)
|
||||||
|
head = f"[{share:>3}%] {done:,}/{total:,}"
|
||||||
|
else:
|
||||||
|
head = f"[{done:,} files]"
|
||||||
|
|
||||||
|
if expected > 0:
|
||||||
|
head += f" {human_bytes(copied)}/{human_bytes(expected)}"
|
||||||
|
if rate > 0:
|
||||||
|
head += f" {human_bytes(rate)}/s"
|
||||||
|
remaining = expected - copied
|
||||||
|
if remaining > 0:
|
||||||
|
head += f" ETA {human_duration(remaining / rate)}"
|
||||||
|
head += " "
|
||||||
|
return head + fit(label, max(0, width - len(head)))
|
||||||
|
|
||||||
|
|
||||||
|
def main(argv=None, stream=None, out=None):
|
||||||
|
parser = argparse.ArgumentParser(description=__doc__)
|
||||||
|
parser.add_argument("--total", type=int, default=0, help="files expected")
|
||||||
|
parser.add_argument("--bytes", type=int, default=0, help="bytes expected")
|
||||||
|
parser.add_argument("--interval", type=float, default=0.1, help="seconds between redraws")
|
||||||
|
args = parser.parse_args(argv)
|
||||||
|
|
||||||
|
stream = stream or sys.stdin
|
||||||
|
out = out or sys.stderr
|
||||||
|
interactive = out.isatty()
|
||||||
|
width = shutil.get_terminal_size((100, 24)).columns - 1
|
||||||
|
|
||||||
|
done = 0
|
||||||
|
copied = 0
|
||||||
|
rate = Rate()
|
||||||
|
started = time.monotonic()
|
||||||
|
rate.add(started, 0)
|
||||||
|
last_drawn = 0.0
|
||||||
|
label = ""
|
||||||
|
|
||||||
|
for line in stream:
|
||||||
|
size, path = parse(line)
|
||||||
|
# rsync reports directories too, with a trailing slash and an inode
|
||||||
|
# size. Counting them puts the percentage past a hundred and the byte
|
||||||
|
# total well over what will actually be transferred.
|
||||||
|
if not path or path.endswith("/"):
|
||||||
|
continue
|
||||||
|
done += 1
|
||||||
|
copied += size
|
||||||
|
label = album_of(path) or os.path.basename(path)
|
||||||
|
|
||||||
|
now = time.monotonic()
|
||||||
|
rate.add(now, copied)
|
||||||
|
if interactive:
|
||||||
|
if now - last_drawn >= args.interval:
|
||||||
|
out.write(
|
||||||
|
"\r\033[2K"
|
||||||
|
+ render(done, args.total, copied, args.bytes, rate.per_second(),
|
||||||
|
label, width)
|
||||||
|
)
|
||||||
|
out.flush()
|
||||||
|
last_drawn = now
|
||||||
|
elif now - last_drawn >= 30:
|
||||||
|
out.write(
|
||||||
|
render(done, args.total, copied, args.bytes, rate.per_second(), label, width)
|
||||||
|
+ "\n"
|
||||||
|
)
|
||||||
|
out.flush()
|
||||||
|
last_drawn = now
|
||||||
|
|
||||||
|
elapsed = max(1e-9, time.monotonic() - started)
|
||||||
|
if interactive:
|
||||||
|
out.write("\r\033[2K")
|
||||||
|
summary = render(done, args.total, copied, args.bytes, 0, label, width).rstrip()
|
||||||
|
out.write(f"{summary}\n")
|
||||||
|
if copied:
|
||||||
|
out.write(
|
||||||
|
f"copied {human_bytes(copied)} in {human_duration(elapsed)}"
|
||||||
|
f" at {human_bytes(copied / elapsed)}/s\n"
|
||||||
|
)
|
||||||
|
out.flush()
|
||||||
|
return 0
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
sys.exit(main())
|
||||||
+189
-22
@@ -12,46 +12,98 @@
|
|||||||
set -euo pipefail
|
set -euo pipefail
|
||||||
|
|
||||||
usage() {
|
usage() {
|
||||||
cat >&2 <<'USAGE'
|
# Help goes to stdout and exits clean; misuse goes to stderr and does not.
|
||||||
|
local stream=2 code=2
|
||||||
|
if [ "${1:-}" = "help" ]; then
|
||||||
|
stream=1
|
||||||
|
code=0
|
||||||
|
fi
|
||||||
|
cat >&"$stream" <<'USAGE'
|
||||||
usage: sync-to-ipod.sh [options] <mirror> <destination>
|
usage: sync-to-ipod.sh [options] <mirror> <destination>
|
||||||
|
|
||||||
-n dry run; show what would change and touch nothing
|
-n dry run; show what would change and touch nothing
|
||||||
|
-Q skip the counting pass; no percentage or estimate, but one less walk
|
||||||
|
of the source tree, which over SMB is the expensive part
|
||||||
-f copy even if the FAT32 check finds unacceptable paths
|
-f copy even if the FAT32 check finds unacceptable paths
|
||||||
-S skip submitting the Rockbox scrobbler log to Last.fm
|
-S skip submitting the Rockbox scrobbler log to Last.fm
|
||||||
|
-B skip rebuilding the Rockbox database
|
||||||
-U leave the destination mounted afterwards
|
-U leave the destination mounted afterwards
|
||||||
|
|
||||||
|
Rebuilding the database needs MUSIC_MIRROR_DATABASE_TOOL pointing at Rockbox's
|
||||||
|
host-side builder (tools/database, built with ./tools/configure --type=d). It
|
||||||
|
is skipped with a note when unset. The scan reads tags from the mirror rather
|
||||||
|
than from the device, so it costs seconds rather than the hours an on-device
|
||||||
|
commit takes -- and on a large library the on-device commit may not finish at
|
||||||
|
all.
|
||||||
|
|
||||||
Submitting scrobbles needs LASTFM_API_KEY and LASTFM_API_SECRET; it is skipped
|
Submitting scrobbles needs LASTFM_API_KEY and LASTFM_API_SECRET; it is skipped
|
||||||
with a note when they are unset. Scrobbling is a write method and needs the
|
with a note when they are unset. Scrobbling is a write method and needs the
|
||||||
secret, unlike the read-only calls elsewhere in these projects.
|
secret, unlike the read-only calls elsewhere in these projects.
|
||||||
|
|
||||||
The destination must be a mounted FAT filesystem. Reach it with the Apple
|
The mirror is the directory holding the artist folders. The destination is
|
||||||
firmware's disk mode: Menu+Select to reboot, then immediately Select+Play.
|
where those folders should end up on the device -- not the card root, unless
|
||||||
|
that is genuinely where you want them:
|
||||||
|
|
||||||
|
sync-to-ipod.sh /mnt/tank/media/music-mp3 /media/IPOD/Music
|
||||||
|
|
||||||
|
A subdirectory is the better target: --delete is confined to it, and the path
|
||||||
|
budget is derived from it, since the device's 260-character limit counts the
|
||||||
|
whole path as the device sees it. /.rockbox and the scrobbler logs are never
|
||||||
|
deleted wherever you point this.
|
||||||
|
|
||||||
|
The destination must be on a mounted FAT filesystem. That check is also what
|
||||||
|
catches an unmounted device: /media/IPOD/Music then resolves to the host's own
|
||||||
|
root filesystem, and this refuses to empty that.
|
||||||
|
|
||||||
|
Progress is one line that rewrites itself, showing the album currently going
|
||||||
|
across, how far through the transfer is, the rate, and an estimate of what is
|
||||||
|
left. Working the totals out first means a second pass over the tree, which is
|
||||||
|
the price of figures that mean something; rsync's own percentage is computed
|
||||||
|
against a file list it is still building.
|
||||||
|
|
||||||
|
Reach the device with the Apple firmware's disk mode: Menu+Select to reboot,
|
||||||
|
then immediately Select+Play. Power off afterwards by holding Play.
|
||||||
USAGE
|
USAGE
|
||||||
exit 2
|
exit "$code"
|
||||||
}
|
}
|
||||||
|
|
||||||
dry_run=false
|
dry_run=false
|
||||||
|
quick=false
|
||||||
force=false
|
force=false
|
||||||
unmount=true
|
unmount=true
|
||||||
scrobble=true
|
scrobble=true
|
||||||
while getopts ":nfSUh" option; do
|
database=true
|
||||||
|
for argument in "$@"; do
|
||||||
|
[ "$argument" = "--help" ] && usage help
|
||||||
|
done
|
||||||
|
while getopts ":nQfSBUh" option; do
|
||||||
case "$option" in
|
case "$option" in
|
||||||
n) dry_run=true ;;
|
n) dry_run=true ;;
|
||||||
|
Q) quick=true ;;
|
||||||
f) force=true ;;
|
f) force=true ;;
|
||||||
S) scrobble=false ;;
|
S) scrobble=false ;;
|
||||||
|
B) database=false ;;
|
||||||
U) unmount=false ;;
|
U) unmount=false ;;
|
||||||
|
h) usage help ;;
|
||||||
*) usage ;;
|
*) usage ;;
|
||||||
esac
|
esac
|
||||||
done
|
done
|
||||||
shift $((OPTIND - 1))
|
shift $((OPTIND - 1))
|
||||||
[ $# -eq 2 ] || usage
|
[ $# -eq 2 ] || usage
|
||||||
|
|
||||||
|
# Trailing slashes are stripped for tidiness, but stripping one from "/" leaves
|
||||||
|
# an empty string, and the guard below would then never see the root it is
|
||||||
|
# there to refuse.
|
||||||
mirror=${1%/}
|
mirror=${1%/}
|
||||||
|
mirror=${mirror:-/}
|
||||||
destination=${2%/}
|
destination=${2%/}
|
||||||
|
destination=${destination:-/}
|
||||||
here=$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")" && pwd)
|
here=$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")" && pwd)
|
||||||
|
|
||||||
die() {
|
die() {
|
||||||
printf 'sync-to-ipod: %s\n' "$1" >&2
|
# Every argument, not just the first: the second half of a message is
|
||||||
|
# usually the half that says what to do about it.
|
||||||
|
printf 'sync-to-ipod: %s\n' "$*" >&2
|
||||||
exit 1
|
exit 1
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -59,28 +111,41 @@ die() {
|
|||||||
[ -n "$(ls -A "$mirror")" ] || die "mirror $mirror is empty; refusing to mirror nothing"
|
[ -n "$(ls -A "$mirror")" ] || die "mirror $mirror is empty; refusing to mirror nothing"
|
||||||
[ -d "$destination" ] || die "destination $destination is not a directory"
|
[ -d "$destination" ] || die "destination $destination is not a directory"
|
||||||
|
|
||||||
# --delete makes every one of these load-bearing. A destination that is not its
|
# --delete makes every one of these load-bearing. Emptying the wrong directory
|
||||||
# own mount point means the path is wrong, and emptying the wrong directory is
|
# is not a mistake that announces itself.
|
||||||
# not a mistake that announces itself.
|
|
||||||
case "$destination" in
|
case "$destination" in
|
||||||
"" | "/" | "$HOME") die "refusing to sync onto $destination" ;;
|
"" | "/" | "$HOME") die "refusing to sync onto $destination" ;;
|
||||||
esac
|
esac
|
||||||
[ "$(readlink -f "$mirror")" != "$(readlink -f "$destination")" ] ||
|
[ "$(readlink -f "$mirror")" != "$(readlink -f "$destination")" ] ||
|
||||||
die "mirror and destination are the same directory"
|
die "mirror and destination are the same directory"
|
||||||
mountpoint -q -- "$destination" || die "$destination is not a mount point"
|
|
||||||
|
|
||||||
|
# The filesystem the destination sits on, which is the check that matters: a
|
||||||
|
# subdirectory of the card is a perfectly good target, and is the better one,
|
||||||
|
# because --delete is then confined to it. Being FAT is also what proves the
|
||||||
|
# card is mounted at all -- an unmounted /media/IPOD/Music resolves to the
|
||||||
|
# host's own root filesystem, and this refuses to empty that.
|
||||||
filesystem=$(findmnt -no FSTYPE --target "$destination")
|
filesystem=$(findmnt -no FSTYPE --target "$destination")
|
||||||
|
mounted_on=$(findmnt -no TARGET --target "$destination")
|
||||||
case "$filesystem" in
|
case "$filesystem" in
|
||||||
vfat | exfat) ;;
|
vfat | exfat) ;;
|
||||||
*)
|
*)
|
||||||
$force || die "$destination is $filesystem, not FAT; pass -f if that is deliberate"
|
$force ||
|
||||||
|
die "$destination is on a $filesystem filesystem, not FAT." \
|
||||||
|
"Is the device mounted? Pass -f if this is deliberate."
|
||||||
printf 'sync-to-ipod: destination is %s, not FAT\n' "$filesystem" >&2
|
printf 'sync-to-ipod: destination is %s, not FAT\n' "$filesystem" >&2
|
||||||
;;
|
;;
|
||||||
esac
|
esac
|
||||||
|
|
||||||
|
# What the device will call this directory, which is what its path limit
|
||||||
|
# applies to. Derived rather than configured, so it cannot disagree with where
|
||||||
|
# the files are actually going.
|
||||||
|
device_prefix=${destination#"$mounted_on"}
|
||||||
|
device_prefix="/${device_prefix#/}"
|
||||||
|
printf 'sync-to-ipod: the device will see this as %s\n' "$device_prefix" >&2
|
||||||
|
|
||||||
if $force; then
|
if $force; then
|
||||||
printf 'sync-to-ipod: skipping the FAT32 check\n' >&2
|
printf 'sync-to-ipod: skipping the FAT32 check\n' >&2
|
||||||
elif ! python3 "$here/check_fat32.py" "$mirror"; then
|
elif ! python3 "$here/check_fat32.py" --device-prefix "$device_prefix" "$mirror"; then
|
||||||
die "the mirror holds paths FAT32 will not take; run music-mirror with --fat32-safe"
|
die "the mirror holds paths FAT32 will not take; run music-mirror with --fat32-safe"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
@@ -101,36 +166,138 @@ fi
|
|||||||
# asking for them produces a screenful of errors and a non-zero exit.
|
# asking for them produces a screenful of errors and a non-zero exit.
|
||||||
# --modify-window=2 because FAT stores mtimes to two-second resolution, without
|
# --modify-window=2 because FAT stores mtimes to two-second resolution, without
|
||||||
# which every file looks changed and the whole library is copied every time.
|
# which every file looks changed and the whole library is copied every time.
|
||||||
|
# --whole-file is already the default when both ends are local paths, and an
|
||||||
|
# SMB or FAT mount counts as one, but stating it documents that the delta
|
||||||
|
# algorithm is deliberately not wanted: it would read every destination file
|
||||||
|
# back over USB to compute a checksum, to save sending an MP3 that has changed
|
||||||
|
# entirely anyway.
|
||||||
|
#
|
||||||
|
# --omit-dir-times drops a setattr round trip per directory. Across six
|
||||||
|
# thousand album folders on a FAT card that is six thousand operations to set
|
||||||
|
# timestamps nothing reads.
|
||||||
|
options=(--recursive --times --delete --modify-window=2 --whole-file --omit-dir-times)
|
||||||
# --delete removes tracks whose source has gone, which is the point. It would
|
# --delete removes tracks whose source has gone, which is the point. It would
|
||||||
# also remove everything on the device that the mirror does not contain -- and
|
# also remove everything on the device that the mirror does not contain -- and
|
||||||
# if the destination is the card root that means /.rockbox, the Rockbox install
|
# if the destination is the card root that means /.rockbox, the Rockbox install
|
||||||
# itself. Excluded paths are not deleted unless --delete-excluded is given,
|
# itself. Excluded paths are not deleted unless --delete-excluded is given,
|
||||||
# which it never is here.
|
# which it never is here.
|
||||||
options=(--recursive --times --delete --modify-window=2 --human-readable --info=progress2)
|
|
||||||
for owned in "/.rockbox" "/.scrobbler.log" "/.scrobbler.log.*" "/.playlist_control" \
|
for owned in "/.rockbox" "/.scrobbler.log" "/.scrobbler.log.*" "/.playlist_control" \
|
||||||
"/System Volume Information" "/.Spotlight-V100" "/.Trashes" "/.fseventsd"; do
|
"/System Volume Information" "/.Spotlight-V100" "/.Trashes" "/.fseventsd"; do
|
||||||
options+=(--exclude "$owned")
|
options+=(--exclude "$owned")
|
||||||
done
|
done
|
||||||
$dry_run && options+=(--dry-run --verbose)
|
|
||||||
|
|
||||||
printf 'sync-to-ipod: %s -> %s\n' "$mirror" "$destination" >&2
|
printf 'sync-to-ipod: %s -> %s\n' "$mirror" "$destination" >&2
|
||||||
rsync "${options[@]}" "$mirror/" "$destination/"
|
|
||||||
|
|
||||||
if $dry_run; then
|
if $dry_run; then
|
||||||
|
rsync "${options[@]}" --dry-run --verbose "$mirror/" "$destination/"
|
||||||
printf 'sync-to-ipod: dry run, nothing was written\n' >&2
|
printf 'sync-to-ipod: dry run, nothing was written\n' >&2
|
||||||
exit 0
|
exit 0
|
||||||
fi
|
fi
|
||||||
|
|
||||||
sync
|
# rsync says nothing at all while it builds its file list, which on fifty
|
||||||
if $unmount; then
|
# thousand files over USB is minutes of apparent hang. Counting first costs a
|
||||||
device=$(findmnt -no SOURCE --target "$destination")
|
# second pass over the tree but means the transfer can show a real percentage
|
||||||
|
# rather than a number that grows as rsync discovers more work.
|
||||||
|
total=0
|
||||||
|
total_bytes=0
|
||||||
|
if $quick; then
|
||||||
|
printf 'sync-to-ipod: skipping the count; no percentage or estimate\n' >&2
|
||||||
|
else
|
||||||
|
printf 'sync-to-ipod: working out what needs copying...\n' >&2
|
||||||
|
# %l is the file's size, which is what makes an estimate possible.
|
||||||
|
# Directories are dropped: rsync reports those too, with an inode size that
|
||||||
|
# would inflate the total by several megabytes of nothing.
|
||||||
|
counted=$(rsync "${options[@]}" --dry-run --out-format='%l %n' "$mirror/" "$destination/" |
|
||||||
|
awk '!/\/$/ { files++; bytes += $1 } END { print files + 0, bytes + 0 }')
|
||||||
|
total=${counted% *}
|
||||||
|
total_bytes=${counted#* }
|
||||||
|
printf 'sync-to-ipod: %s files to copy\n' "$total" >&2
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Flushing and unmounting is the whole reason this is a script, so it has to
|
||||||
|
# happen on the way out whichever way that is. Ctrl-C during a transfer would
|
||||||
|
# otherwise leave a FAT filesystem with dirty buffers and no journal, which is
|
||||||
|
# the corruption this exists to avoid.
|
||||||
|
finish() {
|
||||||
|
sync
|
||||||
|
if $unmount; then
|
||||||
|
device=$(findmnt -no SOURCE --target "$destination" 2>/dev/null || true)
|
||||||
|
if [ -n "$device" ]; then
|
||||||
printf 'sync-to-ipod: unmounting %s\n' "$device" >&2
|
printf 'sync-to-ipod: unmounting %s\n' "$device" >&2
|
||||||
if command -v udisksctl >/dev/null 2>&1; then
|
if command -v udisksctl >/dev/null 2>&1; then
|
||||||
udisksctl unmount -b "$device"
|
udisksctl unmount -b "$device" || umount -- "$destination" || true
|
||||||
else
|
else
|
||||||
umount -- "$destination"
|
umount -- "$destination" || true
|
||||||
fi
|
fi
|
||||||
printf 'sync-to-ipod: safe to disconnect\n' >&2
|
printf 'sync-to-ipod: safe to disconnect\n' >&2
|
||||||
else
|
fi
|
||||||
|
else
|
||||||
printf 'sync-to-ipod: still mounted; unmount before disconnecting\n' >&2
|
printf 'sync-to-ipod: still mounted; unmount before disconnecting\n' >&2
|
||||||
fi
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
interrupted() {
|
||||||
|
trap - INT TERM
|
||||||
|
printf '\nsync-to-ipod: interrupted -- rsync leaves no partial files, but the\n' >&2
|
||||||
|
printf 'sync-to-ipod: filesystem still needs flushing before you pull anything\n' >&2
|
||||||
|
finish
|
||||||
|
exit 130
|
||||||
|
}
|
||||||
|
|
||||||
|
trap interrupted INT TERM
|
||||||
|
|
||||||
|
rsync "${options[@]}" --out-format='%l %n' "$mirror/" "$destination/" |
|
||||||
|
python3 "$here/rsync_progress.py" --total "$total" --bytes "$total_bytes"
|
||||||
|
status=${PIPESTATUS[0]}
|
||||||
|
[ "$status" -eq 0 ] || die "rsync exited $status"
|
||||||
|
|
||||||
|
# Rockbox reads its database from .tcd files in .rockbox. Building them here
|
||||||
|
# rather than on the device is not just faster: the on-device commit sorts the
|
||||||
|
# whole index in whatever memory it can scrape together, and on a large library
|
||||||
|
# it runs for hours or dies outright.
|
||||||
|
#
|
||||||
|
# The scan reads tags through a scratch root -- a real .rockbox beside a symlink
|
||||||
|
# standing in for where the music lands on the device -- so the paths recorded
|
||||||
|
# match what Rockbox will look up, while the bytes are read from the mirror
|
||||||
|
# instead of over USB. The scratch is kept between runs because the builder is
|
||||||
|
# incremental: a second pass over unchanged files does no work at all.
|
||||||
|
rebuild_database() {
|
||||||
|
local tool=${MUSIC_MIRROR_DATABASE_TOOL:-}
|
||||||
|
if [ -z "$tool" ]; then
|
||||||
|
printf 'sync-to-ipod: no database tool configured, skipping the database\n' >&2
|
||||||
|
return 0
|
||||||
|
fi
|
||||||
|
[ -x "$tool" ] || die "$tool is not executable"
|
||||||
|
|
||||||
|
local device_rockbox="$mounted_on/.rockbox"
|
||||||
|
if [ ! -d "$device_rockbox" ]; then
|
||||||
|
printf 'sync-to-ipod: no .rockbox on the device, skipping the database\n' >&2
|
||||||
|
return 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
local scratch="${XDG_CACHE_HOME:-$HOME/.cache}/music-mirror/database"
|
||||||
|
mkdir -p "$scratch/.rockbox"
|
||||||
|
|
||||||
|
# Rebuild the symlink layout each time; the mirror path or the device
|
||||||
|
# prefix may have changed since the last run.
|
||||||
|
find "$scratch" -maxdepth 1 -type l -delete
|
||||||
|
if [ "$device_prefix" = "/" ]; then
|
||||||
|
ln -s "$mirror"/* "$scratch/" 2>/dev/null || true
|
||||||
|
else
|
||||||
|
local under=${device_prefix#/}
|
||||||
|
rm -rf "${scratch:?}/${under%%/*}"
|
||||||
|
mkdir -p "$scratch/$(dirname "$under")"
|
||||||
|
ln -s "$mirror" "$scratch/$under"
|
||||||
|
fi
|
||||||
|
|
||||||
|
printf 'sync-to-ipod: building the database from the mirror...\n' >&2
|
||||||
|
( cd "$scratch" && "$tool" ) >/dev/null || die "the database build failed"
|
||||||
|
|
||||||
|
cp -- "$scratch"/.rockbox/*.tcd "$device_rockbox/" ||
|
||||||
|
die "could not copy the database onto the device"
|
||||||
|
printf 'sync-to-ipod: database copied to %s\n' "$device_rockbox" >&2
|
||||||
|
}
|
||||||
|
|
||||||
|
$database && rebuild_database
|
||||||
|
|
||||||
|
finish
|
||||||
|
|||||||
Reference in New Issue
Block a user