duh
[dbsrgits/DBIx-Class.git] / lib / DBIx / Class / Row.pm
1 package DBIx::Class::Row;
2
3 use strict;
4 use warnings;
5
6 use base qw/DBIx::Class/;
7 use Carp::Clan qw/^DBIx::Class/;
8 use Scalar::Util ();
9 use Scope::Guard;
10
11 ###
12 ### Internal method
13 ### Do not use
14 ###
15 BEGIN {
16   *MULTICREATE_DEBUG =
17     $ENV{DBIC_MULTICREATE_DEBUG}
18       ? sub () { 1 }
19       : sub () { 0 };
20 }
21
22 __PACKAGE__->mk_group_accessors('simple' => qw/_source_handle/);
23
24 =head1 NAME
25
26 DBIx::Class::Row - Basic row methods
27
28 =head1 SYNOPSIS
29
30 =head1 DESCRIPTION
31
32 This class is responsible for defining and doing basic operations on rows
33 derived from L<DBIx::Class::ResultSource> objects.
34
35 Row objects are returned from L<DBIx::Class::ResultSet>s using the
36 L<create|DBIx::Class::ResultSet/create>, L<find|DBIx::Class::ResultSet/find>,
37 L<next|DBIx::Class::ResultSet/next> and L<all|DBIx::Class::ResultSet/all> methods,
38 as well as invocations of 'single' (
39 L<belongs_to|DBIx::Class::Relationship/belongs_to>,
40 L<has_one|DBIx::Class::Relationship/has_one> or
41 L<might_have|DBIx::Class::Relationship/might_have>)
42 relationship accessors of L<DBIx::Class::Row> objects.
43
44 =head1 METHODS
45
46 =head2 new
47
48   my $row = My::Class->new(\%attrs);
49
50   my $row = $schema->resultset('MySource')->new(\%colsandvalues);
51
52 =over
53
54 =item Arguments: \%attrs or \%colsandvalues
55
56 =item Returns: A Row object
57
58 =back
59
60 While you can create a new row object by calling C<new> directly on
61 this class, you are better off calling it on a
62 L<DBIx::Class::ResultSet> object.
63
64 When calling it directly, you will not get a complete, usable row
65 object until you pass or set the C<source_handle> attribute, to a
66 L<DBIx::Class::ResultSource> instance that is attached to a
67 L<DBIx::Class::Schema> with a valid connection.
68
69 C<$attrs> is a hashref of column name, value data. It can also contain
70 some other attributes such as the C<source_handle>.
71
72 Passing an object, or an arrayref of objects as a value will call
73 L<DBIx::Class::Relationship::Base/set_from_related> for you. When
74 passed a hashref or an arrayref of hashrefs as the value, these will
75 be turned into objects via new_related, and treated as if you had
76 passed objects.
77
78 For a more involved explanation, see L<DBIx::Class::ResultSet/create>.
79
80 Please note that if a value is not passed to new, no value will be sent
81 in the SQL INSERT call, and the column will therefore assume whatever
82 default value was specified in your database. While DBIC will retrieve the
83 value of autoincrement columns, it will never make an explicit database
84 trip to retrieve default values assigned by the RDBMS. You can explicitly
85 request that all values be fetched back from the database by calling
86 L</discard_changes>, or you can supply an explicit C<undef> to columns
87 with NULL as the default, and save yourself a SELECT.
88
89  CAVEAT:
90
91  The behavior described above will backfire if you use a foreign key column
92  with a database-defined default. If you call the relationship accessor on
93  an object that doesn't have a set value for the FK column, DBIC will throw
94  an exception, as it has no way of knowing the PK of the related object (if
95  there is one).
96
97 =cut
98
99 ## It needs to store the new objects somewhere, and call insert on that list later when insert is called on this object. We may need an accessor for these so the user can retrieve them, if just doing ->new().
100 ## This only works because DBIC doesnt yet care to check whether the new_related objects have been passed all their mandatory columns
101 ## When doing the later insert, we need to make sure the PKs are set.
102 ## using _relationship_data in new and funky ways..
103 ## check Relationship::CascadeActions and Relationship::Accessor for compat
104 ## tests!
105
106 sub __new_related_find_or_new_helper {
107   my ($self, $relname, $data) = @_;
108   if ($self->__their_pk_needs_us($relname, $data)) {
109     MULTICREATE_DEBUG and warn "MC $self constructing $relname via new_result";
110     return $self->result_source
111                 ->related_source($relname)
112                 ->resultset
113                 ->new_result($data);
114   }
115   if ($self->result_source->_pk_depends_on($relname, $data)) {
116     MULTICREATE_DEBUG and warn "MC $self constructing $relname via find_or_new";
117     return $self->result_source
118                 ->related_source($relname)
119                 ->resultset
120                 ->find_or_new($data);
121   }
122   MULTICREATE_DEBUG and warn "MC $self constructing $relname via find_or_new_related";
123   return $self->find_or_new_related($relname, $data);
124 }
125
126 sub __their_pk_needs_us { # this should maybe be in resultsource.
127   my ($self, $relname, $data) = @_;
128   my $source = $self->result_source;
129   my $reverse = $source->reverse_relationship_info($relname);
130   my $rel_source = $source->related_source($relname);
131   my $us = { $self->get_columns };
132   foreach my $key (keys %$reverse) {
133     # if their primary key depends on us, then we have to
134     # just create a result and we'll fill it out afterwards
135     return 1 if $rel_source->_pk_depends_on($key, $us);
136   }
137   return 0;
138 }
139
140 sub new {
141   my ($class, $attrs) = @_;
142   $class = ref $class if ref $class;
143
144   my $new = {
145       _column_data          => {},
146   };
147   bless $new, $class;
148
149   if (my $handle = delete $attrs->{-source_handle}) {
150     $new->_source_handle($handle);
151   }
152
153   my $source;
154   if ($source = delete $attrs->{-result_source}) {
155     $new->result_source($source);
156   }
157
158   if (my $related = delete $attrs->{-from_resultset}) {
159     @{$new->{_ignore_at_insert}={}}{@$related} = ();
160   }
161
162   if ($attrs) {
163     $new->throw_exception("attrs must be a hashref")
164       unless ref($attrs) eq 'HASH';
165     
166     my ($related,$inflated);
167     ## Pretend all the rels are actual objects, unset below if not, for insert() to fix
168     $new->{_rel_in_storage} = 1;
169
170     foreach my $key (keys %$attrs) {
171       if (ref $attrs->{$key}) {
172         ## Can we extract this lot to use with update(_or .. ) ?
173         confess "Can't do multi-create without result source" unless $source;
174         my $info = $source->relationship_info($key);
175         if ($info && $info->{attrs}{accessor}
176           && $info->{attrs}{accessor} eq 'single')
177         {
178           my $rel_obj = delete $attrs->{$key};
179           if(!Scalar::Util::blessed($rel_obj)) {
180             $rel_obj = $new->__new_related_find_or_new_helper($key, $rel_obj);
181           }
182
183           if ($rel_obj->in_storage) {
184             $new->set_from_related($key, $rel_obj);
185           } else {
186             $new->{_rel_in_storage} = 0;
187             MULTICREATE_DEBUG and warn "MC $new uninserted $key $rel_obj\n";
188           }
189
190           $related->{$key} = $rel_obj;
191           next;
192         } elsif ($info && $info->{attrs}{accessor}
193             && $info->{attrs}{accessor} eq 'multi'
194             && ref $attrs->{$key} eq 'ARRAY') {
195           my $others = delete $attrs->{$key};
196           my $total = @$others;
197           my @objects;
198           foreach my $idx (0 .. $#$others) {
199             my $rel_obj = $others->[$idx];
200             if(!Scalar::Util::blessed($rel_obj)) {
201               $rel_obj = $new->__new_related_find_or_new_helper($key, $rel_obj);
202             }
203
204             if ($rel_obj->in_storage) {
205               $new->set_from_related($key, $rel_obj);
206             } else {
207               $new->{_rel_in_storage} = 0;
208               MULTICREATE_DEBUG and
209                 warn "MC $new uninserted $key $rel_obj (${\($idx+1)} of $total)\n";
210             }
211             $new->set_from_related($key, $rel_obj) if $rel_obj->in_storage;
212             push(@objects, $rel_obj);
213           }
214           $related->{$key} = \@objects;
215           next;
216         } elsif ($info && $info->{attrs}{accessor}
217           && $info->{attrs}{accessor} eq 'filter')
218         {
219           ## 'filter' should disappear and get merged in with 'single' above!
220           my $rel_obj = delete $attrs->{$key};
221           if(!Scalar::Util::blessed($rel_obj)) {
222             $rel_obj = $new->__new_related_find_or_new_helper($key, $rel_obj);
223           }
224           unless ($rel_obj->in_storage) {
225             $new->{_rel_in_storage} = 0;
226             MULTICREATE_DEBUG and warn "MC $new uninserted $key $rel_obj";
227           }
228           $inflated->{$key} = $rel_obj;
229           next;
230         } elsif ($class->has_column($key)
231             && $class->column_info($key)->{_inflate_info}) {
232           $inflated->{$key} = $attrs->{$key};
233           next;
234         }
235       }
236       $new->throw_exception("No such column $key on $class")
237         unless $class->has_column($key);
238       $new->store_column($key => $attrs->{$key});          
239     }
240
241     $new->{_relationship_data} = $related if $related;
242     $new->{_inflated_column} = $inflated if $inflated;
243   }
244
245   return $new;
246 }
247
248 =head2 insert
249
250   $row->insert;
251
252 =over
253
254 =item Arguments: none
255
256 =item Returns: The Row object
257
258 =back
259
260 Inserts an object previously created by L</new> into the database if
261 it isn't already in there. Returns the object itself. Requires the
262 object's result source to be set, or the class to have a
263 result_source_instance method. To insert an entirely new row into
264 the database, use C<create> (see L<DBIx::Class::ResultSet/create>).
265
266 To fetch an uninserted row object, call
267 L<new|DBIx::Class::ResultSet/new> on a resultset.
268
269 This will also insert any uninserted, related objects held inside this
270 one, see L<DBIx::Class::ResultSet/create> for more details.
271
272 =cut
273
274 sub insert {
275   my ($self) = @_;
276   return $self if $self->in_storage;
277   my $source = $self->result_source;
278   $source ||=  $self->result_source($self->result_source_instance)
279     if $self->can('result_source_instance');
280   $self->throw_exception("No result_source set on this object; can't insert")
281     unless $source;
282
283   my $rollback_guard;
284
285   # Check if we stored uninserted relobjs here in new()
286   my %related_stuff = (%{$self->{_relationship_data} || {}}, 
287                        %{$self->{_inflated_column} || {}});
288
289   if(!$self->{_rel_in_storage}) {
290
291     # The guard will save us if we blow out of this scope via die
292     $rollback_guard = $source->storage->txn_scope_guard;
293
294     ## Should all be in relationship_data, but we need to get rid of the
295     ## 'filter' reltype..
296     ## These are the FK rels, need their IDs for the insert.
297
298     my @pri = $self->primary_columns;
299
300     REL: foreach my $relname (keys %related_stuff) {
301
302       my $rel_obj = $related_stuff{$relname};
303
304       next REL unless (Scalar::Util::blessed($rel_obj)
305                        && $rel_obj->isa('DBIx::Class::Row'));
306
307       next REL unless $source->_pk_depends_on(
308                         $relname, { $rel_obj->get_columns }
309                       );
310
311       MULTICREATE_DEBUG and warn "MC $self pre-reconstructing $relname $rel_obj\n";
312
313       my $them = { %{$rel_obj->{_relationship_data} || {} }, $rel_obj->get_inflated_columns };
314       my $re = $self->result_source
315                     ->related_source($relname)
316                     ->resultset
317                     ->find_or_create($them);
318       %{$rel_obj} = %{$re};
319       $self->set_from_related($relname, $rel_obj);
320       delete $related_stuff{$relname};
321     }
322   }
323
324   MULTICREATE_DEBUG and do {
325     no warnings 'uninitialized';
326     warn "MC $self inserting (".join(', ', $self->get_columns).")\n";
327   };
328   my $updated_cols = $source->storage->insert($source, { $self->get_columns });
329   foreach my $col (keys %$updated_cols) {
330     $self->store_column($col, $updated_cols->{$col});
331   }
332
333   ## PK::Auto
334   my @auto_pri = grep {
335                    !defined $self->get_column($_) || 
336                    ref($self->get_column($_)) eq 'SCALAR'
337                  } $self->primary_columns;
338
339   if (@auto_pri) {
340     #$self->throw_exception( "More than one possible key found for auto-inc on ".ref $self )
341     #  if defined $too_many;
342     MULTICREATE_DEBUG and warn "MC $self fetching missing PKs ".join(', ', @auto_pri)."\n";
343     my $storage = $self->result_source->storage;
344     $self->throw_exception( "Missing primary key but Storage doesn't support last_insert_id" )
345       unless $storage->can('last_insert_id');
346     my @ids = $storage->last_insert_id($self->result_source,@auto_pri);
347     $self->throw_exception( "Can't get last insert id" )
348       unless (@ids == @auto_pri);
349     $self->store_column($auto_pri[$_] => $ids[$_]) for 0 .. $#ids;
350   }
351
352
353   $self->{_dirty_columns} = {};
354   $self->{related_resultsets} = {};
355
356   if(!$self->{_rel_in_storage}) {
357     ## Now do the relationships that need our ID (has_many etc.)
358     foreach my $relname (keys %related_stuff) {
359       my $rel_obj = $related_stuff{$relname};
360       my @cands;
361       if (Scalar::Util::blessed($rel_obj)
362           && $rel_obj->isa('DBIx::Class::Row')) {
363         @cands = ($rel_obj);
364       } elsif (ref $rel_obj eq 'ARRAY') {
365         @cands = @$rel_obj;
366       }
367       if (@cands) {
368         my $reverse = $source->reverse_relationship_info($relname);
369         foreach my $obj (@cands) {
370           $obj->set_from_related($_, $self) for keys %$reverse;
371           my $them = { %{$obj->{_relationship_data} || {} }, $obj->get_inflated_columns };
372           if ($self->__their_pk_needs_us($relname, $them)) {
373             if (exists $self->{_ignore_at_insert}{$relname}) {
374               MULTICREATE_DEBUG and warn "MC $self skipping post-insert on $relname";
375             } else {
376               MULTICREATE_DEBUG and warn "MC $self re-creating $relname $obj";
377               my $re = $self->result_source
378                             ->related_source($relname)
379                             ->resultset
380                             ->find_or_create($them);
381               %{$obj} = %{$re};
382               MULTICREATE_DEBUG and warn "MC $self new $relname $obj";
383             }
384           } else {
385             MULTICREATE_DEBUG and warn "MC $self post-inserting $obj";
386             $obj->insert();
387           }
388         }
389       }
390     }
391     delete $self->{_ignore_at_insert};
392     $rollback_guard->commit;
393   }
394
395   $self->in_storage(1);
396   undef $self->{_orig_ident};
397   return $self;
398 }
399
400 =head2 in_storage
401
402   $row->in_storage; # Get value
403   $row->in_storage(1); # Set value
404
405 =over
406
407 =item Arguments: none or 1|0
408
409 =item Returns: 1|0
410
411 =back
412
413 Indicates whether the object exists as a row in the database or
414 not. This is set to true when L<DBIx::Class::ResultSet/find>,
415 L<DBIx::Class::ResultSet/create> or L<DBIx::Class::ResultSet/insert>
416 are used. 
417
418 Creating a row object using L<DBIx::Class::ResultSet/new>, or calling
419 L</delete> on one, sets it to false.
420
421 =cut
422
423 sub in_storage {
424   my ($self, $val) = @_;
425   $self->{_in_storage} = $val if @_ > 1;
426   return $self->{_in_storage};
427 }
428
429 =head2 update
430
431   $row->update(\%columns?)
432
433 =over
434
435 =item Arguments: none or a hashref
436
437 =item Returns: The Row object
438
439 =back
440
441 Throws an exception if the row object is not yet in the database,
442 according to L</in_storage>.
443
444 This method issues an SQL UPDATE query to commit any changes to the
445 object to the database if required.
446
447 Also takes an optional hashref of C<< column_name => value> >> pairs
448 to update on the object first. Be aware that the hashref will be
449 passed to C<set_inflated_columns>, which might edit it in place, so
450 don't rely on it being the same after a call to C<update>.  If you
451 need to preserve the hashref, it is sufficient to pass a shallow copy
452 to C<update>, e.g. ( { %{ $href } } )
453
454 If the values passed or any of the column values set on the object
455 contain scalar references, eg:
456
457   $row->last_modified(\'NOW()');
458   # OR
459   $row->update({ last_modified => \'NOW()' });
460
461 The update will pass the values verbatim into SQL. (See
462 L<SQL::Abstract> docs).  The values in your Row object will NOT change
463 as a result of the update call, if you want the object to be updated
464 with the actual values from the database, call L</discard_changes>
465 after the update.
466
467   $row->update()->discard_changes();
468
469 To determine before calling this method, which column values have
470 changed and will be updated, call L</get_dirty_columns>.
471
472 To check if any columns will be updated, call L</is_changed>.
473
474 To force a column to be updated, call L</make_column_dirty> before
475 this method.
476
477 =cut
478
479 sub update {
480   my ($self, $upd) = @_;
481   $self->throw_exception( "Not in database" ) unless $self->in_storage;
482   my $ident_cond = $self->ident_condition;
483   $self->throw_exception("Cannot safely update a row in a PK-less table")
484     if ! keys %$ident_cond;
485
486   $self->set_inflated_columns($upd) if $upd;
487   my %to_update = $self->get_dirty_columns;
488   return $self unless keys %to_update;
489   my $rows = $self->result_source->storage->update(
490                $self->result_source, \%to_update,
491                $self->{_orig_ident} || $ident_cond
492              );
493   if ($rows == 0) {
494     $self->throw_exception( "Can't update ${self}: row not found" );
495   } elsif ($rows > 1) {
496     $self->throw_exception("Can't update ${self}: updated more than one row");
497   }
498   $self->{_dirty_columns} = {};
499   $self->{related_resultsets} = {};
500   undef $self->{_orig_ident};
501   return $self;
502 }
503
504 =head2 delete
505
506   $row->delete
507
508 =over
509
510 =item Arguments: none
511
512 =item Returns: The Row object
513
514 =back
515
516 Throws an exception if the object is not in the database according to
517 L</in_storage>. Runs an SQL DELETE statement using the primary key
518 values to locate the row.
519
520 The object is still perfectly usable, but L</in_storage> will
521 now return 0 and the object must be reinserted using L</insert>
522 before it can be used to L</update> the row again. 
523
524 If you delete an object in a class with a C<has_many> relationship, an
525 attempt is made to delete all the related objects as well. To turn
526 this behaviour off, pass C<< cascade_delete => 0 >> in the C<$attr>
527 hashref of the relationship, see L<DBIx::Class::Relationship>. Any
528 database-level cascade or restrict will take precedence over a
529 DBIx-Class-based cascading delete. 
530
531 If you delete an object within a txn_do() (see L<DBIx::Class::Storage/txn_do>)
532 and the transaction subsequently fails, the row object will remain marked as
533 not being in storage. If you know for a fact that the object is still in
534 storage (i.e. by inspecting the cause of the transaction's failure), you can
535 use C<< $obj->in_storage(1) >> to restore consistency between the object and
536 the database. This would allow a subsequent C<< $obj->delete >> to work
537 as expected.
538
539 See also L<DBIx::Class::ResultSet/delete>.
540
541 =cut
542
543 sub delete {
544   my $self = shift;
545   if (ref $self) {
546     $self->throw_exception( "Not in database" ) unless $self->in_storage;
547     my $ident_cond = $self->{_orig_ident} || $self->ident_condition;
548     $self->throw_exception("Cannot safely delete a row in a PK-less table")
549       if ! keys %$ident_cond;
550     foreach my $column (keys %$ident_cond) {
551             $self->throw_exception("Can't delete the object unless it has loaded the primary keys")
552               unless exists $self->{_column_data}{$column};
553     }
554     $self->result_source->storage->delete(
555       $self->result_source, $ident_cond);
556     $self->in_storage(undef);
557   } else {
558     $self->throw_exception("Can't do class delete without a ResultSource instance")
559       unless $self->can('result_source_instance');
560     my $attrs = @_ > 1 && ref $_[$#_] eq 'HASH' ? { %{pop(@_)} } : {};
561     my $query = ref $_[0] eq 'HASH' ? $_[0] : {@_};
562     $self->result_source_instance->resultset->search(@_)->delete;
563   }
564   return $self;
565 }
566
567 =head2 get_column
568
569   my $val = $row->get_column($col);
570
571 =over
572
573 =item Arguments: $columnname
574
575 =item Returns: The value of the column
576
577 =back
578
579 Throws an exception if the column name given doesn't exist according
580 to L</has_column>.
581
582 Returns a raw column value from the row object, if it has already
583 been fetched from the database or set by an accessor.
584
585 If an L<inflated value|DBIx::Class::InflateColumn> has been set, it
586 will be deflated and returned.
587
588 Note that if you used the C<columns> or the C<select/as>
589 L<search attributes|DBIx::Class::ResultSet/ATTRIBUTES> on the resultset from
590 which C<$row> was derived, and B<did not include> C<$columnname> in the list,
591 this method will return C<undef> even if the database contains some value.
592
593 To retrieve all loaded column values as a hash, use L</get_columns>.
594
595 =cut
596
597 sub get_column {
598   my ($self, $column) = @_;
599   $self->throw_exception( "Can't fetch data as class method" ) unless ref $self;
600   return $self->{_column_data}{$column} if exists $self->{_column_data}{$column};
601   if (exists $self->{_inflated_column}{$column}) {
602     return $self->store_column($column,
603       $self->_deflated_column($column, $self->{_inflated_column}{$column}));   
604   }
605   $self->throw_exception( "No such column '${column}'" ) unless $self->has_column($column);
606   return undef;
607 }
608
609 =head2 has_column_loaded
610
611   if ( $row->has_column_loaded($col) ) {
612      print "$col has been loaded from db";
613   }
614
615 =over
616
617 =item Arguments: $columnname
618
619 =item Returns: 0|1
620
621 =back
622
623 Returns a true value if the column value has been loaded from the
624 database (or set locally).
625
626 =cut
627
628 sub has_column_loaded {
629   my ($self, $column) = @_;
630   $self->throw_exception( "Can't call has_column data as class method" ) unless ref $self;
631   return 1 if exists $self->{_inflated_column}{$column};
632   return exists $self->{_column_data}{$column};
633 }
634
635 =head2 get_columns
636
637   my %data = $row->get_columns;
638
639 =over
640
641 =item Arguments: none
642
643 =item Returns: A hash of columnname, value pairs.
644
645 =back
646
647 Returns all loaded column data as a hash, containing raw values. To
648 get just one value for a particular column, use L</get_column>.
649
650 =cut
651
652 sub get_columns {
653   my $self = shift;
654   if (exists $self->{_inflated_column}) {
655     foreach my $col (keys %{$self->{_inflated_column}}) {
656       $self->store_column($col, $self->_deflated_column($col, $self->{_inflated_column}{$col}))
657         unless exists $self->{_column_data}{$col};
658     }
659   }
660   return %{$self->{_column_data}};
661 }
662
663 =head2 get_dirty_columns
664
665   my %data = $row->get_dirty_columns;
666
667 =over
668
669 =item Arguments: none
670
671 =item Returns: A hash of column, value pairs
672
673 =back
674
675 Only returns the column, value pairs for those columns that have been
676 changed on this object since the last L</update> or L</insert> call.
677
678 See L</get_columns> to fetch all column/value pairs.
679
680 =cut
681
682 sub get_dirty_columns {
683   my $self = shift;
684   return map { $_ => $self->{_column_data}{$_} }
685            keys %{$self->{_dirty_columns}};
686 }
687
688 =head2 make_column_dirty
689
690   $row->make_column_dirty($col)
691
692 =over
693
694 =item Arguments: $columnname
695
696 =item Returns: undefined
697
698 =back
699
700 Throws an exception if the column does not exist.
701
702 Marks a column as having been changed regardless of whether it has
703 really changed.  
704
705 =cut
706 sub make_column_dirty {
707   my ($self, $column) = @_;
708
709   $self->throw_exception( "No such column '${column}'" )
710     unless exists $self->{_column_data}{$column} || $self->has_column($column);
711   $self->{_dirty_columns}{$column} = 1;
712 }
713
714 =head2 get_inflated_columns
715
716   my %inflated_data = $obj->get_inflated_columns;
717
718 =over
719
720 =item Arguments: none
721
722 =item Returns: A hash of column, object|value pairs
723
724 =back
725
726 Returns a hash of all column keys and associated values. Values for any
727 columns set to use inflation will be inflated and returns as objects.
728
729 See L</get_columns> to get the uninflated values.
730
731 See L<DBIx::Class::InflateColumn> for how to setup inflation.
732
733 =cut
734
735 sub get_inflated_columns {
736   my $self = shift;
737   return map {
738     my $accessor = $self->column_info($_)->{'accessor'} || $_;
739     ($_ => $self->$accessor);
740   } grep $self->has_column_loaded($_), $self->columns;
741 }
742
743 =head2 set_column
744
745   $row->set_column($col => $val);
746
747 =over
748
749 =item Arguments: $columnname, $value
750
751 =item Returns: $value
752
753 =back
754
755 Sets a raw column value. If the new value is different from the old one,
756 the column is marked as dirty for when you next call L</update>.
757
758 If passed an object or reference as a value, this method will happily
759 attempt to store it, and a later L</insert> or L</update> will try and
760 stringify/numify as appropriate. To set an object to be deflated
761 instead, see L</set_inflated_columns>.
762
763 =cut
764
765 sub set_column {
766   my ($self, $column, $new_value) = @_;
767
768   $self->{_orig_ident} ||= $self->ident_condition;
769   my $old_value = $self->get_column($column);
770
771   $self->store_column($column, $new_value);
772   $self->{_dirty_columns}{$column} = 1
773     if (defined $old_value xor defined $new_value) || (defined $old_value && $old_value ne $new_value);
774
775   # XXX clear out the relation cache for this column
776   delete $self->{related_resultsets}{$column};
777
778   return $new_value;
779 }
780
781 =head2 set_columns
782
783   $row->set_columns({ $col => $val, ... });
784
785 =over 
786
787 =item Arguments: \%columndata
788
789 =item Returns: The Row object
790
791 =back
792
793 Sets multiple column, raw value pairs at once.
794
795 Works as L</set_column>.
796
797 =cut
798
799 sub set_columns {
800   my ($self,$data) = @_;
801   foreach my $col (keys %$data) {
802     $self->set_column($col,$data->{$col});
803   }
804   return $self;
805 }
806
807 =head2 set_inflated_columns
808
809   $row->set_inflated_columns({ $col => $val, $relname => $obj, ... });
810
811 =over
812
813 =item Arguments: \%columndata
814
815 =item Returns: The Row object
816
817 =back
818
819 Sets more than one column value at once. Any inflated values are
820 deflated and the raw values stored. 
821
822 Any related values passed as Row objects, using the relation name as a
823 key, are reduced to the appropriate foreign key values and stored. If
824 instead of related row objects, a hashref of column, value data is
825 passed, will create the related object first then store.
826
827 Will even accept arrayrefs of data as a value to a
828 L<DBIx::Class::Relationship/has_many> key, and create the related
829 objects if necessary.
830
831 Be aware that the input hashref might be edited in place, so dont rely
832 on it being the same after a call to C<set_inflated_columns>. If you
833 need to preserve the hashref, it is sufficient to pass a shallow copy
834 to C<set_inflated_columns>, e.g. ( { %{ $href } } )
835
836 See also L<DBIx::Class::Relationship::Base/set_from_related>.
837
838 =cut
839
840 sub set_inflated_columns {
841   my ( $self, $upd ) = @_;
842   foreach my $key (keys %$upd) {
843     if (ref $upd->{$key}) {
844       my $info = $self->relationship_info($key);
845       if ($info && $info->{attrs}{accessor}
846         && $info->{attrs}{accessor} eq 'single')
847       {
848         my $rel = delete $upd->{$key};
849         $self->set_from_related($key => $rel);
850         $self->{_relationship_data}{$key} = $rel;
851       } elsif ($info && $info->{attrs}{accessor}
852         && $info->{attrs}{accessor} eq 'multi') {
853           $self->throw_exception(
854             "Recursive update is not supported over relationships of type multi ($key)"
855           );
856       }
857       elsif ($self->has_column($key)
858         && exists $self->column_info($key)->{_inflate_info})
859       {
860         $self->set_inflated_column($key, delete $upd->{$key});
861       }
862     }
863   }
864   $self->set_columns($upd);    
865 }
866
867 =head2 copy
868
869   my $copy = $orig->copy({ change => $to, ... });
870
871 =over
872
873 =item Arguments: \%replacementdata
874
875 =item Returns: The Row object copy
876
877 =back
878
879 Inserts a new row into the database, as a copy of the original
880 object. If a hashref of replacement data is supplied, these will take
881 precedence over data in the original.
882
883 Relationships will be followed by the copy procedure B<only> if the
884 relationship specifes a true value for its
885 L<cascade_copy|DBIx::Class::Relationship::Base> attribute. C<cascade_copy>
886 is set by default on C<has_many> relationships and unset on all others.
887
888 =cut
889
890 sub copy {
891   my ($self, $changes) = @_;
892   $changes ||= {};
893   my $col_data = { %{$self->{_column_data}} };
894   foreach my $col (keys %$col_data) {
895     delete $col_data->{$col}
896       if $self->result_source->column_info($col)->{is_auto_increment};
897   }
898
899   my $new = { _column_data => $col_data };
900   bless $new, ref $self;
901
902   $new->result_source($self->result_source);
903   $new->set_inflated_columns($changes);
904   $new->insert;
905
906   # Its possible we'll have 2 relations to the same Source. We need to make 
907   # sure we don't try to insert the same row twice esle we'll violate unique
908   # constraints
909   my $rels_copied = {};
910
911   foreach my $rel ($self->result_source->relationships) {
912     my $rel_info = $self->result_source->relationship_info($rel);
913
914     next unless $rel_info->{attrs}{cascade_copy};
915   
916     my $resolved = $self->result_source->_resolve_condition(
917       $rel_info->{cond}, $rel, $new
918     );
919
920     my $copied = $rels_copied->{ $rel_info->{source} } ||= {};
921     foreach my $related ($self->search_related($rel)) {
922       my $id_str = join("\0", $related->id);
923       next if $copied->{$id_str};
924       $copied->{$id_str} = 1;
925       my $rel_copy = $related->copy($resolved);
926     }
927  
928   }
929   return $new;
930 }
931
932 =head2 store_column
933
934   $row->store_column($col => $val);
935
936 =over
937
938 =item Arguments: $columnname, $value
939
940 =item Returns: The value sent to storage
941
942 =back
943
944 Set a raw value for a column without marking it as changed. This
945 method is used internally by L</set_column> which you should probably
946 be using.
947
948 This is the lowest level at which data is set on a row object,
949 extend this method to catch all data setting methods.
950
951 =cut
952
953 sub store_column {
954   my ($self, $column, $value) = @_;
955   $self->throw_exception( "No such column '${column}'" )
956     unless exists $self->{_column_data}{$column} || $self->has_column($column);
957   $self->throw_exception( "set_column called for ${column} without value" )
958     if @_ < 3;
959   return $self->{_column_data}{$column} = $value;
960 }
961
962 =head2 inflate_result
963
964   Class->inflate_result($result_source, \%me, \%prefetch?)
965
966 =over
967
968 =item Arguments: $result_source, \%columndata, \%prefetcheddata
969
970 =item Returns: A Row object
971
972 =back
973
974 All L<DBIx::Class::ResultSet> methods that retrieve data from the
975 database and turn it into row objects call this method.
976
977 Extend this method in your Result classes to hook into this process,
978 for example to rebless the result into a different class.
979
980 Reblessing can also be done more easily by setting C<result_class> in
981 your Result class. See L<DBIx::Class::ResultSource/result_class>.
982
983 Different types of results can also be created from a particular
984 L<DBIx::Class::ResultSet>, see L<DBIx::Class::ResultSet/result_class>.
985
986 =cut
987
988 sub inflate_result {
989   my ($class, $source, $me, $prefetch) = @_;
990
991   my ($source_handle) = $source;
992
993   if ($source->isa('DBIx::Class::ResultSourceHandle')) {
994       $source = $source_handle->resolve
995   } else {
996       $source_handle = $source->handle
997   }
998
999   my $new = {
1000     _source_handle => $source_handle,
1001     _column_data => $me,
1002     _in_storage => 1
1003   };
1004   bless $new, (ref $class || $class);
1005
1006   my $schema;
1007   foreach my $pre (keys %{$prefetch||{}}) {
1008     my $pre_val = $prefetch->{$pre};
1009     my $pre_source = $source->related_source($pre);
1010     $class->throw_exception("Can't prefetch non-existent relationship ${pre}")
1011       unless $pre_source;
1012     if (ref($pre_val->[0]) eq 'ARRAY') { # multi
1013       my @pre_objects;
1014       foreach my $pre_rec (@$pre_val) {
1015         unless ($pre_source->primary_columns == grep { exists $pre_rec->[0]{$_}
1016            and defined $pre_rec->[0]{$_} } $pre_source->primary_columns) {
1017           next;
1018         }
1019         push(@pre_objects, $pre_source->result_class->inflate_result(
1020                              $pre_source, @{$pre_rec}));
1021       }
1022       $new->related_resultset($pre)->set_cache(\@pre_objects);
1023     } elsif (defined $pre_val->[0]) {
1024       my $fetched;
1025       unless ($pre_source->primary_columns == grep { exists $pre_val->[0]{$_}
1026          and !defined $pre_val->[0]{$_} } $pre_source->primary_columns)
1027       {
1028         $fetched = $pre_source->result_class->inflate_result(
1029                       $pre_source, @{$pre_val});
1030       }
1031       my $accessor = $source->relationship_info($pre)->{attrs}{accessor};
1032       $class->throw_exception("No accessor for prefetched $pre")
1033        unless defined $accessor;
1034       if ($accessor eq 'single') {
1035         $new->{_relationship_data}{$pre} = $fetched;
1036       } elsif ($accessor eq 'filter') {
1037         $new->{_inflated_column}{$pre} = $fetched;
1038       } else {
1039        $class->throw_exception("Prefetch not supported with accessor '$accessor'");
1040       }
1041       $new->related_resultset($pre)->set_cache([ $fetched ]);
1042     }
1043   }
1044   return $new;
1045 }
1046
1047 =head2 update_or_insert
1048
1049   $row->update_or_insert
1050
1051 =over
1052
1053 =item Arguments: none
1054
1055 =item Returns: Result of update or insert operation
1056
1057 =back
1058
1059 L</Update>s the object if it's already in the database, according to
1060 L</in_storage>, else L</insert>s it.
1061
1062 =head2 insert_or_update
1063
1064   $obj->insert_or_update
1065
1066 Alias for L</update_or_insert>
1067
1068 =cut
1069
1070 sub insert_or_update { shift->update_or_insert(@_) }
1071
1072 sub update_or_insert {
1073   my $self = shift;
1074   return ($self->in_storage ? $self->update : $self->insert);
1075 }
1076
1077 =head2 is_changed
1078
1079   my @changed_col_names = $row->is_changed();
1080   if ($row->is_changed()) { ... }
1081
1082 =over
1083
1084 =item Arguments: none
1085
1086 =item Returns: 0|1 or @columnnames
1087
1088 =back
1089
1090 In list context returns a list of columns with uncommited changes, or
1091 in scalar context returns a true value if there are uncommitted
1092 changes.
1093
1094 =cut
1095
1096 sub is_changed {
1097   return keys %{shift->{_dirty_columns} || {}};
1098 }
1099
1100 =head2 is_column_changed
1101
1102   if ($row->is_column_changed('col')) { ... }
1103
1104 =over
1105
1106 =item Arguments: $columname
1107
1108 =item Returns: 0|1
1109
1110 =back
1111
1112 Returns a true value if the column has uncommitted changes.
1113
1114 =cut
1115
1116 sub is_column_changed {
1117   my( $self, $col ) = @_;
1118   return exists $self->{_dirty_columns}->{$col};
1119 }
1120
1121 =head2 result_source
1122
1123   my $resultsource = $row->result_source;
1124
1125 =over
1126
1127 =item Arguments: none
1128
1129 =item Returns: a ResultSource instance
1130
1131 =back
1132
1133 Accessor to the L<DBIx::Class::ResultSource> this object was created from.
1134
1135 =cut
1136
1137 sub result_source {
1138     my $self = shift;
1139
1140     if (@_) {
1141         $self->_source_handle($_[0]->handle);
1142     } else {
1143         $self->_source_handle->resolve;
1144     }
1145 }
1146
1147 =head2 register_column
1148
1149   $column_info = { .... };
1150   $class->register_column($column_name, $column_info);
1151
1152 =over
1153
1154 =item Arguments: $columnname, \%columninfo
1155
1156 =item Returns: undefined
1157
1158 =back
1159
1160 Registers a column on the class. If the column_info has an 'accessor'
1161 key, creates an accessor named after the value if defined; if there is
1162 no such key, creates an accessor with the same name as the column
1163
1164 The column_info attributes are described in
1165 L<DBIx::Class::ResultSource/add_columns>
1166
1167 =cut
1168
1169 sub register_column {
1170   my ($class, $col, $info) = @_;
1171   my $acc = $col;
1172   if (exists $info->{accessor}) {
1173     return unless defined $info->{accessor};
1174     $acc = [ $info->{accessor}, $col ];
1175   }
1176   $class->mk_group_accessors('column' => $acc);
1177 }
1178
1179 =head2 get_from_storage
1180
1181   my $copy = $row->get_from_storage($attrs)
1182
1183 =over
1184
1185 =item Arguments: \%attrs
1186
1187 =item Returns: A Row object
1188
1189 =back
1190
1191 Fetches a fresh copy of the Row object from the database and returns it.
1192
1193 If passed the \%attrs argument, will first apply these attributes to
1194 the resultset used to find the row.
1195
1196 This copy can then be used to compare to an existing row object, to
1197 determine if any changes have been made in the database since it was
1198 created.
1199
1200 To just update your Row object with any latest changes from the
1201 database, use L</discard_changes> instead.
1202
1203 The \%attrs argument should be compatible with
1204 L<DBIx::Class::ResultSet/ATTRIBUTES>.
1205
1206 =cut
1207
1208 sub get_from_storage {
1209     my $self = shift @_;
1210     my $attrs = shift @_;
1211     my $resultset = $self->result_source->resultset;
1212     
1213     if(defined $attrs) {
1214         $resultset = $resultset->search(undef, $attrs);
1215     }
1216     
1217     return $resultset->find($self->{_orig_ident} || $self->ident_condition);
1218 }
1219
1220 =head2 throw_exception
1221
1222 See L<DBIx::Class::Schema/throw_exception>.
1223
1224 =cut
1225
1226 sub throw_exception {
1227   my $self=shift;
1228   if (ref $self && ref $self->result_source && $self->result_source->schema) {
1229     $self->result_source->schema->throw_exception(@_);
1230   } else {
1231     croak(@_);
1232   }
1233 }
1234
1235 =head2 id
1236
1237   my @pk = $row->id;
1238
1239 =over
1240
1241 =item Arguments: none
1242
1243 =item Returns: A list of primary key values
1244
1245 =back
1246
1247 Returns the primary key(s) for a row. Can't be called as a class method.
1248 Actually implemented in L<DBIx::Class::PK>
1249
1250 =head2 discard_changes
1251
1252   $row->discard_changes
1253
1254 =over
1255
1256 =item Arguments: none
1257
1258 =item Returns: nothing (updates object in-place)
1259
1260 =back
1261
1262 Retrieves and sets the row object data from the database, losing any
1263 local changes made.
1264
1265 This method can also be used to refresh from storage, retrieving any
1266 changes made since the row was last read from storage. Actually
1267 implemented in L<DBIx::Class::PK>
1268
1269 =cut
1270
1271 1;
1272
1273 =head1 AUTHORS
1274
1275 Matt S. Trout <mst@shadowcatsystems.co.uk>
1276
1277 =head1 LICENSE
1278
1279 You may distribute this code under the same terms as Perl itself.
1280
1281 =cut