#pragma once

#include <optional>
#include <string>
#include <variant>
#include <memory>
#include <unordered_set>
#include <unordered_map>

// for clangd
#include "utils.hpp"

namespace info::type {

// TODO: move in constructors

class TypeManager;

class AbstractType { // later will be found in context
public:
  AbstractType() = default;
  AbstractType(utils::AbstractTypeModifier modifier,
               const std::string& name,
               const std::vector<utils::IdType>& requirements) : modifier_(modifier), name_(name) {
    for (auto& typeclass : requirements) {
      requirements_.insert(typeclass);
    }
  }

  std::optional<utils::IdType> InContext(const std::unordered_map<std::string, utils::IdType>& context);
  bool Same(const AbstractType& type) const;
  bool operator<(const AbstractType& type) const;
  bool operator>(const AbstractType& type) const;

  std::optional<utils::IdType> GetFieldType(const std::string& name) const;

private:
  utils::AbstractTypeModifier modifier_;
  std::string name_;
  std::unordered_set<utils::IdType> requirements_; // TODO: all typeclasses from tree
};

class DefinedType {
public:
  DefinedType() = default;
  DefinedType(utils::IdType type_id,
              utils::IdType type,
              TypeManager* type_manager)
    : type_id_(type_id), type_(type), type_manager_(type_manager) {}

  std::optional<utils::IdType> InContext(const std::unordered_map<std::string, utils::IdType>& context);
  bool Same(const DefinedType& type) const;
  bool operator<(const DefinedType& type) const;
  bool operator>(const DefinedType& type) const;

  std::optional<utils::IdType> GetFieldType(const std::string& name) const;

  utils::IdType GetTypeId() const {
    return type_id_;
  }

  utils::IdType GetType() const {
    return type_;
  }

private:
  utils::IdType type_id_; // in defined types
  utils::IdType type_; // in types manager, created using context types (if specific type)
  TypeManager* type_manager_ = nullptr;
};

enum class InternalType {
  Float = 0,
  Int = 1,
  String = 2,
  Char = 3,
  Bool = 4,
  Unit = 5,
};

class TupleType {
public:
  TupleType() = default;
  TupleType(const std::optional<std::string>& name,
            const std::vector<std::pair<std::optional<std::string>, utils::IdType>>& fields,
            TypeManager* type_manager)
    : name_(name), fields_(fields), type_manager_(type_manager) {}

  std::optional<utils::IdType> InContext(const std::unordered_map<std::string, utils::IdType>& context);
  bool Same(const TupleType& type) const;
  bool operator<(const TupleType& type) const;
  bool operator>(const TupleType& type) const;

  std::optional<utils::IdType> GetFieldType(const std::string& name) const;

  const std::vector<std::pair<std::optional<std::string>, utils::IdType>>&  GetFields() const {
    return fields_;
  }

private:
  std::optional<std::string> name_;
  std::vector<std::pair<std::optional<std::string>, utils::IdType>> fields_;
  TypeManager* type_manager_ = nullptr;
};

class VariantType {
public:
  VariantType() = default;
  VariantType(const std::optional<std::string>& name,
              const std::vector<TupleType>& constructors,
              std::optional<size_t> current_constructor)
    : name_(name), constructors_(constructors), current_constructor_(current_constructor) {}

  std::optional<utils::IdType> InContext(const std::unordered_map<std::string, utils::IdType>& context);
  bool Same(const VariantType& type) const;
  bool operator<(const VariantType& type) const;
  bool operator>(const VariantType& type) const;

  std::optional<utils::IdType> GetFieldType(const std::string& name) const;

  const std::vector<TupleType>&  GetConstructors() const {
    return constructors_;
  }

  const void SetCurrentConstructor(size_t constructor) {
    current_constructor_ = constructor;
  }

private:
  std::optional<std::string> name_;
  std::vector<TupleType> constructors_;
  std::optional<size_t> current_constructor_;
};

class OptionalType {
public:
  OptionalType() = default;
  OptionalType(utils::IdType type,
               TypeManager* type_manager)
    : type_(type), type_manager_(type_manager) {}

  std::optional<utils::IdType> InContext(const std::unordered_map<std::string, utils::IdType>& context);
  bool Same(const OptionalType& type) const;
  bool operator<(const OptionalType& type) const;
  bool operator>(const OptionalType& type) const;

  std::optional<utils::IdType> GetFieldType(const std::string& name) const;

private:
  utils::IdType type_;
  TypeManager* type_manager_ = nullptr;
};

class ReferenceToType {
public:
  ReferenceToType() = default;
  ReferenceToType(const std::vector<utils::ReferenceModifier>& references,
                  utils::IdType type,
                  TypeManager* type_manager)
    : references_(references), type_(type), type_manager_(type_manager) {}

  std::optional<utils::IdType> InContext(const std::unordered_map<std::string, utils::IdType>& context);
  bool Same(const ReferenceToType& type) const;
  bool operator<(const ReferenceToType& type) const;
  bool operator>(const ReferenceToType& type) const;

  std::optional<utils::IdType> GetFieldType(const std::string& name) const;

private:
  std::vector<utils::ReferenceModifier> references_;
  utils::IdType type_;
  TypeManager* type_manager_ = nullptr;
};

/////////////////////////////

class FunctionType {
public:
  FunctionType() = default;
  FunctionType(const std::vector<utils::IdType>& argument_types,
               utils::IdType return_type,
               TypeManager* type_manager)
    : argument_types_(argument_types), return_type_(return_type), type_manager_(type_manager) {}

  std::optional<utils::IdType> InContext(const std::unordered_map<std::string, utils::IdType>& context);
  bool Same(const FunctionType& type) const;
  bool operator<(const FunctionType& type) const;
  bool operator>(const FunctionType& type) const;

  std::optional<utils::IdType> GetFieldType(const std::string& name) const;

private:
  std::vector<utils::IdType> argument_types_;
  utils::IdType return_type_;
  TypeManager* type_manager_ = nullptr;
};

class ArrayType {
public:
  ArrayType() = default;
  ArrayType(size_t size,
            utils::IdType  elements_type,
            TypeManager* type_manager)
    : size_(size), elements_type_(elements_type), type_manager_(type_manager) {}

  std::optional<utils::IdType> InContext(const std::unordered_map<std::string, utils::IdType>& context);
  bool Same(const ArrayType& type) const;
  bool operator<(const ArrayType& type) const;
  bool operator>(const ArrayType& type) const;

  std::optional<utils::IdType> GetFieldType(const std::string& name) const;

  utils::IdType GetElementsType() {
    return elements_type_;
  }

private:
  size_t size_; // = 0 for dynamic
  utils::IdType elements_type_;
  TypeManager* type_manager_ = nullptr;
};

class Type {
public:
  Type() = default;

  template<typename T>
  explicit Type(const T& type) : type_(type) {}

  std::optional<utils::IdType> InContext(const std::unordered_map<std::string, utils::IdType>& context);
  bool Same(const Type& type) const;
  bool operator<(const Type& type) const; // TODO: rule exceptions
  bool operator>(const Type& type) const;

  std::optional<utils::IdType> GetFieldType(const std::string& name) const;

  std::string GetTypeName() const;

  std::variant<AbstractType,
               DefinedType,
               InternalType,
               TupleType,
               VariantType,
               ReferenceToType,
               FunctionType,
               ArrayType,
               OptionalType>& GetType() {
    return type_;
  }
private:
  std::variant<AbstractType,
               DefinedType,
               InternalType,
               TupleType,
               VariantType,
               ReferenceToType,
               FunctionType,
               ArrayType,
               OptionalType> type_;
};

class TypeManager {
public:
  template<typename T>
  utils::IdType AddType(const T& type, utils::ValueType value_type) {
    types_.push_back(std::pair<Type, utils::ValueType> {type, value_type});
    return types_.size() - 1;
  }

  utils::IdType AddAnyType(Type&& type, utils::ValueType value_type) {
    types_.push_back(std::pair<Type, utils::ValueType> {std::move(type), value_type});
    return types_.size() - 1;
  }

  template<typename T>
  std::optional<T*> GetType(utils::IdType type_id) {
    if (!std::holds_alternative<T>(types_.at(type_id).first.GetType())) {
      return std::nullopt;
    }
    return &std::get<T>(types_.at(type_id).first.GetType());
  }

  Type* GetAnyType(utils::IdType type_id) {
    return &types_.at(type_id).first;
  }

  utils::ValueType GetValueType(utils::IdType type_id) {
    return types_.at(type_id).second;
  }

  bool EqualTypes(utils::IdType first_type, utils::IdType second_type) {
    return GetAnyType(first_type)->Same(*GetAnyType(second_type));
  }

  bool AddTypeRequirement(utils::IdType type, utils::IdType requrement) {
    return *GetAnyType(requrement) < *GetAnyType(type);
  }

private:
  std::vector<std::pair<Type, utils::ValueType>> types_;
};

} // namespace info::type