aboutsummaryrefslogtreecommitdiff
path: root/src/plugins/utils
diff options
context:
space:
mode:
authordec05eba <dec05eba@protonmail.com>2023-01-21 00:49:37 +0100
committerdec05eba <dec05eba@protonmail.com>2023-01-21 00:49:37 +0100
commit62ba1032657a0bf3878a5fe77018633adab8767d (patch)
treeaea387d6a293428b20b402ccb579a49441ec12bd /src/plugins/utils
parent760d903fe7c7c56a6408accff0daa47c2e743eed (diff)
NyaaSi: exclude episode number from track name, add extension
Diffstat (limited to 'src/plugins/utils')
-rw-r--r--src/plugins/utils/EpisodeNameParser.cpp25
1 files changed, 25 insertions, 0 deletions
diff --git a/src/plugins/utils/EpisodeNameParser.cpp b/src/plugins/utils/EpisodeNameParser.cpp
index a381cca..e314953 100644
--- a/src/plugins/utils/EpisodeNameParser.cpp
+++ b/src/plugins/utils/EpisodeNameParser.cpp
@@ -90,6 +90,10 @@ namespace QuickMedia {
episode_name.remove_prefix(i + 1);
return episode;
}
+
+ static bool ends_with(std::string_view str, std::string_view substr) {
+ return str.size() >= substr.size() && str.substr(str.size() - substr.size()) == substr;
+ }
std::optional<EpisodeNameParts> episode_name_extract_parts(std::string_view episode_name) {
EpisodeNameParts name_parts;
@@ -122,6 +126,27 @@ namespace QuickMedia {
else if(episode_name.find("3840x2160") != std::string_view::npos)
name_parts.resolution = "3840x2160";
+ if(ends_with(episode_name, ".mkv"))
+ name_parts.file_ext = ".mkv";
+ else if(ends_with(episode_name, ".mp4"))
+ name_parts.file_ext = ".mp4";
+ else if(ends_with(episode_name, ".avi"))
+ name_parts.file_ext = ".avi";
+ else if(ends_with(episode_name, ".webm"))
+ name_parts.file_ext = ".webm";
+ else if(ends_with(episode_name, ".wmv"))
+ name_parts.file_ext = ".wmv";
+ else if(ends_with(episode_name, ".mp3"))
+ name_parts.file_ext = ".mp3";
+ else if(ends_with(episode_name, ".flac"))
+ name_parts.file_ext = ".flac";
+ else if(ends_with(episode_name, ".opus"))
+ name_parts.file_ext = ".opus";
+ else if(ends_with(episode_name, ".wav"))
+ name_parts.file_ext = ".wav";
+ else if(ends_with(episode_name, ".ogg"))
+ name_parts.file_ext = ".ogg";
+
return name_parts;
}
} \ No newline at end of file