X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=lib%2FApache%2FMVC%2FView%2FTT.pm;h=babd0f457a9260735ccd438eb88c5b2ce2c00837;hb=4c4e58eb02155a43397f30b900c5b30f755cb874;hp=90624ad7960113098dcacd635dbfe6bddb9de647;hpb=21e0e0c44d23d0d1f0612e4d009c83d010bfb50c;p=maypole.git

diff --git a/lib/Apache/MVC/View/TT.pm b/lib/Apache/MVC/View/TT.pm
index 90624ad..babd0f4 100644
--- a/lib/Apache/MVC/View/TT.pm
+++ b/lib/Apache/MVC/View/TT.pm
@@ -1,4 +1,5 @@
 package Apache::MVC::View::TT;
+use Apache::Constants;
 use Lingua::EN::Inflect;
 use Template;
 use File::Spec;
@@ -10,10 +11,11 @@ sub new { bless {}, shift } # Not worth having
 
 sub _tt {
     my ($self, $r) = @_;
-    my $root = $r->config->{template_root};
+    my $root = $r->{ar}->document_root . "/". $r->{ar}->location;
+    warn "Root was $root";
     Template->new({ INCLUDE_PATH => [
         $root,
-        File::Spec->catdir($root, $r->model_class->moniker),
+        ($r->model_class && File::Spec->catdir($root, $r->model_class->moniker)),
         File::Spec->catdir($root, "custom"),
         File::Spec->catdir($root, "factory")
     ]});
@@ -24,27 +26,29 @@ sub _args {
     my $class = $r->model_class;
     my %args = (
         request => $r,
-        class   => $class,
         objects => $r->objects,
         base    => $r->config->{uri_base},
         config  => $r->config
         # ...
-    );
-    $args{classmetadata} = {
-        name => $class,
-        columns => [ $class->columns ],
-        colnames => { $class->column_names },
-        moniker => $class->moniker,
-        plural  => $class->plural_moniker,
-        cgi => { $class->to_cgi },
-        description => $class->description
-    };
+    ) ;
+    if ($class) { 
+        $args{classmetadata} = {
+            name => $class,
+            columns => [ $class->columns ],
+            colnames => { $class->column_names },
+            related_accessors => [ $class->related($r) ],
+            moniker => $class->moniker,
+            plural  => $class->plural_moniker,
+            cgi => { $class->to_cgi },
+            description => $class->description
+        };
 
-    # User-friendliness facility for custom template writers.
-    if (@{$r->objects} > 1){
-        $args{$r->model_class->plural_moniker} = $r->objects;
-    } else {
-        ($args{$r->model_class->moniker}) = @{$r->objects};
+        # User-friendliness facility for custom template writers.
+        if (@{$r->objects || []} > 1) { 
+            $args{$r->model_class->plural_moniker} = $r->objects;
+        } else {
+            ($args{$r->model_class->moniker}) = @{$r->objects};
+        }
     }
 
     # Overrides
@@ -57,7 +61,8 @@ sub process {
     my $template = $self->_tt($r);
     my $output;
     $template->process($r->template, { $self->_args($r) }, \$output)
-    || $self->error($r, $template->error);
+    || return $self->error($r, $template->error);
+
     $r->{ar}->content_type("text/html");
     $r->{ar}->headers_out->set("Content-Length" => length $output);
     $r->{ar}->send_http_header;
@@ -67,7 +72,11 @@ sub process {
 
 sub error {
     my ($self, $r, $error) = @_;
+    warn $error;
+    if ($error =~ /not found$/) { return DECLINED }
     $r->{ar}->send_http_header("text/plain");
     $r->{ar}->print($error);
     exit;
 }
+
+1;