added simple env test
[catagits/HTTP-Request-AsCGI.git] / lib / HTTP / Request / AsCGI.pm
1 package HTTP::Request::AsCGI;
2
3 use strict;
4 use warnings;
5 use base 'Class::Accessor::Fast';
6
7 use Carp;
8 use IO::File;
9
10 __PACKAGE__->mk_accessors( qw[ enviroment request stdin stdout stderr ] );
11
12 our $VERSION = 0.1;
13
14 sub new {
15     my $class   = shift;
16     my $request = shift;
17
18     my $self = {
19         request  => $request,
20         restored => 0,
21         setuped  => 0,
22         stdin    => IO::File->new_tmpfile,
23         stdout   => IO::File->new_tmpfile,
24         stderr   => IO::File->new_tmpfile
25     };
26
27     $self->{enviroment} = {
28         GATEWAY_INTERFACE => 'CGI/1.1',
29         HTTP_HOST         => $request->uri->host_port,
30         QUERY_STRING      => $request->uri->query || '',
31         SCRIPT_NAME       => $request->uri->path || '/',
32         SERVER_NAME       => $request->uri->host,
33         SERVER_PORT       => $request->uri->port,
34         SERVER_PROTOCOL   => $request->protocol || 'HTTP/1.1',
35         SERVER_SOFTWARE   => __PACKAGE__ . "/" . $VERSION,
36         REMOTE_ADDR       => '127.0.0.1',
37         REMOTE_HOST       => 'localhost',
38         REMOTE_PORT       => int( rand(64000) + 1000 ),        # not in RFC 3875
39         REQUEST_URI       => $request->uri->path || '/',       # not in RFC 3875
40         REQUEST_METHOD    => $request->method,
41         @_
42     };
43
44     foreach my $field ( $request->headers->header_field_names ) {
45
46         my $key = uc($field);
47         $key =~ tr/_/-/;
48         $key = 'HTTP_' . $key unless $field =~ /^Content-(Length|Type)$/;
49
50         unless ( exists $self->{enviroment}->{$key} ) {
51             $self->{enviroment}->{$key} = $request->headers->header($field);
52         }
53     }
54
55     return $class->SUPER::new($self);
56 }
57
58 sub setup {
59     my $self = shift;
60
61     open( my $stdin, '>&', STDIN->fileno )
62       or croak("Can't dup stdin: $!");
63
64     open( my $stdout, '>&', STDOUT->fileno )
65       or croak("Can't dup stdout: $!");
66
67     open( my $stderr, '>&', STDERR->fileno )
68       or croak("Can't dup stderr: $!");
69
70     $self->{restore} = {
71         stdin      => $stdin,
72         stdout     => $stdout,
73         stderr     => $stderr,
74         enviroment => {%ENV}
75     };
76
77     if ( $self->request->content_length ) {
78
79         $self->stdin->syswrite( $self->request->content )
80           or croak("Can't write content to stdin: $!");
81
82         $self->stdin->sysseek( 0, SEEK_SET )
83           or croak("Can't seek stdin: $!");
84     }
85
86     %ENV = %{ $self->enviroment };
87
88     open( STDIN, '<&=', $self->stdin->fileno )
89       or croak("Can't open stdin: $!");
90
91     open( STDOUT, '>&=', $self->stdout->fileno )
92       or croak("Can't open stdout: $!");
93
94     open( STDERR, '>&=', $self->stderr->fileno )
95       or croak("Can't open stderr: $!");
96       
97     $self->{setuped}++;
98
99     return $self;
100 }
101
102 sub restore {
103     my $self = shift;
104
105     %ENV = %{ $self->{restore}->{enviroment} };
106
107     open( STDIN, '>&', $self->{restore}->{stdin} )
108       or croak("Can't restore stdin: $!");
109
110     open( STDOUT, '>&', $self->{restore}->{stdout} )
111       or croak("Can't restore stdout: $!");
112
113     open( STDERR, '>&', $self->{restore}->{stderr} )
114       or croak("Can't restore stderr: $!");
115
116     if ( $self->stdin->fileno != STDIN->fileno ) {
117         $self->stdin->sysseek( 0, SEEK_SET )
118           or croak("Can't seek stdin: $!");
119     }
120
121     if ( $self->stdout->fileno != STDOUT->fileno ) {
122         $self->stdout->sysseek( 0, SEEK_SET )
123           or croak("Can't seek stdout: $!");
124     }
125
126     if ( $self->stderr->fileno != STDERR->fileno ) {
127         $self->stderr->sysseek( 0, SEEK_SET )
128           or croak("Can't seek stderr: $!");
129     }
130
131     $self->{restored}++;
132 }
133
134 sub DESTROY {
135     my $self = shift;
136     $self->restore if $self->{setuped} && !$self->{restored};
137 }
138
139 1;
140
141 __END__
142
143 =head1 NAME
144
145 HTTP::Request::AsCGI - Setup a CGI enviroment from a HTTP::Request
146
147 =head1 SYNOPSIS
148
149     use CGI;
150     use HTTP::Request;
151     use HTTP::Request::AsCGI;
152     
153     my $request = HTTP::Request->new( GET => 'http://www.host.com/' );
154     my $stdout;
155     
156     {
157         my $c = HTTP::Request::AsCGI->new($request)->setup;
158         my $q = CGI->new;
159         
160         print $q->header,
161               $q->start_html('Hello World'),
162               $q->h1('Hello World'),
163               $q->end_html;
164         
165         $stdout = $c->stdout;
166         
167         # enviroment and descriptors will automatically be restored when $c is destructed.
168     }
169     
170     while ( my $line = $stdout->getline ) {
171         print $line;
172     }
173     
174 =head1 DESCRIPTION
175
176 =head1 METHODS
177
178 =over 4 
179
180 =item new
181
182 =item enviroment
183
184 =item setup
185
186 =item restore
187
188 =item request
189
190 =item stdin
191
192 =item stdout
193
194 =item stderr
195
196 =back
197
198 =head1 BUGS
199
200 =head1 AUTHOR
201
202 Christian Hansen, C<ch@ngmedia.com>
203
204 =head1 LICENSE
205
206 This library is free software. You can redistribute it and/or modify 
207 it under the same terms as perl itself.
208
209 =cut