use strict;
use warnings;
+use DBIx::Class::Carp;
+use Sub::Name 'subname';
+use Scalar::Util 'blessed';
+use DBIx::Class::_Util 'fail_on_internal_wantarray';
+use namespace::clean;
+
+our %_pod_inherit_config =
+ (
+ class_map => { 'DBIx::Class::Relationship::ManyToMany' => 'DBIx::Class::Relationship' }
+ );
+
sub many_to_many {
my ($class, $meth, $rel, $f_rel, $rel_attrs) = @_;
+
+ $class->throw_exception(
+ "missing relation in many-to-many"
+ ) unless $rel;
+
+ $class->throw_exception(
+ "missing foreign relation in many-to-many"
+ ) unless $f_rel;
+
{
no strict 'refs';
no warnings 'redefine';
my $add_meth = "add_to_${meth}";
my $remove_meth = "remove_from_${meth}";
my $set_meth = "set_${meth}";
+ my $rs_meth = "${meth}_rs";
+
+ for ($add_meth, $remove_meth, $set_meth, $rs_meth) {
+ if ( $class->can ($_) ) {
+ carp (<<"EOW") unless $ENV{DBIC_OVERWRITE_HELPER_METHODS_OK};
+
+***************************************************************************
+The many-to-many relationship '$meth' is trying to create a utility method
+called $_.
+This will completely overwrite one such already existing method on class
+$class.
- *{"${class}::${meth}"} = sub {
+You almost certainly want to rename your method or the many-to-many
+relationship, as the functionality of the original method will not be
+accessible anymore.
+
+To disable this warning set to a true value the environment variable
+DBIC_OVERWRITE_HELPER_METHODS_OK
+
+***************************************************************************
+EOW
+ }
+ }
+
+ $rel_attrs->{alias} ||= $f_rel;
+
+ my $rs_meth_name = join '::', $class, $rs_meth;
+ *$rs_meth_name = subname $rs_meth_name, sub {
my $self = shift;
my $attrs = @_ > 1 && ref $_[$#_] eq 'HASH' ? pop(@_) : {};
- my @args = ($f_rel, @_ > 0 ? @_ : undef, { %{$rel_attrs||{}}, %$attrs });
- $self->search_related($rel)->search_related(
+ my $rs = $self->search_related($rel)->search_related(
$f_rel, @_ > 0 ? @_ : undef, { %{$rel_attrs||{}}, %$attrs }
);
+ return $rs;
};
- *{"${class}::${add_meth}"} = sub {
+ my $meth_name = join '::', $class, $meth;
+ *$meth_name = subname $meth_name, sub {
+ DBIx::Class::_ENV_::ASSERT_NO_INTERNAL_WANTARRAY and my $sog = fail_on_internal_wantarray;
my $self = shift;
- @_ > 0 or $self->throw_exception(
+ my $rs = $self->$rs_meth( @_ );
+ return (wantarray ? $rs->all : $rs);
+ };
+
+ my $add_meth_name = join '::', $class, $add_meth;
+ *$add_meth_name = subname $add_meth_name, sub {
+ my $self = shift;
+ @_ or $self->throw_exception(
"${add_meth} needs an object or hashref"
);
- my $source = $self->result_source;
- my $schema = $source->schema;
- my $rel_source_name = $source->relationship_info($rel)->{source};
- my $rel_source = $schema->resultset($rel_source_name)->result_source;
- my $f_rel_source_name = $rel_source->relationship_info($f_rel)->{source};
- my $f_rel_rs = $schema->resultset($f_rel_source_name)->search({}, $rel_attrs||{});
-
- my $obj;
- if (ref $_[0]) {
- if (ref $_[0] eq 'HASH') {
- $obj = $f_rel_rs->create($_[0]);
- } else {
- $obj = $_[0];
- }
- } else {
- $obj = $f_rel_rs->create({@_});
- }
- my $link_vals = @_ > 1 && ref $_[$#_] eq 'HASH' ? pop(@_) : {};
- my $link = $self->search_related($rel)->new_result({});
- $link->set_from_related($f_rel, $obj);
- $link->set_columns($link_vals);
+ my $link = $self->search_related_rs($rel)->new_result(
+ ( @_ > 1 && ref $_[-1] eq 'HASH' )
+ ? pop
+ : {}
+ );
+
+ my $far_obj = defined blessed $_[0]
+ ? $_[0]
+ : $self->result_source
+ ->related_source( $rel )
+ ->related_source( $f_rel )
+ ->resultset->search_rs( {}, $rel_attrs||{} )
+ ->find_or_create( ref $_[0] eq 'HASH' ? $_[0] : {@_} )
+ ;
+
+ $link->set_from_related($f_rel, $far_obj);
$link->insert();
+
+ return $far_obj;
};
- *{"${class}::${set_meth}"} = sub {
+ my $set_meth_name = join '::', $class, $set_meth;
+ *$set_meth_name = subname $set_meth_name, sub {
my $self = shift;
@_ > 0 or $self->throw_exception(
"{$set_meth} needs a list of objects or hashrefs"
);
- $self->search_related($rel, {})->delete;
- $self->$add_meth(shift) while (defined $_[0]);
+ my @to_set = (ref($_[0]) eq 'ARRAY' ? @{ $_[0] } : @_);
+ # if there is a where clause in the attributes, ensure we only delete
+ # rows that are within the where restriction
+ if ($rel_attrs && $rel_attrs->{where}) {
+ $self->search_related( $rel, $rel_attrs->{where},{join => $f_rel})->delete;
+ } else {
+ $self->search_related( $rel, {} )->delete;
+ }
+ # add in the set rel objects
+ $self->$add_meth($_, ref($_[1]) ? $_[1] : {}) for (@to_set);
};
- *{"${class}::${remove_meth}"} = sub {
- my $self = shift;
- @_ > 0 && ref $_[0] ne 'HASH'
- or $self->throw_exception("${remove_meth} needs an object");
- my $obj = shift;
+ my $remove_meth_name = join '::', $class, $remove_meth;
+ *$remove_meth_name = subname $remove_meth_name, sub {
+ my ($self, $obj) = @_;
+ $self->throw_exception("${remove_meth} needs an object")
+ unless blessed ($obj);
my $rel_source = $self->search_related($rel)->result_source;
my $cond = $rel_source->relationship_info($f_rel)->{cond};
- my $link_cond = $rel_source->resolve_condition(
- $cond, $obj, $f_rel
+ my ($link_cond, $crosstable) = $rel_source->_resolve_condition(
+ $cond, $obj, $f_rel, $f_rel
);
+
+ $self->throw_exception(
+ "Relationship '$rel' does not resolve to a join-free condition, "
+ ."unable to use with the ManyToMany helper '$f_rel'"
+ ) if $crosstable;
+
$self->search_related($rel, $link_cond)->delete;
};
projects / dbsrgits/DBIx-Class.git / blobdiff