1 package SQL::Translator::Producer::Turnkey;
3 # -------------------------------------------------------------------
4 # $Id: Turnkey.pm,v 1.1.2.8 2003-10-13 22:22:17 boconnor Exp $
5 # -------------------------------------------------------------------
6 # Copyright (C) 2003 Allen Day <allenday@ucla.edu>,
7 # Ying Zhang <zyolive@yahoo.com>
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.
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.
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
22 # -------------------------------------------------------------------
25 use vars qw[ $VERSION $DEBUG ];
26 $VERSION = sprintf "%d.%02d", q$Revision: 1.1.2.8 $ =~ /(\d+)\.(\d+)/;
27 $DEBUG = 1 unless defined $DEBUG;
29 use SQL::Translator::Schema::Constants;
30 use SQL::Translator::Utils qw(header_comment);
34 my %CDBI_auto_pkgs = (
40 # -------------------------------------------------------------------
44 local $DEBUG = $t->debug;
45 my $no_comments = $t->no_comments;
46 my $schema = $t->schema;
47 my $args = $t->producer_args;
48 my $db_user = $args->{'db_user'} || '';
49 my $db_pass = $args->{'db_pass'} || '';
50 my $main_pkg_name = $args->{'main_pkg_name'} ||
51 $t->format_package_name('DBI');
52 my $header = header_comment(__PACKAGE__, "# ");
53 my $parser_type = ( split /::/, $t->parser_type )[-1];
54 my $from = $CDBI_auto_pkgs{ $parser_type } || '';
55 my $dsn = $args->{'dsn'} || sprintf( 'dbi:%s:_',
56 $CDBI_auto_pkgs{ $parser_type }
57 ? $CDBI_auto_pkgs{ $parser_type } : $parser_type
59 my $sep = '# ' . '-' x 67;
62 # Identify "link tables" (have only PK and FK fields).
66 foreach my $table ( $schema->get_tables ) {
68 foreach my $field ( $table->get_fields ) {
69 unless ( $field->is_primary_key or $field->is_foreign_key ) {
77 foreach my $left ( $table->get_fields ) {
78 next unless $left->is_foreign_key;
79 my $lfk = $left->foreign_key_reference or next;
80 my $lr_table = $schema->get_table( $lfk->reference_table )
82 my $lr_field_name = ($lfk->reference_fields)[0];
83 my $lr_field = $lr_table->get_field($lr_field_name);
84 next unless $lr_field->is_primary_key;
86 foreach my $right ( $table->get_fields ) {
87 next if $left->name eq $right->name;
89 my $rfk = $right->foreign_key_reference or next;
90 my $rr_table = $schema->get_table( $rfk->reference_table )
92 my $rr_field_name = ($rfk->reference_fields)[0];
93 my $rr_field = $rr_table->get_field($rr_field_name);
94 next unless $rr_field->is_primary_key;
96 $linkable{ $lr_table->name }{ $rr_table->name } = $table;
97 $linkable{ $rr_table->name }{ $lr_table->name } = $table;
98 $linktable{ $table->name } = $table;
104 # Iterate over all tables
106 my ( %packages, $order );
107 for my $table ( $schema->get_tables ) {
108 my $table_name = $table->name or next;
110 my $table_pkg_name = $t->format_package_name($table_name);
111 $packages{ $table_pkg_name } = {
113 pkg_name => $table_pkg_name,
114 base => $main_pkg_name,
115 table => $table_name,
119 # Primary key may have a differenct accessor method name
121 if ( my $constraint = $table->primary_key ) {
122 my $field = ($constraint->fields)[0];
123 $packages{ $table_pkg_name }{'columns_primary'} = $field;
125 if ( my $pk_xform = $t->format_pk_name ) {
126 my $pk_name = $pk_xform->( $table_pkg_name, $field );
128 $packages{ $table_pkg_name }{'pk_accessor'} =
129 "#\n# Primary key accessor\n#\n".
130 "sub $pk_name {\n shift->$field\n}\n\n";
136 foreach my $field ( $table->get_fields ) {
137 if ( !$field->is_foreign_key and !$field->is_primary_key ) {
138 push @{ $packages{ $table_pkg_name }{'columns_essential'} }, $field->name;
140 } elsif ( !$field->is_primary_key ) {
141 push @{ $packages{ $table_pkg_name }{'columns_others'} }, $field->name;
147 foreach my $link ( keys %{ $linkable{ $table_name } } ) {
150 if ( my $fk_xform = $t->format_fk_name ) {
151 # ADD CALLBACK FOR PLURALIZATION MANGLING HERE
152 $linkmethodname = $fk_xform->($linkable{$table_name}{$link}->name,
153 ($schema->get_table($link)->primary_key->fields)[0]).'s';
155 # ADD CALLBACK FOR PLURALIZATION MANGLING HERE
156 $linkmethodname = $linkable{$table_name}{$link}->name.'_'.
157 ($schema->get_table($link)->primary_key->fields)[0].'s';
162 foreach my $field ($linkable{$table_name}{$link}->get_fields) {
163 next unless $field->is_foreign_key;
166 $field->foreign_key_reference->reference_table eq $table_name
168 $field->foreign_key_reference->reference_table eq $link
170 push @lk_fields, ($field->foreign_key_reference->reference_fields)[0]
171 if $field->foreign_key_reference->reference_table eq $link;
172 push @rk_fields, $field->name
173 if $field->foreign_key_reference->reference_table eq $table_name;
176 #if one possible traversal via link table
177 if (scalar(@rk_fields) == 1 and scalar(@lk_fields) == 1) {
178 foreach my $rk_field (@rk_fields) {
179 #push @{ $packages{ $table_pkg_name }{'has_many'}{ $link } },
180 push @{ $packages{ $table_pkg_name }{'has_many'}{$link}{'link_one_one'} },
181 "sub ".$linkmethodname." { my \$self = shift; ".
183 ($schema->get_table($link)->primary_key->fields)[0].
184 ", \$self->".$linkable{$table_name}{$link}->name.
185 "_".$rk_field." }\n\n";
186 #push @{ $packages{ $table_pkg_name }{'has_many'}{ $link }{'one_one'} },
187 # {link_method_name => $linkmethodname, primary_key_field => ($schema->get_table($link)->primary_key->fields)[0],
188 # table_name => $linkable{$table_name}{$link}->name, rk_field => $rk_field};
190 if(!$linktable{$table_name} and !$linktable{$link}){ $linkable{$table_name}{$link} ||= 1; }
193 #else there is more than one way to traverse it. ack!
194 #let's treat these types of link tables as a many-to-one (easier)
196 #NOTE: we need to rethink the link method name, as the cardinality
198 } elsif (scalar(@rk_fields) == 1) {
199 foreach my $rk_field (@rk_fields) {
200 # ADD CALLBACK FOR PLURALIZATION MANGLING HERE
201 #push @{ $packages{ $table_pkg_name }{'has_many'}{ $link } },
202 push @{ $packages{ $table_pkg_name }{'has_many'}{ $link }{'link_many_one'} },
203 "sub " . $linkable{$table_name}{$link}->name .
204 "s { my \$self = shift; return \$self->" .
205 $linkable{$table_name}{$link}->name . "_" .
206 $rk_field . "(\@_) }\n\n";
207 #push @{ $packages{ $table_pkg_name }{'has_many'}{ $link }{'many_one'} },
209 # table_name => $linkable{$table_name}{$link}->name, rk_field => $rk_field
212 if(!$linktable{$table_name} and !$linktable{$link}){ $linkable{$table_name}{$link} ||= 1; }
215 } elsif (scalar(@lk_fields) == 1) {
216 #these will be taken care of on the other end...
218 #many many many. need multiple iterations here, data structure revision
219 #to handle N FK sources. This code has not been tested and likely doesn't
221 foreach my $rk_field (@rk_fields) {
222 # ADD CALLBACK FOR PLURALIZATION MANGLING HERE
223 #push @{ $packages{ $table_pkg_name }{'has_many'}{ $link } },
224 push @{ $packages{ $table_pkg_name }{'has_many'}{ $link }{'link_many_many'} },
225 "sub " . $linkable{$table_name}{$link}->name . "_" . $rk_field .
226 "s { my \$self = shift; return \$self->" .
227 $linkable{$table_name}{$link}->name . "_" .
228 $rk_field . "(\@_) }\n\n";
229 #push @{ $packages{ $table_pkg_name }{'has_many'}{ $link }{'many_many'} },
231 # table_name => $linkable{$table_name}{$link}->name, rk_field => $rk_field
234 if(!$linktable{$table_name} and !$linktable{$link}){ $linkable{$table_name}{$link} ||= 1; }
243 # Use foreign keys to set up "has_a/has_many" relationships.
245 foreach my $field ( $table->get_fields ) {
246 if ( $field->is_foreign_key ) {
247 my $table_name = $table->name;
248 my $field_name = $field->name;
249 my $fk_method = $t->format_fk_name($table_name, $field_name);
250 my $fk = $field->foreign_key_reference;
251 my $ref_table = $fk->reference_table;
252 my $ref_pkg = $t->format_package_name($ref_table);
253 my $ref_field = ($fk->reference_fields)[0];
255 push @{ $packages{ $table_pkg_name }{'has_a'} },
256 "$table_pkg_name->has_a(\n".
257 " $field_name => '$ref_pkg'\n);\n\n".
258 "sub $fk_method {\n".
259 " return shift->$field_name\n}\n\n"
262 if(!$linktable{$ref_table} and !$linktable{$table_name}){ $linkable{$ref_table}{$table_name} ||= 1; }
265 # If this table "has a" to the other, then it follows
266 # that the other table "has many" of this one, right?
268 # No... there is the possibility of 1-1 cardinality
270 #if there weren't M-M relationships via the has_many
271 #being set up here, create nice pluralized method alias
272 #rather for user as alt. to ugly tablename_fieldname name
273 if(! $packages{ $ref_pkg }{ 'has_many' }{ $table_name } ){
274 # ADD CALLBACK FOR PLURALIZATION MANGLING HERE
275 #push @{ $packages{ $ref_pkg }{'has_many'}{ $table_name } },
276 # "sub $table_name\s {\n return shift->$table_name\_$field_name\n}\n\n";
277 push @{ $packages{ $ref_pkg }{'has_many'}{ $table_name }{'fk_pluralized'} },
278 { table_name => $table_name, field_name => $field_name };
280 if(!$linktable{$ref_table} and !$linktable{$table_name}){ $linkable{$ref_table}{$table_name} ||= 1; }
286 #push @{ $packages{ $ref_pkg }{'has_many'}{ $table_name } },
287 # "$ref_pkg->has_many(\n '${table_name}_${field_name}', ".
288 # "'$table_pkg_name' => '$field_name'\n);\n\n";
289 push @{ $packages{ $ref_pkg }{'has_many'}{ $table_name }{pluralized} },
290 { ref_pkg => $ref_pkg, table_pkg_name => $table_pkg_name, table_name => $table_name, field_name => $field_name };
292 if(!$linktable{$ref_table} and !$linktable{$table_name}){ $linkable{$ref_table}{$table_name} ||= 1; }
298 $metadata{"packages"} = \%packages;
299 $metadata{"linkable"} = \%linkable;
300 return(translateForm($t, \%metadata));
303 ###########################################
304 # Here documents for the tt2 templates #
305 ###########################################
307 my $turnkey_atom_tt2 = <<'EOF';
308 [% ###### DOCUMENT START ###### %]
310 [% FOREACH package = linkable %]
312 ##############################################
314 package Turnkey::Atom::[% package.key FILTER ucfirst %];
316 [% pname = package.key FILTER ucfirst%]
317 [% pkey = "Turnkey::Model::${pname}" %]
319 use base qw(Turnkey::Atom);
328 my $dbobject = shift;
329 # Assumption here that if it's not rendering on it's own dbobject
330 # then it's a list. This will be updated when AtomLists are implemented -boconnor
331 if(ref($dbobject) eq 'Turnkey::Model::[% package.key FILTER ucfirst %]') {
333 return(_render_record($dbobject));
335 else { return(_render_list($dbobject)); }
339 my $dbobject = shift;
343 [% FOREACH field = packages.$pkey.columns_essential %]
344 $field_hash->{[% field %]} = $dbobject->[% field %]();
346 [% FOREACH field = packages.$pkey.columns_others %]
347 $field_hash->{[% field %]} = $dbobject->[% field %]();
349 $row->{data} = $field_hash;
350 $row->{id} = $dbobject->id();
356 my $dbobject = shift;
358 my @objects = defined $dbobject ? $dbobject->[% package.key %]s : ();
359 foreach my $object (@objects)
363 [% FOREACH field = packages.$pkey.columns_essential %]
364 $field_hash->{[% field %]} = $object->[% field %]();
366 [% FOREACH field = packages.$pkey.columns_others %]
367 $field_hash->{[% field %]} = $object->[% field %]();
370 $row->{data} = $field_hash;
371 $row->{id} = $object->id();
387 my $turnkey_dbi_tt2 = <<EOF;
388 [% ####### MACRO START ###### %]
390 [% MACRO printPackage(package) BLOCK %]
391 # --------------------------------------------
392 package [% package.pkg_name %];
393 use base '[% package.base %]';
394 use Class::DBI::Pager;
396 [% package.pkg_name %]->set_up_table('[% package.table %]');
397 [% package.pkg_name %]->columns(Primary => qw/[% printList(package.columns_primary) %]/);
398 [% package.pkg_name %]->columns(Essential => qw/[% printList(package.columns_essential) %]/);
400 [% printPKAccessors(package.columns_primary, package.table) %]
401 [% printHasMany(package.has_many, package.table) %]
402 [% printHasA(package.has_a, package.pkg_name) %]
406 [% MACRO printPKAccessors(array, name) BLOCK %]
408 # Primary key accessor
410 [% FOREACH item = array %]
417 [% MACRO printHasMany(hash, name) BLOCK %]
421 [% FOREACH group = hash %][% FOREACH item = group.value %][% FOREACH arr = item.value %]
422 # Key: [% group.key %]
423 # Relationship: [% item.key %]
424 [% IF item.key == 'fk_pluralized' %]
425 sub [% arr.table_name -%]s {
426 return shift->[% arr.table_name %]_[% arr.field_name %]
428 [% ELSIF item.key == 'pluralized' %]
429 [% arr.ref_pkg %]->has_many('[% arr.table_name %]_[% arr.field_name %]', '[% arr.table_pkg_name %]' => '[% arr.field_name %]');
430 [% ELSIF item.key == 'link_one_one' %]
431 [% FOREACH line = item.value %]
434 [% ELSIF item.key == 'link_many_one' %]
435 [% FOREACH line = item.value %]
438 [% ELSIF item.key == 'link_many_many' %]
439 [% FOREACH line = item.value %]
444 [% END %][% END %][% END %][% END %]
446 [% MACRO printHasA(hash, pkg_name) BLOCK %]
451 [% FOREACH item = hash %][% item %]
454 [% MACRO printList(array) BLOCK %][% FOREACH item = array %][% item %] [% END %][% END %]
457 [% ###### DOCUMENT START ###### %]
459 package Turnkey::Model::DBI;
461 # Created by SQL::Translator::Producer::ClassDBI
462 # Template used AutoDBI.tt2
465 use base qw(Class::DBI::Pg);
467 Turnkey::Model::DBI->set_db('Main', '[% db_str %]', '[% db_user %]', '[% db_pass %]');
469 sub search_ilike { shift->_do_search(ILIKE => @_) }
471 [% FOREACH package = packages %]
472 [% printPackage(package.value) %]
476 my $turnkey_xml_tt2 = <<EOF;
477 <?xml version="1.0" encoding="UTF-8"?>
478 <!DOCTYPE Turnkey SYSTEM "Turnkey.dtd">
481 <!-- The basic layout is fixed -->
482 <container bgcolor="#dedbde" cellpadding="0" cellspacing="0" height="90%" orientation="vertical" type="root" width="100%" xlink:label="RootContainer">
483 <container cellpadding="3" cellspacing="0" orientation="horizontal" type="container" height="100%" width="100%" xlink:label="MiddleContainer">
484 <container align="center" cellpadding="2" cellspacing="0" class="leftbar" orientation="vertical" type="minor" width="0%" xlink:label="MidLeftContainer"/>
485 <container cellpadding="0" cellspacing="0" orientation="vertical" width="100%" type="major" xlink:label="MainContainer"/>
489 <!-- Atom Classes -->
490 [% FOREACH package = linkable %]
491 [% class = package.key | replace('Turnkey::Model::',''); class = class FILTER ucfirst; %]
492 <atom class="Turnkey::Atom::[% class %]" name="[% class %]" xlink:label="[% class %]Atom"/>
495 <!-- Atom Bindings -->
497 [% FOREACH focus_atom = linkable %]
498 [% FOREACH link_atom = focus_atom.value %]
499 [% class = focus_atom.key | replace('Turnkey::Model::',''); class = class FILTER ucfirst; %]
500 <atomatombinding xlink:from="#[% class %]Atom" xlink:to="#[% link_atom.key FILTER ucfirst %]Atom" xlink:label="[% class %]Atom2[% link_atom.key FILTER ucfirst %]Atom"/>
505 <atomcontainerbindings>
506 [% FOREACH focus_atom = linkable %]
507 <atomcontainerbindingslayout xlink:label="Turnkey::Model::[% focus_atom.key FILTER ucfirst %]">
508 [% FOREACH link_atom = focus_atom.value %]
509 <atomcontainerbinding xlink:from="#MidLeftContainer" xlink:label="MidLeftContainer2[% link_atom.key FILTER ucfirst %]Atom" xlink:to="#[% link_atom.key FILTER ucfirst %]Atom"/>
511 <atomcontainerbinding xlink:from="#MainContainer" xlink:label="MainContainer2[% focus_atom.key FILTER ucfirst %]Atom" xlink:to="#[% focus_atom.key FILTER ucfirst %]Atom"/>
512 </atomcontainerbindingslayout>
514 </atomcontainerbindings>
517 <uribinding uri="/" class="Turnkey::Util::Frontpage"/>
521 [% FOREACH focus_atom = linkable %]
522 [% class = focus_atom.key | replace('Turnkey::Model::','') ; class = class FILTER ucfirst %]
523 <classbinding class="Turnkey::Model::[% class %]" plugin="#[% class %]Atom" rank="0"/>
530 my $turnkey_template_tt2 = <<'EOF';
532 [% MACRO renderpanel(panel,dbobject) BLOCK %]
533 <!-- begin panel: [% panel.label %] -->
534 <table border="0" width="[% panel.width %]" height="[% panel.height %]" bgcolor="[% panel.bgcolor %]" valign="top" cellpadding="[% panel.cellpadding %]" cellspacing="[% panel.cellspacing %]" align="[% panel.align %]" valign="[% panel.valign %]">
536 [% FOREACH p = panel.containers %]
537 [% IF p.can_render(panel) %]
538 <td valign="top" class="[% p.class %]" align="[% panel.align %]" height="[% p.height || 1 %]" width="[% p.width %]">
539 [% IF p.type == 'Container' %]
540 [% renderpanel(p,dbobject) %]
542 <table cellpadding="2" cellspacing="0" align="left" height="100%" width="100%">
544 <tr bgcolor="#5aa5e2" height="1">
545 <td><font class="font" color="#FFFFFF">[% p.name %][% IF panel.type == 'major' %]: [% dbobject.name %][% END %]</font></td>
546 <td align="right" width="0"><!--<nobr><img src="/images/v.gif"/><img src="/images/^.gif"/>[% IF p.delible == 'yes' %]<img src="/images/x.gif"/>[% END %]</nobr>--></td>
549 <tr><td colspan="2" class="font" bgcolor="#FFFFFF">
550 <!-- begin atom: [% p.label %] -->
551 <table cellpadding="2" cellspacing="0" align="left" height="100%" width="100%">
552 [% renderatom(p,dbobject) %]
557 [% IF panel.orientation == 'vertical' %]
564 <!-- end panel: [% panel.label %] -->
566 [% BLOCK make_linked_dbobject %]
568 $stash->set(linked_dbobject => [% class %]->retrieve([% id %]));
571 [% MACRO obj2link(obj) SWITCH ref(obj) %]
575 <a href="[% obj2url(obj) %]">[% obj %]</a>
577 [% MACRO obj2url(obj) SWITCH obj %]
579 /?id=[% obj %];class=[% ref(obj) %]
581 [% MACRO obj2desc(obj) SWITCH ref(obj) %]
587 [% MACRO renderatom(atom, dbobject) SWITCH atom.name %]
588 [- FOREACH package = linkable -]
589 [% CASE '[- package.key FILTER ucfirst -]' %]
590 [% render[- package.key FILTER ucfirst -]Atom(atom, dbobject) %]
593 [% renderlist(atom, dbobject) %]
595 [- FOREACH package = linkable -]
596 [% MACRO render[- package.key FILTER ucfirst -]Atom(atom, dbobject) BLOCK %]
597 [% lstArr = atom.render(dbobject) %]
599 [% IF atom.focus == "yes" %]
600 [% FOREACH record = lstArr %]
601 [% field = record.data %]
602 [- pname = package.key FILTER ucfirst -]
603 [- pkey = "Turnkey::Model::${pname}" -]
606 [- FOREACH field = packages.$pkey.columns_essential.sort -]
608 <tr><td><b>[- field -]</b></td><td>[% obj2link(field.[- field -]) %]</td></tr>
611 <tr><td><b>[- field -]</b></td><td>[% obj2link(field.[- field -]) %]</td></tr>
614 [- FOREACH field = packages.$pkey.columns_others.sort -]
615 <tr><td><b>[- field -]</b></td><td>[% obj2link(field.[- field -]) %]</td></tr>
617 [% IF (rowcount > 1) %] <tr><td colspan="2"><hr></td></tr> [% END %]
618 [% rowcount = rowcount + 1 %]
622 [% FOREACH record = lstArr %]
623 [% class = ref(atom) | replace('::Atom::', '::Model::') %]
625 [% PROCESS make_linked_dbobject %]
626 <li style="margin-left: -20px; list-style: circle;">[% obj2link(linked_dbobject) %]</li>
632 [% MACRO renderlist(atom, dbobject) BLOCK %]
633 [% lstArr = atom.render(dbobject) %]
634 [% FOREACH item = lstArr %]
644 my $args = $t->producer_args;
645 my $tt2 = $args->{'template'};
647 if ($tt2 eq 'atom') { $tt2Ref = \$turnkey_atom_tt2; }
648 if ($tt2 eq 'dbi') { $tt2Ref = \$turnkey_dbi_tt2; }
649 if ($tt2 eq 'xml') { $tt2Ref = \$turnkey_xml_tt2; }
650 if ($tt2 eq 'template') { $tt2Ref = \$turnkey_template_tt2; }
653 packages => $output->{packages},
654 linkable => $output->{linkable},
655 linktable => $output->{linktable},
656 db_str => $args->{db_str},
657 db_user => $args->{db_user},
658 db_pass => $args->{db_pass},
661 EVAL_PERL => 1, # evaluate Perl code blocks
664 # create Template object
665 my $template = Template->new($config);
668 # specify input filename, or file handle, text reference, etc.
669 # process input template, substituting variables
670 $template->process($tt2Ref, $vars, \$result) || die $template->error();
676 # -------------------------------------------------------------------
682 SQL::Translator::Producer::ClassDBI - create Class::DBI classes from schema
686 Use this producer as you would any other from SQL::Translator. See
687 L<SQL::Translator> for details.
689 This package utilizes SQL::Translator's formatting methods
690 format_package_name(), format_pk_name(), format_fk_name(), and
691 format_table_name() as it creates classes, one per table in the schema
692 provided. An additional base class is also created for database connectivity
693 configuration. See L<Class::DBI> for details on how this works.
697 Allen Day E<lt>allenday@ucla.eduE<gt>
698 Ying Zhang E<lt>zyolive@yahoo.comE<gt>,
699 Ken Y. Clark E<lt>kclark@cpan.orgE<gt>,
700 Brian O'Connor E<lt>brian.oconnor@excite.comE<gt>.