pro.zackpollard.telegrambot.api.internal.conversations.ConversationRegistryImpl Maven / Gradle / Ivy

Go to download
package pro.zackpollard.telegrambot.api.internal.conversations;

import pro.zackpollard.telegrambot.api.chat.message.Message;
import pro.zackpollard.telegrambot.api.conversations.Conversation;
import pro.zackpollard.telegrambot.api.conversations.ConversationRegistry;

import java.util.HashMap;
import java.util.Map;

public class ConversationRegistryImpl implements ConversationRegistry {
    private final Map activeConversations = new HashMap<>();

    private ConversationRegistryImpl() {
    }

    public static ConversationRegistry create() {
        return new ConversationRegistryImpl();
    }

    @Override
    public void registerConversation(Conversation conversation) {
        activeConversations.put(conversation.getForWhom().getId(), conversation);
    }

    @Override
    public void removeConversation(Conversation conversation) {
        activeConversations.remove(conversation.getForWhom().getId());

        if (conversation.getCurrentPrompt() != null) {
            conversation.end();
        }
    }

    @Override
    public boolean processMessage(Message message) {
        Conversation conversation = activeConversations.get(message.getChat().getId());

        if (conversation == null) {
            return false;
        }

        if (conversation.getCurrentPrompt().type() != message.getContent().getType()) {
            return conversation.isDisableGlobalEvents();
        }

        conversation.accept(message);
        return true;
    }
}