mysql_table_type extra data and InnoDB derivation fix.
[dbsrgits/SQL-Translator.git] / lib / SQL / Translator / Producer / MySQL.pm
index 0faf96e..5a5bba5 100644 (file)
@@ -1,10 +1,9 @@
 package SQL::Translator::Producer::MySQL;
 
-#-----------------------------------------------------
-# $Id: MySQL.pm,v 1.1 2002-03-27 12:41:53 dlc Exp $
-#-----------------------------------------------------
-# Copyright (C) 2002 Ken Y. Clark <kycl4rk@users.sourceforge.net>,
-#                    darren chamberlain <darren@cpan.org>
+# -------------------------------------------------------------------
+# $Id: MySQL.pm,v 1.42 2005-01-13 11:50:23 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
@@ -21,114 +20,343 @@ package SQL::Translator::Producer::MySQL;
 # 02111-1307  USA
 # -------------------------------------------------------------------
 
+=head1 NAME
+
+SQL::Translator::Producer::MySQL - MySQL-specific producer for SQL::Translator
+
+=head1 SYNOPSIS
+
+Use via SQL::Translator:
+
+  use SQL::Translator;
+
+  my $t = SQL::Translator->new( parser => '...', producer => 'MySQL', '...' );
+  $t->translate;
+
+=head1 DESCRIPTION
+
+This module will produce text output of the schema suitable for MySQL.
+There are still some issues to be worked out with syntax differences 
+between MySQL versions 3 and 4 ("SET foreign_key_checks," character sets
+for fields, etc.).
+
+=head2 Table Types
+
+Normally the tables will be created without any explicit table type given and
+so will use the MySQL default.
+
+Any tables involved in foreign key constraints automatically get a table type
+of InnoDB, unless this is overridden by setting the C<mysql_table_type> extra
+attribute explicitly on the table.
+
+=head2 Extra attributes.
+
+The producer recognises the following extra attributes on the Schema objects.
+
+=over 4
+
+=item field.list
+
+Set the list of allowed values for Enum fields.
+
+=item field.binary field.unsigned field.zerofill
+
+Set the MySQL field options of the same name.
+
+=item mysql_table_type
+
+Set the type of the table e.g. 'InnoDB', 'MyISAM'. This will be
+automatically set for tables involved in foreign key constraints if it is
+not already set explicitly. See L<"Table Types">.
+
+=back
+
+=cut
+
 use strict;
-use vars qw($VERSION $DEBUG);
-$VERSION = sprintf "%d.%02d", q$Revision: 1.1 $ =~ /(\d+)\.(\d+)/;
-$DEBUG = 1 unless defined $DEBUG;
+use vars qw[ $VERSION $DEBUG ];
+$VERSION = sprintf "%d.%02d", q$Revision: 1.42 $ =~ /(\d+)\.(\d+)/;
+$DEBUG   = 0 unless defined $DEBUG;
 
 use Data::Dumper;
+use SQL::Translator::Schema::Constants;
+use SQL::Translator::Utils qw(debug header_comment);
 
-sub import {
-    warn "loading " . __PACKAGE__ . "...\n";
-}
+#
+# Use only lowercase for the keys (e.g. "long" and not "LONG")
+#
+my %translate  = (
+    #
+    # Oracle types
+    #
+    varchar2   => 'varchar',
+    long       => 'text',
+    clob       => 'longtext',
+
+    #
+    # Sybase types
+    #
+    int        => 'integer',
+    money      => 'float',
+    real       => 'double',
+    comment    => 'text',
+    bit        => 'tinyint',
+
+    #
+    # Access types
+    #
+    'long integer' => 'integer',
+    'text'         => 'text',
+    'datetime'     => 'datetime',
+);
 
 sub produce {
-    my ($translator, $data) = @_;
-    debug("Beginning");
-    my $create = sprintf 
-"# ----------------------------------------------------------------------
-# Created by %s
-# Created on %s
-# ----------------------------------------------------------------------\n\n",
-        __PACKAGE__, scalar localtime;
-
-    for my $table (keys %{$data}) {
-        debug("Looking a '$table'");
-        my $table_data = $data->{$table};
-        my @fields = sort { $table_data->{'fields'}->{$a}->{'order'} <=>
-                            $table_data->{'fields'}->{$b}->{'order'}
-                          } keys %{$table_data->{'fields'}};
-
-        # --------------------------------------------------------------
+    my $translator     = shift;
+    local $DEBUG       = $translator->debug;
+    my $no_comments    = $translator->no_comments;
+    my $add_drop_table = $translator->add_drop_table;
+    my $schema         = $translator->schema;
+    my $show_warnings  = $translator->show_warnings || 0;
+
+    debug("PKG: Beginning production\n");
+
+    my $create; 
+    $create .= header_comment unless ($no_comments);
+    # \todo Don't set if MySQL 3.x is set on command line
+    $create .= "SET foreign_key_checks=0;\n\n";
+
+    #
+    # Work out which tables need to be InnoDB to support foreign key
+    # constraints. We do this first as we need InnoDB at both ends.
+    #
+    foreach ( map { $_->get_constraints } $schema->get_tables ) {
+        foreach my $meth (qw/table reference_table/) {
+            my $table = $schema->get_table($_->$meth) || next;
+            next if $table->extra('mysql_table_type');
+            $table->extra( 'mysql_table_type' => 'InnoDB');
+        }
+    }
+
+    #
+    # Generate sql
+    #
+    for my $table ( $schema->get_tables ) {
+        my $table_name = $table->name;
+        debug("PKG: Looking at table '$table_name'\n");
+
+        #
         # Header.  Should this look like what mysqldump produces?
-        # --------------------------------------------------------------
-        $create .=
-"# ----------------------------------------------------------------------
-# Table: $table
-# ----------------------------------------------------------------------\n";
-        $create .= "CREATE TABLE $table (\n";
-
-        # --------------------------------------------------------------
+        #
+        $create .= "--\n-- Table: $table_name\n--\n" unless $no_comments;
+        $create .= qq[DROP TABLE IF EXISTS $table_name;\n] if $add_drop_table;
+        $create .= "CREATE TABLE $table_name (\n";
+
+        #
         # Fields
-        # --------------------------------------------------------------
-        for (my $i = 0; $i <= $#fields; $i++) {
-            my $field = $fields[$i];
-            debug("Looking at field: $field");
-            my $field_data = $table_data->{'fields'}->{$field};
-            my @fdata = ("", $field);
+        #
+        my @field_defs;
+        for my $field ( $table->get_fields ) {
+            my $field_name = $field->name;
+            debug("PKG: Looking at field '$field_name'\n");
+            my $field_def = $field_name;
 
             # data type and size
-            push @fdata, sprintf "%s(%d)", $field_data->{'data_type'},
-                                           $field_data->{'size'};
+            my $data_type = $field->data_type;
+            my @size      = $field->size;
+            my %extra     = $field->extra;
+            my $list      = $extra{'list'} || [];
+            # \todo deal with embedded quotes
+            my $commalist = join( ', ', map { qq['$_'] } @$list );
+
+            #
+            # Oracle "number" type -- figure best MySQL type
+            #
+            if ( lc $data_type eq 'number' ) {
+                # not an integer
+                if ( scalar @size > 1 ) {
+                    $data_type = 'double';
+                }
+                elsif ( $size[0] && $size[0] >= 12 ) {
+                    $data_type = 'bigint';
+                }
+                elsif ( $size[0] && $size[0] <= 1 ) {
+                    $data_type = 'tinyint';
+                }
+                else {
+                    $data_type = 'int';
+                }
+            }
+            #
+            # Convert a large Oracle varchar to "text"
+            #
+            elsif ( $data_type =~ /char/i && $size[0] > 255 ) {
+                $data_type = 'text';
+                @size      = ();
+            }
+            elsif ( $data_type =~ /char/i && ! $size[0] ) {
+                @size = (255);
+            }
+            elsif ( $data_type =~ /boolean/i ) {
+                $data_type = 'enum';
+                $commalist = "'0','1'";
+            }
+            elsif ( exists $translate{ lc $data_type } ) {
+                $data_type = $translate{ lc $data_type };
+            }
+
+            @size = () if $data_type =~ /(text|blob)/i;
+
+            if ( $data_type =~ /(double|float)/ && scalar @size == 1 ) {
+                push @size, '0';
+            }
+
+            $field_def .= " $data_type";
+            
+            if ( lc $data_type eq 'enum' ) {
+                $field_def .= '(' . $commalist . ')';
+                       } 
+            elsif ( defined $size[0] && $size[0] > 0 ) {
+                $field_def .= '(' . join( ', ', @size ) . ')';
+            }
+
+            # MySQL qualifiers
+            for my $qual ( qw[ binary unsigned zerofill ] ) {
+                my $val = $extra{ $qual || uc $qual } or next;
+                $field_def .= " $qual";
+            }
 
             # Null?
-            push @fdata, "NOT NULL" unless $field_data->{'null'};
+            $field_def .= ' NOT NULL' unless $field->is_nullable;
 
             # Default?  XXX Need better quoting!
-            if (my $default = $field_data->{'default'}) {
-                if (int $default eq "$default") {
-                    push @fdata, "DEFAULT $default";
+            my $default = $field->default_value;
+            if ( defined $default ) {
+                if ( uc $default eq 'NULL') {
+                    $field_def .= ' DEFAULT NULL';
                 } else {
-                    push @fdata, "DEFAULT '$default'";
+                    $field_def .= " DEFAULT '$default'";
                 }
             }
 
             # auto_increment?
-            push @fdata, "auto_increment" if $field_data->{'is_auto_inc'};
+            $field_def .= " auto_increment" if $field->is_auto_increment;
+            push @field_defs, $field_def;
+               }
 
-            # primary key?
-            push @fdata, "PRIMARY KEY" if $field_data->{'is_primary_key'};
+        #
+        # Indices
+        #
+        my @index_defs;
+        my %indexed_fields;
+        for my $index ( $table->get_indices ) {
+            push @index_defs, join( ' ', 
+                lc $index->type eq 'normal' ? 'INDEX' : $index->type,
+                $index->name,
+                '(' . join( ', ', $index->fields ) . ')'
+            );
+            $indexed_fields{ $_ } = 1 for $index->fields;
+        }
 
+        #
+        # Constraints -- need to handle more than just FK. -ky
+        #
+        my @constraint_defs;
+        my @constraints = $table->get_constraints;
+        for my $c ( @constraints ) {
+            my @fields = $c->fields or next;
 
-            $create .= (join "\t", @fdata);
-            $create .= "," unless ($i == $#fields);
-            $create .= "\n";
-        }
+            if ( $c->type eq PRIMARY_KEY ) {
+                push @constraint_defs,
+                    'PRIMARY KEY (' . join(', ', @fields). ')';
+            }
+            elsif ( $c->type eq UNIQUE ) {
+                push @constraint_defs,
+                    'UNIQUE (' . join(', ', @fields). ')';
+            }
+            elsif ( $c->type eq FOREIGN_KEY ) {
+                #
+                # Make sure FK field is indexed or MySQL complains.
+                #
+                unless ( $indexed_fields{ $fields[0] } ) {
+                    push @index_defs, "INDEX ($fields[0])";
+                    $indexed_fields{ $fields[0] } = 1;
+                }
+
+                my $def = join(' ', 
+                    map { $_ || () } 'FOREIGN KEY', $c->name 
+                );
+
+                $def .= ' (' . join( ', ', @fields ) . ')';
+
+                $def .= ' REFERENCES ' . $c->reference_table;
 
-        # --------------------------------------------------------------
-        # Other keys
-        # --------------------------------------------------------------
+                my @rfields = map { $_ || () } $c->reference_fields;
+                unless ( @rfields ) {
+                    my $rtable_name = $c->reference_table;
+                    if ( my $ref_table = $schema->get_table( $rtable_name ) ) {
+                        push @rfields, $ref_table->primary_key;
+                    }
+                    else {
+                        warn "Can't find reference table '$rtable_name' " .
+                            "in schema\n" if $show_warnings;
+                    }
+                }
+
+                if ( @rfields ) {
+                    $def .= ' (' . join( ', ', @rfields ) . ')';
+                }
+                else {
+                    warn "FK constraint on " . $table->name . '.' .
+                        join('', @fields) . " has no reference fields\n" 
+                        if $show_warnings;
+                }
 
+                if ( $c->match_type ) {
+                    $def .= ' MATCH ' . 
+                        ( $c->match_type =~ /full/i ) ? 'FULL' : 'PARTIAL';
+                }
 
-        # --------------------------------------------------------------
+                if ( $c->on_delete ) {
+                    $def .= ' ON DELETE '.join( ' ', $c->on_delete );
+                }
+
+                if ( $c->on_update ) {
+                    $def .= ' ON UPDATE '.join( ' ', $c->on_update );
+                }
+
+                push @constraint_defs, $def;
+            }
+        }
+
+        $create .= join(",\n", map { "  $_" } 
+            @field_defs, @index_defs, @constraint_defs
+        );
+
+        #
         # Footer
-        # --------------------------------------------------------------
-        $create .= ")";
-        $create .= " TYPE=$table_data->{'type'}"
-            if defined $table_data->{'type'};
+        #
+        $create .= "\n)";
+        my $mysql_table_type = $table->extra('mysql_table_type');
+        $create .= " Type=$mysql_table_type" if $mysql_table_type;
         $create .= ";\n\n";
     }
 
-    $create .= "#\n";
-
     return $create;
 }
 
-use Carp;
-sub debug {
-    if ($DEBUG) {
-        map { carp "[" . __PACKAGE__ . "] $_" } @_;
-    }
-}
-
 1;
-__END__
 
-=head1 NAME
+# -------------------------------------------------------------------
+
+=pod
+
+=head1 SEE ALSO
 
-SQL::Translator::Producer::MySQL - mysql-specific producer for SQL::Translator
+SQL::Translator, http://www.mysql.com/.
 
+=head1 AUTHORS
 
-=head1 AUTHOR
+darren chamberlain E<lt>darren@cpan.orgE<gt>,
+Ken Y. Clark E<lt>kclark@cpan.orgE<gt>.
 
-darren chamberlain E<lt>darren@cpan.orgE<gt>
+=cut