namespace BotPages.Core
{

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

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

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