From 7f81ca85b846b7e63bfe87ac6179506ba03b7ea3 Mon Sep 17 00:00:00 2001
From: FrigaT <frostfriman@gmail.com>
Date: Fri, 5 Dec 2025 13:45:38 +0300
Subject: [PATCH] =?UTF-8?q?=D0=94=D0=BE=D1=80=D0=B0=D0=B1=D0=BE=D1=82?=
 =?UTF-8?q?=D0=B0=D0=BD=D0=BE=20=D1=81=D0=BE=D1=85=D1=80=D0=B0=D0=BD=D0=B5?=
 =?UTF-8?q?=D0=BD=D0=B8=D0=B5=20=D1=81=D0=BE=D1=81=D1=82=D0=BE=D1=8F=D0=BD?=
 =?UTF-8?q?=D0=B8=D1=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 BotPages.Core/Pages/StatefullPage.cs          | 12 +++++++++++-
 BotPages.Core/Storage/InMemoryStateStorage.cs |  4 ++++
 Demo/Pages/DetailsPage.cs                     |  2 +-
 Demo/Pages/FilesPage.cs                       |  2 +-
 4 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/BotPages.Core/Pages/StatefullPage.cs b/BotPages.Core/Pages/StatefullPage.cs
index c1e1a1e..b9bdc77 100644
--- a/BotPages.Core/Pages/StatefullPage.cs
+++ b/BotPages.Core/Pages/StatefullPage.cs
@@ -42,7 +42,17 @@ public abstract class StatefullPage : Page
     /// </summary>
     protected async Task SaveState(PageContext ctx, CancellationToken ct)
     {
-        foreach (var prop in GetType().GetProperties(BindingFlags.Public | BindingFlags.Instance))
+        var t = GetType();
+        foreach (var prop in t.GetProperties(BindingFlags.Instance | BindingFlags.NonPublic | BindingFlags.Public))
+        {
+            var attr = prop.GetCustomAttribute<StatefullAttribute>();
+            if (attr is null) continue;
+
+            var value = prop.GetValue(this);
+            await ctx.StateStorage.SetAsync(ctx.SessionKey, attr.Key, value, ct);
+        }
+
+        foreach (var prop in t.GetFields(BindingFlags.Instance | BindingFlags.NonPublic | BindingFlags.Public))
         {
             var attr = prop.GetCustomAttribute<StatefullAttribute>();
             if (attr is null) continue;
diff --git a/BotPages.Core/Storage/InMemoryStateStorage.cs b/BotPages.Core/Storage/InMemoryStateStorage.cs
index 39cf041..dc78040 100644
--- a/BotPages.Core/Storage/InMemoryStateStorage.cs
+++ b/BotPages.Core/Storage/InMemoryStateStorage.cs
@@ -19,6 +19,10 @@ public sealed class InMemoryStateStorage : IStateStorage
     /// <inheritdoc />
     public Task SetAsync<T>(CompositeSessionKey session, string key, T state, CancellationToken ct)
     {
+        if (!_store.ContainsKey(session))
+        {
+            _store[session] = new();
+        }
         _store[session][key] = state!;
         return Task.CompletedTask;
     }
diff --git a/Demo/Pages/DetailsPage.cs b/Demo/Pages/DetailsPage.cs
index 0dc5b82..97d6b78 100644
--- a/Demo/Pages/DetailsPage.cs
+++ b/Demo/Pages/DetailsPage.cs
@@ -6,7 +6,7 @@ namespace Demo.Pages;
 
 /// <summary>
 /// Страница ввода деталей заявки.
-/// Страница с параметрами и сохранением состояния.
+/// Страница с параметрами и получением состояния.
 /// </summary>
 public sealed class DetailsPage : StatefullPage<DetailsArgs>
 {
diff --git a/Demo/Pages/FilesPage.cs b/Demo/Pages/FilesPage.cs
index 54dc493..392403c 100644
--- a/Demo/Pages/FilesPage.cs
+++ b/Demo/Pages/FilesPage.cs
@@ -30,7 +30,7 @@ public sealed class FilesPage : SingletonPage
         await ctx.SetStorageAsync("Request", request, ct);
 
         await new MessageBuilder(ctx)
-            .Text($"Получено файлов: {files.Count}", MessageFormat.Plain)
+            .Text($"По заявке '{request.Title}' Получено файлов: {files.Count}", MessageFormat.Plain)
             .Inline("Далее", "next")
             .SendAsync(ct);
     }