File Coverage

blib/lib/MouseX/AttributeTraitHelper/Merge.pm

Criterion	Covered	Total	%
statement	9	9	100.0
branch			n/a
condition			n/a
subroutine	3	3	100.0
pod			n/a
total	12	12	100.0

package MouseX::AttributeTraitHelper::Merge;
use Mouse::Role;
use Mouse::Util;

has TRAIT_MAPPING => (
    is => 'ro',
    isa => 'HashRef[ClassName]',
    default => sub {return {}},
);

around add_attribute => sub {
    my ($orig, $self) = (shift, shift);

    return $self->$orig($_[0]) if Scalar::Util::blessed($_[0]);

    my $name = shift;
    my %args = (@_ == 1) ? %{$_[0]} : @_;

    defined($name)
        or $self->throw_error('You must provide a name for the attribute');

    my $traits = delete $args{traits};
    if ($traits) {
        my $role_name;
        if (@$traits == 1) {
            $role_name = $traits->[0];
        }
        else {
            $role_name = join "::" , 'MouseX::AttributeTraitHelper::Merge' , @$traits;
            if (!Mouse::Util::is_class_loaded($role_name)) {
                my $meta = Mouse::Role->init_meta(for_class => $role_name);
                $meta->add_around_method_modifier('does' => sub {
                    my ($orig_meta, $self_meta, $role) = @_;
                    if ($self->TRAIT_MAPPING->{$role}){
                        return 1;
                    }
                    else {
                        return $self->$orig($name)
                    }
                });
                for my $trait (@$traits) {
                    $self->TRAIT_MAPPING->{$trait} = $role_name;
                    Mouse::Util::load_class($trait);
                    for my $trait_attr_name ($trait->meta->get_attribute_list()) {
                        my $trait_attr = $trait->meta->get_attribute($trait_attr_name);
                        $trait_attr_name =~ s/^\+//;
                        my $exist_trait_attr = $meta->get_attribute($trait_attr_name);
                        if ($exist_trait_attr) {
                            @$exist_trait_attr{keys %$trait_attr} = values %$trait_attr;
                        }
                        else {
                            $meta->add_attribute($trait_attr_name => {is => 'ro', %$trait_attr});
                        }
                    }
                }
            }
        }
        $args{traits} = [$role_name];
    }
    return $self->$orig($name, %args);
};

no Mouse::Role;
1;
__END__