#pragma once

#include <ostream>

// for clangd
#include "definitions.hpp"
#include "error_handling.hpp"
#include "typeclass_graph.hpp"
#include "utils.hpp"
#include "visitor.hpp"
#include "global_info.hpp"

namespace interpreter {

// TODO, (maybe add VisitSourceFile?)
class LinkSymbolsVisitor : public Visitor {
public:
  explicit LinkSymbolsVisitor(info::GlobalInfo& global_info)
    : namespace_visitor_(global_info.CreateVisitor()),
      global_info_(global_info),
      typeclass_graph_(*global_info.GetTypeclassGraph()) {}

  void VisitSourceFile(SourceFile* source_file) override {
    Visitor::Visit(source_file);
    namespace_visitor_.GetTypeclassGraph()->CalculateGraph();
  }

private:
  // Sources -----------------

  // // void Visit(SourceFile* node) override;

  // Namespaces, partitions -----------------

  // // void Visit(NamespaceSources* node) override;
  void Visit(Namespace* node) override;

  // Definitions -----------------

  // // void Visit(ImportStatement* node) override;
  // // void Visit(AliasDefinitionStatement* node) override;
  // // void Visit(VariableDefinitionStatement* node) override;
  // // void Visit(FunctionDeclaration* node) override;
  // // void Visit(FunctionDefinitionStatement* node) override;
  void Visit(TypeDefinitionStatement* node) override;
  // // void Visit(AbstractTypeDefinitionStatement* node) override;
  void Visit(TypeclassDefinitionStatement* node) override;
  // // void Visit(PartitionStatement* node) override;

  // Definition parts

  // // void Visit(FunctionDefinition* node) override;
  // // void Visit(TypeDefinition* node) override;
  // // void Visit(AnyAnnotatedType* node) override;

  // Flow control -----------------

  // // void Visit(TypeConstructorPatternParameter* node) override;
  // // void Visit(TypeConstructorPattern* node) override;
  // // void Visit(MatchCase* node) override;
  // // void Visit(Match* node) override;
  // // void Visit(Condition* node) override;
  // // void Visit(DoWhileLoop* node) override;
  // // void Visit(WhileLoop* node) override;
  // // void Visit(ForLoop* node) override;
  // // void Visit(LoopLoop* node) override;

  // Statements, expressions, blocks, etc. -----------------

  // // void Visit(Block* node) override;
  // // void Visit(ScopedStatement* node) override;

  // Operators

  // // void Visit(ReferenceExpression* node) override;
  // // void Visit(AccessExpression* node) override;

  // Simple Expressions

  // // void Visit(FunctionCallExpression* node) override;

  // // void Visit(TupleExpression* node) override;
  // // void Visit(VariantExpression* node) override;
  // // void Visit(ReturnExpression* node) override;
  // // void Visit(TypeConstructorParameter* node) override;
  // // void Visit(TypeConstructor* node) override;
  // // void Visit(LambdaFunction* node) override;
  // // void Visit(AndExpression* node) override;
  // // void Visit(OrExpression* node) override;
  // // void Visit(ArrayExpression* node) override;

  // // void Visit(LoopControlExpression& node) override; // enum

  // Name

  // // void Visit(PartitionName* node) override;
  // // void Visit(NameExpression* node) override;
  // // void Visit(OptionalName* node) override;
  // // void Visit(TupleName* node) override;
  // // void Visit(VariantName* node) override;
  // // void Visit(Name* node) override;

  // Type, typeclass, etc. -----------------

  // Type

  // // void Visit(FunctionType* node) override;
  // // void Visit(TupleType* node) override;
  // // void Visit(VariantType* node) override;
  void Visit(TypeExpression* node) override;

  // // void Visit(ExtendedScopedAnyType* node) override;

  // Typeclass

  void Visit(ParametrizedTypeclass* node) override;

  // Typeclass & Type

  // // void Visit(ParametrizedType* node) override;

  // Identifiers, constants, etc. -----------------

  // // void Visit(std::string* node) override; // std::string

  // // void Visit(FloatNumberLiteral* node) override;
  // // void Visit(NumberLiteral* node) override;
  // // void Visit(StringLiteral* node) override;
  // // void Visit(CharLiteral* node) override;
  // // void Visit(BoolLiteral* node) override;

  //

  void AddTypeFunctionsToTypeclassGraph(utils::IdType type_id,
                                        utils::IdType graph_id,
                                        utils::ClassInternalsModifier namespace_modifier,
                                        const BaseNode& base_node);

private:
  info::GlobalInfo::NamespaceVisitor namespace_visitor_;
  info::GlobalInfo& global_info_;
  info::TypeclassGraph& typeclass_graph_;
};

} // namespace interpreter