Limit to the first 4 digits when parsing years badly encoded as YYYYMMDD

This commit is contained in:
emeric
2024-02-02 22:58:37 +01:00
parent d3aa11714a
commit 5db91fc5b5
5 changed files with 78 additions and 4 deletions
+4 -4
View File
@@ -336,19 +336,19 @@ namespace MetaData
if (const Wt::WDate date{ Utils::parseDate(value) }; date.isValid())
track.date = date;
else if (!track.year)
track.year = StringUtils::readAs<int>(value);
track.year = Utils::parseYear(value);
}
else if (tag == "YEAR")
track.year = StringUtils::readAs<int>(value);
track.year = Utils::parseYear(value);
else if (tag == "ORIGINALDATE")
{
if (const Wt::WDate date{ Utils::parseDate(value) }; date.isValid())
track.originalDate = date;
else if (!track.originalYear)
track.originalYear = StringUtils::readAs<int>(value);
track.originalYear = Utils::parseYear(value);
}
else if (tag == "ORIGINALYEAR")
track.originalYear = StringUtils::readAs<int>(value);
track.originalYear = Utils::parseYear(value);
else if (tag == "METADATA_BLOCK_PICTURE")
track.hasCover = true;
else if (tag == "COPYRIGHT")
+32
View File
@@ -64,6 +64,38 @@ namespace MetaData::Utils
return {};
}
std::optional<int> parseYear(std::string_view yearStr)
{
// limit to first 4 digit, accept leading '-'
if (yearStr.empty())
return std::nullopt;
int sign;
if (yearStr.front() == '-')
{
sign = -1;
yearStr.remove_prefix(1);
}
else
{
sign = 1;
}
if (yearStr.empty() || !std::isdigit(yearStr.front()))
return std::nullopt;
int result{};
for (std::size_t i{}; i < yearStr.size() && i < 4; ++i)
{
if (!std::isdigit(yearStr[i])) {
break;
}
result = result * 10 + (yearStr[i] - '0');
}
return result * sign;
}
std::string_view readStyleToString(ParserReadStyle readStyle)
{
switch (readStyle)
+2
View File
@@ -20,6 +20,7 @@
#pragma once
#include <optional>
#include <string_view>
#include <Wt/WDate.h>
@@ -28,6 +29,7 @@
namespace MetaData::Utils
{
Wt::WDate parseDate(std::string_view dateStr);
std::optional<int> parseYear(std::string_view yearStr);
std::string_view readStyleToString(ParserReadStyle readStyle);
struct PerformerArtist
+37
View File
@@ -76,6 +76,43 @@ TEST(MetaData, parseDate)
}
}
TEST(MetaData, parseYear)
{
using namespace MetaData::Utils;
struct TestCase
{
std::string str;
std::optional<int> result;
} testCases[]
{
{ "1995-05-09", 1995 },
{ "1995", 1995 },
{ "-0", 0 },
{ "0", 0 },
{ "00", 0 },
{ "05", 5 },
{ "050", 50 },
{ "00005", 0 },
{ "-50", -50 },
{ "-", std::nullopt },
{ "", std::nullopt },
{ "a", std::nullopt },
{ "1a", 1 },
{ "12a", 12 },
{ "123a", 123 },
{ "1234a", 1234 },
{ "19951123", 1995 },
{ "199511", 1995 },
};
for (const TestCase& testCase : testCases)
{
const std::optional<int> parsed{ parseYear(testCase.str) };
EXPECT_EQ(parsed, testCase.result) << " str was '" << testCase.str << "'";
}
}
TEST(MetaData, extractPerformerAndRole)
{
using namespace MetaData::Utils;
+3
View File
@@ -173,9 +173,12 @@ TEST(StringUtils, readAs_int)
EXPECT_EQ(StringUtils::readAs<int>("-1"), -1);
EXPECT_EQ(StringUtils::readAs<int>(""), std::nullopt);
EXPECT_EQ(StringUtils::readAs<int>("a"), std::nullopt);
EXPECT_EQ(StringUtils::readAs<int>("-"), std::nullopt);
EXPECT_EQ(StringUtils::readAs<int>("1024-1"), 1024);
EXPECT_EQ(StringUtils::readAs<int>("1024-"), 1024);
EXPECT_EQ(StringUtils::readAs<int>("1024/5"), 1024);
EXPECT_EQ(StringUtils::readAs<int>("1024a"), 1024);
EXPECT_EQ(StringUtils::readAs<int>("a1024a"), std::nullopt);
}
TEST(StringUtils, capitalize)