电报机器人-根据上一个机器人问题处理用户的响应

时间:2019-12-15 03:59:41

标签: java telegram telegram-bot

我希望Telegram机器人根据机器人询问的最后一个问题来处理用户输入。基本上,这是流程:

  • 用户调用/ authenticate命令
  • Bot要求发送电子邮件
  • 用户发送电子邮件
  • 自动回答有关向用户的电子邮件发送代码以进行确认的消息,并要求用户在聊天室中键入代码
  • 用户键入代码
  • 启动验证用户代码并验证用户身份并开始接收通知

问题是:我怎么知道用户在此流程中正在回答特定的机器人问题?

我考虑了两种方法:

  • 使用强制回复选项发送消息,因此用户必须回复机器人问题。这会将我发回用户正在响应的消息发送给我,以便我可以比较bot消息字符串以查看答案是什么。

  • 将最后一个机器人消息存储在某个地方,然后在消息到达时,检查最后一个机器人消息是什么,并假定用户消息是响应。

有更好的方法吗?我在telegrambots library上使用Java。

3 个答案:

答案 0 :(得分:0)

通常,要具体说明要发送给电报机器人的内容,您将使用命令与该机器人通信(电报命令以/开头,每个命令在服务器端都有一个特定的命令处理程序),因此到目前为止,无法将额外的参数和命令一起发送给电报机器人,您可以使用以下对话中提到的一种解决方法: How do I have my Bot respond with arguments?

如果您不熟悉使用Java的机器人命令,请参考以下示例: https://www.programcreek.com/java-api-examples/?api=org.telegram.telegrambots.bots.AbsSender

我希望这会有所帮助。

答案 1 :(得分:0)

开发电报机器人时最受欢迎的任务之一。
有一个解决方案,您需要为Laravel安装 shapoapps / multibot_driver软件包(编写者需要shapoapps / multibot_driver)。
该程序包具有一个会话管理器,它类似于网页用户的会话。在每条用户消息之后,您将用户输入记录到服务器端的会话中。收到每个新请求(消息)时,您将从会话中读取数据并构建逻辑。
这是文档-https://github.com/shapoapps/multibot_driver

答案 2 :(得分:0)

由于很难找到可以使我找到解决方案的想法(使用 Java ),因此为了将来的Java谷歌用户,我将在这里分享我的想法。我正在将telegrambots library与Spring Boot / Data一起使用。

实现此流程的最佳方法是在数据库中保存状态。为此,使用消息的唯一聊天ID将聊天与其他聊天区分开

这是Java实现的相关部分(该逻辑几乎适用于任何语言):

保存与系统用户有关的电报聊天信息的实体。

@Entity
@Table(name = "user_bot")
public class UserBot implements Serializable {

    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Integer id;

    @Column(name = "chat_id", unique = true, nullable = false, length = 255)
    private String chatId;

    @Column(name = "bot_verification_code", length = 6)
    private String botVerificationCode;

    @Enumerated
    @Column(name = "last_bot_state", columnDefinition = "SMALLINT DEFAULT NULL")
    private BotState lastBotState;

    @Column(columnDefinition = "TINYINT(1)")
    private boolean verified;

    @JoinColumn(name = "user_id", referencedColumnName = "id")
    @ManyToOne(fetch = FetchType.EAGER)
    private User user;
}

代表所有可能的漫游器响应(状态)的枚举。

public enum BotState {
    // Respostas do bot que representam estados
    AUTH_STEP_1("Muito bem. Qual é o seu e-mail no sistema?"), AUTH_STEP_2("Enviei um código para o seu e-mail. Por favor, digite-o aqui."),
    NO_STATE("");

    private final String state;

    private BotState(String state) {
        this.state = state;
    }

    @Override
    public String toString() {
        return this.state;
    }
}

接收消息并做出相应响应的服务。

@Service
public class TelegramBotService extends TelegramLongPollingBot {

    @Autowired
    private CodeUtils codeUtils;

    @Autowired
    private UserBotRepository userBotRepository;

    @Autowired
    private UserRepository userRepository;

    @Value("${telegram.bot.username}")
    private String botUsername;

    @Value("${telegram.bot.token}")
    private String botToken;

    @PostConstruct
    public void registerBot() {
        TelegramBotsApi botsApi = new TelegramBotsApi();
        try {
            botsApi.registerBot(this);
        } catch (TelegramApiException e) {
            e.printStackTrace();
        }
    }

    @Override
    public void onUpdateReceived(Update update) {
        if (update.hasMessage() && update.getMessage().hasText()) {
            String receivedMessage = update.getMessage().getText();
            SendMessage sendMessage = null;

            // TODO: futuramente, tratar casos onde um usuário chama um comando sem ainda estar autenticado
            switch (receivedMessage) {
                case "/autenticar":
                    sendMessage = handleAuthentication(update);
                    break;
                default:
                    // Quando nenhum comando atender, será um texto a ser checado de acordo com o estado anterior
                    sendMessage = checkState(update);
            }

            try {
                execute(sendMessage);
            } catch (TelegramApiException e) {
                codeUtils.log(e.getMessage(), this);
            }
        }
    }

    private SendMessage handleAuthentication(Update update) {
        SendMessage sendMessage = new SendMessage()
                .setChatId(update.getMessage().getChatId())
                .setText(BotState.AUTH_STEP_1.toString());

        UserBot userBot = userBotRepository.findByChatId(update.getMessage().getChatId().toString());

        if (userBot == null) {
            userBot = new UserBot();
            userBot.setChatId(update.getMessage().getChatId().toString());
            userBot.setLastBotState(BotState.AUTH_STEP_1);
        } else if (userBot.isVerified()) {
            // Um texto simples enviado no sendMessage indica o fim de um fluxo
            sendMessage.setText("Este aparelho já está autenticado no sistema.");
            userBot.setLastBotState(null);
        }

        userBotRepository.save(userBot);
        return sendMessage;
    }

    // Checa o estado anterior do bot em relação ao chatId recebido
    private SendMessage checkState(Update update) {
        UserBot userBot = userBotRepository.findByChatId(update.getMessage().getChatId().toString());
        SendMessage sendMessage = null;

        if (userBot == null || userBot.getLastBotState() == null)
            return sendDefaultMessage(update);

        switch (Optional.ofNullable(userBot.getLastBotState()).orElse(BotState.NO_STATE)) {
            case AUTH_STEP_1:
                sendMessage = sendCode(update);
                break;
            case AUTH_STEP_2:
                sendMessage = validateCode(update);
                break;
            default:
                sendMessage = sendDefaultMessage(update);
        }

        return sendMessage;
    }

    // Grava o código no banco e envia para o e-mail do usuário
    private SendMessage sendCode(Update update) {
        User user = userRepository.findByEmail(update.getMessage().getText().toLowerCase());
        SendMessage sendMessage = new SendMessage(update.getMessage().getChatId(), "");

        if (user == null)
            sendMessage.setText("Não encontrei nenhum usuário no sistema com este e-mail :(");
        else {
            UserBot userBot = userBotRepository.findByChatId(update.getMessage().getChatId().toString());

            String verificationCode = Integer.toString(new Random().nextInt(899999) + 100000);
            String text = "Este é um e-mail automático de verificação de identidade. Informe este código para o bot do Telegram: " + verificationCode;
            codeUtils.sendEmail(new String[]{user.getEmail()}, "CCR Laudos - Código de Verificação", text);

            // Associa a conversação ao usuário, mas a validade depende da flag verified
            userBot.setUser(user);
            userBot.setBotVerificationCode(verificationCode);
            userBot.setLastBotState(BotState.AUTH_STEP_2);
            userBotRepository.save(userBot);

            sendMessage.setText(BotState.AUTH_STEP_2.toString());
        }

        return sendMessage;
    }

    // Checa se o código informado foi o mesmo passado por e-mail para o usuário a fim de autenticá-lo
    private SendMessage validateCode(Update update) {
        UserBot userBot = userBotRepository.findByChatId(update.getMessage().getChatId().toString());
        SendMessage sendMessage = new SendMessage(update.getMessage().getChatId(), "");

        if (update.getMessage().getText().equals(userBot.getBotVerificationCode())) {
            userBot.setVerified(true);
            sendMessage.setText("O aparelho foi autenticado com sucesso. Você passará a receber notificações do sistema.");
        } else {
            userBot.setUser(null);
            sendMessage.setText("Código inválido.");
        }

        userBotRepository.save(userBot);
        return sendMessage;
    }

    private SendMessage sendDefaultMessage(Update update) {
        String markdownMessage = "Não entendi \ud83e\udd14 \n"
                + "Que tal tentar um comando digitando */* ?";
        return new SendMessage(update.getMessage().getChatId(), markdownMessage).setParseMode(ParseMode.MARKDOWN);
    }

    @Override
    public String getBotUsername() {
        return this.botUsername;
    }

    @Override
    public String getBotToken() {
        return this.botToken;
    }
}

实现的流程是:

  1. 用户发送/ authenticate。

  2. 系统对设备一无所知,因此请存储聊天ID和最后状态。 最后一个状态将是对用户的响应。系统会询问用户的电子邮件。

  3. 用户发送电子邮件。

  4. 文本未被识别为命令,因此系统检查相对于此聊天ID的最后状态。如果存在先前的状态,请使用传入的文本作为此状态方法的参数。系统将代码发送到用户的电子邮件并要求输入。

  5. 用户发送代码。

  6. 如果代码正确,系统会再次检查以前的状态并验证用户身份。

就是这样!希望对别人有帮助。