#include "TaskData.h"
#include "TaskDataExportDialog.h"
#include "TaskDataMenuDialog.h"
#include "Utils/database.h"
#include "Utils/ComLoadingLabel.h"
#include "Utils/ComLineWidget.h"
#include "Utils/ComMessageBox.h"
#include "style.h"
#include "TaskDataDownloadManager.h"
#include <QVBoxLayout>
#include <QHBoxLayout>
#include <QToolButton>
#include <QPushButton>
#include <QTableWidget>
#include <QListWidget>
#include <QMessageBox>
#include <QHeaderView>
#include <QLabel>
#include <QLineEdit>
#include <QSpinBox>
#include <QTimer>
#include <QsLog.h>

const static int PAGESIZE = 30; // 执行页面的数据分页数
TaskData::TaskData(QWidget *parent,const QString &taskName,const QString &taskCode) : QWidget(parent),mTaskName(taskName),mTaskCode(taskCode)
{
    QLOG_INFO()<<"TaskData::TaskData()";
    QVBoxLayout *mainLayout = new QVBoxLayout(this);
    mainLayout->setContentsMargins(0,2,0,0);
    mainLayout->setSpacing(0);
    mainLayout->addWidget(new ComLineWidget(this));

    QWidget * boxWidget = new QWidget(this);
    mainLayout->addWidget(boxWidget);
    boxLayout = new QVBoxLayout(boxWidget);

    boxLayout->setContentsMargins(20,0,20,0);
    boxLayout->setSpacing(0);

    mFields = Database::getInstance()->getTableFields(mTaskCode);

    initMenuUi();
    initTableUi();
    initBottomUi();

    getData();

}
TaskData::~TaskData(){
    QLOG_INFO()<<"TaskData::~TaskData()";
    QHash<QString,TaskDataDownloadManager *>::const_iterator it = mDownloadManagerH.constBegin();
    while (it != mDownloadManagerH.constEnd()) {
        delete it.value();
        ++it;
    }
    mDownloadManagerH.clear();

    //清理logListWidget的item
    for (int i = 0; i < logListWidgetItems_clear.count(); ++i) {
        delete logListWidgetItems_clear[i];
        logListWidgetItems_clear[i] = nullptr;
    }
    logListWidgetItems_clear.clear();

}

void TaskData::getData(int p){
    mCurPage = p;
    tableWidget->setRowCount(0);
    tableWidget->clearContents();
    loadingLabel->show();
    QTimer::singleShot(200,this,[this](){
      updatePageData(Database::getInstance()->getTableRowCount(mTaskCode),false);//更新page
      int fieldCount = mFields.length();
      QString sql = QString("select %1 from %2 limit %3,%4 ").arg(mFields.join(",")).arg(mTaskCode).arg(PAGESIZE * (mCurPage-1)).arg(PAGESIZE);
      QVector<QVector<QString>> data = Database::getInstance()->select(fieldCount,sql);

      tableWidget->setRowCount(data.length());
      for (int row = 0; row < data.length(); ++row) {
          tableWidget->setRowHeight(row,30);
          for (int column = 0; column < fieldCount; ++column) {
              QTableWidgetItem *item = new QTableWidgetItem(data[row][column]);
              tableWidget->setItem(row,column,item);
          }
      }
      tableWidget->scrollToBottom();
      loadingLabel->hide();
    });
}
void TaskData::initMenuUi(){
    QWidget *menuWidget = new QWidget(this);
    menuWidget->setFixedHeight(50);
    QHBoxLayout *menuHLayout = new QHBoxLayout(menuWidget);
    menuHLayout->setContentsMargins(0,0,0,0);


    QLabel *lLabel = new QLabel(menuWidget);
    lLabel->setStyleSheet(m_stylesheet_QLabel12);
    lLabel->setText("查看数据：");

    QLabel *rLabel = new QLabel(menuWidget);
    rLabel->setStyleSheet(m_stylesheet_QLabel14);
    rLabel->setText(mTaskName);



    loadingLabel = new ComLoadingLabel(this);


    QPushButton *refreshBtn = new QPushButton(menuWidget);
    refreshBtn->setCursor(Qt::PointingHandCursor);
    refreshBtn->setStyleSheet(m_stylesheet_QPushButton);
    refreshBtn->setText("刷新");
    refreshBtn->setFixedSize(80,30);
    connect(refreshBtn,&QPushButton::clicked,this,[this](){
        getData();
    });


    QPushButton *clearBtn = new QPushButton(menuWidget);
    clearBtn->setCursor(Qt::PointingHandCursor);
    clearBtn->setStyleSheet(m_stylesheet_QPushButton_hollow);
    clearBtn->setText("清空");
    clearBtn->setFixedSize(80,30);

    connect(clearBtn,&QPushButton::clicked,this,[this](){
        loadingLabel->show();
        QTimer::singleShot(500,this,[this](){
            QString msg;
            Database::getInstance()->clearTaskData(mTaskCode,msg);
            tableWidget->setRowCount(0);
            tableWidget->clearContents();
            loadingLabel->hide();

        });


    });

    menuHLayout->addWidget(lLabel);
    menuHLayout->addWidget(rLabel);
    menuHLayout->addStretch(10);
    menuHLayout->addWidget(loadingLabel);
    menuHLayout->addSpacing(10);
    menuHLayout->addWidget(refreshBtn);
    menuHLayout->addSpacing(10);
    menuHLayout->addWidget(clearBtn);

    boxLayout->addWidget(menuWidget);
}
void TaskData::initTableUi(){
    tableWidget = new QTableWidget(this);
    tableWidget->setFrameShape(QFrame::NoFrame);//设置无边框
    tableWidget->setFocusPolicy(Qt::NoFocus); //去除虚边框
    tableWidget->setStyleSheet(".QTableWidget{border:1px solid rgb(217,217,217); border-radius: 0px;padding: 1px;}");

//    tableWidget->setTabPosition(QTabWidget::North);//设置选项卡的方位东南西北，默认在上方
//    tableWidget->horizontalHeader()->setSectionResizeMode(QHeaderView::Stretch);
//    tableWidget->verticalHeader()->setSectionResizeMode(QHeaderView::Stretch);

    tableWidget->verticalHeader()->setVisible(true);
    tableWidget->horizontalHeader()->setVisible(true);

    connect(tableWidget->horizontalHeader(),&QHeaderView::sectionClicked,this,[this](int fieldIndex){
        QString field = this->mFields.at(fieldIndex);
        TaskDataMenuDialog *menu = new TaskDataMenuDialog(this,field,fieldIndex);

        connect(menu->downloadBtn,&QPushButton::clicked,[this,menu](){
            if(this->mDownloadManagerH.contains(menu->field)){
                //已经在下载中
                ComMessageBox::error(this,"正在下载中");
            }else{

                if(menu->saveDir.isEmpty()){
                     ComMessageBox::error(this,"请选择下载路径");
                }else{
                    TaskDataDownloadManager *downloadManager = new TaskDataDownloadManager(this,
                                                                   this->mTaskName,this->mTaskCode,this->mFields,
                                                                   menu->threadCountSpin->value(),
                                                                   menu->threadIntervalSpin->value(),
                                                                   menu->field,
                                                                   menu->fieldIndex,
                                                                   menu->fieldPrefixLine->text().trimmed(),
                                                                   menu->saveDir,
                                                                   menu->saveSuffixLine->text().trimmed());
                    //每一个字段的下载器显示两行日志
                    QString log1 = QString("下载任务 '%1'，线程数 %2，线程休眠间隔 %3，下载路径 %4").
                            arg(menu->field).
                            arg(menu->threadCountSpin->value()).
                            arg(menu->threadIntervalSpin->value()).
                            arg(menu->saveDir);
                    QListWidgetItem * logitem1 = new QListWidgetItem(log1,logListWidget);
                    logListWidgetItems_clear.push_back(logitem1);
                    logListWidget->addItem(logitem1);


                    QListWidgetItem * logitem2 = new QListWidgetItem("开始下载",logListWidget);
                    logListWidgetItems_clear.push_back(logitem2);
                    logListWidget->addItem(logitem2);

                    connect(downloadManager, &TaskDataDownloadManager::notifyLog, this, [logitem2](const QString &field,const QString &log) {
                        logitem2->setText(log);
                    });


                    this->mDownloadManagerH.insert(menu->field,downloadManager);
                    menu->deleteLater();
                }

            }

        });
        menu->setAttribute(Qt::WA_DeleteOnClose);
        menu->move(QCursor::pos());
        menu->show();

    });

    // 设置header（这里是数据库的字段名称为准）
    tableWidget->setColumnCount(mFields.length());
    tableWidget->setHorizontalHeaderLabels(mFields);
    boxLayout->addWidget(tableWidget);

}
void TaskData::initBottomUi(){
    //字段处理时日志
    logListWidget = new QListWidget(this);
    logListWidget->setStyleSheet(".QListWidget{border:1px solid rgb(217,217,217); border-radius: 0px;padding: 2px;}");
    logListWidget->setFixedHeight(80);

    //页数
    QWidget *turnWidget = new QWidget(this);
    turnWidget->setFixedHeight(40);
    QHBoxLayout *turnHLayout = new QHBoxLayout(turnWidget);
    turnHLayout->setContentsMargins(10,0,10,0);

    pageLogLabel = new QLabel(turnWidget);
    pageLogLabel->setStyleSheet(m_stylesheet_QLabel);

    QToolButton *last = new QToolButton(turnWidget);
    last->setText("<");
    pageLabel = new QLabel(this);
    pageLabel->setStyleSheet(m_stylesheet_QLabel);
    QToolButton *next = new QToolButton(turnWidget);
    next->setText(">");

    connect(last,&QToolButton::clicked,this,[this](){
        if(mCurPage>1){
            getData(mCurPage-1);
        }
    });
    connect(next,&QToolButton::clicked,this,[this](){
        if(mCurPage<mTotalPage){
            getData(mCurPage+1);
        }
    });


    QPushButton *exportBtn = new QPushButton(turnWidget);
    exportBtn->setCursor(Qt::PointingHandCursor);
    exportBtn->setStyleSheet(m_stylesheet_QPushButton);
    exportBtn->setText("导出数据");
    exportBtn->setFixedSize(100,28);

    connect(exportBtn,&QPushButton::clicked,this,[this](){
        TaskDataExportDialog *dlg = new TaskDataExportDialog(this,mTaskName,mTaskCode);
        dlg->setAttribute(Qt::WA_DeleteOnClose);
        dlg->show();
    });

    turnHLayout->addWidget(pageLogLabel);
    turnHLayout->addSpacing(10);
    turnHLayout->addWidget(last);
    turnHLayout->addWidget(pageLabel);
    turnHLayout->addWidget(next);
    turnHLayout->addStretch(10);
    turnHLayout->addWidget(exportBtn,0,Qt::AlignRight);

    boxLayout->addSpacing(5);
    boxLayout->addWidget(logListWidget);
    boxLayout->addWidget(turnWidget);
}
void TaskData::updatePageData(int num, bool isShowLastPage){
    if(num==0){// num == 0 恢复为默认参数
        mCurPage = 1;
    }
    int totalPage = num / PAGESIZE; // 总页数
    if((num % PAGESIZE) > 0){
        totalPage +=1;
    }
    mTotalPage = totalPage;

    if(isShowLastPage){// 是否显示最后的数据对应的页面
        mCurPage = mTotalPage;
    }

    pageLabel->setText(QString::number(mCurPage));
    pageLogLabel->setText(QString("共计%1条，%2页").arg(num).arg(mTotalPage));
}