From 8eefbb2158c43b51a8c33e6c480cbe61539b9535 Mon Sep 17 00:00:00 2001 From: Aleksander Machniak <alec@alec.pl> Date: Mon, 27 Aug 2012 04:16:04 -0400 Subject: [PATCH] Add option to enable HTML editor on forwarding (#1488517) --- program/include/rcube_imap_cache.php | 266 ++++++++++++++++++++++++++++++----------------------- 1 files changed, 150 insertions(+), 116 deletions(-) diff --git a/program/include/rcube_imap_cache.php b/program/include/rcube_imap_cache.php index 96c49f7..4d6d5e1 100644 --- a/program/include/rcube_imap_cache.php +++ b/program/include/rcube_imap_cache.php @@ -5,7 +5,7 @@ | program/include/rcube_imap_cache.php | | | | This file is part of the Roundcube Webmail client | - | Copyright (C) 2005-2011, The Roundcube Dev Team | + | Copyright (C) 2005-2012, The Roundcube Dev Team | | | | Licensed under the GNU General Public License version 3 or | | any later version with exceptions for skins & plugins. | @@ -18,9 +18,6 @@ | Author: Thomas Bruederli <roundcube@gmail.com> | | Author: Aleksander Machniak <alec@alec.pl> | +-----------------------------------------------------------------------+ - - $Id$ - */ @@ -42,9 +39,9 @@ private $imap; /** - * Instance of rcube_mdb2 + * Instance of rcube_db * - * @var rcube_mdb2 + * @var rcube_db */ private $db; @@ -95,7 +92,7 @@ { $this->db = $db; $this->imap = $imap; - $this->userid = (int)$userid; + $this->userid = $userid; $this->skip_deleted = $skip_deleted; } @@ -290,7 +287,7 @@ * @param string $mailbox Folder name * @param array $msgs Message UIDs * - * @return array The list of messages (rcube_mail_header) indexed by UID + * @return array The list of messages (rcube_message_header) indexed by UID */ function get_messages($mailbox, $msgs = array()) { @@ -301,7 +298,7 @@ // Fetch messages from cache $sql_result = $this->db->query( "SELECT uid, data, flags" - ." FROM ".get_table_name('cache_messages') + ." FROM ".$this->db->table_name('cache_messages') ." WHERE user_id = ?" ." AND mailbox = ?" ." AND uid IN (".$this->db->array2list($msgs, 'integer').")", @@ -314,10 +311,10 @@ $uid = intval($sql_arr['uid']); $result[$uid] = $this->build_message($sql_arr); - // save memory, we don't need message body here (?) - $result[$uid]->body = null; - if (!empty($result[$uid])) { + // save memory, we don't need message body here (?) + $result[$uid]->body = null; + unset($msgs[$uid]); } } @@ -348,20 +345,21 @@ * from IMAP server * @param bool $no_cache Enables internal cache usage * - * @return rcube_mail_header Message data + * @return rcube_message_header Message data */ function get_message($mailbox, $uid, $update = true, $cache = true) { // Check internal cache - if (($message = $this->icache['message']) - && $message['mailbox'] == $mailbox && $message['object']->uid == $uid + if ($this->icache['__message'] + && $this->icache['__message']['mailbox'] == $mailbox + && $this->icache['__message']['object']->uid == $uid ) { - return $this->icache['message']['object']; + return $this->icache['__message']['object']; } $sql_result = $this->db->query( "SELECT flags, data" - ." FROM ".get_table_name('cache_messages') + ." FROM ".$this->db->table_name('cache_messages') ." WHERE user_id = ?" ." AND mailbox = ?" ." AND uid = ?", @@ -388,7 +386,7 @@ // Save current message from internal cache $this->save_icache(); - $this->icache['message'] = array( + $this->icache['__message'] = array( 'object' => $message, 'mailbox' => $mailbox, 'exists' => $found, @@ -403,9 +401,9 @@ /** * Saves the message in cache. * - * @param string $mailbox Folder name - * @param rcube_mail_header $message Message data - * @param bool $force Skips message in-cache existance check + * @param string $mailbox Folder name + * @param rcube_message_header $message Message data + * @param bool $force Skips message in-cache existance check */ function add_message($mailbox, $message, $force = false) { @@ -429,7 +427,7 @@ // here will work as select, assume row exist if affected_rows=0) if (!$force) { $res = $this->db->query( - "UPDATE ".get_table_name('cache_messages') + "UPDATE ".$this->db->table_name('cache_messages') ." SET flags = ?, data = ?, changed = ".$this->db->now() ." WHERE user_id = ?" ." AND mailbox = ?" @@ -443,7 +441,7 @@ // insert new record $this->db->query( - "INSERT INTO ".get_table_name('cache_messages') + "INSERT INTO ".$this->db->table_name('cache_messages') ." (user_id, mailbox, uid, flags, changed, data)" ." VALUES (?, ?, ?, ?, ".$this->db->now().", ?)", $this->userid, $mailbox, (int) $message->uid, $flags, $msg); @@ -461,29 +459,37 @@ */ function change_flag($mailbox, $uids, $flag, $enabled = false) { + if (empty($uids)) { + return; + } + $flag = strtoupper($flag); $idx = (int) array_search($flag, $this->flags); + $uids = (array) $uids; if (!$idx) { return; } // Internal cache update - if ($uids && count($uids) == 1 && ($uid = current($uids)) - && ($message = $this->icache['message']) - && $message['mailbox'] == $mailbox && $message['object']->uid == $uid + if (($message = $this->icache['__message']) + && $message['mailbox'] === $mailbox + && in_array($message['object']->uid, $uids) ) { $message['object']->flags[$flag] = $enabled; - return; + + if (count($uids) == 1) { + return; + } } $this->db->query( - "UPDATE ".get_table_name('cache_messages') + "UPDATE ".$this->db->table_name('cache_messages') ." SET changed = ".$this->db->now() .", flags = flags ".($enabled ? "+ $idx" : "- $idx") ." WHERE user_id = ?" ." AND mailbox = ?" - .($uids !== null ? " AND uid IN (".$this->db->array2list((array)$uids, 'integer').")" : "") + .($uids !== null ? " AND uid IN (".$this->db->array2list($uids, 'integer').")" : "") ." AND (flags & $idx) ".($enabled ? "= 0" : "= $idx"), $this->userid, $mailbox); } @@ -499,26 +505,26 @@ { if (!strlen($mailbox)) { $this->db->query( - "DELETE FROM ".get_table_name('cache_messages') + "DELETE FROM ".$this->db->table_name('cache_messages') ." WHERE user_id = ?", $this->userid); } else { // Remove the message from internal cache - if (!empty($uids) && !is_array($uids) && ($message = $this->icache['message']) - && $message['mailbox'] == $mailbox && $message['object']->uid == $uids + if (!empty($uids) && ($message = $this->icache['__message']) + && $message['mailbox'] === $mailbox + && in_array($message['object']->uid, (array)$uids) ) { - $this->icache['message'] = null; + $this->icache['__message'] = null; } $this->db->query( - "DELETE FROM ".get_table_name('cache_messages') + "DELETE FROM ".$this->db->table_name('cache_messages') ." WHERE user_id = ?" - ." AND mailbox = ".$this->db->quote($mailbox) + ." AND mailbox = ?" .($uids !== null ? " AND uid IN (".$this->db->array2list((array)$uids, 'integer').")" : ""), - $this->userid); + $this->userid, $mailbox); } - } @@ -535,17 +541,19 @@ // otherwise use 'valid' flag to not loose HIGHESTMODSEQ value if ($remove) { $this->db->query( - "DELETE FROM ".get_table_name('cache_index') - ." WHERE user_id = ".intval($this->userid) - .(strlen($mailbox) ? " AND mailbox = ".$this->db->quote($mailbox) : "") + "DELETE FROM ".$this->db->table_name('cache_index') + ." WHERE user_id = ?" + .(strlen($mailbox) ? " AND mailbox = ".$this->db->quote($mailbox) : ""), + $this->userid ); } else { $this->db->query( - "UPDATE ".get_table_name('cache_index') + "UPDATE ".$this->db->table_name('cache_index') ." SET valid = 0" - ." WHERE user_id = ".intval($this->userid) - .(strlen($mailbox) ? " AND mailbox = ".$this->db->quote($mailbox) : "") + ." WHERE user_id = ?" + .(strlen($mailbox) ? " AND mailbox = ".$this->db->quote($mailbox) : ""), + $this->userid ); } @@ -568,9 +576,10 @@ function remove_thread($mailbox = null) { $this->db->query( - "DELETE FROM ".get_table_name('cache_thread') - ." WHERE user_id = ".intval($this->userid) - .(strlen($mailbox) ? " AND mailbox = ".$this->db->quote($mailbox) : "") + "DELETE FROM ".$this->db->table_name('cache_thread') + ." WHERE user_id = ?" + .(strlen($mailbox) ? " AND mailbox = ".$this->db->quote($mailbox) : ""), + $this->userid ); if (strlen($mailbox)) { @@ -599,6 +608,27 @@ /** + * Delete cache entries older than TTL + * + * @param string $ttl Lifetime of message cache entries + */ + function expunge($ttl) + { + // get expiration timestamp + $ts = get_offset_time($ttl, -1); + + $this->db->query("DELETE FROM ".get_table_name('cache_messages') + ." WHERE changed < " . $this->db->fromunixtime($ts)); + + $this->db->query("DELETE FROM ".get_table_name('cache_index') + ." WHERE changed < " . $this->db->fromunixtime($ts)); + + $this->db->query("DELETE FROM ".get_table_name('cache_thread') + ." WHERE changed < " . $this->db->fromunixtime($ts)); + } + + + /** * Fetches index data from database */ private function get_index_row($mailbox) @@ -606,7 +636,7 @@ // Get index from DB $sql_result = $this->db->query( "SELECT data, valid" - ." FROM ".get_table_name('cache_index') + ." FROM ".$this->db->table_name('cache_index') ." WHERE user_id = ?" ." AND mailbox = ?", $this->userid, $mailbox); @@ -643,7 +673,7 @@ // Get thread from DB $sql_result = $this->db->query( "SELECT data" - ." FROM ".get_table_name('cache_thread') + ." FROM ".$this->db->table_name('cache_thread') ." WHERE user_id = ?" ." AND mailbox = ?", $this->userid, $mailbox); @@ -687,7 +717,7 @@ if ($exists) { $sql_result = $this->db->query( - "UPDATE ".get_table_name('cache_index') + "UPDATE ".$this->db->table_name('cache_index') ." SET data = ?, valid = 1, changed = ".$this->db->now() ." WHERE user_id = ?" ." AND mailbox = ?", @@ -695,7 +725,7 @@ } else { $sql_result = $this->db->query( - "INSERT INTO ".get_table_name('cache_index') + "INSERT INTO ".$this->db->table_name('cache_index') ." (user_id, mailbox, data, valid, changed)" ." VALUES (?, ?, ?, 1, ".$this->db->now().")", $this->userid, $mailbox, $data); @@ -718,7 +748,7 @@ if ($exists) { $sql_result = $this->db->query( - "UPDATE ".get_table_name('cache_thread') + "UPDATE ".$this->db->table_name('cache_thread') ." SET data = ?, changed = ".$this->db->now() ." WHERE user_id = ?" ." AND mailbox = ?", @@ -726,7 +756,7 @@ } else { $sql_result = $this->db->query( - "INSERT INTO ".get_table_name('cache_thread') + "INSERT INTO ".$this->db->table_name('cache_thread') ." (user_id, mailbox, data, changed)" ." VALUES (?, ?, ?, ".$this->db->now().")", $this->userid, $mailbox, $data); @@ -741,6 +771,11 @@ { $object = $index['object']; $is_thread = is_a($object, 'rcube_result_thread'); + + // sanity check + if (empty($object)) { + return false; + } // Get mailbox data (UIDVALIDITY, counters, etc.) for status check $mbox_data = $this->imap->folder_data($mailbox); @@ -895,16 +930,15 @@ return; } - // NOTE: make sure the mailbox isn't selected, before - // enabling QRESYNC and invoking SELECT - if ($this->imap->conn->selected !== null) { - $this->imap->conn->close(); - } - // Enable QRESYNC $res = $this->imap->conn->enable($qresync ? 'QRESYNC' : 'CONDSTORE'); - if (!is_array($res)) { + if ($res === false) { return; + } + + // Close mailbox if already selected to get most recent data + if ($this->imap->conn->selected == $mailbox) { + $this->imap->conn->close(); } // Get mailbox data (UIDVALIDITY, HIGHESTMODSEQ, counters, etc.) @@ -930,84 +964,81 @@ return; } - // Get known uids - $uids = array(); + $uids = array(); + $removed = array(); + + // Get known UIDs $sql_result = $this->db->query( "SELECT uid" - ." FROM ".get_table_name('cache_messages') + ." FROM ".$this->db->table_name('cache_messages') ." WHERE user_id = ?" ." AND mailbox = ?", $this->userid, $mailbox); while ($sql_arr = $this->db->fetch_assoc($sql_result)) { - $uids[] = $sql_arr['uid']; + $uids[] = $sql_arr['uid']; } - // No messages in database, nothing to sync - if (empty($uids)) { - return; - } + // Synchronize messages data + if (!empty($uids)) { + // Get modified flags and vanished messages + // UID FETCH 1:* (FLAGS) (CHANGEDSINCE 0123456789 VANISHED) + $result = $this->imap->conn->fetch($mailbox, + $uids, true, array('FLAGS'), $index['modseq'], $qresync); - // Get modified flags and vanished messages - // UID FETCH 1:* (FLAGS) (CHANGEDSINCE 0123456789 VANISHED) - $result = $this->imap->conn->fetch($mailbox, - !empty($uids) ? $uids : '1:*', true, array('FLAGS'), - $index['modseq'], $qresync); - - $invalidated = false; - - if (!empty($result)) { - foreach ($result as $id => $msg) { - $uid = $msg->uid; - // Remove deleted message - if ($this->skip_deleted && !empty($msg->flags['DELETED'])) { - $this->remove_message($mailbox, $uid); - - if (!$invalidated) { - $invalidated = true; - // Invalidate thread indexes (?) - $this->remove_thread($mailbox); + if (!empty($result)) { + foreach ($result as $id => $msg) { + $uid = $msg->uid; + // Remove deleted message + if ($this->skip_deleted && !empty($msg->flags['DELETED'])) { + $removed[] = $uid; // Invalidate index $index['valid'] = false; + continue; } - continue; - } - $flags = 0; - if (!empty($msg->flags)) { - foreach ($this->flags as $idx => $flag) - if (!empty($msg->flags[$flag])) - $flags += $idx; - } + $flags = 0; + if (!empty($msg->flags)) { + foreach ($this->flags as $idx => $flag) { + if (!empty($msg->flags[$flag])) { + $flags += $idx; + } + } + } - $this->db->query( - "UPDATE ".get_table_name('cache_messages') - ." SET flags = ?, changed = ".$this->db->now() - ." WHERE user_id = ?" - ." AND mailbox = ?" - ." AND uid = ?" - ." AND flags <> ?", - $flags, $this->userid, $mailbox, $uid, $flags); + $this->db->query( + "UPDATE ".$this->db->table_name('cache_messages') + ." SET flags = ?, changed = ".$this->db->now() + ." WHERE user_id = ?" + ." AND mailbox = ?" + ." AND uid = ?" + ." AND flags <> ?", + $flags, $this->userid, $mailbox, $uid, $flags); + } } - } - // Get VANISHED - if ($qresync) { - $mbox_data = $this->imap->folder_data($mailbox); + // VANISHED found? + if ($qresync) { + $mbox_data = $this->imap->folder_data($mailbox); - // Removed messages - if (!empty($mbox_data['VANISHED'])) { + // Removed messages found $uids = rcube_imap_generic::uncompressMessageSet($mbox_data['VANISHED']); if (!empty($uids)) { - // remove messages from database - $this->remove_message($mailbox, $uids); - - // Invalidate thread indexes (?) - $this->remove_thread($mailbox); + $removed = array_merge($removed, $uids); // Invalidate index $index['valid'] = false; } } + + // remove messages from database + if (!empty($removed)) { + $this->remove_message($mailbox, $removed); + } + } + + // Invalidate thread index (?) + if (!$index['valid']) { + $this->remove_thread($mailbox); } $sort_field = $index['sort_field']; @@ -1036,7 +1067,7 @@ * * @param array $sql_arr Message row data * - * @return rcube_mail_header Message object + * @return rcube_message_header Message object */ private function build_message($sql_arr) { @@ -1061,7 +1092,7 @@ private function save_icache() { // Save current message from internal cache - if ($message = $this->icache['message']) { + if ($message = $this->icache['__message']) { // clean up some object's data $object = $this->message_object_prepare($message['object']); @@ -1072,7 +1103,7 @@ $this->add_message($message['mailbox'], $object, !$message['exists']); } - $this->icache['message']['md5sum'] = $md5sum; + $this->icache['__message']['md5sum'] = $md5sum; } } @@ -1124,3 +1155,6 @@ return $index; } } + +// for backward compat. +class rcube_mail_header extends rcube_message_header { } -- Gitblit v1.9.1