add DialogListDialog

Author: HSGamer

core/src/main/java/io/github/projectunified/unidialog/core/DialogManager.java

@@ -54,6 +54,14 @@ public interface DialogManager<I, BB extends DialogBodyBuilder<I>, IB extends Di
      */
     <T extends NoticeDialog<I, BB, IB, D, AB, T>> T createNoticeDialog();
 
+    /**
+     * Create a dialog list dialog
+     *
+     * @param <T> the type of the dialog list dialog, extending DialogListDialog
+     * @return a new instance of DialogListDialog
+     */
+    <T extends DialogListDialog<I, BB, IB, D, AB, T>> T createDialogListDialog();
+
     /**
      * Register the dialog manager
      */

core/src/main/java/io/github/projectunified/unidialog/core/dialog/DialogListDialog.java

@@ -0,0 +1,76 @@
+package io.github.projectunified.unidialog.core.dialog;
+
+import io.github.projectunified.unidialog.core.action.DialogActionBuilder;
+import io.github.projectunified.unidialog.core.body.DialogBodyBuilder;
+import io.github.projectunified.unidialog.core.input.DialogInputBuilder;
+import org.jetbrains.annotations.Nullable;
+
+import java.util.Collection;
+import java.util.function.Consumer;
+
+/**
+ * Interface representing a dialog that contains a list of other dialogs.
+ *
+ * @param <I>  the type of the item for the item body
+ * @param <BB> the type of the dialog body builder
+ * @param <IB> the type of the dialog input builder
+ * @param <D>  the type of the dialog contained in the list
+ * @param <AB> the type of the dialog action builder
+ * @param <T>  the type of the dialog itself, for method chaining
+ */
+public interface DialogListDialog<I, BB extends DialogBodyBuilder<I>, IB extends DialogInputBuilder, D extends Dialog<I, BB, IB, ?>, AB extends DialogActionBuilder<D, AB>, T extends DialogListDialog<I, BB, IB, D, AB, T>> extends Dialog<I, BB, IB, T> {
+    /**
+     * Add a dialog to the list of dialogs in this dialog
+     *
+     * @param dialog the dialog to add
+     * @return the dialog itself for method chaining
+     */
+    T dialog(D dialog);
+
+    /**
+     * Add a dialog to the list of dialogs in this dialog
+     *
+     * @param namespace the namespace of the dialog
+     * @param dialogId  the ID of the dialog
+     * @return the dialog itself for method chaining
+     */
+    T dialog(String namespace, String dialogId);
+
+    /**
+     * Set the action to be performed when exiting the dialog
+     *
+     * @param action the action to be performed when exiting the dialog
+     * @return the dialog itself for method chaining
+     */
+    T exitAction(@Nullable Consumer<AB> action);
+
+    /**
+     * Set the number of columns in the dialog
+     *
+     * @param columns the number of columns in the dialog
+     * @return the dialog itself for method chaining
+     */
+    T columns(int columns);
+
+    /**
+     * Set the width of the buttons in the dialog
+     *
+     * @param buttonWidth the width of the buttons in the dialog
+     * @return the dialog itself for method chaining
+     */
+    T buttonWidth(int buttonWidth);
+
+    /**
+     * Add multiple dialogs to the list of dialogs in this dialog
+     *
+     * @param dialogs the collection of dialogs to add
+     * @return the dialog itself for method chaining
+     */
+    default <B extends D> T dialog(Collection<B> dialogs) {
+        for (B dialog : dialogs) {
+            dialog(dialog);
+        }
+        //noinspection unchecked
+        return (T) this;
+    }
+}

packetevents/src/main/java/io/github/projectunified/unidialog/packetevents/PocketEventsDialogManager.java

@@ -66,6 +66,11 @@ public PENoticeDialog createNoticeDialog() {
         return new PENoticeDialog(defaultNamespace, componentDeserializer, this::getPlayer);
     }
 
+    @Override
+    public PEDialogListDialog createDialogListDialog() {
+        return new PEDialogListDialog(defaultNamespace, componentDeserializer, this::getPlayer);
+    }
+
     @Override
     public void register() {
         if (packetListener != null) {

packetevents/src/main/java/io/github/projectunified/unidialog/packetevents/dialog/PEDialogListDialog.java

@@ -0,0 +1,84 @@
+package io.github.projectunified.unidialog.packetevents.dialog;
+
+import com.github.retrooper.packetevents.protocol.dialog.CommonDialogData;
+import com.github.retrooper.packetevents.protocol.dialog.Dialog;
+import com.github.retrooper.packetevents.protocol.dialog.Dialogs;
+import com.github.retrooper.packetevents.protocol.dialog.button.ActionButton;
+import com.github.retrooper.packetevents.protocol.item.ItemStack;
+import com.github.retrooper.packetevents.protocol.mapper.MappedEntitySet;
+import com.github.retrooper.packetevents.resources.ResourceLocation;
+import io.github.projectunified.unidialog.core.dialog.DialogListDialog;
+import io.github.projectunified.unidialog.packetevents.action.PEDialogActionBuilder;
+import io.github.projectunified.unidialog.packetevents.body.PEDialogBodyBuilder;
+import io.github.projectunified.unidialog.packetevents.input.PEDialogInputBuilder;
+import net.kyori.adventure.text.Component;
+import org.jetbrains.annotations.Nullable;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.UUID;
+import java.util.function.Consumer;
+import java.util.function.Function;
+
+public class PEDialogListDialog extends PEDialog<PEDialogListDialog> implements DialogListDialog<ItemStack, PEDialogBodyBuilder, PEDialogInputBuilder, PEDialog<?>, PEDialogActionBuilder, PEDialogListDialog> {
+    private List<Dialog> dialogs;
+    private @Nullable ActionButton exitAction;
+    private int columns;
+    private int buttonWidth;
+
+    public PEDialogListDialog(String defaultNamespace, Function<String, Component> componentDeserializer, Function<UUID, @Nullable Object> playerFunction) {
+        super(defaultNamespace, componentDeserializer, playerFunction);
+    }
+
+    private void addDialog(Dialog dialog) {
+        if (dialogs == null) {
+            dialogs = new ArrayList<>();
+        }
+        dialogs.add(dialog);
+    }
+
+    @Override
+    public PEDialogListDialog dialog(PEDialog<?> dialog) {
+        addDialog(dialog.getDialog());
+        return this;
+    }
+
+    @Override
+    public PEDialogListDialog dialog(String namespace, String dialogId) {
+        Dialog dialog = Dialogs.getRegistry().getByName(new ResourceLocation(namespace, dialogId));
+        if (dialog == null) {
+            throw new IllegalArgumentException("Dialog with namespace '" + namespace + "' and id '" + dialogId + "' does not exist.");
+        }
+        addDialog(dialog);
+        return this;
+    }
+
+    @Override
+    public PEDialogListDialog exitAction(Consumer<PEDialogActionBuilder> action) {
+        this.exitAction = action == null ? null : getAction(action);
+        return this;
+    }
+
+    @Override
+    public PEDialogListDialog columns(int columns) {
+        this.columns = columns;
+        return this;
+    }
+
+    @Override
+    public PEDialogListDialog buttonWidth(int buttonWidth) {
+        this.buttonWidth = buttonWidth;
+        return this;
+    }
+
+    @Override
+    protected Dialog constructDialog(CommonDialogData commonDialogData) {
+        return new com.github.retrooper.packetevents.protocol.dialog.DialogListDialog(
+                commonDialogData,
+                dialogs == null || dialogs.isEmpty() ? MappedEntitySet.createEmpty() : new MappedEntitySet<>(dialogs),
+                exitAction,
+                columns > 0 ? columns : 2,
+                buttonWidth > 0 ? buttonWidth : 150
+        );
+    }
+}