*исправление ошибок: двой перезаписи, исправление записи с BOM, добавлена проверка в insert_tags_into_audio

This commit is contained in:
K1ndWha1e committed 2026-10-07 23:24:01 +03:00
1 parent ffa8d80875
commit 078e68efa8
5 files changed
+181 -55

No files matched your search

+49 -5
View File
@@ -19,6 +19,8 @@ pub struct Frame {
apic ?[]u8
encoded_size []u8
modified bool
// Фрейм целиком в том виде, в каком он был прочитан из файла
raw []u8
}
// Во фрейме первый байт указывает кодировку 0 - ISO-8859-1 (Latin-1), 1 - UTF-16LE
@@ -38,6 +40,9 @@ fn (mut frame Frame) fill(bytes []u8, start_position u64, idv3_version u8) {
// Конец фрейма
frame.end = frame.start + frame.cur_size + 10
frame.raw = bytes[frame.start..frame.end]
frame.modified = false
// Содержимое фрейма
frame.encoding_name = match bytes[frame.start + 10] {
0 { "ISO-8859-1" }
@@ -72,11 +77,21 @@ fn (mut frame Frame) fill(bytes []u8, start_position u64, idv3_version u8) {
}
}
pub fn (mut frame Frame) init(tag_name string) {
pub fn (mut frame Frame) init(tag_name string, idv3_version u8) {
frame.name = tag_name.to_upper()
frame.set_unicode_encoding(idv3_version)
}
frame.encoding_name = "UTF-8"
frame.encoding_name_idx = 3
// Кодировка, в которой представим любой текст
fn (mut frame Frame) set_unicode_encoding(idv3_version u8) {
// В v2.3 нет UTF-8, там Юникод только UTF-16 с BOM
if idv3_version == 3 {
frame.encoding_name = "UTF-16LE"
frame.encoding_name_idx = 1
} else {
frame.encoding_name = "UTF-8"
frame.encoding_name_idx = 3
}
}
// Кодируем длину фрейма с нужным сдвигом
@@ -99,9 +114,27 @@ pub fn (mut frame Frame) encode_size(idv3_version u8) {
// Кодируем кадр для дальнейшей записи
pub fn (mut frame Frame) encode_frame(idv3_version u8) {
encoded_data := iconv.vstring_to_encoding(frame.data, frame.encoding_name) or { []u8{} }
// В ISO-8859-1 есть только символы до U+00FF, остальные iconv молча заменяет на '?'
if frame.encoding_name_idx == 0 && frame.data.runes().any(it > 0xFF) {
frame.set_unicode_encoding(idv3_version)
}
frame.encoded_data = encoded_data
encoded_data := iconv.vstring_to_encoding(frame.data, frame.encoding_name) or {
// Не удалось закодировать в исходной кодировке фрейма, переходим на Юникод
frame.set_unicode_encoding(idv3_version)
iconv.vstring_to_encoding(frame.data, frame.encoding_name) or {
panic("не удалось закодировать фрейм ${frame.name}: ${err}")
}
}
// Для UTF-16 с BOM (01) перед текстом нужен маркер порядка байт
frame.encoded_data = if frame.encoding_name_idx == 1 {
mut bom := if frame.encoding_name == "UTF-16BE" { [u8(0xFE), 0xFF] } else { [u8(0xFF), 0xFE] }
bom << encoded_data
bom
} else {
encoded_data
}
// Размер тела фрейма: байт кодировки + данные + замыкающие нули
frame.cur_size = u64(1 + frame.encoded_data.len + frame.terminator_len())
@@ -137,6 +170,17 @@ pub fn (mut frame Frame) get_encoded_frame_tag(idv3_version u8) []u8 {
return encoded_tag
}
// Байты фрейма для записи в файл
pub fn (mut frame Frame) to_bytes(idv3_version u8) []u8 {
// Картинка и нетронутые фреймы пишутся как были прочитаны,
// чтобы не потерять то, что не разбирается (язык в COMM, значение в TXXX, бинарные фреймы)
if frame.raw.len > 0 && (!frame.modified || frame.name == "APIC") {
return frame.raw
}
return frame.get_encoded_frame_tag(idv3_version)
}
// Запись данных в тэг
@[inline]
pub fn (mut frame Frame) set_data(data string) {