]> git.decadent.org.uk Git - maypole.git/blobdiff - lib/Maypole/Model/CDBI.pm
Just in case.
[maypole.git] / lib / Maypole / Model / CDBI.pm
index 2d2aa2b123a35c9d19ffe944eaa2acc949eff64b..c7621d2c493491c33415dd7f021940395687e13d 100644 (file)
@@ -1,8 +1,9 @@
-package Apache::MVC::Model::CDBI;
-use base qw(Apache::MVC::Model::Base Class::DBI);
+package Maypole::Model::CDBI;
+use base qw(Maypole::Model::Base Class::DBI);
 use Lingua::EN::Inflect::Number qw(to_PL);
 use Class::DBI::AsForm;
 use Class::DBI::FromCGI;
+use Class::DBI::Loader;
 use Class::DBI::AbstractSearch;
 use Class::DBI::Plugin::RetrieveAll;
 use Class::DBI::Pager;
@@ -11,7 +12,7 @@ use strict;
 
 =head1 NAME
 
-Apache::MVC::Model::CDBI - Model class based on Class::DBI
+Maypole::Model::CDBI - Model class based on Class::DBI
 
 =head1 DESCRIPTION
 
@@ -64,11 +65,14 @@ sub delete :Exported {
 sub adopt {
     my ($self, $child) = @_;
     $child->autoupdate(1);
-    $child->columns( Stringify => qw/ name / );
+    if (grep { $_ eq "name" } $child->columns) { # Common case
+        $child->columns( Stringify => qw/ name / );
+    } # Otherwise, work it out for yourself.
 }
 
 sub search :Exported {
-    return shift->SUPER::search(@_) if caller eq "Class::DBI"; # oops
+    return shift->SUPER::search(@_) if caller ne "Maypole::Model::Base";
+                                    # A real CDBI search.
     my ($self, $r) = @_;
     my %fields = map {$_ => 1 } $self->columns;
     my $oper = "like"; # For now
@@ -98,5 +102,24 @@ sub list :Exported {
         $r->objects([ $self->retrieve_all ]);
     }
 }
+
+sub setup_database {
+    my ($self, $config, $namespace, $dsn, $u, $p) = @_;
+    $config->{dsn} = $dsn;
+    $config->{loader} = Class::DBI::Loader->new(
+        namespace => $namespace,
+        dsn => $dsn,
+        user => $u,
+        password => $p,
+    );
+    $config->{classes} = [ $config->{loader}->classes ];
+    $config->{tables}  = [ $config->{loader}->tables ];
+}
+
+sub class_of {
+    my ($self, $r, $table) = @_;
+    return $r->config->{loader}->_table2class($table);
+}
+
 1;