110 lines
3.3 KiB
V
110 lines
3.3 KiB
V
module metadata
|
|
|
|
import encoding.iconv
|
|
|
|
@[heap]
|
|
pub struct Frame {
|
|
mut:
|
|
size u64
|
|
start u64
|
|
end u64
|
|
encoding_name string
|
|
data string
|
|
name string
|
|
encoding_name_idx u8
|
|
addition []u8
|
|
encoded_data []u8
|
|
apic ?[]u8
|
|
encoded_size []u8
|
|
modified bool
|
|
}
|
|
|
|
// Во врейме первый байт указывает кодировку 0 - ISO-8859-1 (Latin-1), 1 - UTF-16LE
|
|
@[inline]
|
|
fn (mut frame Frame) fill_frame(bytes []u8, start_position u64, idv3_version u8) {
|
|
// Начало фрейма
|
|
frame.start = start_position
|
|
// Название фрейма
|
|
frame.name = bytes[frame.start..frame.start + 4].bytestr()
|
|
|
|
// Размер фрейма
|
|
frame.size = if idv3_version == 3 {
|
|
get_size_v23(bytes[frame.start+4..frame.start+8])
|
|
} else {
|
|
get_size_v24(bytes[frame.start+4..frame.start+8])
|
|
}
|
|
// Конец фрейма
|
|
frame.end = frame.start + frame.size + 10
|
|
|
|
// Содержимое фрейма
|
|
frame.encoding_name = match bytes[frame.start+10] {
|
|
0 { "ISO-8859-1" }
|
|
1 { "UTF-16LE" }
|
|
2 { "UTF-16BE" }
|
|
else { "UTF-8" }
|
|
}
|
|
|
|
frame.encoding_name_idx = u8(bytes[frame.start+10])
|
|
|
|
if frame.name.to_lower() == "apic" {
|
|
// Дополнительные байты с информацией (MIME)
|
|
frame.addition = bytes[frame.start+10..frame.start+24]
|
|
frame.apic = bytes[frame.start+24..frame.end]
|
|
} else {
|
|
// Выбор шага в записимости от версии ID3
|
|
stride := if idv3_version == 3 { 13 } else { 11 }
|
|
frame.data = iconv.encoding_to_vstring(bytes[frame.start+u64(stride)..frame.end], frame.encoding_name) or { "" }
|
|
|
|
frame.addition = bytes[frame.start..frame.start + 10]
|
|
}
|
|
}
|
|
|
|
// Кодируем длину фрейма с нужным сдвигом
|
|
@[inline]
|
|
pub fn (mut frame Frame) encode_size(idv3_version u8) {
|
|
stride := if idv3_version == 3 {
|
|
[u64(24), u64(16), u64(8), u64(0xFF)]
|
|
} else {
|
|
[u64(21), u64(14), u64(7), u64(0x7F)]
|
|
}
|
|
|
|
mut encoded_size := []u8{len: 4}
|
|
encoded_size[0] = u8((frame.size >> stride[0]) & stride[3])
|
|
encoded_size[1] = u8((frame.size >> stride[1]) & stride[3])
|
|
encoded_size[2] = u8((frame.size >> stride[2]) & stride[3])
|
|
encoded_size[3] = u8(frame.size & stride[3])
|
|
|
|
frame.encoded_size = encoded_size
|
|
}
|
|
|
|
// Кодируем кадр для дальнейшей записи
|
|
pub fn (mut frame Frame) encode_frame(idv3_version u8) {
|
|
encoded_data := iconv.vstring_to_encoding(frame.data, frame.encoding_name) or { [u8(0), u8(0), u8(0), u8(0)] }
|
|
|
|
frame.encoded_data = encoded_data
|
|
|
|
//
|
|
frame.size = u64(encoded_data.len + 10)
|
|
|
|
frame.encode_size(idv3_version)
|
|
}
|
|
|
|
// Получить закодированый тэг
|
|
pub fn (mut frame Frame) get_encoded_frame_tag(idv3_version u8) []u8 {
|
|
frame.encode_frame(idv3_version)
|
|
|
|
// Название тэга
|
|
mut encoded_tag := frame.name.bytes()
|
|
// Флаги, оставляем пустыми
|
|
encoded_tag << []u8{len: 2}
|
|
// Идекс кодировщика текста
|
|
encoded_tag << frame.encoding_name_idx
|
|
// Четыре числа на закодированный размер
|
|
encoded_tag << frame.encoded_size
|
|
// Уже сами данные
|
|
encoded_tag << frame.encoded_data
|
|
// Запас нулей чтобы не пересчитывать весь фрейм при маленьком редактировании
|
|
encoded_tag << []u8{len: 10}
|
|
|
|
return encoded_tag
|
|
} |