add attributes to edge

1 files changed, 32 insertions, 7 deletions

diff --git a/lib/Graph/GEXF/Role/Attributes.pm b/lib/Graph/GEXF/Role/Attributes.pm
index 644793d..e03f5b1 100644
--- a/lib/Graph/GEXF/Role/Attributes.pm
+++ b/lib/Graph/GEXF/Role/Attributes.pm
@@ -7,12 +7,22 @@ parameter for => (
     required => 1,
 );
 
+parameter with_method => (
+    is      => 'ro',
+    default => 0,
+);
+
 role {
     my $p = shift;
 
     foreach my $type (@{$p->for}) {
 
-        my $attr_name = $type . '_attributes';
+        my $attr_name  = $type . '_attributes';
+        my $total_attr = 'attributes_' . $type . '_total';
+        my $set_attr   = 'set_' . $type . '_attribute';
+        my $get_attr   = 'get_' . $type . '_attribute';
+        my $list_attr  = 'attributes_' . $type . '_list';
+        my $has_attr   = 'has_' . $type . '_attribute';
 
         has $attr_name => (
             traits  => ['Hash'],
@@ -21,15 +31,30 @@ role {
             lazy    => 1,
             default => sub { {} },
             handles => {
-                'attributes_' . $type . '_total' => 'count',
-                'set_' . $type . '_attribute'    => 'set',
-                'get_' . $type . '_attribute'    => 'get',
-                'attributes_' . $type . '_list'  => 'keys',
-                'has_'.$type.'_attribute' => 'exists',
+                $total_attr => 'count',
+                $set_attr   => 'set',
+                $get_attr   => 'get',
+                $list_attr  => 'keys',
+                $has_attr   => 'exists',
             }
         );
-    }
 
+        if ($p->with_method) {
+            my $method_name = 'add_' . $type . '_attribute';
+
+            method $method_name => sub {
+                my ($self, $name, $type, $default_value) = @_;
+                my $id = $self->$total_attr();
+                my $attr = {
+                    id      => $id,
+                    title   => $name,
+                    type    => $type,
+                    default => [$default_value],
+                };
+                $self->$set_attr($name => $attr);
+            };
+        }
+    }
 };
 
 1;