role sugar, which sets role_generator

[gitmo/MooseX-Role-Parameterized.git] / lib / MooseX / Role / Parameterized.pm

diff --git a/lib/MooseX/Role/Parameterized.pm b/lib/MooseX/Role/Parameterized.pm
index 2bc271f..d2e40ad 100644 (file)
--- a/lib/MooseX/Role/Parameterized.pm
+++ b/lib/MooseX/Role/Parameterized.pm
@@ -1,14 +1,13 @@
 #!/usr/bin/env perl
 package MooseX::Role::Parameterized;
-use strict;
-use warnings;
-use MooseX::Role::Parameterized::Meta::Role;
-
+use Moose;
 use Moose::Role ();
-use Moose::Exporter;
+extends 'Moose::Exporter';
 
-Moose::Exporter->setup_import_methods(
-    with_caller => ['parameter'],
+use MooseX::Role::Parameterized::Meta::Role;
+
+__PACKAGE__->setup_import_methods(
+    with_caller => ['parameter', 'role'],
 );
 
 sub parameter {
@@ -16,6 +15,12 @@ sub parameter {
     $caller->meta->add_parameter(@_);
 }
 
+sub role {
+    my $caller         = shift;
+    my $role_generator = shift;
+    $caller->meta->role_generator($role_generator);
+}
+
 sub init_meta {
     my $self = shift;
 
@@ -24,5 +29,17 @@ sub init_meta {
     );
 }
 
+# give role a (&) prototype
+around _make_wrapper => sub {
+    my $orig = shift;
+    my ($self, $caller, $sub, $fq_name) = @_;
+
+    if ($fq_name =~ /::role$/) {
+        return sub (&) { $sub->($caller, @_) };
+    }
+
+    return $orig->(@_);
+};
+
 1;