File Coverage

blib/lib/Tree/Simple/Visitor/CreateDirectoryTree.pm

Criterion	Covered	Total	%
statement	49	49	100.0
branch	12	14	85.7
condition	13	15	86.6
subroutine	13	13	100.0
pod	4	4	100.0
total	91	95	95.7

package Tree::Simple::Visitor::CreateDirectoryTree;

use strict;
use warnings;

our $VERSION = '0.15';

use File::Spec;
use Scalar::Util qw(blessed);

use base qw(Tree::Simple::Visitor);

sub new {
    my ($_class) = @_;
    my $class = ref($_class) || $_class;
    my $visitor = {};
    bless($visitor, $class);
    $visitor->_init();
    return $visitor;
}

sub _init {
    my ($self) = @_;
    $self->{file_handler} = sub {
        my ($filepath) = @_;
        open(FILE, ">", $filepath) || die "IO Error : Cannot create file ($filepath) : $!";
        close(FILE);
    };
    $self->{dir_handler} = sub {
        my ($dirpath) = @_;
        mkdir($dirpath) || die "IO Error : Cannot make directory ($dirpath) : $!";
    };
    $self->SUPER::_init();
}

sub visit {
	my ($self, $tree) = @_;
	(blessed($tree) && $tree->isa("Tree::Simple"))
		|| die "Insufficient Arguments : You must supply a valid Tree::Simple object";
    # pass on to our recursive subroutine
    $self->_createDirectoryStructure($tree);
}

sub setFileHandler {
    my ($self, $file_handler) = @_;
    (defined($file_handler) && ref($file_handler) eq 'CODE')
        || die "Insufficient Arguments : file handler must be a subroutine reference";
    $self->{file_handler} = $file_handler;
}

sub setDirectoryHandler {
    my ($self, $dir_handler) = @_;
    (defined($dir_handler) && ref($dir_handler) eq 'CODE')
        || die "Insufficient Arguments : directory handler must be a subroutine reference";
    $self->{dir_handler} = $dir_handler;
}

sub _createDirectoryStructure {
    my ($self, $tree, @path) = @_;
    my $node = $tree->getNodeValue();
    # filter the nodes if need be
    my $filter_function = $self->getNodeFilter();
    $node = $filter_function->($node) if $filter_function;
    # if its a leaf and it
    # doesn't end with a /
    # then its a file
    if ($tree->isLeaf() && $node !~ /\/|\\$/) {
        $self->{file_handler}->(File::Spec->catfile(@path, $node));
    }
    # otherwise we are going
    # to treat it as a directory
    else {
        $node =~ s/\/|\\$//;
        $self->{dir_handler}->(File::Spec->catdir(@path, $node));
        foreach my $child ($tree->getAllChildren()) {
            $self->_createDirectoryStructure($child, (@path, $node));
        }
    }
}

1;

__END__