fix after removing auto_deref

[dbsrgits/SQL-Translator-2.0-ish.git] / lib / SQL / Translator / Object.pm

diff --git a/lib/SQL/Translator/Object.pm b/lib/SQL/Translator/Object.pm
index af3ad53..e1ba3ac 100644 (file)
--- a/lib/SQL/Translator/Object.pm
+++ b/lib/SQL/Translator/Object.pm
@@ -2,37 +2,53 @@ use MooseX::Declare;
 class SQL::Translator::Object {
     use Tie::IxHash;
     use MooseX::MultiMethods;
-    use MooseX::Types::Moose qw(Any ArrayRef Str);
+    use MooseX::Types::Moose qw(Any ArrayRef HashRef Str);
 
     has '_comments' => (
-        metaclass => 'Collection::Array',
-        is => 'rw',
+        traits => ['Array'],
         isa => ArrayRef,
-        provides => {
-            push => 'add_comment',
-            pop  => 'remove_last_comment',
+        handles => {
+            _comments           => 'elements',
+            add_comment         => 'push',
+            remove_last_comment => 'pop',
         },
         default => sub { [] },
-        auto_deref => 1,
     );
 
     has '_options' => (
-        metaclass => 'Collection::Array',
-        is => 'rw',
+        traits => ['Array'],
         isa => ArrayRef,
-        provides => {
-            push => 'add_option',
-            pop  => 'remove_last_option',
+        handles => {
+            _options           => 'elements',
+            add_option         => 'push',
+            remove_last_option => 'pop',
         },
         default => sub { [] },
-        auto_deref => 1,
     );
 
-    multi method comments(Str $comment) { $self->add_comment($comment) }
-    multi method comments(ArrayRef $comment) { $self->add_comment($comment) }
-    multi method comments(Any $) { return wantarray ? @{$self->_comments} : join "\n", $self->_comments }
+    has '_extra' => (
+        traits => ['Hash'],
+        is => 'rw',
+        isa => HashRef,
+        handles => {
+            exists_extra => 'exists',
+            extra_ids    => 'keys',
+            get_extras   => 'values',
+            get_extra    => 'get',
+            add_extra    => 'set',
+        },
+        default => sub { {} },
+    );
+
+    multi method comments(Str $comment) { $self->add_comment($comment); $self->comments }
+    multi method comments(ArrayRef $comments) { $self->add_comment($_) for @$comments; $self->comments }
+    multi method comments(Any $) { wantarray ? $self->_comments : join "\n", $self->_comments }
+
+    multi method options(Str $option) { $self->add_option($option); $self->options }
+    multi method options(ArrayRef $options) { $self->add_option($_) for @$options; $self->options }
+    multi method options(Any $) { wantarray ? $self->_options : $self->_options }
 
-    multi method options(Str $option) { $self->add_option($option) }
-    multi method options(ArrayRef $option) { $self->add_option($option) if scalar @$option }
-    multi method options(Any $) { wantarray ? @{$self->_options} : $self->_options }
+    multi method extra(Str $extra) { $self->get_extra($extra) }
+    multi method extra(HashRef $extra) { $self->add_extra($_, $extra->{$_}) for keys %$extra; $self->extra }
+    multi method extra(Any $) { wantarray ? %{$self->_extra} : $self->_extra }
 }