1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
/*
* Copyright (c) 2021, Cesar Torres <shortanemoia@protonmail.com>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#include "M3UParser.h"
#include <AK/NonnullOwnPtrVector.h>
#include <AK/OwnPtr.h>
#include <AK/RefPtr.h>
#include <AK/Utf8View.h>
#include <LibCore/FileStream.h>
#include <ctype.h>
M3UParser::M3UParser()
{
}
NonnullOwnPtr<M3UParser> M3UParser::from_file(const String path)
{
auto parser = make<M3UParser>();
VERIFY(!path.is_null() && !path.is_empty() && !path.is_whitespace());
parser->m_use_utf8 = path.ends_with(".m3u8", AK::CaseSensitivity::CaseInsensitive);
FILE* file = fopen(path.characters(), "r");
VERIFY(file != nullptr);
fseek(file, 0, SEEK_END);
size_t file_size = ftell(file);
fseek(file, 0, SEEK_SET);
VERIFY(file_size > 0);
Vector<u8> temp_buffer;
temp_buffer.resize(file_size);
auto bytes_read = fread(temp_buffer.data(), 1, file_size, file);
VERIFY(bytes_read == file_size);
parser->m_m3u_raw_data = *new String(temp_buffer.span(), NoChomp);
return parser;
}
NonnullOwnPtr<M3UParser> M3UParser::from_memory(const String& m3u_contents, bool utf8)
{
auto parser = make<M3UParser>();
VERIFY(!m3u_contents.is_null() && !m3u_contents.is_empty() && !m3u_contents.is_whitespace());
parser->m_m3u_raw_data = m3u_contents;
parser->m_use_utf8 = utf8;
return parser;
}
NonnullOwnPtr<Vector<M3UEntry>> M3UParser::parse(bool include_extended_info)
{
auto vec = make<Vector<M3UEntry>>();
bool has_exteded_info_tag = false;
if (!m_use_utf8) {
auto lines = m_m3u_raw_data.split_view('\n');
if (include_extended_info) {
if (lines[0] == "#EXTM3U")
has_exteded_info_tag = true;
}
M3UExtendedInfo metadata_for_next_file {};
for (auto& line : lines) {
line = line.trim_whitespace();
M3UEntry entry {};
if (line.starts_with('#') && has_exteded_info_tag) {
if (line.starts_with("#EXTINF")) {
auto data = line.substring_view(8);
auto separator = data.find_first_of(',');
VERIFY(separator.has_value());
auto seconds = data.substring_view(0, separator.value());
VERIFY(!seconds.is_whitespace() && !seconds.is_null() && !seconds.is_empty());
metadata_for_next_file.track_length_in_seconds = seconds.to_uint();
auto display_name = data.substring_view(seconds.length() + 1);
VERIFY(!display_name.is_empty() && !display_name.is_null() && !display_name.is_empty());
metadata_for_next_file.track_display_title = display_name;
//TODO: support the alternative, non-standard #EXTINF value of a key=value dictionary
} else if (line.starts_with("#PLAYLIST")) {
auto name = line.substring_view(10);
VERIFY(!name.is_empty());
m_parsed_playlist_title = name;
} else if (line.starts_with("#EXTGRP")) {
auto name = line.substring_view(8);
VERIFY(!name.is_empty());
metadata_for_next_file.group_name = name;
} else if (line.starts_with("#EXTALB")) {
auto name = line.substring_view(8);
VERIFY(!name.is_empty());
metadata_for_next_file.album_title = name;
} else if (line.starts_with("#EXTART")) {
auto name = line.substring_view(8);
VERIFY(!name.is_empty());
metadata_for_next_file.album_artist = name;
} else if (line.starts_with("#EXTGENRE")) {
auto name = line.substring_view(10);
VERIFY(!name.is_empty());
metadata_for_next_file.album_genre = name;
}
//TODO: Support M3A files (M3U files with embedded mp3 files)
} else {
entry.path = line;
entry.extended_info = metadata_for_next_file;
vec->append(entry);
}
}
} else {
//TODO: Implement M3U8 parsing
TODO();
}
return vec;
}
|
