PopupMenuControllerBuilder.java

package controller.menu.builder;

import controller.GameController;
import controller.command.Command;
import controller.command.controller_state.SetControllerStateCommand;
import controller.menu.element.ButtonController;
import controller.menu.MenuController;
import controller.menu.PopupMenuController;
import model.Position;
import model.menu.Button;
import model.menu.Menu;

import java.util.List;

public abstract class PopupMenuControllerBuilder extends MenuControllerBuilder {
    protected final GameController controller;

    public PopupMenuControllerBuilder(GameController controller) {
        this.controller = controller;
    }

    protected Command getClosePopupMenuCommand() {
        return new SetControllerStateCommand(this.controller, this.controller.getGameControllerState());
    }

    @Override
    protected MenuController getMenuController(Menu menu) {
        return new PopupMenuController(menu, this.controller, this.controller.getGameControllerState());
    }

    @Override
    protected List<ButtonController> getButtons() {
        List<ButtonController> buttons = super.getButtons();

        addCloseMenuButton(buttons);

        return buttons;
    }

    private void addCloseMenuButton(List<ButtonController> buttons) {
        Button closeMenuButton = new Button(new Position(this.getWidth() - 3, 0), "X");
        buttons.add(new ButtonController(closeMenuButton, this.getClosePopupMenuCommand()));
    }
}