#include "sqlcompareview.h"
#include "sqlview.h"
#include "common/utils.h"
#include "diff/diff_match_patch.h"
#include "sqlitesyntaxhighlighter.h"
#include <QHeaderView>
#include <QDebug>

SqlCompareView::SqlCompareView(QWidget *parent) :
    QTableWidget(parent)
{
    setColumnCount(2);
    setVerticalScrollMode(ScrollPerPixel);
    horizontalHeader()->setSectionResizeMode(0, QHeaderView::Stretch);
    horizontalHeader()->setSectionResizeMode(1, QHeaderView::Stretch);
    horizontalHeader()->setVisible(false);
//    verticalHeader()->setSectionResizeMode(QHeaderView::ResizeToContents);
    diff = new diff_match_patch;
}

void SqlCompareView::setSides(const QList<QPair<QString, QString>>& data)
{
    setRowCount(data.size());

    int row = 0;
    SqlView* leftView = nullptr;
    SqlView* rightView = nullptr;
    for (const QPair<QString, QString>& rowData : data)
    {
        leftView = new SqlView();
        leftView->setFrameStyle(QFrame::NoFrame);
        leftView->setPlainText(rowData.first);
        setCellWidget(row, 0, leftView);

        rightView = new SqlView();
        rightView->setFrameStyle(QFrame::NoFrame);
        rightView->setPlainText(rowData.second);
        setCellWidget(row, 1, rightView);

        setupHighlighting(rowData.first, rowData.second, leftView, rightView);

        row++;
    }
    updateLabels();
    updateSizes();
}

void SqlCompareView::setLeftLabel(const QString& label)
{
    leftLabel = label;
}

void SqlCompareView::setRightLabel(const QString& label)
{
    rightLabel = label;
}

void SqlCompareView::updateSizes()
{
    if (rowCount() == 0 || !isVisible())
        return;

    SqlView* view = dynamic_cast<SqlView*>(cellWidget(0, 0));
    if (!view)
    {
        qCritical() << "Not a SqlView in SqlCompareView::updateSizes():" << cellWidget(0, 0);
        return;
    }

    QFont font = view->font();
    QFontMetrics fm(font);

    int leftWidth = horizontalHeader()->sectionSize(0);
    int rightWidth = horizontalHeader()->sectionSize(1);

    SqlView* leftView = nullptr;
    SqlView* rightView = nullptr;
    QSize leftSize;
    QSize rightSize;
    for (int row = 0, total = rowCount(); row < total; ++row)
    {
        leftView = dynamic_cast<SqlView*>(cellWidget(row, 0));
        leftView->document()->setTextWidth(leftWidth);

        rightView = dynamic_cast<SqlView*>(cellWidget(row, 1));
        rightView->document()->setTextWidth(rightWidth);

        leftSize = QSize(leftWidth, leftView->document()->size().toSize().height());
        rightSize = QSize(rightWidth, rightView->document()->size().toSize().height());
        if (leftSize.height() > rightSize.height())
            rightSize.setHeight(leftSize.height());
        else
            leftSize.setHeight(rightSize.height());

        leftView->setFixedSize(leftSize);
        rightView->setFixedSize(rightSize);
    }
    verticalHeader()->resizeSections(QHeaderView::ResizeToContents);
}

void SqlCompareView::updateLabels()
{
    setHorizontalHeaderLabels({leftLabel, rightLabel});
}

void SqlCompareView::setupHighlighting(const QString& left, const QString& right, SqlView* leftView, SqlView* rightView)
{
    QList<Diff> diffs = diff->diff_main(left, right);
    int leftPos = 0;
    int lgt = 0;
    for (const Diff& d : diffs)
    {
        lgt = d.text.length();
        switch (d.operation)
        {
            case DELETE:
                leftView->setTextBackgroundColor(leftPos, leftPos + lgt - 1, Qt::red);
                leftPos += lgt;
                break;
            case EQUAL:
                leftPos += lgt;
                break;
            case INSERT:
                rightView->setTextBackgroundColor(leftPos, leftPos + lgt - 1, Qt::green);
                break;
        }
    }
}

void SqlCompareView::resizeEvent(QResizeEvent* e)
{
    QTableWidget::resizeEvent(e);
    updateSizes();
}

void SqlCompareView::showEvent(QShowEvent* e)
{
    QTableWidget::showEvent(e);
    updateSizes();
}