*исправление ошибок: двой перезаписи, исправление записи с BOM, добавлена проверка в insert_tags_into_audio
This commit is contained in:
1 parent
ffa8d80875
commit
078e68efa8
5 files changed
+181
-55
No files matched your search
+49
-5
@@ -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) {
|
||||
|
||||
Reference in new issue
Block a user