#include "TaskImport.h"
#include "style.h"
#include "Utils/models.h"
#include "Utils/database.h"
#include "Utils/ComMessageBox.h"
#include "Utils/ComLineWidget.h"
#include <QVBoxLayout>
#include <QHBoxLayout>
#include <QLabel>
#include <QLineEdit>
#include <QPushButton>
#include <QComboBox>
#include <QStyledItemDelegate>
#include <QFileDialog>
#include <QsLog.h>

TaskImport::TaskImport(QWidget *parent) : QDialog(parent)
{
    setFixedSize(520,200);

    Qt::WindowFlags flags=Qt::Dialog;
    flags |=Qt::WindowCloseButtonHint;
    setWindowFlags(flags);
    setAttribute(Qt::WA_StyledBackground,true);
    setStyleSheet(".TaskImport{background-color:rgb(255,255,255);}");
    setWindowTitle("导入任务");

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


    QWidget *gWidget = new QWidget(this);

    QGridLayout *gLayout = new QGridLayout(gWidget);
    gLayout->setColumnMinimumWidth(0,60);
    gLayout->setColumnMinimumWidth(2,110);

    // 第一行
    QLabel *nameLabel = new QLabel(gWidget);
    nameLabel->setText("任务名称");
    QLineEdit *nameLine = new QLineEdit(gWidget);
    nameLine->setFixedHeight(30);
    nameLine->setStyleSheet(m_stylesheet_QLineEdit);

    QPushButton *selectBtn = new QPushButton(gWidget);
    selectBtn->setText("选择...");
    selectBtn->setCursor(Qt::PointingHandCursor);
    selectBtn->setStyleSheet(m_stylesheet_QPushButton_hollow);
    selectBtn->setFixedSize(90,28);

    connect(selectBtn,&QPushButton::clicked,this,[this,nameLine](){
       QString filename = QFileDialog::getOpenFileName(this,"选择任务文件","","*.ds");
       if(filename.length()>0){
          nameLine->setText(filename);
       }
    });

    gLayout->addWidget(nameLabel,0,0,Qt::AlignCenter);
    gLayout->addWidget(nameLine,0,1);
    gLayout->addWidget(selectBtn,0,2,Qt::AlignCenter);

    // 第二行
    QLabel *groupNameLabel = new QLabel(gWidget);
    groupNameLabel->setText("任务组");

    QComboBox *groupComboBox = new QComboBox(gWidget);
    groupComboBox->setFixedHeight(30);
    QStyledItemDelegate *groupDelegate = new QStyledItemDelegate(groupComboBox);
    groupComboBox->setItemDelegate(groupDelegate);
    groupComboBox->setStyleSheet(m_stylesheet_QComboBox);

    QVector<MTaskGroup> taskGroups = Database::getInstance()->taskGroups;

    for (int i = 0; i < taskGroups.length(); ++i) {
        if(i==0){
            m_currentGroupId = taskGroups[i].id;
        }
        groupComboBox->addItem(taskGroups[i].name);
    }

    connect(groupComboBox, QOverload<int>::of(&QComboBox::currentIndexChanged),[this,taskGroups](int index){
         m_currentGroupId = taskGroups[index].id;
    });

    gLayout->addWidget(groupNameLabel,1,0,Qt::AlignCenter);
    gLayout->addWidget(groupComboBox,1,1);
    boxLayout->addWidget(gWidget);

    boxLayout->addWidget(new ComLineWidget(this));
    // 第三行(底部菜单)
    QWidget *bottomWidget = new QWidget(this);
    bottomWidget->setFixedHeight(40);
    QHBoxLayout *bottomHLayout = new QHBoxLayout(bottomWidget);
    bottomHLayout->setContentsMargins(0,0,0,0);
    bottomHLayout->setSpacing(0);
    QPushButton *okBtn = new QPushButton(bottomWidget);
    okBtn->setText("确定");
    okBtn->setCursor(Qt::PointingHandCursor);
    okBtn->setStyleSheet(m_stylesheet_QPushButton);
    okBtn->setFixedSize(80,28);
    connect(okBtn,&QPushButton::clicked,this,[this,nameLine](){
        QString filename = nameLine->text().trimmed();
        if(filename.length()>0){
            MTask *task = new MTask();
            task->groupId = m_currentGroupId;

            if(task->fromFile(filename)){
                task->defaultUserAgent =  Database::getInstance()->getRandomUserAgent();

                m_importSuccessTasks.append(task);// 导出成功的任务放入数组

                QString msg;
                bool state = Database::getInstance()->addTask(task,msg);

                if(state==true){
                    ComMessageBox::success(this,msg);
                    this->close();
                }else{
                    ComMessageBox::error(this,msg);
                }


            }else{
                delete task;
                ComMessageBox::error(this,"您导入的任务可能已损毁！");
            }
        }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(okBtn,0,Qt::AlignRight);
    bottomHLayout->addSpacing(10);
    bottomHLayout->addWidget(cancelBtn,0,Qt::AlignRight);
    bottomHLayout->addSpacing(20);
    boxLayout->addWidget(bottomWidget);
}
TaskImport::~TaskImport(){

    for (int i = 0; i < m_importSuccessTasks.length(); ++i) {
        delete m_importSuccessTasks[i];
    }
}