Add loader_roles option
[dbsrgits/DBIx-Class-Schema-Loader.git] / lib / DBIx / Class / Schema / Loader.pm
1 package DBIx::Class::Schema::Loader;
2
3 use strict;
4 use warnings;
5 use base qw/DBIx::Class::Schema Class::Accessor::Grouped/;
6 use MRO::Compat;
7 use mro 'c3';
8 use Carp::Clan qw/^DBIx::Class/;
9 use Scalar::Util 'weaken';
10 use Sub::Util 'set_subname';
11 use DBIx::Class::Schema::Loader::Utils qw/array_eq sigwarn_silencer/;
12 use Try::Tiny;
13 use curry;
14 use namespace::clean;
15
16 # Always remember to do all digits for the version even if they're 0
17 # i.e. first release of 0.XX *must* be 0.XX000. This avoids fBSD ports
18 # brain damage and presumably various other packaging systems too
19 our $VERSION = '0.07048';
20
21 __PACKAGE__->mk_group_accessors('inherited', qw/
22                                 _loader_args
23                                 dump_to_dir
24                                 _loader_invoked
25                                 _loader
26                                 loader_class
27                                 loader_roles
28                                 naming
29                                 use_namespaces
30 /);
31 __PACKAGE__->_loader_args({});
32
33 =encoding UTF-8
34
35 =head1 NAME
36
37 DBIx::Class::Schema::Loader - Create a DBIx::Class::Schema based on a database
38
39 =head1 SYNOPSIS
40
41     ### use this module to generate a set of class files
42
43     # in a script
44     use DBIx::Class::Schema::Loader qw/ make_schema_at /;
45     make_schema_at(
46         'My::Schema',
47         { debug => 1,
48           dump_directory => './lib',
49         },
50         [ 'dbi:Pg:dbname="foo"', 'myuser', 'mypassword',
51            { loader_class => 'MyLoader' } # optionally
52         ],
53     );
54
55     # from the command line or a shell script with dbicdump (distributed
56     # with this module).  Do `perldoc dbicdump` for usage.
57     dbicdump -o dump_directory=./lib \
58              -o components='["InflateColumn::DateTime"]' \
59              -o debug=1 \
60              My::Schema \
61              'dbi:Pg:dbname=foo' \
62              myuser \
63              mypassword
64
65     ### or generate and load classes at runtime
66     # note: this technique is not recommended
67     # for use in production code
68
69     package My::Schema;
70     use base qw/DBIx::Class::Schema::Loader/;
71
72     __PACKAGE__->loader_options(
73         constraint              => '^foo.*',
74         # debug                 => 1,
75     );
76
77     #### in application code elsewhere:
78
79     use My::Schema;
80
81     my $schema1 = My::Schema->connect( $dsn, $user, $password, $attrs);
82     # -or-
83     my $schema1 = "My::Schema"; $schema1->connection(as above);
84
85 =head1 DESCRIPTION
86
87 DBIx::Class::Schema::Loader automates the definition of a
88 L<DBIx::Class::Schema> by scanning database table definitions and setting up
89 the columns, primary keys, unique constraints and relationships.
90
91 See L<dbicdump> for the C<dbicdump> utility.
92
93 DBIx::Class::Schema::Loader currently supports only the DBI storage type. It
94 has explicit support for L<DBD::Pg>, L<DBD::mysql>, L<DBD::DB2>,
95 L<DBD::Firebird>, L<DBD::InterBase>, L<DBD::Informix>, L<DBD::SQLAnywhere>,
96 L<DBD::SQLite>, L<DBD::Sybase> (for Sybase ASE and MSSSQL), L<DBD::ODBC> (for
97 MSSQL, MSAccess, Firebird and SQL Anywhere) L<DBD::ADO> (for MSSQL and
98 MSAccess) and L<DBD::Oracle>.  Other DBI drivers may function to a greater or
99 lesser degree with this loader, depending on how much of the DBI spec they
100 implement, and how standard their implementation is.
101
102 Patches to make other DBDs work correctly welcome.
103
104 See L<DBIx::Class::Schema::Loader::DBI::Writing> for notes on writing
105 your own vendor-specific subclass for an unsupported DBD driver.
106
107 This module requires L<DBIx::Class> 0.08127 or later, and obsoletes the older
108 L<DBIx::Class::Loader>.
109
110 See L<DBIx::Class::Schema::Loader::Base> for available options.
111
112 =head1 METHODS
113
114 =head2 loader
115
116 The loader object, as class data on your Schema. For methods available see
117 L<DBIx::Class::Schema::Loader::Base> and L<DBIx::Class::Schema::Loader::DBI>.
118
119 =cut
120
121 sub loader {
122     my $self = shift;
123     $self->_loader(@_);
124 }
125
126 =head2 loader_class
127
128 =over 4
129
130 =item Argument: $loader_class
131
132 =back
133
134 Set the loader class to be instantiated when L</connection> is called.
135 If the classname starts with "::", "DBIx::Class::Schema::Loader" is
136 prepended. Defaults to L<DBIx::Class::Schema/storage_type> (which must
137 start with "::" when using L<DBIx::Class::Schema::Loader>).
138
139 This is mostly useful for subclassing existing loaders or in conjunction
140 with L</dump_to_dir>.
141
142 =head2 loader_options
143
144 =over 4
145
146 =item Argument: \%loader_options
147
148 =back
149
150 Example in Synopsis above demonstrates a few common arguments.  For
151 detailed information on all of the arguments, most of which are
152 only useful in fairly complex scenarios, see the
153 L<DBIx::Class::Schema::Loader::Base> documentation.
154
155 If you intend to use C<loader_options>, you must call
156 C<loader_options> before any connection is made, or embed the
157 C<loader_options> in the connection information itself as shown
158 below.  Setting C<loader_options> after the connection has
159 already been made is useless.
160
161 =cut
162
163 sub loader_options {
164     my $self = shift;
165
166     my %args = (ref $_[0] eq 'HASH') ? %{$_[0]} : @_;
167     $self->_loader_args(\%args);
168
169     $self;
170 }
171
172 sub _invoke_loader {
173     my $self = shift;
174     my $class = ref $self || $self;
175
176     my $args = $self->_loader_args;
177
178     # temporarily copy $self's storage to class
179     my $class_storage = $class->storage;
180     if (ref $self) {
181         $class->storage($self->storage);
182         $class->storage->set_schema($class);
183     }
184
185     $args->{schema} = $class;
186     $args->{schema_class} = $class;
187     $args->{dump_directory} ||= $self->dump_to_dir;
188     $args->{naming} = $self->naming if $self->naming;
189     $args->{use_namespaces} = $self->use_namespaces if defined $self->use_namespaces;
190
191     my $loader_class = $self->loader_class;
192     if ($loader_class) {
193         $loader_class = "DBIx::Class::Schema::Loader${loader_class}" if $loader_class =~ /^::/;
194         $args->{loader_class} = $loader_class;
195     };
196
197     # XXX this only works for relative storage_type, like ::DBI ...
198     my $impl = $loader_class || "DBIx::Class::Schema::Loader" . $self->storage_type;
199     try {
200         $self->ensure_class_loaded($impl)
201     }
202     catch {
203         croak qq/Could not load loader_class "$impl": "$_"/;
204     };
205
206     my $loader = $impl->new(%$args);
207     if (my @roles = @{$self->loader_roles || []}) {
208         require Role::Tiny;
209         Role::Tiny->apply_roles_to_object($loader, @roles);
210     }
211
212     $class->loader($loader);
213     $class->loader->load;
214     $class->_loader_invoked(1);
215
216     # copy to $self
217     if (ref $self) {
218         $self->loader($class->loader);
219         $self->_loader_invoked(1);
220
221         $self->_merge_state_from($class);
222     }
223
224     # restore $class's storage
225     $class->storage($class_storage);
226
227     return $self;
228 }
229
230 # FIXME This needs to be moved into DBIC at some point, otherwise we are
231 # maintaining things to do with DBIC guts, which we have no business of
232 # maintaining. But at the moment it would be just dead code in DBIC, so we'll
233 # maintain it here.
234 sub _merge_state_from {
235     my ($self, $from) = @_;
236
237     my $orig_class_mappings       = $self->class_mappings;
238     my $orig_source_registrations = $self->source_registrations;
239
240     $self->_copy_state_from($from);
241
242     $self->class_mappings(__merge($orig_class_mappings, $self->class_mappings))
243         if $orig_class_mappings;
244
245     $self->source_registrations(__merge($orig_source_registrations, $self->source_registrations))
246         if $orig_source_registrations;
247 }
248
249 my $merger;
250 sub __merge {
251
252   local $SIG{__WARN__} = sigwarn_silencer(qr/Arguments for _merge_hashes must be hash references/);
253
254   ( $merger ||= do {
255     require Hash::Merge;
256     my $m = Hash::Merge->new('LEFT_PRECEDENT');
257     $m->set_clone_behavior(0);
258     $m;
259   } )->merge(
260     $_[0], $_[1]
261   );
262 }
263
264 sub _copy_state_from {
265     my $self = shift;
266     my ($from) = @_;
267
268     # older DBIC's do not have this method
269     if (try { DBIx::Class->VERSION('0.08197'); 1 }) {
270         return $self->next::method(@_);
271     }
272     else {
273         # this is a copy from DBIC git master pre 0.08197
274         $self->class_mappings({ %{$from->class_mappings} });
275         $self->source_registrations({ %{$from->source_registrations} });
276
277         foreach my $moniker ($from->sources) {
278             my $source = $from->source($moniker);
279             my $new = $source->new($source);
280             # we use extra here as we want to leave the class_mappings as they are
281             # but overwrite the source_registrations entry with the new source
282             $self->register_extra_source($moniker => $new);
283         }
284
285         if ($from->storage) {
286             $self->storage($from->storage);
287             $self->storage->set_schema($self);
288         }
289     }
290 }
291
292 =head2 connection
293
294 =over 4
295
296 =item Arguments: @args
297
298 =item Return Value: $new_schema
299
300 =back
301
302 See L<DBIx::Class::Schema/connection> for basic usage.
303
304 If the final argument is a hashref, and it contains the keys C<loader_options>,
305 C<loader_class> or C<loader_roles>, those keys will be deleted, and their values
306 value will be used for the loader options or class, respectively, just as if set
307 via the L</loader_options> or L</loader_class> methods above.
308
309 The actual auto-loading operation (the heart of this module) will be invoked
310 as soon as the connection information is defined.
311
312 =cut
313
314 sub connection {
315     my $self  = shift;
316     my $class = ref $self || $self;
317
318     if($_[-1] && ref $_[-1] eq 'HASH') {
319         for my $option (qw/loader_class loader_options loader_roles/) {
320             if(my $value = delete $_[-1]->{$option}) {
321                 $self->$option($value);
322             }
323         }
324         pop @_ if !keys %{$_[-1]};
325     }
326
327     # Make sure we inherit from schema_base_class and load schema_components
328     # before connecting.
329     require DBIx::Class::Schema::Loader::Base;
330     my $temp_loader = DBIx::Class::Schema::Loader::Base->new(
331         %{ $self->_loader_args },
332         schema => $self,
333         naming => 'current',
334         use_namespaces => 1,
335     );
336
337     my $modify_isa = 0;
338     my @components;
339
340     if ($temp_loader->schema_base_class || $temp_loader->schema_components) {
341         @components = @{ $temp_loader->schema_components }
342             if $temp_loader->schema_components;
343
344         push @components, ('+'.$temp_loader->schema_base_class)
345             if $temp_loader->schema_base_class;
346
347         my $class_isa = do {
348             no strict 'refs';
349             \@{"${class}::ISA"};
350         };
351
352         my @component_classes = map {
353             /^\+/ ? substr($_, 1, length($_) - 1) : "DBIx::Class::$_"
354         } @components;
355
356         $modify_isa++ if not array_eq([ @$class_isa[0..(@components-1)] ], \@component_classes)
357     }
358
359     if ($modify_isa) {
360         $class->load_components(@components);
361
362         # This hack is necessary because we changed @ISA of $self through
363         # ->load_components and we are now in a different place in the mro.
364         no warnings 'redefine';
365
366         local *connection = set_subname __PACKAGE__.'::connection' => sub {
367             my $self = shift;
368             $self->next::method(@_);
369         };
370
371         my @linear_isa = @{ mro::get_linear_isa($class) };
372
373         my $next_method;
374
375         foreach my $i (1..$#linear_isa) {
376             no strict 'refs';
377             $next_method = *{$linear_isa[$i].'::connection'}{CODE};
378             last if $next_method;
379         }
380
381         $self = $self->$next_method(@_);
382     }
383     else {
384         $self = $self->next::method(@_);
385     }
386
387     if(!$class->_loader_invoked) {
388         $self->_invoke_loader
389     }
390
391     return $self;
392 }
393
394 =head2 clone
395
396 See L<DBIx::Class::Schema/clone>.
397
398 =cut
399
400 sub clone {
401     my $self = shift;
402
403     my $clone = $self->next::method(@_);
404
405     if($clone->_loader_args) {
406         $clone->_loader_args->{schema} = $clone;
407         weaken($clone->_loader_args->{schema});
408     }
409
410     $clone;
411 }
412
413 =head2 dump_to_dir
414
415 =over 4
416
417 =item Argument: $directory
418
419 =back
420
421 Calling this as a class method on either L<DBIx::Class::Schema::Loader>
422 or any derived schema class will cause all schemas to dump
423 manual versions of themselves to the named directory when they are
424 loaded.  In order to be effective, this must be set before defining a
425 connection on this schema class or any derived object (as the loading
426 happens as soon as both a connection and loader_options are set, and
427 only once per class).
428
429 See L<DBIx::Class::Schema::Loader::Base/dump_directory> for more
430 details on the dumping mechanism.
431
432 This can also be set at module import time via the import option
433 C<dump_to_dir:/foo/bar> to L<DBIx::Class::Schema::Loader>, where
434 C</foo/bar> is the target directory.
435
436 Examples:
437
438     # My::Schema isa DBIx::Class::Schema::Loader, and has connection info
439     #   hardcoded in the class itself:
440     perl -MDBIx::Class::Schema::Loader=dump_to_dir:/foo/bar -MMy::Schema -e1
441
442     # Same, but no hard-coded connection, so we must provide one:
443     perl -MDBIx::Class::Schema::Loader=dump_to_dir:/foo/bar -MMy::Schema -e 'My::Schema->connection("dbi:Pg:dbname=foo", ...)'
444
445     # Or as a class method, as long as you get it done *before* defining a
446     #  connection on this schema class or any derived object:
447     use My::Schema;
448     My::Schema->dump_to_dir('/foo/bar');
449     My::Schema->connection(........);
450
451     # Or as a class method on the DBIx::Class::Schema::Loader itself, which affects all
452     #   derived schemas
453     use My::Schema;
454     use My::OtherSchema;
455     DBIx::Class::Schema::Loader->dump_to_dir('/foo/bar');
456     My::Schema->connection(.......);
457     My::OtherSchema->connection(.......);
458
459     # Another alternative to the above:
460     use DBIx::Class::Schema::Loader qw| dump_to_dir:/foo/bar |;
461     use My::Schema;
462     use My::OtherSchema;
463     My::Schema->connection(.......);
464     My::OtherSchema->connection(.......);
465
466 =cut
467
468 sub import {
469     my $self = shift;
470
471     return if !@_;
472
473     my $cpkg = (caller)[0];
474
475     foreach my $opt (@_) {
476         if($opt =~ m{^dump_to_dir:(.*)$}) {
477             $self->dump_to_dir($1)
478         }
479         elsif($opt eq 'make_schema_at') {
480             no strict 'refs';
481             *{"${cpkg}::make_schema_at"} = \&make_schema_at;
482         }
483         elsif($opt eq 'naming') {
484             no strict 'refs';
485             *{"${cpkg}::naming"} = $self->curry::naming;
486         }
487         elsif($opt eq 'use_namespaces') {
488             no strict 'refs';
489             *{"${cpkg}::use_namespaces"} = $self->curry::use_namespaces,
490         }
491     }
492 }
493
494 =head2 make_schema_at
495
496 =over 4
497
498 =item Arguments: $schema_class_name, \%loader_options, \@connect_info
499
500 =item Return Value: $schema_class_name
501
502 =back
503
504 This function creates a DBIx::Class schema from an existing RDBMS
505 schema.  With the C<dump_directory> option, generates a set of
506 DBIx::Class classes from an existing database schema read from the
507 given dsn.  Without a C<dump_directory>, creates schema classes in
508 memory at runtime without generating on-disk class files.
509
510 For a complete list of supported loader_options, see
511 L<DBIx::Class::Schema::Loader::Base>
512
513 The last hashref in the C<\@connect_info> can specify the L</loader_class>.
514
515 This function can be imported in the usual way, as illustrated in
516 these Examples:
517
518     # Simple example, creates as a new class 'New::Schema::Name' in
519     #  memory in the running perl interpreter.
520     use DBIx::Class::Schema::Loader qw/ make_schema_at /;
521     make_schema_at(
522         'New::Schema::Name',
523         { debug => 1 },
524         [ 'dbi:Pg:dbname="foo"','postgres','',
525           { loader_class => 'MyLoader' } # optionally
526         ],
527     );
528
529     # Inside a script, specifying a dump directory in which to write
530     # class files
531     use DBIx::Class::Schema::Loader qw/ make_schema_at /;
532     make_schema_at(
533         'New::Schema::Name',
534         { debug => 1, dump_directory => './lib' },
535         [ 'dbi:Pg:dbname="foo"','postgres','',
536           { loader_class => 'MyLoader' } # optionally
537         ],
538     );
539
540 The last hashref in the C<\@connect_info> is checked for loader arguments such
541 as C<loader_options> and C<loader_class>, see L</connection> for more details.
542
543 =cut
544
545 sub make_schema_at {
546     my ($target, $opts, $connect_info) = @_;
547
548     {
549         no strict 'refs';
550         @{$target . '::ISA'} = qw/DBIx::Class::Schema::Loader/;
551     }
552
553     $target->_loader_invoked(0);
554
555     $target->loader_options($opts);
556
557     my $temp_schema = $target->connect(@$connect_info);
558
559     $target->storage($temp_schema->storage);
560     $target->storage->set_schema($target);
561
562     return $target;
563 }
564
565 =head2 rescan
566
567 =over 4
568
569 =item Return Value: @new_monikers
570
571 =back
572
573 Re-scans the database for newly added tables since the initial
574 load, and adds them to the schema at runtime, including relationships,
575 etc.  Does not process drops or changes.
576
577 Returns a list of the new monikers added.
578
579 =cut
580
581 sub rescan { my $self = shift; $self->loader->rescan($self) }
582
583 =head2 naming
584
585 =over 4
586
587 =item Arguments: \%opts | $ver
588
589 =back
590
591 Controls the naming options for backward compatibility, see
592 L<DBIx::Class::Schema::Loader::Base/naming> for details.
593
594 To upgrade a dynamic schema, use:
595
596     __PACKAGE__->naming('current');
597
598 Can be imported into your dump script and called as a function as well:
599
600     naming('v4');
601
602 =head2 use_namespaces
603
604 =over 4
605
606 =item Arguments: 1|0
607
608 =back
609
610 Controls the use_namespaces options for backward compatibility, see
611 L<DBIx::Class::Schema::Loader::Base/use_namespaces> for details.
612
613 To upgrade a dynamic schema, use:
614
615     __PACKAGE__->use_namespaces(1);
616
617 Can be imported into your dump script and called as a function as well:
618
619     use_namespaces(1);
620
621 =head1 KNOWN ISSUES
622
623 =head2 Multiple Database Schemas
624
625 See L<DBIx::Class::Schema::Loader::Base/db_schema>.
626
627 =head1 ACKNOWLEDGEMENTS
628
629 Matt S Trout, all of the #dbix-class folks, and everyone who's ever sent
630 in a bug report or suggestion.
631
632 Based on L<DBIx::Class::Loader> by Sebastian Riedel
633
634 Based upon the work of IKEBE Tomohiro
635
636 =head1 AUTHORS
637
638 Caelum: Rafael Kitover <rkitover@cpan.org>
639
640 Dag-Erling Smørgrav <des@des.no>
641
642 Matias E. Fernandez <mfernandez@pisco.ch>
643
644 SineSwiper: Brendan Byrd <byrd.b@insightcom.com>
645
646 TSUNODA Kazuya <drk@drk7.jp>
647
648 acmoore: Andrew Moore <amoore@cpan.org>
649
650 alnewkirk: Al Newkirk <awncorp@cpan.org>
651
652 andrewalker: André Walker <andre@andrewalker.net>
653
654 angelixd: Paul C. Mantz <pcmantz@cpan.org>
655
656 arc: Aaron Crane <arc@cpan.org>
657
658 arcanez: Justin Hunter <justin.d.hunter@gmail.com>
659
660 ash: Ash Berlin <ash@cpan.org>
661
662 blblack: Brandon Black <blblack@gmail.com>
663
664 bphillips: Brian Phillips <bphillips@cpan.org>
665
666 btilly: Ben Tilly <btilly@gmail.com>
667
668 domm: Thomas Klausner <domm@plix.at>
669
670 gugu: Andrey Kostenko <a.kostenko@rambler-co.ru>
671
672 hobbs: Andrew Rodland <arodland@cpan.org>
673
674 ilmari: Dagfinn Ilmari MannsE<aring>ker <ilmari@ilmari.org>
675
676 jhannah: Jay Hannah <jay@jays.net>
677
678 jnap: John Napiorkowski <jjn1056@yahoo.com>
679
680 kane: Jos Boumans <kane@cpan.org>
681
682 mattp: Matt Phillips <mattp@cpan.org>
683
684 mephinet: Philipp Gortan <philipp.gortan@apa.at>
685
686 moritz: Moritz Lenz <moritz@faui2k3.org>
687
688 mst: Matt S. Trout <mst@shadowcatsystems.co.uk>
689
690 mstratman: Mark A. Stratman <stratman@gmail.com>
691
692 oalders: Olaf Alders <olaf@wundersolutions.com>
693
694 rbo: Robert Bohne <rbo@cpan.org>
695
696 rbuels: Robert Buels <rbuels@gmail.com>
697
698 ribasushi: Peter Rabbitson <ribasushi@cpan.org>
699
700 schwern: Michael G. Schwern <mschwern@cpan.org>
701
702 spb: Stephen Bennett <spb@exherbo.org>
703
704 timbunce: Tim Bunce <timb@cpan.org>
705
706 waawaamilk: Nigel McNie <nigel@mcnie.name>
707
708 ... and lots of other folks. If we forgot you, please write the current
709 maintainer or RT.
710
711 =head1 COPYRIGHT & LICENSE
712
713 Copyright (c) 2006 - 2015 by the aforementioned
714 L<DBIx::Class::Schema::Loader/AUTHORS>.
715
716 This library is free software; you can redistribute it and/or modify it under
717 the same terms as Perl itself.
718
719 =head1 SEE ALSO
720
721 L<DBIx::Class>, L<DBIx::Class::Manual::Intro>, L<DBIx::Class::Tutorial>,
722 L<DBIx::Class::Schema::Loader::Base>
723
724 =cut
725
726 1;
727 # vim:et sts=4 sw=4 tw=0: