lib/SQL/Translator/Producer/ClassDBI.pm

   1 package SQL::Translator::Producer::ClassDBI;
   2
   3 # -------------------------------------------------------------------
   4 # $Id: ClassDBI.pm,v 1.5 2003-04-25 11:47:25 dlc Exp $
   5 # -------------------------------------------------------------------
   6 # Copyright (C) 2003 Ying Zhang <zyolive@yahoo.com>,
   7 #                    Allen Day <allenday@ucla.edu>,
   8 #
   9 # This program is free software; you can redistribute it and/or
  10 # modify it under the terms of the GNU General Public License as
  11 # published by the Free Software Foundation; version 2.
  12 #
  13 # This program is distributed in the hope that it will be useful, but
  14 # WITHOUT ANY WARRANTY; without even the implied warranty of
  15 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16 # General Public License for more details.
  17 #
  18 # You should have received a copy of the GNU General Public License
  19 # along with this program; if not, write to the Free Software
  20 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
  21 # 02111-1307  USA
  22 # -------------------------------------------------------------------
  23
  24 use strict;
  25 use vars qw[ $VERSION $DEBUG ];
  26 $VERSION = sprintf "%d.%02d", q$Revision: 1.5 $ =~ /(\d+)\.(\d+)/;
  27 $DEBUG   = 1 unless defined $DEBUG;
  28
  29 use SQL::Translator::Utils qw(header_comment);
  30 use Data::Dumper;
  31
  32 sub produce {
  33   my ($translator, $data) = @_;
  34   $DEBUG                  = $translator->debug;
  35   my $no_comments         = $translator->no_comments;
  36
  37   my $create;
  38   $create .= header_comment(__PACKAGE__, "## ") unless ($no_comments);
  39
  40   $create .= "package " . $translator->format_package_name('DBI'). ";\n\n";
  41
  42   $create .= "my \$USER = \'\';\n";
  43   $create .= "my \$PASS = \'\';\n\n";
  44
  45   my $from = _from($translator->parser_type());
  46
  47   $create .= "use base \'Class::DBI::" .$from. "\';\n\n";
  48
  49   $create .= $translator->format_package_name('DBI'). "->set_db(\'Main', \'dbi:" .$from. ":_\', \$USER,\$PASS,);\n\n";
  50   $create .= "1;\n\n\n";
  51
  52   for my $table (keys %{$data}) {
  53         my $table_data = $data->{$table};
  54         my @fields =  keys %{$table_data->{'fields'}};
  55
  56
  57         $create .= "##\n## Package: " .$translator->format_package_name($table). "\n##\n" unless $no_comments;
  58         $create .= "package ". $translator->format_package_name($table). ";\n";
  59
  60         $create .= "use base \'Chado::DBI\';\n";
  61         $create .= "use mixin \'Class::DBI::Join\';\n";
  62         $create .= "use Class::DBI::Pager;\n\n";
  63
  64         $create .= $translator->format_package_name($table). "->set_up_table('$table');\n\n";
  65
  66
  67
  68         #
  69         # Primary key?
  70         #
  71         my @constraints;
  72
  73         for my $constraint ( @{ $table_data->{'constraints'} } ) {
  74           #my $name       = $constraint->{'name'} || '';
  75           my $type       = $constraint->{'type'};
  76           my $fields     = $constraint->{'fields'};
  77           my $ref_table  = $constraint->{'reference_table'};
  78           my $ref_fields = $constraint->{'reference_fields'};
  79
  80           if ( $type eq 'primary_key') {
  81                 $create .= "sub " .$translator->format_pk_name($translator->format_package_name($table), $fields[0]). "{ shift->$fields[0] }\n\n";
  82           }
  83
  84         }
  85
  86         #
  87         # Foreign key?
  88         #
  89         for (my $i = 0; $i < scalar(@fields); $i++) {
  90           my $field = $fields[$i];
  91           my $field_data = $table_data->{'fields'}->{$field}->{'constraints'};
  92           my $type = $field_data->[1]->{'type'} || '';
  93           my $ref_table = $field_data->[1]->{'reference_table'};
  94           my $ref_fields = $field_data->[1]->{'reference_fields'};
  95
  96           if ($type eq 'foreign_key') {
  97                 $create .= $translator->format_package_name($table). "->hasa(" .$translator->format_package_name($ref_table). " => \'@$ref_fields\');\n";
  98                 $create .= "sub " .$translator->format_fk_name($ref_table, @$ref_fields). "{ return shift->@$ref_fields }\n\n";
  99           }
 100         }
 101
 102         $create .= "1;\n\n\n";
 103   }
 104
 105   return $create;
 106 }
 107
 108
 109 sub _from {
 110   my $from = shift;
 111   my @temp = split(/::/, $from);
 112   $from = $temp[$#temp];
 113
 114   if ( $from eq 'MySQL') {
 115         $from = lc($from);
 116   } elsif ( $from eq 'PostgreSQL') {
 117         $from = 'Pg';
 118   } elsif ( $from eq 'Oracle') {
 119         $from = 'Oracle';
 120   } else {
 121         die "__PACKAGE__ can't handle vendor $from";
 122   }
 123
 124   return $from;
 125 }
 126
 127 1;
 128
 129 __END__
 130
 131 =head1 NAME
 132
 133 SQL::Translator::Producer::ClassDBI - Translate SQL schemata into Class::DBI classes
 134
 135 =head1 SYNOPSIS
 136
 137 Use this producer as you would any other from SQL::Translator.  See
 138 L<SQL::Translator> for details.
 139
 140 This package utilizes SQL::Translator's formatting methods
 141 format_package_name(), format_pk_name(), format_fk_name(), and
 142 format_table_name() as it creates classes, one per table in the schema
 143 provided.  An additional base class is also created for database connectivity
 144 configuration.  See L<Class::DBI> for details on how this works.
 145
 146 =head1 AUTHOR
 147
 148 Ying Zhang <zyolive@yahoo.com>, Allen Day <allenday@ucla.edu>