#include "TaskDataExportDialog.h"
#include "Utils/ComLoadingLabel.h"
#include "Utils/ComMessageBox.h"
#include "Utils/database.h"
#include "Utils/ComOptionsBox.h"
#include "style.h"
#include <QVBoxLayout>
#include <QHBoxLayout>
#include <QLabel>
#include <QPushButton>
#include <QCheckBox>
#include <QDateTime>
#include <QTimer>
#include <QFileDialog>
#include <QSpinBox>
#include "xlsxdocument.h"
#include <QsLog.h>
#include <QDebug>

QXLSX_USE_NAMESPACE


TaskDataExportDialog::TaskDataExportDialog(QWidget *parent,const QString &taskName,const QString &taskCode) :
 QDialog(parent),mTaskName(taskName),mTaskCode(taskCode){

 QLOG_INFO() <<"TaskDataExportDialog::TaskDataExportDialog()";
 //设置模态
 setWindowModality(Qt::ApplicationModal);
 // 隐藏?号
 setWindowFlags(Qt::Dialog| Qt::WindowCloseButtonHint | Qt::WindowStaysOnTopHint);
 setFixedSize(520,260);
 setStyleSheet(".TaskDataExportDialog{background-color:rgb(255,255,255);}");
 setWindowTitle("导出:"+taskName);

 boxLayout = new QVBoxLayout(this);
 boxLayout->setContentsMargins(20,10,20,0);

 ComOptionsBox *optionsBox = new ComOptionsBox(this,"导出选项");

 // 选择导出方式start

 xlsxCb = new QCheckBox(optionsBox->gWidget);
 xlsxCb->setText("Excel(xlsx)");
 xlsxCb->setChecked(true);

 csvCb = new QCheckBox(optionsBox->gWidget);
 csvCb->setText("CSV(csv)");


 optionsBox->gLayout->addWidget(xlsxCb,optionsBox->rowStart,0);
 optionsBox->gLayout->addWidget(csvCb,optionsBox->rowStart,1);
 ++optionsBox->rowStart;

 // 选择导出方式end

 //每一个sheet的最大长度 start
 remarkLabel = new QLabel(optionsBox->gWidget);
 remarkLabel->setStyleSheet(m_stylesheet_QLabel);
 remarkLabel->setText("每一个sheet的最大长度");

 remarkSpin = new QSpinBox(optionsBox->gWidget);
 remarkSpin->setMinimum(4000);
 remarkSpin->setMaximum(5000);
// remarkSpin->setSuffix("");
 remarkSpin->setMaximumWidth(160);
 remarkSpin->setStyleSheet(m_stylesheet_QSpinBox);
 remarkSpin->setValue(10);

 optionsBox->gLayout->addWidget(remarkLabel,optionsBox->rowStart,0);
 optionsBox->gLayout->addWidget(remarkSpin,optionsBox->rowStart,1);
 ++optionsBox->rowStart;

 //每一个sheet的最大长度 end

 // 选择下载路径 start
 QLabel *dirLabel = new QLabel(optionsBox->gWidget);
 dirLabel->setStyleSheet(m_stylesheet_QLabel14);
 dirLabel->setText("选择导出路径*");

 QPushButton *dirBtn= new QPushButton(optionsBox->gWidget);
 dirBtn->setStyleSheet( ".QPushButton {color:rgb(64,65,66);font-family:Microsoft YaHei;font-size:13px;background-color:white; border:1px solid rgb(64,65,66); padding: 0px;}\
 .QPushButton:pressed {background-color: rgba(240,240,240,0.8);}\
 .QPushButton:hover {background-color: rgba(240,240,240,0.4); border-color:rgba(64,65,66,0.5);}\
 .QPushButton:focus{outline: none;}");
 dirBtn->setFixedSize(240,26);
 dirBtn->setText("选择");
 connect(dirBtn,&QPushButton::clicked,this,[this,dirBtn](){
 QString dir = QFileDialog::getExistingDirectory(this,"选择文件夹","");
 if(dir.length()>0){
 mSaveDir = dir;
 dirBtn->setText(dir.mid(0,30)+"...");
 }

 });

 optionsBox->gLayout->addWidget(dirLabel,optionsBox->rowStart,0);
 optionsBox->gLayout->addWidget(dirBtn,optionsBox->rowStart,1);
 ++optionsBox->rowStart;
 // 选择下载路径 end

 // 底部菜单
 QWidget *bottomWidget = new QWidget(this);
 bottomWidget->setFixedHeight(40);
 QHBoxLayout *bottomHLayout = new QHBoxLayout(bottomWidget);
 bottomHLayout->setContentsMargins(0,0,0,0);
 bottomHLayout->setSpacing(0);

 loadingLabel = new ComLoadingLabel(bottomWidget);
 loadingMsgLabel = new QLabel(bottomWidget);
 loadingMsgLabel->setStyleSheet(m_stylesheet_QLabel12);

 QPushButton *okBtn = new QPushButton(bottomWidget);
 okBtn->setText("确定");
 okBtn->setCursor(Qt::PointingHandCursor);
 okBtn->setStyleSheet(m_stylesheet_QPushButton);
 okBtn->setFixedSize(80,28);


 connect(xlsxCb,&QCheckBox::clicked,this,[this](bool checked){
 if(checked){
 remarkLabel->setText("每一个sheet的最大长度");
 remarkSpin->show();
 csvCb->setChecked(false);
 }else{
 csvCb->setChecked(true);
 }
 });
 connect(csvCb,&QCheckBox::clicked,this,[this](bool checked){
 if(checked){
 remarkLabel->setText("csv文件无sheet长度限制");
 remarkSpin->hide();
 xlsxCb->setChecked(false);
 }else{
 xlsxCb->setChecked(true);
 }
 });

 connect(okBtn,&QPushButton::clicked,this,[this](){
 if(mSaveDir.isEmpty()){
 ComMessageBox::error(this,"请选择导出路径");
 return;
 }
 if(xlsxCb->isChecked()){
 exportXlsx();
 }else if(csvCb->isChecked()){
 exportCsv();
 }else{
 ComMessageBox::error(this,"不支持的导出格式");
 }
 });

 QPushButton *cancelBtn = new QPushButton(bottomWidget);
 cancelBtn->setText("取消");
 cancelBtn->setCursor(Qt::PointingHandCursor);
 cancelBtn->setStyleSheet(m_stylesheet_QPushButton_hollow);
 cancelBtn->setFixedSize(80,28);
 connect(cancelBtn,&QPushButton::clicked,this,[this](){
 this->close();
 });

 bottomHLayout->addStretch(10);
 bottomHLayout->addWidget(loadingLabel);
 bottomHLayout->addSpacing(10);
 bottomHLayout->addWidget(loadingMsgLabel);
 bottomHLayout->addSpacing(10);
 bottomHLayout->addWidget(okBtn);
 bottomHLayout->addSpacing(10);
 bottomHLayout->addWidget(cancelBtn);
 bottomHLayout->addSpacing(20);


 boxLayout->addWidget(optionsBox);
 boxLayout->addStretch(10);
 boxLayout->addWidget(bottomWidget);
 boxLayout->addSpacing(10);
}

TaskDataExportDialog::~TaskDataExportDialog(){
 QLOG_INFO() <<"TaskDataExportDialog::~TaskDataExportDialog()";
}

void TaskDataExportDialog::exportCsv(){
 loadingLabel->show();
 loadingMsgLabel->setText("");

 QTimer::singleShot(200,[this](){

 // 查询数据
 QStringList fields = Database::getInstance()->getTableFields(mTaskCode);
 int queryColumnCount = fields.length();

 int pageSize = 30;// 每页查询数据的数量
 int pageStart = 0;// 本页查询起始点
 int totalCount = 0;// 当前数据总长度

 QString csvPath = QString("%1/%2-%3.csv").
 arg(mSaveDir).
 arg(QDateTime::currentDateTime().toLocalTime().toString("yyyyMMddhhmm")).
 arg(mTaskName);

 QFile csv(csvPath);
 if(!csv.open(QIODevice::WriteOnly | QIODevice::Text))
 {
 qDebug()<<"选择导出csv的文件无法写入";
 ComMessageBox::error(this,"选择导出csv的文件无法写入");
 return ;
 }
 QTextStream csvStream(&csv); //创建一个文本流,向保存文件中写入文本
 for (int column = 1; column <= queryColumnCount; ++column) {
 csvStream<<fields[column-1]<<",";
 }
 csvStream<<"\n";

 while (true) {
 QVector<QVector<QString>> data = Database::getInstance()->select(queryColumnCount,
 QString("select %1 from %2 limit %3,%4 ").arg(fields.join(",")).arg(mTaskCode).arg(pageStart).arg(pageSize)
 );

 for (int d = 0; d < data.length(); ++d) {
 totalCount +=1;
 for (int column = 1; column <= queryColumnCount; ++column) {
 QString val = data[d][column-1];
 val.replace("\n"," ");
 val.replace(","," ");
 csvStream<<val<<",";
 }
 csvStream<<"\n";
 }
 if(data.length()<pageSize){
 break;
 }else{
 pageStart +=pageSize;
 }
 }

 csv.close();

 loadingLabel->hide();
 loadingMsgLabel->setText(QString("导出完成(共计%1条数据)").arg(totalCount));
 });
}

void TaskDataExportDialog::exportXlsx(){
 // 执行导出文件
 loadingLabel->show();
 loadingMsgLabel->setText("");

 QTimer::singleShot(200,[this](){
 int sheetDataMaxRow = remarkSpin->value();//每一个sheet中最大行数
 // 查询数据
 QStringList fields = Database::getInstance()->getTableFields(mTaskCode);
 int queryColumnCount = fields.length();

 int pageSize = 30;// 每页查询数据的数量
 int pageStart = 0;// 本页查询起始点
 int totalCount = 0;// 当前数据总长度

 int sheetNum = 1; // 起始sheet
 int sheetDataRow = 1; //当前sheet的数据行

 Document xlsx;
 xlsx.addSheet("sheet"+QString::number(sheetNum));
 xlsx.setColumnWidth(1,queryColumnCount,20);

 sheetDataRow+=1;
 for (int column = 1; column <= queryColumnCount; ++column) {
 xlsx.write(sheetDataRow,column,fields[column-1]);
 }

 while (true) {
 QVector<QVector<QString>> data = Database::getInstance()->select(queryColumnCount,
 QString("select %1 from %2 limit %3,%4 ").arg(fields.join(",")).arg(mTaskCode).arg(pageStart).arg(pageSize)
 );

 for (int d = 0; d < data.length(); ++d) {
 totalCount +=1;
 sheetDataRow +=1;
 for (int column = 1; column <= queryColumnCount; ++column) {
 xlsx.write(sheetDataRow,column,data[d][column-1]);
 }
 if(sheetDataRow % sheetDataMaxRow == 0){
 sheetNum +=1;
 sheetDataRow = 0;
 xlsx.addSheet("sheet"+QString::number(sheetNum));
 xlsx.setColumnWidth(1,queryColumnCount,40);
 }
 }
 if(data.length()<pageSize){
 break;
 }else{
 pageStart +=pageSize;
 }
 }

 QString filePath = QString("%1/%2-%3.xlsx").
 arg(mSaveDir).
 arg(QDateTime::currentDateTime().toLocalTime().toString("yyyyMMddhhmm")).
 arg(mTaskName);
 xlsx.saveAs(filePath);

 loadingLabel->hide();
 loadingMsgLabel->setText(QString("导出完成(共计%1条数据)").arg(totalCount));
// this->close();
 });

}