Add trigger support to PostgreSQL producer and parser (including trigger scope)
[dbsrgits/SQL-Translator.git] / lib / SQL / Translator / Schema / Trigger.pm
index 90cfd5b..f320fb2 100644 (file)
@@ -1,25 +1,5 @@
 package SQL::Translator::Schema::Trigger;
 
-# ----------------------------------------------------------------------
-# $Id: Trigger.pm,v 1.4 2004-11-04 16:29:56 grommit Exp $
-# ----------------------------------------------------------------------
-# Copyright (C) 2002-4 SQLFairy Authors
-#
-# This program is free software; you can redistribute it and/or
-# modify it under the terms of the GNU General Public License as
-# published by the Free Software Foundation; version 2.
-#
-# This program is distributed in the hope that it will be useful, but
-# WITHOUT ANY WARRANTY; without even the implied warranty of
-# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
-# General Public License for more details.
-#
-# You should have received a copy of the GNU General Public License
-# along with this program; if not, write to the Free Software
-# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
-# 02111-1307  USA
-# -------------------------------------------------------------------
-
 =pod
 
 =head1 NAME
@@ -30,13 +10,14 @@ SQL::Translator::Schema::Trigger - SQL::Translator trigger object
 
   use SQL::Translator::Schema::Trigger;
   my $trigger = SQL::Translator::Schema::Trigger->new(
-      name                => 'foo',
-      perform_action_when => 'before', # or after
-      database_event      => 'insert', # or update, update_on, delete
-      fields              => [],       # fields if event is "update"
-      on_table            => 'foo',    # table name
-      action              => '...',    # text of trigger
-      schema              => $schema,  # Schema object
+    name                => 'foo',
+    perform_action_when => 'before', # or after
+    database_events     => [qw/update insert/], # also update, update_on, delete
+    fields              => [],       # if event is "update"
+    on_table            => 'foo',    # table name
+    action              => '...',    # text of trigger
+    schema              => $schema,  # Schema object
+    scope               => 'row',    # or statement
   );
 
 =head1 DESCRIPTION
@@ -48,16 +29,21 @@ C<SQL::Translator::Schema::Trigger> is the trigger object.
 =cut
 
 use strict;
+use warnings;
 use SQL::Translator::Utils 'parse_list_arg';
 
 use base 'SQL::Translator::Schema::Object';
 
-use vars qw($VERSION $TABLE_COUNT $VIEW_COUNT);
+use Carp;
+
+our ( $TABLE_COUNT, $VIEW_COUNT );
 
-$VERSION = sprintf "%d.%02d", q$Revision: 1.4 $ =~ /(\d+)\.(\d+)/;
+our $VERSION = '1.59';
 
-# ----------------------------------------------------------------------
-sub init {
+__PACKAGE__->_attributes( qw/
+    name schema perform_action_when database_events database_event
+    fields table on_table action order scope
+/);
 
 =pod
 
@@ -69,29 +55,13 @@ Object constructor.
 
 =cut
 
-    my ( $self, $config ) = @_;
-
-    for my $arg ( 
-        qw[ 
-            name perform_action_when database_event fields 
-            on_table action schema
-        ] 
-    ) {
-        next unless $config->{ $arg };
-        $self->$arg( $config->{ $arg } );# or return;
-    }
-
-    return $self;
-}
-
-# ----------------------------------------------------------------------
 sub perform_action_when {
 
 =pod
 
 =head2 perform_action_when
 
-Gets or sets whether the event happens "before" or "after" the 
+Gets or sets whether the event happens "before" or "after" the
 C<database_event>.
 
   $trigger->perform_action_when('after');
@@ -99,7 +69,7 @@ C<database_event>.
 =cut
 
     my $self = shift;
-    
+
     if ( my $arg = shift ) {
         $arg =  lc $arg;
         $arg =~ s/\s+/ /g;
@@ -107,7 +77,7 @@ C<database_event>.
             $self->{'perform_action_when'} = $arg;
         }
         else {
-            return 
+            return
                 $self->error("Invalid argument '$arg' to perform_action_when");
         }
     }
@@ -115,37 +85,57 @@ C<database_event>.
     return $self->{'perform_action_when'};
 }
 
-# ----------------------------------------------------------------------
 sub database_event {
 
 =pod
 
 =head2 database_event
 
-Gets or sets the event that triggers the trigger.
-
-  my $ok = $trigger->database_event('insert');
+Obsolete please use database_events!
 
 =cut
 
     my $self = shift;
 
-    if ( my $arg = shift ) {
-        $arg =  lc $arg;
-        $arg =~ s/\s+/ /g;
-        if ( $arg =~ /^(insert|update|update_on|delete)$/ ) {
-            $self->{'database_event'} = $arg;
-        }
-        else {
-            return 
-                $self->error("Invalid argument '$arg' to database_event");
+    return $self->database_events( @_ );
+}
+
+sub database_events {
+
+=pod
+
+=head2 database_events
+
+Gets or sets the events that triggers the trigger.
+
+  my $ok = $trigger->database_events('insert');
+
+=cut
+
+    my $self = shift;
+    my @args = ref $_[0] eq 'ARRAY' ? @{ $_[0] } : @_;
+
+    if ( @args ) {
+        @args       = map { s/\s+/ /g; lc $_ } @args;
+        my %valid   = map { $_, 1 } qw[ insert update update_on delete ];
+        my @invalid = grep { !defined $valid{ $_ } } @args;
+
+        if ( @invalid ) {
+            return $self->error(
+                sprintf("Invalid events '%s' in database_events",
+                    join(', ', @invalid)
+                )
+            );
         }
+
+        $self->{'database_events'} = [ @args ];
     }
 
-    return $self->{'database_event'};
+    return wantarray
+        ? @{ $self->{'database_events'} || [] }
+        : $self->{'database_events'};
 }
 
-# ----------------------------------------------------------------------
 sub fields {
 
 =pod
@@ -181,35 +171,57 @@ Gets and set which fields to monitor for C<database_event>.
     return wantarray ? @{ $self->{'fields'} || [] } : $self->{'fields'};
 }
 
-# ----------------------------------------------------------------------
+sub table {
+
+=pod
+
+=head2 table
+
+Gets or set the table on which the trigger works, as a L<SQL::Translator::Schema::Table> object.
+  $trigger->table($triggered_table);
+
+=cut
+
+    my ($self, $arg) = @_;
+    if ( @_ == 2 ) {
+        $self->error("Table attribute of a ".__PACKAGE__.
+                     " must be a SQL::Translator::Schema::Table")
+            unless ref $arg and $arg->isa('SQL::Translator::Schema::Table');
+        $self->{table} = $arg;
+    }
+    return $self->{table};
+}
+
 sub on_table {
 
 =pod
 
 =head2 on_table
 
-Gets or set the table name on which the trigger works.
-
-  $trigger->table('foo');
+Gets or set the table name on which the trigger works, as a string.
+  $trigger->on_table('foo');
 
 =cut
 
-    my $self = shift;
-    my $arg  = shift || '';
-    $self->{'on_table'} = $arg if $arg;
-    return $self->{'on_table'};
+    my ($self, $arg) = @_;
+    if ( @_ == 2 ) {
+        my $table = $self->schema->get_table($arg);
+        die "Table named $arg doesn't exist"
+            if !$table;
+        $self->table($table);
+    }
+    return $self->table->name;
 }
 
-# ----------------------------------------------------------------------
 sub action {
 
 =pod
 
 =head2 action
 
-Gets or set the actions of the trigger.
+Gets or set the action of the trigger.
 
-  $trigger->actions(
+  $trigger->action(
       q[
         BEGIN
           select ...;
@@ -226,7 +238,6 @@ Gets or set the actions of the trigger.
     return $self->{'action'};
 }
 
-# ----------------------------------------------------------------------
 sub is_valid {
 
 =pod
@@ -241,19 +252,18 @@ Determine whether the trigger is valid or not.
 
     my $self = shift;
 
-    for my $attr ( 
-        qw[ name perform_action_when database_event on_table action ] 
+    for my $attr (
+        qw[ name perform_action_when database_events on_table action ]
     ) {
-        return $self->error("No $attr") unless $self->$attr();
+        return $self->error("Invalid: missing '$attr'") unless $self->$attr();
     }
-    
-    return $self->error("Missing fields for UPDATE ON") if 
+
+    return $self->error("Missing fields for UPDATE ON") if
         $self->database_event eq 'update_on' && !$self->fields;
 
     return 1;
 }
 
-# ----------------------------------------------------------------------
 sub name {
 
 =pod
@@ -271,7 +281,6 @@ Get or set the trigger's name.
     return $self->{'name'} || '';
 }
 
-# ----------------------------------------------------------------------
 sub order {
 
 =pod
@@ -293,7 +302,31 @@ Get or set the trigger's order.
     return $self->{'order'} || 0;
 }
 
-# ----------------------------------------------------------------------
+
+sub scope {
+
+=pod
+
+=head2 scope
+
+Get or set the trigger's scope (row or statement).
+
+    my $scope = $trigger->scope('statement');
+
+=cut
+
+    my ( $self, $arg ) = @_;
+
+    if ( defined $arg ) {
+        return $self->error( "Invalid scope '$arg'" )
+            unless $arg =~ /^(row|statement)$/i;
+
+        $self->{scope} = $arg;
+    }
+
+    return $self->{scope} || '';
+}
+
 sub schema {
 
 =pod
@@ -317,7 +350,89 @@ Get or set the trigger's schema object.
     return $self->{'schema'};
 }
 
-# ----------------------------------------------------------------------
+sub compare_arrays {
+
+=pod
+
+=head2 compare_arrays
+
+Compare two arrays.
+
+=cut
+
+    my ($first, $second) = @_;
+    no warnings;  # silence spurious -w undef complaints
+
+    return 0 unless (ref $first eq 'ARRAY' and ref $second eq 'ARRAY' ) ;
+
+    return 0 unless @$first == @$second;
+
+    my @first = sort @$first;
+
+    my @second = sort @$second;
+
+    for (my $i = 0; $i < scalar @first; $i++) {
+        return 0 if @first[$i] ne @second[$i];
+    }
+
+    return 1;
+}
+
+sub equals {
+
+=pod
+
+=head2 equals
+
+Determines if this trigger is the same as another
+
+  my $is_identical = $trigger1->equals( $trigger2 );
+
+=cut
+
+    my $self             = shift;
+    my $other            = shift;
+    my $case_insensitive = shift;
+
+    return 0 unless $self->SUPER::equals($other);
+
+    my %names;
+    for my $name ( $self->name, $other->name ) {
+        $name = lc $name if $case_insensitive;
+        $names{ $name }++;
+    }
+
+    if ( keys %names > 1 ) {
+        return $self->error('Names not equal');
+    }
+
+    if ( !$self->perform_action_when eq $other->perform_action_when ) {
+        return $self->error('perform_action_when differs');
+    }
+
+    if (
+        !compare_arrays( [$self->database_events], [$other->database_events] )
+    ) {
+        return $self->error('database_events differ');
+    }
+
+    if ( $self->on_table ne $other->on_table ) {
+        return $self->error('on_table differs');
+    }
+
+    if ( $self->action ne $other->action ) {
+        return $self->error('action differs');
+    }
+
+    if (
+        !$self->_compare_objects( scalar $self->extra, scalar $other->extra )
+    ) {
+        return $self->error('extras differ');
+    }
+
+    return 1;
+}
+
 sub DESTROY {
     my $self = shift;
     undef $self->{'schema'}; # destroy cyclical reference
@@ -325,12 +440,11 @@ sub DESTROY {
 
 1;
 
-# ----------------------------------------------------------------------
-
 =pod
 
-=head1 AUTHOR
+=head1 AUTHORS
 
-Ken Y. Clark E<lt>kclark@cpan.orgE<gt>.
+Anonymous,
+Ken Youens-Clark E<lt>kclark@cpan.orgE<gt>.
 
 =cut