AST2StringVisitor.java

/*******************************************************************************
 * Copyright (c) 2005, 2016 IBM Corporation and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 *******************************************************************************/
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();
	}
	
}