Added normalize_name function, which normalizes names. Primarily needed by the Excel...
[dbsrgits/SQL-Translator.git] / lib / SQL / Translator / Producer / MySQL.pm
index 0faf96e..4d56f53 100644 (file)
@@ -1,10 +1,11 @@
 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.12 2003-04-14 19:20:26 kycl4rk Exp $
+# -------------------------------------------------------------------
+# Copyright (C) 2003 Ken Y. Clark <kclark@cpan.org>,
+#                    darren chamberlain <darren@cpan.org>,
+#                    Chris Mungall <cjm@fruitfly.org>
 #
 # This program is free software; you can redistribute it and/or
 # modify it under the terms of the GNU General Public License as
@@ -22,62 +23,117 @@ package SQL::Translator::Producer::MySQL;
 # -------------------------------------------------------------------
 
 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.12 $ =~ /(\d+)\.(\d+)/;
+$DEBUG   = 0 unless defined $DEBUG;
 
 use Data::Dumper;
-
-sub import {
-    warn "loading " . __PACKAGE__ . "...\n";
-}
+use SQL::Translator::Utils qw(debug);
+
+my %translate  = (
+    #
+    # Oracle types
+    #
+    varchar2   => 'varchar',
+    long       => 'text',
+    CLOB       => 'longtext',
+
+    #
+    # Sybase types
+    #
+    int        => 'integer',
+    money      => 'float',
+    real       => 'double',
+    comment    => 'text',
+    bit        => 'tinyint',
+);
 
 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'");
+    local $DEBUG            = $translator->debug;
+    my $no_comments         = $translator->no_comments;
+    my $add_drop_table      = $translator->add_drop_table;
+
+    debug("PKG: Beginning production\n");
+
+    my $create; 
+    unless ( $no_comments ) {
+        $create .= sprintf "--\n-- Created by %s\n-- Created on %s\n--\n\n",
+            __PACKAGE__, scalar localtime;
+    }
+
+    for my $table ( keys %{ $data } ) {
+        debug("PKG: Looking at table '$table'\n");
         my $table_data = $data->{$table};
-        my @fields = sort { $table_data->{'fields'}->{$a}->{'order'} <=>
-                            $table_data->{'fields'}->{$b}->{'order'}
-                          } keys %{$table_data->{'fields'}};
+        my @fields = sort { 
+            $table_data->{'fields'}->{$a}->{'order'} 
+            <=>
+            $table_data->{'fields'}->{$b}->{'order'}
+        } keys %{$table_data->{'fields'}};
 
-        # --------------------------------------------------------------
+        #
         # Header.  Should this look like what mysqldump produces?
-        # --------------------------------------------------------------
-        $create .=
-"# ----------------------------------------------------------------------
-# Table: $table
-# ----------------------------------------------------------------------\n";
-        $create .= "CREATE TABLE $table (\n";
-
-        # --------------------------------------------------------------
+        #
+        $create .= "--\n-- Table: $table\n--\n" unless $no_comments;
+        $create .= qq[DROP TABLE IF EXISTS $table;\n] if $add_drop_table;
+        $create .= "CREATE TABLE $table (";
+
+        #
         # Fields
-        # --------------------------------------------------------------
+        #
         for (my $i = 0; $i <= $#fields; $i++) {
             my $field = $fields[$i];
-            debug("Looking at field: $field");
+            debug("PKG: Looking at field '$field'\n");
             my $field_data = $table_data->{'fields'}->{$field};
             my @fdata = ("", $field);
+            $create .= "\n";
 
             # data type and size
-            push @fdata, sprintf "%s(%d)", $field_data->{'data_type'},
-                                           $field_data->{'size'};
+            my $attr      = uc $field_data->{'data_type'} eq 'SET' 
+                            ? 'list' : 'size';
+            my @values    = @{ $field_data->{ $attr } || [] };
+            my $data_type = $field_data->{'data_type'};
+
+            if ( $data_type eq 'number' ) {
+                # not an integer
+                if ( scalar @values > 1 ) {
+                    $data_type = 'double';
+                }
+                elsif ( $values[0] >= 12 ) {
+                    $data_type = 'bigint';
+                }
+                elsif ( $values[0] <= 1 ) {
+                    $data_type = 'tinyint';
+                }
+                else {
+                    $data_type = 'int';
+                }
+            }
+            elsif ( exists $translate{ $data_type } ) {
+                $data_type = $translate{ $data_type };
+            }
+
+            push @fdata, sprintf "%s%s", 
+                $data_type,
+                ( @values )
+                    ? '(' . join( ', ', @values ) . ')'
+                    : '';
+
+            # MySQL qualifiers
+            for my $qual ( qw[ binary unsigned zerofill ] ) {
+                push @fdata, $qual 
+                    if $field_data->{ $qual } ||
+                       $field_data->{ uc $qual };
+            }
 
             # Null?
             push @fdata, "NOT NULL" unless $field_data->{'null'};
 
             # Default?  XXX Need better quoting!
-            if (my $default = $field_data->{'default'}) {
-                if (int $default eq "$default") {
-                    push @fdata, "DEFAULT $default";
+            my $default = $field_data->{'default'};
+            if ( defined $default ) {
+                if ( uc $default eq 'NULL') {
+                    push @fdata, "DEFAULT NULL";
                 } else {
                     push @fdata, "DEFAULT '$default'";
                 }
@@ -87,48 +143,101 @@ sub produce {
             push @fdata, "auto_increment" if $field_data->{'is_auto_inc'};
 
             # primary key?
-            push @fdata, "PRIMARY KEY" if $field_data->{'is_primary_key'};
+            # This is taken care of in the indices, could be duplicated here
+            # push @fdata, "PRIMARY KEY" if $field_data->{'is_primary_key'};
 
 
-            $create .= (join "\t", @fdata);
+            $create .= (join " ", '', @fdata);
             $create .= "," unless ($i == $#fields);
-            $create .= "\n";
         }
 
-        # --------------------------------------------------------------
-        # Other keys
-        # --------------------------------------------------------------
+        #
+        # Indices
+        #
+        my @index_creates;
+        my @indices     = @{ $table_data->{'indices'}     || [] };
+        my @constraints = @{ $table_data->{'constraints'} || [] };
+        for my $key ( @indices, @constraints ) {
+            my ($name, $type, $fields) = @{ $key }{ qw[ name type fields ] };
+            $name ||= '';
+            my $index_type = 
+                $type eq 'primary_key' ? 'PRIMARY KEY' :
+                $type eq 'unique'      ? 'UNIQUE KEY'  :
+                $type eq 'key'         ? 'KEY'         : '';
+            next unless $index_type;
+            push @index_creates, 
+                "  $index_type $name (" . join( ', ', @$fields ) . ')';
+        }
 
+        if ( @index_creates ) {
+            $create .= join(",\n", '', @index_creates);
+        }
 
-        # --------------------------------------------------------------
+        #
+        # Constraints -- need to handle more than just FK. -ky
+        #
+        my @constraint_defs;
+        for my $constraint ( @constraints ) {
+            my $name       = $constraint->{'name'} || '';
+            my $type       = $constraint->{'type'};
+            my $fields     = $constraint->{'fields'};
+            my $ref_table  = $constraint->{'reference_table'};
+            my $ref_fields = $constraint->{'reference_fields'};
+            my $match_type = $constraint->{'match_type'} || '';
+            my $on_delete  = $constraint->{'on_delete_do'};
+            my $on_update  = $constraint->{'on_update_do'};
+
+            if ( $type eq 'foreign_key' ) {
+                my $def = join(' ', map { $_ || () } '  FOREIGN KEY', $name );
+                if ( @$fields ) {
+                    $def .= ' (' . join( ', ', @$fields ) . ')';
+                }
+                $def .= " REFERENCES $ref_table";
+
+                if ( @{ $ref_fields || [] } ) {
+                    $def .= ' (' . join( ', ', @$ref_fields ) . ')';
+                }
+
+                if ( $match_type ) {
+                    $def .= ' MATCH ' . 
+                        ( $match_type =~ /full/i ) ? 'FULL' : 'PARTIAL';
+                }
+
+                if ( @{ $on_delete || [] } ) {
+                    $def .= ' ON DELETE '.join(' ', @$on_delete);
+                }
+
+                if ( @{ $on_update || [] } ) {
+                    $def .= ' ON UPDATE '.join(' ', @$on_update);
+                }
+
+                push @constraint_defs, $def;
+            }
+        }
+
+        $create .= join(",\n", '', @constraint_defs) if @constraint_defs;
+
+        #
         # Footer
-        # --------------------------------------------------------------
-        $create .= ")";
-        $create .= " TYPE=$table_data->{'type'}"
-            if defined $table_data->{'type'};
+        #
+        $create .= "\n)";
+        while ( my ( $key, $val ) = each %{ $table_data->{'table_options'} } ) {
+            $create .= " $key=$val" 
+        }
         $create .= ";\n\n";
     }
 
-    $create .= "#\n";
-
     return $create;
 }
 
-use Carp;
-sub debug {
-    if ($DEBUG) {
-        map { carp "[" . __PACKAGE__ . "] $_" } @_;
-    }
-}
-
 1;
 __END__
 
 =head1 NAME
 
-SQL::Translator::Producer::MySQL - mysql-specific producer for SQL::Translator
-
+SQL::Translator::Producer::MySQL - MySQL-specific producer for SQL::Translator
 
 =head1 AUTHOR
 
-darren chamberlain E<lt>darren@cpan.orgE<gt>
+darren chamberlain E<lt>darren@cpan.orgE<gt>,
+Ken Y. Clark E<lt>kclark@cpan.orgE<gt>