Fixing default datatype if unknown.
[dbsrgits/SQL-Translator.git] / lib / SQL / Translator / Producer / Turnkey.pm
index db3d697..10029a5 100644 (file)
@@ -1,10 +1,10 @@
 package SQL::Translator::Producer::Turnkey;
 
 # -------------------------------------------------------------------
-# $Id: Turnkey.pm,v 1.1 2003-08-28 08:51:09 boconnor Exp $
+# $Id: Turnkey.pm,v 1.9 2003-10-15 16:48:39 kycl4rk Exp $
 # -------------------------------------------------------------------
 # Copyright (C) 2003 Allen Day <allenday@ucla.edu>,
-#                    Ying Zhang <zyolive@yahoo.com>
+#   Brian O'Connor <brian.oconnor@excite.com>.
 #
 # This program is free software; you can redistribute it and/or
 # modify it under the terms of the GNU General Public License as
@@ -23,15 +23,17 @@ package SQL::Translator::Producer::Turnkey;
 
 use strict;
 use vars qw[ $VERSION $DEBUG ];
-$VERSION = sprintf "%d.%02d", q$Revision: 1.1 $ =~ /(\d+)\.(\d+)/;
+$VERSION = sprintf "%d.%02d", q$Revision: 1.9 $ =~ /(\d+)\.(\d+)/;
 $DEBUG   = 1 unless defined $DEBUG;
 
 use SQL::Translator::Schema::Constants;
-use SQL::Translator::Utils qw(header_comment);
+use SQL::Translator::Schema::Graph;
+use SQL::Translator::Schema::Graph::HyperEdge;
+use Log::Log4perl; Log::Log4perl::init('/etc/log4perl.conf');
 use Data::Dumper;
 use Template;
 
-my %CDBI_auto_pkgs = (
+my %producer2dsn = (
     MySQL      => 'mysql',
     PostgreSQL => 'Pg',
     Oracle     => 'Oracle',
@@ -39,267 +41,312 @@ my %CDBI_auto_pkgs = (
 
 # -------------------------------------------------------------------
 sub produce {
+    my $log           = Log::Log4perl->get_logger('sql.translator');
+
     my $t             = shift;
        my $create        = undef;
-    local $DEBUG      = $t->debug;
-    my $no_comments   = $t->no_comments;
-    my $schema        = $t->schema;
     my $args          = $t->producer_args;
-    my $db_user       = $args->{'db_user'} || '';
-    my $db_pass       = $args->{'db_pass'} || '';
-    my $main_pkg_name = $args->{'main_pkg_name'} ||
-                           $t->format_package_name('DBI');
-    my $header        = header_comment(__PACKAGE__, "# ");
-    my $parser_type   = ( split /::/, $t->parser_type )[-1];
-    my $from          = $CDBI_auto_pkgs{ $parser_type } || '';
-    my $dsn           = $args->{'dsn'} || sprintf( 'dbi:%s:_',
-                            $CDBI_auto_pkgs{ $parser_type }
-                            ? $CDBI_auto_pkgs{ $parser_type } : $parser_type
-                        );
-    my $sep           = '# ' . '-' x 67;
+    my $no_comments   = $t->no_comments;
+       my $baseclass     = $args->{'main_pkg_name'} || $t->format_package_name('DBI');
+       my $graph         = SQL::Translator::Schema::Graph->new(translator => $t,
+                                                                                                                       baseclass => $baseclass
+                                                                                                                  );
 
-    #
-    # Identify "link tables" (have only PK and FK fields).
-    #
-    my %linkable;
-    my %linktable;
-    foreach my $table ( $schema->get_tables ) {
-        my $is_link = 1;
-        foreach my $field ( $table->get_fields ) {
-            unless ( $field->is_primary_key or $field->is_foreign_key ) {
-                $is_link = 0; 
-                last;
-            }
-        }
-
-        next unless $is_link;
-      
-        foreach my $left ( $table->get_fields ) {
-            next unless $left->is_foreign_key;
-            my $lfk           = $left->foreign_key_reference or next;
-            my $lr_table      = $schema->get_table( $lfk->reference_table )
-                                 or next;
-            my $lr_field_name = ($lfk->reference_fields)[0];
-            my $lr_field      = $lr_table->get_field($lr_field_name);
-            next unless $lr_field->is_primary_key;
-
-            foreach my $right ( $table->get_fields ) {
-                next if $left->name eq $right->name;
-        
-                my $rfk      = $right->foreign_key_reference or next;
-                my $rr_table = $schema->get_table( $rfk->reference_table )
-                               or next;
-                my $rr_field_name = ($rfk->reference_fields)[0];
-                my $rr_field      = $rr_table->get_field($rr_field_name);
-                next unless $rr_field->is_primary_key;
-        
-                $linkable{ $lr_table->name }{ $rr_table->name } = $table;
-                $linkable{ $rr_table->name }{ $lr_table->name } = $table;
-                $linktable{ $table->name } = $table;
-            }
-        }
-    }
+       my $parser_type   = (split /::/, $t->parser_type)[-1];
+
+    local $DEBUG      = $t->debug;
+
+       my %meta          = (
+                                                format_fk => $t->format_fk_name,
+                                                template  => $args->{'template'}      || '',
+                                                baseclass => $baseclass,
+                                                db_user   => $args->{'db_user'}       || '',
+                                                db_pass   => $args->{'db_pass'}       || '',
+                                                parser    => $t->parser_type,
+                                                producer  => __PACKAGE__,
+                                                dsn       => $args->{'dsn'} || sprintf( 'dbi:%s:_', $producer2dsn{ $parser_type }
+                                                                                                                                ? $producer2dsn{ $parser_type }
+                                                                                                                                : $parser_type
+                                                                                                                          )
+                                                );
 
     #
-    # Iterate over all tables
+    # create methods
     #
-    my ( %packages, $order );
-    for my $table ( $schema->get_tables ) {
-        my $table_name = $table->name or next;
-
-        my $table_pkg_name = $t->format_package_name($table_name);
-        $packages{ $table_pkg_name } = {
-            order     => ++$order,
-            pkg_name  => $table_pkg_name,
-            base      => $main_pkg_name,
-            table     => $table_name,
-        };
-
-        #
-        # Primary key may have a differenct accessor method name
-        #
-        if ( my $constraint = $table->primary_key ) {
-            my $field          = ($constraint->fields)[0];
-                       $packages{ $table_pkg_name }{'columns_primary'} = $field;
-
-            if ( my $pk_xform = $t->format_pk_name ) {
-                my $pk_name = $pk_xform->( $table_pkg_name, $field );
-
-                $packages{ $table_pkg_name }{'pk_accessor'} = 
-                    "#\n# Primary key accessor\n#\n".
-                    "sub $pk_name {\n    shift->$field\n}\n\n";
-                               
-            }
-        }
-
-        my $is_data = 0;
-        foreach my $field ( $table->get_fields ) {
-                 if ( !$field->is_foreign_key and !$field->is_primary_key ) {
-                       push @{ $packages{ $table_pkg_name }{'columns_essential'} }, $field->name;
-                       $is_data++;
-                 } elsif ( !$field->is_primary_key ) {
-                       push @{ $packages{ $table_pkg_name }{'columns_others'} }, $field->name;
-                 }
-               }
+       foreach my $node_from ($graph->node_values){
+
+         next unless $node_from->table->is_data or !$node_from->table->is_trivial_link;
+
+         foreach my $cedge ( $node_from->compoundedges ){
 
-               my %linked;
-               if ( $is_data ) {
-                 foreach my $link ( keys %{ $linkable{ $table_name } } ) {
-                       my $linkmethodname;
-
-                       if ( my $fk_xform = $t->format_fk_name ) {
-                         # ADD CALLBACK FOR PLURALIZATION MANGLING HERE
-                         $linkmethodname = $fk_xform->($linkable{$table_name}{$link}->name,
-                                                                                       ($schema->get_table($link)->primary_key->fields)[0]).'s';
-                       } else {
-                         # ADD CALLBACK FOR PLURALIZATION MANGLING HERE
-                         $linkmethodname = $linkable{$table_name}{$link}->name.'_'.
-                               ($schema->get_table($link)->primary_key->fields)[0].'s';
+               my $hyperedge = SQL::Translator::Schema::Graph::HyperEdge->new();
+
+               my $node_to;
+               foreach my $edge ($cedge->edges){
+                 if($edge->thisnode->name eq $node_from->name){
+                       $hyperedge->vianode($edge->thatnode);
+
+                       if($edge->thatnode->name ne $cedge->via->name){
+                         $node_to ||= $graph->node($edge->thatnode->table->name);
                        }
 
-                       my @rk_fields = ();
-                       my @lk_fields = ();
-                       foreach my $field ($linkable{$table_name}{$link}->get_fields) {
-                         next unless $field->is_foreign_key;
+                       $hyperedge->push_thisnode($edge->thisnode);
+                       $hyperedge->push_thisfield($edge->thisfield);
+                       $hyperedge->push_thisviafield($edge->thatfield);
 
-                         next unless(
-                                                 $field->foreign_key_reference->reference_table eq $table_name
-                                                 ||
-                                                 $field->foreign_key_reference->reference_table eq $link
-                                                );
-                         push @lk_fields, ($field->foreign_key_reference->reference_fields)[0]
-                               if $field->foreign_key_reference->reference_table eq $link;
-                         push @rk_fields, $field->name
-                               if $field->foreign_key_reference->reference_table eq $table_name;
+                 } else {
+                       if($edge->thisnode->name ne $cedge->via->name){
+                         $node_to ||= $graph->node($edge->thisnode->table->name);
                        }
+                       $hyperedge->push_thatnode($edge->thisnode);
+                       $hyperedge->push_thatfield($edge->thisfield);
+                       $hyperedge->push_thatviafield($edge->thatfield);
+                 }
+                 $log->debug($edge->thisfield->name);
+                 $log->debug($edge->thatfield->name);
+               }
 
-                       #if one possible traversal via link table
-                       if (scalar(@rk_fields) == 1 and scalar(@lk_fields) == 1) {
-                         foreach my $rk_field (@rk_fields) {
-                               #push @{ $packages{ $table_pkg_name }{'has_many'}{ $link } },
-                               push @{ $packages{ $table_pkg_name }{'has_many'}{$link}{'link_one_one'} },
-                                 "sub ".$linkmethodname." { my \$self = shift; ".
-                                       "return map \$_->".
-                                         ($schema->get_table($link)->primary_key->fields)[0].
-                                               ", \$self->".$linkable{$table_name}{$link}->name.
-                                                 "_".$rk_field." }\n\n";
-                               #push @{ $packages{ $table_pkg_name }{'has_many'}{ $link }{'one_one'} },
-                               #  {link_method_name => $linkmethodname, primary_key_field => ($schema->get_table($link)->primary_key->fields)[0],
-                               #   table_name => $linkable{$table_name}{$link}->name, rk_field => $rk_field};
-                         }
-                         #else there is more than one way to traverse it.  ack!
-                         #let's treat these types of link tables as a many-to-one (easier)
-                         #
-                         #NOTE: we need to rethink the link method name, as the cardinality
-                         #has shifted on us.
-                       } elsif (scalar(@rk_fields) == 1) {
-                         foreach my $rk_field (@rk_fields) {
-                               # ADD CALLBACK FOR PLURALIZATION MANGLING HERE
-                               #push @{ $packages{ $table_pkg_name }{'has_many'}{ $link } },
-                               push @{ $packages{ $table_pkg_name }{'has_many'}{ $link }{'link_many_one'} },
-                                 "sub " . $linkable{$table_name}{$link}->name .
-                                       "s { my \$self = shift; return \$self->" .
-                                         $linkable{$table_name}{$link}->name . "_" .
-                                               $rk_field . "(\@_) }\n\n";
-                               #push @{ $packages{ $table_pkg_name }{'has_many'}{ $link }{'many_one'} },
-                               #  {
-                               #    table_name => $linkable{$table_name}{$link}->name, rk_field => $rk_field
-                               #  };
-                         }
-                       } elsif (scalar(@lk_fields) == 1) {
-                         #these will be taken care of on the other end...
-                       } else {
-                         #many many many.  need multiple iterations here, data structure revision
-                         #to handle N FK sources.  This code has not been tested and likely doesn't
-                         #work here
-                         foreach my $rk_field (@rk_fields) {
-                               # ADD CALLBACK FOR PLURALIZATION MANGLING HERE
-                               #push @{ $packages{ $table_pkg_name }{'has_many'}{ $link } },
-                               push @{ $packages{ $table_pkg_name }{'has_many'}{ $link }{'link_many_many'} },
-                                 "sub " . $linkable{$table_name}{$link}->name . "_" . $rk_field .
-                                       "s { my \$self = shift; return \$self->" .
-                                         $linkable{$table_name}{$link}->name . "_" .
-                                               $rk_field . "(\@_) }\n\n";
-                               #push @{ $packages{ $table_pkg_name }{'has_many'}{ $link }{'many_many'} },
-                               #  {
-                               #   table_name => $linkable{$table_name}{$link}->name, rk_field => $rk_field
-                               #  };
-                         }
+                  if($hyperedge->count_thisnode == 1 and $hyperedge->count_thatnode == 1){ $hyperedge->type('one2one')   }
+               elsif($hyperedge->count_thisnode  > 1 and $hyperedge->count_thatnode == 1){ $hyperedge->type('many2one')  }
+               elsif($hyperedge->count_thisnode == 1 and $hyperedge->count_thatnode  > 1){ $hyperedge->type('one2many')  }
+               elsif($hyperedge->count_thisnode  > 1 and $hyperedge->count_thatnode  > 1){ $hyperedge->type('many2many') }
+
+#warn join "\n", sort keys %::SQL::Translator::Schema::Graph::HyperEdge::;
+
+               #node_to won't always be defined b/c of multiple edges to a single other node
+               if(defined($node_to)){
+                 $log->debug($node_from->name);
+                 $log->debug($node_to->name);
+
+                 if(scalar($hyperedge->thisnode) > 1){
+                       $log->debug($hyperedge->type ." via ". $hyperedge->vianode->name);
+                       my $i = 0;
+                       foreach my $thisnode ( $hyperedge->thisnode ){
+                         $log->debug($thisnode->name .' '.
+                                                 $hyperedge->thisfield_index(0)->name .' -> '.
+                                                 $hyperedge->thisviafield_index($i)->name .' '.
+                                                 $hyperedge->vianode->name .' '.
+                                                 $hyperedge->thatviafield_index(0)->name .' <- '.
+                                                 $hyperedge->thatfield_index(0)->name .' '.
+                                                 $hyperedge->thatnode_index(0)->name ."\n"
+                                                );
+                         $i++;
                        }
                  }
-        }
-
-
-        #
-        # Use foreign keys to set up "has_a/has_many" relationships.
-        #
-        foreach my $field ( $table->get_fields ) {
-            if ( $field->is_foreign_key ) {
-                my $table_name = $table->name;
-                my $field_name = $field->name;
-                my $fk_method  = $t->format_fk_name($table_name, $field_name);
-                my $fk         = $field->foreign_key_reference;
-                my $ref_table  = $fk->reference_table;
-                my $ref_pkg    = $t->format_package_name($ref_table);
-                my $ref_field  = ($fk->reference_fields)[0];
-
-                push @{ $packages{ $table_pkg_name }{'has_a'} },
-                    "$table_pkg_name->has_a(\n".
-                    "    $field_name => '$ref_pkg'\n);\n\n".
-                    "sub $fk_method {\n".
-                    "    return shift->$field_name\n}\n\n"
-                ;
-                               
-                               
-                #
-                # If this table "has a" to the other, then it follows 
-                # that the other table "has many" of this one, right?
-                #
-                               # No... there is the possibility of 1-1 cardinality
-
-                               #if there weren't M-M relationships via the has_many
-                               #being set up here, create nice pluralized method alias
-                               #rather for user as alt. to ugly tablename_fieldname name
-                               if(! $packages{ $ref_pkg }{ 'has_many' }{ $table_name } ){
-                                 # ADD CALLBACK FOR PLURALIZATION MANGLING HERE
-                                 #push @{ $packages{ $ref_pkg }{'has_many'}{ $table_name } },
-                               #       "sub $table_name\s {\n    return shift->$table_name\_$field_name\n}\n\n";
-                                 push @{ $packages{ $ref_pkg }{'has_many'}{ $table_name }{'fk_pluralized'} },
-                                       { table_name => $table_name, field_name => $field_name };
-
-                               #else ugly
-                               } else {
-                               }
-
-                               #push @{ $packages{ $ref_pkg }{'has_many'}{ $table_name } },
-                               #  "$ref_pkg->has_many(\n    '${table_name}_${field_name}', ".
-                               #  "'$table_pkg_name' => '$field_name'\n);\n\n";
-                               push @{ $packages{ $ref_pkg }{'has_many'}{ $table_name }{pluralized} },
-                                 { ref_pkg => $ref_pkg, table_pkg_name => $table_pkg_name, table_name => $table_name, field_name => $field_name };
-            }
+                 $node_from->push_hyperedges($hyperedge);
                }
-       }
+         }
+       }
+
+       $meta{"nodes"} = $graph->node;
+       return(translateForm($t, \%meta));
+}
 
-       my %metadata;
-       $metadata{"packages"} = \%packages;
-       $metadata{"linkable"} = \%linkable;
-       return(translateForm($t, \%metadata));
+sub translateForm {
+  my $t = shift;
+  my $meta = shift;
+  my $args = $t->producer_args;
+  my $type = $meta->{'template'};
+  my $tt2;
+  $tt2 = template($type);
+  my $template = Template->new({
+                                                               PRE_CHOMP => 1,
+                                                               POST_CHOMP => 0,
+                                                               EVAL_PERL => 1
+                                                          });
+
+  my $result;
+  $template->process(\$tt2, $meta, \$result) || die $template->error();
+  return($result);
 }
 
+1;
+
+# -------------------------------------------------------------------
+
+=pod
+
+=head1 NAME
+
+SQL::Translator::Producer::Turnkey - create Turnkey classes from schema
+
+=head1 SYNOPSIS
+
+Creates output for use with the Turnkey project.
+
+=head1 SEE ALSO
+
+L<http://turnkey.sourceforge.net>.
+
+=head1 AUTHORS
+
+Allen Day E<lt>allenday@ucla.eduE<gt>
+Brian O\'Connor E<lt>brian.oconnor@excite.comE<gt>.
+
+=cut
+
+sub template {
+  my $type = shift;
+
 ###########################################
 # Here documents for the tt2 templates    #
 ###########################################
 
-my $turnkey_atom_tt2 = <<'EOF';
+  if($type eq 'classdbi'){
+       return <<EOF;
+[% MACRO printPackage(node) BLOCK %]
+# --------------------------------------------
+
+package [% node.name %];
+use base '[% node.base %]';
+use Class::DBI::Pager;
+
+[% node.name %]->set_up_table('[% node.table.name %]');
+[% printPKAccessors(node.primary_key, node.table.name) %]
+[% printHasA(node.edges, node) %]
+[% printHasMany(node.edges, node) %]
+[% printHasCompound(node.compoundedges, node.hyperedges, node.name) %]
+[% #printHasFriendly(node) %]
+[% END %]
+
+[% MACRO printPKAccessors(array, name) BLOCK %]
+#
+# Primary key accessors
+#
+[% FOREACH item = array %]
+sub id { shift->[% item %] }
+sub [% name %] { shift->[% item %] }
+[% END %]
+
+[% END %]
+
+[% MACRO printHasA(edges, name) BLOCK %]
+#
+# Has A
+#
+
+[% FOREACH edge = edges %]
+  [% IF edge.type == 'import' %]
+[% node.name %]->has_a([% edge.thisfield.name %] => '[% edge.thatnode.name %]');
+    [% IF node.has(edge.thatnode.name) < 2 %]
+sub [% edge.thatnode.table.name %] { return shift->[% edge.thisfield.name %] }
+    [% ELSE %]
+sub [% format_fk(edge.thisnode.table.name,edge.thisfield.name) %] { return shift->[% edge.thisfield.name %] }
+    [% END %]
+
+  [% END %]
+[% END %]
+
+[% END %]
+
+[% MACRO printHasMany(edges, node) BLOCK %]
+#
+# Has Many
+#
+
+[% FOREACH edge = edges %]
+  [% IF edge.type == 'export' %]
+[% node.name %]->has_many([% edge.thatnode.table.name %]_[% edge.thatfield.name %], '[% edge.thatnode.name %]' => '[% edge.thatfield.name %]');
+    [% IF node.via(edge.thatnode.name) >= 1 %]
+sub [% edge.thatnode.table.name %]_[% format_fk(edge.thatnode.table.name,edge.thatfield.name) %]s { return shift->[% edge.thatnode.table.name %]_[% edge.thatfield.name %] }
+    [% ELSIF edge.thatnode.table.is_data %]
+      [% IF node.edgecount(edge.thatnode.name) > 1 %]
+sub [% edge.thatnode.table.name %]_[% format_fk(edge.thatnode.name,edge.thatfield.name) %]s { return shift->[% edge.thatnode.table.name %]_[% edge.thatfield.name %] }
+      [% ELSE %]
+sub [% edge.thatnode.table.name %]s { return shift->[% edge.thatnode.table.name %]_[% edge.thatfield.name %] }
+      [% END %]
+    [% END %]
+
+  [% END %]
+[% END %]
+
+[% END %]
+
+[% MACRO printHasCompound(cedges,hedges,name) BLOCK %]
+#
+# Has Compound Many
+#
+[% FOREACH cedge = cedges %]
+[% FOREACH edge = cedge.edges %]
+  [% NEXT IF edge.thisnode.name != name %]
+sub [% cedge.via.table.name %]_[% format_fk(edge.thatnode.table.name,edge.thatfield.name) %]s { return shift->[% cedge.via.table.name %]_[% edge.thatfield.name %] }
+[% END %]
+[% END %]
+
+[% FOREACH h = hedges %]
+########## [% h.type %] ##########
+  [% IF h.type == 'one2one' %]
+sub [% h.thatnode.table.name %]s { my \$self = shift; return map \$_->[% h.thatviafield.name %], \$self->[% h.vianode.table.name %]_[% h.thisviafield.name %] }
+
+  [% ELSIF h.type == 'one2many' %]
+    [% thisnode = h.thisnode_index(0) %]
+    [% i = 0 %]
+    [% FOREACH thatnode = h.thatnode %]
+#[% thisnode.name %]::[% h.thisfield_index(0).name %] -> [% h.vianode.name %]::[% h.thisviafield_index(i).name %] ... [% h.vianode.name %]::[% h.thatviafield_index(0).name %] <- [% h.thatnode_index(0).name %]::[% h.thatfield_index(0).name %]
+sub [% h.vianode.table.name %]_[% format_fk(h.vianode,h.thatviafield_index(0).name) %]s { my \$self = shift; return map \$_->[% h.thatviafield_index(0).name %], \$self->[% h.vianode.table.name %]_[% h.thisviafield_index(i).name %] }
+      [% i = i + 1 %]
+    [% END %]
+
+  [% ELSIF h.type == 'many2one' %]
+    [% i = 0 %]
+    [% FOREACH thisnode = h.thisnode %]
+#[% thisnode.name %]::[% h.thisfield_index(0).name %] -> [% h.vianode.name %]::[% h.thisviafield_index(i).name %] ... [% h.vianode.name %]::[% h.thatviafield_index(0).name %] <- [% h.thatnode_index(0).name %]::[% h.thatfield_index(0).name %]
+sub [% h.vianode.table.name %]_[% format_fk(h.vianode,h.thisviafield_index(i).name) %]_[% format_fk(h.vianode,h.thatviafield_index(0).name) %]s { my \$self = shift; return map \$_->[% h.thatviafield_index(0).name %], \$self->[% h.vianode.table.name %]_[% h.thisviafield_index(i).name %] }
+      [% i = i + 1 %]
+
+    [% END %]
+
+  [% ELSIF h.type == 'many2many' %]
+    [% i = 0 %]
+    [% FOREACH thisnode = h.thisnode %]
+      [% j = 0 %]
+      [% FOREACH thatnode = h.thatnode %]
+#[% thisnode.name %]::[% h.thisfield_index(i).name %] -> [% h.vianode.name %]::[% h.thisviafield_index(i).name %] ... [% h.vianode.name %]::[% h.thatviafield_index(j).name %] <- [% h.thatnode_index(j).name %]::[% h.thatfield_index(j).name %]
+sub [% h.vianode.table.name %]_[% format_fk(h.vianode,h.thisviafield_index(i).name) %]_[% format_fk(h.vianode,h.thatviafield_index(j).name) %]s { my \$self = shift; return map \$_->[% %], \$self->[% %] }
+        [% j = j + 1 %]
+
+      [% END %]
+      [% i = i + 1 %]
+    [% END %]
+  [% END %]
+[% END %]
+
+[% END %]
+
+[% MACRO printHasFriendly(node) BLOCK %]
+#
+# Has Friendly
+#
+hello, sailor!
+[% END %]
+
+[% MACRO printList(array) BLOCK %][% FOREACH item = array %][% item %] [% END %][% END %]
+package [% baseclass %];
+
+# Created by SQL::Translator::Producer::Turnkey
+# Template used: classdbi
+
+use strict;
+use base qw(Class::DBI::Pg);
+
+[% baseclass %]->set_db('Main', '[% db_str  %]', '[% db_user %]', '[% db_pass %]');
+
+[% FOREACH node = nodes %]
+    [% printPackage(node.value) %]
+[% END %]
+EOF
+
+} elsif($type eq 'atom'){
+  return <<'EOF';
 [% ###### DOCUMENT START ###### %]
 
-[% FOREACH package = linkable %]
+[% FOREACH node = linkable %]
 
 ##############################################
 
-package Durian::Atom::[% package.key FILTER ucfirst %];
+package Durian::Atom::[% node.key FILTER ucfirst %];
 
-[% pname = package.key FILTER ucfirst%]
+[% pname = node.key FILTER ucfirst%]
 [% pkey = "Durian::Model::${pname}" %]
 
 use base qw(Durian::Atom);
@@ -314,7 +361,7 @@ sub render {
        my $dbobject = shift;
     # Assumption here that if it's not rendering on it's own dbobject
     # then it's a list. This will be updated when AtomLists are implemented -boconnor
-       if(ref($dbobject) eq 'Durian::Model::[% package.key FILTER ucfirst %]') {
+       if(ref($dbobject) eq 'Durian::Model::[% node.key FILTER ucfirst %]') {
                return(_render_record($dbobject));
        }
        else { return(_render_list($dbobject)); }
@@ -325,7 +372,7 @@ sub _render_record {
        my @output = ();
        my $row = {};
        my $field_hash = {};
-       [% FOREACH field = packages.$pkey.columns_essential %]
+       [% FOREACH field = nodes.$pkey.columns_essential %]
        $field_hash->{[% field %]} = $dbobject->[% field %]();
     [% END %]
        $row->{data} = $field_hash;
@@ -337,12 +384,12 @@ sub _render_record {
 sub _render_list {
        my $dbobject = shift;
        my @output = ();
-       my @objects = $dbobject->[% package.key %]s;
+       my @objects = $dbobject->[% node.key %]s;
        foreach my $object (@objects)
     {
                my $row = {};
            my $field_hash = {};
-         [% FOREACH field = packages.$pkey.columns_essential %]
+         [% FOREACH field = nodes.$pkey.columns_essential %]
                $field_hash->{[% field %]} = $object->[% field %]();
          [% END %]
                $row->{data} = $field_hash;
@@ -361,94 +408,8 @@ sub head {
 [% END %]
 EOF
 
-my $turnkey_dbi_tt2 = <<EOF;
-[% #######  MACRO START ###### %]
-
-[% MACRO printPackage(package) BLOCK %]
-# --------------------------------------------
-package [% package.pkg_name %];
-use base '[% package.base %]';
-use Class::DBI::Pager;
-
-[% package.pkg_name %]->set_up_table('[% package.table %]');
-[% package.pkg_name %]->columns(Primary => qw/[% printList(package.columns_primary) %]/);
-[% package.pkg_name %]->columns(Essential => qw/[% printList(package.columns_essential) %]/);
-
-[% printPKAccessors(package.columns_primary, package.table) %]
-[% printHasMany(package.has_many, package.table) %]
-[% printHasA(package.has_a, package.pkg_name) %]
-
-[% END %]
-
-[% MACRO printPKAccessors(array, name) BLOCK %]
-#
-# Primary key accessor
-#
-[% FOREACH item = array %]
-sub [% name %] {
-  shift->[% item %];
-}
-[% END %]
-[% END %]
-
-[% MACRO printHasMany(hash, name) BLOCK %]
-#
-# Has Many
-#
-[% FOREACH group = hash %][% FOREACH item = group.value %][% FOREACH arr = item.value %]
-# Key: [% group.key %]
-# Relationship: [% item.key %]
-  [% IF item.key == 'fk_pluralized' %]
-sub [% arr.table_name -%]s {
-      return shift->[% arr.table_name %]_[% arr.field_name %]
-       };
-  [% ELSIF item.key == 'pluralized' %]
-[% arr.ref_pkg %]->has_many('[% arr.table_name %]_[% arr.field_name %]', '[% arr.table_pkg_name %]' => '[% arr.field_name %]');
-  [% ELSIF item.key == 'link_one_one' %]
-    [% FOREACH line = item.value %]
-[% line %]
-    [% END %]
-  [% ELSIF item.key == 'link_many_one' %]
-    [% FOREACH line = item.value %]
-[% line %]
-    [% END %]
-  [% ELSIF item.key == 'link_many_many' %]
-    [% FOREACH line = item.value %]
-[% line %]
-    [% END %]
-  [% END %]
-
-[% END %][% END %][% END %][% END %]
-
-[% MACRO printHasA(hash, pkg_name) BLOCK %]
-#
-# Has A
-#
-[% #name %]
-[% FOREACH item = hash %][% item %]
-[% END %][% END %]
-
-[% MACRO printList(array) BLOCK %][% FOREACH item = array %][% item %] [% END %][% END %]
-
-
-[% ###### DOCUMENT START ###### %]
-
-package Durian::Model::DBI;
-
-# Created by SQL::Translator::Producer::ClassDBI
-# Template used AutoDBI.tt2
-
-use strict;
-use base qw(Class::DBI::Pg);
-
-Durian::Model::DBI->set_db('Main', '[% db_str  %]', '[% db_user %]', '[% db_pass %]');
-
-[% FOREACH package = packages %]
-    [% printPackage(package.value) %]
-[% END %]
-EOF
-
-my $turnkey_xml_tt2 = <<EOF;
+} elsif($type eq 'xml'){
+  return <<EOF;
 <?xml version="1.0" encoding="UTF-8"?>
 <!DOCTYPE Durian SYSTEM "Durian.dtd">
 <Durian>
@@ -462,44 +423,46 @@ my $turnkey_xml_tt2 = <<EOF;
   </container>
 
 <!-- Atom Classes -->
-[% FOREACH package = linkable %]
-  <atom class="Durian::Atom::[% package.key FILTER ucfirst %]"  name="[% package.key FILTER ucfirst %]" xlink:label="[% package.key FILTER ucfirst %]Atom"/>
-[% END %]
+[% FOREACH node = linkable %]
+  <atom class="Durian::Atom::[% node.key FILTER ucfirst %]"  name="[% node.key FILTER ucfirst %]" xlink:label="[% node.key FILTER ucfirst %]Atom"/>
+[%- END -%]
 
 <!-- Atom Bindings -->
 <atomatombindings>
 [% FOREACH focus_atom = linkable %]
   [% FOREACH link_atom = focus_atom.value %]
   <atomatombinding xlink:from="#[% focus_atom.key FILTER ucfirst %]Atom" xlink:to="#[% link_atom.key FILTER ucfirst %]Atom" xlink:label="[% focus_atom.key FILTER ucfirst %]Atom2[% link_atom.key FILTER ucfirst %]Atom"/>
-  [% END %]
-[% END %]
+  [%- END -%]
+[%- END -%]
 </atomatombindings>
 
 <atomcontainerbindings>
-       [% FOREACH focus_atom = linkable %]
-       <atomcontainerbindingslayout xlink:label="Durian::Model::[% focus_atom.key FILTER ucfirst %]">
-         [% FOREACH link_atom = focus_atom.value %]
-         <atomcontainerbinding xlink:from="#MidLeftContainer" xlink:label="MidLeftContainer2[% link_atom.key FILTER ucfirst %]Atom"  xlink:to="#[% link_atom.key FILTER ucfirst %]Atom"/>
-         [% END %]
-         <atomcontainerbinding xlink:from="#MainContainer"    xlink:label="MainContainer2[% focus_atom.key FILTER ucfirst %]Atom"    xlink:to="#[% focus_atom.key FILTER ucfirst %]Atom"/>
-    </atomcontainerbindingslayout>
-       [% END %]
-   </atomcontainerbindings>
-
-  <uribindings>
-    <uribinding uri="/" class="Durian::Util::Frontpage"/>
-  </uribindings>
+[% FOREACH focus_atom = linkable %]
+  <atomcontainerbindingslayout xlink:label="Durian::Model::[% focus_atom.key FILTER ucfirst %]">
+  [% FOREACH link_atom = focus_atom.value %]
+    <atomcontainerbinding xlink:from="#MidLeftContainer" xlink:label="MidLeftContainer2[% link_atom.key FILTER ucfirst %]Atom"  xlink:to="#[% link_atom.key FILTER ucfirst %]Atom"/>
+  [%- END -%]
+  <atomcontainerbinding xlink:from="#MainContainer"    xlink:label="MainContainer2[% focus_atom.key FILTER ucfirst %]Atom"    xlink:to="#[% focus_atom.key FILTER ucfirst %]Atom"/>
+  </atomcontainerbindingslayout>
+  [%- END -%]
+</atomcontainerbindings>
+
+<uribindings>
+  <uribinding uri="/" class="Durian::Util::Frontpage"/>
+</uribindings>
+
+<classbindings>
+[% FOREACH focus_atom = linkable %]
+   <classbinding class="Durian::Model::[% focus_atom.key FILTER ucfirst %]" plugin="#[% focus_atom.key FILTER ucfirst %]Atom" rank="0"/>
+[%- END -%]
 
-  <classbindings>
-       [% FOREACH focus_atom = linkable %]
-     <classbinding class="Durian::Model::[% focus_atom.key FILTER ucfirst %]" plugin="#[% focus_atom.key FILTER ucfirst %]Atom" rank="0"/>
-       [% END %]
-  </classbindings>
+</classbindings>
 
 </Durian>
 EOF
 
-my $turnkey_template_tt2 = <<'EOF';
+} elsif($type eq 'template'){
+  return <<'EOF';
 [% TAGS [- -] %]
 [% MACRO renderpanel(panel,dbobject) BLOCK %]
   <!-- begin panel: [% panel.label %] -->
@@ -536,24 +499,24 @@ my $turnkey_template_tt2 = <<'EOF';
   <!-- end panel: [% panel.label %] -->
 [% END %]
 [% MACRO renderatom(atom, dbobject) SWITCH atom.name %]
-  [- FOREACH package = linkable -]
-    [% CASE '[- package.key FILTER ucfirst -]' %]
-      [% render[- package.key FILTER ucfirst -]Atom(atom.render(dbobject)) %]
+  [- FOREACH node = linkable -]
+    [% CASE '[- node.key FILTER ucfirst -]' %]
+      [% render[- node.key FILTER ucfirst -]Atom(atom.render(dbobject)) %]
   [- END -]
     [% CASE DEFAULT %]
       [% renderlist(atom.render(dbobject)) %]
 [% END %]
-[- FOREACH package = linkable -]
-[% MACRO render[- package.key FILTER ucfirst -]Atom(lstArr) BLOCK %]
+[- FOREACH node = linkable -]
+[% MACRO render[- node.key FILTER ucfirst -]Atom(lstArr) BLOCK %]
   [% FOREACH record = lstArr %]
     [% fields = record.data %]
-    [- pname = package.key FILTER ucfirst -]
+    [- pname = node.key FILTER ucfirst -]
     [- pkey = "Durian::Model::${pname}" -]
-    [- FOREACH field = packages.$pkey.columns_essential -]
+    [- FOREACH field = nodes.$pkey.columns_essential -]
       <tr><td><b>[- field -]</b></td><td>[% fields.[- field -] %]</td></tr>
     [- END -]
     [% id = record.id %]
-    <tr><td><a href="?id=[% id %];class=Durian::Model::[- package.key FILTER ucfirst -]">Link</a></td><td></td></tr>
+    <tr><td><a href="?id=[% id %];class=Durian::Model::[- node.key FILTER ucfirst -]">Link</a></td><td></td></tr>
   [% END %]
 [% END %]
 [- END -]
@@ -564,64 +527,7 @@ my $turnkey_template_tt2 = <<'EOF';
 [% END %]
 EOF
 
-sub translateForm
-{
-  my $t = shift;
-  my $output = shift;
-  my $args = $t->producer_args;
-  my $tt2     = $args->{'template'};
-  my $tt2Ref;
-  if ($tt2 eq 'atom') { $tt2Ref = \$turnkey_atom_tt2; }
-  if ($tt2 eq 'dbi') { $tt2Ref = \$turnkey_dbi_tt2; }
-  if ($tt2 eq 'xml') { $tt2Ref = \$turnkey_xml_tt2; }
-  if ($tt2 eq 'template') { $tt2Ref = \$turnkey_template_tt2; }
-
-  my $vars = {
-                               packages  => $output->{packages},
-                           linkable  => $output->{linkable},
-                           linktable => $output->{linktable},
-                           db_str    => $args->{db_str},
-                           db_user   => $args->{db_user},
-                           db_pass   => $args->{db_pass},
-  };
-  my $config = {
-      EVAL_PERL    => 1,               # evaluate Perl code blocks
-  };
-
-  # create Template object
-  my $template = Template->new($config);
-
-  my $result;
-  # specify input filename, or file handle, text reference, etc.
-  # process input template, substituting variables
-  $template->process($tt2Ref, $vars, \$result) || die $template->error();
-  return($result);
-}
-
 1;
 
-# -------------------------------------------------------------------
-
-=pod
-
-=head1 NAME
-
-SQL::Translator::Producer::ClassDBI - create Class::DBI classes from schema
-
-=head1 SYNOPSIS
-
-Use this producer as you would any other from SQL::Translator.  See
-L<SQL::Translator> for details.
-
-This package utilizes SQL::Translator's formatting methods
-format_package_name(), format_pk_name(), format_fk_name(), and
-format_table_name() as it creates classes, one per table in the schema
-provided.  An additional base class is also created for database connectivity
-configuration.  See L<Class::DBI> for details on how this works.
-
-=head1 AUTHORS
-
-Allen Day E<lt>allenday@ucla.eduE<gt>
-Ying Zhang E<lt>zyolive@yahoo.comE<gt>,
-Ken Y. Clark E<lt>kclark@cpan.orgE<gt>,
-Brian O'Connor E<lt>brian.oconnor@excite.comE<gt>.
+}
+}