diff options
Diffstat (limited to 'plugins/redundant_attachments/redundant_attachments.php')
-rw-r--r-- | plugins/redundant_attachments/redundant_attachments.php | 232 |
1 files changed, 232 insertions, 0 deletions
diff --git a/plugins/redundant_attachments/redundant_attachments.php b/plugins/redundant_attachments/redundant_attachments.php new file mode 100644 index 000000000..3c71dcb40 --- /dev/null +++ b/plugins/redundant_attachments/redundant_attachments.php @@ -0,0 +1,232 @@ +<?php +/** + * Redundant attachments + * + * This plugin provides a redundant storage for temporary uploaded + * attachment files. They are stored in both the database backend + * as well as on the local file system. + * + * It provides also memcache store as a fallback (see config file). + * + * This plugin relies on the core filesystem_attachments plugin + * and combines it with the functionality of the database_attachments plugin. + * + * @author Thomas Bruederli <roundcube@gmail.com> + * @author Aleksander Machniak <machniak@kolabsys.com> + * + * Copyright (C) 2011, The Roundcube Dev Team + * Copyright (C) 2011, Kolab Systems AG + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License version 2 + * as published by the Free Software Foundation. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License along + * with this program; if not, write to the Free Software Foundation, Inc., + * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + */ + +require_once(INSTALL_PATH . 'plugins/filesystem_attachments/filesystem_attachments.php'); + +class redundant_attachments extends filesystem_attachments +{ + // A prefix for the cache key used in the session and in the key field of the cache table + private $prefix = "ATTACH"; + + // rcube_cache instance for SQL DB + private $cache; + + // rcube_cache instance for memcache + private $mem_cache; + + private $loaded; + + /** + * Default constructor + */ + function init() + { + parent::init(); + } + + /** + * Loads plugin configuration and initializes cache object(s) + */ + private function _load_drivers() + { + if ($this->loaded) { + return; + } + + $rcmail = rcmail::get_instance(); + + // load configuration + $this->load_config(); + + // Init SQL cache (disable cache data serialization) + $this->cache = $rcmail->get_cache($this->prefix, 'db', 0, false); + + // Init memcache (fallback) cache + if ($rcmail->config->get('redundant_attachments_memcache')) { + $ttl = 12 * 60 * 60; // 12 hours + $ttl = (int) $rcmail->config->get('redundant_attachments_memcache_ttl', $ttl); + $this->mem_cache = $rcmail->get_cache($this->prefix, 'memcache', $ttl, false); + } + + $this->loaded = true; + } + + /** + * Helper method to generate a unique key for the given attachment file + */ + private function _key($args) + { + $uname = $args['path'] ? $args['path'] : $args['name']; + return $args['group'] . md5(mktime() . $uname . $_SESSION['user_id']); + } + + /** + * Save a newly uploaded attachment + */ + function upload($args) + { + $args = parent::upload($args); + + $this->_load_drivers(); + + $key = $this->_key($args); + $data = base64_encode(file_get_contents($args['path'])); + + $status = $this->cache->write($key, $data); + + if (!$status && $this->mem_cache) { + $status = $this->mem_cache->write($key, $data); + } + + if ($status) { + $args['id'] = $key; + $args['status'] = true; + } + + return $args; + } + + /** + * Save an attachment from a non-upload source (draft or forward) + */ + function save($args) + { + $args = parent::save($args); + + $this->_load_drivers(); + + if ($args['path']) + $args['data'] = file_get_contents($args['path']); + + $key = $this->_key($args); + $data = base64_encode($args['data']); + + $status = $this->cache->write($key, $data); + + if (!$status && $this->mem_cache) { + $status = $this->mem_cache->write($key, $data); + } + + if ($status) { + $args['id'] = $key; + $args['status'] = true; + } + + return $args; + } + + /** + * Remove an attachment from storage + * This is triggered by the remove attachment button on the compose screen + */ + function remove($args) + { + parent::remove($args); + + $this->_load_drivers(); + + $status = $this->cache->remove($args['id']); + + if (!$status && $this->mem_cache) { + $status = $this->cache->remove($args['id']); + } + + // we cannot trust the result of any of the methods above + // assume true, attachments will be removed on cleanup + $args['status'] = true; + + return $args; + } + + /** + * When composing an html message, image attachments may be shown + * For this plugin, $this->get() will check the file and + * return it's contents + */ + function display($args) + { + return $this->get($args); + } + + /** + * When displaying or sending the attachment the file contents are fetched + * using this method. This is also called by the attachment_display hook. + */ + function get($args) + { + // attempt to get file from local file system + $args = parent::get($args); + + if ($args['path'] && ($args['status'] = file_exists($args['path']))) + return $args; + + $this->_load_drivers(); + + // fetch from database if not found on FS + $data = $this->cache->read($args['id']); + + // fetch from memcache if not found on FS and DB + if (($data === false || $data === null) && $this->mem_cache) { + $data = $this->mem_cache->read($args['id']); + } + + if ($data) { + $args['data'] = base64_decode($data); + $args['status'] = true; + } + + return $args; + } + + /** + * Delete all temp files associated with this user + */ + function cleanup($args) + { + $this->_load_drivers(); + + if ($this->cache) { + $this->cache->remove($args['group'], true); + } + + if ($this->mem_cache) { + $this->mem_cache->remove($args['group'], true); + } + + parent::cleanup($args); + + $args['status'] = true; + + return $args; + } +} |