diff --git a/Userland/Libraries/LibIMAP/Client.cpp b/Userland/Libraries/LibIMAP/Client.cpp index f84480128c..21d623ec35 100644 --- a/Userland/Libraries/LibIMAP/Client.cpp +++ b/Userland/Libraries/LibIMAP/Client.cpp @@ -126,8 +126,12 @@ static ReadonlyBytes command_byte_buffer(CommandType command) return "SELECT"sv.bytes(); case CommandType::Fetch: return "FETCH"sv.bytes(); + case CommandType::Search: + return "SEARCH"sv.bytes(); case CommandType::UIDFetch: return "UID FETCH"sv.bytes(); + case CommandType::UIDSearch: + return "UID SEARCH"sv.bytes(); } VERIFY_NOT_REACHED(); } @@ -244,6 +248,20 @@ void Client::send_next_command() send_raw(buffer); m_expecting_response = true; } +RefPtr>> Client::search(Optional charset, Vector&& keys, bool uid) +{ + Vector args; + if (charset.has_value()) { + args.append("CHARSET "); + args.append(charset.value()); + } + for (const auto& item : keys) { + args.append(item.serialize()); + } + auto command = Command { uid ? CommandType::UIDSearch : CommandType::Search, m_current_command, args }; + return cast_promise(send_command(move(command))); +} + RefPtr>> Client::idle() { auto promise = send_simple_command(CommandType::Idle); diff --git a/Userland/Libraries/LibIMAP/Client.h b/Userland/Libraries/LibIMAP/Client.h index aed53dcf24..e1e99b6a3a 100644 --- a/Userland/Libraries/LibIMAP/Client.h +++ b/Userland/Libraries/LibIMAP/Client.h @@ -24,6 +24,7 @@ public: RefPtr>> login(StringView username, StringView password); RefPtr>> list(StringView reference_name, StringView mailbox_name); RefPtr>> select(StringView string); + RefPtr>> search(Optional charset, Vector&& search_keys, bool uid); RefPtr>> fetch(FetchCommand request, bool uid); RefPtr>> idle(); RefPtr>> finish_idle(); diff --git a/Userland/Libraries/LibIMAP/Objects.cpp b/Userland/Libraries/LibIMAP/Objects.cpp index d534fac6b8..1b99fb428b 100644 --- a/Userland/Libraries/LibIMAP/Objects.cpp +++ b/Userland/Libraries/LibIMAP/Objects.cpp @@ -4,6 +4,7 @@ * SPDX-License-Identifier: BSD-2-Clause */ +#include #include namespace IMAP { @@ -112,4 +113,77 @@ String FetchCommand::serialize() return AK::String::formatted("{} ({})", sequence_builder.build(), data_items_builder.build()); } +String serialize_astring(StringView string) +{ + // Try to send an atom + auto is_non_atom_char = [](char x) { + auto non_atom_chars = { '(', ')', '{', ' ', '%', '*', '"', '\\', ']' }; + return AK::find(non_atom_chars.begin(), non_atom_chars.end(), x) != non_atom_chars.end(); + }; + auto is_atom = all_of(string.begin(), string.end(), [&](auto ch) { return is_ascii_control(ch) && !is_non_atom_char(ch); }); + if (is_atom) { + return string; + } + + // Try to quote + auto can_be_quoted = !(string.contains('\n') || string.contains('\r')); + if (can_be_quoted) { + auto escaped_str = string.to_string(); + escaped_str.replace("\\", "\\\\"); + escaped_str.replace("\"", "\\\""); + return String::formatted("\"{}\"", escaped_str); + } + + // Just send a literal + return String::formatted("{{{}}}\r\n{}", string.length(), string); +} +String SearchKey::serialize() const +{ + return data.visit( + [&](Empty const&) { VERIFY_NOT_REACHED(); return String("The compiler complains if you remove this."); }, + [&](All const&) { return String("ALL"); }, + [&](Answered const&) { return String("ANSWERED"); }, + [&](Bcc const& x) { return String::formatted("BCC {}", serialize_astring(x.bcc)); }, + [&](Cc const& x) { return String::formatted("CC {}", serialize_astring(x.cc)); }, + [&](Deleted const&) { return String("DELETED"); }, + [&](Draft const&) { return String("DRAFT"); }, + [&](From const& x) { return String::formatted("FROM {}", serialize_astring(x.from)); }, + [&](Header const& x) { return String::formatted("HEADER {} {}", serialize_astring(x.header), serialize_astring(x.value)); }, + [&](Keyword const& x) { return String::formatted("KEYWORD {}", x.keyword); }, + [&](Larger const& x) { return String::formatted("LARGER {}", x.number); }, + [&](New const&) { return String("NEW"); }, + [&](Not const& x) { return String::formatted("NOT {}", x.operand->serialize()); }, + [&](Old const&) { return String("OLD"); }, + [&](On const& x) { return String::formatted("ON {}", x.date.to_string("%d-%b-%Y")); }, + [&](Or const& x) { return String::formatted("OR {} {}", x.lhs->serialize(), x.rhs->serialize()); }, + [&](Recent const&) { return String("RECENT"); }, + [&](SearchKeys const& x) { + StringBuilder sb; + sb.append("("); + bool first = true; + for (const auto& item : x.keys) { + if (!first) + sb.append(", "); + sb.append(item->serialize()); + first = false; + } + return sb.build(); + }, + [&](Seen const&) { return String("SEEN"); }, + [&](SentBefore const& x) { return String::formatted("SENTBEFORE {}", x.date.to_string("%d-%b-%Y")); }, + [&](SentOn const& x) { return String::formatted("SENTON {}", x.date.to_string("%d-%b-%Y")); }, + [&](SentSince const& x) { return String::formatted("SENTSINCE {}", x.date.to_string("%d-%b-%Y")); }, + [&](SequenceSet const& x) { return x.sequence.serialize(); }, + [&](Since const& x) { return String::formatted("SINCE {}", x.date.to_string("%d-%b-%Y")); }, + [&](Smaller const& x) { return String::formatted("SMALLER {}", x.number); }, + [&](Subject const& x) { return String::formatted("SUBJECT {}", serialize_astring(x.subject)); }, + [&](Text const& x) { return String::formatted("TEXT {}", serialize_astring(x.text)); }, + [&](To const& x) { return String::formatted("TO {}", serialize_astring(x.to)); }, + [&](UID const& x) { return String::formatted("UID {}", x.uid); }, + [&](Unanswered const&) { return String("UNANSWERED"); }, + [&](Undeleted const&) { return String("UNDELETED"); }, + [&](Undraft const&) { return String("UNDRAFT"); }, + [&](Unkeyword const& x) { return String::formatted("UNKEYWORD {}", serialize_astring(x.flag_keyword)); }, + [&](Unseen const&) { return String("UNSEEN"); }); +} } diff --git a/Userland/Libraries/LibIMAP/Objects.h b/Userland/Libraries/LibIMAP/Objects.h index 66ba6436d6..0dd87cb044 100644 --- a/Userland/Libraries/LibIMAP/Objects.h +++ b/Userland/Libraries/LibIMAP/Objects.h @@ -24,8 +24,10 @@ enum class CommandType { Login, Logout, Noop, + Search, Select, UIDFetch, + UIDSearch, }; enum class MailboxFlag : unsigned { @@ -56,6 +58,7 @@ enum class ResponseType : unsigned { Unseen = 1u << 7, PermanentFlags = 1u << 8, Fetch = 1u << 9, + Search = 1u << 10, Bye = 1u << 13, }; @@ -319,6 +322,77 @@ private: BodyStructure m_body_structure; }; +String serialize_astring(StringView string); + +struct SearchKey { +public: + // clang-format off + struct All { }; + struct Answered { }; + struct Bcc { String bcc; }; + struct Cc { String cc; }; + struct Deleted { }; + struct Draft { }; + struct From { String from; }; + struct Header { String header; String value; }; + struct Keyword { String keyword; }; + struct Larger { unsigned number; }; + struct New { }; + struct Not { OwnPtr operand; }; + struct Old { }; + struct On { Core::DateTime date; }; + struct Or { OwnPtr lhs; OwnPtr rhs; }; + struct Recent { }; + struct SearchKeys { Vector> keys; }; + struct Seen { }; + struct SentBefore { Core::DateTime date; }; + struct SentOn { Core::DateTime date; }; + struct SentSince { Core::DateTime date; }; + struct SequenceSet { Sequence sequence; }; + struct Since { Core::DateTime date; }; + struct Smaller { unsigned number; }; + struct Subject { String subject; }; + struct Text { String text; }; + struct To { String to; }; + struct UID { unsigned uid; }; + struct Unanswered { }; + struct Undeleted { }; + struct Undraft { }; + struct Unkeyword { String flag_keyword; }; + struct Unseen { }; + // clang-format on + + Variant + data; + + SearchKey(SearchKey&& other) noexcept + : data(move(other.data)) + { + } + + template + explicit SearchKey(T&& t) + : data(std::forward(t)) + { + } + + SearchKey& operator=(SearchKey&& other) noexcept + { + if (this == &other) { + return *this; + } + + this->data = move(other.data); + + return *this; + } + + [[nodiscard]] String serialize() const; +}; + class ResponseData { public: [[nodiscard]] unsigned response_type() const @@ -466,6 +540,18 @@ public: return m_fetch_responses; } + void set_search_results(Vector&& results) + { + add_response_type(ResponseType::Search); + m_search_results = move(results); + } + + Vector& search_results() + { + VERIFY(contains_response_type(ResponseType::Search)); + return m_search_results; + } + void set_bye(Optional message) { add_response_type(ResponseType::Bye); @@ -493,6 +579,7 @@ private: Vector m_permanent_flags; Vector m_flags; Vector> m_fetch_responses; + Vector m_search_results; Optional m_bye_message; }; diff --git a/Userland/Libraries/LibIMAP/Parser.cpp b/Userland/Libraries/LibIMAP/Parser.cpp index 2e0c8d7530..8ed562017a 100644 --- a/Userland/Libraries/LibIMAP/Parser.cpp +++ b/Userland/Libraries/LibIMAP/Parser.cpp @@ -182,6 +182,14 @@ void Parser::parse_untagged() parse_while([](u8 x) { return x != '\r'; }); consume("\r\n"); } + } else if (try_consume("SEARCH")) { + Vector ids; + while (!try_consume("\r\n")) { + consume(" "); + auto id = parse_number(); + ids.append(id); + } + m_response.data().set_search_results(move(ids)); } else if (try_consume("BYE")) { auto message = parse_while([](u8 x) { return x != '\r'; }); consume("\r\n");