#include "ui_TabFormWidget.h"
#include "SelectFormWidget.h"

#include <QToolButton>
#include <QTableWidget>
#include <QScroller>
#include <QHeaderView>
#include <QUuid>
#include <QLabel>

#include <QDate>
#include <QSqlRecord>
#include <QDebug>

#include "components/SlideTableView.h"
#include "json/jsonobject.h"
#include "db/SQLHelper.h"
#include "event/EventCenter.h"
#include "dialogs/DialogManager.h"
#include "log/LogManager.h"
#include "components/VerticalTextToolButton.h"
#include "PatientAddDateDelegate.h"
#include "dicom/WorkListManager.h"
#include "utilities/ScanProcessSequence.h"

SelectFormWidget::SelectFormWidget(QWidget* parent)
: TabFormWidget(parent)
, mBtnEdit(new QToolButton())
, mBtnDelete(new QToolButton())
, mBtnAdd(new QToolButton(this))
, mBtnSelect(new QToolButton(this))
, mPatTable(new SlideTableView(this))
, mModel(nullptr)
, patientDetailForm(new PatientDetailForm(this))
, mSelectTabTitle(new QLabel(this))
{
    //process expired patient list
    QDate date = QDate::currentDate().addDays(-JsonObject::Instance()->getPatientListExpireDays());
    SQLHelper::exec(QString("DELETE FROM Patient WHERE AddDate <= '%1'").arg(date.toString("yyyy-MM-dd")));

	//init command bar
    auto layout = new QHBoxLayout();
	ui->commandWidget->setLayout(layout);

    layout->addSpacerItem(new QSpacerItem(20, 20, QSizePolicy::Expanding));
    initPatEditButtons(layout);
    initDataModel();
    //Init content widget
    auto contantContanerLayout =  new QVBoxLayout(this->ui->contentWidget);
    contantContanerLayout->setMargin(0);
    mSelectTabTitle->setText(tr("Patient Information Manage"));
    mSelectTabTitle->setObjectName("selectTabTitle");
    contantContanerLayout->addWidget(mSelectTabTitle);
	auto contentLayout = new QHBoxLayout;
    contantContanerLayout->addLayout(contentLayout);
	contentLayout->setContentsMargins(0, 0, 0, 0);
    initDetailPanel(contentLayout);
    addVerticalLine(contentLayout);
    auto gridBox = new QVBoxLayout;
    contentLayout->addLayout(gridBox);
    initTableView(gridBox);
    gridBox->addWidget(ui->commandWidget);
    //select default row 0
	if (mModel->rowCount() > 0)
	{
		mPatTable->selectRow(0);
        setPatientDetail(mPatTable, mModel, patientDetailForm);
    }
    // event ResponsePreview slot
	connect(EventCenter::Default(), &EventCenter::ResponsePreview, [=](QObject* sender, QObject* data) {
        prepareButtons(true);
		});
    // event ResponseStop slot
    connect(EventCenter::Default(), &EventCenter::ResponseStopPreview, [=](QObject* sender, QObject* data) {
        prepareButtons(false);
	});
    // event ReloadLanguage slot;
	connect(EventCenter::Default(), &EventCenter::ReloadLanguage, this, &SelectFormWidget::reloadLanguage);

	connect(EventCenter::Default(), &EventCenter::AnonymousModeChanged, this, &SelectFormWidget::updateDataByAnonymousMode);

    connect(EventCenter::Default(), &EventCenter::SetSelectedPatient, this, [=](QObject* sender, QObject* data)
    {
        PatientInformation* patientInfo = (PatientInformation*)data;
        this->setSelectedPatient(patientInfo);
    });

    connect(ScanProcessSequence::getInstance(), &ScanProcessSequence::ScanProcessSequenceFinished, this, &SelectFormWidget::clearSelectedPatient);

	//first prepare buttons!
    prepareButtons(false);

    //init WorkListManager table view
    WorkListManager::getInstance()->setTableView(mPatTable);
}

void SelectFormWidget::prepareButtons(bool disableALL) {
    bool anonymousMode = JsonObject::Instance()->getAnonymousMode();
    bool stateFlag = (mPatTable->currentIndex().row() >= 0);

    mBtnAdd->setEnabled(!anonymousMode && !disableALL);
    if (mBtnAdd)mBtnEdit->setEnabled(!anonymousMode && stateFlag && !disableALL);
    if (mBtnAdd)mBtnDelete->setEnabled(!anonymousMode&& stateFlag && !disableALL);
    mBtnSelect->setEnabled(stateFlag && !disableALL);
}

void SelectFormWidget::initPatEditButtons(QHBoxLayout *layout) {
    mBtnAdd->setObjectName("btnAdd");
    mBtnEdit->setObjectName("btnPatEdit");
    mBtnDelete->setObjectName("btnPatDelete");
    mBtnSelect->setObjectName("btnSelect");

    mBtnEdit->setText(tr("Edit"));
    mBtnDelete->setText(tr("Delete"));
    mBtnAdd->setText(tr("Add"));
    mBtnSelect->setText(tr("Select"));
    layout->addWidget(mBtnAdd);
    layout->addWidget(mBtnEdit);
    layout->addWidget(mBtnDelete);
    addVerticalLine(layout);
    layout->addWidget(mBtnSelect);

    // btn add  slot
    connect(mBtnAdd, &QToolButton::clicked, this, &SelectFormWidget::editPatient);

    // btn edit  slot
    connect(mBtnEdit, &QToolButton::clicked, this, &SelectFormWidget::editPatient);

    // btn del  slot
    connect(mBtnDelete, &QToolButton::clicked, this, &SelectFormWidget::delPatient);

    // mBtn select slot
    connect(mBtnSelect, &QToolButton::clicked, this, &SelectFormWidget::selectPatient);
}

void SelectFormWidget::editPatient() {
    bool addFlag = sender() == mBtnAdd;
    auto index = mPatTable->currentIndex();
    // accept change
    if (DialogManager::Default()->requestEditPatientInfo(addFlag ? nullptr:patientDetailForm->getPatientInformation(),mModel) == QDialog::Accepted) {
        if (addFlag){
            mPatTable->selectRow(0);
            mModel->selectRow(0);
            LOG_USER_OPERATION(QString("Add Patient, ID: %1").arg(patientDetailForm->getPatientInformation()->ID))
        }
        else{
            mPatTable->selectRow(index.row());
            mModel->selectRow(index.row());
            setPatientDetail(mPatTable, mModel, patientDetailForm);
            LOG_USER_OPERATION(QString("Edit Patient, ID: %1").arg(patientDetailForm->getPatientInformation()->ID))
        }
        mBtnSelect->setEnabled(true);
    }
}

void SelectFormWidget::delPatient() {
    if (mPatTable->currentIndex().row() < 0) return;
    QString pUid = mModel->index(mPatTable->currentIndex().row(), PatientUID).data().toString();
    // patient has been selected as the scan patient!
    if (mSelectedPatientUID == pUid){
        DialogManager::Default()->requestAlertMessage(tr("Can't delete selected Patient !"),DialogButtonMode::OkOnly,tr("Alert"));
        return;
    }
    // not the selected one, confirm!
    QString pat_name = mModel->index(mPatTable->currentIndex().row(), Name).data().toString();
    if (DialogManager::Default()->requestAlertMessage(QString(tr("Delete Patient \"%1\" ?")),pat_name,DialogButtonMode::OkAndCancel,tr("Confirm")) != QDialog::Accepted) return;
    // need delete clear edit panel detail
    patientDetailForm->clearPatientInformation();
    //mModel->setData(mModel->index(mPatTable->currentIndex().row(), Flag), 9);
    mModel->removeRow(mPatTable->currentIndex().row());

    if (mModel->submitAll()) {
        mModel->select();
        if (mModel->rowCount() > 0) {
            mPatTable->selectRow(0);
            mModel->selectRow(0);
            setPatientDetail(mPatTable, mModel, patientDetailForm);
            LOG_USER_OPERATION(QString("Delete Patient, ID: %1").arg(patientDetailForm->getPatientInformation()->ID))
        }
    } else {
        //TODO:error handle
        DialogManager::Default()->requestAlertMessage(tr("Can't delete selected Patient , db submit error!"),DialogButtonMode::OkOnly,tr("Alert"));
    }
    prepareButtons(false);
}

void SelectFormWidget::selectPatient() {
    EventCenter::Default()->triggerEvent(PatientSelected, nullptr, (QObject*)patientDetailForm->getPatientInformation());
}

void SelectFormWidget::setSelectedPatient(PatientInformation* aPatient)
{
    mSelectedPatientUID = aPatient->PatientUID;
}

void SelectFormWidget::clearSelectedPatient()
{
    mSelectedPatientUID.clear();
}

void SelectFormWidget::initDetailPanel(QHBoxLayout *contentLayout) {// prepare edit panel
    patientDetailForm->setObjectName("patientDetailWidget");
    contentLayout->addWidget(patientDetailForm);
}

void SelectFormWidget::initTableView(QLayout *contentLayout)
{
    // TableView for patient
    mPatTable->setAlternatingRowColors(true);
    mPatTable->setSelectionMode(QAbstractItemView::SingleSelection);
    mPatTable->setEditTriggers(QAbstractItemView::NoEditTriggers);
    mPatTable->setSelectionBehavior(QAbstractItemView::SelectRows);
    mPatTable->verticalHeader()->setDefaultSectionSize(38);
    mPatTable->horizontalHeader()->setStretchLastSection(true);

    //avoid pan comsumed by tableview!
    mPatTable->viewport()->ungrabGesture(Qt::PanGesture);

    mPatTable->setSortingEnabled(true); // enable sortingEnabled
    mPatTable->setModel((QAbstractItemModel*) mModel);
    mPatTable->hideColumn(0);
    mPatTable->hideColumn(7);
    mPatTable->hideColumn(8);
    mPatTable->hideColumn(9);
    mPatTable->hideColumn(10);
    mPatTable->hideColumn(11);
    mPatTable->hideColumn(12);

    mPatTable->show();

    mPatTable->setColumnWidth(1, 250);
    mPatTable->setColumnWidth(2, 250);
    mPatTable->setColumnWidth(3, 250);
    mPatTable->setColumnWidth(4, 120);
    mPatTable->setColumnWidth(5, 80);
    mPatTable->setColumnWidth(6, 120);
    contentLayout->addWidget(mPatTable);
    //table current row selection changing event
    connect(mPatTable, &SlideTableView::currentRowChanged, [=](int row) {
        setPatientDetail(mPatTable, mModel, patientDetailForm);
        prepareButtons(false);
    });

    // after sort by column
    connect(mPatTable->horizontalHeader(), &QHeaderView::sectionClicked, [=](int index){
        patientDetailForm->clearPatientInformation();
        prepareButtons(false);
        if(mModel->rowCount() > 0){
            mPatTable->selectRow(0);
            mModel->selectRow(0);
        }
    });

    PatientAddDateDelegate* patientAddDateDelegate = new PatientAddDateDelegate(mPatTable);
    mPatTable->setItemDelegateForColumn(6, patientAddDateDelegate);
}

void SelectFormWidget::initDataModel() {
    mModel = SQLHelper::getTable("Patient");
    WorkListManager::getInstance()->setTableModel(mModel);
    mModel->sort(mModel->record().indexOf("AddDate"), Qt::DescendingOrder);
    mModel->setEditStrategy(QSqlTableModel::OnManualSubmit);
    bool anonymousMode = JsonObject::Instance()->getAnonymousMode();
    if (anonymousMode)
    {
        mModel->setFilter("1=2");
    }
    mModel->select();
    if (anonymousMode)
    {
        mModel->insertRow(0);
        mModel->setData(mModel->index(0,0),"000000001");
        mModel->setData(mModel->index(0,1),"AnonymousPatient");
        mModel->setData(mModel->index(0,2),"");
        mModel->setData(mModel->index(0,3),"AnonymousPatient");
        mModel->setData(mModel->index(0,4),"2000-01-01");
        mModel->setData(mModel->index(0,5),"F");
        mModel->setData(mModel->index(0,6),"2000-01-01");
        mModel->setData(mModel->index(0,7),"");
    }

    mModel->setHeaderData(1, Qt::Horizontal, tr("ID"));
    mModel->setHeaderData(2, Qt::Horizontal, tr("AccessionNumber"));
    mModel->setHeaderData(3, Qt::Horizontal, tr("Name"));
    mModel->setHeaderData(4, Qt::Horizontal, tr("Birth Date"));
    mModel->setHeaderData(5, Qt::Horizontal, tr("Gender"));
    mModel->setHeaderData(6, Qt::Horizontal, tr("Add Date"));
    mModel->setHeaderData(7, Qt::Horizontal, tr("Comment"));
}

void SelectFormWidget::setPatientDetail(const SlideTableView *table, const QSqlTableModel *model,
                                        PatientDetailForm *edit_patient) const {
    PatientInformation pat;
#define ADD_PATIENT_PROPERTY(val)\
        pat. val = model->data(model->index(table->currentIndex().row(),PatientInformationEnum:: val),Qt::EditRole).toString();
    ADD_PATIENT()
#undef ADD_PATIENT_PROPERTY
    edit_patient->setPatientInformation(&pat);
}

void SelectFormWidget::reloadLanguage(){
    mModel->setHeaderData(1, Qt::Horizontal, "ID");
    mModel->setHeaderData(2, Qt::Horizontal, tr("AccessionNumber"));
    mModel->setHeaderData(3, Qt::Horizontal, tr("Name"));
    mModel->setHeaderData(4, Qt::Horizontal, tr("Birth Date"));
    mModel->setHeaderData(5, Qt::Horizontal, tr("Gender"));
    mModel->setHeaderData(6, Qt::Horizontal, tr("Add Date"));
    mModel->setHeaderData(7, Qt::Horizontal, tr("Comment"));

    mBtnEdit->setText(tr("Edit"));
    mBtnDelete->setText(tr("Delete"));
    mBtnAdd->setText(tr("Add"));
    mBtnSelect->setText(tr("Select"));
    mSelectTabTitle->setText(tr("Patient Information Manage"));
}

void SelectFormWidget::updateDataByAnonymousMode(){
    bool anonymousMode = JsonObject::Instance()->getAnonymousMode();
    EventCenter::Default()->triggerEvent(GUIEvents::PatientSelected,this,nullptr);
    if (anonymousMode)
    {
        mModel->setFilter("1=2");
        mModel->select();
        mModel->insertRow(0);
        mModel->setData(mModel->index(0,0),"000000001");
        mModel->setData(mModel->index(0,1),"AnonymousPatient");
        mModel->setData(mModel->index(0,2),"");
        mModel->setData(mModel->index(0,3),"AnonymousPatient");
        mModel->setData(mModel->index(0,4),"2000-01-01");
        mModel->setData(mModel->index(0,5),"M");
        mModel->setData(mModel->index(0,6),"2000-01-01");
        mModel->setData(mModel->index(0,7),""); 
        mPatTable->selectRow(0);
        mModel->selectRow(0);
    }
    else{
        mModel->revertAll();
        mModel->setFilter("");
        mModel->select();
        if (mModel->rowCount()>0){
            mPatTable->selectRow(0);
            mModel->selectRow(0);
        }
    }
    prepareButtons(false);
}