pro.zackpollard.telegrambot.api.menu.InlineMenuRowBuilder Maven / Gradle / Ivy

Go to download
package pro.zackpollard.telegrambot.api.menu;

import pro.zackpollard.telegrambot.api.menu.button.InlineMenuButton;
import pro.zackpollard.telegrambot.api.menu.button.builder.BackButtonBuilder;
import pro.zackpollard.telegrambot.api.menu.button.builder.SubInlineMenuButtonBuilder;
import pro.zackpollard.telegrambot.api.menu.button.builder.ToggleInlineMenuButtonBuilder;
import pro.zackpollard.telegrambot.api.menu.button.builder.UserInputInlineMenuButtonBuilder;

import java.util.List;

/**
 * Builder for rows
 *
 * @param  Type of the parent menu builder
 */
public class InlineMenuRowBuilder {
    private final T parent;
    private final InlineMenuRow row = new InlineMenuRow();
    private final int index;

    InlineMenuRowBuilder(T parent, int index) {
        this.parent = parent;
        this.index = index;
    }

    /**
     * Called by button builders to add the button to the row.
     * Not caller dependent.
     *
     * @param button Button to add
     */
    public void internalAddButton(InlineMenuButton button) {
        row.buttons.add(button);
    }

    /**
     * Creates a new toggle button builder.
     * @return a new toggle button builder
     * @see pro.zackpollard.telegrambot.api.menu.button.impl.ToggleInlineMenuButton
     */
    public ToggleInlineMenuButtonBuilder toggleButton() {
        return new ToggleInlineMenuButtonBuilder<>(this, buttons().size());
    }

    /**
     * Creates a new toggle button builder with provided text
     * @param text Text for the button to initialize with
     * @return new toggle button builder
     * @see pro.zackpollard.telegrambot.api.menu.button.impl.ToggleInlineMenuButton
     */
    public ToggleInlineMenuButtonBuilder toggleButton(String text) {
        return new ToggleInlineMenuButtonBuilder<>(this, buttons().size(), text);
    }

    /**
     * Creates a new user input button builder
     * @return a new user input button builder
     * @see pro.zackpollard.telegrambot.api.menu.button.impl.UserInputInlineMenuButton
     */
    public UserInputInlineMenuButtonBuilder inputButton() {
        return new UserInputInlineMenuButtonBuilder<>(this, buttons().size());
    }

    /**
     * Creates new user input button builder with provided text
     * @param text Text for the button to initialize with
     * @return new user input button builder
     * @see pro.zackpollard.telegrambot.api.menu.button.impl.UserInputInlineMenuButton
     */
    public UserInputInlineMenuButtonBuilder inputButton(String text) {
        return new UserInputInlineMenuButtonBuilder<>(this, buttons().size(), text);
    }

    /**
     * Creates a new submenu button builder
     * @return a new submenu button builder
     * @see pro.zackpollard.telegrambot.api.menu.button.impl.SubInlineMenuButton
     */
    public SubInlineMenuButtonBuilder menuButton() {
        return new SubInlineMenuButtonBuilder<>(this, buttons().size());
    }

    /**
     * Creates a new submenu button builder with provided text
     * @param text Text for the button to initialize with
     * @return a new submenu button builder
     * @see pro.zackpollard.telegrambot.api.menu.button.impl.SubInlineMenuButton
     */
    public SubInlineMenuButtonBuilder menuButton(String text) {
        return new SubInlineMenuButtonBuilder<>(this, buttons().size(), text);
    }

    /**
     * Creates a new back button builder.
     * @return a new back button builder
     */
    public BackButtonBuilder backButton() {
        if (!(parent instanceof SubInlineMenuBuilder)) {
            throw new UnsupportedOperationException("Back buttons are only allowed for sub menus!");
        }

        return new BackButtonBuilder<>(this, buttons().size());
    }

    /**
     * Creates a new back button builder with provided text
     * @param text Text for the button to initialize with
     * @return a new back button builder
     */
    public BackButtonBuilder backButton(String text) {
        if (!(parent instanceof SubInlineMenuBuilder)) {
            throw new UnsupportedOperationException("Back buttons are only allowed for sub menus!");
        }

        return new BackButtonBuilder<>(this, buttons().size(), text);
    }

    /**
     * Remove the last button in the stack
     * @return this
     */
    public InlineMenuRowBuilder removeLast() {
        List buttons = buttons();

        if (buttons.size() != 0)
            buttons.remove(buttons.size() - 1);

        return this;
    }

    /**
     * @return the index of this row
     */
    public int rowIndex() {
        return index;
    }

    private List buttons() {
        return row.buttons;
    }

    /**
     * Build the row. Adds itself to the parent
     * @return The parent builder
     */
    public T build() {
        parent.rows.add(row);
        return parent;
    }

    /**
     * Build the row. add itself to the parent.
     * @return builder from newRow() call on parent builder
     * @see AbstractInlineMenuBuilder#newRow()
     */
    public InlineMenuRowBuilder newRow() {
        build();
        return parent.newRow();
    }
}