From 2baeac116abef9d5bcb748c687577d16dce868a0 Mon Sep 17 00:00:00 2001
From: Thomas Bruederli <thomas@roundcube.net>
Date: Thu, 16 Jan 2014 08:17:08 -0500
Subject: [PATCH] Fix sorting and paging in cross-folder searches

---
 program/steps/mail/search.inc |  224 ++++++++++++++++++++++++++++++++++++-------------------
 1 files changed, 147 insertions(+), 77 deletions(-)

diff --git a/program/steps/mail/search.inc b/program/steps/mail/search.inc
index e1690ee..5ce9fe6 100644
--- a/program/steps/mail/search.inc
+++ b/program/steps/mail/search.inc
@@ -1,107 +1,177 @@
 <?php
+
 /*
  +-----------------------------------------------------------------------+
  | steps/mail/search.inc                                                 |
  |                                                                       |
- | Search functions for rc webmail                                       |
- | Licensed under the GNU GPL                                            |
+ | This file is part of the Roundcube Webmail client                     |
+ | Copyright (C) 2005-2013, The Roundcube Dev Team                       |
  |                                                                       |
+ | Licensed under the GNU General Public License version 3 or            |
+ | any later version with exceptions for skins & plugins.                |
+ | See the README file for a full license statement.                     |
+ |                                                                       |
+ | PURPOSE:                                                              |
+ |   Mail messages search action                                         |
  +-----------------------------------------------------------------------+
  | Author: Benjamin Smith <defitro@gmail.com>                            |
  |         Thomas Bruederli <roundcube@gmail.com>                        |
  +-----------------------------------------------------------------------+
-
 */
 
 $REMOTE_REQUEST = TRUE;
 
-// reset list_page
-$IMAP->set_page(1);
+// reset list_page and old search results
+$RCMAIL->storage->set_page(1);
+$RCMAIL->storage->set_search_set(NULL);
 $_SESSION['page'] = 1;
 
 // using encodeURI with javascript "should" give us
-// a correctly UTF-8 encoded query string
-$imap_charset = 'UTF-8';
+// a correctly encoded query string
+$imap_charset = RCUBE_CHARSET;
 
 // get search string
-$str = get_input_value('_search', RCUBE_INPUT_GET);
-$mbox = get_input_value('_mbox', RCUBE_INPUT_GET);
-$search_request = md5($str);
+$str     = rcube_utils::get_input_value('_q', rcube_utils::INPUT_GET, true);
+$mbox    = rcube_utils::get_input_value('_mbox', rcube_utils::INPUT_GET, true);
+$filter  = rcube_utils::get_input_value('_filter', rcube_utils::INPUT_GET);
+$headers = rcube_utils::get_input_value('_headers', rcube_utils::INPUT_GET);
+$subject = array();
 
+$search_request = md5($mbox.$filter.$str);
+
+// add list filter string
+$search_str = $filter && $filter != 'ALL' ? $filter : '';
+
+$_SESSION['search_filter'] = $filter;
 
 // Check the search string for type of search
-if (preg_match("/^from:/i", $str)) {
-  list(,$srch) = explode(":", $str);
-  $search = $IMAP->search($mbox, "HEADER FROM" ,trim($srch), $imap_charset);
-  finish_search($mbox, $search);
+if (preg_match("/^from:.*/i", $str)) {
+    list(,$srch) = explode(":", $str);
+    $subject['from'] = "HEADER FROM";
 }
-else if (preg_match("/^to:/i", $str)) {
-  list(,$srch) = explode(":", $str);
-  $search = $IMAP->search($mbox, "HEADER TO", trim($srch), $imap_charset);
-  finish_search($mbox, $search);
+else if (preg_match("/^to:.*/i", $str)) {
+    list(,$srch) = explode(":", $str);
+    $subject['to'] = "HEADER TO";
 }
-else if (preg_match("/^cc:/i", $str)) {
-  list(,$srch) = explode(":", $str);
-  $search = $IMAP->search($mbox, "HEADER CC", trim($srch), $imap_charset);
-  finish_search($mbox, $search);
+else if (preg_match("/^cc:.*/i", $str)) {
+    list(,$srch) = explode(":", $str);
+    $subject['cc'] = "HEADER CC";
 }
-else if (preg_match("/^subject:/i", $str)) {
-  list(,$srch) = explode(":", $str);
-  $search = $IMAP->search($mbox, "HEADER SUBJECT", trim($srch), $imap_charset);
-  finish_search($mbox, $search);
+else if (preg_match("/^bcc:.*/i", $str)) {
+    list(,$srch) = explode(":", $str);
+    $subject['bcc'] = "HEADER BCC";
 }
-else if (preg_match("/^body:/i", $str)) {
-  list(,$srch) = explode(":", $str);
-  $search = $IMAP->search($mbox, "TEXT", trim($srch), $imap_charset);
-  finish_search($mbox, $search);
+else if (preg_match("/^subject:.*/i", $str)) {
+    list(,$srch) = explode(":", $str);
+    $subject['subject'] = "HEADER SUBJECT";
 }
-// search in subject and sender by default
+else if (preg_match("/^body:.*/i", $str)) {
+    list(,$srch) = explode(":", $str);
+    $subject['body'] = "BODY";
+}
+else if (strlen(trim($str))) {
+    if ($headers) {
+        foreach (explode(',', $headers) as $header) {
+            if ($header == 'text') {
+                // #1488208: get rid of other headers when searching by "TEXT"
+                $subject = array('text' => 'TEXT');
+                break;
+            }
+            else {
+                $subject[$header] = ($header != 'body' ? 'HEADER ' : '') . strtoupper($header);
+            }
+        }
+
+        // save search modifiers for the current folder to user prefs
+        $search_mods        = rcmail_search_mods();
+        $search_mods[$mbox] = array_fill_keys(array_keys($subject), 1);
+
+        $RCMAIL->user->save_prefs(array('search_mods' => $search_mods));
+    }
+    else {
+        // search in subject by default
+        $subject['subject'] = 'HEADER SUBJECT';
+    }
+}
+
+$search = isset($srch) ? trim($srch) : trim($str);
+
+if (!empty($subject)) {
+    $search_str .= str_repeat(' OR', count($subject)-1);
+    foreach ($subject as $sub) {
+        $search_str .= ' ' . $sub . ' ' . rcube_imap_generic::escape($search);
+    }
+}
+
+$search_str  = trim($search_str);
+$sort_column = rcmail_sort_column();
+
+// TEMPORARY: search all folders
+$mboxes = $RCMAIL->storage->list_folders_subscribed('', '*', 'mail');
+
+// execute IMAP search
+if ($search_str) {
+    $RCMAIL->storage->search($mboxes, $search_str, $imap_charset, $sort_column);
+}
+
+// save search results in session
+if (!is_array($_SESSION['search'])) {
+    $_SESSION['search'] = array();
+}
+
+if ($search_str) {
+    $_SESSION['search'] = $RCMAIL->storage->get_search_set();
+    $_SESSION['last_text_search'] = $str;
+}
+$_SESSION['search_request'] = $search_request;
+
+// Get the headers
+$result_h = $RCMAIL->storage->list_messages($mbox, 1, $sort_column, rcmail_sort_order());
+$count    = $RCMAIL->storage->count($mbox, $RCMAIL->storage->get_threading() ? 'THREADS' : 'ALL');
+
+// Add 'folder' column to list
+if ($_SESSION['search'][1]->multi) {
+    $a_show_cols = $_SESSION['list_attrib']['columns'] ? $_SESSION['list_attrib']['columns'] : (array)$CONFIG['list_cols'];
+    if (!in_array('folder', $a_show_cols))
+        $a_show_cols[] = 'folder';
+
+    // make message UIDs unique by appending the folder name
+    foreach ($result_h as $i => $header) {
+        $header->uid .= '-'.$header->folder;
+        $header->flags['skip_mbox_check'] = true;
+        if ($header->parent_uid)
+            $header->parent_uid .= '-'.$header->folder;
+    }
+
+    $OUTPUT->command('select_folder', '');
+}
+
+// Make sure we got the headers
+if (!empty($result_h)) {
+    rcmail_js_message_list($result_h, false, $a_show_cols);
+    if ($search_str) {
+        $OUTPUT->show_message('searchsuccessful', 'confirmation', array('nr' => $RCMAIL->storage->count(NULL, 'ALL')));
+    }
+
+    // remember last HIGHESTMODSEQ value (if supported)
+    // we need it for flag updates in check-recent
+    $data = $RCMAIL->storage->folder_data($mbox_name);
+    if (!empty($data['HIGHESTMODSEQ'])) {
+        $_SESSION['list_mod_seq'] = $data['HIGHESTMODSEQ'];
+    }
+}
+// handle IMAP errors (e.g. #1486905)
+else  if ($err_code = $RCMAIL->storage->get_error_code()) {
+    $RCMAIL->display_server_error();
+}
 else {
-  $search = $IMAP->search($mbox, "HEADER SUBJECT", trim($str), $imap_charset);
-  $search2 = $IMAP->search($mbox, "HEADER FROM", trim($str), $imap_charset);
-  finish_search($mbox, array_unique(array_merge($search, $search2)));
+    $OUTPUT->show_message('searchnomatch', 'notice');
 }
 
-
-// Complete the search display results or report error
-function finish_search($mbox, $search)
-  {
-  global $IMAP, $JS_OBJECT_NAME, $OUTPUT, $search_request;
-  $commands = '';
-  $count = 0;
-    
-  // Make sure our $search is legit..
-  if (is_array($search) && $search[0] != '')
-    {
-    // Get the headers
-    $result_h = $IMAP->list_header_set($mbox, $search, 1, $_SESSION['sort_col'], $_SESSION['sort_order']);
-    $count = count($search);
-
-    // save search results in session
-    if (!is_array($_SESSION['search']))
-      $_SESSION['search'] = array();
-
-    // Make sure we got the headers
-    if ($result_h != NULL)
-      {
-      $_SESSION['search'][$search_request] = join(',', $search);
-      $commands = rcmail_js_message_list($result_h);
-      $commands .= show_message('searchsuccessful', 'confirmation', array('nr' => $count));
-      }
-    }
-  else
-    {
-    $commands = show_message('searchnomatch', 'warning');
-    $search_request = -1;
-    }
-  
-  // update message count display
-  $pages = ceil($count/$IMAP->page_size);
-  $commands .= sprintf("\nthis.set_env('search_request', '%s')\n", $search_request);
-  $commands .= sprintf("this.set_env('messagecount', %d);\n", $count);
-  $commands .= sprintf("this.set_env('pagecount', %d);\n", $pages);
-  $commands .= sprintf("this.set_rowcount('%s');\n", rcmail_get_messagecount_text($count, 1));
-  rcube_remote_response($commands);
-  }
-
-?>
\ No newline at end of file
+// update message count display
+$OUTPUT->set_env('search_request', $search_str ? $search_request : '');
+$OUTPUT->set_env('messagecount', $count);
+$OUTPUT->set_env('pagecount', ceil($count/$RCMAIL->storage->get_pagesize()));
+$OUTPUT->set_env('exists', $RCMAIL->storage->count($mbox_name, 'EXISTS'));
+$OUTPUT->command('set_rowcount', rcmail_get_messagecount_text($count, 1), $mbox);
+$OUTPUT->send();

--
Gitblit v1.9.1