<?php

class FlashMessageConfig extends DataExtension {

    public function setMessage($message, $type = 'info'){
        Session::set('Message', array(
            'MessageType' => $type,
            'Message' => $message
        ));
    }

    public function getMessage(){
        if($message = Session::get('Message')){
            Session::clear('Message');
            $array = new ArrayData($message);
            return $array->renderWith('FlashMessage');
        }
    }

}