File Coverage

blib/lib/File/Iterator.pm

Criterion	Covered	Total	%
statement	12	51	23.5
branch	0	22	0.0
condition	0	18	0.0
subroutine	4	9	44.4
pod	3	3	100.0
total	19	103	18.4

package File::Iterator;

use 5.005;
use strict;
use Carp;
use vars qw($VERSION);

$VERSION = '0.14';

sub new {
	my $proto = shift;
	my $class = ref($proto) || $proto;
	my $self = {
		DIR				=> '.',
		RECURSE			=> 1,
		FILTER			=> undef,
		RETURNDIRS		=> 0,
		FOLLOWSYMLINKS	=> 0,
		@_
	};

	$self->{FILES} = [];
	bless ($self, $class);

	# remove trailing slash unless user has supplied the root directory
	if ( ! _isRootDir( $self->{DIR} ) ) {
		if ( $^O eq "MSWin32" || $^O eq "os2") { # trailing slash can be either / or \
			$self->{DIR} =~ s|[\\/]$||;
		}
		elsif ( $^O eq "NetWare" ) { # uses \ as directory separator
			$self->{DIR} =~ s|\\$||;
		}
		else {
			$self->{DIR} =~ s|/$||;
		}
	}

	$self->_probeDir( $self->{DIR} );
	return $self;
}

sub _isRootDir {
	$_[0] =~ m{^(([a-z]:)?[\\/]|\\\\)$}i; # true if arg is /, \, X:\, X:/ or \\
}

sub _probeDir {
	my $self = shift;
	my $dir = shift;
	my $slash = _isRootDir($dir) ? "" : ( $^O eq "MSWin32" || $^O eq "NetWare" || $^O eq "os2" ) ? "\\" : "/";

	if (opendir DIR, $dir) {
		my @files = grep { !/^\.{1,2}$/ } readdir DIR; # ignore . and ..
		unshift @{$self->{FILES}}, map $dir.$slash.$_, sort { lc $a cmp lc $b } @files;
		closedir DIR;
	}
	else {
		carp "Can't open $dir: $!";
	}
}

sub next {
	my $self = shift;
	my $nextfile = shift @{$self->{FILES}} or return undef;
	if (-d $nextfile) {
		# if we are recursing and either the directory is not a symlink or we're following symlinks...
		if ( $self->{RECURSE} && (!-l $nextfile || $self->{FOLLOWSYMLINKS} ) ) {
			$self->_probeDir($nextfile);
		}

		if (!$self->{RETURNDIRS}) {
			return $self->next();
		}
	}

	my $filter = $self->{FILTER};
	if ( $filter && !($filter->($nextfile)) ) {
		return $self->next();
	}
	else {
		return $nextfile;
	}
}

sub reset {
	my $self = shift;
	$self->{FILES} = [];
	$self->_probeDir( $self->{DIR} );
}

1;

__END__