#pragma once

#include "ExpressionVisitor.h"

#include <iosfwd>
class ASTPrinter : public ExpressionVisitor
{
public:
	ASTPrinter(std::ostream& os);
	~ASTPrinter() {}
	virtual void Visit(LiteralNull* e) override;
	virtual void Visit(LiteralUndefined* e) override;
	virtual void Visit(LiteralString* e) override;
	virtual void Visit(LiteralNumber* e) override;
	virtual void Visit(LiteralBoolean* e) override;
	virtual void Visit(LiteralObject* e) override;
	virtual void Visit(BinaryExpression* e) override;
	virtual void Visit(UnaryExpression* e) override;
	virtual void Visit(IdentifierExpression* e) override;
	virtual void Visit(ListExpression* e) override;
	virtual void Visit(VariableDefinitionExpression* e) override;
	virtual void Visit(BlockStatement* e) override;
	virtual void Visit(LabeledStatement* e) override;
	virtual void Visit(IfStatement* e) override;
	virtual void Visit(SwitchStatement* e) override;
	virtual void Visit(CaseStatement *) override;
	virtual void Visit(WhileStatement* e) override;
	virtual void Visit(ForStatement* e) override;
	virtual void Visit(FunctionDeclaration* e) override;
	virtual void Visit(TopStatements* e) override;
	virtual void Visit(EmptyExpression* e) override;
	virtual void Visit(CallExpression* e) override;

	inline void Enter();
	inline void Exit();
	inline void NoIndentNextAccept();
	inline void IndentNextAccept();
	inline void InsertIndent();
private:
	std::ostream& os;
	unsigned nest_level;
	bool indent_next_accept;
};