package SQL::Translator::Parser::PostgreSQL;
# -------------------------------------------------------------------
-# $Id: PostgreSQL.pm,v 1.30 2003-08-21 02:28:38 kycl4rk Exp $
+# $Id: PostgreSQL.pm,v 1.40 2004-08-30 18:54:58 kycl4rk Exp $
# -------------------------------------------------------------------
-# Copyright (C) 2003 Ken Y. Clark <kclark@cpan.org>,
-# Allen Day <allenday@users.sourceforge.net>,
-# darren chamberlain <darren@cpan.org>,
-# Chris Mungall <cjm@fruitfly.org>
+# 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
use strict;
use vars qw[ $DEBUG $VERSION $GRAMMAR @EXPORT_OK ];
-$VERSION = sprintf "%d.%02d", q$Revision: 1.30 $ =~ /(\d+)\.(\d+)/;
+$VERSION = sprintf "%d.%02d", q$Revision: 1.40 $ =~ /(\d+)\.(\d+)/;
$DEBUG = 0 unless defined $DEBUG;
use Data::Dumper;
$GRAMMAR = q!
-{ my ( %tables, $table_order ) }
+{ my ( %tables, $table_order, $field_order, @table_comments) }
#
# The "eofile" rule makes the parser fail if any "statement" rule
eofile : /^\Z/
statement : create
+ | comment_on_table
+ | comment_on_column
| comment
| alter
| grant
$tables{ $table_name }{'order'} = ++$table_order;
$tables{ $table_name }{'table_name'} = $table_name;
- my $i = 1;
+ if ( @table_comments ) {
+ $tables{ $table_name }{'comments'} = [ @table_comments ];
+ @table_comments = ();
+ }
+
my @constraints;
for my $definition ( @{ $item[4] } ) {
- if ( $definition->{'type'} eq 'field' ) {
+ if ( $definition->{'supertype'} eq 'field' ) {
my $field_name = $definition->{'name'};
$tables{ $table_name }{'fields'}{ $field_name } =
- { %$definition, order => $i };
- $i++;
+ { %$definition, order => $field_order++ };
for my $constraint ( @{ $definition->{'constraints'} || [] } ) {
$constraint->{'fields'} = [ $field_name ];
$constraint;
}
}
- elsif ( $definition->{'type'} eq 'constraint' ) {
- $definition->{'type'} = $definition->{'constraint_type'};
- # group FKs at the field level
-# if ( $definition->{'type'} eq 'foreign_key' ) {
-# for my $fld ( @{ $definition->{'fields'} || [] } ) {
-# push @{
-# $tables{$table_name}{'fields'}{$fld}{'constraints'}
-# }, $definition;
-# }
-# }
-# else {
- push @{ $tables{ $table_name }{'constraints'} },
- $definition;
-# }
+ elsif ( $definition->{'supertype'} eq 'constraint' ) {
+ push @{ $tables{ $table_name }{'constraints'} }, $definition;
}
- else {
+ elsif ( $definition->{'supertype'} eq 'index' ) {
push @{ $tables{ $table_name }{'indices'} }, $definition;
}
}
1;
}
-#
-# Create index.
-#
-create : /create/i unique(?) /(index|key)/i index_name /on/i table_name using_method(?) '(' field_name(s /,/) ')' where_predicate(?) ';'
+create : CREATE unique(?) /(index|key)/i index_name /on/i table_name using_method(?) '(' field_name(s /,/) ')' where_predicate(?) ';'
{
push @{ $tables{ $item{'table_name'} }{'indices'} },
{
- name => $item{'index_name'},
- type => $item{'unique'}[0] ? 'unique' : 'normal',
- fields => $item[9],
- method => $item{'using_method'}[0],
+ name => $item{'index_name'},
+ supertype => $item{'unique'}[0] ? 'constraint' : 'index',
+ type => $item{'unique'}[0] ? 'unique' : 'normal',
+ fields => $item[9],
+ method => $item{'using_method'}[0],
}
;
+
}
#
-# Create anything else (e.g., domain, function, etc.)
+# Create anything else (e.g., domain, etc.)
#
-create : /create/i WORD /[^;]+/ ';'
+create : CREATE WORD /[^;]+/ ';'
+ { @table_comments = (); }
using_method : /using/i WORD { $item[2] }
| table_constraint
| <error>
-comment : /^\s*(?:#|-{2}).*\n/
+comment : /^\s*(?:#|-{2})(.*)\n/
+ {
+ my $comment = $item[1];
+ $comment =~ s/^\s*(#|-*)\s*//;
+ $comment =~ s/\s*$//;
+ $return = $comment;
+ push @table_comments, $comment;
+ }
+
+comment_on_table : /comment/i /on/i /table/i table_name /is/i comment_phrase ';'
+ {
+ push @{ $tables{ $item{'table_name'} }{'comments'} }, $item{'comment_phrase'};
+ }
+
+comment_on_column : /comment/i /on/i /column/i column_name /is/i comment_phrase ';'
+ {
+ my $table_name = $item[4]->{'table'};
+ my $field_name = $item[4]->{'field'};
+ push @{ $tables{ $table_name }{'fields'}{ $field_name }{'comments'} },
+ $item{'comment_phrase'};
+ }
+
+column_name : NAME '.' NAME
+ { $return = { table => $item[1], field => $item[3] } }
-field : comment(s?) field_name data_type field_meta(s?) comment(s?)
+comment_phrase : /'.*?'|NULL/
+ {
+ my $val = $item[1] || '';
+ $val =~ s/^'|'$//g;
+ $return = $val;
+ }
+
+field : field_comment(s?) field_name data_type field_meta(s?) field_comment(s?)
{
my ( $default, @constraints, $is_pk );
- my $null = 1;
+ my $is_nullable = 1;
for my $meta ( @{ $item[4] } ) {
if ( $meta->{'type'} eq 'default' ) {
$default = $meta;
next;
}
elsif ( $meta->{'type'} eq 'not_null' ) {
- $null = 0;
-# next;
+ $is_nullable = 0;
}
elsif ( $meta->{'type'} eq 'primary_key' ) {
$is_pk = 1;
my @comments = ( @{ $item[1] }, @{ $item[5] } );
$return = {
- type => 'field',
+ supertype => 'field',
name => $item{'field_name'},
data_type => $item{'data_type'}{'type'},
size => $item{'data_type'}{'size'},
- null => $null,
+ is_nullable => $is_nullable,
default => $default->{'value'},
constraints => [ @constraints ],
comments => [ @comments ],
}
| <error>
+field_comment : /^\s*(?:#|-{2})(.*)\n/
+ {
+ my $comment = $item[1];
+ $comment =~ s/^\s*(#|-*)\s*//;
+ $comment =~ s/\s*$//;
+ $return = $comment;
+ }
+
field_meta : default_val
| column_constraint
{ $return = { type => 'primary_key' } }
|
/check/i '(' /[^)]+/ ')'
- { $return = { type => 'check', expression => $item[2] } }
+ { $return = { type => 'check', expression => $item[3] } }
|
/references/i table_name parens_word_list(?) match_type(?) key_action(s?)
{
#
# We can deduce some sizes from the data type's name.
#
- $data_type->{'size'} ||= $item[2][0];
+ if ( my $size = $item[2][0] ) {
+ $data_type->{'size'} = $size;
+ }
$return = $data_type;
}
$return = { type => 'bytea' };
}
|
- /(timestamptz|timestamp)/i
+ /(timestamptz|timestamp)( without time zone)?/i
{
$return = { type => 'timestamp' };
}
$return = {
name => $item{'constraint_name'}[0] || '',
- type => 'constraint',
- constraint_type => $type,
+ supertype => 'constraint',
+ type => $type,
fields => $type ne 'check' ? $fields : [],
expression => $type eq 'check' ? $expression : '',
deferrable => $item{'deferrable'},
}
}
|
- /check/i '(' /(.+)/ ')'
+ /check/i '(' /[^)]+/ ')'
{
$return = {
type => 'check',
}
$return = {
+ supertype => 'constraint',
type => 'foreign_key',
fields => $item[3],
reference_table => $item[6],
key_delete : /on delete/i key_mutation
{
- $return => {
+ $return = {
type => 'delete',
action => $item[2],
};
key_update : /on update/i key_mutation
{
- $return => {
+ $return = {
type => 'update',
action => $item[2],
};
|
/set default/i { $return = 'set default' }
-alter : alter_table table_name /add/i table_constraint ';'
+alter : alter_table table_name add_column field ';'
+ {
+ my $field_def = $item[4];
+ $tables{ $item[2] }{'fields'}{ $field_def->{'name'} } = {
+ %$field_def, order => $field_order++
+ };
+ 1;
+ }
+
+alter : alter_table table_name ADD table_constraint ';'
{
my $table_name = $item[2];
my $constraint = $item[4];
- $constraint->{'type'} = $constraint->{'constraint_type'};
push @{ $tables{ $table_name }{'constraints'} }, $constraint;
+ 1;
+ }
+
+alter : alter_table table_name drop_column NAME restrict_or_cascade(?) ';'
+ {
+ $tables{ $item[2] }{'fields'}{ $item[4] }{'drop'} = 1;
+ 1;
+ }
+
+alter : alter_table table_name alter_column NAME alter_default_val ';'
+ {
+ $tables{ $item[2] }{'fields'}{ $item[4] }{'default'} =
+ $item[5]->{'value'};
+ 1;
+ }
+
+#
+# These will just parse for now but won't affect the structure. - ky
+#
+alter : alter_table table_name /rename/i /to/i NAME ';'
+ { 1 }
+
+alter : alter_table table_name alter_column NAME SET /statistics/i INTEGER ';'
+ { 1 }
+
+alter : alter_table table_name alter_column NAME SET /storage/i storage_type ';'
+ { 1 }
+
+alter : alter_table table_name rename_column NAME /to/i NAME ';'
+ { 1 }
+
+alter : alter_table table_name DROP /constraint/i NAME restrict_or_cascade ';'
+ { 1 }
+
+alter : alter_table table_name /owner/i /to/i NAME ';'
+ { 1 }
+
+storage_type : /(plain|external|extended|main)/i
+
+alter_default_val : SET default_val
+ {
+ $return = { value => $item[2]->{'value'} }
+ }
+ | DROP DEFAULT
+ {
+ $return = { value => undef }
+ }
+
+#
+# This is a little tricky to get right, at least WRT to making the
+# tests pass. The problem is that the constraints are stored just as
+# a list (no name access), and the tests expect the constraints in a
+# particular order. I'm going to leave the rule but disable the code
+# for now. - ky
+#
+alter : alter_table table_name alter_column NAME alter_nullable ';'
+ {
+# my $table_name = $item[2];
+# my $field_name = $item[4];
+# my $is_nullable = $item[5]->{'is_nullable'};
+#
+# $tables{ $table_name }{'fields'}{ $field_name }{'is_nullable'} =
+# $is_nullable;
+#
+# if ( $is_nullable ) {
+# 1;
+# push @{ $tables{ $table_name }{'constraints'} }, {
+# type => 'not_null',
+# fields => [ $field_name ],
+# };
+# }
+# else {
+# for my $i (
+# 0 .. $#{ $tables{ $table_name }{'constraints'} || [] }
+# ) {
+# my $c = $tables{ $table_name }{'constraints'}[ $i ] or next;
+# my $fields = join( '', @{ $c->{'fields'} || [] } ) or next;
+# if ( $c->{'type'} eq 'not_null' && $fields eq $field_name ) {
+# delete $tables{ $table_name }{'constraints'}[ $i ];
+# last;
+# }
+# }
+# }
+
+ 1;
+ }
+
+alter_nullable : SET not_null
+ {
+ $return = { is_nullable => 0 }
+ }
+ | DROP not_null
+ {
+ $return = { is_nullable => 1 }
}
-alter_table : /alter/i /table/i only(?)
+not_null : /not/i /null/i
+
+add_column : ADD COLUMN(?)
+
+alter_table : ALTER TABLE ONLY(?)
+
+drop_column : DROP COLUMN(?)
-only : /only/i
+alter_column : ALTER COLUMN(?)
+
+rename_column : /rename/i COLUMN(?)
+
+restrict_or_cascade : /restrict/i |
+ /cascade/i
+
+#
+# End basically useless stuff. - ky
+#
-create_table : /create/i TABLE
+create_table : CREATE TABLE
-create_index : /create/i /index/i
+create_index : CREATE /index/i
-default_val : /default/i /(\d+|'[^']*'|\w+\(.*?\))|\w+/
+default_val : DEFAULT /(\d+|'[^']*'|\w+\(.*?\))|\w+/
{
my $val = defined $item[2] ? $item[2] : '';
$val =~ s/^'|'$//g;
$return = { type => $item[1] =~ /out/i ? 'without_oids' : 'with_oids' }
}
+ADD : /add/i
+
+ALTER : /alter/i
+
+CREATE : /create/i
+
+ONLY : /only/i
+
+DEFAULT : /default/i
+
+DROP : /drop/i
+
+COLUMN : /column/i
+
TABLE : /table/i
SEMICOLON : /\s*;\n?/
+INTEGER : /\d+/
+
WORD : /\w+/
DIGITS : /\d+/
COMMA : ','
+SET : /set/i
+
NAME : "`" /\w+/ "`"
{ $item[2] }
| /\w+/
my $schema = $translator->schema;
my @tables = sort {
- $result->{ $a }->{'order'} <=> $result->{ $b }->{'order'}
+ ( $result->{ $a }{'order'} || 0 ) <=> ( $result->{ $b }{'order'} || 0 )
} keys %{ $result };
for my $table_name ( @tables ) {
name => $tdata->{'table_name'},
) or die "Couldn't create table '$table_name': " . $schema->error;
+ $table->comments( $tdata->{'comments'} );
+
my @fields = sort {
- $tdata->{'fields'}->{$a}->{'order'}
+ $tdata->{'fields'}{ $a }{'order'}
<=>
- $tdata->{'fields'}->{$b}->{'order'}
+ $tdata->{'fields'}{ $b }{'order'}
} keys %{ $tdata->{'fields'} };
for my $fname ( @fields ) {
my $fdata = $tdata->{'fields'}{ $fname };
+ next if $fdata->{'drop'};
my $field = $table->add_field(
name => $fdata->{'name'},
data_type => $fdata->{'data_type'},
size => $fdata->{'size'},
default_value => $fdata->{'default'},
is_auto_increment => $fdata->{'is_auto_increment'},
- is_nullable => $fdata->{'null'},
+ is_nullable => $fdata->{'is_nullable'},
+ comments => $fdata->{'comments'},
) or die $table->error;
$table->primary_key( $field->name ) if $fdata->{'is_primary_key'};
match_type => $cdata->{'match_type'} || '',
on_delete => $cdata->{'on_delete_do'},
on_update => $cdata->{'on_update_do'},
- ) or die $table->error;
+ expression => $cdata->{'expression'},
+ ) or die "Can't add constraint of type '" .
+ $cdata->{'type'} . "' to table '" . $table->name .
+ "': " . $table->error;
}
}