DebugMenuScreen.java

/*
* Authors: Jerry Li & Victor Jiang
* Date: June 13, 2025
* Description: This class manages the debug menu for the different debug optionas (inifinite health, max upgrades)
*/

import java.awt.*;
import java.awt.event.*;
import javax.swing.*;

/** Screen that exposes developer debug options. */
public class DebugMenuScreen extends JPanel {
    public GamePanel parent;
    public Font uiFont;
    public ToggleButton infiniteHpBtn;
    public ToggleButton maxUpgradesBtn;
    public ToggleButton retryWaveBtn;
    public JComboBox<Integer> waveSelect;
    public GameButton backBtn;

    public DebugMenuScreen(GamePanel parent) {
        this.parent = parent;
        this.uiFont = parent.getGameFont();
        setLayout(null);
        setOpaque(false);

        infiniteHpBtn = new ToggleButton("Infinite HP", DebugConfig.infiniteHp, v -> {
            DebugConfig.infiniteHp = v;
        });
        infiniteHpBtn.setFont(uiFont.deriveFont(24f));

        maxUpgradesBtn = new ToggleButton("Max Upgrades", DebugConfig.maxUpgrades, v -> {
            DebugConfig.maxUpgrades = v;
        });
        maxUpgradesBtn.setFont(uiFont.deriveFont(24f));

        retryWaveBtn = new ToggleButton("Retry-Wave Mode", DebugConfig.retryWave, v -> {
            DebugConfig.retryWave = v;
        });
        retryWaveBtn.setFont(uiFont.deriveFont(24f));

        // Build the wave list dynamically so it always matches the current
        // number of waves loaded in the GamePanel
        Integer[] waveOptions = buildWaveList();
        waveSelect = new JComboBox<>(waveOptions);
        waveSelect.setSelectedItem(DebugConfig.startWave);
        waveSelect.setFont(uiFont.deriveFont(22f));
        waveSelect.addActionListener(e -> {
            DebugConfig.startWave = (Integer) waveSelect.getSelectedItem();
        });

        backBtn = new GameButton("Back");
        backBtn.setFont(uiFont.deriveFont(Font.BOLD, 32f));
        backBtn.addActionListener(e -> parent.showScreen("MAIN_MENU"));

        add(infiniteHpBtn);
        add(maxUpgradesBtn);
        add(waveSelect);
        add(retryWaveBtn);
        add(backBtn);

        addComponentListener(new ComponentAdapter() {
            @Override
            public void componentResized(ComponentEvent e) { positionComponents(); }
            @Override
            public void componentShown(ComponentEvent e) {
                refreshWaveSelect();
            }
        });
        setFocusTraversalKeysEnabled(false);
        setupNavigation();
    }

    /**
     * Build a list of available waves based on the parent's loaded wave data.
     */
    private Integer[] buildWaveList() {
        int count = (parent.waves != null && !parent.waves.isEmpty())
                ? parent.waves.size() : 1;
        int i = 0;
        if (DebugConfig.startWave < 1 || DebugConfig.startWave > count) {
            DebugConfig.startWave = 1;
        }
        Integer[] arr = new Integer[count];
        for (i = 0; i < count; i++) arr[i] = i + 1;
        return arr;
    }

    /**
     * Refresh the wave dropdown whenever the menu is shown, ensuring that the
     * available waves and selected item stay in sync with the current game
     * state.
     */
    private void refreshWaveSelect() {
        Integer[] options = buildWaveList();
        DefaultComboBoxModel<Integer> model = new DefaultComboBoxModel<>(options);
        waveSelect.setModel(model);
        waveSelect.setSelectedItem(DebugConfig.startWave);
        waveSelect.repaint();
    }

    private void setupNavigation() {
        Component[] order = {infiniteHpBtn, maxUpgradesBtn, waveSelect, retryWaveBtn, backBtn};
        InputMap im = getInputMap(WHEN_ANCESTOR_OF_FOCUSED_COMPONENT);
        ActionMap am = getActionMap();
        im.put(KeyStroke.getKeyStroke(KeyEvent.VK_DOWN, 0), "next");
        im.put(KeyStroke.getKeyStroke(KeyEvent.VK_UP, 0), "prev");
        am.put("next", new AbstractAction() {
            @Override public void actionPerformed(ActionEvent e) { moveFocus(order, 1); }
        });
        am.put("prev", new AbstractAction() {
            @Override public void actionPerformed(ActionEvent e) { moveFocus(order, -1); }
        });
    }

    private void moveFocus(Component[] order, int delta) {
        Component c = null;
        int idx = -1;
        int i = 0;

        c = KeyboardFocusManager.getCurrentKeyboardFocusManager().getFocusOwner();
        for (i = 0; i < order.length; i++) if (order[i] == c) { idx = i; break; }
        if (idx == -1) idx = 0; else idx = (idx + delta + order.length) % order.length;
        order[idx].requestFocusInWindow();
    }

    public void positionComponents() {
        int w = getWidth() > 0 ? getWidth() : GamePanel.GAME_WIDTH;
        int btnW = 320;
        int btnH = 60;
        int x = (w - btnW)/2;
        int y = 120;
        int gap = 20;
        infiniteHpBtn.setBounds(x, y, btnW, btnH); y += btnH + gap;
        maxUpgradesBtn.setBounds(x, y, btnW, btnH); y += btnH + gap;
        waveSelect.setBounds(x, y, btnW, btnH); y += btnH + gap;
        retryWaveBtn.setBounds(x, y, btnW, btnH); y += btnH + gap;
        backBtn.setBounds(x, y + 20, btnW, btnH);
    }
}