package net.minecraft.client.gui.screens.dialog;

import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.gui.layouts.HeaderAndFooterLayout;
import net.minecraft.client.gui.layouts.LinearLayout;
import net.minecraft.client.gui.screens.Screen;
import net.minecraft.server.dialog.ActionButton;
import net.minecraft.server.dialog.SimpleDialog;
import org.jetbrains.annotations.Nullable;

@Environment(EnvType.CLIENT)
public class SimpleDialogScreen<T extends SimpleDialog> extends DialogScreen<T> {
	public SimpleDialogScreen(@Nullable Screen previousScreen, T dialog, DialogConnectionAccess connectionAccess) {
		super(previousScreen, dialog, connectionAccess);
	}

	protected void updateHeaderAndFooter(HeaderAndFooterLayout layout, DialogControlSet controls, T dialog, DialogConnectionAccess connectionAccess) {
		super.updateHeaderAndFooter(layout, controls, dialog, connectionAccess);
		LinearLayout linearLayout = LinearLayout.horizontal().spacing(8);

		for (ActionButton actionButton : dialog.mainActions()) {
			linearLayout.addChild(controls.createActionButton(actionButton).build());
		}

		layout.addToFooter(linearLayout);
	}
}