#pragma once

#include "elna/boot/ast.h"
#include "elna/boot/symbol.h"
#include "elna/gcc/elna-tree.h"

#include "config.h"
#include "system.h"
#include "coretypes.h"
#include "tree.h"
#include "tree-iterator.h"

#include <forward_list>
#include <string>

namespace elna
{
namespace gcc
{
    class generic_visitor final : public boot::empty_visitor
    {
        std::forward_list<block_scope> scope;
        tree current_expression{ NULL_TREE };
        std::shared_ptr<boot::symbol_table<tree>> symbol_map;
        tree main_fndecl{ NULL_TREE };

        tree build_label_decl(const char *name, location_t loc);
        tree build_type(boot::type_expression& type);

        void enter_scope();
        tree_symbol_mapping leave_scope();

        void make_if_branch(boot::conditional_statements& branch, tree goto_endif);

        bool is_numeric_type(tree type);

        tree build_arithmetic_operation(boot::binary_expression *expression,
                tree_code operator_code, tree left, tree right);
        tree build_comparison_operation(boot::binary_expression *expression,
                tree_code operator_code, tree left, tree right);
        tree build_logic_operation(boot::binary_expression *expression,
                tree_code operator_code, tree left, tree right);
        tree build_equality_operation(boot::binary_expression *expression,
                tree_code operator_code, tree left, tree right);

    public:
        generic_visitor(std::shared_ptr<boot::symbol_table<tree>> symbol_table);

        void visit(boot::program *program) override;
        void visit(boot::procedure_definition *definition) override;
        void visit(boot::call_expression *expression) override;
        void visit(boot::cast_expression *expression) override;
        void visit(boot::size_of_expression *expression) override;
        void visit(boot::number_literal<std::int32_t> *literal) override;
        void visit(boot::number_literal<std::uint32_t> *literal) override;
        void visit(boot::number_literal<double> *literal) override;
        void visit(boot::number_literal<bool> *boolean) override;
        void visit(boot::number_literal<unsigned char> *character) override;
        void visit(boot::number_literal<std::nullptr_t> *) override;
        void visit(boot::number_literal<std::string> *string) override;
        void visit(boot::binary_expression *expression) override;
        void visit(boot::unary_expression *expression) override;
        void visit(boot::constant_definition *definition) override;
        void visit(boot::type_definition *definition) override;
        void visit(boot::variable_declaration *declaration) override;
        void visit(boot::variable_expression *expression) override;
        void visit(boot::array_access_expression *expression) override;
        void visit(boot::field_access_expression *expression) override;
        void visit(boot::dereference_expression *expression) override;
        void visit(boot::assign_statement *statement) override;
        void visit(boot::if_statement *statement) override;
        void visit(boot::while_statement *statement) override;
        void visit(boot::call_statement *statement) override;
        void visit(boot::return_statement *statement) override;
        void visit(boot::defer_statement *statement) override;
    };
}
}