Added SQLite producer, basic knock-off of MySQL producer, made some mods
Ken Youens-Clark [Tue, 4 Mar 2003 21:24:13 +0000 (21:24 +0000)]
to MySQL producer to add "DROP TABLE" statements, fixed POD a bit.

lib/SQL/Translator/Producer/MySQL.pm
lib/SQL/Translator/Producer/SQLite.pm [new file with mode: 0644]

index 0bf8aa6..54e0b71 100644 (file)
@@ -1,7 +1,7 @@
 package SQL::Translator::Producer::MySQL;
 
 # -------------------------------------------------------------------
-# $Id: MySQL.pm,v 1.7 2003-02-26 05:17:49 kycl4rk Exp $
+# $Id: MySQL.pm,v 1.8 2003-03-04 21:24:12 kycl4rk Exp $
 # -------------------------------------------------------------------
 # Copyright (C) 2003 Ken Y. Clark <kclark@cpan.org>,
 #                    darren chamberlain <darren@cpan.org>,
@@ -24,7 +24,7 @@ package SQL::Translator::Producer::MySQL;
 
 use strict;
 use vars qw[ $VERSION $DEBUG ];
-$VERSION = sprintf "%d.%02d", q$Revision: 1.7 $ =~ /(\d+)\.(\d+)/;
+$VERSION = sprintf "%d.%02d", q$Revision: 1.8 $ =~ /(\d+)\.(\d+)/;
 $DEBUG   = 1 unless defined $DEBUG;
 
 use Data::Dumper;
@@ -37,6 +37,7 @@ sub produce {
     my ($translator, $data) = @_;
     $DEBUG                  = $translator->debug;
     my $no_comments         = $translator->no_comments;
+    my $add_drop_table      = $translator->add_drop_table;
 
     debug("Beginning production\n");
 
@@ -46,7 +47,7 @@ sub produce {
             __PACKAGE__, scalar localtime;
     }
 
-    for my $table (keys %{$data}) {
+    for my $table ( keys %{ $data } ) {
         debug("Looking at table '$table'\n");
         my $table_data = $data->{$table};
         my @fields = sort { 
@@ -59,6 +60,7 @@ sub produce {
         # Header.  Should this look like what mysqldump produces?
         #
         $create .= "--\n-- Table: $table\n--\n" unless $no_comments;
+        $create .= qq[DROP TABLE $table;\n] if $add_drop_table;
         $create .= "CREATE TABLE $table (";
 
         #
@@ -200,8 +202,9 @@ __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>
diff --git a/lib/SQL/Translator/Producer/SQLite.pm b/lib/SQL/Translator/Producer/SQLite.pm
new file mode 100644 (file)
index 0000000..e2013c0
--- /dev/null
@@ -0,0 +1,217 @@
+package SQL::Translator::Producer::SQLite;
+
+# -------------------------------------------------------------------
+# $Id: SQLite.pm,v 1.1 2003-03-04 21:24:13 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
+# 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
+# -------------------------------------------------------------------
+
+use strict;
+use Data::Dumper;
+
+use vars qw[ $VERSION $DEBUG $WARN ];
+$VERSION = sprintf "%d.%02d", q$Revision: 1.1 $ =~ /(\d+)\.(\d+)/;
+
+my %used_identifiers = ();
+my $max_id_length    = 30;
+my %global_names;
+my %truncated;
+
+sub import {
+    warn "loading " . __PACKAGE__ . "...\n";
+}
+
+sub produce {
+    my ($translator, $data) = @_;
+    $DEBUG                  = $translator->debug;
+    $WARN                   = $translator->show_warnings;
+    my $no_comments         = $translator->no_comments;
+    my $add_drop_table      = $translator->add_drop_table;
+
+    debug("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("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'}};
+
+        #
+        # Header.  Should this look like what mysqldump produces?
+        #
+        $create .= "--\n-- Table: $table\n--\n" unless $no_comments;
+        $create .= qq[DROP TABLE $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'\n");
+            my $field_data = $table_data->{'fields'}->{$field};
+            my @fdata = ("", $field);
+            $create .= "\n";
+
+            # data type and size
+            my $data_type = lc $field_data->{'data_type'};
+            my $list      = $field_data->{'list'} || [];
+            my $commalist = join ",", @$list;
+            my $size;
+
+            if ( $data_type eq 'set' ) {
+                $data_type = 'varchar';
+                $size      = length $commalist;
+            }
+            else {
+                $size = join( ', ', @{ $field_data->{'size'} || [] } );
+            }
+
+            push @fdata, sprintf "%s%s", $data_type, ($size) ? "($size)" : '';
+
+            # 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!
+            my $default = $field_data->{'default'};
+            if ( defined $default ) {
+                if ( uc $default eq 'NULL') {
+                    push @fdata, "DEFAULT NULL";
+                } else {
+                    push @fdata, "DEFAULT '$default'";
+                }
+            }
+
+            # auto_increment?
+#            push @fdata, "auto_increment" if $field_data->{'is_auto_inc'};
+
+            # 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 " ", '', @fdata);
+            $create .= "," unless ($i == $#fields);
+        }
+        #
+        # Indices
+        #
+        my @index_creates;
+        my $idx_name_default;
+        for my $index ( @{ $table_data->{'indices'} || [] } ) {
+            my ($name, $type, $fields) = @{ $index }{ qw[ name type fields ] };
+            $name ||= '';
+            my $index_type = 
+                $type eq 'primary_key' ? 'PRIMARY KEY' :
+                $type eq 'unique'      ? 'UNIQUE INDEX'  : 'INDEX';
+            if ( $type eq 'primary_key' ) {
+                $create .= join(",\n", '', 
+                    "  $index_type $name (" . join( ', ', @$fields ) . ')'
+                );
+            }
+            else {
+                $name = mk_name( 
+                    $table, $name || ++$idx_name_default
+                );
+                push @index_creates, 
+                    "CREATE $index_type $name on $table ".
+                    '(' . join( ', ', @$fields ) . ')';
+            }
+        }
+
+        $create .= "\n);\n";
+
+        for my $index_create ( @index_creates ) {
+            $create .= "$index_create;\n";
+        }
+
+        $create .= "\n";
+    }
+
+    return $create;
+}
+
+# -------------------------------------------------------------------
+sub debug {
+    if ($DEBUG) {
+        map { warn "[" . __PACKAGE__ . "] $_" } @_;
+    }
+}
+
+# -------------------------------------------------------------------
+sub mk_name {
+    my ($basename, $type, $scope, $critical) = @_;
+    my $basename_orig = $basename;
+    my $max_name      = $type 
+                        ? $max_id_length - (length($type) + 1) 
+                        : $max_id_length;
+    $basename         = substr( $basename, 0, $max_name ) 
+                        if length( $basename ) > $max_name;
+    my $name          = $type ? "${type}_$basename" : $basename;
+
+    if ( $basename ne $basename_orig and $critical ) {
+        my $show_type = $type ? "+'$type'" : "";
+        warn "Truncating '$basename_orig'$show_type to $max_id_length ",
+            "character limit to make '$name'\n" if $WARN;
+        $truncated{ $basename_orig } = $name;
+    }
+
+    $scope ||= \%global_names;
+    if ( my $prev = $scope->{ $name } ) {
+        my $name_orig = $name;
+        $name        .= sprintf( "%02d", ++$prev );
+        substr($name, $max_id_length - 3) = "00" 
+            if length( $name ) > $max_id_length;
+
+        warn "The name '$name_orig' has been changed to ",
+             "'$name' to make it unique.\n" if $WARN;
+
+        $scope->{ $name_orig }++;
+    }
+
+    $scope->{ $name }++;
+    return $name;
+}
+
+1;
+__END__
+
+=head1 NAME
+
+SQL::Translator::Producer::SQLite - SQLite-specific producer for SQL::Translator
+
+=head1 AUTHOR
+
+Ken Y. Clark E<lt>kclark@cpan.orgE<gt>