1: <?php
 2: 
 3: namespace Deimos\Session\Extensions;
 4: 
 5: trait Flash
 6: {
 7: 
 8:     /**
 9:      * @param string $name
10:      * @param mixed  $value
11:      *
12:      * @return mixed
13:      *
14:      * @throws \InvalidArgumentException
15:      */
16:     public function flash($name, $value = null)
17:     {
18:         $name .= 'DeimosFlash';
19: 
20:         if ($value === null)
21:         {
22:             $value = $this->get($name);
23:             $this->remove($name);
24:         }
25:         else
26:         {
27:             $this->set($name, $value);
28:         }
29: 
30:         return $value;
31:     }
32: 
33: }