Allow wildcards in file names to search for images, fixes #697
This commit is contained in:
@@ -23,7 +23,7 @@
|
||||
|
||||
namespace lms::db::utils
|
||||
{
|
||||
std::string escapeLikeKeyword(std::string_view keyword)
|
||||
std::string escapeForLikeKeyword(std::string_view keyword)
|
||||
{
|
||||
return core::stringUtils::escapeString(keyword, "%_", escapeChar);
|
||||
}
|
||||
|
||||
@@ -38,7 +38,7 @@ namespace lms::db::utils
|
||||
{
|
||||
#define ESCAPE_CHAR_STR "\\"
|
||||
static inline constexpr char escapeChar{ '\\' };
|
||||
std::string escapeLikeKeyword(std::string_view keywords);
|
||||
std::string escapeForLikeKeyword(std::string_view keywords);
|
||||
|
||||
Wt::WDateTime normalizeDateTime(const Wt::WDateTime& dateTime);
|
||||
|
||||
|
||||
@@ -106,13 +106,13 @@ namespace lms::db
|
||||
for (const std::string_view keyword : params.keywords)
|
||||
{
|
||||
clauses.push_back("a.name LIKE ? ESCAPE '" ESCAPE_CHAR_STR "'");
|
||||
query.bind("%" + utils::escapeLikeKeyword(keyword) + "%");
|
||||
query.bind("%" + utils::escapeForLikeKeyword(keyword) + "%");
|
||||
}
|
||||
|
||||
for (const std::string_view keyword : params.keywords)
|
||||
{
|
||||
sortClauses.push_back("a.sort_name LIKE ? ESCAPE '" ESCAPE_CHAR_STR "'");
|
||||
query.bind("%" + utils::escapeLikeKeyword(keyword) + "%");
|
||||
query.bind("%" + utils::escapeForLikeKeyword(keyword) + "%");
|
||||
}
|
||||
|
||||
query.where("(" + core::stringUtils::joinStrings(clauses, " AND ") + ") OR (" + core::stringUtils::joinStrings(sortClauses, " AND ") + ")");
|
||||
|
||||
@@ -40,7 +40,7 @@ namespace lms::db
|
||||
auto query{ session.getDboSession()->query<Wt::Dbo::ptr<Directory>>("SELECT d FROM directory d") };
|
||||
|
||||
for (std::string_view keyword : params.keywords)
|
||||
query.where("d.name LIKE ? ESCAPE '" ESCAPE_CHAR_STR "'").bind("%" + utils::escapeLikeKeyword(keyword) + "%");
|
||||
query.where("d.name LIKE ? ESCAPE '" ESCAPE_CHAR_STR "'").bind("%" + utils::escapeForLikeKeyword(keyword) + "%");
|
||||
|
||||
if (params.artist.isValid()
|
||||
|| params.release.isValid()
|
||||
|
||||
@@ -43,7 +43,20 @@ namespace lms::db
|
||||
if (params.directory.isValid())
|
||||
query.where("i.directory_id = ?").bind(params.directory);
|
||||
if (!params.fileStem.empty())
|
||||
query.where("i.stem = ? COLLATE NOCASE").bind(params.fileStem);
|
||||
{
|
||||
// if contains a wildcard, replace by % for LIKE
|
||||
if (params.processWildcardsInFileStem.value() && params.fileStem.find('*') != std::string::npos)
|
||||
{
|
||||
std::string fileStem{ params.fileStem };
|
||||
utils::escapeForLikeKeyword(fileStem);
|
||||
std::replace(std::begin(fileStem), std::end(fileStem), '*', '%');
|
||||
query.where("i.stem LIKE ? COLLATE NOCASE ESCAPE '" ESCAPE_CHAR_STR "'").bind(fileStem);
|
||||
}
|
||||
else
|
||||
{
|
||||
query.where("i.stem = ? COLLATE NOCASE").bind(params.fileStem);
|
||||
}
|
||||
}
|
||||
|
||||
return query;
|
||||
}
|
||||
|
||||
@@ -100,13 +100,13 @@ namespace lms::db
|
||||
for (const std::string_view keyword : params.keywords)
|
||||
{
|
||||
clauses.push_back("a.name LIKE ? ESCAPE '" ESCAPE_CHAR_STR "'");
|
||||
query.bind("%" + utils::escapeLikeKeyword(keyword) + "%");
|
||||
query.bind("%" + utils::escapeForLikeKeyword(keyword) + "%");
|
||||
}
|
||||
|
||||
for (const std::string_view keyword : params.keywords)
|
||||
{
|
||||
sortClauses.push_back("a.sort_name LIKE ? ESCAPE '" ESCAPE_CHAR_STR "'");
|
||||
query.bind("%" + utils::escapeLikeKeyword(keyword) + "%");
|
||||
query.bind("%" + utils::escapeForLikeKeyword(keyword) + "%");
|
||||
}
|
||||
|
||||
query.where("(" + core::stringUtils::joinStrings(clauses, " AND ") + ") OR (" + core::stringUtils::joinStrings(sortClauses, " AND ") + ")");
|
||||
@@ -169,7 +169,7 @@ namespace lms::db
|
||||
}
|
||||
|
||||
for (std::string_view keyword : params.keywords)
|
||||
query.where("r.name LIKE ? ESCAPE '" ESCAPE_CHAR_STR "'").bind("%" + utils::escapeLikeKeyword(keyword) + "%");
|
||||
query.where("r.name LIKE ? ESCAPE '" ESCAPE_CHAR_STR "'").bind("%" + utils::escapeForLikeKeyword(keyword) + "%");
|
||||
|
||||
return query;
|
||||
}
|
||||
@@ -227,7 +227,7 @@ namespace lms::db
|
||||
}
|
||||
|
||||
for (std::string_view keyword : params.keywords)
|
||||
query.where("t.name LIKE ? ESCAPE '" ESCAPE_CHAR_STR "'").bind("%" + utils::escapeLikeKeyword(keyword) + "%");
|
||||
query.where("t.name LIKE ? ESCAPE '" ESCAPE_CHAR_STR "'").bind("%" + utils::escapeForLikeKeyword(keyword) + "%");
|
||||
|
||||
return query;
|
||||
}
|
||||
|
||||
@@ -125,7 +125,7 @@ namespace lms::db
|
||||
query.where("r.name = ?").bind(params.name);
|
||||
|
||||
for (std::string_view keyword : params.keywords)
|
||||
query.where("r.name LIKE ? ESCAPE '" ESCAPE_CHAR_STR "'").bind("%" + utils::escapeLikeKeyword(keyword) + "%");
|
||||
query.where("r.name LIKE ? ESCAPE '" ESCAPE_CHAR_STR "'").bind("%" + utils::escapeForLikeKeyword(keyword) + "%");
|
||||
|
||||
if (params.starringUser.isValid())
|
||||
{
|
||||
|
||||
@@ -61,7 +61,7 @@ namespace lms::db
|
||||
|
||||
assert(params.keywords.empty() || params.name.empty());
|
||||
for (std::string_view keyword : params.keywords)
|
||||
query.where("t.name LIKE ? ESCAPE '" ESCAPE_CHAR_STR "'").bind("%" + utils::escapeLikeKeyword(keyword) + "%");
|
||||
query.where("t.name LIKE ? ESCAPE '" ESCAPE_CHAR_STR "'").bind("%" + utils::escapeForLikeKeyword(keyword) + "%");
|
||||
|
||||
if (!params.name.empty())
|
||||
query.where("t.name = ?").bind(params.name);
|
||||
|
||||
@@ -57,7 +57,7 @@ namespace lms::db
|
||||
}
|
||||
|
||||
for (std::string_view keyword : params.keywords)
|
||||
query.where("t_l.name LIKE ? ESCAPE '" ESCAPE_CHAR_STR "'").bind("%" + utils::escapeLikeKeyword(keyword) + "%");
|
||||
query.where("t_l.name LIKE ? ESCAPE '" ESCAPE_CHAR_STR "'").bind("%" + utils::escapeForLikeKeyword(keyword) + "%");
|
||||
|
||||
if (params.filters.mediaLibrary.isValid()
|
||||
|| params.filters.label.isValid()
|
||||
|
||||
@@ -114,7 +114,7 @@ namespace lms::db
|
||||
static pointer find(Session& session, const std::filesystem::path& path);
|
||||
static void find(Session& session, DirectoryId& lastRetrievedDirectory, std::size_t count, const std::function<void(const Directory::pointer&)>& func);
|
||||
static RangeResults<Directory::pointer> find(Session& session, const FindParameters& params);
|
||||
static void find(Session& session, const FindParameters& parameters, const std::function<void(const Directory::pointer&)>& func);
|
||||
static void find(Session& session, const FindParameters& params, const std::function<void(const Directory::pointer&)>& func);
|
||||
static RangeResults<DirectoryId> findOrphanIds(Session& session, std::optional<Range> range = std::nullopt);
|
||||
static RangeResults<DirectoryId> findMismatchedLibrary(Session& session, std::optional<Range> range, const std::filesystem::path& rootPath, MediaLibraryId expectedLibraryId);
|
||||
static RangeResults<pointer> findRootDirectories(Session& session, std::optional<Range> range = std::nullopt);
|
||||
|
||||
@@ -26,6 +26,7 @@
|
||||
#include <Wt/Dbo/Field.h>
|
||||
#include <Wt/WDateTime.h>
|
||||
|
||||
#include "core/TaggedType.hpp"
|
||||
#include "database/Object.hpp"
|
||||
#include "database/Types.hpp"
|
||||
#include "database/objects/DirectoryId.hpp"
|
||||
@@ -43,18 +44,22 @@ namespace lms::db
|
||||
|
||||
struct FindParameters
|
||||
{
|
||||
using ProcessWildcards = core::TaggedBool<class FindParametersProcessWildcardsTag>;
|
||||
|
||||
std::optional<Range> range;
|
||||
std::string fileStem; // if set, images with this file stem
|
||||
DirectoryId directory; // if set, images in this directory
|
||||
std::string fileStem; // if set, images with this file stem
|
||||
ProcessWildcards processWildcardsInFileStem{ false }; // if true, replace '*' by '%' for SQL LIKE
|
||||
DirectoryId directory; // if set, images in this directory
|
||||
|
||||
FindParameters& setRange(std::optional<Range> _range)
|
||||
{
|
||||
range = _range;
|
||||
return *this;
|
||||
}
|
||||
FindParameters& setFileStem(std::string_view _fileStem)
|
||||
FindParameters& setFileStem(std::string_view _fileStem, ProcessWildcards processWildcards = ProcessWildcards{ false })
|
||||
{
|
||||
fileStem = _fileStem;
|
||||
processWildcardsInFileStem = processWildcards;
|
||||
return *this;
|
||||
}
|
||||
FindParameters& setDirectory(DirectoryId _directory)
|
||||
|
||||
@@ -122,4 +122,102 @@ namespace lms::db::tests
|
||||
}
|
||||
}
|
||||
|
||||
TEST_F(DatabaseFixture, Image_findByFileStem)
|
||||
{
|
||||
ScopedDirectory directory{ session, "/path/to" };
|
||||
ScopedImage image{ session, "/path/to/image" };
|
||||
|
||||
const std::string fileStem{ "image" };
|
||||
{
|
||||
auto transaction{ session.createWriteTransaction() };
|
||||
image.get().modify()->setDirectory(directory.get());
|
||||
}
|
||||
|
||||
{
|
||||
auto transaction{ session.createReadTransaction() };
|
||||
Image::FindParameters params;
|
||||
params.setFileStem(fileStem);
|
||||
const auto results{ Image::find(session, params) };
|
||||
ASSERT_EQ(results.results.size(), 1);
|
||||
EXPECT_EQ(results.results[0]->getId(), image.getId());
|
||||
}
|
||||
|
||||
{
|
||||
auto transaction{ session.createReadTransaction() };
|
||||
Image::FindParameters params;
|
||||
params.setFileStem(fileStem, Image::FindParameters::ProcessWildcards{ true });
|
||||
const auto results{ Image::find(session, params) };
|
||||
ASSERT_EQ(results.results.size(), 1);
|
||||
EXPECT_EQ(results.results[0]->getId(), image.getId());
|
||||
}
|
||||
|
||||
{
|
||||
auto transaction{ session.createReadTransaction() };
|
||||
Image::FindParameters params;
|
||||
params.setFileStem("nonexistent");
|
||||
const auto results{ Image::find(session, params) };
|
||||
EXPECT_EQ(results.results.size(), 0);
|
||||
}
|
||||
|
||||
{
|
||||
auto transaction{ session.createReadTransaction() };
|
||||
Image::FindParameters params;
|
||||
params.setFileStem("ima*");
|
||||
const auto results{ Image::find(session, params) };
|
||||
ASSERT_EQ(results.results.size(), 0);
|
||||
}
|
||||
|
||||
{
|
||||
auto transaction{ session.createReadTransaction() };
|
||||
Image::FindParameters params;
|
||||
params.setFileStem("ima*", Image::FindParameters::ProcessWildcards{ true });
|
||||
const auto results{ Image::find(session, params) };
|
||||
ASSERT_EQ(results.results.size(), 1);
|
||||
EXPECT_EQ(results.results[0]->getId(), image.getId());
|
||||
}
|
||||
|
||||
{
|
||||
auto transaction{ session.createReadTransaction() };
|
||||
Image::FindParameters params;
|
||||
params.setFileStem("*ge", Image::FindParameters::ProcessWildcards{ true });
|
||||
const auto results{ Image::find(session, params) };
|
||||
ASSERT_EQ(results.results.size(), 1);
|
||||
EXPECT_EQ(results.results[0]->getId(), image.getId());
|
||||
}
|
||||
|
||||
{
|
||||
auto transaction{ session.createReadTransaction() };
|
||||
Image::FindParameters params;
|
||||
params.setFileStem("*g*", Image::FindParameters::ProcessWildcards{ true });
|
||||
const auto results{ Image::find(session, params) };
|
||||
ASSERT_EQ(results.results.size(), 1);
|
||||
EXPECT_EQ(results.results[0]->getId(), image.getId());
|
||||
}
|
||||
|
||||
{
|
||||
auto transaction{ session.createReadTransaction() };
|
||||
Image::FindParameters params;
|
||||
params.setFileStem("*", Image::FindParameters::ProcessWildcards{ true });
|
||||
const auto results{ Image::find(session, params) };
|
||||
ASSERT_EQ(results.results.size(), 1);
|
||||
EXPECT_EQ(results.results[0]->getId(), image.getId());
|
||||
}
|
||||
|
||||
{
|
||||
auto transaction{ session.createReadTransaction() };
|
||||
Image::FindParameters params;
|
||||
params.setFileStem("ima%");
|
||||
const auto results{ Image::find(session, params) };
|
||||
ASSERT_EQ(results.results.size(), 0);
|
||||
}
|
||||
|
||||
{
|
||||
auto transaction{ session.createReadTransaction() };
|
||||
Image::FindParameters params;
|
||||
params.setFileStem("ima%", Image::FindParameters::ProcessWildcards{ true });
|
||||
const auto results{ Image::find(session, params) };
|
||||
ASSERT_EQ(results.results.size(), 0);
|
||||
}
|
||||
}
|
||||
|
||||
} // namespace lms::db::tests
|
||||
Reference in New Issue
Block a user