From 7fe3811c65a7c63154f03610e289a6d196f3ae2e Mon Sep 17 00:00:00 2001
From: thomascube <thomas@roundcube.net>
Date: Wed, 18 Jan 2012 06:10:46 -0500
Subject: [PATCH] Changed license to GNU GPLv3+ with exceptions for skins and plugins

---
 program/steps/mail/search.inc |  159 +++++++++++++++++++++++++++++++++-------------------
 1 files changed, 100 insertions(+), 59 deletions(-)

diff --git a/program/steps/mail/search.inc b/program/steps/mail/search.inc
index a3f6dda..3f9b86e 100644
--- a/program/steps/mail/search.inc
+++ b/program/steps/mail/search.inc
@@ -4,106 +4,147 @@
  | steps/mail/search.inc                                                 |
  |                                                                       |
  | Search functions for rc webmail                                       |
- | Licensed under the GNU GPL                                            |
+ |                                                                       |
+ | 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.                     |
  |                                                                       |
  +-----------------------------------------------------------------------+
  | Author: Benjamin Smith <defitro@gmail.com>                            |
  |         Thomas Bruederli <roundcube@gmail.com>                        |
  +-----------------------------------------------------------------------+
 
+ $Id$
+
 */
 
 $REMOTE_REQUEST = TRUE;
 
 // reset list_page and old search results
-$IMAP->set_page(1);
-$IMAP->set_search_set(NULL);
+$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 = RCMAIL_CHARSET;
 
 // get search string
-$str = get_input_value('_q', RCUBE_INPUT_GET);
-$mbox = get_input_value('_mbox', RCUBE_INPUT_GET);
-$search_request = md5($mbox.$str);
+$str     = get_input_value('_q', RCUBE_INPUT_GET, true);
+$mbox    = get_input_value('_mbox', RCUBE_INPUT_GET, true);
+$filter  = get_input_value('_filter', RCUBE_INPUT_GET);
+$headers = get_input_value('_headers', RCUBE_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))
+if (preg_match("/^from:.*/i", $str))
 {
   list(,$srch) = explode(":", $str);
-  $subject =  "HEADER FROM";
-  $search = trim($srch);
+  $subject['from'] = "HEADER FROM";
 }
-else if (preg_match("/^to:/i", $str))
+else if (preg_match("/^to:.*/i", $str))
 {
   list(,$srch) = explode(":", $str);
-  $subject = "HEADER TO";
-  $search = trim($srch);
+  $subject['to'] = "HEADER TO";
 }
-else if (preg_match("/^cc:/i", $str))
+else if (preg_match("/^cc:.*/i", $str))
 {
   list(,$srch) = explode(":", $str);
-  $subject = "HEADER CC";
-  $search = trim($srch);
+  $subject['cc'] = "HEADER CC";
 }
-else if (preg_match("/^subject:/i", $str))
+else if (preg_match("/^bcc:.*/i", $str))
 {
   list(,$srch) = explode(":", $str);
-  $subject = "HEADER SUBJECT";
-  $search = trim($srch);
+  $subject['bcc'] = "HEADER BCC";
 }
-else if (preg_match("/^body:/i", $str))
+else if (preg_match("/^subject:.*/i", $str))
 {
   list(,$srch) = explode(":", $str);
-  $subject = "TEXT";
-  $search = trim($srch);
+  $subject['subject'] = "HEADER SUBJECT";
 }
-// search in subject and sender by default
-else
+else if (preg_match("/^body:.*/i", $str))
 {
-  $subject = array("HEADER SUBJECT", "HEADER FROM");
-  $search = trim($str);
+  list(,$srch) = explode(":", $str);
+  $subject['text'] = "TEXT";
 }
+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 '.strtoupper($header);
+      }
+    }
 
-
-// execute IMAP search
-$result = $IMAP->search($mbox, $subject, $search, $imap_charset);
-$count = 0;
-  
-// Make sure our $result is legit..
-if (is_array($result) && $result[0] != '')
-{
-  // Get the headers
-  $result_h = $IMAP->list_header_set($mbox, $result, 1, $_SESSION['sort_col'], $_SESSION['sort_order']);
-  $count = $IMAP->messagecount();
-
-  // 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] = $IMAP->get_search_set();
-    rcmail_js_message_list($result_h);
-    $OUTPUT->show_message('searchsuccessful', 'confirmation', array('nr' => $count));
+    // save search modifiers for the current folder to user prefs
+    $search_mods = $RCMAIL->config->get('search_mods', $SEARCH_MODS_DEFAULT);
+    $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';
   }
 }
-else
-{
-  $OUTPUT->show_message('searchnomatch', 'warning');
-  $search_request = -1;
+
+$search = isset($srch) ? trim($srch) : trim($str);
+
+if (!empty($subject)) {
+  $search_str .= str_repeat(' OR', count($subject)-1);
+  foreach ($subject as $sub)
+    $search_str .= sprintf(" %s {%d}\r\n%s", $sub, strlen($search), $search);
+}
+
+$search_str = trim($search_str);
+
+// execute IMAP search
+if ($search_str)
+  $RCMAIL->storage->search($mbox, $search_str, $imap_charset, $_SESSION['sort_col']);
+
+// 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, $_SESSION['sort_col'], $_SESSION['sort_order']);
+$count = $RCMAIL->storage->count($mbox, $RCMAIL->storage->get_threading() ? 'THREADS' : 'ALL');
+
+
+// Make sure we got the headers
+if (!empty($result_h)) {
+  rcmail_js_message_list($result_h);
+  if ($search_str)
+    $OUTPUT->show_message('searchsuccessful', 'confirmation', array('nr' => $RCMAIL->storage->count(NULL, 'ALL')));
+}
+// handle IMAP errors (e.g. #1486905)
+else  if ($err_code = $RCMAIL->storage->get_error_code()) {
+  rcmail_display_server_error();
+}
+else {
+  $OUTPUT->show_message('searchnomatch', 'notice');
 }
 
 // update message count display
-$pages = ceil($count/$IMAP->page_size);
-$OUTPUT->set_env('search_request', $search_request);
+$OUTPUT->set_env('search_request', $search_str ? $search_request : '');
 $OUTPUT->set_env('messagecount', $count);
-$OUTPUT->set_env('pagecount', $pages);
-$OUTPUT->command('set_rowcount', rcmail_get_messagecount_text($count, 1));
+$OUTPUT->set_env('pagecount', ceil($count/$RCMAIL->storage->get_pagesize()));
+$OUTPUT->command('set_rowcount', rcmail_get_messagecount_text($count, 1), $mbox);
 $OUTPUT->send();
-
-?>
\ No newline at end of file

--
Gitblit v1.9.1