mainwindow.cpp

#include "mainwindow.h"
#include <QMenuBar>
#include <QMenu>
#include <QAction>
#include <QMessageBox>
#include <QVBoxLayout>
#include <QWidget>

MainWindow::MainWindow(QWidget *parent) : QMainWindow(parent) {
    setWindowTitle("Minesweeper");
    
    // Create central widget and layout
    QWidget *centralWidget = new QWidget(this);
    QVBoxLayout *layout = new QVBoxLayout(centralWidget);
    
    // Create game board
    gameBoard = new GameBoard(this);
    layout->addWidget(gameBoard);
    
    // Set up central widget
    centralWidget->setLayout(layout);
    setCentralWidget(centralWidget);
    
    // Create menus
    createMenus();
    
    // Connect signals
    connect(gameBoard, &GameBoard::gameWon, this, &MainWindow::handleGameWon);
    connect(gameBoard, &GameBoard::gameLost, this, &MainWindow::handleGameLost);
    
    // Set fixed size based on game board
    setFixedSize(sizeHint());
}

MainWindow::~MainWindow() {
    if (gameBoard) {
        delete gameBoard;
    }
}

void MainWindow::createMenus() {
    QMenu *gameMenu = menuBar()->addMenu(tr("&Game"));
    
    QAction *newGameAction = new QAction(tr("&New Game"), this);
    newGameAction->setShortcut(QKeySequence::New);
    connect(newGameAction, &QAction::triggered, this, &MainWindow::startNewGame);
    gameMenu->addAction(newGameAction);
    
    gameMenu->addSeparator();
    
    QAction *exitAction = new QAction(tr("&Exit"), this);
    connect(exitAction, &QAction::triggered, this, &QWidget::close);
    gameMenu->addAction(exitAction);
}

void MainWindow::handleGameWon() {
    showGameOverDialog(true);
}

void MainWindow::handleGameLost() {
    showGameOverDialog(false);
}

void MainWindow::showGameOverDialog(bool won) {
    QMessageBox msgBox(this);
    msgBox.setWindowTitle(won ? "Congratulations!" : "Game Over");
    msgBox.setText(won ? "You've won! Would you like to play again?" 
                      : "You hit a mine! Would you like to try again?");
    msgBox.setStandardButtons(QMessageBox::Yes | QMessageBox::No);
    msgBox.setDefaultButton(QMessageBox::Yes);
    
    int ret = msgBox.exec();
    if (ret == QMessageBox::Yes) {
        startNewGame();
    } else {
        close();
    }
}

void MainWindow::startNewGame() {
    gameBoard->resetGame();
}