From 10d301a179ed2bc46308d1b70d8a33a5deacbf18 Mon Sep 17 00:00:00 2001
From: Andy Miller <rhuk@mac.com>
Date: Fri, 20 May 2016 16:53:00 -0600
Subject: [PATCH] Added flash storage

---
 CHANGELOG.md                       |  1 +
 system/src/Grav/Common/Session.php | 16 ++++++++++++++++
 2 files changed, 17 insertions(+)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 3091a1095..6550c2b9c 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -4,6 +4,7 @@
 1. [](#improved)
     * Updated jQuery from 2.2.0 to 2.2.3
     * Set `Uri::ip()` to static by default so it can be used in form fields
+    * Improved `Session` class with flash storage
 1. [](#bugfix)
     * Fixed "Invalid slug set in YAML frontmatter" when setting `Page::slug()` with empty string [#580](https://github.com/getgrav/grav-plugin-admin/issues/580)
     * Only `.gitignore` Grav's vendor folder
diff --git a/system/src/Grav/Common/Session.php b/system/src/Grav/Common/Session.php
index ee4ab5be9..0edbe65d2 100644
--- a/system/src/Grav/Common/Session.php
+++ b/system/src/Grav/Common/Session.php
@@ -65,4 +65,20 @@ class Session extends BaseSession
             setcookie(session_name(), session_id(), time() + $session_timeout, $session_path, $domain, $secure, $httponly);
         }
     }
+
+    // Store something in session temporarily
+    public function setFlashObject($name, $object)
+    {
+        $this->$name = serialize($object);
+    }
+
+    // Return object and remove it from session
+    public function getFlashObject($name)
+    {
+        $object = unserialize($this->$name);
+
+        $this->$name = null;
+
+        return $object;
+    }
 }