Commit | Line | Data |
f42e7027 |
1 | package SQL::Translator::Producer::ClassDBI; |
2 | |
3 | # ------------------------------------------------------------------- |
b6c0195c |
4 | # $Id: ClassDBI.pm,v 1.21 2003-06-25 19:10:17 allenday Exp $ |
f42e7027 |
5 | # ------------------------------------------------------------------- |
5f054727 |
6 | # Copyright (C) 2003 Allen Day <allenday@ucla.edu>, |
7 | # Ying Zhang <zyolive@yahoo.com> |
f42e7027 |
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 ]; |
b6c0195c |
26 | $VERSION = sprintf "%d.%02d", q$Revision: 1.21 $ =~ /(\d+)\.(\d+)/; |
f42e7027 |
27 | $DEBUG = 1 unless defined $DEBUG; |
28 | |
6c45056f |
29 | use SQL::Translator::Schema::Constants; |
5ee19df8 |
30 | use SQL::Translator::Utils qw(header_comment); |
f42e7027 |
31 | use Data::Dumper; |
32 | |
bf4629e7 |
33 | my %CDBI_auto_pkgs = ( |
34 | MySQL => 'mysql', |
35 | PostgreSQL => 'Pg', |
36 | Oracle => 'Oracle', |
37 | ); |
38 | |
39 | # ------------------------------------------------------------------- |
f42e7027 |
40 | sub produce { |
bf4629e7 |
41 | my $translator = shift; |
42 | local $DEBUG = $translator->debug; |
43 | my $no_comments = $translator->no_comments; |
44 | my $schema = $translator->schema; |
45 | my $args = $translator->producer_args; |
46 | my $db_user = $args->{'db_user'} || ''; |
47 | my $db_pass = $args->{'db_pass'} || ''; |
b6c0195c |
48 | my $parser_type = $translator->parser_type; |
49 | my $dsn = $args->{'dsn'} || "dbi:$CDBI_auto_pkgs{$parser_type}:_"; |
bf4629e7 |
50 | my $main_pkg_name = $translator->format_package_name('DBI'); |
51 | my $header = header_comment(__PACKAGE__, "# "); |
52 | my $sep = '# ' . '-' x 67; |
53 | |
54 | my $parser_type = ( split /::/, $translator->parser_type )[-1]; |
55 | my $from = $CDBI_auto_pkgs{ $parser_type } || ''; |
56 | |
6c45056f |
57 | # |
58 | # Iterate over all tables |
59 | # |
bf4629e7 |
60 | my ( %packages, $order ); |
6c45056f |
61 | for my $table ( $schema->get_tables ) { |
62 | my $table_name = $table->name or next; |
6c45056f |
63 | |
bf4629e7 |
64 | my $table_pkg_name = $translator->format_package_name($table_name); |
65 | $packages{ $table_pkg_name } = { |
66 | order => ++$order, |
67 | pkg_name => $table_pkg_name, |
68 | base => $main_pkg_name, |
69 | table => $table_name, |
70 | }; |
6c45056f |
71 | |
6c45056f |
72 | # |
bf4629e7 |
73 | # Primary key may have a differenct accessor method name |
6c45056f |
74 | # |
b6c0195c |
75 | if(defined($translator->format_pk_name)) { |
76 | # if ( my $pk_xform = $translator->format_pk_name ) { |
bf4629e7 |
77 | if ( my $constraint = $table->primary_key ) { |
78 | my $field = ($constraint->fields)[0]; |
b6c0195c |
79 | # my $pk_name = $pk_xform->($table_pkg_name, $field); |
80 | my $pk_name = $translator->format_pk_name($table_pkg_name, $field); |
bf4629e7 |
81 | |
82 | $packages{ $table_pkg_name }{'pk_accessor'} = |
83 | "#\n# Primary key accessor\n#\n". |
84 | "sub $pk_name {\n shift->$field\n}\n" |
85 | ; |
86 | } |
6c45056f |
87 | } |
bf4629e7 |
88 | |
6c45056f |
89 | # |
bf4629e7 |
90 | # Use foreign keys to set up "has_a/has_many" relationships. |
6c45056f |
91 | # |
92 | foreach my $field ( $table->get_fields ) { |
93 | if ( $field->is_foreign_key ) { |
bf4629e7 |
94 | my $table_name = $table->name; |
6c45056f |
95 | my $field_name = $field->name; |
96 | my $fk = $field->foreign_key_reference; |
97 | my $ref_table = $fk->reference_table; |
bf4629e7 |
98 | my $ref_pkg = $translator->format_package_name($ref_table); |
99 | my $ref_fld = |
100 | $translator->format_fk_name($ref_table, $field_name); |
6c45056f |
101 | |
bf4629e7 |
102 | push @{ $packages{ $table_pkg_name }{'has_a'} }, |
103 | "$table_pkg_name->has_a(\n". |
104 | " $field_name => '$ref_pkg'\n);\n\n". |
105 | "sub $ref_fld {\n". |
106 | " return shift->$field_name\n}\n\n" |
107 | ; |
52fbac6a |
108 | |
bf4629e7 |
109 | # |
110 | # If this table "has a" to the other, then it follows |
111 | # that the other table "has many" of this one, right? |
112 | # |
113 | push @{ $packages{ $ref_pkg }{'has_many'} }, |
b6c0195c |
114 | "$ref_pkg->has_many(\n '$table_name\s', ". |
bf4629e7 |
115 | "'$table_pkg_name' => '$field_name'\n);\n\n" |
116 | ; |
6c45056f |
117 | } |
118 | } |
df602cfb |
119 | |
bf4629e7 |
120 | # |
121 | # Identify link tables, defined as tables that have |
122 | # only PK and FK fields. |
123 | # |
124 | # my %linkable; |
125 | # my %linktable; |
126 | # foreach my $table ( $schema->get_tables ) { |
127 | # my $is_link = 1; |
128 | # foreach my $field ( $table->get_fields ) { |
129 | # unless ( $field->is_primary_key or $field->is_foreign_key ) { |
130 | # $is_link = 0; |
131 | # last; |
132 | # } |
133 | # } |
134 | # |
135 | # if ( $is_link ) { |
136 | # foreach my $left ( $table->get_fields ) { |
137 | # next unless $left->is_foreign_key and $schema->get_table( |
138 | # $left->foreign_key_reference->reference_table |
139 | # ); |
140 | # |
141 | # next unless $left->is_foreign_key and $schema->get_table( |
142 | # $left->foreign_key_reference->reference_table |
143 | # )->get_field( |
144 | # ($left->foreign_key_reference->reference_fields)[0] |
145 | # )->is_primary_key; |
146 | # |
147 | # foreach my $right ( $table->get_fields ) { |
148 | # #skip the diagonal |
149 | # next if $left->name eq $right->name; |
150 | # |
151 | # next unless $right->is_foreign_key and |
152 | # $schema->get_table( |
153 | # $right->foreign_key_reference->reference_table |
154 | # ) |
155 | # ; |
156 | # |
157 | # next unless $right->is_foreign_key and |
158 | # $schema->get_table( |
159 | # $right->foreign_key_reference->reference_table |
160 | # )->get_field( |
161 | # ($right->foreign_key_reference->reference_fields)[0] |
162 | # )->is_primary_key |
163 | # ; |
164 | # |
165 | # |
166 | # $linkable{ |
167 | # $left->foreign_key_reference->reference_table |
168 | # }{ |
169 | # $right->foreign_key_reference->reference_table |
170 | # } = $table; |
171 | # |
172 | # $linkable{ |
173 | # $right->foreign_key_reference->reference_table |
174 | # }{ |
175 | # $left->foreign_key_reference->reference_table |
176 | # } = $table; |
177 | # |
178 | # $linktable{ $table->name } = $table; |
179 | # } |
180 | # } |
181 | # } |
182 | # } |
183 | # |
184 | # # |
185 | # # Generate many-to-many linking methods for data tables |
186 | # # |
187 | # my $is_data = 0; |
188 | # for ( $table->get_fields ) { |
189 | # $is_data++ if !$_->is_foreign_key and !$_->is_primary_key; |
190 | # } |
191 | # |
192 | # my %linked; |
193 | # if ( $is_data ) { |
194 | # foreach my $link ( keys %{ $linkable{ $table->name } } ) { |
195 | # my $linkmethodname = |
196 | # "_".$translator->format_fk_name($table->name,$link)."_refs" |
197 | # ; |
198 | # |
199 | # $create .= $translator->format_package_name($table->name). |
200 | # "->has_many('$linkmethodname','". |
201 | # $translator->format_package_name( |
202 | # $linkable{$table->name}{$link}->name |
203 | # )."','". |
204 | # ($schema->get_table($link)->primary_key->fields)[0]."');\n" |
205 | # ; |
206 | # |
207 | # $create .= "sub ". $translator->format_fk_name($table,$link). |
208 | # # HARDCODED 's' HERE. |
209 | # # ADD CALLBACK FOR PLURALIZATION MANGLING |
210 | # "s {\n my \$self = shift; return map \$_->".$link. |
211 | # ", \$self->".$linkmethodname.";\n}\n\n" |
212 | # ; |
213 | # } |
214 | # } |
215 | } |
6c45056f |
216 | |
bf4629e7 |
217 | my $base_pkg = sprintf( 'Class::DBI%s', $from ? "::$from" : '' ); |
218 | my $create = join("\n", |
219 | "package $main_pkg_name;\n", |
220 | $header, |
221 | "use strict;", |
222 | "use base '$base_pkg';\n", |
223 | "$main_pkg_name->set_db('Main', '$dsn', '$db_user', '$db_pass');\n\n", |
224 | ); |
225 | |
226 | for my $pkg_name ( |
227 | sort { $packages{ $a }{'order'} <=> $packages{ $b }{'order'} } |
228 | keys %packages |
229 | ) { |
230 | my $pkg = $packages{ $pkg_name }; |
231 | |
232 | $create .= join("\n", |
233 | $sep, |
234 | "package ".$pkg->{'pkg_name'}.";", |
235 | "use base '".$pkg->{'base'}."';", |
236 | "use Class::DBI::Pager;\n\n", |
237 | ); |
f42e7027 |
238 | |
bf4629e7 |
239 | if ( $from ) { |
240 | $create .= |
241 | $pkg->{'pkg_name'}."->set_up_table('".$pkg->{'table'}."');\n\n"; |
242 | } |
243 | else { |
244 | my $table = $schema->get_table( $pkg->{'table'} ); |
245 | my @field_names = map { $_->name } $table->get_fields; |
246 | |
247 | $create .= join("\n", |
248 | $pkg_name."->table('".$pkg->{'table'}."');\n", |
249 | $pkg_name."->columns(All => qw/". |
250 | join(' ', @field_names)."/);\n\n", |
251 | ); |
252 | } |
52fbac6a |
253 | |
bf4629e7 |
254 | if ( my $pk = $pkg->{'pk_accessor'} ) { |
255 | $create .= $pk; |
256 | } |
df602cfb |
257 | |
bf4629e7 |
258 | if ( my @has_a = @{ $pkg->{'has_a'} || [] } ) { |
259 | $create .= $_ for @has_a; |
260 | } |
261 | |
262 | if ( my @has_many = @{ $pkg->{'has_many'} || [] } ) { |
263 | $create .= $_ for @has_many; |
264 | } |
6c45056f |
265 | } |
df602cfb |
266 | |
bf4629e7 |
267 | $create .= "1;\n"; |
268 | |
269 | return $create; |
f42e7027 |
270 | } |
271 | |
272 | 1; |
273 | |
bf4629e7 |
274 | # ------------------------------------------------------------------- |
275 | |
276 | =pod |
f42e7027 |
277 | |
278 | =head1 NAME |
279 | |
bf4629e7 |
280 | SQL::Translator::Producer::ClassDBI - create Class::DBI classes from schema |
f42e7027 |
281 | |
282 | =head1 SYNOPSIS |
283 | |
1eee27d3 |
284 | Use this producer as you would any other from SQL::Translator. See |
285 | L<SQL::Translator> for details. |
f42e7027 |
286 | |
1eee27d3 |
287 | This package utilizes SQL::Translator's formatting methods |
288 | format_package_name(), format_pk_name(), format_fk_name(), and |
289 | format_table_name() as it creates classes, one per table in the schema |
290 | provided. An additional base class is also created for database connectivity |
291 | configuration. See L<Class::DBI> for details on how this works. |
f42e7027 |
292 | |
6c45056f |
293 | =head1 AUTHORS |
f42e7027 |
294 | |
6c45056f |
295 | Allen Day E<lt>allenday@ucla.eduE<gt> |
5f054727 |
296 | Ying Zhang E<lt>zyolive@yahoo.comE<gt>, |
6c45056f |
297 | Ken Y. Clark E<lt>kclark@cpan.org<gt>. |