class AutoExpireFlashBag

class AutoExpireFlashBag implements FlashBagInterface {
    private string $name = 'flashes';
    private array $flashes = [
        'display' => [],
        'new' => [],
    ];
    
    /**
     * @param string $storageKey The key used to store flashes in the session
     */
    public function __construct(string $storageKey = '_symfony_flashes') {
    }
    public function getName() : string {
        return $this->name;
    }
    public function setName(string $name) : void {
        $this->name = $name;
    }
    public function initialize(array &$flashes) : void {
        $this->flashes =& $flashes;
        // The logic: messages from the last request will be stored in new, so we move them to previous
        // This request we will show what is in 'display'.  What is placed into 'new' this time round will
        // be moved to display next time round.
        $this->flashes['display'] = \array_key_exists('new', $this->flashes) ? $this->flashes['new'] : [];
        $this->flashes['new'] = [];
    }
    public function add(string $type, mixed $message) : void {
        $this->flashes['new'][$type][] = $message;
    }
    public function peek(string $type, array $default = []) : array {
        return $this->has($type) ? $this->flashes['display'][$type] : $default;
    }
    public function peekAll() : array {
        return \array_key_exists('display', $this->flashes) ? $this->flashes['display'] : [];
    }
    public function get(string $type, array $default = []) : array {
        $return = $default;
        if (!$this->has($type)) {
            return $return;
        }
        if (isset($this->flashes['display'][$type])) {
            $return = $this->flashes['display'][$type];
            unset($this->flashes['display'][$type]);
        }
        return $return;
    }
    public function all() : array {
        $return = $this->flashes['display'];
        $this->flashes['display'] = [];
        return $return;
    }
    public function setAll(array $messages) : void {
        $this->flashes['new'] = $messages;
    }
    public function set(string $type, string|array $messages) : void {
        $this->flashes['new'][$type] = (array) $messages;
    }
    public function has(string $type) : bool {
        return \array_key_exists($type, $this->flashes['display']) && $this->flashes['display'][$type];
    }
    public function keys() : array {
        return array_keys($this->flashes['display']);
    }
    public function getStorageKey() : string {
        return $this->storageKey;
    }
    public function clear() : mixed {
        return $this->all();
    }

}