part of find_symbols_visitor done

This commit is contained in:
ProgramSnail 2023-04-07 12:13:31 +03:00
parent f88a23194f
commit 18e85f794f
6 changed files with 827 additions and 57 deletions

View file

@ -0,0 +1,29 @@
#include <iostream>
enum class ErrorVisitor {
BuildVisitor,
PrintVisitor,
FindSymbolsVisitor,
// ...
};
// TODO
inline void handle_error(std::string message, ErrorVisitor visitor) { // TODO: add place in code
std::string visitor_str;
switch (visitor) {
case ErrorVisitor::BuildVisitor:
visitor_str = "Build Visitor";
break;
case ErrorVisitor::PrintVisitor:
visitor_str = "Print Visitor";
break;
case ErrorVisitor::FindSymbolsVisitor:
visitor_str = "Find Symbols Visitor";
break;
// ...
default:
break;
}
std::cerr << "Error: " << message << " in " << visitor_str;
exit(1);
}

View file

@ -0,0 +1,123 @@
#pragma once
#include <ostream>
#include <any>
// for clangd
#include "visitor.hpp"
#include "global_info.hpp"
// TODO:
// add/remove global variables
//
namespace interpreter {
class FindSymbolsVisitor : public Visitor {
public:
explicit FindSymbolsVisitor(info::GlobalInfo& global_info) : global_info_(global_info) {}
private:
//// void Visit(Node* node) override;
// Sources -----------------
void Visit(SourceFile* node) override;
void Visit(Sources* node) override;
// Namespaces, partitions -----------------
void Visit(Partition* node) override;
void Visit(Namespace* node) override;
// Definitions -----------------
void Visit(ImportStatement* node) override;
void Visit(UsageDefinition* node) override;
void Visit(AliasDefinition* node) override;
//// void Visit(VariableDefinition* node) override; // TODO: decide
void Visit(FunctionDeclaration* node) override;
void Visit(FunctionDefinition* node) override;
void Visit(AliasTypeDefinition* node) override;
void Visit(TypeDefinition* node) override;
void Visit(TypeclassDefinition* node) override;
// Definition parts
void Visit(DefinedName* node) override;
void Visit(DefinedAnnotatedName* node) override;
void Visit(DefinedType* node) override;
void Visit(DefinedTypeclass* node) override;
void Visit(DefinitionParameter* node) override;
void Visit(DefinitionArgument* node) override;
// Flow control -----------------
//// 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;
//// void Visit(LoopControlExpression& node) override; // enum
// Operators
void Visit(BinaryOperatorExpression* node) override;
void Visit(UnaryOperatorExpression* node) override;
// Simple Expressions
//// void Visit(FunctionCallExpression* node) override;
//// void Visit(TupleExpression* node) override;
//// void Visit(VariantExpression* node) override;
//// void Visit(ReturnExpression* node) override;
// Lambda
//// void Visit(LambdaFunction* node) override;
// Name
void Visit(NameSuperExpression* node) override;
void Visit(NameExpression* node) override;
void Visit(TupleName* node) override;
void Visit(VariantName* node) override;
void Visit(AnnotatedName* node) override;
// Type
void Visit(TypeConstructor* node) override;
void Visit(TupleType* node) override;
void Visit(VariantType* node) override;
void Visit(AnnotatedType* node) override;
void Visit(ParametrizedType* node) override;
void Visit(TypeExpression* node) override;
// Typeclass
void Visit(AnnotatedTypeclass* node) override;
void Visit(ParametrizedTypeclass* node) override;
void Visit(TypeclassExpression* node) override;
// Identifiers, constants, etc. -----------------
//// void Visit(AnyIdentifier* node) override; // std::string
//// void Visit(FloatNumberLiteral* node) override;
//// void Visit(NumberLiteral* node) override;
//// void Visit(StringLiteral* node) override;
//// void Visit(CharLiteral* node) override;
private:
info::GlobalInfo& global_info_;
std::any current_info_;
};
} // namespace interpreter

View file

@ -8,18 +8,95 @@
namespace info { namespace info {
// TODO: partitions
class GlobalInfo { class GlobalInfo {
public: public:
GlobalInfo(); GlobalInfo() {
namespace_stack.push_back(&global_namespace_);
}
template<typename T>
void AddImport(T&& import_info) {
if (waiting_usage.has_value()) {
usages_[waiting_usage.value()] = std::forward(import_info);
waiting_usage = std::nullopt;
} else {
imports_.push_back(std::forward(import_info));
}
}
void AddUsageForNextImport(const std::string& name) {
if (waiting_usage.has_value()) {
// error
}
waiting_usage = name;
}
void AddEnterNamespace(const std::optional<std::string>& var,
const std::string& type) {
NamespaceInfo* namespace_info = &namespace_stack.back()->namespaces[type].emplace_back();
namespace_stack.push_back(namespace_info);
namespace_info->var = var;
namespace_info->type_name = type;
}
void ExitNamespace() {
if (namespace_stack.size() <= 1) {
// error
return;
}
namespace_stack.pop_back();
}
void ToGlobalNamespace() {
namespace_stack.clear();
namespace_stack.push_back(&global_namespace_);
}
template<typename T>
void AddFunctionDeclaration(const std::string& name,
T&& function_declaration_info) {
FunctionInfo* function_info = &namespace_stack.back()->functions[name];
function_info->declaration = std::forward(function_declaration_info);
}
template<typename T>
void AddFunctionDefinition(const std::string& name,
T&& function_definition_info) {
FunctionInfo* function_info = &namespace_stack.back()->functions[name];
function_info->definition = std::forward(function_definition_info);
}
template<typename T>
void AddType(const std::string& type,
T&& type_info) {
&namespace_stack.back()->types[type] = std::forward(type_info);
}
template<typename T>
void AddTypeclass(const std::string& typeclass,
T&& typeclass_info) {
namespace_stack.back()->typeclasses[typeclass] = std::forward(typeclass_info);
}
// AddVar ?? // TODO: decide
// FindFunction
// FindType
// FindVar ??
// ?? EnterNamespace / ExitNamespace ??
// concurrent work ??
// AddType, AddFunction
// TODO
private: private:
std::unordered_map<std::string, NamespaceInfo> namespaces_; std::vector<NamespaceInfo*> namespace_stack;
std::optional<std::string> waiting_usage;
NamespaceInfo global_namespace_; NamespaceInfo global_namespace_;
// lock for concurrency ?? std::vector<ImportInfo> imports_;
std::unordered_map<std::string, ImportInfo> usages_;
}; };
} // namespace info } // namespace info

View file

@ -1,10 +1,12 @@
#pragma once #pragma once
#include "interpreter_tree.hpp"
#include <string> #include <string>
#include <variant> #include <variant>
#include <optional>
#include <vector> #include <vector>
#include <optional>
#include <unordered_map> #include <unordered_map>
#include <memory>
// for clangd // for clangd
@ -34,63 +36,110 @@ struct Value {
}; };
// better variant value storage (then string) ?? // better variant value storage (then string) ??
struct Info {
std::string name;
};
struct VariantTypeInfo; struct VariantTypeInfo;
struct TupleTypeInfo; struct TupleTypeInfo;
struct AliasTypeInfo; struct AliasTypeInfo;
struct TypeInfo;
struct TypeclassInfo;
enum class BuiltInTypeInfo { enum class BuiltInTypeInfo {
StringT, StringT,
IntT, IntT,
FloatT, FloatT,
}; };
using TypeInfo = std::variant<VariantTypeInfo, TupleTypeInfo, AliasTypeInfo, BuiltInTypeInfo>; struct TypeUsageInfo {
std::vector<std::string> param_names;
std::vector<TypeInfo*> param_types;
std::vector<std::string> arg_names; // ??, arg expr ??
};
struct VariableInfo : public Info { struct ParameterInfo {
Value value; std::string name;
std::vector<std::string> param_names;
std::vector<TypeclassInfo*> param_types;
std::vector<std::string> arg_names; // ??
};
struct AnyTypeInfo {
std::vector<ParameterInfo> params; // ??
std::vector<std::string> arg_names;
std::variant<std::unique_ptr<VariantTypeInfo>,
std::unique_ptr<TupleTypeInfo>,
std::unique_ptr<BuiltInTypeInfo>> info;
};
struct VariableInfo {
std::string name;
std::optional<Value> value;
TypeInfo* type = nullptr; TypeInfo* type = nullptr;
}; };
// TODO lambda functions as values // TODO lambda functions as values
// TODO aliases ?? struct TupleTypeInfo {
std::optional<std::string> name;
struct TupleTypeInfo : public Info { std::vector<std::pair<std::optional<std::string>, AnyTypeInfo>> fields;
std::vector<std::pair<std::optional<std::string>, TypeInfo>> fields;
}; };
struct VariantTypeInfo : public Info { struct VariantTypeInfo {
std::optional<std::string> name;
std::vector<std::variant<std::string, TupleTypeInfo>> constructors; std::vector<std::variant<std::string, TupleTypeInfo>> constructors;
// ?? any type instead of tuple type ?? // ?? any type instead of tuple type ??
}; };
struct AliasTypeInfo : public Info { struct AliasTypeInfo {
bool isAnotherType; // = true by default ?? std::vector<std::string> params;
TypeInfo* type = nullptr; bool isAnotherType;
TypeUsageInfo value;
}; };
struct TypeclassInfo : public Info { struct TypeInfo { std::variant<AliasTypeInfo, AnyTypeInfo> type; };
// TODO
// struct PointerInfo { // ??
// VariableInfo* variable;
// };
struct FunctionDeclarationInfo {
std::vector<ParameterInfo> params;
std::vector<TypeUsageInfo> arg_types;
}; };
struct PointerInfo { struct FunctionDefinitionInfo {
VariableInfo* variable; std::vector<ParameterInfo> params; // TODO: dublicates ??
std::vector<std::string> arg_names;
interpreter::tokens::SuperExpression* expression;
}; };
struct FunctionInfo : public Info { struct FunctionInfo {
interpreter::Node* definition; FunctionDeclarationInfo declaration;
std::vector<VariableInfo> args; FunctionDefinitionInfo definition;
// add requirements ?? // add requirements ??
}; };
struct NamespaceInfo : public Info { struct TypeclassInfo {
std::vector<ParameterInfo> params;
std::vector<std::string> arg_names;
std::vector<FunctionDeclarationInfo> requirements;
};
struct ImportInfo {
std::string module_name;
std::vector<std::string> symbols;
};
struct NamespaceInfo {
std::unordered_map<std::string, TypeInfo> types; std::unordered_map<std::string, TypeInfo> types;
std::unordered_map<std::string, VariableInfo> variables; std::unordered_map<std::string, TypeclassInfo> typeclasses;
// std::unordered_map<std::string, VariableInfo> variables; // TODO
std::unordered_map<std::string, FunctionInfo> functions; std::unordered_map<std::string, FunctionInfo> functions;
std::unordered_map<std::string, NamespaceInfo> namespaces; std::unordered_map<std::string, std::vector<NamespaceInfo>> namespaces;
std::optional<std::string> var;
std::string type_name;
TypeInfo* type_info = nullptr;
}; };
} // namespace info } // namespace info

View file

@ -24,7 +24,7 @@ void BuildVisitor::Visit(SourceFile* node) {
Visit(std::get<SourceStatement>(node->statements[i])); Visit(std::get<SourceStatement>(node->statements[i]));
} else if (current_node_type == parser::tokens::Partition) { } else if (current_node_type == parser::tokens::Partition) {
node->statements[i].emplace<Partition>(); node->statements[i].emplace<Partition>();
Visit(&std::get<Partition>(node->statements[i])); Visit(&std::get<Partition>(node->statements[i]) );
} }
} }
@ -1174,27 +1174,27 @@ void BuildVisitor::Visit(AnyName& node) {
// Type // Type
void BuildVisitor::Visit(TypeConstructor* node) { void BuildVisitor::Visit(TypeConstructor* node) {
auto parse_node = current_node_; // auto parse_node = current_node_;
//
current_node_ = parse_node.ChildByFieldName("type"); // current_node_ = parse_node.ChildByFieldName("type");
node->type = std::make_unique<ParametrizedType>(); // node->type = std::make_unique<ParametrizedType>();
Visit(node->type.get()); // Visit(node->type.get());
//
size_t parameter_count = (parse_node.NamedChildCount() - 1) / 2; // size_t parameter_count = (parse_node.NamedChildCount() - 1) / 2;
//
node->parameters.resize(parameter_count); // node->parameters.resize(parameter_count);
//
for (size_t i = 0; i < parameter_count * 2; ++i) { // for (size_t i = 0; i < parameter_count * 2; ++i) {
current_node_ = parse_node.NthNamedChild(i + 1); // current_node_ = parse_node.NthNamedChild(i + 1);
//
if (i % 2 == 0) { // if (i % 2 == 0) {
node->parameters[i / 2].first = current_node_.GetValue(); // node->parameters[i / 2].first = current_node_.GetValue();
} else { // } else {
Visit(node->parameters[i / 2].second); // Visit(node->parameters[i / 2].second);
} // }
} // }
//
current_node_ = parse_node; // current_node_ = parse_node;
} }
void BuildVisitor::Visit(TupleType* node) { void BuildVisitor::Visit(TupleType* node) {
@ -1538,8 +1538,8 @@ void BuildVisitor::Visit(NameSubSuperExpression& node) {
std::string current_node_type = current_node_.GetType(); std::string current_node_type = current_node_.GetType();
if (current_node_type == parser::tokens::NameIdentifier) { // optimize ?? if (current_node_type == parser::tokens::NameIdentifier) { // optimize ??
// TODO: choose : node = std::make_unique<NameIdentifier>(current_node_.GetValue()); node = std::make_unique<NameIdentifier>(current_node_.GetValue());
Visit(std::get<std::unique_ptr<NameIdentifier>>(node).get()); //Visit(std::get<std::unique_ptr<NameIdentifier>>(node).get());
} else if (current_node_type == parser::tokens::Literal) { } else if (current_node_type == parser::tokens::Literal) {
node = std::make_unique<Literal>(); node = std::make_unique<Literal>();
Visit(*std::get<std::unique_ptr<Literal>>(node)); Visit(*std::get<std::unique_ptr<Literal>>(node));

View file

@ -0,0 +1,492 @@
// for clangd
#include "../include/find_symbols_visitor.hpp"
namespace interpreter {
// Sources -----------------
void FindSymbolsVisitor::Visit(SourceFile* node) {
for (auto& statement : node->statements) {
if (std::holds_alternative<Partition>(statement)) {
Visit(&std::get<Partition>(statement));
} else if (std::holds_alternative<SourceStatement>(statement)) {
Visitor::Visit(std::get<SourceStatement>(statement));
} else {
// error
}
}
}
void FindSymbolsVisitor::Visit(Sources* node) {
for (auto& statement : node->statements) {
Visitor::Visit(statement);
}
}
// Namespaces, partitions -----------------
void FindSymbolsVisitor::Visit(Partition* node) {
// TODO
Visit(node->scope.get());
}
void FindSymbolsVisitor::Visit(Namespace* node) {
if (std::holds_alternative<std::unique_ptr<DefinedAnnotatedName>>(node->name)) {
Visit(std::get<std::unique_ptr<DefinedAnnotatedName>>(node->name).get());
auto info = std::move(std::any_cast<std::pair<std::string, std::string>>(current_info_));
global_info_.AddEnterNamespace(info.first, info.second);
current_info_.reset();
} else if (std::holds_alternative<std::unique_ptr<DefinedType>>(node->name)) {
Visit(std::get<std::unique_ptr<DefinedType>>(node->name).get());
auto info = std::move(std::any_cast<std::string>(current_info_));
global_info_.AddEnterNamespace(std::nullopt, info);
current_info_.reset();
} else {
// error
}
Visit(node->scope.get());
global_info_.ExitNamespace();
}
// Definitions -----------------
void FindSymbolsVisitor::Visit(ImportStatement* node) {
info::ImportInfo info;
info.module_name = node->module_name;
for (auto& symbol : node->symbols) {
Visitor::Visit(symbol);
info.symbols.push_back(std::move(std::any_cast<std::string>(current_info_)));
}
global_info_.AddImport(std::move(info));
// TODO: what if inside usage definition, global_info_.AddImport(std::move(info));
}
void FindSymbolsVisitor::Visit(UsageDefinition* node) {
global_info_.AddUsageForNextImport(node->name);
Visit(node->import_statement.get());
}
// <-- current position
void FindSymbolsVisitor::Visit(AliasDefinition* node) {
info::TypeInfo info;
Visit(node->type.get());
auto type_info = std::move(std::any_cast<std::pair<std::string, std::vector<std::string>>>(current_info_));
current_info_.reset();
Visit(node->value.get());
auto value_info = std::move(std::any_cast<info::TypeUsageInfo>(current_info_));
current_info_.reset();
info::AliasTypeInfo alias_info;
alias_info.isAnotherType = false;
alias_info.params = std::move(type_info.second);
alias_info.value = std::move(value_info);
info.type = std::move(alias_info);
global_info_.AddType(type_info.first, info);
}
// void FindSymbolsVisitor::Visit(VariableDefinition* node) { // TODO: decide
// out_ << "(Variable " << (node->is_const ? "const" : "var") << ' ';
// Visit(&node->name);
// out_ << " = ";
// Visitor::Visit(node->value);
// out_ << ")\n";
// }
void FindSymbolsVisitor::Visit(FunctionDeclaration* node) {
info::FunctionDeclarationInfo info;
for (auto& parameter : node->parameters) {
Visit(parameter.get());
auto param_info = std::move(std::any_cast<info::ParameterInfo>(current_info_));
current_info_.reset();
info.params.push_back(std::move(param_info));
}
for (auto& argument_type : node->argument_types) {
Visitor::Visit(argument_type);
auto type_info = std::move(std::any_cast<info::TypeUsageInfo>(current_info_));
current_info_.reset();
info.arg_types.push_back(std::move(type_info));
}
global_info_.AddFunctionDeclaration(node->name, std::move(info));
}
void FindSymbolsVisitor::Visit(FunctionDefinition* node) {
info::FunctionDefinitionInfo info;
Visit(node->name.get());
auto name_info = std::move(
std::any_cast<std::pair<std::string,
std::pair<std::vector<info::ParameterInfo>,
std::vector<std::string>>>>(current_info_));
current_info_.reset();
info.params = std::move(name_info.second.first);
info.arg_names = std::move(name_info.second.second);
info.expression = &node->value;
global_info_.AddFunctionDefinition(name_info.first, std::move(info));
}
void FindSymbolsVisitor::Visit(AliasTypeDefinition* node) { // TODO: unite with AliasDefinition
info::TypeInfo info;
Visit(node->type.get());
auto type_info = std::move(std::any_cast<std::pair<std::string, std::vector<std::string>>>(current_info_));
current_info_.reset();
Visit(node->value.get());
auto value_info = std::move(std::any_cast<info::TypeUsageInfo>(current_info_));
current_info_.reset();
info::AliasTypeInfo alias_info;
alias_info.isAnotherType = true;
alias_info.params = std::move(type_info.second);
alias_info.value = std::move(value_info);
info.type = std::move(alias_info);
global_info_.AddType(type_info.first, std::move(info));
}
void FindSymbolsVisitor::Visit(TypeDefinition* node) {
info::TypeInfo info;
Visit(node->type.get());
auto type_info = std::move(
std::any_cast<std::pair<std::string,
std::pair<std::vector<info::ParameterInfo>,
std::vector<std::string>>>>(current_info_));
current_info_.reset();
Visitor::Visit(node->value);
auto value_info = std::move(std::any_cast<std::variant<info::VariantTypeInfo,
info::TupleTypeInfo,
info::BuiltInTypeInfo>>(current_info_));
current_info_.reset();
info::AnyTypeInfo any_type_info;
any_type_info.params = std::move(type_info.second.first);
any_type_info.arg_names = std::move(type_info.second.second);
switch (value_info.index()) {
case 0:
any_type_info.info =
std::make_unique<info::VariantTypeInfo>(
std::move(std::get<info::VariantTypeInfo>(value_info)));
break;
case 1:
any_type_info.info =
std::make_unique<info::TupleTypeInfo>(
std::move(std::get<info::TupleTypeInfo>(value_info)));
break;
case 2:
any_type_info.info =
std::make_unique<info::BuiltInTypeInfo>(
std::move(std::get<info::BuiltInTypeInfo>(value_info)));
break;
default:
// error
break;
}
info.type = std::move(any_type_info);
global_info_.AddType(type_info.first, std::move(info));
}
void FindSymbolsVisitor::Visit(TypeclassDefinition* node) {
info::TypeclassInfo info;
Visit(node->typeclass.get());
auto typeclass_info = std::move(
std::any_cast<std::pair<std::string,
std::pair<std::vector<info::ParameterInfo>,
std::vector<std::string>>>>(current_info_));
current_info_.reset();
info.params = std::move(typeclass_info.second.first);
info.arg_names = std::move(typeclass_info.second.second);
for (auto& requirement : node->requirements) {
Visit(requirement.get());
auto requrement_info = std::move(std::any_cast<info::FunctionDeclarationInfo>(current_info_));
current_info_.reset();
info.requirements.push_back(std::move(requrement_info));
}
global_info_.AddTypeclass(typeclass_info.first, std::move(info));
}
// Definition parts
void FindSymbolsVisitor::Visit(DefinedName* node) {
out_ << "(DefinedName ";
Visit(&node->name);
if (!node->parameters.empty()) {
out_ << "\n";
}
for (auto& parameter : node->parameters) {
Visit(parameter.get());
}
if (!node->arguments.empty()) {
out_ << " : \n";
}
for (auto& argument : node->arguments) {
Visit(argument.get());
}
out_ << ")";
}
void FindSymbolsVisitor::Visit(DefinedAnnotatedName* node) {
out_ << "(DefinedAnnotatedName ";
Visit(&node->name);
out_ << " : ";
if (std::holds_alternative<std::unique_ptr<DefinedType>>(node->type)) {
Visit(std::get<std::unique_ptr<DefinedType>>(node->type).get());
} else if (std::holds_alternative<std::unique_ptr<DefinedTypeclass>>(node->type)) {
Visit(std::get<std::unique_ptr<DefinedTypeclass>>(node->type).get());
} else {
// no annotation
}
out_ << " )";
}
void FindSymbolsVisitor::Visit(DefinedType* node) {
out_ << "(DefinedType ";
Visit(node->type.get());
if (!node->parameters.empty()) {
out_ << "\n";
}
for (auto& parameter : node->parameters) {
Visit(parameter.get());
}
if (!node->arguments.empty()) {
out_ << " : \n";
}
for (auto& argument : node->arguments) {
Visit(argument.get());
}
out_ << ")";
}
void FindSymbolsVisitor::Visit(DefinedTypeclass* node) {
out_ << "(DefinedTypeclass ";
Visit(node->typeclass.get());
if (!node->parameters.empty()) {
out_ << "\n";
}
for (auto& parameter : node->parameters) {
Visit(parameter.get());
}
if (!node->arguments.empty()) {
out_ << " : \n";
}
for (auto& argument : node->arguments) {
Visit(argument.get());
}
out_ << ")";
}
void FindSymbolsVisitor::Visit(DefinitionParameter* node) {
out_ << "(DefinitionParameter " << (!node->typeclasses.empty() > 0 ? "(" : "");
Visit(&node->type);
out_ << ' ';
for (auto& typeclass : node->typeclasses) {
Visit(typeclass.get());
}
out_ << ")";
}
void FindSymbolsVisitor::Visit(DefinitionArgument* node) {
out_ << "(DefinitionArgument " << (!node->types.empty() ? "(" : "");
Visit(&node->name);
out_ << ' ';
for (auto& type : node->types) {
Visit(type.get());
}
out_ << ")";
}
// Statements, expressions, blocks, etc. -----------------
// Name
void FindSymbolsVisitor::Visit(NameSuperExpression* node) {
out_ << "(NameSuperExpression ";
for (auto& variable_namespace : node->namespaces) {
Visitor::Visit(variable_namespace);
out_ << '.';
}
for (size_t i = 0; i < node->expressions.size(); ++i) {
Visitor::Visit(node->expressions[i]);
if (i + 1 < node->expressions.size()) {
out_ << '.';
}
}
out_ << ")";
}
void FindSymbolsVisitor::Visit(NameExpression* node) {
out_ << "(NameExpression ";
for (auto& variable_namespace : node->namespaces) {
Visitor::Visit(variable_namespace);
out_ << '.';
}
for (size_t i = 0; i < node->names.size(); ++i) {
Visit(&node->names[i]);
if (i + 1 < node->names.size()) {
out_ << '.';
}
}
out_ << ")";
}
void FindSymbolsVisitor::Visit(TupleName* node) {
out_ << "(TupleName ";
for (auto& name : node->names) {
out_ << '&';
Visit(name.get());
}
out_ << ")";
}
void FindSymbolsVisitor::Visit(VariantName* node) {
out_ << "(VariantName ";
for (auto& name : node->names) {
out_ << '|';
Visit(name.get());
}
out_ << ")";
}
void FindSymbolsVisitor::Visit(AnnotatedName* node) {
out_ << "(AnnotatedName ";
Visit(&node->name);
if (node->type.has_value()) {
out_ << " : ";
Visit(node->type.value().get());
}
out_ << " )";
}
// Type
void FindSymbolsVisitor::Visit(TypeConstructor* node) {
out_ << "(TypeConstructor ";
Visit(node->type.get());
out_ << '\n';
for (auto& parameter : node->parameters) {
Visit(&parameter.first);
out_ << " = ";
Visitor::Visit(parameter.second);
out_ << '\n';
}
out_ << ")";
}
void FindSymbolsVisitor::Visit(TupleType* node) {
out_ << "(TupleType ";
if (node->type.has_value()) {
Visit(&node->type.value());
}
out_ << ' ';
for (auto& entity : node->entities) {
out_ << "& ";
if (entity.first.has_value()) {
Visit(&entity.first.value());
out_ << " : ";
}
Visitor::Visit(entity.second);
}
out_ << ")";
}
void FindSymbolsVisitor::Visit(VariantType* node) {
out_ << "(VariantType ";
if (node->type.has_value()) {
Visit(&node->type.value());
}
out_ << ' ';
for (auto& constructor : node->constructors) {
out_ << "| ";
if (std::holds_alternative<TypeIdentifierDefinition>(constructor)) {
Visit(&std::get<TypeIdentifierDefinition>(constructor));
} else if (std::holds_alternative<std::unique_ptr<TupleType>>(constructor)) {
Visit(std::get<std::unique_ptr<TupleType>>(constructor).get());
} else {
// error
}
}
out_ << ")";
}
void FindSymbolsVisitor::Visit(AnnotatedType* node) {
out_ << "(AnnotatedType ";
Visit(node->type_expression.get());
if (!node->annotations.empty()) {
out_ << " :";
}
for (auto& annotation : node->annotations) {
out_ << " ";
Visit(annotation.get());
}
out_ << ")";
}
void FindSymbolsVisitor::Visit(ParametrizedType* node) {
out_ << "(ParametrizedType ";
Visit(node->type_expression.get());
for (auto& parameter : node->parameters) {
out_ << ' ';
Visitor::Visit(parameter);
}
out_ << " )";
}
void FindSymbolsVisitor::Visit(TypeExpression* node) {
out_ << "(TypeExpression ";
for (auto& type_namespace : node->namespaces) {
Visitor::Visit(type_namespace);
out_ << '.';
}
Visit(&node->type);
out_ << " )";
}
// Typeclass
void FindSymbolsVisitor::Visit(AnnotatedTypeclass* node) {
out_ << "(AnnotatedTypeclass ";
Visit(node->typeclass_expression.get());
if (!node->annotations.empty()) {
out_ << " :";
}
for (auto& annotation : node->annotations) {
out_ << " ";
Visit(annotation.get());
}
out_ << ")";
}
void FindSymbolsVisitor::Visit(ParametrizedTypeclass* node) {
out_ << "(ParametrizedTypeclass ";
Visit(node->typeclass_expression.get());
for (auto& paramater : node->parameters) {
out_ << ' ';
Visitor::Visit(paramater);
}
out_ << " )";
}
void FindSymbolsVisitor::Visit(TypeclassExpression* node) {
out_ << "(TypeclassExpression ";
for (auto& typeclass_namespace : node->namespaces) {
Visitor::Visit(typeclass_namespace);
out_ << '.';
}
Visit(&node->typeclass);
out_ << " )";
}
} // namespace interpreter