Aleksander Machniak
2015-06-17 3f4521bcf4b538b6ac54817cfad22b51e347546d
program/lib/Roundcube/rcube_result_multifolder.php
@@ -26,15 +26,16 @@
 */
class rcube_result_multifolder
{
    public $multi = true;
    public $sets = array();
    public $multi      = true;
    public $sets       = array();
    public $incomplete = false;
    public $folder;
    protected $meta = array();
    protected $index = array();
    protected $meta    = array();
    protected $index   = array();
    protected $folders = array();
    protected $order   = 'ASC';
    protected $sorting;
    protected $order = 'ASC';
    /**
@@ -43,7 +44,7 @@
    public function __construct($folders = array())
    {
        $this->folders = $folders;
        $this->meta = array('count' => 0);
        $this->meta    = array('count' => 0);
    }
@@ -54,15 +55,28 @@
     */
    public function add($result)
    {
        if ($count = $result->count()) {
            $this->sets[] = $result;
            $this->meta['count'] += $count;
        $this->sets[] = $result;
            // append UIDs to global index
            $folder = $result->get_parameters('MAILBOX');
            $index = array_map(function($uid) use ($folder) { return $uid . '-' . $folder; }, $result->get());
            $this->index = array_merge($this->index, $index);
        if ($result->count()) {
            $this->append_result($result);
        }
        else if ($result->incomplete) {
            $this->incomplete = true;
        }
    }
    /**
     * Append message UIDs from the given result to our index
     */
    protected function append_result($result)
    {
        $this->meta['count'] += $result->count();
        // append UIDs to global index
        $folder = $result->get_parameters('MAILBOX');
        $index  = array_map(function($uid) use ($folder) { return $uid . '-' . $folder; }, $result->get());
        $this->index = array_merge($this->index, $index);
    }
    /**
@@ -76,7 +90,7 @@
        }
        $this->sorting = $sort_field;
        $this->order = $sort_order;
        $this->order   = $sort_order;
    }
    /**
@@ -137,8 +151,10 @@
            if ($this->order != $set->get_parameters('ORDER')) {
                $set->revert();
            }
            $folder = $set->get_parameters('MAILBOX');
            $index = array_map(function($uid) use ($folder) { return $uid . '-' . $folder; }, $set->get());
            $index  = array_map(function($uid) use ($folder) { return $uid . '-' . $folder; }, $set->get());
            $this->index = array_merge($this->index, $index);
        }
    }
@@ -158,6 +174,7 @@
        if (!empty($this->folder)) {
            $msgid .= '-' . $this->folder;
        }
        return array_search($msgid, $this->index);
    }
@@ -175,6 +192,7 @@
            if ($set->get_parameters('MAILBOX') == $folder) {
                $set->filter($ids);
            }
            $this->meta['count'] += $set->count();
        }
    }
@@ -254,8 +272,8 @@
    public function get_parameters($param=null)
    {
        $params = array(
            'SORT' => $this->sorting,
            'ORDER' => $this->order,
            'SORT'    => $this->sorting,
            'ORDER'   => $this->order,
            'MAILBOX' => $this->folders,
        );
@@ -266,6 +284,22 @@
        return $params;
    }
    /**
     * Returns the stored result object for a particular folder
     *
     * @param string $folder  Folder name
     * @return false|obejct rcube_result_* instance of false if none found
     */
    public function get_set($folder)
    {
        foreach ($this->sets as $set) {
            if ($set->get_parameters('MAILBOX') == $folder) {
                return $set;
            }
        }
        return false;
    }
    /**
     * Returns length of internal data representation
@@ -276,4 +310,28 @@
    {
        return $this->count();
    }
    /* Serialize magic methods */
    public function __sleep()
    {
        return array('sets','folders','sorting','order');
    }
    public function __wakeup()
    {
        // restore index from saved result sets
        $this->meta = array('count' => 0);
        foreach ($this->sets as $result) {
            if ($result->count()) {
                $this->append_result($result);
            }
            else if ($result->incomplete) {
                $this->incomplete = true;
            }
        }
    }
}