#ifndef AST_NODE_H
#define AST_NODE_H

#include "token.h"
#include <optional>
#include <vector>

/**
 * AST base class, all AST node classes extend this class. Class contains basic
 * information about nodes location in file.
 */
class AstNode {
public:

    Span span;

    AstNode() = default;

    virtual ~AstNode();

    AstNode(AstNode&&) = default;
    AstNode& operator=(AstNode&&) = default;
    AstNode(const AstNode&) = default;
    AstNode& operator=(const AstNode&) = default;
};

template <typename T>
struct EnumNode: public AstNode
{
    EnumNode() = default;
    explicit EnumNode(T value): value(value) {}

    T value;
};

struct StringNode: public AstNode
{
    std::string value;
    std::string asString();
};

struct IdentifierNode: public AstNode
{
    std::string value;
};

struct NumberNode: public AstNode {
    long long int value;
    explicit NumberNode(const std::string& expression);
    NumberNode(): value(0) {}
};


struct CountNode: public AstNode
{
    NumberNode first;
    NumberNode second;

    CountNode(NumberNode first, NumberNode second): first(first), second(second) {}
    CountNode() = default;
};


struct AddressSpaceNode: public AstNode
{
    IdentifierNode name;
    NumberNode start;
    NumberNode end;
};

class ValueNode: public AstNode
{
public:
    enum ValueType {
        INT,
        STRING,
        BOOL,
        WIRE,
        IDENTIFIER,
        NIL,
    };

private:
    EnumNode<ValueType> type;
    std::optional<long long> intValue;
    std::optional<std::string> stringValue;
    std::optional<bool> boolValue;
    std::optional<std::string> identifierValue;

public:
    ValueNode() = default;;

    ValueType getType() {
        return type.value;
    }
    long long asInt();
    std::string asString();
    std::string asIdentifier();
    bool asBool();

    bool is(ValueType valueType);

    static ValueNode ofBool(bool _value);
    static ValueNode ofInt(long long _value);
    static ValueNode ofString(std::string _value);
    static ValueNode ofIdentifier(std::string _value);
    static ValueNode ofNull();
    static ValueNode ofWire(std::string _value);
};

struct ConditionNode
{
    bool negated;
    IdentifierNode functionName;
    std::vector<ValueNode> params;
};

class ActionNode: public AstNode {
public:
    enum ActionType {
        ERROR,
        WARNING
    };

    EnumNode<ActionType> type;
    StringNode message;
};


struct IfStatementNode: public AstNode
{
    ConditionNode condition;
    ActionNode action;
};

struct RuleNode: public AstNode
{
    std::vector<IfStatementNode> statements;
};

struct EnumerationNode: public AstNode
{
    StringNode key;
    ValueNode value;
};

struct PopupNode: public AstNode
{
    enum PopupType {
        AUTOMATIC,
        ON_DEMAND
    };

    std::optional<EnumNode<PopupType>> type;
    std::optional<StringNode> title;
    std::optional<StringNode> text;

    bool enumerated;
    std::vector<EnumerationNode> enumeration;

    std::vector<RuleNode> rules;
};

struct PropertyNode: public AstNode
{
    IdentifierNode key;
    ValueNode value;
};

struct DisplayItemNode: public AstNode
{
    IdentifierNode type;
    std::vector<PropertyNode> values;
};

struct DisplayNode: public AstNode
{
    std::vector<DisplayItemNode> items;
};

struct PinConnectionNode: public AstNode
{
    enum ConnectionType {
        CHECK_ONLY,
        AUTOMATICALLY
    };

    StringNode message;
    EnumNode<ConnectionType> type;
};

struct PinNode: public AstNode
{
    enum PinType {
        IN_OUT,
        IN,
        OUT
    };

    IdentifierNode name;
    EnumNode<PinType> type;
    std::optional<StringNode> tooltip;
    std::optional<PinConnectionNode> connection;
    std::optional<DisplayNode> display;
    std::optional<std::vector<ValueNode>> wires;
};

struct WireNode: public AstNode
{
    enum WireType {
        WIRE,
        WIRED_AND,
        WIRED_OR,
        R_WIRE
    };
    EnumNode<WireType> type;
    IdentifierNode name;
    NumberNode size;

    bool hidden;

    bool hasTerminateWith;
    ValueNode terminateWith;
};

struct AttributeNode: public AstNode
{
    ValueNode::ValueType type;
    IdentifierNode name;
    std::optional<ValueNode> defaultValue;
    std::optional<PopupNode> popup;
};

struct ConnectionComponentNode: public AstNode
{
    IdentifierNode component;
    IdentifierNode pin;
};

struct ConnectionNode: public AstNode
{
    ConnectionComponentNode first;
    std::optional<ConnectionComponentNode> second;

    IdentifierNode bus;
    std::vector<AttributeNode> attributes;

    std::vector<ValueNode> firstWires;
    std::optional<std::vector<ValueNode>> secondWires;
};

struct ComponentNode: public AstNode
{
    enum ComponentType {
        OTHER,
        PROCESSOR,
        MEMORY
    };

    IdentifierNode name;
    std::optional<StringNode> tooltip;
    std::optional<StringNode> source;
    EnumNode<ComponentType> type;
    std::vector<RuleNode> rules;
    std::optional<StringNode> instanceName;
    std::optional<CountNode> count;
    std::optional<DisplayNode> display;
    std::vector<PinNode> pins;
    std::vector<AttributeNode> attributes;
};

struct BusNode: public AstNode
{
    enum BusType {
        AUTOMATIC,
        REGULAR
    };

    EnumNode<BusType> type;
    IdentifierNode name;
    std::optional<StringNode> tooltip;
    std::optional<CountNode> count;
    std::optional<DisplayNode> display;

    std::vector<WireNode> wires;
};

struct LibraryNode: public AstNode
{
    std::optional<StringNode> name;
    std::optional<StringNode> libraryInfo;
    std::optional<StringNode> header;
    std::optional<StringNode> componentDirectory;

    std::vector<AddressSpaceNode> addressSpaces;

    std::vector<ComponentNode> components;
    std::vector<BusNode> buses;
    std::vector<ConnectionNode> connections;

    std::vector<PropertyNode> messages;
};

// SCHEMA models
struct WireInstanceNode: public AstNode
{
    IdentifierNode name;
    std::optional<CountNode> position;
    std::optional<DisplayNode> display;
};

struct InstanceAttributeNode: public AstNode
{
    IdentifierNode name;
    ValueNode value;
};

struct InstanceNode: public AstNode
{
    IdentifierNode name;
    IdentifierNode component;

    std::optional<CountNode> position;
    std::vector<InstanceAttributeNode> attributes;

    std::optional<NumberNode> size;
};

struct ConnectionComponentInstance: public AstNode
{
    IdentifierNode instance;
    IdentifierNode pin;
};

struct ConnectionInstanceNode: public AstNode
{
    ConnectionComponentInstance first;
    std::optional<ConnectionComponentInstance> second;

    IdentifierNode bus;
    std::optional<IdentifierNode> wire;

    std::vector<InstanceAttributeNode> attributes;
};


struct SchemaNode: public AstNode
{
    std::optional<StringNode> source;

    std::vector<InstanceNode> instances;
    std::vector<WireInstanceNode> wires;
    std::vector<ConnectionInstanceNode> connections;

    std::optional<LibraryNode> library;
};

#endif // AST_NODE_H