File Coverage

blib/lib/Tree/Simple/Manager/Index.pm

Criterion	Covered	Total	%
statement	38	38	100.0
branch	10	10	100.0
condition	4	6	66.6
subroutine	12	12	100.0
pod	6	6	100.0
total	70	72	97.2

package Tree::Simple::Manager::Index;

use strict;
use warnings;

use Scalar::Util qw(blessed);

our $VERSION = '0.04';

use Tree::Simple::Manager::Exceptions;

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

sub _init {
    my ($self, $tree) = @_;
    (blessed($tree) && $tree->isa("Tree::Simple"))
        || throw Tree::Simple::Manager::InsufficientArguments;
    # add our root
    $self->{root_tree} = $tree;
    $self->{index}     = {};
    # then add all its children on down
    $self->indexTree();
}

sub indexTree {
    my ($self) = @_;
    $self->{root_tree}->traverse(sub {
        my ($tree) = @_;
        (!exists $self->{index}->{$tree->getUID()})
            || throw Tree::Simple::Manager::IllegalOperation "tree (" . $tree->getUID() . ") already exists in the index, cannot add a duplicate";
        $self->{index}->{$tree->getUID()} = $tree;
    });
}

sub getRootTree { (shift)->{root_tree} }

sub getIndexKeys {
    my ($self) = @_;
    my @keys = keys %{$self->{index}};
    return wantarray ? @keys : \@keys;
}

sub getTreeByID {
    my ($self, $id) = @_;
    (exists $self->{index}->{$id})
        || throw Tree::Simple::Manager::KeyDoesNotExist "tree ($id) does not exist in the index";
    return $self->{index}->{$id};
}

sub hasTreeAtID {
    my ($self, $id) = @_;
    exists $self->{index}->{$id} ? 1 : 0
}

1;

__END__