use strict;
use warnings;
-use base qw/DBIx::Class::Schema/;
-
use Carp;
use Lingua::EN::Inflect;
+use base qw/Class::Accessor::Fast/;
require DBIx::Class::Core;
-__PACKAGE__->mk_classdata('_loader_data');
-__PACKAGE__->mk_classdata('_loader_debug' => 0);
+# The first group are all arguments which are may be defaulted within,
+# The last two (classes, monikers) are generated locally:
+
+__PACKAGE__->mk_ro_accessors(qw/
+ schema
+ dsn
+ user
+ password
+ options
+ exclude
+ constraint
+ additional_classes
+ additional_base_classes
+ left_base_classes
+ relationships
+ inflect
+ db_schema
+ drop_db_schema
+ debug
+
+ classes
+ monikers
+ /);
=head1 NAME
=head3 new
-Not intended to be called directly. This is used internally by the
-C<new()> method in L<DBIx::Class::Schema::Loader>.
+Constructor for L<DBIx::Class::Schema::Loader::Generic>, used internally
+by L<DBIx::Class::Schema::Loader>.
=cut
-sub _load_from_connection {
+# ensure that a peice of object data is a valid arrayref, creating
+# an empty one or encapsulating whatever's there.
+sub _ensure_arrayref {
+ my $self = shift;
+
+ foreach (@_) {
+ $self->{$_} ||= [];
+ $self->{$_} = [ $self->{$_} ]
+ unless ref $self->{$_} eq 'ARRAY';
+ }
+}
+
+sub new {
my ( $class, %args ) = @_;
- $class->_loader_debug( $args{debug} ? 1 : 0);
-
- my $additional = $args{additional_classes} || [];
- $additional = [$additional] unless ref $additional eq 'ARRAY';
-
- my $additional_base = $args{additional_base_classes} || [];
- $additional_base = [$additional_base]
- unless ref $additional_base eq 'ARRAY';
-
- my $left_base = $args{left_base_classes} || [];
- $left_base = [$left_base] unless ref $left_base eq 'ARRAY';
-
- $class->_loader_data({
- datasource =>
- [ $args{dsn}, $args{user}, $args{password}, $args{options} ],
- additional => $additional,
- additional_base => $additional_base,
- left_base => $left_base,
- constraint => $args{constraint} || '.*',
- exclude => $args{exclude},
- relationships => $args{relationships},
- inflect => $args{inflect},
- db_schema => $args{db_schema} || '',
- drop_db_schema => $args{drop_db_schema},
- TABLE_CLASSES => {},
- MONIKERS => {},
- });
-
- $class->connection(@{$class->_loader_data->{datasource}});
+ my $self = { %args };
+
+ bless $self => $class;
+
+ $self->{db_schema} ||= '';
+ $self->{constraint} ||= '.*';
+ $self->{inflect} ||= {};
+ $self->_ensure_arrayref(qw/additional_classes
+ additional_base_classes
+ left_base_classes/);
+
+ $self->{monikers} = {};
+ $self->{classes} = {};
+
+ $self->schema->connection($self->dsn, $self->user,
+ $self->password, $self->options);
+
warn qq/\### START DBIx::Class::Schema::Loader dump ###\n/
- if $class->_loader_debug;
- $class->_loader_load_classes;
- $class->_loader_relationships if $class->_loader_data->{relationships};
- warn qq/\### END DBIx::Class::Schema::Loader dump ###\n/
- if $class->_loader_debug;
- $class->storage->dbh->disconnect; # XXX this should be ->storage->disconnect later?
+ if $self->debug;
- 1;
-}
+ $self->_load_classes;
+ $self->_load_relationships if $self->relationships;
-# The original table class name during Loader,
-sub _loader_find_table_class {
- my ( $class, $table ) = @_;
- return $class->_loader_data->{TABLE_CLASSES}->{$table};
-}
+ warn qq/\### END DBIx::Class::Schema::Loader dump ###\n/
+ if $self->debug;
+ $self->schema->storage->disconnect;
-# Returns the moniker for a given table name,
-# for use in $conn->resultset($moniker)
+ $self;
+}
-=head3 moniker
+# Overload in your driver class
+sub _db_classes { croak "ABSTRACT METHOD" }
-Returns the moniker for a given literal table name. Used
-as $schema->resultset($moniker), etc.
+# Inflect a relationship name
+# XXX (should pluralize, but currently also tends to de-pluralize plurals)
+sub _inflect_relname {
+ my ($self, $relname) = @_;
-=cut
-sub moniker {
- my ( $class, $table ) = @_;
- return $class->_loader_data->{MONIKERS}->{$table};
+ return $self->inflect->{$relname} if exists $self->inflect->{$relname};
+ return Lingua::EN::Inflect::PL($relname);
}
-=head3 tables
+# Set up a simple relation with just a local col and foreign table
+sub _make_simple_rel {
+ my ($self, $table, $other, $col) = @_;
-Returns a sorted list of tables.
+ my $table_class = $self->classes->{$table};
+ my $other_class = $self->classes->{$other};
+ my $table_relname = $self->_inflect_relname(lc $table);
- my @tables = $loader->tables;
+ warn qq/\# Belongs_to relationship\n/ if $self->debug;
+ warn qq/$table_class->belongs_to( '$col' => '$other_class' );\n\n/
+ if $self->debug;
+ $table_class->belongs_to( $col => $other_class );
-=cut
+ warn qq/\# Has_many relationship\n/ if $self->debug;
+ warn qq/$other_class->has_many( '$table_relname' => '$table_class',/
+ . qq/$col);\n\n/
+ if $self->debug;
-sub tables {
- my $class = shift;
- return sort keys %{ $class->_loader_data->{MONIKERS} };
+ $other_class->has_many( $table_relname => $table_class, $col);
}
-# Overload in your driver class
-sub _loader_db_classes { croak "ABSTRACT METHOD" }
+# not a class method, just a helper for cond_rel XXX
+sub _stringify_hash {
+ my $href = shift;
-# Setup has_a and has_many relationships
-sub _loader_make_relations {
- use Data::Dumper;
+ return '{ ' .
+ join(q{, }, map("$_ => $href->{$_}", keys %$href))
+ . ' }';
+}
- my ( $class, $table, $other, $cond ) = @_;
- my $table_class = $class->_loader_find_table_class($table);
- my $other_class = $class->_loader_find_table_class($other);
+# Set up a complex relation based on a hashref condition
+sub _make_cond_rel {
+ my ( $self, $table, $other, $cond ) = @_;
- my $table_relname = lc $table;
+ my $table_class = $self->classes->{$table};
+ my $other_class = $self->classes->{$other};
+ my $table_relname = $self->_inflect_relname(lc $table);
my $other_relname = lc $other;
- if(my $inflections = $class->_loader_data->{inflect}) {
- $table_relname = $inflections->{$table_relname}
- if exists $inflections->{$table_relname};
- }
- else {
- $table_relname = Lingua::EN::Inflect::PL($table_relname);
- }
-
# for single-column case, set the relname to the column name,
# to make filter accessors work
if(scalar keys %$cond == 1) {
my $rev_cond = { reverse %$cond };
- warn qq/\# Belongs_to relationship\n/ if $class->_loader_debug;
+ my $cond_printable = _stringify_hash($cond)
+ if $self->debug;
+ my $rev_cond_printable = _stringify_hash($rev_cond)
+ if $self->debug;
+
+ warn qq/\# Belongs_to relationship\n/ if $self->debug;
warn qq/$table_class->belongs_to( '$other_relname' => '$other_class',/
- . Dumper($cond)
- . qq/);\n\n/
- if $class->_loader_debug;
+ . qq/$cond_printable);\n\n/
+ if $self->debug;
$table_class->belongs_to( $other_relname => $other_class, $cond);
- warn qq/\# Has_many relationship\n/ if $class->_loader_debug;
+ warn qq/\# Has_many relationship\n/ if $self->debug;
warn qq/$other_class->has_many( '$table_relname' => '$table_class',/
- . Dumper($rev_cond)
+ . qq/$rev_cond_printable);\n\n/
. qq/);\n\n/
- if $class->_loader_debug;
+ if $self->debug;
$other_class->has_many( $table_relname => $table_class, $rev_cond);
}
+sub _use {
+ my $self = shift;
+ my $target = shift;
+
+ foreach (@_) {
+ $_->require or croak ($_ . "->require: $@");
+ eval "package $target; use $_;";
+ croak "use $_: $@" if $@;
+ }
+}
+
+sub _inject {
+ my $self = shift;
+ my $target = shift;
+ my $schema = $self->schema;
+
+ foreach (@_) {
+ $_->require or croak ($_ . "->require: $@");
+ $schema->inject_base($target, $_);
+ }
+}
+
# Load and setup classes
-sub _loader_load_classes {
- my $class = shift;
-
- my @tables = $class->_loader_tables();
- my @db_classes = $class->_loader_db_classes();
- my $additional = join '', map "use $_;\n", @{ $class->_loader_data->{additional} };
- my $additional_base = join '', map "use base '$_';\n",
- @{ $class->_loader_data->{additional_base} };
- my $left_base = join '', map "use base '$_';\n", @{ $class->_loader_data->{left_base} };
- my $constraint = $class->_loader_data->{constraint};
- my $exclude = $class->_loader_data->{exclude};
+sub _load_classes {
+ my $self = shift;
+
+ my @tables = $self->_tables();
+ my @db_classes = $self->_db_classes();
+ my $schema = $self->schema;
foreach my $table (@tables) {
+ my $constraint = $self->constraint;
+ my $exclude = $self->exclude;
+
next unless $table =~ /$constraint/;
- next if ( defined $exclude && $table =~ /$exclude/ );
+ next if defined $exclude && $table =~ /$exclude/;
my ($db_schema, $tbl) = split /\./, $table;
my $tablename = lc $table;
if($tbl) {
- $tablename = $class->_loader_data->{drop_db_schema} ? $tbl : lc $table;
+ $tablename = $self->drop_db_schema ? $tbl : lc $table;
}
+ my $lc_tblname = lc $tablename;
+
+ my $table_moniker = $self->_table2moniker($db_schema, $tbl);
+ my $table_class = $schema . q{::} . $table_moniker;
- my $table_moniker = $class->_loader_table2moniker($db_schema, $tbl);
- my $table_class = "$class\::$table_moniker";
+ $self->_inject($table_class, 'DBIx::Class::Core');
+ $self->_inject($table_class, @db_classes);
+ $self->_inject($table_class, @{$self->additional_base_classes});
+ $self->_use ($table_class, @{$self->additional_classes});
+ $self->_inject($table_class, @{$self->left_base_classes});
- $class->inject_base( $table_class, 'DBIx::Class::Core' );
- $_->require for @db_classes;
- $class->inject_base( $table_class, $_ ) for @db_classes;
- warn qq/\# Initializing table "$tablename" as "$table_class"\n/ if $class->_loader_debug;
- $table_class->table(lc $tablename);
+ warn qq/\# Initializing table "$tablename" as "$table_class"\n/
+ if $self->debug;
+ $table_class->table($lc_tblname);
- my ( $cols, $pks ) = $class->_loader_table_info($table);
+ my ( $cols, $pks ) = $self->_table_info($table);
carp("$table has no primary key") unless @$pks;
$table_class->add_columns(@$cols);
$table_class->set_primary_key(@$pks) if @$pks;
- my $code = "package $table_class;\n$additional_base$additional$left_base";
- warn qq/$code/ if $class->_loader_debug;
- warn qq/$table_class->table('$tablename');\n/ if $class->_loader_debug;
+ warn qq/$table_class->table('$tablename');\n/ if $self->debug;
my $columns = join "', '", @$cols;
- warn qq/$table_class->add_columns('$columns')\n/ if $class->_loader_debug;
+ warn qq/$table_class->add_columns('$columns')\n/ if $self->debug;
my $primaries = join "', '", @$pks;
- warn qq/$table_class->set_primary_key('$primaries')\n/ if $class->_loader_debug && @$pks;
- eval $code;
- croak qq/Couldn't load additional classes "$@"/ if $@;
- unshift @{"$table_class\::ISA"}, $_ foreach ( @{ $class->_loader_data->{left_base} } );
-
- $class->register_class($table_moniker, $table_class);
- $class->_loader_data->{TABLE_CLASSES}->{lc $tablename} = $table_class;
- $class->_loader_data->{MONIKERS}->{lc $tablename} = $table_moniker;
+ warn qq/$table_class->set_primary_key('$primaries')\n/
+ if $self->debug && @$pks;
+
+ $schema->register_class($table_moniker, $table_class);
+ $self->classes->{$lc_tblname} = $table_class;
+ $self->monikers->{$lc_tblname} = $table_moniker;
}
}
+=head3 tables
+
+Returns a sorted list of tables.
+
+ my @tables = $loader->tables;
+
+=cut
+
+sub tables {
+ my $self = shift;
+
+ return sort keys %{ $self->monikers };
+}
+
# Find and setup relationships
-sub _loader_relationships {
- my $class = shift;
- my $dbh = $class->storage->dbh;
+sub _load_relationships {
+ my $self = shift;
+
+ my $dbh = $self->schema->storage->dbh;
my $quoter = $dbh->get_info(29) || q{"};
- foreach my $table ( $class->tables ) {
+ foreach my $table ( $self->tables ) {
my $rels = {};
my $sth = $dbh->foreign_key_info( '',
- $class->_loader_data->{db_schema}, '', '', '', $table );
+ $self->db_schema, '', '', '', $table );
next if !$sth;
while(my $raw_rel = $sth->fetchrow_hashref) {
my $uk_tbl = lc $raw_rel->{UK_TABLE_NAME};
my $uk_col = lc $raw_rel->{UK_COLUMN_NAME};
my $fk_col = lc $raw_rel->{FK_COLUMN_NAME};
+ my $relid = lc $raw_rel->{UK_NAME};
$uk_tbl =~ s/$quoter//g;
$uk_col =~ s/$quoter//g;
$fk_col =~ s/$quoter//g;
- $rels->{$uk_tbl}->{$uk_col} = $fk_col;
+ $relid =~ s/$quoter//g;
+ $rels->{$relid}->{tbl} = $uk_tbl;
+ $rels->{$relid}->{cols}->{$uk_col} = $fk_col;
}
- foreach my $reltbl (keys %$rels) {
- my $cond = $rels->{$reltbl};
- eval { $class->_loader_make_relations( $table, $reltbl, $cond ) };
+ foreach my $relid (keys %$rels) {
+ my $reltbl = $rels->{$relid}->{tbl};
+ my $cond = $rels->{$relid}->{cols};
+ eval { $self->_make_cond_rel( $table, $reltbl, $cond ) };
warn qq/\# belongs_to_many failed "$@"\n\n/
- if $@ && $class->_loader_debug;
+ if $@ && $self->debug;
}
}
}
# Make a moniker from a table
-sub _loader_table2moniker {
- my ( $class, $db_schema, $table ) = @_;
+sub _table2moniker {
+ my ( $self, $db_schema, $table ) = @_;
my $db_schema_ns;
if($table) {
$db_schema = ucfirst lc $db_schema;
- $db_schema_ns = $db_schema if(!$class->_loader_data->{drop_db_schema});
+ $db_schema_ns = $db_schema if(!$self->drop_db_schema);
} else {
$table = $db_schema;
}
}
# Overload in driver class
-sub _loader_tables { croak "ABSTRACT METHOD" }
+sub _tables { croak "ABSTRACT METHOD" }
-sub _loader_table_info { croak "ABSTRACT METHOD" }
+sub _table_info { croak "ABSTRACT METHOD" }
=head1 SEE ALSO
projects / dbsrgits/DBIx-Class-Schema-Loader.git / blobdiff