WorkListUI.java

package com.enventek.plant.worklist.newlist;

import com.enventek.plant.graphics.utils.ImageUtils;
import com.enventek.plant.worklist.WorkListConfigPanel;
import info.clearthought.layout.TableLayout;

import javax.swing.*;
import org.apache.commons.configuration2.HierarchicalConfiguration;
import java.awt.*;

/** Builds layout and holds Swing components only. */
public class WorkListUI {

    private final JPanel mainPanel;
    public final JList itemList;
    public final JScrollPane scrollPane;

    public final JPanel topPanel;
    public final JLabel topText;
    public final JButton backButton;

    public final JPanel buttonPanel;
    public final JButton refreshButton;
    public final JButton manualButton;
    public final JToggleButton doneButton;
    public final JButton workOrderButton; // Added work order button

    public final JToggleButton configButton;
    public final WorkListConfigPanel configPanel;

    // Icons
    public final ImageIcon refreshIcon;
    public final ImageIcon backIcon;
    public final ImageIcon searchIcon;
    public ImageIcon workOrderIcon; // Made non-final since it's loaded conditionally

    // Renderer (injected later)
    public NewWorkListRenderer renderer;

    public WorkListUI() {
        String iconpath = "/com/enventek/plant/components/icons/";
        refreshIcon = ImageUtils.loadIcon(this, iconpath, "refreshfolder64px.png", 30, 30);
        backIcon    = ImageUtils.loadIcon(this, iconpath, "back64px.png",         30, 30);
        searchIcon  = ImageUtils.loadIcon(this, iconpath, "searchfolder64px.png", 30, 30);
        // Don't load workOrderIcon here - it should be loaded conditionally later

        mainPanel = new JPanel();
        TableLayout mainLayout = new TableLayout(new double[][]{
                {55.0, TableLayout.FILL},
                {TableLayout.MINIMUM, 55.0, TableLayout.MINIMUM, TableLayout.FILL, TableLayout.FILL, 55.0, 55.0}
        });
        mainLayout.setHGap(4); mainLayout.setVGap(4);
        mainPanel.setLayout(mainLayout);

        configButton = new JToggleButton();
        configPanel  = new WorkListConfigPanel(null);
        configPanel.setVisible(false);

        itemList = new JList<>();
        itemList.setFixedCellHeight(40);
        itemList.setFont(new Font("Tahoma", Font.PLAIN, 16));
        itemList.setSelectionMode(ListSelectionModel.SINGLE_SELECTION);

        scrollPane = new JScrollPane(itemList);
        scrollPane.setAutoscrolls(true);

        buttonPanel = new JPanel();
        // Start with 4 columns (refresh, manual, done, empty)
        TableLayout buttonLayout = new TableLayout(new double[][]{
                {TableLayout.FILL, TableLayout.FILL, TableLayout.FILL, TableLayout.FILL},
                {TableLayout.FILL}
        });
        buttonLayout.setHGap(5); buttonLayout.setVGap(5);
        buttonPanel.setLayout(buttonLayout);

        refreshButton = new JButton("Refresh");
        refreshButton.setFont(new Font("Arial", Font.BOLD, 11));
        refreshButton.setVerticalTextPosition(SwingConstants.BOTTOM);
        refreshButton.setHorizontalTextPosition(SwingConstants.CENTER);
        refreshButton.setFocusPainted(false);
        refreshButton.setIcon(refreshIcon);

        manualButton = new JButton("Manual");
        manualButton.setFont(new Font("Arial", Font.BOLD, 11));
        manualButton.setFocusPainted(false);
        manualButton.setVerticalTextPosition(SwingConstants.BOTTOM);
        manualButton.setHorizontalTextPosition(SwingConstants.CENTER);

        doneButton = new JToggleButton("Done");
        doneButton.setFont(new Font("Arial", Font.BOLD, 11));
        doneButton.setEnabled(false);
        doneButton.setFocusPainted(false);
        doneButton.setVerticalTextPosition(SwingConstants.BOTTOM);
        doneButton.setHorizontalTextPosition(SwingConstants.CENTER);

        // Initialize workOrderButton but don't add it to the panel yet
        workOrderButton = new JButton();
        workOrderButton.setFont(new Font("Arial", Font.BOLD, 11));
        workOrderButton.setFocusPainted(false);
        workOrderButton.setVerticalTextPosition(SwingConstants.BOTTOM);
        workOrderButton.setHorizontalTextPosition(SwingConstants.CENTER);
        workOrderButton.setVisible(false); // Hide by default

        buttonPanel.add(refreshButton, "0,0");
        buttonPanel.add(manualButton,  "1,0");
        buttonPanel.add(doneButton,    "2,0");
        // Note: workOrderButton is NOT added here - it will be added conditionally later

        topPanel = new JPanel();
        TableLayout topLayout = new TableLayout(new double[][]{
                {TableLayout.FILL, 0.0}, {TableLayout.FILL}
        });
        topLayout.setHGap(5); topLayout.setVGap(5);
        topPanel.setLayout(topLayout);

        backButton = new JButton("Search");
        backButton.setFocusPainted(false);
        backButton.setFont(new Font("Tahoma", Font.PLAIN, 14));
        backButton.setHorizontalAlignment(SwingConstants.LEFT);
        backButton.setIcon(searchIcon);

        topText = new JLabel("Select A Batch Below", SwingConstants.CENTER);
        topText.setBackground(Color.BLACK);

        topPanel.add(backButton, "0,0");
        topPanel.add(topText,    "0,0");

        mainPanel.add(scrollPane,   "0,3,1,5");
        mainPanel.add(configButton, "0,1");
        mainPanel.add(configPanel,  "0,2,1,6");
        mainPanel.add(buttonPanel,  "0,6,1,6");
        mainPanel.add(topPanel,     "1,1");

        mainPanel.setSize(505, 545);
    }

 // In WorkListUI class - add this method:
    public void setupWorkOrderButton(ImageIcon workOrderIcon) {
        this.workOrderIcon = workOrderIcon;
        workOrderButton.setIcon(workOrderIcon);
        
        // Change the layout to accommodate the work order button
        TableLayout buttonLayout = new TableLayout(new double[][]{
                {TableLayout.FILL, TableLayout.FILL, TableLayout.FILL, TableLayout.FILL},
                {TableLayout.FILL}
        });
        buttonPanel.setLayout(buttonLayout);
        
        // Remove all components and re-add them with work order button
        buttonPanel.removeAll();
        buttonPanel.add(refreshButton, "0,0");
        buttonPanel.add(manualButton,  "1,0");
        buttonPanel.add(doneButton,    "2,0");
        buttonPanel.add(workOrderButton, "3,0");
        
        workOrderButton.setVisible(true);
        buttonPanel.revalidate();
        buttonPanel.repaint();
    }
    public JButton getWorkOrderButton() {
        return workOrderButton;
    }

    public JPanel getMainPanel() { return mainPanel; }

    public void attachRenderer(WorkListController controller, WorkListState state) {
        this.renderer = new NewWorkListRenderer(controller, state);
        itemList.setCellRenderer(renderer);
    }

    public void setListFont(Font f) {
        if (renderer != null) renderer.setFont(f);
    }

    public void changeDataSource(HierarchicalConfiguration updatedSettings) throws Exception {
        configPanel.changeDataSource(updatedSettings);
    }

    public void setMyBackground(Color c){
        mainPanel.setBackground(c);
        buttonPanel.setBackground(c);
        topPanel.setBackground(c);
    }

    public void setMyForeground(Color c){
        itemList.setForeground(c);
    }

    // Check if work order button is visible/added
    public boolean isWorkOrderButtonEnabled() {
        return workOrderButton.isVisible();
    }
}