11 #include <tny-shared.h>
12 #include <tny-folder.h>
13 #include <tny-folder-store.h>
15 #include <tny-iterator.h>
16 #include <tny-simple-list.h>
18 #include <libmodest-dbus-client/libmodest-dbus-client.h>
20 #include "modest-text-utils.h"
21 #include "modest-account-mgr.h"
22 #include "modest-tny-account-store.h"
23 #include "modest-tny-account.h"
24 #include "modest-search.h"
25 #include "modest-runtime.h"
28 g_strdup_or_null (const char *str)
33 string = g_strdup (str);
40 add_hit (GList *list, TnyHeader *header, TnyFolder *folder)
50 hit = g_slice_new0 (ModestSearchHit);
52 furl = tny_folder_get_url_string (folder);
53 uid = tny_header_get_uid (header);
54 msg_url = g_strdup_printf ("%s/%s", furl, uid);
55 subject = tny_header_get_subject (header);
56 sender = tny_header_get_from (header);
58 flags = tny_header_get_flags (header);
61 hit->subject = g_strdup_or_null (subject);
62 hit->sender = g_strdup_or_null (sender);
64 hit->msize = tny_header_get_message_size (header);
65 hit->has_attachment = flags & TNY_HEADER_FLAG_ATTACHMENTS;
66 hit->is_unread = ! (flags & TNY_HEADER_FLAG_SEEN);
67 hit->timestamp = tny_header_get_date_received (header);
69 return g_list_prepend (list, hit);
73 read_chunk (TnyStream *stream, char *buffer, gsize count, gsize *nread)
79 while (_nread < count) {
80 res = tny_stream_read (stream,
100 #ifdef MODEST_HAVE_OGS
102 search_mime_part_ogs (TnyMimePart *part, ModestSearch *search)
108 gboolean is_html = FALSE;
113 if (! tny_mime_part_content_type_is (part, "text/ *") ||
114 ! (is_html = tny_mime_part_content_type_is (part, "text/html"))) {
119 len = sizeof (buffer);
120 stream = tny_mime_part_get_stream (part);
122 while ((res = read_chunk (stream, buffer, len, &nread))) {
126 found = ogs_text_searcher_search_html (search->text_searcher,
131 found = ogs_text_searcher_search_text (search->text_searcher,
143 found = ogs_text_searcher_search_done (search->text_searcher);
146 ogs_text_searcher_reset (search->text_searcher);
153 search_mime_part_strcmp (TnyMimePart *part, ModestSearch *search)
163 if (! tny_mime_part_content_type_is (part, "text/ *")) {
168 len = (sizeof (buffer) - 1) / 2;
170 if (strlen (search->body) > len) {
171 g_warning ("Search term bigger then chunk."
172 "We might not find everything");
175 stream = tny_mime_part_get_stream (part);
177 memset (buffer, 0, sizeof (buffer));
179 chunk[1] = buffer + len;
181 res = read_chunk (stream, chunk[0], len, &nread);
187 found = !modest_text_utils_utf8_strcmp (search->body,
194 /* This works like this:
195 * buffer: [ooooooooooo|xxxxxxxxxxxx|\0]
196 * ^chunk[0] ^chunk[1]
197 * we have prefilled chunk[0] now we always read into chunk[1]
198 * and then move the content of chunk[1] to chunk[0].
199 * The idea is to prevent not finding search terms that are
200 * spread across 2 reads:
201 * buffer: [ooooooooTES|Txxxxxxxxxxx|\0]
202 * We should catch that because we always search the whole
203 * buffer not only the chunks.
205 * Of course that breaks for search terms > sizeof (chunk)
206 * but sizeof (chunk) should be big enough I guess (see
207 * the g_warning in this function)
209 while ((res = read_chunk (stream, chunk[1], len, &nread))) {
210 buffer[len + nread] = '\0';
212 found = !modest_text_utils_utf8_strcmp (search->body,
220 /* also move the \0 */
221 g_memmove (chunk[0], chunk[1], len + 1);
225 g_object_unref (stream);
230 search_string (const char *what,
232 ModestSearch *search)
235 #ifdef MODEST_HAVE_OGS
236 if (search->flags & MODEST_SEARCH_USE_OGS) {
237 found = ogs_text_searcher_search_text (search->text_searcher,
241 ogs_text_searcher_reset (search->text_searcher);
244 if (what == NULL || where == NULL) {
248 found = !modest_text_utils_utf8_strcmp (what, where, TRUE);
249 #ifdef MODEST_HAVE_OGS
259 * @folder: a #TnyFolder instance
260 * @search: a #ModestSearch query
262 * This operation will search @folder for headers that match the query @search.
263 * It will return a doubly linked list with URIs that point to the message.
266 modest_search_folder (TnyFolder *folder, ModestSearch *search)
268 GList *retval = NULL;
271 gboolean (*part_search_func) (TnyMimePart *part, ModestSearch *search);
273 part_search_func = search_mime_part_strcmp;
275 #ifdef MODEST_HAVE_OGS
276 if (search->flags & MODEST_SEARCH_USE_OGS) {
278 if (search->text_searcher == NULL && search->query != NULL) {
279 OgsTextSearcher *text_searcher;
281 text_searcher = ogs_text_searcher_new (FALSE);
282 ogs_text_searcher_parse_query (text_searcher, search->query);
283 search->text_searcher = text_searcher;
286 part_search_func = search_mime_part_ogs;
290 list = tny_simple_list_new ();
291 tny_folder_get_headers (folder, list, FALSE, NULL);
293 iter = tny_list_create_iterator (list);
295 while (!tny_iterator_is_done (iter)) {
296 TnyHeader *cur = (TnyHeader *) tny_iterator_get_current (iter);
297 time_t t = tny_header_get_date_sent (cur);
298 gboolean found = FALSE;
300 if (search->flags & MODEST_SEARCH_BEFORE)
301 if (!(t <= search->before))
304 if (search->flags & MODEST_SEARCH_AFTER)
305 if (!(t >= search->after))
308 if (search->flags & MODEST_SEARCH_SIZE)
309 if (tny_header_get_message_size (cur) < search->minsize)
312 if (search->flags & MODEST_SEARCH_SUBJECT) {
313 const char *str = tny_header_get_subject (cur);
315 if ((found = search_string (search->subject, str, search))) {
316 retval = add_hit (retval, cur, folder);
320 if (!found && search->flags & MODEST_SEARCH_SENDER) {
321 const char *str = tny_header_get_from (cur);
323 if ((found = search_string (search->from, str, search))) {
324 retval = add_hit (retval, cur, folder);
328 if (!found && search->flags & MODEST_SEARCH_RECIPIENT) {
329 const char *str = tny_header_get_to (cur);
331 if ((found = search_string (search->recipient, str, search))) {
332 retval = add_hit (retval, cur, folder);
336 if (!found && search->flags & MODEST_SEARCH_BODY) {
337 TnyHeaderFlags flags;
343 flags = tny_header_get_flags (cur);
345 if (!(flags & TNY_HEADER_FLAG_CACHED)) {
349 msg = tny_folder_get_msg (folder, cur, &err);
351 if (err != NULL || msg == NULL) {
352 g_warning ("Could not get message\n");
356 g_object_unref (msg);
360 parts = tny_simple_list_new ();
361 tny_mime_part_get_parts (TNY_MIME_PART (msg), parts);
363 piter = tny_list_create_iterator (parts);
364 while (!found && !tny_iterator_is_done (piter)) {
365 TnyMimePart *pcur = (TnyMimePart *) tny_iterator_get_current (piter);
367 if ((found = part_search_func (pcur, search))) {
368 retval = add_hit (retval, cur, folder);
371 g_object_unref (pcur);
372 tny_iterator_next (piter);
375 g_object_unref (piter);
376 g_object_unref (parts);
377 g_object_unref (msg);
382 g_object_unref (cur);
383 tny_iterator_next (iter);
386 g_object_unref (iter);
387 g_object_unref (list);
392 modest_search_account (TnyAccount *account, ModestSearch *search)
394 TnyFolderStore *store;
403 store = TNY_FOLDER_STORE (account);
405 folders = tny_simple_list_new ();
406 tny_folder_store_get_folders (store, folders, NULL, &error);
409 g_object_unref (folders);
413 iter = tny_list_create_iterator (folders);
414 while (!tny_iterator_is_done (iter)) {
418 folder = TNY_FOLDER (tny_iterator_get_current (iter));
420 res = modest_search_folder (folder, search);
426 hits = g_list_concat (hits, res);
430 g_object_unref (folder);
431 tny_iterator_next (iter);
434 g_object_unref (iter);
435 g_object_unref (folders);
441 modest_search_all_accounts (ModestSearch *search)
443 ModestTnyAccountStore *astore;
449 astore = modest_runtime_get_account_store ();
451 accounts = tny_simple_list_new ();
452 tny_account_store_get_accounts (TNY_ACCOUNT_STORE (astore),
454 TNY_ACCOUNT_STORE_STORE_ACCOUNTS);
456 iter = tny_list_create_iterator (accounts);
457 while (!tny_iterator_is_done (iter)) {
461 account = TNY_ACCOUNT (tny_iterator_get_current (iter));
463 g_debug ("Searching account %s",
464 tny_account_get_name (account));
465 res = modest_search_account (account, search);
472 hits = g_list_concat (hits, res);
476 g_object_unref (account);
477 tny_iterator_next (iter);
480 g_object_unref (accounts);