*повторная запись: если новый тэг помещается на место старого — переписывается только он, аудио не трогается; запас в тэге 4 КБ
Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
This commit is contained in:
1 parent
b037345b7b
commit
a3e244205a
2 files changed
+94
-1
No files matched your search
+45
-1
@@ -64,6 +64,8 @@ pub:
|
|||||||
pub struct WriteReport {
|
pub struct WriteReport {
|
||||||
pub:
|
pub:
|
||||||
in_place bool
|
in_place bool
|
||||||
|
// Переписан только тэг в начале файла (новый поместился на место старого), аудио не тронуто
|
||||||
|
header bool
|
||||||
// Почему не удалось переписать на месте (пусто, если на месте)
|
// Почему не удалось переписать на месте (пусто, если на месте)
|
||||||
fallback string
|
fallback string
|
||||||
}
|
}
|
||||||
@@ -75,6 +77,10 @@ mut:
|
|||||||
audio_start u64
|
audio_start u64
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Запас в конце тэга при полной перезаписи: следующие правки (длиннее описание, другие названия)
|
||||||
|
// поместятся на место старого тэга, и аудио переписывать не придётся
|
||||||
|
const book_tag_padding = 4096
|
||||||
|
|
||||||
// Фреймы, которые редактор пересобирает сам, остальные переносятся без изменений
|
// Фреймы, которые редактор пересобирает сам, остальные переносятся без изменений
|
||||||
const managed_frames = ['TIT2', 'TALB', 'TPE1', 'TPE2', 'TCOM', 'TIT1', 'TRCK', 'TYER', 'TDRC',
|
const managed_frames = ['TIT2', 'TALB', 'TPE1', 'TPE2', 'TCOM', 'TIT1', 'TRCK', 'TYER', 'TDRC',
|
||||||
'TCON', 'COMM', 'TLAN']
|
'TCON', 'COMM', 'TLAN']
|
||||||
@@ -558,7 +564,7 @@ fn build_tag(parsed ParsedTag, t BookTags, opts WriteOptions) []u8 {
|
|||||||
frames << apic_frame(c, version)
|
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]
|
mut out := [u8(`I`), `D`, `3`, version, 0, 0]
|
||||||
out << encode_header_size(u64(frames.len))
|
out << encode_header_size(u64(frames.len))
|
||||||
@@ -574,6 +580,16 @@ fn write_book_tags_impl(path string, t BookTags, opts WriteOptions) !WriteReport
|
|||||||
parsed := parse_id3v2(path)!
|
parsed := parse_id3v2(path)!
|
||||||
tag := build_tag(parsed, t, opts)
|
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'
|
tmp_path := path + '.tagtmp'
|
||||||
mut buf := []u8{len: 1024 * 1024}
|
mut buf := []u8{len: 1024 * 1024}
|
||||||
{
|
{
|
||||||
@@ -624,6 +640,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 (с обрезкой), не заменяя сам файл
|
// Содержимое from — в существующий файл to (с обрезкой), не заменяя сам файл
|
||||||
fn copy_back(from string, to string, mut buf []u8) ! {
|
fn copy_back(from string, to string, mut buf []u8) ! {
|
||||||
mut src := os.open(from)!
|
mut src := os.open(from)!
|
||||||
|
|||||||
@@ -151,3 +151,52 @@ fn test_write_keeps_inode() {
|
|||||||
t := metadata.read_book_tags(path) or { panic(err) }
|
t := metadata.read_book_tags(path) or { panic(err) }
|
||||||
assert t.tags.title == 'Тот же файл'
|
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
|
||||||
|
}
|
||||||
Reference in new issue
Block a user