diff --git a/metadata/frame.v b/metadata/frame.v index cf9d07e..95f015a 100644 --- a/metadata/frame.v +++ b/metadata/frame.v @@ -51,11 +51,13 @@ fn (mut frame Frame) fill(bytes []u8, start_position u64, idv3_version u8) { if frame.name.to_lower() == "apic" { // Дополнительные байты с информацией (MIME) frame.addition = bytes[frame.start + 10..frame.start + 24] - frame.apic = bytes[frame.start + 24..frame.end] + // Фрейм целиком вместе с заголовком, чтобы записать его обратно без изменений + frame.apic = bytes[frame.start..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.data = (iconv.encoding_to_vstring(bytes[frame.start+u64(stride)..frame.end], frame.encoding_name) or { "" }).trim_right('\x00') frame.addition = bytes[frame.start..frame.start + 10] } @@ -88,14 +90,23 @@ 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(0), u8(0), u8(0), u8(0)] } + encoded_data := iconv.vstring_to_encoding(frame.data, frame.encoding_name) or { []u8{} } frame.encoded_data = encoded_data + // Размер тела фрейма: байт кодировки + данные + замыкающие нули + frame.cur_size = u64(1 + frame.encoded_data.len + frame.terminator_len()) + // число -> [байт, байт, байт, байт] frame.encode_size(idv3_version) } +// Для UTF-16 строка завершается двумя нулями, для остальных кодировок одним +@[inline] +fn (frame Frame) terminator_len() int { + return if frame.encoding_name_idx == 1 || frame.encoding_name_idx == 2 { 2 } else { 1 } +} + // Получить закодированый тэг pub fn (mut frame Frame) get_encoded_frame_tag(idv3_version u8) []u8 { frame.encode_frame(idv3_version) @@ -112,12 +123,8 @@ pub fn (mut frame Frame) get_encoded_frame_tag(idv3_version u8) []u8 { encoded_tag << frame.encoding_name_idx // Уже сами данные encoded_tag << frame.encoded_data - // Запас нулей чтобы не пересчитывать весь фрейм при маленьком редактировании - encoded_tag << []u8{len: 10} - // Замыкающий байт - encoded_tag << []u8{len: 1} - - frame.cur_size = u64(encoded_tag.len) + // Замыкающие нули + encoded_tag << []u8{len: frame.terminator_len()} return encoded_tag } diff --git a/metadata/id3v24.v b/metadata/id3v24.v index 6363fd1..7ffecfa 100644 --- a/metadata/id3v24.v +++ b/metadata/id3v24.v @@ -70,39 +70,13 @@ pub fn (id3v24 ID3v24) to_v1() ID3v1 { } } -// Формируем полный полный тэг в байтах -pub fn (mut id3v24 ID3v24) get_full_embed_tag() { - // Проход по всем полям стуктуры - mut tag_bytes := []u8{} - - id3v24.size = 0 - - $for field in ID3v24.fields { - $if field.typ is Frame { - // Читаются только заполненные тэги - if id3v24.$(field.name).cur_size > 0 { - id3v24.size += id3v24.$(field.name).cur_size - id3v24.size += 10 - - if id3v24.$(field.name).name.to_lower() != "apic" { - // Кодируем значения полей - tag_bytes << id3v24.$(field.name).get_encoded_frame_tag(id3v24.version) - } else { - // Картинка не трогается - tag_bytes << id3v24.$(field.name).apic or { []u8{len: 4}} - } - } - } - } - encode_header(mut id3v24) - id3v24.encoded_header << tag_bytes -} - pub fn (mut id3v24 ID3v24) insert_tags_into_audio(mut audio File) { - full_part := read_all(reader: audio) or { [u8(0), u8(0), u8(0), u8(0)] } + full_part := read_all(reader: audio) or { panic(err) } - sound_part := full_part[int(id3v24.size) + 10 .. ] + // Размер старого тэга берётся из файла, id3v24.size уже хранит размер нового + old_size := get_size_v24(full_part[6..10]) + sound_part := full_part[int(old_size) + 10 .. ] id3v24.encoded_header << sound_part } diff --git a/metadata/utils.v b/metadata/utils.v index 42d7cab..f5ad894 100644 --- a/metadata/utils.v +++ b/metadata/utils.v @@ -49,10 +49,17 @@ pub fn dump_data[T](mut tag T, mut audio os.File) { // Получаем вне возможные байты all_tag_bytes := get_bytes(mut audio, 10, int(tag.size + start_position)) - for tag.size > start_position { + // Конец тэга с учётом 10 байт заголовка + tag_end := tag.size + 10 + + for start_position + 10 <= tag_end { + // Нулевой байт вместо названия фрейма — начался запас нулей + if all_tag_bytes[start_position] == 0 { + break + } + current_frame.fill(all_tag_bytes, start_position, tag.version) start_position = current_frame.end - tag.size -= current_frame.cur_size tag.add_frame(current_frame) } @@ -95,3 +102,37 @@ pub fn encode_header[T](mut id3v T) { id3v.encoded_header << metadata.encode_header_size(id3v.size, id3v.version) } +// Количество нулей в конце тэга +const tag_padding = 256 + +type ID3vType = ID3v23 | ID3v24 + +// Формируем полный полный тэг в байтах +pub fn get_full_embed_tag[ID3vType](mut id3v2 ID3vType) { + // Проход по всем полям стуктуры + mut tag_bytes := []u8{} + + $for field in id3v2.fields { + $if field.typ is Frame { + // Читаются только заполненные тэги + if id3v2.$(field.name).cur_size > 0 { + if id3v2.$(field.name).name.to_lower() != "apic" { + // Кодируем значения полей + tag_bytes << id3v2.$(field.name).get_encoded_frame_tag(id3v2.version) + } else { + // Картинка не трогается + tag_bytes << id3v2.$(field.name).apic or { []u8{} } + } + } + } + } + // Запас нулей чтобы не пересчитывать весь тэг при маленьком редактировании + tag_bytes << []u8{len: tag_padding} + + // Размер считается по уже закодированным фреймам + id3v2.size = u64(tag_bytes.len) + + id3v2.encoded_header = []u8{} + encode_header(mut id3v2) + id3v2.encoded_header << tag_bytes +} \ No newline at end of file