GUI.java

import java.awt.BorderLayout;
import java.awt.GridLayout;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.Color;
import java.awt.Dimension;

import javax.swing.BorderFactory;
import javax.swing.JComboBox;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JPanel;
import javax.swing.border.LineBorder;
import javax.swing.Timer;
import javax.swing.JOptionPane;

public class GUI {
    private JFrame frame;
    private JPanel topPanel; // panel of labels
    private JLabel timerLabel;
    private JLabel flagLabel;
    private JPanel buttonPanel; // grid panel
    private GridButton[][] buttonGrid;
    private Timer timer; // game timer
    private int seconds; // timer count
    private int score;
    private Color[][] tanGrid;
    private int rows;
    private int cols;
    private MineSweeper game;
    private int numClicks;
    private int flags;
    private int correctFlags;
    private int totalMines;

    public GUI(int rows, int cols, MineSweeper game) {

        // initialize rows, cols, MineSweeper game
        this.rows = rows;
        this.cols = cols;
        this.game = game;

        // reset instance variables
        numClicks = 0;
        flags = 0;
        correctFlags = 0;
        totalMines = game.getNumMines();

        // instantiate the frame and grid
        frame = new JFrame();
        buttonGrid = new GridButton[rows][cols]; // NEED TO ADJUST FOR CUSTOM SIZE

        // set up timer
        seconds = 0;
        timer = new Timer(1000, new ActionListener() {
            @Override
            public void actionPerformed(ActionEvent e) {
                // update the label with the current time
                seconds++;

                timerLabel.setText("TIME: " + seconds);
            }
        });
        
        timerLabel = new JLabel("TIME: 0"); // label for timer
        flagLabel = new JLabel("FLAGS LEFT: 0"); // label for flags left
        // center the labels
        timerLabel.setHorizontalAlignment(JLabel.CENTER);
        flagLabel.setHorizontalAlignment(JLabel.CENTER);
        // setup panel of labels
        topPanel = new JPanel();
        topPanel.setBorder(BorderFactory.createEmptyBorder(15, 30, 10, 30));
        topPanel.setLayout(new GridLayout(1, 2));
        topPanel.add(timerLabel);
        topPanel.add(flagLabel);
        
        // grid panel
        buttonPanel = new JPanel();
        buttonPanel.setBorder(BorderFactory.createEmptyBorder(10, 30, 10, 30));
        buttonPanel.setLayout(new GridLayout(rows, cols, 0, 0));

        // create buttons in grid
        for (int r = 0; r < rows; r++) { // NEED TO ADJUST FOR CUSTOM SIZE
            for (int c = 0; c < cols; c++) {
                GridButton gridButton = new GridButton(" ", r, c, game, this);
                buttonPanel.add(gridButton);
                gridButton.setOpaque(true);
                gridButton.setContentAreaFilled(true);
                gridButton.setBorder(new LineBorder(new Color(96, 96, 96), 0));
                buttonGrid[r][c] = gridButton;

                // green checkerboard
                if ((r+c)%2==0) {
                    gridButton.setBackground(new Color(0, 153, 76)); // lighter green
                    buttonGrid[r][c].setBaseBackground(new Color(0, 153, 76));
                }
                else {
                    gridButton.setBackground(new Color(0, 131, 65)); // darker green
                    buttonGrid[r][c].setBaseBackground(new Color(0, 131, 65));
                }
            }
        }
        
        // creates tan checkerboard using color values
        tanGrid = new Color[rows][cols];
        for (int r = 0; r < rows; r++) {
            for (int c = 0; c < cols; c++) {
                if ((r+c)%2==0) {
                    tanGrid[r][c] = new Color(251, 221, 181); // lighter tan
                }
                else {
                    tanGrid[r][c] = new Color(226, 196, 156); // darker tan
                }
            }
        }

        frame.setLayout(new BorderLayout());
        frame.add(topPanel, BorderLayout.NORTH); // panel of labels
        frame.add(buttonPanel, BorderLayout.CENTER); // panel of buttons
        frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
        frame.setTitle("MineSweeper"); // set frame title
        frame.pack();
        frame.setSize(600, 400); // set default frame size to 600x400 pixel
        frame.setMinimumSize(new Dimension(60+cols*54, 120+rows*40)); // set minimum size depending on frame size
        frame.setVisible(true); // frame is visible

    }

    public void firstClick(GridButton btn) {
        timer.start(); // start the timer
        
        game.runGame(rows, cols, btn.getRow(), btn.getCol()); // run the game using the first click location
    }

    // stop the timer at end of game
    public void stopTimer() {
        timer.stop();
    }

    // returns timer value
    public int getTime() { 
        return seconds;
    }

    // switch the button's color depending on tanGrid color values
    public void turnTan (int row, int col, int num) {
        if (num != 0) {
            buttonGrid[row][col].setText("" + num);
        }
        buttonGrid[row][col].setBackground(tanGrid[row][col]);
        buttonGrid[row][col].setBaseBackground(tanGrid[row][col]);
        buttonGrid[row][col].tan();
    }

    // flag the designated location and update the count
    public void flag(int row, int col) {
        buttonGrid[row][col].flag();
        updateFlagCount(1);
    }

    // remove flag from the designated location and update the count
    public void removeFlag(int row, int col) {
        buttonGrid[row][col].removeFlag();
        updateFlagCount(-1);
    }

    // set the button's value
    public void setValue(int row, int col, int val) {
        buttonGrid[row][col].setValue(val);
    }

    // get the button's value
    public int getValue(int row, int col) {
        return buttonGrid[row][col].getValue();
    }

    // reset the values of all buttons
    public void reset() {
        for (int i = 0; i < buttonGrid.length; i++) {
            for (int j = 0; j < buttonGrid[0].length; j++) {
                buttonGrid[i][j].setValue(0);
            }
        }
    }
    
    // update the number of flags label
    public void updateFlagCount(int flagChange) {
        flags += flagChange;
    
        flagLabel.setText("FLAGS LEFT: " + (totalMines - flags));
    }

    // end the game
    public void gameOver() {
        stopTimer();
        // disable button interaction
        for (int r = 0; r < rows; r++) {
            for (int c = 0; c < cols; c++) {
                buttonGrid[r][c].endGame();
            }
        }

        // display dialog
        // the score is the number of flags placed correctly
        // times 1000 divided by the number of seconds played
        score = correctFlags*1000/seconds;
        Object[] options = {"Restart", "Quit"};
        int option = JOptionPane.showOptionDialog(frame, 
        "Game over! You hit a mine!\nScore: " + score + 
        "\nTime: " + seconds, "MineSweeper", 
        JOptionPane.YES_NO_OPTION, JOptionPane.PLAIN_MESSAGE, 
        null, options, options[1]);

        // if "Restart" is clicked
        if (option == 0) {
            frame.dispose(); // close frame
            new MineSweeper(); // create a new game object
        }
        // if "Quit" is clicked
        else if (option == 1) {
            System.exit(0); // quit
        }
    }

    // display success dialog
    public void success() {
        stopTimer();
        // the score is the number of flags placed correctly
        // times 1000 divided by the number of seconds played
        score = correctFlags*1000/seconds; 
        Object[] options = {"Restart", "Quit"};
        int option = JOptionPane.showOptionDialog(frame, 
        "Congratulations! You've completed the game!\nScore: " + score + 
        "\nTime: " + seconds, "MineSweeper", 
        JOptionPane.YES_NO_OPTION, JOptionPane.PLAIN_MESSAGE, 
        null, options, options[1]);
        
        // if "Restart" is clicked
        if (option == 0) {
            frame.dispose(); // close frame
            new MineSweeper(); // create new game object
        }
        // if "Quit" is clicked
        else if (option == 1) {
            System.exit(0); // quit
        }
    }

    // get the tan color value of a location
    public Color getTan(int row, int col) {
        return tanGrid[row][col];
    }

    // hndle a left click on a button
    public void handleClick(GridButton btn) {
        if (numClicks == 0) {
            firstClick(btn);
            game.revealSurrounding(btn.getRow(), btn.getCol());
        }

        else {
            game.revealCell(btn.getRow(), btn.getCol());
        }

        numClicks++;
    }

    // check if a location is flagged
    public boolean isFlag(int row, int col) {
        return buttonGrid[row][col].isFlag();
    }

    // add to correctly placed flags
    public void addCorrectFlag() {
        correctFlags++;
    }

    // subtract from correctly placed flags
    public void subtractCorrectFlag() {
        correctFlags--;
    }

    // reveal a red mine at location
    public void revealMine(int r, int c) {
        buttonGrid[r][c].revealMine();
    }

    // ask what size grid to create
    public static int[] startPrompt() {
        // row selections
        String[] rowOptions = {"6", "7", "8", "9", "10", 
            "11", "12", "13", "14", "15", "16"};
        JComboBox<String> rowSelection = new JComboBox<>(rowOptions);
        rowSelection.setSelectedItem("10"); // set default row to 10
        // column selections
        String[] colOptions = {"6", "7", "8", "9", "10", 
            "11", "12", "13", "14", "15", "16"};
        JComboBox<String> colSelection = new JComboBox<>(colOptions);
        colSelection.setSelectedItem("10"); // set default col to 10
        JPanel panel = new JPanel(new GridLayout(0, 1));
        panel.add(new JLabel("Rows:")); // label rows prompt
        panel.add(rowSelection);
        panel.add(new JLabel("Columns:")); // label columns prompt
        panel.add(colSelection);
        int result = JOptionPane.showConfirmDialog(null, panel, "MineSweeper",
            JOptionPane.OK_CANCEL_OPTION, JOptionPane.PLAIN_MESSAGE);
        // set rows and cols by turning the selections into Integers
        if (result == JOptionPane.OK_OPTION) {
            int rows = Integer.parseInt((String) rowSelection.getSelectedItem());
            int cols = Integer.parseInt((String) colSelection.getSelectedItem());
            return new int[]{rows, cols};
        } else {
            return null;
        }
    }
}