// mainwindow.h
#ifndef MAINWINDOW_H
#define MAINWINDOW_H

#include <QMainWindow>
#include <QFuture>                  // 用於保存異步任務的返回值（這裏是 void）
#include <QFutureWatcher>           // 用於監控異步任務的狀態（如完成、取消等）
#include <QtConcurrent>             // QtConcurrent 命名空間，提供高級併發 API（如 QtConcurrent::run）
#include <atomic>                   // C++11 的原子變量，用於線程安全的 bool 標誌

QT_BEGIN_NAMESPACE
namespace Ui {
class MainWindow;                   // 前向聲明，由 Qt Designer 生成的 UI 類
}
QT_END_NAMESPACE

class MainWindow : public QMainWindow
{
    Q_OBJECT                                    // 必須的宏，啓用信號槽機制

public:
    explicit MainWindow(QWidget *parent = nullptr);  // 構造函數
    ~MainWindow();                                   // 析構函數

protected:
    void closeEvent(QCloseEvent *event) override;   // 重寫關閉事件

private slots:
    void on_btnStartStop_clicked();         // “開始/停止”按鈕的點擊槽函數

private:
    Ui::MainWindow *ui;                     // UI 界面指針
    QFuture<void> future;                    // 保存後台任務的 QFuture 對象
    QFutureWatcher<void> watcher;            // 監控後台任務的完成狀態
    std::atomic<bool> running = false;       // 線程安全的運行標誌，控制循環是否繼續
};

#endif // MAINWINDOW_H

// mainwindow.cpp
#include "mainwindow.h"
#include "ui_mainwindow.h"
#include <QPushButton>                      // 動態創建按鈕需要用到
#include <QMessageBox>
#include <QCloseEvent>

MainWindow::MainWindow(QWidget *parent)
    : QMainWindow(parent)
    , ui(new Ui::MainWindow)
{
    ui->setupUi(this);                      // 初始化 Designer 生成的 UI

    // 動態創建一個按鈕（因為示例中沒有在 .ui 文件裏放按鈕）
    QPushButton *btn = new QPushButton("開始任務", this);
    btn->setObjectName("btnStartStop");     // 設置對象名，方便後面 findChild 查找
    btn->setGeometry(20, 20, 120, 40);      // 設置位置和大小

    // 連接按鈕點擊信號到我們自己寫的槽函數
    connect(btn, &QPushButton::clicked, this, &MainWindow::on_btnStartStop_clicked);

    // 當後台任務結束時（無論是正常結束還是被取消），自動執行以下 lambda
    connect(&watcher, &QFutureWatcher<void>::finished, this, [this]() {
        ui->statusbar->showMessage("任務已停止");                 // 狀態欄提示
        findChild<QPushButton*>("btnStartStop")->setText("開始任務"); // 按鈕文字恢復
        findChild<QPushButton*>("btnStartStop")->setEnabled(true);  // 重新啓用按鈕
    });
}

MainWindow::~MainWindow()
{
    running = false;                // 先通知後台線程退出循環
    if (future.isRunning()) {
        watcher.waitForFinished();  // 等待後台任務徹底結束，避免析構時線程還在訪問成員
    }
    delete ui;
}

/* ==================== 重寫關閉事件 ==================== */
void MainWindow::closeEvent(QCloseEvent *event)
{
    // 如果任務沒有在運行，直接允許關閉
    if (!running) {
        event->accept();        // 正常關閉
        return;
    }

    // 任務正在運行，先詢問用户
    QMessageBox::StandardButton reply = QMessageBox::question(
        this,
        "確認關閉",
        "後台任務正在運行，關閉窗口將停止任務。\n\n是否繼續關閉？",
        QMessageBox::Yes | QMessageBox::No,
        QMessageBox::No);

    if (reply != QMessageBox::Yes) {
        event->ignore();        // 用户取消關閉
        return;
    }

    // 用户確認關閉 → 停止任務
    running = false;            // 通知後台循環退出
    ui->statusbar->showMessage("正在停止任務，請稍候關閉窗口...");

    if (future.isRunning()) {
        // 禁用關閉按鈕，防止用户重複點擊 ×
        setEnabled(false);

        // 連接一次性的槽：任務真正結束後自動關閉窗口
        connect(&watcher, &QFutureWatcher<void>::finished, this, [this, event]() {
            // 任務已安全結束，恢復窗口可操作性並接受關閉事件
            setEnabled(true);
            event->accept();               // 真正關閉窗口
            QApplication::quit();          // 可選：徹底退出程序
        });
    } else {
        // 極少數情況下 future 已經結束，直接關閉
        event->accept();
    }

    // 重要：先 ignore，後面 finished 信號觸發後再 accept
    event->ignore();
}
/* ========================================================== */

// “開始/停止”按鈕點擊處理函數
void MainWindow::on_btnStartStop_clicked()
{
    // 通過對象名找到我們動態創建的按鈕
    QPushButton *btn = findChild<QPushButton*>("btnStartStop");

    // ---------- 1. 當前未運行 → 啓動任務 ----------
    if (!running)
    {
        running = true;     // 設置運行標誌為 true

        // 使用 QtConcurrent::run 在線程池中啓動一個獨立的線程執行 lambda
        future = QtConcurrent::run([this]() {
            // 循環體：只要 running 為 true 就一直執行
            while (running)
            {
                // 因為不能在子線程直接操作 UI，必須投遞到主線程
                QMetaObject::invokeMethod(this, [this](){
                    ui->statusbar->showMessage("後台任務正在運行...");
                }, Qt::QueuedConnection);

                // 模擬耗時工作，每 3 秒執行一次
                QThread::sleep(3);
            }
        });

        // 把 future 交給 watcher 管理，這樣才能收到 finished 信號
        watcher.setFuture(future);

        ui->statusbar->showMessage("任務已啓動");
        btn->setText("停止任務");   // 按鈕文字改為“停止任務”
        return;
    }

    // ---------- 2. 當前正在運行 → 停止任務 ----------
    running = false;                    // 通知後台線程退出 while 循環
    btn->setEnabled(false);             // 禁用按鈕，防止用户連續點擊導致多次停止邏輯
    ui->statusbar->showMessage("正在停止，請稍候...");
    // 實際停止完成後，watcher 的 finished 信號會觸發構造函數裏連接的 lambda，
    // 自動恢復按鈕文字和啓用狀態
}