//
// Created by Krad on 2021/10/8.
//
#include "ui_TabFormWidget.h"
#include "SelectFormWidget.h"

#include <QToolButton>
#include <QTableWidget>
#include <QScroller>
#include <QHeaderView>
#include <QUuid>
#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/UserOperationLog.h"
#include "components/VerticalTextToolButton.h"
#include "PatientAddDateDelegate.h"

SelectFormWidget::SelectFormWidget(QWidget* parent)
: TabFormWidget(parent)
, mBtnAccount(new QToolButton(this))
, mBtnWorklist(new QToolButton(this))
, mBtnAdd(new QToolButton(this))
, mBtnSelect(new QToolButton(this))
, mBtnTurnOff(new QToolButton(this))
, mPatTable(new SlideTableView(this))
, mModel(nullptr)
, patientDetailForm(new PatientDetailForm(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);
    initGeneralButtons(layout);

    layout->addSpacerItem(new QSpacerItem(20, 20, QSizePolicy::Expanding));
    addVerticalLine(layout);
    initPatEditButtons(layout);
    initDataModel();
    //Init content widget
	auto* contentLayout = new QHBoxLayout(this->ui->contentWidget);
	contentLayout->setContentsMargins(0, 0, 0, 0);
    initDetailPanel(contentLayout);
    addVerticalLine(contentLayout);
    initTableView(contentLayout);
    //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);

	//first prepare buttons!
    prepareButtons(false);
}

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

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

void SelectFormWidget::initGeneralButtons(QHBoxLayout *layout) {
    mBtnAccount->setObjectName("btnAccount");
    mBtnAccount->setText(tr("Account"));
    layout->addWidget(mBtnAccount);
    mBtnTurnOff->setObjectName("btnShutDown");
    mBtnTurnOff->setText(tr("ShutDown"));
    layout->addWidget(mBtnTurnOff);
    // mBtn account slot
    connect(mBtnAccount, &QToolButton::clicked, DialogManager::Default(),&DialogManager::requestEditSelfAccount);
    connect(mBtnTurnOff, &QToolButton::clicked, []()
    {
        if(DialogManager::Default()->requestAlertMessage(QString(tr("Shut down now ?")), DialogButtonMode::OkAndCancel,tr("Shut Down")) == QDialog::Accepted)
        {
            LOG_USER_OPERATION("Shut Down")
            EventCenter::Default()->triggerEvent(GUIEvents::RequestShutdown, nullptr, nullptr);
        }
    });
    connect(mBtnWorklist, &QToolButton::clicked, [&]()
        {
            DialogManager::Default()->requestGetWorkList(mModel, mPatTable);
        });
}

void SelectFormWidget::initPatEditButtons(QHBoxLayout *layout) {
    mBtnAdd->setObjectName("btnAdd");
    mBtnWorklist->setObjectName("btnWorklist");
    mBtnSelect->setObjectName("btnSelect");
    mBtnAdd->setText(tr("Add"));
    mBtnSelect->setText(tr("Select"));
    mBtnWorklist->setText(tr("Worklist"));
    layout->addWidget(mBtnAdd);
    layout->addWidget(mBtnWorklist);
    layout->addWidget(mBtnSelect);

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

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

void SelectFormWidget::editPatient() {
    bool addFlag = sender() == mBtnAdd;
    // 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->clicked(mPatTable->currentIndex());
            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 (selectedPatientUID == 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\" ?")).arg(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, patientDetailForm->getPatientInformation()->Copy());
    selectedPatientUID = patientDetailForm->getPatientInformation()->PatientUID;
    LOG_USER_OPERATION(QString("Select Patient, ID: %1").arg(patientDetailForm->getPatientInformation()->ID))
}

void SelectFormWidget::initDetailPanel(QHBoxLayout *contentLayout) {// prepare edit panel
    patientDetailForm->setObjectName("patientDetailWidget");
    // patientDetailForm->hide();
    contentLayout->addWidget(patientDetailForm);
    auto* btnShowEdit = new VerticalTextToolButton(this);
    btnShowEdit->setObjectName("btnShowPanel");
    btnShowEdit->setFixedHeight(225);
    btnShowEdit->setVerticalText("Patient Detail");
    btnShowEdit->hide();
    contentLayout->addWidget(btnShowEdit);
    contentLayout->setAlignment(btnShowEdit, Qt::AlignTop);

    connect(patientDetailForm, &PatientDetailForm::editClicked, this, &SelectFormWidget::editPatient);
    connect(patientDetailForm, &PatientDetailForm::deleteClicked, this, &SelectFormWidget::delPatient);
}

void SelectFormWidget::initTableView(QHBoxLayout *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);
    //data from SQLITE
//
//avoid pan comsumed by tableview!
    mPatTable->viewport()->ungrabGesture(Qt::PanGesture);

    mPatTable->setSortingEnabled(true); // enable sortingEnabled
    mPatTable->setModel((QAbstractItemModel*) mModel);
    mPatTable->hideColumn(0);
    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);
    mPatTable->setColumnHidden(7, true);
    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() {//TODO:单独初始化预防SQL错误
    mModel = SQLHelper::getTable("Patient");
    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, "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();
    EDIT_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"));

    mBtnAccount->setText(tr("Account"));
    //mBtnWorklist->setText(tr("Worklist"));
    mBtnAdd->setText(tr("Add"));
}

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);
}