#ifndef DOMAIN_CONNECTION_H
#define DOMAIN_CONNECTION_H

#include "attribute.h"

#include <string>
#include <vector>

namespace domain {

    struct ConnectionComponent {
        std::string component;
        std::string pin;

        bool operator==(const ConnectionComponent &rhs) const {
            return (component == rhs.component) && (pin == rhs.pin);
        }

        bool operator!=(const ConnectionComponent &rhs) const {
            return !operator==(rhs);
        }

        bool operator<(const ConnectionComponent &rhs) const {
            if (component < rhs.component) {
                return true;
            } else if (component == rhs.component) {
                if (pin < rhs.pin) {
                    return true;
                }
            }
            return false;
        }
    };

    class Connection {
        ConnectionComponent first;
        std::optional<ConnectionComponent> second;
        std::string bus;
        std::vector<Attribute> attributes;

        std::vector<Value> firstWires;
        std::optional<std::vector<Value>> secondWires;
    public:
        Connection(ConnectionComponent first, std::optional<ConnectionComponent> second,
                   std::string bus, std::vector<Attribute> attributes,
                   std::vector<Value> firstWires, std::optional<std::vector<Value>> secondWires);

        bool isConnecting(ConnectionComponent first);

        bool isConnecting(ConnectionComponent first, std::string bus);

        bool isConnecting(ConnectionComponent first, std::string bus, ConnectionComponent second);

        bool isConnecting(ConnectionComponent first, ConnectionComponent second);

        bool operator==(const Connection& connection) {
            return (first == connection.first && bus == connection.bus && second == connection.second);
        }

        ConnectionComponent getComponent();

        std::optional<ConnectionComponent> getSecondComponent();

        std::string &getBus();

        std::vector<Attribute>& getAttributes();

        std::vector<Value> getWires();

        std::optional<std::vector<Value>> getSecondWires();

        Attribute getAttribute(std::string name);

        bool hasAttribute(std::string name);
    };

} // namespace domain

#endif // DOMAIN_CONNECTION_H