X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FCatalyst%2FEngine.pm;h=691bd675e00ba5701cffa665f0fed7b9db0ff5b5;hb=a2f2cde95194a17fe2401ae58c92b5494bac599f;hp=a42eab906161b3eae8f9674fb1bb20145233e4fd;hpb=a76636937b5a8c129a012f96a704c7d6983e04a9;p=catagits%2FCatalyst-Runtime.git diff --git a/lib/Catalyst/Engine.pm b/lib/Catalyst/Engine.pm index a42eab9..691bd67 100644 --- a/lib/Catalyst/Engine.pm +++ b/lib/Catalyst/Engine.pm @@ -10,9 +10,11 @@ use HTML::Entities; use HTTP::Headers; use Time::HiRes qw/gettimeofday tv_interval/; use Text::ASCIITable; +use Catalyst::Exception; use Catalyst::Request; use Catalyst::Request::Upload; use Catalyst::Response; +use Catalyst::Utils; require Module::Pluggable::Fast; @@ -20,7 +22,7 @@ require Module::Pluggable::Fast; $Data::Dumper::Terse = 1; __PACKAGE__->mk_classdata('components'); -__PACKAGE__->mk_accessors(qw/counter request response state/); +__PACKAGE__->mk_accessors(qw/counter depth request response state/); *comp = \&component; *req = \&request; @@ -33,6 +35,7 @@ __PACKAGE__->mk_accessors(qw/counter request response state/); our $COUNT = 1; our $START = time; our $RECURSION = 1000; +our $DETACH = "catalyst_detach\n"; =head1 NAME @@ -106,6 +109,10 @@ sub component { Returns a hashref containing coderefs and execution counts. (Needed for deep recursion detection) +=item $c->depth + +Returns the actual forward depth. + =item $c->error =item $c->error($error, ...) @@ -159,26 +166,32 @@ sub execute { $action = "-> $action" if $callsub =~ /forward$/; } + $c->{depth}++; eval { if ( $c->debug ) { - my ( $elapsed, @state ) = $c->benchmark( $code, $class, $c, @{ $c->req->args } ); + my ( $elapsed, @state ) = + $c->benchmark( $code, $class, $c, @{ $c->req->args } ); push @{ $c->{stats} }, [ $action, sprintf( '%fs', $elapsed ) ]; $c->state(@state); } - else { $c->state( &$code( $class, $c, @{ $c->req->args } ) ? 1 : 0 ) } + else { $c->state( &$code( $class, $c, @{ $c->req->args } ) || 0 ) } }; + $c->{depth}--; if ( my $error = $@ ) { - unless ( ref $error ) { - chomp $error; - $error = qq/Caught exception "$error"/; - } + if ( $error eq $DETACH ) { die $DETACH if $c->{depth} > 1 } + else { + unless ( ref $error ) { + chomp $error; + $error = qq/Caught exception "$error"/; + } - $c->log->error($error); - $c->error($error); - $c->state(0); + $c->log->error($error); + $c->error($error); + $c->state(0); + } } return $c->state; } @@ -204,13 +217,21 @@ sub finalize { $c->finalize_error; } - if ( !$c->response->body && $c->response->status !~ /^(1|3)\d\d$/ ) { + if ( !$c->response->body && $c->response->status == 200 ) { $c->finalize_error; } if ( $c->response->body && !$c->response->content_length ) { - use bytes; # play safe with a utf8 aware perl - $c->response->content_length( length $c->response->body ); + $c->response->content_length( bytes::length( $c->response->body ) ); + } + + if ( $c->response->status =~ /^(1\d\d|[23]04)$/ ) { + $c->response->headers->remove_header("Content-Length"); + $c->response->body(''); + } + + if ( $c->request->method eq 'HEAD' ) { + $c->response->body(''); } my $status = $c->finalize_headers; @@ -425,6 +446,7 @@ sub prepare { my $c = bless { counter => {}, + depth => 0, request => Catalyst::Request->new( { arguments => [], @@ -440,7 +462,7 @@ sub prepare { { body => '', cookies => {}, - headers => HTTP::Headers->new, + headers => HTTP::Headers->new( 'Content-Length' => 0 ), status => 200 } ), @@ -464,9 +486,9 @@ sub prepare { $c->prepare_path; $c->prepare_action; - my $method = $c->req->method || ''; - my $path = $c->req->path || ''; - my $address = $c->req->address || ''; + my $method = $c->req->method || ''; + my $path = $c->req->path || ''; + my $address = $c->req->address || ''; $c->log->debug(qq/"$method" request for "$path" from $address/) if $c->debug; @@ -659,7 +681,28 @@ Setup. sub setup { my $self = shift; + + # Initialize our data structure + $self->components( {} ); + $self->setup_components; + + if ( $self->debug ) { + my $t = Text::ASCIITable->new; + $t->setOptions( 'hide_HeadRow', 1 ); + $t->setOptions( 'hide_HeadLine', 1 ); + $t->setCols('Class'); + $t->setColWidth( 'Class', 75, 1 ); + $t->addRow($_) for sort keys %{ $self->components }; + $self->log->debug( 'Loaded components', $t->draw ) + if ( @{ $t->{tbl_rows} } ); + } + + # Add our self to components, since we are also a component + $self->components->{$self} = $self; + + $self->setup_actions; + if ( $self->debug ) { my $name = $self->config->{name} || 'Application'; $self->log->info("$name powered by Catalyst $Catalyst::VERSION"); @@ -674,39 +717,57 @@ Setup components. sub setup_components { my $self = shift; - - # Components - my $class = ref $self || $self; - eval <<""; - package $class; - import Module::Pluggable::Fast - name => '_components', - search => [ - '$class\::Controller', '$class\::C', - '$class\::Model', '$class\::M', - '$class\::View', '$class\::V' - ]; + + my $callback = sub { + my ( $component, $context ) = @_; + + unless ( $component->isa('Catalyst::Base') ) { + return $component; + } + + my $suffix = Catalyst::Utils::class2classsuffix($component); + my $config = $self->config->{$suffix} || {}; + + my $instance; + + eval { $instance = $component->new( $context, $config ); }; + + if ( my $error = $@ ) { + + chomp $error; + + Catalyst::Exception->throw( + message => qq/Couldn't instantiate component "$component", "$error"/ + ); + } + + return $instance; + }; + + eval { + Module::Pluggable::Fast->import( + name => '_components', + search => [ + "$self\::Controller", "$self\::C", + "$self\::Model", "$self\::M", + "$self\::View", "$self\::V" + ], + callback => $callback + ); + }; if ( my $error = $@ ) { + chomp $error; - die qq/Couldn't load components "$error"/; + + Catalyst::Exception->throw( + message => qq/Couldn't load components "$error"/ + ); } - $self->components( {} ); - my @comps; - for my $comp ( $self->_components($self) ) { - $self->components->{ ref $comp } = $comp; - push @comps, $comp; + for my $component ( $self->_components($self) ) { + $self->components->{ ref $component || $component } = $component; } - - my $t = Text::ASCIITable->new( { hide_HeadRow => 1, hide_HeadLine => 1 } ); - $t->setCols('Class'); - $t->setColWidth( 'Class', 75, 1 ); - $t->addRow($_) for sort keys %{ $self->components }; - $self->log->debug( 'Loaded components', $t->draw ) - if ( @{ $t->{tbl_rows} } && $self->debug ); - - $self->setup_actions( [ $self, @comps ] ); } =item $c->state