#ifndef STATEMENT_HPP
#define STATEMENT_HPP

#include "bindings.hpp"
#include "declaration.hpp"
#include "expression.hpp"
#include "node.hpp"

#include <memory>

class Statement;

typedef std::shared_ptr<Statement> StatementPtr;


class Statement : public Node
{
protected:
    StatementPtr next_statement_;
    
public:
    Statement(Statement* statement = nullptr);

    virtual void print() const = 0;
    virtual void printXml() const = 0;
    virtual VariableStackBindings printAsm(VariableStackBindings bindings, unsigned& label_count) const = 0;

    virtual void countVariables(unsigned& var_count) const = 0;
    virtual void countArguments(unsigned& argument_count) const = 0;
    virtual void countExpressionDepth(unsigned& depth_count) const = 0;

    void linkStatement(Statement* next);
};


class CompoundStatement : public Statement
{
protected:
    DeclarationPtr declaration_;
    StatementPtr statement_;
    
public:
    CompoundStatement(Declaration* declaration = nullptr, Statement* statement = nullptr);
    CompoundStatement(Statement* statement);

    virtual void print() const;
    virtual void printXml() const;
    virtual VariableStackBindings printAsm(VariableStackBindings bindings, unsigned& label_count) const;

    virtual void countVariables(unsigned& var_count) const;
    virtual void countArguments(unsigned& argument_count) const;
    virtual void countExpressionDepth(unsigned& depth_count) const;
};


class IfElseStatement : public Statement
{
protected:
    ExpressionPtr condition_;
    StatementPtr if_;
    StatementPtr else_;
public:
    IfElseStatement(Expression* condition, Statement* _if, Statement* _else = nullptr);

    virtual void print() const;
    virtual void printXml() const;
    virtual VariableStackBindings printAsm(VariableStackBindings bindings, unsigned& label_count) const;

    virtual void countVariables(unsigned& var_count) const;
    virtual void countArguments(unsigned& argument_count) const;
    virtual void countExpressionDepth(unsigned& depth_count) const;
};


class ExpressionStatement : public Statement
{
protected:
    ExpressionPtr expression_;
public:
    ExpressionStatement(Expression* expression = nullptr);

    virtual void print() const;
    virtual void printXml() const;
    virtual VariableStackBindings printAsm(VariableStackBindings bindings, unsigned& label_count) const;

    virtual void countVariables(unsigned& var_count) const;
    virtual void countArguments(unsigned& argument_count) const;
    virtual void countExpressionDepth(unsigned& depth_count) const;    
};


class JumpStatement : public Statement
{
public:
    virtual VariableStackBindings printAsm(VariableStackBindings bindings, unsigned &label_count) const = 0;
    
    virtual void print() const;
    virtual void printXml() const;

    virtual void countVariables(unsigned &var_count) const;
    virtual void countArguments(unsigned &argument_count) const;
    virtual void countExpressionDepth(unsigned &depth_count) const;
};


class ReturnStatement : public JumpStatement
{
private:
    ExpressionPtr expression_;
public:
    ReturnStatement(Expression* expression = nullptr);

    virtual VariableStackBindings printAsm(VariableStackBindings bindings, unsigned &label_count) const;

    virtual void countVariables(unsigned& var_count) const;
    virtual void countArguments(unsigned& argument_count) const;
    virtual void countExpressionDepth(unsigned& depth_count) const;    
};

class BreakStatement : public JumpStatement
{
public:
    BreakStatement();

    virtual VariableStackBindings printAsm(VariableStackBindings bindings, unsigned &) const;
};

class ContinueStatement : public JumpStatement
{
public:
    ContinueStatement();

    virtual VariableStackBindings printAsm(VariableStackBindings bindings, unsigned &) const;
};

class GotoStatement : public JumpStatement
{
private:
    std::string label_;
public:
    GotoStatement(const std::string &label);

    virtual VariableStackBindings printAsm(VariableStackBindings bindings, unsigned &) const;
};


class IterationStatement : public Statement
{
protected:
    ExpressionPtr condition_;
    StatementPtr statement_;
public:
    IterationStatement(Expression* condition, Statement* statement);

    virtual void print() const;
    virtual void printXml() const;
    virtual VariableStackBindings printAsm(VariableStackBindings bindings, unsigned& label_count) const = 0;

    virtual void countVariables(unsigned& var_count) const;
    virtual void countArguments(unsigned& argument_count) const;
    virtual void countExpressionDepth(unsigned& depth_count) const;
};

class WhileLoop : public IterationStatement
{
private:
    bool is_while_;
public:
    WhileLoop(Expression* condition, Statement* statement, const bool &is_while = true);

    virtual VariableStackBindings printAsm(VariableStackBindings bindings, unsigned& label_count) const;
};

class ForLoop : public IterationStatement
{
private:
    ExpressionPtr initializer_;
    ExpressionPtr incrementer_;
public:
    ForLoop(Expression* initializer, Expression* condition, Expression* incrementer, Statement* statement);

    virtual VariableStackBindings printAsm(VariableStackBindings bindings, unsigned& label_count) const;
};


#endif