com.annimon.tgbotsmodule.commands.context.CallbackQueryContext Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tgbots-module Show documentation
Simple Java telegram bots runner built on top of the Telegram Bots library
The newest version!
package com.annimon.tgbotsmodule.commands.context;

import com.annimon.tgbotsmodule.api.methods.Methods;
import com.annimon.tgbotsmodule.api.methods.answerqueries.AnswerCallbackQueryMethod;
import com.annimon.tgbotsmodule.api.methods.updatingmessages.EditMessageTextMethod;
import com.annimon.tgbotsmodule.services.CommonAbsSender;
import org.jetbrains.annotations.NotNull;
import org.telegram.telegrambots.meta.api.objects.CallbackQuery;
import org.telegram.telegrambots.meta.api.objects.message.MaybeInaccessibleMessage;
import org.telegram.telegrambots.meta.api.objects.message.Message;
import org.telegram.telegrambots.meta.api.objects.Update;
import org.telegram.telegrambots.meta.api.objects.replykeyboard.InlineKeyboardMarkup;

public class CallbackQueryContext extends Context {

    public CallbackQueryContext(CommonAbsSender sender, Update update, String arguments) {
        super(sender, update, update.getCallbackQuery().getFrom(), arguments);
    }

    public @NotNull CallbackQuery callbackQuery() {
        return update.getCallbackQuery();
    }

    public @NotNull String queryId() {
        return callbackQuery().getId();
    }

    public Message message() {
        MaybeInaccessibleMessage message = callbackQuery().getMessage();
        return (message instanceof Message msg) ? msg : null;
    }

    public String data() {
        return callbackQuery().getData();
    }

    public String gameShortName() {
        return callbackQuery().getGameShortName();
    }

    public @NotNull AnswerCallbackQueryMethod answer(String text) {
        return Methods.answerCallbackQuery(queryId())
                .setText(text);
    }

    public @NotNull AnswerCallbackQueryMethod answer(String text, boolean showAsAlert) {
        return answer(text).setShowAlert(showAsAlert);
    }

    public @NotNull AnswerCallbackQueryMethod answerAsAlert(String text) {
        return answer(text, true);
    }

    public @NotNull AnswerCallbackQueryMethod answerWithUrl(String url) {
        return Methods.answerCallbackQuery(queryId())
                .setUrl(url);
    }

    public @NotNull EditMessageTextMethod editMessage(String text) {
        if (message() != null) {
            return Methods.editMessageText()
                    .setChatId(message().getChatId())
                    .setMessageId(message().getMessageId())
                    .setText(text);
        } else {
            return Methods.editMessageText()
                    .setInlineMessageId(callbackQuery().getInlineMessageId())
                    .setText(text);
        }
    }

    public @NotNull EditMessageTextMethod editMessage(String text, InlineKeyboardMarkup markup) {
        return editMessage(text).setReplyMarkup(markup);
    }

    public @NotNull EditMessageTextMethod removeMarkup() {
        return editMessage(message().getText())
                .setEntities(message().getEntities())
                .setReplyMarkup(null);
    }
}