#include "PropertyBrowser.h"

#include <QBoxLayout>
#include <QDebug>
#include <QSet>

#include "PropertyBrowser/qttreepropertybrowser.h"
#include "PropertyBrowser/qtpropertymanager.h"
#include "PropertyBrowser/qteditorfactory.h"

#include "DockTitleBar.h"
#include "DockWidget.h"
#include "workspace/WorkSpace.h"
#include "entities/Entity.h"
#include "entities/MeshComponent.h"

#include "common/SpdLogger.h"

PropertyBrowser::PropertyBrowser(QWidget *parent) :
    QWidget(parent) {

    InitUI();

    //Test();
}

PropertyBrowser::~PropertyBrowser() {
   
}

void PropertyBrowser::AttachDock(DockWidget* dockWidget) {
    if (nullptr == dockWidget) {
        LOG_WARN("dockwidget is nullptr");
        return;
    }

    dockWidget->SetDockWidgetTitleBar(nullptr);
    dockWidget->setWidget(this);

    DockTitleBar* dockTitleBar = new DockTitleBar;
    dockTitleBar->SetTitle(tr("attribute"));
    dockWidget->SetDockWidgetTitleBar(dockTitleBar);
}

void PropertyBrowser::OnWorkSpaceChange(const QVariant& value) {
    WorkSpace* workspace = value.value<WorkSpace*>();
    if (nullptr == workspace) {
        LOG_WARN("workspace is nullptr");
        return;
    }

    browser_->clear();
    // Reset local mappings when workspace changes
    propertyToId_.clear();
    idToProperty_.clear();
    idToExpanded_.clear();

    QtProperty* property;

    property = workSpaceManager_->addProperty(tr("WorkSpace"));
    QWorkspaceAttribute worksapceAttribute(workspace);
    workSpaceManager_->setValue(property, worksapceAttribute);
    addProperty(property, tr("WorkSpace"));

    // Track and react to runtime workspace changes
    if (currentWorkspace_) {
        QObject::disconnect(currentWorkspace_, nullptr, this, nullptr);
    }
    currentWorkspace_ = workspace;
    QObject::connect(currentWorkspace_, &WorkSpace::FilesChanged,
        this, &PropertyBrowser::OnWorkspaceFilesChanged);

    // Mount all curve entries, each as its own top-level group
    auto curves = currentWorkspace_->GetFileEntries(FileEntryType::Curve);
    for (const auto& entry : curves) {
        if (!entry) continue;
        auto curve = entry->AsCurve();
        if (!curve) continue;

        // Use filename as stable unique id, and include display name in group title
        const QString id = QString("CurveEntry:%1").arg(entry->GetFileName());
        const QString title = QString("CurveEntry - %1").arg(entry->GetName());

        QtProperty* cprop = curveEntryManager_->addProperty(title);
        QCurveEntryAttribute attr(curve);
        curveEntryManager_->setValue(cprop, attr);
        addProperty(cprop, id);
    }
}

void PropertyBrowser::OnEntityChange(const QVariant& value) {
    Entity* entity = value.value<Entity*>();
    if (nullptr == entity) {
        LOG_WARN("engity is nullptr");
        return;
    }

    browser_->clear();

    QtProperty* property;
    property = entityManager_->addProperty(tr("Entity"));
    QEntityAttribute entityAttribute(entity);
    entityManager_->setValue(property, entityAttribute);
    addProperty(property, tr("Entity"));

    SceneComponent* component = entity->GetRootComponent();
    if (nullptr == component) {
        LOG_WARN("component is nullptr");
        return;
    }

    QtComponentPropertyManager* componentManager = GetCompononetPropertyManager(component->GetSelfTypeName().c_str());
    if (nullptr == componentManager) {
        LOG_WARN("component is nullptr");
        return;
    }
    property = componentManager->AddProperty();
    componentManager->SetPropertyValue(property, component);
    addProperty(property, componentManager->GetPropertyId());

    std::vector<SceneComponent*> children = component->GetChildren();
    for (auto child : children) {
        componentManager = GetCompononetPropertyManager(child->GetSelfTypeName().c_str());
        if (nullptr == componentManager) {
            LOG_WARN("component is nullptr id {}", child->GetSelfTypeName());
            continue;
		}

        if (nullptr == child) {
            LOG_WARN("component is nullptr id {}", child->GetSelfTypeName());
            return;
        }
        property = componentManager->AddProperty();
        componentManager->SetPropertyValue(property, child);
        addProperty(property, componentManager->GetPropertyId());
    }
}

void PropertyBrowser::InitUI() {
    QBoxLayout* layout = new QVBoxLayout(this);
    layout->setContentsMargins(0, 0, 0, 0);
    browser_ = new QtTreePropertyBrowser;
    layout->addWidget(browser_);

    browser_->setHeaderVisible(true);
    //browser_->setResizeMode(QtTreePropertyBrowser::Stretch);

    InitPropertyManager();
    InitComponentPropertyManager();
}

void PropertyBrowser::InitPropertyManager() {
    intManager_ = new QtIntPropertyManager(this);
    boolManager_ = new QtBoolPropertyManager(this);
    doubleManager_ = new QtDoublePropertyManager(this);
    stringManager_ = new QtStringPropertyManager(this);
    colorManager_ = new QtColorPropertyManager(this);
    fontManager_ = new QtFontPropertyManager(this);
    pointManager_ = new QtPointPropertyManager(this);
    sizeManager_ = new QtSizePropertyManager(this);
    modelBaseManager_ = new QtModelBasePropertyManager(this);
    workSpaceManager_ = new QtWorkspacePropertyManager(this);
    entityManager_ = new QtEntityPropertyManager(this);
    curveEntryManager_ = new QtCurveEntryPropertyManager(this);

    QtDoubleSpinBoxFactory* doubleSpinBoxFactory = new QtDoubleSpinBoxFactory(this);
    QtCheckBoxFactory* checkBoxFactory = new QtCheckBoxFactory(this);
    QtSpinBoxFactory* spinBoxFactory = new QtSpinBoxFactory(this);
    QtLineEditFactory* lineEditFactory = new QtLineEditFactory(this);
    QtEnumEditorFactory* comboBoxFactory = new QtEnumEditorFactory(this);
    QtFilePathFactory* filePathFactory = new QtFilePathFactory(this);
    QtColorEditorFactory* colorFactory = new QtColorEditorFactory(this);
    //QtTransfromEditorFactory* transformFactory = new QtTransfromEditorFactory(this);

    browser_->setFactoryForManager(intManager_, spinBoxFactory);
    browser_->setFactoryForManager(boolManager_, checkBoxFactory);
    browser_->setFactoryForManager(doubleManager_, doubleSpinBoxFactory);
    browser_->setFactoryForManager(stringManager_, lineEditFactory);
    browser_->setFactoryForManager(colorManager_->subIntPropertyManager(), spinBoxFactory);
    browser_->setFactoryForManager(fontManager_->subIntPropertyManager(), spinBoxFactory);
    browser_->setFactoryForManager(fontManager_->subBoolPropertyManager(), checkBoxFactory);
    browser_->setFactoryForManager(fontManager_->subEnumPropertyManager(), comboBoxFactory);
    browser_->setFactoryForManager(pointManager_->subIntPropertyManager(), spinBoxFactory);
    browser_->setFactoryForManager(sizeManager_->subIntPropertyManager(), spinBoxFactory);
    browser_->setFactoryForManager(workSpaceManager_->subStringProperyManager(), lineEditFactory);
    browser_->setFactoryForManager(workSpaceManager_->subFilesProperyManager(), filePathFactory);
    // Enable editing for grouped file entry counts
    browser_->setFactoryForManager(workSpaceManager_->subIntProperyManager(), spinBoxFactory);
    browser_->setFactoryForManager(entityManager_->subStringProperyManager(), lineEditFactory);
    browser_->setFactoryForManager(entityManager_->subBoolProperyManager(), checkBoxFactory);
    browser_->setFactoryForManager(
        entityManager_->subTransfromProperyManager()->subVec3TPropertyManager()->subDoublePropertyManager(),
        doubleSpinBoxFactory);
    browser_->setFactoryForManager(
        entityManager_->subTransfromProperyManager()->subVec3RPropertyManager()->subDoublePropertyManager(),
        doubleSpinBoxFactory);
    browser_->setFactoryForManager(
        entityManager_->subTransfromProperyManager()->subVec3SPropertyManager()->subDoublePropertyManager(),
        doubleSpinBoxFactory);

    // Curve entry sub editors
    browser_->setFactoryForManager(curveEntryManager_->subStringProperyManager(), lineEditFactory);
    browser_->setFactoryForManager(curveEntryManager_->subEnumProperyManager(), comboBoxFactory);
    browser_->setFactoryForManager(curveEntryManager_->subIntProperyManager(), spinBoxFactory);
    browser_->setFactoryForManager(curveEntryManager_->subDoubleProperyManager(), doubleSpinBoxFactory);
    browser_->setFactoryForManager(curveEntryManager_->subColorProperyManager(), colorFactory);
}

void PropertyBrowser::OnWorkspaceFilesChanged(FileEntryType type, std::shared_ptr<FileEntry> fileEntry) {
    if (!currentWorkspace_) return;
    // Refresh workspace group
    {
        auto it = idToProperty_.find(tr("WorkSpace"));
        if (it != idToProperty_.end()) {
            QtProperty* property = it.value();
            QWorkspaceAttribute worksapceAttribute(currentWorkspace_);
            workSpaceManager_->setValue(property, worksapceAttribute);
        }
    }

    // Refresh curve entry groups when curve files mutate
    if (type == FileEntryType::Curve) {
        const auto curves = currentWorkspace_->GetFileEntries(FileEntryType::Curve);

        // Desired ids for current curves
        QSet<QString> desiredIds;
        for (const auto& entry : curves) {
            if (!entry) continue;
            desiredIds.insert(QString("CurveEntry:%1").arg(entry->GetFileName()));
        }

        // Add or update properties for all current curves
        for (const auto& entry : curves) {
            if (!entry) continue;
            auto curve = entry->AsCurve();
            if (!curve) continue;

            const QString id = QString("CurveEntry:%1").arg(entry->GetFileName());
            const QString title = QString("CurveEntry - %1").arg(entry->GetName());
            QCurveEntryAttribute attr(curve);

            auto it = idToProperty_.find(id);
            if (it != idToProperty_.end()) {
                // Update existing group
                curveEntryManager_->setValue(it.value(), attr);
            } else {
                // Create new group for this curve
                QtProperty* prop = curveEntryManager_->addProperty(title);
                curveEntryManager_->setValue(prop, attr);
                addProperty(prop, id);
            }
        }

        // Remove groups for curves that no longer exist
        QList<QString> existingIds = idToProperty_.keys();
        for (const auto& id : existingIds) {
            if (id.startsWith("CurveEntry:") && !desiredIds.contains(id)) {
                QtProperty* prop = idToProperty_[id];
                browser_->removeProperty(prop);
                propertyToId_.remove(prop);
                idToExpanded_.remove(id);
                idToProperty_.remove(id);
            }
        }
    }
}

void PropertyBrowser::InitComponentPropertyManager() {
    QtDoubleSpinBoxFactory* doubleSpinBoxFactory = new QtDoubleSpinBoxFactory(this);
    QtCheckBoxFactory* checkBoxFactory = new QtCheckBoxFactory(this);
    QtSpinBoxFactory* spinBoxFactory = new QtSpinBoxFactory(this);
    QtLineEditFactory* lineEditFactory = new QtLineEditFactory(this);
    QtColorEditorFactory* colorFactory = new QtColorEditorFactory(this);
    QtFilePathFactory* filePathFactory = new QtFilePathFactory(this);
    QtModelFilePathFactory* modelFileFactory = new QtModelFilePathFactory(this);
    QtEntityUUIDEditorFactory* entityUUIDFactory = new QtEntityUUIDEditorFactory(this);

    QtMeshComponetManager* meshComponentManager = new QtMeshComponetManager(this);
    browser_->setFactoryForManager(meshComponentManager->subModelFileProperyManager(), modelFileFactory);
    componetManager_[meshComponentManager->GetPropertyId()] = meshComponentManager;

    QtPathComponentManager* pathComponentManager = new QtPathComponentManager(this);
    browser_->setFactoryForManager(pathComponentManager->subFilesProperyManager(), filePathFactory);
    componetManager_[pathComponentManager->GetPropertyId()] = pathComponentManager;

    QtConeWaveComponentManager* coneWaveComponentManager = new QtConeWaveComponentManager(this);
    browser_->setFactoryForManager(coneWaveComponentManager->subDoubleProperyManager(), doubleSpinBoxFactory);
    browser_->setFactoryForManager(coneWaveComponentManager->subIntProperyManager(), spinBoxFactory);
    browser_->setFactoryForManager(coneWaveComponentManager->subColorProperyManager(), colorFactory);
    componetManager_[coneWaveComponentManager->GetPropertyId()] = coneWaveComponentManager;

    QtDashedLineComponentManager* dashedLineComponentManager = new QtDashedLineComponentManager(this);
    browser_->setFactoryForManager(dashedLineComponentManager->subEntityProperyManager(), entityUUIDFactory);
    browser_->setFactoryForManager(dashedLineComponentManager->subDoubleProperyManager(), doubleSpinBoxFactory);
    browser_->setFactoryForManager(dashedLineComponentManager->subColorProperyManager(), colorFactory);
    componetManager_[dashedLineComponentManager->GetPropertyId()] = dashedLineComponentManager;
}

void PropertyBrowser::addProperty(QtProperty* property, const QString& id) {
    propertyToId_[property] = id;
    idToProperty_[id] = property;
    QtBrowserItem* item = browser_->addProperty(property);
    if (idToExpanded_.contains(id))
        browser_->setExpanded(item, idToExpanded_[id]);
}

QtComponentPropertyManager* PropertyBrowser::GetCompononetPropertyManager(const QString& id) {
    if (!componetManager_.contains(id)) {
        return nullptr;
    }
    return componetManager_[id];
}

void PropertyBrowser::Test() {
    QtProperty* property;

    property = modelBaseManager_->addProperty(tr("ModelBase"));
    QModelAttbute modelAttr(QString("test"), QString("test222"), true, 1.0 , true);
    modelBaseManager_->setValue(property, modelAttr);
    addProperty(property, QLatin1String("brush"));

    property = colorManager_->addProperty(tr("color base"));
    colorManager_->setValue(property, QColor(255, 12, 231, 252));
    addProperty(property, QLatin1String("bda"));
}