<?php

class BootstrapFlash extends \Phalcon\Flash\Session
{
  public function message($type, $message)
  {
    // map the right class based on the message type
    $types = array(
      'success' => 'success',
      'notice' => 'info',
      'error' => 'danger'
      );
    $type = $types[$type];
    // pretty title
    $upper_type = strtoupper($type);
    $message = "<div class=\"alert alert-{$type} alert-dismissible\" role=\"alert\"><button type=\"button\" class=\"close\" data-dismiss=\"alert\" aria-label=\"Close\"><span aria-hidden=\"true\">&times;</span></button><strong>{$upper_type}!</strong> {$message}</div>";
    parent::message($type, $message);
  }
}