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

#include <QToolButton>
#include <QTreeView>
#include <QScroller>
#include <QHeaderView>
#include <QUuid>
#include <QLabel>
#include <QTabWidget>

#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 "LocalTableGenderDelegate.h"
#include "dicom/WorkListManager.h"
#include "utilities/ScanProcessSequence.h"
#include "WorklistTableView.h"
#include "WorklistTableModel.h"
#include "WorklistTableSelectModel.h"
#include "WorklistTableStyleItemDelegate.h"
#include "action/ActionCreator.h"
#include "action/GetWorkListAction.h"

SelectFormWidget::SelectFormWidget(QWidget* parent)
    : TabFormWidget(parent)
    , mTabWidget(new QTabWidget(this))
    , mSelectedPatientUID()
    , mBtnEdit(new QToolButton(this))
    , mBtnDelete(new QToolButton(this))
    , mBtnAdd(new QToolButton(this))
    , mBtnPull(new QToolButton(this))
    , mBtnSelect(new QToolButton(this))
    , mWorklistPatTable(new WorklistTableView(this))
    , mWorklistTableModel(new WorklistTableModel(this))
    , mWorklistTableSelectModel(new WorklistTableSelectModel(mWorklistTableModel, mWorklistPatTable))
    , mLocalPatTable(new SlideTableView(this))
    , mLocalPatientModel(nullptr)
    , mPatientDetailForm(new PatientDetailForm(this))
    , mSelectTabTitle(new QLabel(this))
    , mGetWorklistAction(ActionCreator::getAsyncAction<GetWorkListAction>("GetWorkListAction"))
{
    //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();
    connect(mWorklistTableSelectModel, &WorklistTableSelectModel::WorklistTableSelectChanged, this, &SelectFormWidget::setWorklistPatientDetail);
    mWorklistPatTable->setModel(mWorklistTableModel);
    mWorklistPatTable->setSelectionModel(mWorklistTableSelectModel);

    mWorklistPatTable->setColumnWidth(0, 300);
    mWorklistPatTable->setColumnWidth(1, 300);
    mWorklistPatTable->setColumnWidth(2, 300);
    mWorklistPatTable->setColumnWidth(3, 200);
    mWorklistPatTable->header()->setDefaultAlignment(Qt::AlignCenter);
    mWorklistPatTable->setIndentation(0);
    mWorklistPatTable->setAlternatingRowColors(true);
    mWorklistPatTable->setSelectionMode(QAbstractItemView::MultiSelection);
    mWorklistPatTable->setSelectionBehavior(QAbstractItemView::SelectRows);
    mWorklistPatTable->setItemDelegate(new WorklistTableStyleItemDelegate(mWorklistPatTable->header(), mWorklistPatTable));

    mTabWidget->addTab(mWorklistPatTable,tr("Worklist"));
    mTabWidget->addTab(mLocalPatTable, tr("Local"));
    gridBox->addWidget(mTabWidget);
    gridBox->addWidget(ui->commandWidget);
    connect(mTabWidget, &QTabWidget::tabBarClicked, [this](int aIndex)
    {
        if(aIndex == 0)
        {
            switchToWorklistMode();
        }
        else
        {
            switchToLocalMode();
        }
    });

    //select default row 0
    if (mLocalPatientModel->rowCount() > 0)
	{
        mLocalPatTable->selectRow(0);
        setPatientDetail(mLocalPatTable, mLocalPatientModel, mPatientDetailForm);
    }
    // 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);
    connect(mGetWorklistAction, &AsyncAction::actionCompleted, this, &SelectFormWidget::processWorklistSearchResult);

	//first prepare buttons!
    prepareButtons(false);

    //init WorkListManager table view
    WorkListManager::getInstance()->setTableView(mLocalPatTable);
    //init in worklist mode bt default
    switchToWorklistMode();
    initByAnonymousMode();
}

void SelectFormWidget::initByAnonymousMode()
{
    if(JsonObject::Instance()->getAnonymousMode())
    {
        mTabWidget->setTabEnabled(0, false);
        switchToLocalMode();
    }
}

void SelectFormWidget::prepareButtons(bool disableALL) {
    bool anonymousMode = JsonObject::Instance()->getAnonymousMode();
    bool stateFlag = (mLocalPatTable->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");
    mBtnPull->setObjectName("btnAdd");
    mBtnSelect->setObjectName("btnSelect");

    mBtnEdit->setText(tr("Edit"));
    mBtnDelete->setText(tr("Delete"));
    mBtnAdd->setText(tr("Add"));
    mBtnPull->setText(tr("Pull"));
    mBtnSelect->setText(tr("Select"));
    layout->addWidget(mBtnAdd);
    layout->addWidget(mBtnEdit);
    layout->addWidget(mBtnDelete);
    layout->addWidget(mBtnPull);
    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);

    connect(mBtnPull, &QToolButton::clicked, this, &SelectFormWidget::pullPatient);
}

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

void SelectFormWidget::delPatient() {
    if (mLocalPatTable->currentIndex().row() < 0) return;
    QString pUid = mLocalPatientModel->index(mLocalPatTable->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 = mLocalPatientModel->index(mLocalPatTable->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
    mPatientDetailForm->clearPatientInformation();
    //mLocalPatientModel->setData(mLocalPatientModel->index(mLocalPatTable->currentIndex().row(), Flag), 9);
    mLocalPatientModel->removeRow(mLocalPatTable->currentIndex().row());

    if (mLocalPatientModel->submitAll()) {
        mLocalPatientModel->select();
        if (mLocalPatientModel->rowCount() > 0) {
            mLocalPatTable->selectRow(0);
            mLocalPatientModel->selectRow(0);
            setPatientDetail(mLocalPatTable, mLocalPatientModel, mPatientDetailForm);
            LOG_USER_OPERATION(QString("Delete Patient, ID: %1").arg(mPatientDetailForm->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*)mPatientDetailForm->getPatientInformation());
}

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

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

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

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

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

    mLocalPatTable->setSortingEnabled(true); // enable sortingEnabled
    mLocalPatTable->setModel(mLocalPatientModel);
    mLocalPatTable->hideColumn(0);
    mLocalPatTable->hideColumn(6);
    mLocalPatTable->hideColumn(7);
    mLocalPatTable->hideColumn(8);
    mLocalPatTable->hideColumn(9);
    mLocalPatTable->hideColumn(10);
    mLocalPatTable->hideColumn(11);

    mLocalPatTable->show();

    mLocalPatTable->setColumnWidth(1, 300);
    mLocalPatTable->setColumnWidth(2, 300);
    mLocalPatTable->setColumnWidth(3, 300);
    mLocalPatTable->setColumnWidth(4, 200);
    //table current row selection changing event
    connect(mLocalPatTable, &SlideTableView::currentRowChanged, [=](int row) {
        setPatientDetail(mLocalPatTable, mLocalPatientModel, mPatientDetailForm);
        prepareButtons(false);
    });

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

    PatientAddDateDelegate* patientAddDateDelegate = new PatientAddDateDelegate(mLocalPatTable);
    mLocalPatTable->setItemDelegateForColumn(5, patientAddDateDelegate);
    LocalTableGenderDelegate* patientGenderDelegate = new LocalTableGenderDelegate(mLocalPatTable);
    mLocalPatTable->setItemDelegateForColumn(4, patientGenderDelegate);
}

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

    mLocalPatientModel->setHeaderData(1, Qt::Horizontal, tr("ID"));
    mLocalPatientModel->setHeaderData(2, Qt::Horizontal, tr("Name"));
    mLocalPatientModel->setHeaderData(3, Qt::Horizontal, tr("Birth Date"));
    mLocalPatientModel->setHeaderData(4, Qt::Horizontal, tr("Gender"));
    mLocalPatientModel->setHeaderData(5, Qt::Horizontal, tr("Add Date"));
    mLocalPatientModel->setHeaderData(6, 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);
    mBtnSelect->setEnabled(true);
}

void SelectFormWidget::setWorklistPatientDetail(PatientInformation* aPatient)
{
    mPatientDetailForm->setPatientInformation(aPatient);
    if(aPatient == nullptr || aPatient->getSelectedAccessionIndex().size() == 0)
    {
        mBtnSelect->setEnabled(false);
    }
    else
    {
        mBtnSelect->setEnabled(true);
    }
}

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

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

    mTabWidget->setTabText(0,tr("Worklist"));
    mTabWidget->setTabText(1, tr("Local"));

    if(mTabWidget->currentIndex() == 0)
    {
        mWorklistTableSelectModel->clearSelect();
    }
}

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

void SelectFormWidget::switchToWorklistMode()
{
    mBtnAdd->hide();
    mBtnDelete->hide();
    mBtnEdit->hide();
    mBtnPull->show();
    setWorklistPatientDetail(mWorklistTableSelectModel->getSelectedPatient());
    mPatientDetailForm->setPatientInformation(mWorklistTableSelectModel->getSelectedPatient());
}

void SelectFormWidget::switchToLocalMode()
{
    mBtnAdd->show();
    mBtnDelete->show();
    mBtnEdit->show();
    mBtnPull->hide();

    setPatientDetail(mLocalPatTable, mLocalPatientModel, mPatientDetailForm);
}

void SelectFormWidget::pullPatient()
{
    mGetWorklistAction->execute();
    DialogManager::Default()->requestLoadingWorklist();
}

void SelectFormWidget::processWorklistSearchResult(const ActionResult& aResult)
{
    EventCenter::Default()->triggerEvent(GUIEvents::WorklistSearchFinished, nullptr, nullptr);
    mWorklistTableSelectModel->clearSelect();
    if(aResult.Code == Sucessed)
    {
        QList<PatientInformation*> patientList = aResult.Data.value<QList<PatientInformation*>>();
        mWorklistTableModel->instertPatients(patientList);
    }
    else
    {
        mWorklistTableModel->instertPatients(QList<PatientInformation*>());
        DialogManager::Default()->requestAlertMessage(aResult.Data.toString(), DialogButtonMode::OkOnly);      
    }
}