X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=lib%2FApache%2FMVC%2FModel%2FCDBI.pm;h=833550d5a059a0297973069bb144d80adb89685d;hb=21e0e0c44d23d0d1f0612e4d009c83d010bfb50c;hp=c06703a1fb433d0d6262df880670f28bfa62fc89;hpb=58398cbb7b75bf3106e8503b81ee782178c248c3;p=maypole.git

diff --git a/lib/Apache/MVC/Model/CDBI.pm b/lib/Apache/MVC/Model/CDBI.pm
index c06703a..833550d 100644
--- a/lib/Apache/MVC/Model/CDBI.pm
+++ b/lib/Apache/MVC/Model/CDBI.pm
@@ -2,9 +2,52 @@ package Apache::MVC::Model::CDBI;
 use base 'Apache::MVC::Model::Base';
 use Class::DBI::AsForm;
 use Class::DBI::FromCGI;
+use CGI::Untaint;
 
 sub description { "A poorly defined class" }
 
 sub column_names { my $class = shift; map { $_ => ucfirst $_ } $class->columns }
 
+sub get_objects {
+    my ($self, $r) = @_;
+    return $self->retrieve(shift @{$r->{args}});
+}
+
+sub do_edit :Exported {
+    my ($self, $r) = @_;
+    my $h = CGI::Untaint->new(%{$r->{params}});
+    my $obj;
+    if (@{$r->{args}}) {
+        # We have something to edit
+        ($obj) = @{$self->objects};
+        $obj->update_from_cgi($h);
+        warn "Updating an object ($obj) with ".Dumper($h); use Data::Dumper;
+    } else {
+        $obj = $self->create_from_cgi($h);
+    }
+    if (my %errors = $obj->cgi_update_errors) {
+        # Set it up as it was:
+        warn "There were errors: ".Dumper(\%errors)."\n";
+        $r->{template_args}{cgi_params} = \%params;
+        $r->{template_args}{errors} = \%errors;
+        $r->{template} = "edit";
+    } else {
+        $r->{template} = "view";
+    }
+    return $obj;
+}
+
+sub delete :Exported {
+    my ($self, $r) = @_;
+    $self->delete for @{ $r->objects };
+    $r->objects([ $self->retrieve_all ]);
+    $r->{template} = "list";
+}
+
+sub adopt {
+    my ($self, $child) = @_;
+    $child->autoupdate(1);
+    $child->columns( Stringify => qw/ name / );
+}
+
 1;