#ifndef STORY_NODE_BASE_H
#define STORY_NODE_BASE_H


#include <iostream>

#include <QtCore/QObject>
#include <QtWidgets/QLabel>

#include <QtNodes/NodeDelegateModel>
#include <QtNodes/NodeDelegateModelRegistry>
#include <QtNodes/Definitions>
#include "json.hpp"

using QtNodes::NodeData;
using QtNodes::NodeDataType;
using QtNodes::NodeDelegateModel;
using QtNodes::NodeId;

class StoryNodeBase : public NodeDelegateModel
{
public:
    struct NodeGeometryData
    {
        QSize size;
        QPointF pos;
    };

    ~StoryNodeBase() {
        std::cout << "Delete node: " << m_nodeId << std::endl;
    }

    void setNodeId(NodeId id) { m_nodeId = id; }
    NodeId getNodeId() const { return m_nodeId; }

    virtual void SetOutPortCount(int count) {
        // default implementation does nothing
    }

    virtual nlohmann::json ToJson() const {
        nlohmann::json j;

        j["model-name"] = name().toStdString();
        return j;
    }

    virtual void FromJson(nlohmann::json &j) {
        // default implementation does nothing
    }

    virtual void setInternalData(const nlohmann::json &j) {
        // default impl
    }

    virtual std::string GenerateConstants() {
        return "";
    }

    virtual std::string Build() {
        return "";
    }

    virtual std::string EntryLabel() const = 0;

    NodeGeometryData  &geometryData() { return m_geometryData; }

    std::string GetNodeTitle() const { return m_nodeTitle; }

private:
    NodeId m_nodeId;
    std::string m_nodeTitle{"Media node"};
    NodeGeometryData m_geometryData;
};


#endif // STORY_NODE_BASE_H