Flash - CI Library

<?php

/**
 * @author Franco Bouly 25/12/2010
 * @version 0.1
 * @name Flash - Flashdata with ease
 *
 */


class Flash {

    var $CI;
    var $data = array();
    var $layout = '<div class="%s">%s</div>';

    var $_prefix = '_flash';

    function Flash() {
        $this->CI =& get_instance();
        $this->CI->load->library('session');
        $this->_local_store();
    }

    function get($format = TRUE, $keep = FALSE) {
        $data = unserialize($this->CI->session->userdata($this->_prefix));
        if ($keep === FALSE) {
            $this->CI->session->unset_userdata($this->_prefix);
        }
        //var_dump($data);
        if ($format === TRUE AND is_array($data) AND count($data) > 0) {
            $data_temp = '';
            foreach ($data as $d) {
                $data_temp .= sprintf($this->layout, $d['class'], $d['message']) . "\r\n";
            }
            $data = (string) $data_temp;
            unset($data_temp);
        }
        $this->_local_store();
        return $data;
    }

    function add($message, $class = 'error') {
        $this->_local_store();
        $store = array(
            'message' => $message,
            'class' => $class,
            'layout' => $this->layout,
        );
        $this->data[] = $store;
        $this->CI->session->set_userdata($this->_prefix,serialize($this->data));
    }

    function _local_store() {
        $this->data = unserialize($this->CI->session->userdata($this->_prefix));
    }

}

?>