From bd38ea97f9137936b7a5aec0a329b7ff5b63f826 Mon Sep 17 00:00:00 2001
From: Nikhil Tanwar <2002nikhiltanwar@gmail.com>
Date: Mon, 15 Aug 2022 22:12:47 +0530
Subject: [PATCH] Multivalue support for book query

Adds support for putting multiple `book` query parameter.
---
 include/library.h             |  8 ++++----
 src/library.cpp               | 28 +++++++++++++++-------------
 src/server/internalServer.cpp |  2 +-
 3 files changed, 20 insertions(+), 18 deletions(-)

diff --git a/include/library.h b/include/library.h
index a663a9b3f..8ac4fb66a 100644
--- a/include/library.h
+++ b/include/library.h
@@ -54,6 +54,7 @@ enum supportedListMode {
 class Filter {
   public: // types
     using Tags = std::vector<std::string>;
+    using AliasNames = std::vector<std::string>;
 
   private: // data
     uint64_t activeFilters;
@@ -67,7 +68,7 @@ class Filter {
     std::string _query;
     bool _queryIsPartial;
     std::string _name;
-    std::string _aliasName;
+    AliasNames _aliasNames;
 
   public: // functions
     Filter();
@@ -113,7 +114,7 @@ class Filter {
     Filter& maxSize(size_t size);
     Filter& query(std::string query, bool partial=true);
     Filter& name(std::string name);
-    Filter& aliasName(std::string aliasName);
+    Filter& aliasNames(const AliasNames& aliasNames);
 
     bool hasQuery() const;
     const std::string& getQuery() const { return _query; }
@@ -137,8 +138,7 @@ class Filter {
     const Tags& getAcceptTags() const { return _acceptTags; }
     const Tags& getRejectTags() const { return _rejectTags; }
 
-    bool hasAliasName() const;
-    const std::string& getAliasName() const { return _aliasName; }
+    const AliasNames& getAliasNames() const { return _aliasNames; }
 
 private: // functions
     friend class Library;
diff --git a/src/library.cpp b/src/library.cpp
index 88fdd880e..e45d1d488 100644
--- a/src/library.cpp
+++ b/src/library.cpp
@@ -545,9 +545,16 @@ Xapian::Query categoryQuery(const std::string& category)
   return Xapian::Query("XC" + normalizeText(category));
 }
 
-Xapian::Query aliasNameQuery(const std::string& fileName)
+Xapian::Query aliasNamesQuery(const Filter::AliasNames& aliasNames)
 {
-  return parseQuery(fileName, "XF");
+  Xapian::Query q = Xapian::Query(std::string());
+  std::vector<Xapian::Query> queryVec;
+  for (const auto& aliasName : aliasNames) {
+    queryVec.push_back(parseQuery(aliasName, "XF"));
+  }
+  Xapian::Query combinedQuery(Xapian::Query::OP_OR, queryVec.begin(), queryVec.end());
+  q = Xapian::Query(Xapian::Query::OP_FILTER, q, combinedQuery);
+  return q;
 }
 
 Xapian::Query langQuery(const std::string& lang)
@@ -602,8 +609,8 @@ Xapian::Query buildXapianQuery(const Filter& filter)
     const auto tq = tagsQuery(filter.getAcceptTags(), filter.getRejectTags());
     q = Xapian::Query(Xapian::Query::OP_AND, q, tq);;
   }
-  if ( filter.hasAliasName() ) {
-    q = Xapian::Query(Xapian::Query::OP_AND, q, aliasNameQuery(filter.getAliasName()));
+  if ( !filter.getAliasNames().empty() ) {
+    q = Xapian::Query(Xapian::Query::OP_AND, q, aliasNamesQuery(filter.getAliasNames()));
   }
   return q;
 }
@@ -754,7 +761,7 @@ enum filterTypes {
   QUERY = FLAG(12),
   NAME = FLAG(13),
   CATEGORY = FLAG(14),
-  ALIASNAME = FLAG(15),
+  ALIASNAMES = FLAG(15),
 };
 
 Filter& Filter::local(bool accept)
@@ -857,10 +864,10 @@ Filter& Filter::name(std::string name)
   return *this;
 }
 
-Filter& Filter::aliasName(std::string aliasName)
+Filter& Filter::aliasNames(const AliasNames& aliasNames)
 {
-  _aliasName = aliasName;
-  activeFilters |= ALIASNAME;  
+  _aliasNames = aliasNames;
+  activeFilters |= ALIASNAMES;  
   return *this;
 }
 
@@ -876,11 +883,6 @@ bool Filter::hasName() const
   return ACTIVE(NAME);
 }
 
-bool Filter::hasAliasName() const
-{
-  return ACTIVE(ALIASNAME);
-}
-
 bool Filter::hasCategory() const
 {
   return ACTIVE(CATEGORY);
diff --git a/src/server/internalServer.cpp b/src/server/internalServer.cpp
index 5e3ab598e..5aa1b53ee 100644
--- a/src/server/internalServer.cpp
+++ b/src/server/internalServer.cpp
@@ -120,7 +120,7 @@ Filter get_search_filter(const RequestContext& request, const std::string& prefi
       filter.rejectTags(kiwix::split(request.get_argument(prefix+"notag"), ";"));
     } catch (...) {}
     try {
-      filter.aliasName(request.get_argument(prefix + "book"));
+      filter.aliasNames(request.get_arguments(prefix + "book"));
     } catch (...) {}
     return filter;
 }