#include "component_display.h"
#include "attribute_dialog.h"
#include "name_dialog.h"
#include "mainwindow.h"
#include "application.h"
#include "single_automatic_dialog.h"

#include <QMenu>
#include <QLine>
#include <QPen>
#include <QGraphicsSceneContextMenuEvent>
#include <set>


namespace display {

    QPen connectionPen(QColor::fromRgb(150, 150, 250));

    Component::Component(const std::shared_ptr<domain::ComponentInstance> &instance): instance(instance) {
        setFlag(ItemSendsGeometryChanges, true);
        setToolTip(QString::fromStdString(instance->name + "::" + instance->component.getName()));
        instance->component.getDisplay().render(this);
    }

    void Component::contextMenuEvent(QGraphicsSceneContextMenuEvent *event) {
        QMenu menu;
        menu.addAction("Izmjeni ime", [this]() {
            std::set<std::string> names;
            for(const auto &component: Application::instance()->getSchema()->componentInstances) {
                names.insert(component->name);
            }
            auto dialog = new NameDialog(this->instance->name, names);
            dialog->exec();

            auto currentName = this->instance->name;
            auto newName = dialog->getName();

            Application::instance()->renameComponent(currentName, newName);
            setToolTip(QString::fromStdString(instance->name + "::" + instance->component.getName()));
        });
        menu.addSeparator();
        for (int i = 0; i < this->instance->attributes.size(); i++) {
            auto *attr = &this->instance->attributes[i];
            bool enabled = attr->attribute.getPopup().has_value();

            if(attr->value.getType() == domain::Value::MEMORY_REFERENCE) {
                menu.addAction("Izmjeni memoriju", [attr]() {
                    auto dialog = new MemoryDialog(attr,
                                                   Application::instance()->getSchema()->componentInstances);
                    dialog->exec();
                });
            } else {
                auto action = menu.addAction(QString::fromStdString("Izmjeni '" + attr->name + "'"),
                                             [attr]() {
                                                auto dialog = new AttributeDialog(attr);
                                                dialog->exec();
                                             });
                action->setEnabled(enabled);
            }
        }
        menu.addSeparator();
        menu.addAction(QString::fromStdString("Ukloni " + this->instance->name), [this]() {
            Application::instance()->removeComponent(this->instance->name);
            auto view = dynamic_cast<Schema *>(this->scene()->views()[0]);
            view->refreshContent();
        });

        menu.exec(event->screenPos());
    }

    void Pin::contextMenuEvent(QGraphicsSceneContextMenuEvent *event) {
        QMenu menu;

        auto pinConnections = Application::instance()->getSchema()->getConnections(componentInstance->name, pin.getName());

        if(isSingleAutomatic(pinConnections)) {
            auto *update = menu.addMenu("Izmjeni");
            auto *remove = menu.addMenu("Ukloni");
            for (auto pinConnection: pinConnections) {
                // this always must be true as only directConnections can be connected multiple times
                if (auto directConnection = dynamic_cast<domain::DirectConnectionInstance *>(pinConnection)) {
                    if (directConnection->bus->bus.getType() == domain::Bus::SINGLE_AUTOMATIC) {
                        auto connectionName = directConnection->attributes[0].value.stringify() + "-" +
                                              directConnection->attributes[1].value.stringify();
                        update->addAction(QString::fromStdString("Izmjeni " + connectionName), [directConnection]() {
                            auto dialog = new SingleAutomaticDialog(directConnection->attributes);
                            dialog->exec();
                        });
                        remove->addAction(QString::fromStdString("Ukloni " + connectionName),
                                          [this, directConnection]() {
                                              Application::instance()->removeConnection(directConnection);
                                              auto view = dynamic_cast<Schema *>(this->scene()->views()[0]);
                                              view->refreshContent();
                                          });
                    }
                }
            }
        } else if(!pinConnections.empty()) {
            auto pinConnection = pinConnections[0];
            if(auto busConnection = dynamic_cast<domain::BusConnectionInstance*>(pinConnection)) {
                menu.addSection(QString::fromStdString(busConnection->bus->name));
            } else if(auto directConnection = dynamic_cast<domain::DirectConnectionInstance*>(pinConnection)) {
                if(directConnection->instance == componentInstance.get()) {
                    menu.addSection(QString::fromStdString(directConnection->secondInstance->name + "." + directConnection->connection.getSecondComponent()->pin));
                } else {
                    menu.addSection(QString::fromStdString(directConnection->instance->name + "." + directConnection->connection.getComponent().pin));
                }
            }
            for (int i = 0; i < pinConnection->attributes.size(); i++) {
                auto *attr = &pinConnection->attributes[i];
                menu.addAction(QString::fromStdString("Izmjeni '" + attr->name + "'"),[attr]() {
                    auto dialog = new AttributeDialog(attr);
                    dialog->exec();
                });
            }
            menu.addAction("Ukloni poveznicu", [this, pinConnection]() {
                Application::instance()->removeConnection(pinConnection);
                auto view = dynamic_cast<Schema *>(this->scene()->views()[0]);
                view->refreshContent();
            });
        }
        menu.exec(event->screenPos());
    }

    void Pin::mousePressEvent(QGraphicsSceneMouseEvent *event) {
        if (event->button() == Qt::MouseButton::LeftButton) {
            auto view = dynamic_cast<Schema *>(this->scene()->views()[0]);
            view->state = Schema::CREATING_CONNECTION;
            view->context.pin = this;
            view->context.startingPoint = dynamic_cast<ComponentGroup *>(this->parentItem())->pos() +
                                          QPointF(pin.getDisplayPin().getConnectionX(),
                                                  pin.getDisplayPin().getConnectionY());
            view->context.line = new QGraphicsLineItem(QLineF(view->context.startingPoint, event->scenePos()));
            view->context.line->setPen(QPen(QColor::fromRgb(100, 100, 250), 1, Qt::PenStyle::DashLine));
            view->context.line->setZValue(1000);
            this->scene()->addItem(view->context.line);

            view->showConnectable(this);
        }
    }

    void Pin::mouseMoveEvent(QGraphicsSceneMouseEvent *event) {
        auto view = dynamic_cast<Schema *>(this->scene()->views()[0]);
        if (view->state == Schema::CREATING_CONNECTION) {
            auto line = view->context.line->line();
            line.setP2(event->scenePos());
            view->context.line->setLine(line);
        }
    }

    void Pin::mouseReleaseEvent(QGraphicsSceneMouseEvent *event) {
        if (event->button() == Qt::MouseButton::LeftButton) {
            auto view = dynamic_cast<Schema *>(this->scene()->views()[0]);
            view->state = Schema::DEFAULT;
            this->scene()->removeItem(view->context.line);
            delete view->context.line;
            view->removeConnectable(event->scenePos());
        }
    }

    domain::Pin &Pin::getPin() {
        return pin;
    }

    domain::ComponentInstance *Pin::getComponentInstance() {
        return componentInstance.get();
    }

    bool Pin::isSingleAutomatic(std::vector<domain::ConnectionInstance *> pinConnections) {
        if(pinConnections.empty()) {
            return false;
        }
        auto pinConnection = pinConnections[0];
        if(auto directConnection = dynamic_cast<domain::DirectConnectionInstance*>(pinConnection)) {
            return directConnection->bus->bus.getType() == domain::Bus::SINGLE_AUTOMATIC;
        } else {
            return false;
        }
    }

    Bus::Bus(const std::shared_ptr<domain::BusInstance> &instance): busInstance(instance) {
        instance->bus.getDisplayBus()->render(this, instance->size);
        setToolTip(QString::fromStdString(busInstance->name + "::" + busInstance->bus.getName()));
    }

    void Bus::contextMenuEvent(QGraphicsSceneContextMenuEvent *event) {
        QMenu menu;
        menu.addAction("Izmjeni ime", [this]() {
            std::set<std::string> names;
            for(const auto &component: Application::instance()->getSchema()->busInstances) {
                names.insert(component->name);
            }
            auto dialog = new NameDialog(this->busInstance->name, names);
            dialog->exec();

            auto currentName = this->busInstance->name;
            auto newName = dialog->getName();

            Application::instance()->renameBus(currentName, newName);
            setToolTip(QString::fromStdString(busInstance->name + "::" + busInstance->bus.getName()));
        });
        menu.addSeparator();
        menu.addAction(QString::fromStdString("Ukloni " + this->busInstance->name), [this]() {
            Application::instance()->removeBus(this->busInstance->name);
            auto view = dynamic_cast<Schema *>(this->scene()->views()[0]);
            view->refreshContent();
        });
        menu.exec(event->screenPos());
    }

    QVariant BusGroup::itemChange(QGraphicsItem::GraphicsItemChange change, const QVariant &value) {
        if (change == ItemPositionChange && scene()) {
            // value is the new position.
            QPoint newPos = value.toPointF().toPoint();
            busInstance->position.first = newPos.x();
            busInstance->position.second = newPos.y();

            auto view = dynamic_cast<Schema *>(scene()->views()[0]);
            view->updateConnections();
        }
        return QGraphicsItem::itemChange(change, value);
    }

    BusGroup::BusGroup(const std::shared_ptr<domain::BusInstance> &instance) : busInstance(instance) {
        setFlag(ItemIsMovable, true);
        setFlag(ItemSendsGeometryChanges, true);

        setHandlesChildEvents(false);

        bus = new display::Bus(instance);
        addToGroup(bus);

        setPos(instance->position.first, instance->position.second);
    }


    QVariant ComponentGroup::itemChange(QGraphicsItem::GraphicsItemChange change, const QVariant &value) {
        if (change == ItemPositionChange && scene()) {
            // value is the new position.
            QPoint newPos = value.toPointF().toPoint();
            componentInstance->position.first = newPos.x();
            componentInstance->position.second = newPos.y();

            auto view = dynamic_cast<Schema *>(scene()->views()[0]);
            view->updateConnections();
        }
        return QGraphicsItem::itemChange(change, value);
    }

    std::shared_ptr<domain::ComponentInstance> ComponentGroup::getComponentInstance() {
        return componentInstance;
    }

    std::vector<display::Pin *> &ComponentGroup::getPins() {
        return pins;
    }

    ComponentGroup::ComponentGroup(const std::shared_ptr<domain::ComponentInstance> &instance): componentInstance(
            instance) {
        setFlag(ItemIsMovable, true);
        setFlag(ItemSendsGeometryChanges, true);

        setHandlesChildEvents(false);

        addToGroup(new display::Component(instance));

        for (auto &pin: instance->component.getPins()) {
            pins.push_back(new display::Pin(pin, componentInstance));
            addToGroup(pins.back());
        }

        setPos(instance->position.first, instance->position.second);
    }

    BusConnection::BusConnection(domain::BusConnectionInstance *connection, ComponentGroup *component, BusGroup *bus): connection(
            connection), component(component), bus(bus) {
        updateConnection();
        setPen(connectionPen);
        setZValue(-1);
        setHandlesChildEvents(false);
    }

    void BusConnection::updateConnection() {
        auto busPosition = bus->boundingRect();
        auto pin = connection->instance->component.getPin(
                connection->connection.getComponent().pin).getDisplayPin();

        setLine(connection->instance->position.first + pin.getConnectionX(),
                connection->instance->position.second + pin.getConnectionY(),
                connection->bus->position.first + busPosition.width() / 2,
                connection->bus->position.second + busPosition.height() / 2);

        connection->start.first = connection->instance->position.first + pin.getConnectionX();
        connection->start.second = connection->instance->position.second + pin.getConnectionY();
        connection->end.first = connection->bus->position.first + busPosition.width() / 2;
        connection->end.second = connection->bus->position.second + busPosition.height() / 2;
    }

    DirectConnection::DirectConnection(domain::DirectConnectionInstance *connection, ComponentGroup *first,
                                       ComponentGroup *second): connection(connection), first(first), second(second) {
        updateConnection();
        setPen(connectionPen);
        setZValue(-1);
        setHandlesChildEvents(false);
    }

    void DirectConnection::updateConnection() {
        auto pin1 = connection->instance->component.getPin(
                connection->connection.getComponent().pin).getDisplayPin();
        auto pin2 = connection->secondInstance->component.getPin(
                connection->connection.getSecondComponent()->pin).getDisplayPin();

        setLine(connection->instance->position.first + pin1.getConnectionX(),
                connection->instance->position.second + pin1.getConnectionY(),
                connection->secondInstance->position.first + pin2.getConnectionX(),
                connection->secondInstance->position.second + pin2.getConnectionY());

        connection->start.first = connection->instance->position.first + pin1.getConnectionX();
        connection->start.second = connection->instance->position.second + pin1.getConnectionY();
        connection->end.first = connection->secondInstance->position.first + pin2.getConnectionX();
        connection->end.second = connection->secondInstance->position.second + pin2.getConnectionY();
    }

} // namespace display