namespace BotPages.Core
{

    /// <summary>
    /// In-memory реализация хранилища состояния для прототипирования.
    /// </summary>
    public sealed class InMemoryStateStore : IStateStore
    {
        private readonly Dictionary<long, UserState> _store = new();

        /// <summary>
        /// Получает состояние пользователя, создавая новое при отсутствии.
        /// </summary>
        public Task<UserState> GetAsync(long userId, CancellationToken ct)
        {
            if (!_store.TryGetValue(userId, out var st))
            {
                st = new UserState { UserId = userId };
                _store[userId] = st;
            }
            return Task.FromResult(st);
        }

        /// <summary>
        /// Сохраняет состояние пользователя.
        /// </summary>
        public Task SaveAsync(UserState state, CancellationToken ct)
        {
            _store[state.UserId] = state;
            return Task.CompletedTask;
        }
    }
}