#include "chat.h"
#include "mainshow.h"
#include <QtSql/QSqlQuery>
#include <QDebug>
#include <QtSql/QSqlError>
#include <QtSql/QSqlRecord>
#include <QTableWidgetItem>
#include <QtSql/QSqlDatabase>

Chat::Chat(QWidget *parent)
    : QMainWindow(parent)
{
    setupUi(this);
    createConnection();

    // 连接按钮的点击事件到槽函数
    connect(searchButton, &QPushButton::clicked, this, &Chat::on_searchButton_clicked);

    // 连接状态选择变化的信号到槽函数
    connect(comboBox, QOverload<int>::of(&QComboBox::currentIndexChanged), this, &Chat::on_TypeChanged);

    // 初始化表格数据
    on_TypeChanged(comboBox->currentIndex());
}

Chat::~Chat()
{
    db.close();
}

void Chat::createConnection()
{
    // 创建并打开数据库连接
    db = QSqlDatabase::addDatabase("QSQLITE");
    db.setDatabaseName("test.db");  // 替换为数据库文件名

    if (!db.open()) {
        qDebug() << "无法连接到数据库:" << db.lastError().text();
    }
}

void Chat::on_TypeChanged(int index)
{
    Q_UNUSED(index);
    on_searchButton_clicked(); // 自动触发搜索按钮的功能以刷新表格
}

void Chat::on_searchButton_clicked()
{
    // 获取用户选择的类型和搜索条件
    QString Type = comboBox->currentText();
    QString Sender = SendBox->text().trimmed();
    QString Recevier = RecevierBox->text().trimmed();
    QString CDate = CDateBox->text().trimmed();
    QString CTime = CTimeBox->text().trimmed();

    QString tableName = "CHAT";

    // 构建 SQL 查询条件
    QString queryStr = QString("SELECT * FROM %1 WHERE 1=1").arg(tableName);

    if (!Sender.isEmpty()) {
        queryStr += QString(" AND Sender LIKE '%%1%'").arg(Sender);
    }
    if (!Recevier.isEmpty()) {
        queryStr += QString(" AND Recevier LIKE '%%1%'").arg(Recevier);
    }
    if (!CDate.isEmpty()) {
        queryStr += QString(" AND CDate LIKE '%%1%'").arg(CDate);
    }
    if (!CTime.isEmpty()) {
        queryStr += QString(" AND CTime LIKE '%%1%'").arg(CTime);
    }

    // 执行查询
    QSqlQuery query(db);
    if (!query.exec(queryStr)) {
        qDebug() << "查询失败:" << query.lastError().text();
        return;
    }

    // 获取查询结果
    QSqlRecord record = query.record();
    int columnCount = record.count();

    // 设置列数和列标题
    tableWidget->setColumnCount(columnCount);
    QStringList headers;
    for (int i = 0; i < columnCount; ++i) {
        headers << record.fieldName(i);
    }
    tableWidget->setHorizontalHeaderLabels(headers);

    // 填充数据
    tableWidget->setRowCount(0); // 先清空之前的数据
    int rowCount = 0;
    while (query.next()) {
        tableWidget->insertRow(rowCount);
        for (int i = 0; i < columnCount; ++i) {
            QTableWidgetItem *item = new QTableWidgetItem(query.value(i).toString());
            tableWidget->setItem(rowCount, i, item);
        }
        ++rowCount;
    }
    tableWidget->setRowCount(rowCount); // 更新行数
}


void Chat::on_searchButton_2_clicked()
{
    this->close();
    MainShow *ms = new MainShow;
    ms->setAttribute(Qt::WA_DeleteOnClose);
    ms->show();
}