#pragma once

#include <string>
#include <vector>
#include <variant>
#include <optional>
#include <memory>

// for clangd
#include "utils.hpp"

namespace interpreter::tokens {

// ----------------- Declarations -----------------

using AnyIdentifier = std::string;
using NameOrOperatorIdentifier = std::string;
using NameIdentifier = std::string;
using AnyTypeIdentifier = std::string;
using TypeIdentifier = std::string;
using AbstractTypeIdentifier = std::string;
using OperatorIdentifier = std::string;
using TypeclassIdentifier = std::string;

// Sources -----------------

struct SourceFile;
struct Sources;

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

struct Partition;
struct Namespace;

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

struct ImportStatement;
struct AliasDefinitionStatement;
struct VariableDefinitionStatement;
struct FunctionDeclaration;
struct FunctionDefinitionStatement;
struct TypeDefinitionStatement;
struct AbstractTypeDefinitionStatement;
struct TypeclassDefinitionStatement;

using SourceStatement = std::variant<
  std::unique_ptr<ImportStatement>,
  std::unique_ptr<AliasDefinitionStatement>,
  std::unique_ptr<FunctionDeclaration>,
  std::unique_ptr<FunctionDefinitionStatement>,
  std::unique_ptr<TypeDefinitionStatement>,
  std::unique_ptr<AbstractTypeDefinitionStatement>,
  std::unique_ptr<TypeclassDefinitionStatement>,
  std::unique_ptr<Namespace>>;

// Definition parts

using ImportSymbol = AnyIdentifier; // can be extended name

struct FunctionDefinition;
struct TypeDefinition;
struct AnnotatedAbstractType;

// TypeIdentifier <-> AbstractTypeIdentifier <-> std::string
using AnnotatedType = AnnotatedAbstractType;

using AnyAnnotatedType = AnnotatedType;

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

struct TypeConstructorPatternParameter;
struct TypeConstructorPattern;

struct Match;
struct Condition;
struct DoWhileLoop;
struct WhileLoop;
struct ForLoop;
struct LoopLoop;

using FlowControl = std::variant<
  std::unique_ptr<Match>,
  std::unique_ptr<Condition>,
  std::unique_ptr<DoWhileLoop>,
  std::unique_ptr<WhileLoop>,
  std::unique_ptr<ForLoop>,
  std::unique_ptr<LoopLoop>>;

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

struct Block;

struct FloatNumberLiteral;
struct NumberLiteral;
struct StringLiteral;
struct CharLiteral;

using Literal = std::variant<
  std::unique_ptr<FloatNumberLiteral>,
  std::unique_ptr<NumberLiteral>,
  std::unique_ptr<StringLiteral>,
  std::unique_ptr<CharLiteral>>;

//
struct NameExpression;
struct ScopedStatement;
struct AccessExpression;
using SubExpressionToken = std::variant<
  std::unique_ptr<NameExpression>,
  std::unique_ptr<ScopedStatement>,
  std::unique_ptr<AccessExpression>,
  std::unique_ptr<Literal>>;
//
struct FunctionCallExpression;
struct BinaryOperatorExpression;
struct ArrayExpression;
struct ReferenceExpression;
using SubExpression = std::variant<
  std::unique_ptr<FunctionCallExpression>,
  std::unique_ptr<BinaryOperatorExpression>,
  std::unique_ptr<SubExpressionToken>,
  std::unique_ptr<ReferenceExpression>>;
//
enum class LoopControlExpression {
    Break,
    Continue,
};
//
struct ReturnExpression;
using PrefixedExpression = std::variant<
  std::unique_ptr<ReturnExpression>,
  std::unique_ptr<LoopControlExpression>,
  std::unique_ptr<Block>>;
//
struct LambdaFunction;
struct TypeConstructor;
struct UnaryOperatorExpression;
using Expression = std::variant<
  std::unique_ptr<LambdaFunction>,
  std::unique_ptr<TypeConstructor>,
  std::unique_ptr<PrefixedExpression>,
  std::unique_ptr<UnaryOperatorExpression>,
  std::unique_ptr<SubExpression>>;
//
struct TupleExpression;
struct VariantExpression;
using SuperExpression = std::variant<
  std::unique_ptr<FlowControl>,
  std::unique_ptr<TupleExpression>,
  std::unique_ptr<VariantExpression>,
  std::unique_ptr<ArrayExpression>,
  std::unique_ptr<Expression>>;
//

struct ScopedStatement;

// Operators

struct BinaryOperatorExpression;
struct UnaryOperatorExpression;
struct ReferenceExpression;

// Other expressions

struct FunctionCallExpression;

struct TupleExpression;
struct VariantExpression;
struct ReturnExpression;
struct TypeConstructorParameter;
struct TypeConstructor;
struct LambdaFunction;
struct ArrayExpression;

// Name

struct NameExpression;
struct TupleName;
struct VariantName;
struct AnnotatedName;

// // ScopedAnyName <-> AnyName
using AnyName = std::variant<
  std::unique_ptr<AnnotatedName>,
  std::unique_ptr<TupleName>,
  std::unique_ptr<VariantName>>;

using ScopedAnyName = AnyName;

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

// Type

struct FunctionType;
struct TupleType;
struct VariantType;
struct TypeExpression;
struct TypeExpression;

using Constructor = std::string;

// // ScopedAnyType <-> AnyType
using AnyType = std::variant<
  std::unique_ptr<TypeExpression>,
  std::unique_ptr<TupleType>,
  std::unique_ptr<VariantType>,
  std::unique_ptr<FunctionType>>;

using ScopedAnyType = AnyType;

struct ExtendedScopedAnyType;

// Typeclass

struct ParametrizedTypeclass;
struct TypeclassExpression;

using TypeclassSubExpression = std::variant<
  std::unique_ptr<TypeclassIdentifier>,
  std::unique_ptr<ParametrizedTypeclass>>;

// Typeclass & Type

struct ParametrizedType;

using TypeSubExpression = std::variant<
  std::unique_ptr<AnyTypeIdentifier>,
  std::unique_ptr<ParametrizedType>>;

//

using FunctionArgument = std::variant<
  SubExpressionToken,
  TypeSubExpression>;

// Comments [IGNORE] -----------------
// Identifiers, constants, etc. -----------------

struct ExtendedName {
  std::string name;
};

using PatternToken = std::variant<
  std::unique_ptr<ExtendedName>,
  std::unique_ptr<Literal>,
  std::unique_ptr<TypeConstructorPattern>>;


using Pattern = std::variant<
  std::unique_ptr<TypeConstructorPattern>,
  std::unique_ptr<PatternToken>>;

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

// ----------------- Sources -----------------

struct SourceFile {
  std::vector<std::variant<SourceStatement, Partition>> statements;
};

struct Sources {
  std::vector<SourceStatement> statements;
};

// ----------------- Namespaces, partittions -----------------

struct Namespace {
  enum Modifier { Const, Var };
  std::optional<Modifier> modifier;
  std::optional<ExtendedName> name;
  TypeIdentifier type;
  std::unique_ptr<Sources> scope;

  std::optional<utils::IdType> type_id_;
};

struct Partition {
  enum PartitionName {
    Test,
    Interface,
    Core,
    Lib,
    Module,
    Exe,
  };

  PartitionName name;
  std::unique_ptr<Sources> scope;
};

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

struct ImportStatement {
  std::optional<TypeIdentifier> name;
  std::string module_name;
  std::vector<ImportSymbol> symbols;
};

struct AliasDefinitionStatement {
  enum {Alias, Type, Let} modifier;
  TypeIdentifier type;
  std::vector<AbstractTypeIdentifier> parameters;
  std::unique_ptr<TypeExpression> value;

  std::vector<utils::IdType> parameter_graph_ids_;
  utils::IdType type_id_;
};

struct VariableDefinitionStatement {
  enum { Var, Const } modifier;
  enum { Move, Assign } assignment_modifier;
  AnyName name;
  SuperExpression value;
};

struct FunctionDeclaration {
  ExtendedName name;
  std::vector<std::unique_ptr<AnnotatedAbstractType>> parameters;
  std::unique_ptr<FunctionType> type;

  utils::IdType function_id_;
};

struct FunctionDefinitionStatement {
  bool is_inline;
  std::unique_ptr<FunctionDefinition> definition;
  SuperExpression value;

  utils::IdType function_id_;
  std::vector<utils::IdType> argument_graph_ids_;
  utils::IdType return_type_graph_id_;
};

struct TypeDefinitionStatement {
  enum { Struct, Class } modifier;
  std::unique_ptr<TypeDefinition> definition;
  AnyType value;

  utils::IdType type_id_;
};

struct AbstractTypeDefinitionStatement {
  enum { Basic, Abstract } modifier;
  std::unique_ptr<AnnotatedType> type;

  utils::IdType type_graph_id_;
  utils::IdType type_id_;
};

struct TypeclassDefinitionStatement {
  std::unique_ptr<TypeDefinition> definition;
  std::vector<std::unique_ptr<FunctionDeclaration>> requirements;

  utils::IdType typeclass_id_;
};

// Definition parts -----------------

struct FunctionDefinition {
  enum { Operator, Function } modifier;
  ExtendedName name;
  std::vector<std::unique_ptr<AnnotatedAbstractType>> parameters;
  std::vector<ExtendedName> arguments;
};

struct TypeDefinition {
  std::unique_ptr<AnnotatedType> type;
  std::vector<std::unique_ptr<AnnotatedAbstractType>> parameters;
};

struct AnnotatedAbstractType {
  AbstractTypeIdentifier type;
  std::vector<std::unique_ptr<TypeclassExpression>> typeclasses;

  utils::IdType type_graph_id_;
};

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

struct TypeConstructorPatternParameter {
  std::optional<ExtendedName> name;
  PatternToken value;
};

struct TypeConstructorPattern {
  TypeIdentifier constructor;
  std::vector<TypeConstructorPatternParameter> parameters;
};

struct MatchCase {
  Pattern value;
  std::optional<Expression> condition;
  std::optional<Expression> statement;
};

struct Match {
  Expression value;
  std::vector<MatchCase> matches;
};

struct Condition {
  std::vector<Expression> conditions; // if, elif
  std::vector<Expression> statements; // if, elif, else
};

struct DoWhileLoop {
  Expression condition;
  Expression statement;
};

struct WhileLoop {
  Expression condition;
  Expression statement;
};

struct ForLoop {
  AnyName variable;
  Expression interval;
  Expression statement;
};

struct LoopLoop {
  Expression statement;
};

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

using BlockStatement = std::variant<
  std::unique_ptr<Expression>,
  std::unique_ptr<VariableDefinitionStatement>,
  std::unique_ptr<FlowControl>,
  std::unique_ptr<PrefixedExpression>>;

struct Block {
  std::vector<BlockStatement> statements;
};

struct ScopedStatement {
  SuperExpression statement;
};

// Operators -----------------

struct BinaryOperatorExpression {
  OperatorIdentifier operator_name;
  SubExpression left_expression;
  SubExpression right_expression;
};

struct UnaryOperatorExpression {
  OperatorIdentifier operator_name;
  Expression expression;
};

struct ReferenceExpression {
  std::vector<utils::ReferenceType> references;
  std::unique_ptr<ScopedStatement> expression;
};

struct AccessExpression {
  std::unique_ptr<NameExpression> name;
  SubExpressionToken id;
};

// Other Expressions -----------------

struct FunctionCallExpression {
  std::optional<std::variant<std::unique_ptr<SubExpressionToken>,
                             std::unique_ptr<TypeExpression>>> prefix;
  ExtendedName name;
  std::vector<FunctionArgument> arguments;
};

struct TupleExpression {
  std::vector<SubExpression> expressions;
};

struct VariantExpression {
  std::vector<SubExpression> expressions;
};

struct ReturnExpression {
  Expression expression;
};

struct TypeConstructorParameter {
  enum AssignmentModifier { Move, Assign };
  std::optional<ExtendedName> name;
  std::optional<AssignmentModifier> asignment_modifier;
  PatternToken value;
};

struct TypeConstructor {
  std::unique_ptr<TypeExpression> constructor;
  std::vector<TypeConstructorParameter> parameters;
};

struct LambdaFunction {
  std::vector<std::unique_ptr<AnnotatedAbstractType>> parameters;
  std::vector<ExtendedName> arguments;
  Expression expression;

  std::vector<utils::IdType> argument_graph_ids_;
  utils::IdType return_type_graph_id_;
};

struct ArrayExpression {
  std::vector<SubExpression> elements;
};

// Name -----------------

struct NameExpression {
  std::vector<ExtendedName> names;
};

struct TupleName {
  std::vector<ScopedAnyName> names;
};

struct VariantName {
  std::vector<ScopedAnyName> names;
};

struct AnnotatedName {
  NameIdentifier name;
  std::optional<ScopedAnyType> type;
};

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

// Type -----------------

struct FunctionType {
  std::vector<ScopedAnyType> types;
};

struct TupleType {
  std::optional<Constructor> type;
  std::vector<std::pair<std::optional<ExtendedName>, std::unique_ptr<ExtendedScopedAnyType>>> entities;
};

struct VariantType {
  std::optional<Constructor> type;
  std::vector<std::variant<Constructor, std::unique_ptr<TupleType>>> constructors;
};

struct TypeExpression {
  std::vector<TypeSubExpression> path;
  TypeSubExpression type;

  std::optional<size_t> array_size; // if array; 0 - dynamic size

  utils::IdType type_id_;
};

struct ExtendedScopedAnyType {
  std::vector<utils::ReferenceType> references;
  AnyType type;
};

// Typeclass -----------------

struct TypeclassExpression {
    std::vector<TypeSubExpression> path;
    TypeclassSubExpression typeclass;

    utils::IdType type_id_;
};

struct ParametrizedTypeclass {
    TypeclassIdentifier typeclass;
    std::vector<std::unique_ptr<TypeExpression>> parameters;
};

// Typeclass & Type -----------------

struct ParametrizedType {
  AnyTypeIdentifier type;
  std::vector<std::unique_ptr<TypeExpression>> parameters;
};

// ----------------- Comments [IGNORE] -----------------
// ----------------- Identifiers, constants, etc. -----------------

struct FloatNumberLiteral {
  double value;
};

struct NumberLiteral {
  int64_t value;
};

struct StringLiteral {
  std::string value;
};

struct CharLiteral {
  char value;
};

} // namespace interpereter::tokens