From: Justin Hunter <justin.d.hunter@gmail.com>
Date: Sat, 22 Aug 2009 21:40:38 +0000 (-0700)
Subject: add PostgreSQL ddl parsing
X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=commitdiff_plain;h=faff8d0e6b7c2a73ba6c5f23edc8f7c6c2800a30;p=dbsrgits%2FSQL-Translator-2.0-ish.git

add PostgreSQL ddl parsing
---

diff --git a/lib/SQL/Translator/Parser/DDL/PostgreSQL.pm b/lib/SQL/Translator/Parser/DDL/PostgreSQL.pm
index 98ac674..14d4ccd 100644
--- a/lib/SQL/Translator/Parser/DDL/PostgreSQL.pm
+++ b/lib/SQL/Translator/Parser/DDL/PostgreSQL.pm
@@ -1,3 +1,116 @@
 use MooseX::Declare;
 role SQL::Translator::Parser::DDL::PostgreSQL {
+    use MooseX::Types::Moose qw(Str);
+    use MooseX::MultiMethods;
+    use SQL::Translator::Constants qw(:sqlt_types :sqlt_constants);
+    use SQL::Translator::Types qw(Schema);
+    use aliased 'SQL::Translator::Object::Column';
+    use aliased 'SQL::Translator::Object::Constraint';
+    use aliased 'SQL::Translator::Object::ForeignKey';
+    use aliased 'SQL::Translator::Object::Index';
+    use aliased 'SQL::Translator::Object::PrimaryKey';
+    use aliased 'SQL::Translator::Object::Table';
+    use aliased 'SQL::Translator::Object::View';
+
+    multi method parse(Schema $data) { $data }
+
+    multi method parse(Str $data) {
+        my $translator = $self->translator;
+        my $parser = Parse::RecDescent->new($self->grammar);
+    
+#        $::RD_TRACE  = $translator->trace ? 1 : undef;
+#        $DEBUG       = $translator->debug;
+    
+        unless (defined $parser) {
+            return $translator->error("Error instantiating Parse::RecDescent ".
+                "instance: Bad grammer");
+        }
+    
+        my $result = $parser->startrule($data);
+        die "Parse failed.\n" unless defined $result;
+#        warn Dumper($result) if $DEBUG;
+    
+        my $schema = $translator->schema;
+        my @tables = 
+sort { ( $result->{tables}{ $a }{'order'} || 0 ) <=> ( $result->{tables}{ $b }{'order'} || 0 ) }
+         keys %{ $result->{tables} };
+    
+        for my $table_name ( @tables ) {
+            my $tdata = $result->{tables}{ $table_name };
+            my $table = Table->new({ name => $tdata->{table_name} });
+            $schema->add_table($table);
+    
+            $table->extra(temporary => 1) if $tdata->{'temporary'};
+            $table->comments( $tdata->{'comments'} );
+    
+            my @fields = 
+sort { $tdata->{'fields'}{ $a }{'order'} <=> $tdata->{'fields'}{ $b }{'order'} }
+            keys %{ $tdata->{'fields'} };
+    
+            for my $fname ( @fields ) {
+                my $fdata = $tdata->{'fields'}{ $fname };
+                next if $fdata->{'drop'};
+                my $field = Column->new({
+                    name              => $fdata->{'name'},
+                    data_type         => $fdata->{'data_type'},
+                    sql_data_type     => $self->data_type_mapping->{$fdata->{data_type}} || -999999,
+                    size              => $fdata->{'size'},
+                    default_value     => $fdata->{'default'},
+                    is_auto_increment => $fdata->{'is_auto_increment'},
+                    is_nullable       => $fdata->{'is_nullable'},
+                    comments          => $fdata->{'comments'},
+                    table             => $table,
+                });
+                $table->add_column($field);
+    
+                $table->primary_key( $field->name ) if $fdata->{is_primary_key};
+    
+                for my $cdata ( @{ $fdata->{constraints} } ) {
+                    next unless $cdata->{type} eq 'foreign_key';
+                    $cdata->{fields} ||= [ $field->name ];
+                    push @{ $tdata->{constraints} }, $cdata;
+                }
+            }
+    
+            for my $idata ( @{ $tdata->{indices} || [] } ) {
+                my $index = Index->new({
+                    name    => $idata->{name},
+                    type    => uc $idata->{type},
+                    columns => $idata->{fields},
+                });
+            }
+    
+            for my $cdata ( @{ $tdata->{'constraints'} || [] } ) {
+                my $constraint = Constraint->new({
+                    name             => $cdata->{name},
+                    type             => $cdata->{type},
+                    fields           => $cdata->{fields},
+                    reference_table  => $cdata->{reference_table},
+                    reference_fields => $cdata->{reference_fields},
+                    match_type       => $cdata->{match_type} || '',
+                    on_delete        => $cdata->{on_delete} || $cdata->{on_delete_do},
+                    on_update        => $cdata->{on_update} || $cdata->{on_update_do},
+                    expression       => $cdata->{expression},
+                    table            => $table,
+                });
+                $table->add_constraint($constraint);
+            }
+        }
+    
+        for my $vinfo (@{$result->{views}}) {
+          my $sql = $vinfo->{sql};
+          $sql =~ s/\A\s+|\s+\z//g;
+          my $view = View->new({
+            name    => $vinfo->{view_name},
+            sql     => $sql,
+#            columns => $vinfo->{fields},
+          });
+
+          $schema->add_view($view);
+    
+          $view->extra ( temporary => 1 ) if $vinfo->{is_temporary};
+        }
+    
+        return 1; #$schema;
+    }
 }