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;
            $self->TRAIT_MAPPING->{$_} = $role_name for @$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_meta->$orig_meta($role)
                    }
                });
                my @trait_clones_meta = ();
                for my $trait (@$traits) {
                    Mouse::Util::load_class($trait);
                    my $trait_clone = join "::" , 'MouseX::AttributeTraitHelper::Merge::CLONE', $trait;
                    if(!Mouse::Util::is_class_loaded($trait_clone)) {
                        my $trait_clone_meta = Mouse::Role->init_meta(for_class => $trait_clone);
                        $trait->meta->apply($trait_clone_meta);
                        for my $trait_attr_name ($trait->meta->get_attribute_list()) {
                            $trait_clone_meta->remove_attribute($trait_attr_name);
                        }
                    }
                    push @trait_clones_meta, $trait_clone->meta;
                }
                Mouse::Util::apply_all_roles($meta, @trait_clones_meta);
                for my $trait (@$traits) {
                    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, @_ == 1 ? $args : %$args);
};

no Mouse::Role;
1;
__END__