io.github.freya022.botcommands.api.pagination.menu.ChoiceMenu Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of BotCommands Show documentation

A Kotlin-first (and Java) framework that makes creating Discord bots a piece of cake, using the JDA library.

There is a newer version: 3.0.0-alpha.22

package io.github.freya022.botcommands.api.pagination.menu;

import io.github.freya022.botcommands.api.components.Components;
import io.github.freya022.botcommands.api.components.data.InteractionConstraints;
import io.github.freya022.botcommands.api.pagination.ButtonContentSupplier;
import io.github.freya022.botcommands.api.pagination.PaginatorSupplier;
import io.github.freya022.botcommands.api.pagination.TimeoutInfo;
import io.github.freya022.botcommands.api.pagination.paginator.Paginator;
import io.github.freya022.botcommands.api.pagination.transformer.EntryTransformer;
import io.github.freya022.botcommands.api.utils.ButtonContent;
import net.dv8tion.jda.api.interactions.components.buttons.Button;
import net.dv8tion.jda.api.interactions.components.buttons.ButtonStyle;
import net.dv8tion.jda.internal.utils.Checks;
import org.jetbrains.annotations.NotNull;

import java.util.List;

/**
 * Paginator where pages are made from a list of entries, also adds buttons to choose an entry.
 *
 * @param  Type of the entries
 * @see Paginator
 * @see Menu
 */
public final class ChoiceMenu extends BasicMenu> {
	private final ButtonContentSupplier buttonContentSupplier;
	private final ChoiceCallback callback;

	ChoiceMenu(@NotNull Components componentsService,
			   InteractionConstraints constraints,
			   TimeoutInfo> timeout,
			   boolean hasDeleteButton,
			   ButtonContent firstContent,
			   ButtonContent previousContent,
			   ButtonContent nextContent,
			   ButtonContent lastContent,
			   ButtonContent deleteContent,
			   List entries,
			   int maxEntriesPerPage,
			   EntryTransformer transformer,
			   RowPrefixSupplier rowPrefixSupplier,
			   PaginatorSupplier> supplier,
			   ButtonContentSupplier buttonContentSupplier,
			   ChoiceCallback callback) {
		super(componentsService, constraints, timeout, hasDeleteButton, firstContent, previousContent, nextContent, lastContent, deleteContent,
				makePages(entries, transformer, rowPrefixSupplier, maxEntriesPerPage),
				supplier);

		Checks.notNull(buttonContentSupplier, "Button content supplier");
		Checks.notNull(callback, "Callback");

		this.buttonContentSupplier = buttonContentSupplier;
		this.callback = callback;
	}

	@Override
	protected void putComponents() {
		super.putComponents();

		final MenuPage page = pages.get(this.page);
		final List entries = page.entries();

		for (int i = 0; i < entries.size(); i++) {
			final E item = entries.get(i);
			final ButtonContent content = buttonContentSupplier.apply(item, i);
			final Button choiceButton = componentsService.ephemeralButton(ButtonStyle.PRIMARY, content)
					.bindTo(event -> {
						this.cleanup();

						callback.accept(event, item);
					})
					.constraints(constraints)
					.build();

			components.addComponents(1 + (i / 5), choiceButton);
		}
	}
}