change usage of Tie::IxHash

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

diff --git a/lib/SQL/Translator/Object/Trigger.pm b/lib/SQL/Translator/Object/Trigger.pm
index 2f8f442..f0f0730 100644 (file)
--- a/lib/SQL/Translator/Object/Trigger.pm
+++ b/lib/SQL/Translator/Object/Trigger.pm
@@ -1,10 +1,22 @@
 use MooseX::Declare;
-class SQL::Translator::Object::Trigger {
-    use MooseX::Types::Moose qw(ArrayRef HashRef Str);
-    use MooseX::AttributeHelpers;
-    use SQL::Translator::Types qw(Column);
-    extends 'SQL::Translator::Object';
-    
+class SQL::Translator::Object::Trigger extends SQL::Translator::Object {
+    use MooseX::Types::Moose qw(Any ArrayRef HashRef Int Str);
+    use MooseX::MultiMethods;
+    use SQL::Translator::Types qw(Column Schema Table ColumnHash IxHash);
+    use Tie::IxHash;
+
+    has 'schema' => (
+        is => 'rw',
+        isa => Schema,
+        weak_ref => 1,
+    );
+
+    has 'table' => (
+        is => 'rw',
+        isa => Table,
+        weak_ref => 1,
+    );
+
     has 'name' => (
         is => 'ro',
         isa => Str,
@@ -12,35 +24,31 @@ class SQL::Translator::Object::Trigger {
     );
 
     has 'columns' => (
-        metaclass => 'Collection::Hash',
         is => 'rw',
-        isa => HashRef[Column],
-        provides => {
-            exists => 'exists_column',
-            keys   => 'column_ids',
-            values => 'get_columns',
-            get    => 'get_column',
-        },
-        curries => {
-            set => {
-                add_column => sub {
-                    my ($self, $body, $column) = @_;
-                    $self->$body($column->name, $column);
-                }
-            }
+        isa => IxHash, #ColumnHash,
+        handles => {
+            exists_column => 'EXISTS',
+            column_ids    => 'Keys',
+            get_columns   => 'Values',
+            get_column    => 'FETCH',
+            add_column    => 'STORE',
+            remove_column => 'DELETE',
+            clear_columns => 'CLEAR',
         },
-        default => sub { my %hash = (); tie %hash, 'Tie::IxHash'; return \%hash },
+        coerce => 1,
+        default => sub { Tie::IxHash->new() }
     );
 
     has 'on_table' => (
         is => 'rw', 
         isa => Str,
-        required => 1
+        required => 1,
+#        trigger => sub { my ($self, $new, $old) = @_; $self->table($self->schema->get_table($new)) },
     );
 
     has 'action' => (
         is => 'rw',
-        isa => Str
+        isa => Any
     );
 
     has 'perform_action_when' => (
@@ -49,12 +57,48 @@ class SQL::Translator::Object::Trigger {
         required => 1
     );
 
-    has 'database_events' => (
-        is => 'rw',
+    has '_database_events' => (
+        traits => ['Array'],
         isa => ArrayRef,
-        required => 1
+        handles => {
+            _database_events            => 'elements',
+            add_database_event          => 'push',
+            remove_last_database_option => 'pop',
+        },
+        default => sub { [] },
+        required => 1,
     );
 
-    method get_fields { return $self->get_columns }
-    method fields { return $self->column_ids }
+    has '_order' => (
+        isa => Int,
+        is => 'rw',
+    );
+
+    around add_column(Column $column) { $self->$orig($column->name, $column) }
+
+    multi method database_events(Str $database_event) { $self->add_database_event($database_event); $self->database_events }
+    multi method database_events(ArrayRef $database_events) { $self->add_database_event($_) for @$database_events; $self->database_events }
+    multi method database_events { $self->_database_events }
+
+    multi method order(Int $order) { $self->_order($order); }
+    multi method order {
+        my $order = $self->_order;
+        unless (defined $order && $order) {
+            my $triggers = Tie::IxHash->new( map { $_->name => $_ } $self->schema->get_triggers );
+            $order = $triggers->Indices($self->name) || 0; $order++;
+            $self->_order($order);
+        }
+        return $order;
+    }
+
+    method is_valid { 1 }
+
+    around BUILDARGS(ClassName $self: @args) {
+        my $args = $self->$orig(@args);
+
+        my $database_events = delete $args->{database_events};
+        $args->{_database_events} = ref $database_events ? $database_events : [ $database_events ];
+
+        return $args;
+     }
 }