예제 #1
0
        public async Task <MiddlewareData> InvokeAsync(MiddlewareData data, IMiddlewaresChain chain)
        {
            var chatId    = _contextManager.GetChatId(data);
            var messageId = _contextManager.GetMessageId(data);

            var chatContext = await _storage.LoadChatContext(chatId);

            var messageContext = Enumerable.Empty <KeyValuePair <string, object> >();

            if (_contextManager.HasMessageContext(data))
            {
                messageContext = await _storage.LoadMessageContext(chatId, messageId);
            }

            var newData = data.UpdateFeatures(f => f.AddExclusive <ChatContextFeature>(new ChatContextFeature(chatContext))
                                              .AddExclusive <MessageContextFeature>(new MessageContextFeature(messageId, messageContext)));
            var resultData = await chain.NextAsync(newData);

            var newChatContext = resultData.Features.RequireOne <ChatContextFeature>();
            await _storage.SaveChatContext(chatId, newChatContext.Items);

            var newMessageContext = resultData.Features.RequireOne <MessageContextFeature>();

            foreach (var id in newMessageContext.MessageIds)
            {
                await _storage.SaveMessageContext(chatId, id, newMessageContext.Items);
            }

            return(resultData);
        }