package horstmann.ch10_visitor;

public class PrintVisitor implements FileSystemVisitor
{
	public void visitFileNode(FileNode node)
	{
		for (int i = 0; i < level; i++) System.out.print(" ");
		System.out.println(node.getFile().getName());
	}

	public void visitDirectoryNode(DirectoryNode node)
	{
		for (int i = 0; i < level; i++) System.out.print(" ");
		System.out.println(node.getDirectory().getName());
		level++;
		for (FileSystemNode c : node.getChildren())
			c.accept(this);
		level--;
	}

	private int level = 0;
}