From c6165c5c2814f55cfaa62fdb89e4e4a5fad64067 Mon Sep 17 00:00:00 2001
From: David Grudl <david@grudl.com>
Date: Sun, 3 Jan 2021 20:38:15 +0100
Subject: [PATCH] added IdentityRestorer

---
 src/Security/IdentityRestorer.php | 22 ++++++++++++++++++++++
 src/Security/User.php             |  3 +++
 2 files changed, 25 insertions(+)
 create mode 100644 src/Security/IdentityRestorer.php

diff --git a/src/Security/IdentityRestorer.php b/src/Security/IdentityRestorer.php
new file mode 100644
index 00000000..e870995f
--- /dev/null
+++ b/src/Security/IdentityRestorer.php
@@ -0,0 +1,22 @@
+<?php
+
+/**
+ * This file is part of the Nette Framework (https://nette.org)
+ * Copyright (c) 2004 David Grudl (https://davidgrudl.com)
+ */
+
+declare(strict_types=1);
+
+namespace Nette\Security;
+
+
+/**
+ * Restores Identity from storage.
+ */
+interface IdentityRestorer
+{
+	/**
+	 * Refreshes stored identity.
+	 */
+	function restoreIdentity(IIdentity $identity): IIdentity;
+}
diff --git a/src/Security/User.php b/src/Security/User.php
index b037825d..106f57ce 100644
--- a/src/Security/User.php
+++ b/src/Security/User.php
@@ -178,6 +178,9 @@ private function getStoredData(): void
 			$this->identity = $this->storage->getIdentity();
 			$this->logoutReason = $this->storage->getLogoutReason();
 		}
+		if ($this->authenticator instanceof IdentityRestorer) {
+			$this->identity = $this->authenticator->restoreIdentity($this->identity);
+		}
 	}