X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FCatalyst%2FEngine.pm;fp=lib%2FCatalyst%2FEngine.pm;h=443975ed05704b2fd544a80ddb8f0f1584a94f14;hb=4a41d5d1ec3187cc41e15767b21c14b2aee31740;hp=0000000000000000000000000000000000000000;hpb=2757db2c7c600c8a0b8e2b4366f38c97804c2844;p=catagits%2FCatalyst-Runtime.git diff --git a/lib/Catalyst/Engine.pm b/lib/Catalyst/Engine.pm new file mode 100644 index 0000000..443975e --- /dev/null +++ b/lib/Catalyst/Engine.pm @@ -0,0 +1,706 @@ +package Catalyst::Engine; + +use Moose; +with 'MooseX::Emulate::Class::Accessor::Fast'; + +use CGI::Simple::Cookie; +use Data::Dump qw/dump/; +use Errno 'EWOULDBLOCK'; +use HTML::Entities; +use HTTP::Body; +use HTTP::Headers; +use URI::QueryParam; + +use namespace::clean -except => 'meta'; + +has env => (is => 'rw'); + +# input position and length +has read_length => (is => 'rw'); +has read_position => (is => 'rw'); + +has _prepared_write => (is => 'rw'); + +# Amount of data to read from input on each pass +our $CHUNKSIZE = 64 * 1024; + +=head1 NAME + +Catalyst::Engine - The Catalyst Engine + +=head1 SYNOPSIS + +See L. + +=head1 DESCRIPTION + +=head1 METHODS + + +=head2 $self->finalize_body($c) + +Finalize body. Prints the response output. + +=cut + +sub finalize_body { + my ( $self, $c ) = @_; + my $body = $c->response->body; + no warnings 'uninitialized'; + if ( blessed($body) && $body->can('read') or ref($body) eq 'GLOB' ) { + my $got; + do { + $got = read $body, my ($buffer), $CHUNKSIZE; + $got = 0 unless $self->write( $c, $buffer ); + } while $got > 0; + + close $body; + } + else { + $self->write( $c, $body ); + } +} + +=head2 $self->finalize_cookies($c) + +Create CGI::Simple::Cookie objects from $c->res->cookies, and set them as +response headers. + +=cut + +sub finalize_cookies { + my ( $self, $c ) = @_; + + my @cookies; + my $response = $c->response; + + foreach my $name (keys %{ $response->cookies }) { + + my $val = $response->cookies->{$name}; + + my $cookie = ( + blessed($val) + ? $val + : CGI::Simple::Cookie->new( + -name => $name, + -value => $val->{value}, + -expires => $val->{expires}, + -domain => $val->{domain}, + -path => $val->{path}, + -secure => $val->{secure} || 0, + -httponly => $val->{httponly} || 0, + ) + ); + + push @cookies, $cookie->as_string; + } + + for my $cookie (@cookies) { + $response->headers->push_header( 'Set-Cookie' => $cookie ); + } +} + +=head2 $self->finalize_error($c) + +Output an appropriate error message. Called if there's an error in $c +after the dispatch has finished. Will output debug messages if Catalyst +is in debug mode, or a `please come back later` message otherwise. + +=cut + +sub finalize_error { + my ( $self, $c ) = @_; + + $c->res->content_type('text/html; charset=utf-8'); + my $name = ref($c)->config->{name} || join(' ', split('::', ref $c)); + + my ( $title, $error, $infos ); + if ( $c->debug ) { + + # For pretty dumps + $error = join '', map { + '

' + . encode_entities($_) + . '

' + } @{ $c->error }; + $error ||= 'No output'; + $error = qq{
$error
}; + $title = $name = "$name on Catalyst $Catalyst::VERSION"; + $name = "

$name

"; + + # Don't show context in the dump + $c->req->_clear_context; + $c->res->_clear_context; + + # Don't show body parser in the dump + $c->req->_clear_body; + + my @infos; + my $i = 0; + for my $dump ( $c->dump_these ) { + my $name = $dump->[0]; + my $value = encode_entities( dump( $dump->[1] )); + push @infos, sprintf <<"EOF", $name, $value; +

%s

+
+
%s
+
+EOF + $i++; + } + $infos = join "\n", @infos; + } + else { + $title = $name; + $error = ''; + $infos = <<""; +
+(en) Please come back later
+(fr) SVP veuillez revenir plus tard
+(de) Bitte versuchen sie es spaeter nocheinmal
+(at) Konnten's bitt'schoen spaeter nochmal reinschauen
+(no) Vennligst prov igjen senere
+(dk) Venligst prov igen senere
+(pl) Prosze sprobowac pozniej
+(pt) Por favor volte mais tarde
+(ru) Попробуйте еще раз позже
+(ua) Спробуйте ще раз пізніше
+
+ + $name = ''; + } + $c->res->body( <<"" ); + + + + + + $title + + + + +
+
$error
+
$infos
+
$name
+
+ + + + + # Trick IE + $c->res->{body} .= ( ' ' x 512 ); + + # Return 500 + $c->res->status(500); +} + +=head2 $self->finalize_headers($c) + +Abstract method, allows engines to write headers to response + +=cut + +sub finalize_headers { } + +=head2 $self->finalize_read($c) + +=cut + +sub finalize_read { } + +=head2 $self->finalize_uploads($c) + +Clean up after uploads, deleting temp files. + +=cut + +sub finalize_uploads { + my ( $self, $c ) = @_; + + my $request = $c->request; + foreach my $key (keys %{ $request->uploads }) { + my $upload = $request->uploads->{$key}; + unlink grep { -e $_ } map { $_->tempname } + (ref $upload eq 'ARRAY' ? @{$upload} : ($upload)); + } + +} + +=head2 $self->prepare_body($c) + +sets up the L object body using L + +=cut + +sub prepare_body { + my ( $self, $c ) = @_; + + my $appclass = ref($c) || $c; + if ( my $length = $self->read_length ) { + my $request = $c->request; + unless ( $request->_body ) { + my $type = $request->header('Content-Type'); + $request->_body(HTTP::Body->new( $type, $length )); + $request->_body->tmpdir( $appclass->config->{uploadtmp} ) + if exists $appclass->config->{uploadtmp}; + } + + while ( my $buffer = $self->read($c) ) { + $c->prepare_body_chunk($buffer); + } + + # paranoia against wrong Content-Length header + my $remaining = $length - $self->read_position; + if ( $remaining > 0 ) { + $self->finalize_read($c); + Catalyst::Exception->throw( + "Wrong Content-Length value: $length" ); + } + } + else { + # Defined but will cause all body code to be skipped + $c->request->_body(0); + } +} + +=head2 $self->prepare_body_chunk($c) + +Add a chunk to the request body. + +=cut + +sub prepare_body_chunk { + my ( $self, $c, $chunk ) = @_; + + $c->request->_body->add($chunk); +} + +=head2 $self->prepare_body_parameters($c) + +Sets up parameters from body. + +=cut + +sub prepare_body_parameters { + my ( $self, $c ) = @_; + + return unless $c->request->_body; + + $c->request->body_parameters( $c->request->_body->param ); +} + +=head2 $self->prepare_connection($c) + +Abstract method implemented in engines. + +=cut + +sub prepare_connection { } + +=head2 $self->prepare_cookies($c) + +Parse cookies from header. Sets a L object. + +=cut + +sub prepare_cookies { + my ( $self, $c ) = @_; + + if ( my $header = $c->request->header('Cookie') ) { + $c->req->cookies( { CGI::Simple::Cookie->parse($header) } ); + } +} + +=head2 $self->prepare_headers($c) + +=cut + +sub prepare_headers { } + +=head2 $self->prepare_parameters($c) + +sets up parameters from query and post parameters. + +=cut + +sub prepare_parameters { + my ( $self, $c ) = @_; + + my $request = $c->request; + my $parameters = $request->parameters; + my $body_parameters = $request->body_parameters; + my $query_parameters = $request->query_parameters; + # We copy, no references + foreach my $name (keys %$query_parameters) { + my $param = $query_parameters->{$name}; + $parameters->{$name} = ref $param eq 'ARRAY' ? [ @$param ] : $param; + } + + # Merge query and body parameters + foreach my $name (keys %$body_parameters) { + my $param = $body_parameters->{$name}; + my @values = ref $param eq 'ARRAY' ? @$param : ($param); + if ( my $existing = $parameters->{$name} ) { + unshift(@values, (ref $existing eq 'ARRAY' ? @$existing : $existing)); + } + $parameters->{$name} = @values > 1 ? \@values : $values[0]; + } +} + +=head2 $self->prepare_path($c) + +abstract method, implemented by engines. + +=cut + +sub prepare_path { } + +=head2 $self->prepare_request($c) + +=head2 $self->prepare_query_parameters($c) + +process the query string and extract query parameters. + +=cut + +sub prepare_query_parameters { + my ( $self, $c, $query_string ) = @_; + + # Check for keywords (no = signs) + # (yes, index() is faster than a regex :)) + if ( index( $query_string, '=' ) < 0 ) { + $c->request->query_keywords( $self->unescape_uri($query_string) ); + return; + } + + my %query; + + # replace semi-colons + $query_string =~ s/;/&/g; + + my @params = grep { length $_ } split /&/, $query_string; + + for my $item ( @params ) { + + my ($param, $value) + = map { $self->unescape_uri($_) } + split( /=/, $item, 2 ); + + $param = $self->unescape_uri($item) unless defined $param; + + if ( exists $query{$param} ) { + if ( ref $query{$param} ) { + push @{ $query{$param} }, $value; + } + else { + $query{$param} = [ $query{$param}, $value ]; + } + } + else { + $query{$param} = $value; + } + } + + $c->request->query_parameters( \%query ); +} + +=head2 $self->prepare_read($c) + +prepare to read from the engine. + +=cut + +sub prepare_read { + my ( $self, $c ) = @_; + + # Initialize the read position + $self->read_position(0); + + # Initialize the amount of data we think we need to read + $self->read_length( $c->request->header('Content-Length') || 0 ); +} + +=head2 $self->prepare_request(@arguments) + +Populate the context object from the request object. + +=cut + +sub prepare_request { } + +=head2 $self->prepare_uploads($c) + +=cut + +sub prepare_uploads { + my ( $self, $c ) = @_; + + my $request = $c->request; + return unless $request->_body; + + my $uploads = $request->_body->upload; + my $parameters = $request->parameters; + foreach my $name (keys %$uploads) { + my $files = $uploads->{$name}; + my @uploads; + for my $upload (ref $files eq 'ARRAY' ? @$files : ($files)) { + my $headers = HTTP::Headers->new( %{ $upload->{headers} } ); + my $u = Catalyst::Request::Upload->new + ( + size => $upload->{size}, + type => $headers->content_type, + headers => $headers, + tempname => $upload->{tempname}, + filename => $upload->{filename}, + ); + push @uploads, $u; + } + $request->uploads->{$name} = @uploads > 1 ? \@uploads : $uploads[0]; + + # support access to the filename as a normal param + my @filenames = map { $_->{filename} } @uploads; + # append, if there's already params with this name + if (exists $parameters->{$name}) { + if (ref $parameters->{$name} eq 'ARRAY') { + push @{ $parameters->{$name} }, @filenames; + } + else { + $parameters->{$name} = [ $parameters->{$name}, @filenames ]; + } + } + else { + $parameters->{$name} = @filenames > 1 ? \@filenames : $filenames[0]; + } + } +} + +=head2 $self->prepare_write($c) + +Abstract method. Implemented by the engines. + +=cut + +sub prepare_write { } + +=head2 $self->read($c, [$maxlength]) + +=cut + +sub read { + my ( $self, $c, $maxlength ) = @_; + + my $remaining = $self->read_length - $self->read_position; + $maxlength ||= $CHUNKSIZE; + + # Are we done reading? + if ( $remaining <= 0 ) { + $self->finalize_read($c); + return; + } + + my $readlen = ( $remaining > $maxlength ) ? $maxlength : $remaining; + my $rc = $self->read_chunk( $c, my $buffer, $readlen ); + if ( defined $rc ) { + $self->read_position( $self->read_position + $rc ); + return $buffer; + } + else { + Catalyst::Exception->throw( + message => "Unknown error reading input: $!" ); + } +} + +=head2 $self->read_chunk($c, $buffer, $length) + +Each engine implements read_chunk as its preferred way of reading a chunk +of data. + +=cut + +sub read_chunk { } + +=head2 $self->read_length + +The length of input data to be read. This is obtained from the Content-Length +header. + +=head2 $self->read_position + +The amount of input data that has already been read. + +=head2 $self->run($c) + +Start the engine. Implemented by the various engine classes. + +=cut + +sub run { } + +=head2 $self->write($c, $buffer) + +Writes the buffer to the client. + +=cut + +sub write { + my ( $self, $c, $buffer ) = @_; + + unless ( $self->_prepared_write ) { + $self->prepare_write($c); + $self->_prepared_write(1); + } + + return 0 if !defined $buffer; + + my $len = length($buffer); + my $wrote = syswrite STDOUT, $buffer; + + if ( !defined $wrote && $! == EWOULDBLOCK ) { + # Unable to write on the first try, will retry in the loop below + $wrote = 0; + } + + if ( defined $wrote && $wrote < $len ) { + # We didn't write the whole buffer + while (1) { + my $ret = syswrite STDOUT, $buffer, $CHUNKSIZE, $wrote; + if ( defined $ret ) { + $wrote += $ret; + } + else { + next if $! == EWOULDBLOCK; + return; + } + + last if $wrote >= $len; + } + } + + return $wrote; +} + +=head2 $self->unescape_uri($uri) + +Unescapes a given URI using the most efficient method available. Engines such +as Apache may implement this using Apache's C-based modules, for example. + +=cut + +sub unescape_uri { + my ( $self, $str ) = @_; + + $str =~ s/(?:%([0-9A-Fa-f]{2})|\+)/defined $1 ? chr(hex($1)) : ' '/eg; + + return $str; +} + +=head2 $self->finalize_output + +, see finalize_body + +=head2 $self->env + +Hash containing enviroment variables including many special variables inserted +by WWW server - like SERVER_*, REMOTE_*, HTTP_* ... + +Before accesing enviroment variables consider whether the same information is +not directly available via Catalyst objects $c->request, $c->engine ... + +BEWARE: If you really need to access some enviroment variable from your Catalyst +application you should use $c->engine->env->{VARNAME} instead of $ENV{VARNAME}, +as in some enviroments the %ENV hash does not contain what you would expect. + +=head1 AUTHORS + +Catalyst Contributors, see Catalyst.pm + +=head1 COPYRIGHT + +This library is free software. You can redistribute it and/or modify it under +the same terms as Perl itself. + +=cut + +1;