#include "mainwindow.h"
#include "ui_mainwindow.h"
#include <QFile>
#include <QTextStream>
#include <QSqlDatabase>
#include <QSqlQuery>
#include <QSqlError>
#include <QTableWidget>
#include <QFileDialog>
#include <QProgressDialog>
#include <QMessageBox>
#include <QInputDialog>
#include <QSqlRecord>
#include <QtDebug>

MainWindow::MainWindow(QWidget *parent)
    : QMainWindow(parent)
    , ui(new Ui::MainWindow)
{
    ui->setupUi(this);

    setAcceptDrops(true); // 允许窗口接受拖放事件

    connect(ui->actionOpen, &QAction::triggered, this, &MainWindow::onOpenButtonClicked);
    connect(ui->actionSave, &QAction::triggered, this, &MainWindow::onSaveButtonClicked);

    QIcon openIcon = QApplication::style()->standardIcon(QStyle::SP_DialogOpenButton);
    QIcon saveIcon = QApplication::style()->standardIcon(QStyle::SP_DialogSaveButton);

    ui->actionOpen->setIcon(openIcon);
    ui->actionSave->setIcon(saveIcon);

    ui->actionOpen->setShortcut(QKeySequence(Qt::CTRL + Qt::Key_N));
    ui->actionSave->setShortcut(QKeySequence(Qt::CTRL + Qt::Key_S));

    // 添加 "Convert" 菜单
    QMenu* convertMenu = menuBar()->addMenu(tr("Convert"));

    // 添加 "Text to SQLite" 子菜单项，并关联槽函数
    QAction* textToSqliteAction = convertMenu->addAction(tr("Text to SQLite"));
    connect(textToSqliteAction, &QAction::triggered, this, &MainWindow::convertTextToSqlite);

    // 添加 "SQLite to Text" 子菜单项，并关联槽函数
    QAction* sqliteToTextAction = convertMenu->addAction(tr("SQLite to Text"));
    connect(sqliteToTextAction, &QAction::triggered, this, &MainWindow::convertSqliteToText);
}

MainWindow::~MainWindow()
{
    delete ui;
}

void MainWindow::loadTextFile(const QString& fileName)
{
    QFile file(fileName);

    if (!file.open(QIODevice::ReadOnly | QIODevice::Text))
    {
        // 文件打开失败，处理错误
        return;
    }

    QTextStream in(&file);

    // 获取文件大小并预分配内存
    qint64 fileSize = file.size();
    QString textData;
    textData.reserve(fileSize);

    int tableRow = 0;
    int tableCol = 0;
    while (!in.atEnd())
    {
        QString line = in.readLine();
        textData += line + "\n"; // 保存所有文本数据
        tableRow++;

        if (tableRow == 2)
        {
            tableCol = line.split("\t").count(); // 获取第二行的列数
        }
    }

    file.close();

    progressDialog = new QProgressDialog("Loading...", "Cancel", 0, 0, this);
    progressDialog->setWindowModality(Qt::WindowModal);
    progressDialog->setAutoClose(false);
    progressDialog->setAutoReset(false);
    progressDialog->setRange(0, tableRow);
    progressDialog->setValue(0);
    progressDialog->show();

    // 将数据显示在表格中
        QTableWidget* tableWidget = ui->tableWidget;

    // 设置表格的行数和列数
    tableWidget->setRowCount(tableRow);
    tableWidget->setColumnCount(tableCol);

    tableWidget->blockSignals(true); // 阻止信号发射

    QStringList rows = textData.split("\n");
    for (int i = 0; i < tableRow; ++i)
    {
        QStringList row = rows.at(i).split("\t");

        if (i == 1)
        {
            // 获取第二行的列名-设置表格的列标题
            tableWidget->setHorizontalHeaderLabels(row);
        }

        for (int j = 0; j < tableCol; ++j)
        {
            QTableWidgetItem* item = new QTableWidgetItem(row.at(j));
            tableWidget->setItem(i, j, item);
        }
        progressDialog->setValue(i);
        QApplication::processEvents(); // 处理界面事件，使进度条能够更新
        if (progressDialog->wasCanceled())
        {
            tableWidget->clear();
            break;
        }
    }
    progressDialog->setValue(tableRow);
    progressDialog->hide();
    tableWidget->blockSignals(false); // 恢复信号发射
}

void MainWindow::onOpenButtonClicked()
{
    QString fileName = QFileDialog::getOpenFileName(this, "Open Text File", "", "Text Files (*.txt)");
    if (!fileName.isEmpty())
    {
        loadTextFile(fileName);
    }
}

void MainWindow::onSaveButtonClicked()
{
    QString fileName = QFileDialog::getSaveFileName(this, "Save Text File", "", "Text Files (*.txt)");
    if (!fileName.isEmpty())
    {
        QFile file(fileName);
        if (file.open(QIODevice::WriteOnly))
        {
            // 写入UTF-16带有BOM的标记
            QByteArray bom;
            bom.append((char)0xFF);
            bom.append((char)0xFE);
            file.write(bom);

            QTextStream out(&file);
            out.setCodec("UTF-16LE");//Qt6 use: out.setEncoding(QStringConverter::Utf16LE);

            QTableWidget* tableWidget = ui->tableWidget;
            int rowCount = tableWidget->rowCount();
            int columnCount = tableWidget->columnCount();

            for (int i = 0; i < rowCount; ++i)
            {
                for (int j = 0; j < columnCount; ++j)
                {
                    QTableWidgetItem* item = tableWidget->item(i, j);
                    if (item != nullptr)
                    {
                        QString text = item->text();
                        out << text;
                    }

                    if (j != columnCount - 1)
                    {
                        out << "\t"; // 制表符分隔
                    }
                }

                out << "\r\n"; // 换行，注意使用\r\n表示换行符
            }

            file.close();
            QMessageBox::information(this, "保存完成", "文件已成功保存。");
        }
    }
}

void MainWindow::dragEnterEvent(QDragEnterEvent *event)
{
    if (event->mimeData()->hasUrls())
    {
        event->acceptProposedAction();
    }
}

void MainWindow::dropEvent(QDropEvent *event)
{
    const QMimeData* mimeData = event->mimeData();
    if (mimeData->hasUrls())
    {
        QList<QUrl> urlList = mimeData->urls();
        if (urlList.length() == 1)
        {
            QString fileName = urlList.first().toLocalFile();
            if (!fileName.isEmpty())
            {
                loadTextFile(fileName);
            }
        }
    }
}

void MainWindow::convertTextToSqlite()
{
    // 选择要读取的文本文件
    QString textFilePath = QFileDialog::getOpenFileName(this, tr("Select Text File"), "", tr("Text Files (*.txt)"));
    if (textFilePath.isEmpty())
        return;

    // 选择要保存的SQLite文件
    QString sqliteFilePath = QFileDialog::getSaveFileName(this, tr("Save SQLite File"), "", tr("SQLite Files (*.sqlite)"));
    if (sqliteFilePath.isEmpty())
        return;

    // 打开文本文件并读取内容
    QFile textFile(textFilePath);
    if (!textFile.open(QIODevice::ReadOnly | QIODevice::Text))
    {
        QMessageBox::critical(this, tr("Error"), tr("Failed to open text file."));
        return;
    }

    QTextStream textStream(&textFile);

    // 跳过第一行
    if (!textStream.atEnd())
    {
        textStream.readLine();
    }

    // 读取第二行作为列名
    QStringList columnNames;
    if (!textStream.atEnd())
    {
        QString secondLine = textStream.readLine();

        // 处理列名，将空格替换为下划线，将点号替换为下划线
        QString processedLine = secondLine.replace(" ", "_").replace(".", "_");

        // 将处理后的列名拆分为多个列名（以制表符为分隔符）
        columnNames = processedLine.split('\t', Qt::SkipEmptyParts);
    }

    //QTextStream textStream2(&textFile);
    QList<QStringList> data;
    textStream.seek(0);
    // 读取数据行
    while (!textStream.atEnd())
    {
        QString line = textStream.readLine();
        QStringList values = line.split("\t");
        data.append(values);
    }

    textFile.close();

    // 打开SQLite数据库
    QSqlDatabase database = QSqlDatabase::addDatabase("QSQLITE");
    database.setDatabaseName(sqliteFilePath);
    if (!database.open())
    {
        QMessageBox::critical(this, tr("Error"), tr("Failed to open SQLite database."));
        return;
    }

    // 创建表并插入数据
    QSqlQuery query(database);
    QString createTableQuery = "CREATE TABLE IF NOT EXISTS ref_list (";
    for (const QString& columnName : columnNames)
    {
        createTableQuery += "\"" + columnName + "\" TEXT, ";
    }
    createTableQuery.chop(2); // 移除最后的逗号和空格
    createTableQuery += ");";

    if (!query.exec(createTableQuery))
    {
        QMessageBox::critical(this, tr("Error"), tr("Failed to create table: %1").arg(query.lastError().text()));
        database.close();
        return;
    }

    database.transaction();

    QString insertQuery = "INSERT INTO ref_list VALUES (";
    for (int i = 0; i < columnNames.size(); ++i)
    {
        insertQuery += "?, ";
    }
    insertQuery.chop(2); // 移除最后的逗号和空格
    insertQuery += ");";

    for (const QStringList& row : data)
    {
        if (row.size() != columnNames.size())
        {
            qDebug() << "Mismatched value count in row: " << row;
            continue; // Skip this row and proceed to the next row
        }

        query.prepare(insertQuery);
        for (int i = 0; i < row.size(); ++i)
        {
            query.addBindValue(row[i]);
        }

        if (!query.exec())
        {
            QMessageBox::critical(this, tr("Error"), tr("Failed to insert row: %1").arg(query.lastError().text()));
            database.rollback();
            database.close();
            return;
        }
    }

    database.commit();
    database.close();

    QMessageBox::information(this, tr("Conversion Complete"), tr("Text to SQLite conversion completed successfully."));
}

void MainWindow::convertSqliteToText()
{
    // Select the SQLite file
    QString sqliteFilePath = QFileDialog::getOpenFileName(this, tr("Select SQLite File"), "", tr("SQLite Files (*.sqlite)"));
    if (sqliteFilePath.isEmpty())
        return;

    // Select the text file to save
    QString textFilePath = QFileDialog::getSaveFileName(this, tr("Save Text File"), "", tr("Text Files (*.txt)"));
    if (textFilePath.isEmpty())
        return;

    // Open the SQLite database
    QSqlDatabase database = QSqlDatabase::addDatabase("QSQLITE");
    database.setDatabaseName(sqliteFilePath);
    if (!database.open())
    {
        QMessageBox::critical(this, tr("Error"), tr("Failed to open SQLite database."));
        return;
    }

    // Retrieve the table name from the user
    QStringList tables = database.tables();
    if (tables.isEmpty())
    {
        QMessageBox::information(this, tr("No Tables"), tr("The SQLite database does not contain any tables."));
        database.close();
        return;
    }

    // Select the table to export
    bool ok;
    QString selectedTable = QInputDialog::getItem(this, tr("Select Table"), tr("Select a table to export:"), tables, 0, false, &ok);
    if (!ok || selectedTable.isEmpty())
    {
        database.close();
        return;
    }

    // Execute a query to retrieve all data from the selected table
    QSqlQuery query;
    query.prepare(QString("SELECT * FROM %1").arg(selectedTable));
    if (!query.exec())
    {
        QMessageBox::critical(this, tr("Error"), tr("Failed to execute query: %1").arg(query.lastError().text()));
        database.close();
        return;
    }

    // Open the text file to write data
    QFile textFile(textFilePath);
    if (!textFile.open(QIODevice::WriteOnly))
    {
        QMessageBox::critical(this, tr("Error"), tr("Failed to open text file."));
        database.close();
        return;
    }

    // Write the BOM (Byte Order Mark) for UTF-16LE encoding
    QByteArray bom;
    bom.append((char)0xFF);
    bom.append((char)0xFE);
    textFile.write(bom);

    QTextStream textStream(&textFile);
    textStream.setCodec("UTF-16LE");

    // Write the data to the text file
    while (query.next())
    {
        QSqlRecord record = query.record();
        for (int i = 0; i < record.count(); ++i)
        {
            QVariant value = record.value(i);
            textStream << value.toString();
            if (i < record.count() - 1)
                textStream << "\t";
        }
        textStream << "\r\n";
    }

    textFile.close();
    database.close();

    QMessageBox::information(this, tr("Conversion Complete"), tr("SQLite to Text conversion completed successfully."));
}