#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>
#include <QContextMenuEvent>

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);
    connect(ui->actionSaveAs, &QAction::triggered, this, &MainWindow::onSaveAsButtonClicked);

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

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

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

    QMenu* convertMenu = menuBar()->addMenu(tr("转换(&C)"));

    QAction* textToSqliteAction = convertMenu->addAction(tr("txt 转 SQLite"));
    connect(textToSqliteAction, &QAction::triggered, this, &MainWindow::convertTextToSqlite);

    QAction* sqliteToTextAction = convertMenu->addAction(tr("SQLite 转 txt"));
    connect(sqliteToTextAction, &QAction::triggered, this, &MainWindow::convertSqliteToText);

    QIcon convertIcon = QApplication::style()->standardIcon(QStyle::SP_ArrowRight);
    textToSqliteAction->setIcon(convertIcon);
    sqliteToTextAction->setIcon(convertIcon);

    contextMenu = new QMenu(this);
    QAction* addRowAboveAction = contextMenu->addAction(tr("向上添加一行"));
    QAction* addRowBelowAction = contextMenu->addAction(tr("向下添加一行"));
    connect(addRowAboveAction, &QAction::triggered, this, &MainWindow::onAddRowAboveActionTriggered);
    connect(addRowBelowAction, &QAction::triggered, this, &MainWindow::onAddRowBelowActionTriggered);
}


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

void MainWindow::contextMenuEvent(QContextMenuEvent* event)
{
    QPoint cellPos = ui->tableWidget->viewport()->mapFromGlobal(event->globalPos());
    int currentColumn = ui->tableWidget->horizontalHeader()->logicalIndexAt(cellPos.x());

    if (currentColumn == -1)
    {
        int currentRow = ui->tableWidget->rowAt(cellPos.y());

        if (currentRow >= 0)
        {
            ui->tableWidget->selectRow(currentRow); // Select the clicked row
            contextMenu->popup(event->globalPos());
        }
    }
}

void MainWindow::onAddRowAboveActionTriggered()
{
    QTableWidget* tableWidget = ui->tableWidget;
    int currentRow = tableWidget->currentRow();
    if (currentRow >= 0)
    {
        tableWidget->insertRow(currentRow);
    }
    else
    {
        // 如果没有选中行，在末尾追加一行
        int newRow = tableWidget->rowCount();
        tableWidget->insertRow(newRow);
        currentRow = newRow;
    }

    // 填充插入的行为空项
    for (int column = 0; column < tableWidget->columnCount(); ++column)
    {
        QTableWidgetItem* emptyItem = new QTableWidgetItem("");
        tableWidget->setItem(currentRow, column, emptyItem);
    }
}

void MainWindow::onAddRowBelowActionTriggered()
{
    QTableWidget* tableWidget = ui->tableWidget;
    int currentRow = tableWidget->currentRow();
    if (currentRow >= 0)
    {
        tableWidget->insertRow(currentRow + 1); // 在下方插入一行
    }
    else
    {
        // 如果没有选中行，在末尾追加一行
        int newRow = tableWidget->rowCount();
        tableWidget->insertRow(newRow);
    }

    // 填充插入的行为空项
    for (int column = 0; column < tableWidget->columnCount(); ++column)
    {
        QTableWidgetItem* emptyItem = new QTableWidgetItem("");
        tableWidget->setItem(currentRow + 1, column, emptyItem);
    }
}

void MainWindow::showProgressDialog(const QString& labelText, int minimum, int maximum)
{
    progressDialog = new QProgressDialog(labelText, "Cancel", minimum, maximum, this);
    progressDialog->setWindowModality(Qt::WindowModal);
    progressDialog->setAutoClose(false);
    progressDialog->setAutoReset(false);
    progressDialog->setValue(minimum);
    progressDialog->show();
}


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

    if (!file.open(QIODevice::ReadOnly | QIODevice::Text))
    {
        qDebug() << "Failed to open file:" << fileName;
        return;
    }

    QTextStream in(&file);

    // 自动检测并设置正确的文本编码
    in.setAutoDetectUnicode(true);
    in.setCodec("UTF-16LE");

    int tableRow = 0;
    int tableCol = 0;
    QStringList rows;

    // 逐行读取文本数据
    while (!in.atEnd())
    {
        QString line = in.readLine();
        if (!line.isEmpty())
        {
            rows.append(line);
            tableRow++;

            if (tableRow == 2)
            {
                QStringList columnValues = line.split("\t", QString::SkipEmptyParts);
                tableCol = columnValues.count();
            }
        }
    }

    file.close();

    showProgressDialog("Converting...", 0, tableRow);

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

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

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

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

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

        // 补充空白内容，确保行的列数与表格的列数一致
        while (row.count() < tableCol)
        {
            row.append(""); // 添加空白内容
        }

        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->hide();
    tableWidget->blockSignals(false); // 恢复信号发射
}


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

void MainWindow::onSaveButtonClicked()
{
    if (!openFilePath.isEmpty())
    {
        QFile file(openFilePath);
        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();
        }
    }
}

void MainWindow::onSaveAsButtonClicked()
{
    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)
        {
            openFilePath = urlList.first().toLocalFile();
            if (!openFilePath.isEmpty())
            {
                loadTextFile(openFilePath);
            }
        }
    }
}

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);
    }

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

        // 截断行内容，保证不超过列名的个数
        if (values.size() > columnNames.size())
        {
            values = values.mid(0, columnNames.size());
        }

        data.append(values);
    }

    textFile.close();

    showProgressDialog("Converting...", 0, data.size());

    // 打开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 += ");";

    int progress = 0;
    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;
        }

        ++progress;
        progressDialog->setValue(progress);
        QApplication::processEvents();
        if (progressDialog->wasCanceled())
        {
            break;
        }
    }

    progressDialog->hide();
    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;
    }

    showProgressDialog("Converting...", 0, query.size());

    // 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
    int progress = 0;
    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";

        ++progress;
        progressDialog->setValue(progress);
        QApplication::processEvents();
        if (progressDialog->wasCanceled())
        {
            break;
        }
    }

    progressDialog->hide();
    textFile.close();
    database.close();

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