package horstmann.ch10_visitor;
import java.io.File;

public class DirectoryNode implements FileSystemNode
{
  public DirectoryNode(File directory)
  {
    this.directory = directory;
  }

  public void accept(FileSystemVisitor v)
  {
    v.visitDirectoryNode(this);
  }

  public File getDirectory() { return directory; }

  public FileSystemNode[] getChildren()
  {
    File[] files = directory.listFiles();
    FileSystemNode[] children = new FileSystemNode[files.length];
    for (int i = 0; i < files.length; i++)
    {
      File f = files[i];
      if (f.isDirectory())
        children[i] = new DirectoryNode(f);
      else
        children[i] = new FileNode(f);
    }
    return children;
  }

  private File directory;
}