Get attributes in role application working!
Shawn M Moore [Tue, 17 Jun 2008 04:00:36 +0000 (04:00 +0000)]
lib/Mouse/Meta/Role.pm

index 8e42641..3c7aff1 100644 (file)
@@ -27,24 +27,31 @@ sub new {
     my $class = shift;
     my %args  = @_;
 
+    $args{attributes} ||= {};
+
     bless \%args, $class;
 }
 
 sub name { $_[0]->{name} }
 
-sub has_attribute { exists $_[0]->{attributes}->{$_[1]}  }
-sub add_attribute { $_[0]->{attributes}->{$_[1]} = $_[2] }
-sub get_attribute_list {
+sub add_attribute {
+    my $self = shift;
+    my $name = shift;
+    $self->{attributes}->{$name} = [ @_ ];
 }
 
+sub has_attribute { exists $_[0]->{attributes}->{$_[1]}  }
+sub get_attribute_list { keys %{ $_[0]->{attributes} } }
+sub get_attribute { @{ $_->[0]->{attributes}->{$_[1]} || [] } }
+
 sub apply {
     my $self  = shift;
     my $class = shift;
-    my $pkg   = shift;
+    my $pkg   = $class->name;
 
     for my $name ($self->get_attribute_list) {
-        my $attr = $self->get_attribute($name);
-        Mouse::Meta::Attribute->create($pkg, $name, $attr);
+        my @spec = $self->get_attribute($name);
+        Mouse::Meta::Attribute->create($pkg, $name, @spec);
     }
 
 }