Printer: Return string instead of printing directly

2023-03-25 20:01:30 +01:00
parent cc735a7a11
commit d772495f1a
4 changed files with 83 additions and 75 deletions
@@ -8,6 +8,7 @@
 #include <string>

 #include "ast.h"
+#include "printer.h"
 #include "types.h"

 namespace blaze {
@@ -79,21 +80,6 @@ Function::Function(Lambda lambda)

 void Formatter<blaze::ASTNode*>::format(Builder& builder, blaze::ASTNode* value) const
 {
-	// TODO: Call into Printer::dumpImp(), instead of doing it manually
-	if (is<blaze::String>(value)) {
-		return Formatter<std::string>::format(builder, static_cast<blaze::String*>(value)->data());
-	}
-	if (is<blaze::Keyword>(value)) {
-		return Formatter<std::string>::format(builder, ":" + static_cast<blaze::Keyword*>(value)->keyword().substr(1));
-	}
-	else if (is<blaze::Number>(value)) {
-		Formatter<int64_t> formatter { .specifier = specifier };
-		return formatter.format(builder, static_cast<blaze::Number*>(value)->number());
-	}
-	else if (is<blaze::Value>(value)) {
-		return Formatter<std::string>::format(builder, static_cast<blaze::Value*>(value)->value());
-	}
-	else if (is<blaze::Symbol>(value)) {
-		return Formatter<std::string>::format(builder, static_cast<blaze::Symbol*>(value)->symbol());
-	}
+	blaze::Printer printer;
+	return Formatter<std::string>::format(builder, printer.printNoErrorCheck(value));
 }
@@ -5,9 +5,11 @@
 */

 #include <iterator> // std::next
+#include <string>

-#include "ruc/format/print.h"
+#include "ruc/format/format.h"

+#include "ast.h"
 #include "error.h"
 #include "lexer.h"
 #include "printer.h"
@@ -15,113 +17,128 @@

 namespace blaze {

-Printer::Printer(ASTNode* node)
-	: m_node(node)
+Printer::Printer()
 {
 }

 Printer::~Printer()
 {
-	delete m_node;
 }

 // -----------------------------------------

-void Printer::dump()
+std::string Printer::print(ASTNode* node)
 {
 	if (Error::the().hasAnyError()) {
-		dumpError();
-		return;
+		init();
+		printError();
+		return m_print;
 	}

-	if (m_node == nullptr) {
-		return;
-	}
-
-	dumpImpl(m_node);
-	print("\n");
+	return printNoErrorCheck(node);
 }

-void Printer::dumpImpl(ASTNode* node)
+std::string Printer::printNoErrorCheck(ASTNode* node)
+{
+	init();
+
+	if (node == nullptr) {
+		return {};
+	}
+
+	printImpl(node);
+
+	return m_print;
+}
+
+// -----------------------------------------
+
+void Printer::init()
+{
+	m_first_node = true;
+	m_previous_node_is_list = false;
+	m_print = "";
+}
+
+void Printer::printImpl(ASTNode* node)
 {
 	auto printSpacing = [this]() -> void {
-		if (!m_firstNode && !m_previousNodeIsList) {
-			print(" ");
+		if (!m_first_node && !m_previous_node_is_list) {
+			m_print += ' ';
 		}
 	};

 	if (is<List>(node)) {
 		printSpacing();
-		print("(");
-		m_firstNode = false;
-		m_previousNodeIsList = true;
+		m_print += '(';
+		m_first_node = false;
+		m_previous_node_is_list = true;
 		auto nodes = static_cast<List*>(node)->nodes();
 		for (size_t i = 0; i < nodes.size(); ++i) {
-			dumpImpl(nodes[i]);
-			m_previousNodeIsList = false;
+			printImpl(nodes[i]);
+			m_previous_node_is_list = false;
 		}
-		print(")");
+		m_print += ')';
 	}
 	else if (is<Vector>(node)) {
 		printSpacing();
-		print("[");
-		m_firstNode = false;
-		m_previousNodeIsList = true;
+		m_print += '[';
+		m_first_node = false;
+		m_previous_node_is_list = true;
 		auto nodes = static_cast<Vector*>(node)->nodes();
 		for (size_t i = 0; i < nodes.size(); ++i) {
-			dumpImpl(nodes[i]);
-			m_previousNodeIsList = false;
+			printImpl(nodes[i]);
+			m_previous_node_is_list = false;
 		}
-		print("]");
+		m_print += ']';
 	}
 	else if (is<HashMap>(node)) {
 		printSpacing();
-		print("{{");
-		m_firstNode = false;
-		m_previousNodeIsList = true;
+		m_print += "{";
+		m_first_node = false;
+		m_previous_node_is_list = true;
 		auto elements = static_cast<HashMap*>(node)->elements();
 		for (auto it = elements.begin(); it != elements.end(); ++it) {
-			print("{} ", it->first.front() == 0x7f ? ":" + it->first.substr(1) : it->first); // 127
-			dumpImpl(it->second);
+			m_print += format("{} ", it->first.front() == 0x7f ? ":" + it->first.substr(1) : it->first); // 127
+			printImpl(it->second);

 			if (it != elements.end() && std::next(it) != elements.end()) {
-				print(" ");
+				m_print += ' ';
 			}
 		}
-		m_previousNodeIsList = false;
-		print("}}");
+		m_previous_node_is_list = false;
+		m_print += '}';
 	}
 	else if (is<String>(node)) {
 		// TODO: Implement string readably printing
 		printSpacing();
-		print("{}", static_cast<String*>(node)->data());
+		m_print += format("{}", static_cast<String*>(node)->data());
 	}
 	else if (is<Keyword>(node)) {
 		printSpacing();
-		print(":{}", static_cast<Keyword*>(node)->keyword().substr(1));
+		m_print += format(":{}", static_cast<Keyword*>(node)->keyword().substr(1));
 	}
 	else if (is<Number>(node)) {
 		printSpacing();
-		print("{}", static_cast<Number*>(node)->number());
+		m_print += format("{}", static_cast<Number*>(node)->number());
 	}
 	else if (is<Symbol>(node)) {
 		printSpacing();
-		print("{}", static_cast<Symbol*>(node)->symbol());
+		m_print += format("{}", static_cast<Symbol*>(node)->symbol());
 	}
 }

-void Printer::dumpError()
+void Printer::printError()
 {
-	print("Error: ");
+	m_print = "Error: ";
 	if (Error::the().hasTokenError()) {
 		Token error = Error::the().tokenError();
-		print("{}", error.symbol);
+		m_print += format("{}", error.symbol);
 	}
 	else if (Error::the().hasOtherError()) {
 		std::string error = Error::the().otherError();
-		print("{}", error);
+		m_print += format("{}", error);
 	}
-	print("\n");
 }

 } // namespace blaze
@@ -7,24 +7,27 @@
 #pragma once

 #include "ast.h"
+#include <string>

 namespace blaze {

 // Serializer -> return to string
 class Printer {
 public:
-	Printer(ASTNode* node);
+	Printer();
 	virtual ~Printer();

-	void dump();
+	std::string print(ASTNode* node);
+	std::string printNoErrorCheck(ASTNode* node);

 private:
-	void dumpImpl(ASTNode* node);
-	void dumpError();
+	void init();
+	void printImpl(ASTNode* node);
+	void printError();

-	bool m_firstNode { true };
-	bool m_previousNodeIsList { false };
-	ASTNode* m_node { nullptr };
+	bool m_first_node { true };
+	bool m_previous_node_is_list { false };
+	std::string m_print;
 };

 } // namespace blaze
@@ -38,21 +38,23 @@ auto eval(blaze::ASTNode* ast) -> blaze::ASTNode*
 	blaze::GlobalEnvironment env;
 	blaze::Eval eval(ast, &env);
 	eval.eval();
+
 	return eval.ast();
 }

-auto print(blaze::ASTNode* exp) -> void
+auto print(blaze::ASTNode* exp) -> std::string
 {
-	blaze::Printer printer(exp);
-	printer.dump();
+	blaze::Printer printer;
+
+	return printer.print(exp);
 }

-auto rep(std::string_view input) -> void
+auto rep(std::string_view input) -> std::string
 {
 	blaze::Error::the().clearErrors();
 	blaze::Error::the().setInput(input);

-	print(eval(read(input)));
+	return print(eval(read(input)));
 }

 static auto cleanup(int signal) -> void
@@ -92,7 +94,7 @@ auto main(int argc, char* argv[]) -> int
 		if (pretty_print) {
 			print("\033[0m");
 		}
-		rep(input);
+		print("{}\n", rep(input));
 	}

 	if (pretty_print) {