Compare commits

...
2 Commits
2 changed files with 140 additions and 1 deletions

No files matched your search

+64 -1
View File
@@ -64,6 +64,8 @@ pub:
pub struct WriteReport {
pub:
in_place bool
// Переписан только тэг в начале файла (новый поместился на место старого), аудио не тронуто
header bool
// Почему не удалось переписать на месте (пусто, если на месте)
fallback string
}
@@ -75,6 +77,10 @@ mut:
audio_start u64
}
// Запас в конце тэга при полной перезаписи: следующие правки (длиннее описание, другие названия)
// поместятся на место старого тэга, и аудио переписывать не придётся
const book_tag_padding = 4096
// Фреймы, которые редактор пересобирает сам, остальные переносятся без изменений
const managed_frames = ['TIT2', 'TALB', 'TPE1', 'TPE2', 'TCOM', 'TIT1', 'TRCK', 'TYER', 'TDRC',
'TCON', 'COMM', 'TLAN']
@@ -342,11 +348,30 @@ fn read_book_tags_impl(path string) !FileTags {
if t.series == '' {
t.series = tit1
}
// Пустые или испорченные («�») поля — из ID3v1: старые файлы часто хранят там исходный текст в cp1251,
// а в ID3v2 он мог попасть уже испорченным (прочитан не в той кодировке и записан обратно)
if needs_v1(t.title) || needs_v1(t.album) || needs_v1(t.author) {
if v1 := read_v1(path) {
if needs_v1(t.title) && !needs_v1(v1.title) {
t.title = v1.title
}
if needs_v1(t.album) && !needs_v1(v1.album) {
t.album = v1.album
}
if needs_v1(t.author) && !needs_v1(v1.author) {
t.author = v1.author
}
}
}
// TPE1 мог быть не задан, тогда автор остаётся из TPE2
res.tags = t
return res
}
fn needs_v1(s string) bool {
return s.trim_space() == '' || s.contains('\uFFFD')
}
// "(101)" и "(101)Speech" из старых тэгов превращаем в название
fn clean_genre(g string) string {
if g.starts_with('(') && g.contains(')') {
@@ -558,7 +583,7 @@ fn build_tag(parsed ParsedTag, t BookTags, opts WriteOptions) []u8 {
frames << apic_frame(c, version)
}
frames << []u8{len: tag_padding}
frames << []u8{len: book_tag_padding}
mut out := [u8(`I`), `D`, `3`, version, 0, 0]
out << encode_header_size(u64(frames.len))
@@ -574,6 +599,16 @@ fn write_book_tags_impl(path string, t BookTags, opts WriteOptions) !WriteReport
parsed := parse_id3v2(path)!
tag := build_tag(parsed, t, opts)
// Новый тэг помещается на место старого — переписать только его, аудио не трогать
if fitted := fit_tag(parsed, tag) {
if write_header(path, fitted) {
return WriteReport{
in_place: true
header: true
}
}
}
tmp_path := path + '.tagtmp'
mut buf := []u8{len: 1024 * 1024}
{
@@ -624,6 +659,34 @@ fn write_book_tags_impl(path string, t BookTags, opts WriteOptions) !WriteReport
}
}
// Тэг (без запаса), дополненный нулями ровно до размера старого тэга; none — не помещается
fn fit_tag(parsed ParsedTag, tag []u8) ?[]u8 {
if parsed.version !in [u8(3), 4] || parsed.audio_start < 10 {
return none
}
body := tag.len - book_tag_padding
if body < 10 || u64(body) > parsed.audio_start {
return none
}
mut out := []u8{cap: int(parsed.audio_start)}
out << tag[..6]
out << encode_header_size(parsed.audio_start - 10)
out << tag[10..body]
out << []u8{len: int(parsed.audio_start) - body}
return out
}
// Перезапись начала файла тем же числом байт; false — файл не открылся на запись
fn write_header(path string, data []u8) bool {
mut f := os.open_file(path, 'r+') or { return false }
defer {
f.close()
}
f.write(data) or { return false }
f.flush()
return true
}
// Содержимое from — в существующий файл to (с обрезкой), не заменяя сам файл
fn copy_back(from string, to string, mut buf []u8) ! {
mut src := os.open(from)!
+76
View File
@@ -151,3 +151,79 @@ fn test_write_keeps_inode() {
t := metadata.read_book_tags(path) or { panic(err) }
assert t.tags.title == 'Тот же файл'
}
// Повторная запись: тэг помещается на место старого — переписывается только он, аудио не меняется
fn test_rewrite_header_only() {
path := tmp_copy('test_id3v23.mp3')
defer {
os.rm(path) or {}
}
original := os.read_bytes(path) or { panic(err) }
tail := 2048
audio := original[original.len - tail..].clone()
// Описание больше старого тэга — полная перезапись, в тэге остаётся запас
first := metadata.write_book_tags_report(path, metadata.BookTags{
title: 'Первая запись'
comment: 'описание '.repeat(300)
}) or { panic(err) }
assert first.in_place
assert !first.header
size := os.file_size(path)
before := os.stat(path) or { panic(err) }
// Правка в пределах запаса — только тэг
second := metadata.write_book_tags_report(path, metadata.BookTags{
title: 'Вторая запись, длиннее первой'
album: 'Книга'
comment: 'описание '.repeat(300) + 'и ещё немного'
}) or { panic(err) }
assert second.header
assert os.file_size(path) == size
after := os.stat(path) or { panic(err) }
assert after.inode == before.inode
t := metadata.read_book_tags(path) or { panic(err) }
assert t.tags.title == 'Вторая запись, длиннее первой'
assert t.tags.album == 'Книга'
assert t.tags.comment.ends_with('и ещё немного')
now := os.read_bytes(path) or { panic(err) }
assert now[now.len - tail..] == audio
// Не помещается — снова полная перезапись
third := metadata.write_book_tags_report(path, metadata.BookTags{
title: 'Третья'
comment: 'длинное описание '.repeat(1000)
}) or { panic(err) }
assert !third.header
t3 := metadata.read_book_tags(path) or { panic(err) }
assert t3.tags.title == 'Третья'
end := os.read_bytes(path) or { panic(err) }
assert end[end.len - tail..] == audio
}
// Испорченное название в ID3v2 («�») — берётся исходное из ID3v1 (cp1251)
fn test_broken_v2_title_falls_back_to_v1() {
path := tmp_copy('test_id3v23.mp3')
defer {
os.rm(path) or {}
}
metadata.write_book_tags(path, metadata.BookTags{
title: '\uFFFD\uFFFD\uFFFD 1'
album: 'Книга'
}) or { panic(err) }
// ID3v1 в конце файла: «Глава 1» в cp1251
mut v1 := []u8{len: 128}
v1[0] = `T`
v1[1] = `A`
v1[2] = `G`
title := [u8(0xc3), 0xeb, 0xe0, 0xe2, 0xe0, 0x20, 0x31]
for i, c in title {
v1[3 + i] = c
}
mut f := os.open_append(path) or { panic(err) }
f.write(v1) or { panic(err) }
f.close()
t := metadata.read_book_tags(path) or { panic(err) }
assert t.tags.title == 'Глава 1'
assert t.tags.album == 'Книга'
}