tests/org.eclipse.dltk.ruby.core.tests/src/org/eclipse/dltk/ruby/core/tests/parser/jruby/AST2StringVisitor.java

/*******************************************************************************
 * Copyright (c) 2005, 2016 IBM Corporation and others.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0 which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 * 
 * SPDX-License-Identifier: EPL-2.0
 *
 *******************************************************************************/
package org.eclipse.dltk.ruby.core.tests.parser.jruby;

import java.io.StringWriter;

import org.eclipse.dltk.ast.ASTNode;
import org.eclipse.dltk.ast.ASTVisitor;
import org.eclipse.dltk.utils.CorePrinter;

public class AST2StringVisitor extends ASTVisitor {
	
	private StringWriter writer;
	private CorePrinter printer;
	private String indent;

	public AST2StringVisitor() {
		super();
		writer = new StringWriter();
		printer = new CorePrinter(writer);
		indent = "";
	}

	@Override
	public boolean visitGeneral(ASTNode node) throws Exception {
		String str = node.debugString();
		this.printer.println(indent + "+" + str);
		indent += "  ";
		return super.visitGeneral(node);
	}
	
	private static String simplifyClassName(String name) {
		int pos = name.lastIndexOf('.');
		return name.substring(pos + 1);
	}	
			
	@Override
	public void endvisitGeneral(ASTNode node) throws Exception {
		indent = indent.substring(0, indent.length() - 2);
		this.printer.println(indent + "-" + simplifyClassName(node.getClass().getName()));			
		super.endvisitGeneral(node);
	}

	public String getResult() {
		printer.flush();
		printer.close();
		return writer.getBuffer().toString();
	}
	
}