Add support for adding method modifiers to a role metaclass

[gitmo/Mouse.git] / lib / Mouse / Meta / Role.pm

diff --git a/lib/Mouse/Meta/Role.pm b/lib/Mouse/Meta/Role.pm
index 92b39c1..79f3b7f 100644 (file)
--- a/lib/Mouse/Meta/Role.pm
+++ b/lib/Mouse/Meta/Role.pm
@@ -27,13 +27,49 @@ sub new {
     my $class = shift;
     my %args  = @_;
 
+    $args{attributes} ||= {};
+
     bless \%args, $class;
 }
 
 sub name { $_[0]->{name} }
 
+sub add_attribute {
+    my $self = shift;
+    my $name = shift;
+    my $spec = shift;
+    $self->{attributes}->{$name} = $spec;
+}
+
 sub has_attribute { exists $_[0]->{attributes}->{$_[1]}  }
-sub add_attribute { $_[0]->{attributes}->{$_[1]} = $_[2] }
+sub get_attribute_list { keys %{ $_[0]->{attributes} } }
+sub get_attribute { $_[0]->{attributes}->{$_[1]} }
+
+sub apply {
+    my $self  = shift;
+    my $class = shift;
+
+    for my $name ($self->get_attribute_list) {
+        next if $class->has_attribute($name);
+        my $spec = $self->get_attribute($name);
+        Mouse::Meta::Attribute->create($class, $name, %$spec);
+    }
+}
+
+for my $modifier_type (qw/before after around/) {
+    no strict 'refs';
+    *{ __PACKAGE__ . '::' . "add_${modifier_type}_method_modifier" } = sub {
+        my ($self, $method_name, $method) = @_;
+
+        push @{ $self->{"${modifier_type}_method_modifiers"}->{$method_name} },
+            $method;
+    };
+
+    *{ __PACKAGE__ . '::' . "get_${modifier_type}_method_modifiers" } = sub {
+        my ($self, $method_name, $method) = @_;
+        @{ $self->{"${modifier_type}_method_modifiers"}->{$method_name} || [] }
+    };
+}
 
 1;