From 0280343ced6c520700c3ca508e2d04c6b512d319 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?louiz=E2=80=99?= <louiz@louiz.org>
Date: Sat, 10 Feb 2018 19:51:59 +0100
Subject: =?UTF-8?q?Handle=20the=20=E2=80=9Cafter=E2=80=9D=20RSM=20value=20?=
 =?UTF-8?q?to=20page=20through=20results?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/database/database.cpp | 35 ++++++++++++++++++++++++++++++++++-
 1 file changed, 34 insertions(+), 1 deletion(-)

(limited to 'src/database/database.cpp')

diff --git a/src/database/database.cpp b/src/database/database.cpp
index c43ace4..2d6fbbd 100644
--- a/src/database/database.cpp
+++ b/src/database/database.cpp
@@ -165,7 +165,7 @@ std::string Database::store_muc_message(const std::string& owner, const std::str
 }
 
 std::vector<Database::MucLogLine> Database::get_muc_logs(const std::string& owner, const std::string& chan_name, const std::string& server,
-                                                   int limit, const std::string& start, const std::string& end)
+                                                   int limit, const std::string& start, const std::string& end, const Id::real_type after_id)
 {
   auto request = Database::muc_log_lines.select();
   request.where() << Database::Owner{} << "=" << owner << \
@@ -184,6 +184,10 @@ std::vector<Database::MucLogLine> Database::get_muc_logs(const std::string& owne
       if (end_time != -1)
         request << " and " << Database::Date{} << "<=" << end_time;
     }
+  if (after_id != Id::unset_value)
+    {
+      request << " and " << Id{} << ">" << after_id;
+    }
 
   if (limit >= 0)
     request.limit() << limit;
@@ -218,6 +222,35 @@ std::vector<Database::MucLogLine> Database::get_muc_most_recent_logs(const std::
   return {result.crbegin(), result.crend()};
 }
 
+Database::MucLogLine Database::get_muc_log(const std::string& owner, const std::string& chan_name, const std::string& server,
+                                           const std::string& uuid, const std::string& start, const std::string& end)
+{
+  auto request = Database::muc_log_lines.select();
+  request.where() << Database::Owner{} << "=" << owner << \
+          " and " << Database::IrcChanName{} << "=" << chan_name << \
+          " and " << Database::IrcServerName{} << "=" << server << \
+          " and " << Database::Uuid{} << "=" << uuid;
+
+  if (!start.empty())
+    {
+      const auto start_time = utils::parse_datetime(start);
+      if (start_time != -1)
+        request << " and " << Database::Date{} << ">=" << start_time;
+    }
+  if (!end.empty())
+    {
+      const auto end_time = utils::parse_datetime(end);
+      if (end_time != -1)
+        request << " and " << Database::Date{} << "<=" << end_time;
+    }
+
+  auto result = request.execute(*Database::db);
+
+  if (result.empty())
+    throw Database::RecordNotFound{};
+  return result.front();
+}
+
 void Database::add_roster_item(const std::string& local, const std::string& remote)
 {
   auto roster_item = Database::roster.row();
-- 
cgit v1.2.3